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