FSOD-code icon indicating copy to clipboard operation
FSOD-code copied to clipboard

Published 20 hours ago • verified publisher icon fanq15

RuntimeError

Open jinweiLiu opened this issue 3 years ago • 1 comments

"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?

jinweiLiu avatar Mar 05 '21 13:03 jinweiLiu

I found the solution, #40

Stephanessy avatar Apr 19 '21 21:04 Stephanessy