RuntimeError: Error(s) in loading state_dict for BaseVideoModel: size mismatch for head.out.weight: copying a param with shape torch.Size([400, 768]) from checkpoint, the shape in current model is torch.Size([5, 768]). size mismatch for head.out.bias: copying a param with shape torch.Size([400]) from checkpoint, the shape in current model is torch.Size([5]).

[lyxlplhy]

RuntimeError: Error(s) in loading state_dict for BaseVideoModel: size mismatch for head.out.weight: copying a param with shape torch.Size([400, 768]) from checkpoint, the shape in current model is torch.Size([5, 768]). size mismatch for head.out.bias: copying a param with shape torch.Size([400]) from checkpoint, the shape in current model is torch.Size([5]).

[lyxlplhy]

I have resolved it.
Add two lines of code before the code in/TAdaConv main/tadconv/utils/checkpoint. py: mismatch=ms. load_sate_dict (checkpoint ["model_state"], strict=False):
Del checkpoint ["model_state"] ["head. out. weight"]
Del checkpoint ["model_state"] ["head. out. bias"]