Hello,
I got this error when loading the weights with load_state_dict
RuntimeError: Error(s) in loading state_dict for ViT3:
size mismatch for pos_embedding: copying a param with shape torch.Size([1, 3137, 1024]) from checkpoint, the shape in current model is torch.Size([1, 1569, 1024]).
what's wrong ? also same problem with the 2D model
Hello, I got this error when loading the weights with load_state_dict
RuntimeError: Error(s) in loading state_dict for ViT3: size mismatch for pos_embedding: copying a param with shape torch.Size([1, 3137, 1024]) from checkpoint, the shape in current model is torch.Size([1, 1569, 1024]).
what's wrong ? also same problem with the 2D model