2020-08-30から1日間の記事一覧
pytorchでモデルを学習させているときにlossがnanになって時間を結構溶かしたのでメモ。 BCEではなくBCEWithLogitsLossを使っているか 2値分類問題で自分で最終レイヤをいじったモデルはnanにならなくてOSSのモデル実装を使うとなぜかnanになる現象にぶちあ…
pytorchでモデルを学習させているときにlossがnanになって時間を結構溶かしたのでメモ。 BCEではなくBCEWithLogitsLossを使っているか 2値分類問題で自分で最終レイヤをいじったモデルはnanにならなくてOSSのモデル実装を使うとなぜかnanになる現象にぶちあ…