エントリーの編集
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
Pytorch 訓練時にベストモデルを更新する方法 - Qiita
記事へのコメント0件
- 注目コメント
- 新着コメント
このエントリーにコメントしてみましょう。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています
- バナー広告なし
- ミュート機能あり
- ダークモード搭載
関連記事
Pytorch 訓練時にベストモデルを更新する方法 - Qiita
検証時の推定精度が最大値になるたびにbest_modelを更新する例を考えてみます。以下の例ではbest_model... 検証時の推定精度が最大値になるたびにbest_modelを更新する例を考えてみます。以下の例ではbest_modelがmodel自体を参照しているため、modelが更新されるたびにbest_modelのパラメータも更新されてしまいます。これではダメです。 model = model.eval() best_accuracy = 0 ... for epoch in range(100): for idx, data in enumerate(data_loader): ... if cur_accuracy > best_accuracy: best_model = model best_accuracy = cur_accuracy torch.save(best_model.state_dict(), 'model.pt') import copy ... model = model.