この記事は， Pythonを利用して研究を行なっていく中で私がつまずいてしまったポイントをまとめていくものです。同じような状況で苦しんでいる方々の参考になれば嬉しいです。Pythonつまずきポイント集の目次は以下のページをご覧ください。
【超初心者お悩み解決】Pythonつまずきポイント記事まとめページ
この記事は，Pythonを利用して研究を行なっていく中で私がつまずいてしまったポイントをまとめていくものです。同じような状況で苦しんで...
スポンサーリンク
環境
●Ubuntu 16.04
●Python 3.7.3
●conda 4.7.12
●pytorch 1.2.0
現象
「reduce failed to synchronize: device-side assert triggered」というエラーが出てしまう。
原因
損失関数の「nn.functional.binary_cross_entropy」に[0,1]を取らない値を突っ込んでしまったことが原因でした。
解決方法
「nn.functional.binary_cross_entropy」に突っ込む値は「torch.sigmoid」や「torch.nn.functional.softmax」を利用しましょう。
