What's mean 'shape '[10, 65536]' is invalid for input of size 1966080'

1

File “D:/360Downloads/VS/MyDataset.py”, line 233, in
x=x.view(x.size(0),256*256)

RuntimeError: shape ‘[10, 65536]’ is invalid for input of size 1966080

2

The shape of your x is (10, 3, 256, 256).
use x = x.view(x.size(0), -1)
or x = x.view(x.size(0), 3 * 256 * 256)

3

Boy, this is your second topic I responded to.
You REALLY need to be more careful with your code…:rofl:

4

all right,thans for your answer