Pytorch 所遇BUG彙總---持續更新

BUG_1

RuntimeError: cuda runtime error (59) : device-side assert triggered at C:/w/1/s/tmp_conda_3.5_160740/conda/conda-bld/pytorch_1570810133449/work/aten/src\THC/generic/THCTensorMath.cu:26
           

問題原因：再用pytorch時進行損失值計算時，标簽值應該再0-class_size-1内，是以标簽值不能大于标簽的個數。

解決方法：

在資料集類裡面的

def __getitem__(self, index):

函數中裡面添加label處理，将label限制再範圍内。比如我的标簽是1-18，是以我再處理的時直接

label=label-1

，就可以了。