回答編集履歴
1
コード追加
test
CHANGED
|
@@ -27,3 +27,25 @@
|
|
|
27
27
|
参考
|
|
28
28
|
|
|
29
29
|
[pytorchで書いたモデルを保存するときの落とし穴](https://qiita.com/jyori112/items/aad5703c1537c0139edb)
|
|
30
|
+
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
【追記】
|
|
34
|
+
|
|
35
|
+
```python
|
|
36
|
+
|
|
37
|
+
train_model(model_ft,dataset_sizes,criterion,optimizer,40,False)
|
|
38
|
+
|
|
39
|
+
```
|
|
40
|
+
|
|
41
|
+
の一番最後の「False」を「True」に変えてください
|
|
42
|
+
|
|
43
|
+
「False」だと、下記の「if」のブロック内に入らないので、「torch.save(...」が実行されず、ファイルが保存されません
|
|
44
|
+
|
|
45
|
+
```python
|
|
46
|
+
|
|
47
|
+
if(is_saved):
|
|
48
|
+
|
|
49
|
+
torch.save(model.state_dict(), './original_model_{}.pth'.format(epoch))
|
|
50
|
+
|
|
51
|
+
```
|