前提・実現したいこと
keras-yoloでepoch51~100以降で学習する際にエラーが表示される
epoch51以降は、バッチサイズを8に減らさないと学習されない状態です。
epoch51以降もバッチサイズをデフォルトの32で学習する方法を知りたいです。
発生している問題・エラーメッセージ
Allocator (GPU_0_bfc) ran out of memory trying to allocate 2.17GiB. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.
該当のソースコード
ソースコード
試したこと
ここに問題に対して試したことを記載してください。
補足情報(FW/ツールのバージョンなど)
ここにより詳細な情報を記載してください。
あなたの回答
tips
プレビュー