回答編集履歴
1
コード追加
answer
CHANGED
|
@@ -12,4 +12,15 @@
|
|
|
12
12
|
みたいにして読み込めばいいはず
|
|
13
13
|
|
|
14
14
|
参考
|
|
15
|
-
[pytorchで書いたモデルを保存するときの落とし穴](https://qiita.com/jyori112/items/aad5703c1537c0139edb)
|
|
15
|
+
[pytorchで書いたモデルを保存するときの落とし穴](https://qiita.com/jyori112/items/aad5703c1537c0139edb)
|
|
16
|
+
|
|
17
|
+
【追記】
|
|
18
|
+
```python
|
|
19
|
+
train_model(model_ft,dataset_sizes,criterion,optimizer,40,False)
|
|
20
|
+
```
|
|
21
|
+
の一番最後の「False」を「True」に変えてください
|
|
22
|
+
「False」だと、下記の「if」のブロック内に入らないので、「torch.save(...」が実行されず、ファイルが保存されません
|
|
23
|
+
```python
|
|
24
|
+
if(is_saved):
|
|
25
|
+
torch.save(model.state_dict(), './original_model_{}.pth'.format(epoch))
|
|
26
|
+
```
|