回答編集履歴
1
d
answer
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
> RuntimeError: Expected 4-dimensional input for 4-dimensional weight [16, 3, 3, 3], but got 2-dimensional input of size [32, 196608] instead
|
|
2
2
|
|
|
3
|
-
4次元配列の入力を期待しているのに、2次元配列が入力として
|
|
3
|
+
4次元配列の入力を期待しているのに、2次元配列が入力として与えられたというエラーです。
|
|
4
4
|
原因は images.view(images.shape[0], -1) で4次元配列を2次元配列にしていることです。
|
|
5
5
|
|
|
6
6
|
> cnnで一次元にしないとなると通常どのように設定すればいいのでしょうか....
|