回答編集履歴
3
修正
test
CHANGED
@@ -36,7 +36,7 @@
|
|
36
36
|
|
37
37
|
loss = criterion(outputs, labels)
|
38
38
|
|
39
|
-
epoch_loss.append(loss)
|
39
|
+
epoch_loss.append(float(loss))
|
40
40
|
|
41
41
|
|
42
42
|
|
2
修正
test
CHANGED
@@ -18,7 +18,7 @@
|
|
18
18
|
|
19
19
|
for epoch in range(300): # loop over the dataset multiple times
|
20
20
|
|
21
|
-
epoch_loss =
|
21
|
+
epoch_loss = []
|
22
22
|
|
23
23
|
for i, data in enumerate(train_loader, 0):
|
24
24
|
|
@@ -36,11 +36,11 @@
|
|
36
36
|
|
37
37
|
loss = criterion(outputs, labels)
|
38
38
|
|
39
|
-
epoch_loss
|
39
|
+
epoch_loss.append(loss)
|
40
40
|
|
41
|
-
|
42
41
|
|
42
|
+
|
43
|
-
losses.append(epoch_loss
|
43
|
+
losses.append(np.mean(epoch_loss)) # そのエポックの損失の平均を記録
|
44
44
|
|
45
45
|
```
|
46
46
|
|
1
修正
test
CHANGED
@@ -2,7 +2,7 @@
|
|
2
2
|
|
3
3
|
|
4
4
|
|
5
|
-
エポックごとにそのエポックの loss の平均を
|
5
|
+
エポックごとにそのエポックの loss の平均を計算し、それをリストに追加していって、あとから matplotlib で描画すればよいと思います。
|
6
6
|
|
7
7
|
|
8
8
|
|