山下誠二郎のITとフォートナイトと野球ブログ
id:electric-city
【Pytorch】CUDA error: device-side assert triggered で発生したエラー
エラー CUDA error: device-side assert triggered ビビりました。 単刀直入によるとネットワークに入力したデータの形式が間違っている場合に、 発生するエラーだそうです。 やろうとしたこと CIFAR10の画像から2クラスのみを取得して、二値分類を試そうとした。 調査 Jupyter Notebookで一行ずつ実装していたのですが、 …