训练nnUNet CUDA error: device-side assert triggered 报错

在使用pytorch训练网络时，有可能会出现这个错误。

看了网上一些解决方案，确实是因为数据标签问题导致的。

我的情况是使用nnUNet时报错的，比如我在标签描述中写了三个类，可是输入的数据标签可能有四类或者五类，这样很明显输入数据与类别不一致。

解决方法：在输入数据中确定标签类别，保持一致。