RuntimeError

[jinweiLiu]

"RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 3, 7, 7], but got input of size [2, 10, 3, 320, 320] instead"
when train use a gpu it occured. How can I fix it?

[Stephanessy]

I found the solution, #40