回答編集履歴
1
書式の修正（軽微）
answer
CHANGED
|
@@ -5,12 +5,12 @@
|
|
|
5
5
|
num_workers=2)
|
|
6
6
|
```
|
|
7
7
|
|
|
8
|
-
で指定してあるので，`forward()`の入力テンソルのサイズは
|
|
8
|
+
で指定してあるので，`forward()`の入力テンソルのサイズは`(32, 1, 28, 28)`となっており，`view()`等は不要です．
|
|
9
9
|
実際のミスは別のところに3つあります．
|
|
10
10
|
|
|
11
|
-
1. `self.conv1(x)`
|
|
11
|
+
1. `self.conv1(x)`を呼び忘れている
|
|
12
12
|
3. `conv1`のあとのReLUを呼び忘れている
|
|
13
|
-
2. `conv1`
|
|
13
|
+
2. `conv1`の出力は，`(32, 32, 26, 26)`なので，`d1`の入力チャネル数は`32 * 26 * 26`になる
|
|
14
14
|
|
|
15
15
|
それらを直すと以下のようになります．
|
|
16
16
|
|
|
@@ -32,7 +32,7 @@
|
|
|
32
32
|
```
|
|
33
33
|
|
|
34
34
|
|
|
35
|
-
|
|
35
|
+
さらに`to(device)`もしておけば動くと思います．
|
|
36
36
|
|
|
37
37
|
|
|
38
38
|
```diff
|