Pytorchで保存したモデルについて

Question

前提・実現したいこと

普段Kerasやtfを使うことが多いのですが、最近Pytorchの勉強を始めました。
Pytorchで学習後モデルを保存する場合、以下のようにコードを書くと学びました。

torch.save(the_model.state_dict(), PATH)

これに関連した質問なのですが、上記の方法で保存されたモデルは、学習エポックの内1番性能が良かったモデルが保存されるのでしょうか。
それとも、学習の最終エポックのモデルが保存されるのでしょうか。
初歩的な質問で恐れ入りますが、ご教示いただければ幸いです。

Answer 1

これに関連した質問なのですが、上記の方法で保存されたモデルは、学習エポックの内1番性能が良かったモデルが保存されるのでしょうか。
それとも、学習の最終エポックのモデルが保存されるのでしょうか。

ファイル名を固定にしているなら、既存のファイルは上書きされるので、最後に torch.save(the_model.state_dict(), PATH) が実行されたときのモデルです。

「学習エポックの内1番性能が良かったモデルを保存したい」場合はそのようにコードを自分で書く必要があります。

Comments

[SuzuAya]

>tiitoi様
ご回答ありがとうございます！大変参考になりました。