状況

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年からフリーランスのシステムエンジニアになりました。 出身は沖縄県で、プロフィール画像も沖縄で撮った写真です。 個人ブログも書いており、こちらにはそこでの記事の一部を載せる予定です。

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

• 二項分布 - 2024年8月17日
• Pythonの便利ライブラリ – Typer - 2024年8月15日
• 自由意志論争の対立軸の整理 - 2024年4月28日