MAE-pytorch
MAE-pytorch copied to clipboard
RuntimeError: Given normalized_shape=[768], expected input with shape [*, 768], but got input of size[12]
I'm running mae's finetune"return torch.layer_norm(input, normalized_shape, weight, bias, eps, torch.backends.cudnn.enabled) RuntimeError: Given normalized_shape=[768], expected input with shape [*, 768], but got input of size[12]",Please tell me what the problem is.