RuntimeError

[xytywh]

when I run the program,I meet blow error:

RuntimeError :Expected 4-dimensional input for 4-dimensional weight [30,1,5,5],but got 5- dimensional input of size [16,2,1,256,256] instead

I don't know why?please help me!

[HarryAB]

Have you solved the problem?I have the same problem as you

[wanyan-li]

Have you solved the problem?I have the same problem as you

hello, have you solved this problem?

[niefengxxx]

have anyone solved this problem?................

[Yogurt2019]

Same problem! Have anyone solved it?

[Yogurt2019]

I solved it just add three lines under train()->images, labels = images.cuda(), labels.cuda():

`shape = list(images.shape)

images = images.view(shape[0]*shape[1], *shape[2:])

labels = labels.view(-1)`

just combine the 1st and the 2nd dimension of images and labels.