runtime error

True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Drop path rate: 0.0 No L_MHRA: True Double L_MHRA: True Use checkpoint: False Checkpoint number: [0] Drop path rate: 0.0 Drop path rate: 0.0 Drop path rate: 0.0 Drop path rate: 0.0 Traceback (most recent call last): File "app.py", line 36, in <module> model.load_state_dict(state_dict) File "/home/user/.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1406, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for Uniformerv2: size mismatch for backbone.conv1.weight: copying a param with shape torch.Size([1024, 3, 1, 14, 14]) from checkpoint, the shape in current model is torch.Size([1024, 3, 3, 14, 14]). size mismatch for backbone.transformer.proj.2.weight: copying a param with shape torch.Size([339, 1024]) from checkpoint, the shape in current model is torch.Size([400, 1024]). size mismatch for backbone.transformer.proj.2.bias: copying a param with shape torch.Size([339]) from checkpoint, the shape in current model is torch.Size([400]).

Container logs: