状況

YOLOv5のtrain.pyでの学習時、box_lossとobj_lossがnanになる

解決策

バッチサイズを小さくする

原因

lossがnanになる問題は様々な原因で起こりますが、上記解決策で解決した場合はメモリ不足が原因と思われます。
loss計算時にメモリが不足していると、異常終了などはせずにlossをnanとして処理続行する実装になっています。

参考URL

• https://github.com/ultralytics/yolov5/issues/6907

The following two tabs change content below.

• この記事を書いた人
• 最新の記事

神谷 全俊

2018年からフリーランスのシステムエンジニアになりました。　出身は沖縄県で、プロフィール画像も沖縄で撮った写真です。　ITについては他のSEの方が述べられているので、記事にはIT関連でないことを書いていく予定です。

最新記事 by 神谷 全俊 (全て見る)

• 【本の備忘録】自由意志の向こう側 決定論をめぐる哲学史 – はじめに - 2023年12月16日
• YOLOv5での学習時にlossがnanになった時の対処法 - 2023年11月3日
• Python入門動画を投稿しました - 2023年5月4日