1. 程式人生 > >Pytorch BUG 持續填坑

Pytorch BUG 持續填坑

BUG1

在做多分類的時候,計算LOSS 報錯:cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-bld/pytorch_1513368888240/work/torch/lib/THC/generated/../THCReduceAll.cuh:33

這是因為 在計算loss的時候 data label 定義出了問題,在用到在用log_softmax +cross_entropy 在做多分類的時候: 損失函式的輸入如下:label 序號應該從0 開始 定義。

  • input – Variable (
    N,C)
    (N,C)
     where C = number of classes
  • target – Variable (N)(N) where each value is 0 <= targets[i] <= C-1

解決辦法

將目標label 改為從0 開始[1,2,3..20]->[0,1,2...19]

-----------------------------------------------------------------------------------