アカデミック

【Pythonお悩み解決】「reduce failed to synchronize: device-side assert triggered」というエラーメッセージの解決方法

この記事は, Pythonを利用して研究を行なっていく中で私がつまずいてしまったポイントをまとめていくものです。同じような状況で苦しんでいる方々の参考になれば嬉しいです。Pythonつまずきポイント集の目次は以下のページをご覧ください。

【超初心者お悩み解決】Pythonつまずきポイント記事まとめページ この記事は,Pythonを利用して研究を行なっていく中で私がつまずいてしまったポイントをまとめていくものです。同じような状況で苦しんで...

本記事で紹介する解決策がBestという保証はできません。正確な情報を発信するように心掛けていますが図らずも誤った情報を記載してしまう場合があります。もしご指摘等がありましたら,コメント欄またはお問い合わせページよりご連絡下さい。

読みたい場所へジャンプ!

環境

●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」を利用しましょう。

COMMENT

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です