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