回答編集履歴
1
書式の修正（軽微）
test
CHANGED
|
@@ -12,17 +12,17 @@
|
|
|
12
12
|
|
|
13
13
|
|
|
14
14
|
|
|
15
|
-
で指定してあるので，`forward()`の入力テンソルのサイズは
|
|
15
|
+
で指定してあるので，`forward()`の入力テンソルのサイズは`(32, 1, 28, 28)`となっており，`view()`等は不要です．
|
|
16
16
|
|
|
17
17
|
実際のミスは別のところに3つあります．
|
|
18
18
|
|
|
19
19
|
|
|
20
20
|
|
|
21
|
-
1. `self.conv1(x)`
|
|
21
|
+
1. `self.conv1(x)`を呼び忘れている
|
|
22
22
|
|
|
23
23
|
3. `conv1`のあとのReLUを呼び忘れている
|
|
24
24
|
|
|
25
|
-
2. `conv1`
|
|
25
|
+
2. `conv1`の出力は，`(32, 32, 26, 26)`なので，`d1`の入力チャネル数は`32 * 26 * 26`になる
|
|
26
26
|
|
|
27
27
|
|
|
28
28
|
|
|
@@ -66,7 +66,7 @@
|
|
|
66
66
|
|
|
67
67
|
|
|
68
68
|
|
|
69
|
-
|
|
69
|
+
さらに`to(device)`もしておけば動くと思います．
|
|
70
70
|
|
|
71
71
|
|
|
72
72
|
|