Fastai CUDA错误:已触发deviceside断言

2024-04-27 17:52:02 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用Fastai对Kaggle数据集进行图像分割-TGS盐识别挑战。我用colab在CPU和GPU上都试过这段代码

  • 使用CPU链接到笔记本文件:bit.ly/2izomx\u TGS\u CPU
  • 使用GPU链接到笔记本文件:bit.ly/2Vq0ROZ\u TGS\u GPU

在CPU笔记本中,我遇到以下错误: Target 255 is out of bounds

我在open_mask中设置了div=True,还设置了num_workers=0

在GPU笔记本中,我遇到以下错误: CUDA error: device-side assert triggered

我已采用借据的自定义度量。 Fastai版本:1.0.60

请看一看,并建议我如何解决这个问题


Tags: 文件数据代码图像gpu链接错误ly