回答編集履歴

3

修正

2020/09/03 08:31

投稿

tiitoi
tiitoi

スコア21956

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

修正

2020/09/03 08:31

投稿

tiitoi
tiitoi

スコア21956

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 = 0
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 += loss
39
+ epoch_loss.append(loss)
40
40
 
41
-
42
41
 
42
+
43
- losses.append(epoch_loss.mean()) # そのエポックの損失の平均を記録
43
+ losses.append(np.mean(epoch_loss)) # そのエポックの損失の平均を記録
44
44
 
45
45
  ```
46
46
 

1

修正

2020/09/03 08:14

投稿

tiitoi
tiitoi

スコア21956

test CHANGED
@@ -2,7 +2,7 @@
2
2
 
3
3
 
4
4
 
5
- エポックごとにそのエポックの loss の平均を記録して、あとから matplotlib で描画すればよいと思います。
5
+ エポックごとにそのエポックの loss の平均を計算、それをリストに追加しいって、あとから matplotlib で描画すればよいと思います。
6
6
 
7
7
 
8
8