TPN icon indicating copy to clipboard operation
TPN copied to clipboard

RuntimeError: Expected 5-dimensional input for 5-dimensional weight 64 3 1 7 7, but got 4-dimensional input of size [12, 10, 224, 224] instead

Open Tramac opened this issue 4 years ago • 0 comments

img_group shape=(4, 3, 10, 224, 224) # NCTHW

After operator: img_group.reshape((-1,) + img_group.shape[2:]), the shape of x is: (12, 10, 224, 224), and leads to the error above in here.

Is this a code bug?

Tramac avatar Nov 22 '21 13:11 Tramac