fallpoke https://blog.hatena.ne.jp/fallpoke/ 誰かの技術置き場 https://fallpoke-tech.hatenadiary.jp/ Pytorch Python 備忘録 画像処理 お疲れ様です。 今回はPytorchで学習の途中再開をするためのコードのメモです。 長期間モデル学習を実行する際に予期せぬトラブルで処理が止まってしまった場合などにも使えると思います。 ソースコード 以前作成した画像分類のプロジェクトに実装しています。 github.com 実装 モデルの保存時 下記modules/trainer.pyのTrainerクラスのモデル保存用メソッドです。 modelのパラメータに加えてoptimizerのパラメータを保存するのがポイントになっています。 ちなみに下記は最新epochのモデルを保存するためのメソッドですが、これを毎epoch行うようにしています。 … 190 <iframe src="https://hatenablog-parts.com/embed?url=https%3A%2F%2Ffallpoke-tech.hatenadiary.jp%2Fentry%2F2025%2F11%2F26%2F205134" title="Pytorchのモデル学習の中断と再開の処理を実装したメモ【備忘録】 - 誰かの技術置き場" class="embed-card embed-blogcard" scrolling="no" frameborder="0" style="display: block; width: 100%; height: 190px; max-width: 500px; margin: 10px 0px;"></iframe> https://cdn-ak.f.st-hatena.com/images/fotolife/f/fallpoke/20251126/20251126204745.png Hatena Blog https://hatena.blog 2025-11-26 20:51:34 rich https://fallpoke-tech.hatenadiary.jp/entry/2025/11/26/205134 1.0 100%