errorの内容

pytorchでCrossEntropyLossをlossの計算に使った場合errorが出る。

RuntimeError: CUDA error: an illegal memory access was encountered

原因

pytorchのimplementationミスだそうです。 labelが[0 ~ NUM_CLASS-1]の範囲を超える事によるerror 今回はクラスに分類されない物体を255と設定していたのでエラーとなりました。

・labelを[0 ~ NUM_CLASS-1]の中で設定する

or

・pytorch 1.5.1以上にupdateする

https://discuss.pytorch.org/t/i-get-runtimeerror-cuda-error-an-illegal-memory-access-was-encountered/88118