vit-pytorch

vit-pytorch copied to clipboard

My code: `if __name__ == '__main__': import torch import cv2 input_size = 224 img = cv2.imread("OIP-C.jpg") img = cv2.resize(img,(224,224)) img = rearrange(img, 'h w c -> c h w') img...

hustCYQ

Issue of Given normalized_shape=[768], expected input with shape [*, 768], but got input of size[9, 768, 14, 14]

1

@likelyzhao Hello, when I used your code to finetine my dataset, I got a problem like this, could you tell me how to slove it? Thank you! 

helloworld-wu