質問編集履歴
2
保存方法の変更、質問の追加
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -7,6 +7,12 @@
|
|
|
7
7
|
について教えていただきたいです。
|
|
8
8
|
なお、100epoch(episode)ごとにモデルを保存しています。
|
|
9
9
|
|
|
10
|
+
### 追記
|
|
11
|
+
モデル全体を保存するように変更したところ、学習済みモデルから再開することに成功しました。
|
|
12
|
+
しかし、この方法は一般的には非推奨となっており、またなぜこれで成功したのか分かっていない状態です。
|
|
13
|
+
ですので、成功した要因等についても教えていただきたいです。
|
|
14
|
+
以下のソースコードに該当部分を追記しておきます。
|
|
15
|
+
|
|
10
16
|
環境
|
|
11
17
|
-
|
|
12
18
|
- windows10
|
|
@@ -37,6 +43,15 @@
|
|
|
37
43
|
self.loss = checkpoint['loss']
|
|
38
44
|
self.epoch = checkpoint['epoch']
|
|
39
45
|
```
|
|
46
|
+
|
|
47
|
+
### 追記
|
|
48
|
+
```
|
|
49
|
+
保存
|
|
50
|
+
torch.save(model, 'save/to/path/model.pt')
|
|
51
|
+
|
|
52
|
+
読み込み
|
|
53
|
+
model = torch.load('load/from/path/model.pt')
|
|
54
|
+
```
|
|
40
55
|
### 補足情報（FW/ツールのバージョンなど）
|
|
41
56
|
|
|
42
57
|
以下に該当ソースコードを載せておきます。
|
1
脱字の修正
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -39,7 +39,7 @@
|
|
|
39
39
|
```
|
|
40
40
|
### 補足情報（FW/ツールのバージョンなど）
|
|
41
41
|
|
|
42
|
-
以下に
|
|
42
|
+
以下に該当ソースコードを載せておきます。
|
|
43
43
|
```
|
|
44
44
|
import sys
|
|
45
45
|
import os
|