回答編集履歴
1
修正
answer
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
> これに関連した質問なのですが、上記の方法で保存されたモデルは、学習エポックの内1番性能が良かったモデルが保存されるのでしょうか。
|
|
2
2
|
> それとも、学習の最終エポックのモデルが保存されるのでしょうか。
|
|
3
3
|
|
|
4
|
-
最後に `torch.save(the_model.state_dict(), PATH)` が実行されたときのモデルです。
|
|
4
|
+
ファイル名を固定にしているなら、既存のファイルは上書きされるので、最後に `torch.save(the_model.state_dict(), PATH)` が実行されたときのモデルです。
|
|
5
5
|
|
|
6
6
|
「学習エポックの内1番性能が良かったモデルを保存したい」場合はそのようにコードを自分で書く必要があります。
|