Pytorch CUDA ERROR: device-side assert triggered at

遇到  CUDA ERROR: device-side assert triggered at  這個問題，我出現的原因目前有2種

1. 模型預測中包含了label不存在的類别，預測中部分預測類别超出了label中最大類别

2. 交叉熵函數，傳入的模型預測tensor并不是N * Num_Class * H * W     (Num_Class: 類别數量)