Python Fastai-CUDA错误：触发设备端断言

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

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

在CPU笔记本中，我遇到以下错误：

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

在GPU笔记本中，我遇到以下错误：

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

---

请看一看，并向我建议如何解决此问题。

错误在于您的数据集中有255个，但您的模型没有那么多的输出。所以，要么清理数据集以删除无效类，要么修改模型（如果该类是有效类）。您好@akshayk07，谢谢，但我是新手。您是否可以查看一下代码并告诉我如何修复它？