C:\Users\Smile\Desktop\manga-image-translator-beta-0.2.0>translate_demo.py --mode web --use-inpainting --use-cuda
Namespace(box_threshold=0.7, image='', inpainting_size=2048, mode='web', size=2048, text_threshold=0.5, unclip_ratio=2.2, use_cuda=True, use_inpainting=True)
-- Loading models
Traceback (most recent call last):
File "C:\Users\Smile\Desktop\manga-image-translator-beta-0.2.0\translate_demo.py", line 784, in
asyncio.run(main(args.mode))
File "C:\Users\Smile\AppData\Local\Programs\Python\Python38\lib\asyncio\runners.py", line 43, in run
return loop.run_until_complete(main)
File "C:\Users\Smile\AppData\Local\Programs\Python\Python38\lib\asyncio\base_events.py", line 616, in run_until_complete
return future.result()
File "C:\Users\Smile\Desktop\manga-image-translator-beta-0.2.0\translate_demo.py", line 749, in main
dictionary, model_ocr = load_ocr_model()
File "C:\Users\Smile\Desktop\manga-image-translator-beta-0.2.0\translate_demo.py", line 481, in load_ocr_model
model.load_state_dict(torch.load('ocr.ckpt', map_location='cpu'), strict=False)
File "C:\Users\Smile\AppData\Roaming\Python\Python38\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 OCR:
size mismatch for backbone.ConvNet.conv0_1.weight: copying a param with shape torch.Size([40, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([32, 3, 3, 3]).
size mismatch for backbone.ConvNet.bn0_1.weight: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for backbone.ConvNet.bn0_1.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for backbone.ConvNet.bn0_1.running_mean: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for backbone.ConvNet.bn0_1.running_var: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([32]).
size mismatch for backbone.ConvNet.conv0_2.weight: copying a param with shape torch.Size([40, 40, 3, 3]) from checkpoint, the shape in current model is torch.Size([64, 32, 3, 3]).
size mismatch for backbone.ConvNet.layer1.0.conv1.weight: copying a param with shape torch.Size([80, 40, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 64, 3, 3]).
size mismatch for backbone.ConvNet.layer1.0.bn1.weight: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.bn1.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.bn1.running_mean: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.bn1.running_var: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.conv2.weight: copying a param with shape torch.Size([80, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).
size mismatch for backbone.ConvNet.layer1.0.bn2.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.bn2.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.bn2.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.bn2.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.0.downsample.0.weight: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([128, 64, 1, 1]).
size mismatch for backbone.ConvNet.layer1.0.downsample.1.weight: copying a param with shape torch.Size([80, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.conv1.weight: copying a param with shape torch.Size([80, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).
size mismatch for backbone.ConvNet.layer1.1.bn1.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.bn1.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.bn1.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.bn1.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.conv2.weight: copying a param with shape torch.Size([80, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).
size mismatch for backbone.ConvNet.layer1.1.bn2.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.bn2.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.bn2.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.1.bn2.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.conv1.weight: copying a param with shape torch.Size([80, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).
size mismatch for backbone.ConvNet.layer1.2.bn1.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.bn1.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.bn1.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.bn1.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.conv2.weight: copying a param with shape torch.Size([80, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).
size mismatch for backbone.ConvNet.layer1.2.bn2.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.bn2.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.bn2.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer1.2.bn2.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.conv1.weight: copying a param with shape torch.Size([80, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).
size mismatch for backbone.ConvNet.bn1.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.bn1.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.bn1.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.bn1.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([128]).
size mismatch for backbone.ConvNet.layer2.0.conv1.weight: copying a param with shape torch.Size([160, 80, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 128, 3, 3]).
size mismatch for backbone.ConvNet.layer2.0.bn1.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.bn1.bias: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.bn1.running_mean: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.bn1.running_var: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.conv2.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.0.bn2.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.bn2.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.bn2.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.bn2.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.0.downsample.0.weight: copying a param with shape torch.Size([80]) from checkpoint, the shape in current model is torch.Size([256, 128, 1, 1]).
size mismatch for backbone.ConvNet.layer2.0.downsample.1.weight: copying a param with shape torch.Size([160, 80, 1, 1]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.conv1.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.1.bn1.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.bn1.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.bn1.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.bn1.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.conv2.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.1.bn2.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.bn2.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.bn2.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.1.bn2.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.conv1.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.2.bn1.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.bn1.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.bn1.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.bn1.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.conv2.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.2.bn2.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.bn2.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.bn2.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.2.bn2.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.conv1.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.3.bn1.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.bn1.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.bn1.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.bn1.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.conv2.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.3.bn2.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.bn2.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.bn2.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.3.bn2.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.conv1.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.4.bn1.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.bn1.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.bn1.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.bn1.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.conv2.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer2.4.bn2.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.bn2.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.bn2.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer2.4.bn2.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.conv2.weight: copying a param with shape torch.Size([160, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([256, 256, 3, 3]).
size mismatch for backbone.ConvNet.bn2.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.bn2.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.bn2.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.bn2.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([256]).
size mismatch for backbone.ConvNet.layer3.0.conv1.weight: copying a param with shape torch.Size([320, 160, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 256, 3, 3]).
size mismatch for backbone.ConvNet.layer3.0.bn1.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.bn1.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.bn1.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.bn1.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.0.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.0.downsample.0.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([512, 256, 1, 1]).
size mismatch for backbone.ConvNet.layer3.0.downsample.1.weight: copying a param with shape torch.Size([320, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.1.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.1.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.1.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.2.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.2.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.2.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.3.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.3.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.3.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.4.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.4.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.4.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.5.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.5.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.5.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.6.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer3.6.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer3.6.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.conv3.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.bn3.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn3.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn3.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn3.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.0.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.0.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.0.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.1.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.1.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.1.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.2.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.2.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.2.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.3.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.3.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.3.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.conv1.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.4.bn1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.bn1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.bn1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.bn1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.conv2.weight: copying a param with shape torch.Size([320, 320, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 512, 3, 3]).
size mismatch for backbone.ConvNet.layer4.4.bn2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.bn2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.bn2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.layer4.4.bn2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.conv4_1.weight: copying a param with shape torch.Size([320, 320, 2, 2]) from checkpoint, the shape in current model is torch.Size([512, 512, 2, 2]).
size mismatch for backbone.ConvNet.bn4_1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn4_1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn4_1.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn4_1.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.conv4_2.weight: copying a param with shape torch.Size([320, 320, 2, 2]) from checkpoint, the shape in current model is torch.Size([512, 512, 2, 2]).
size mismatch for backbone.ConvNet.bn4_2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn4_2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn4_2.running_mean: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for backbone.ConvNet.bn4_2.running_var: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.0.self_attn.in_proj_weight: copying a param with shape torch.Size([960, 320]) from checkpoint, the shape in current model is torch.Size([1536, 512]).
size mismatch for encoders.layers.0.self_attn.in_proj_bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1536]).
size mismatch for encoders.layers.0.self_attn.out_proj.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([512, 512]).
size mismatch for encoders.layers.0.self_attn.out_proj.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.0.linear1.weight: copying a param with shape torch.Size([2048, 320]) from checkpoint, the shape in current model is torch.Size([2048, 512]).
size mismatch for encoders.layers.0.linear2.weight: copying a param with shape torch.Size([320, 2048]) from checkpoint, the shape in current model is torch.Size([512, 2048]).
size mismatch for encoders.layers.0.linear2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.0.norm1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.0.norm1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.0.norm2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.0.norm2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.1.self_attn.in_proj_weight: copying a param with shape torch.Size([960, 320]) from checkpoint, the shape in current model is torch.Size([1536, 512]).
size mismatch for encoders.layers.1.self_attn.in_proj_bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1536]).
size mismatch for encoders.layers.1.self_attn.out_proj.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([512, 512]).
size mismatch for encoders.layers.1.self_attn.out_proj.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.1.linear1.weight: copying a param with shape torch.Size([2048, 320]) from checkpoint, the shape in current model is torch.Size([2048, 512]).
size mismatch for encoders.layers.1.linear2.weight: copying a param with shape torch.Size([320, 2048]) from checkpoint, the shape in current model is torch.Size([512, 2048]).
size mismatch for encoders.layers.1.linear2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.1.norm1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.1.norm1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.1.norm2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for encoders.layers.1.norm2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.self_attn.in_proj_weight: copying a param with shape torch.Size([960, 320]) from checkpoint, the shape in current model is torch.Size([1536, 512]).
size mismatch for decoders.layers.0.self_attn.in_proj_bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1536]).
size mismatch for decoders.layers.0.self_attn.out_proj.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([512, 512]).
size mismatch for decoders.layers.0.self_attn.out_proj.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.multihead_attn.in_proj_weight: copying a param with shape torch.Size([960, 320]) from checkpoint, the shape in current model is torch.Size([1536, 512]).
size mismatch for decoders.layers.0.multihead_attn.in_proj_bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1536]).
size mismatch for decoders.layers.0.multihead_attn.out_proj.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([512, 512]).
size mismatch for decoders.layers.0.multihead_attn.out_proj.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.linear1.weight: copying a param with shape torch.Size([2048, 320]) from checkpoint, the shape in current model is torch.Size([2048, 512]).
size mismatch for decoders.layers.0.linear2.weight: copying a param with shape torch.Size([320, 2048]) from checkpoint, the shape in current model is torch.Size([512, 2048]).
size mismatch for decoders.layers.0.linear2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.norm1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.norm1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.norm2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.norm2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.norm3.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.0.norm3.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.self_attn.in_proj_weight: copying a param with shape torch.Size([960, 320]) from checkpoint, the shape in current model is torch.Size([1536, 512]).
size mismatch for decoders.layers.1.self_attn.in_proj_bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1536]).
size mismatch for decoders.layers.1.self_attn.out_proj.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([512, 512]).
size mismatch for decoders.layers.1.self_attn.out_proj.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.multihead_attn.in_proj_weight: copying a param with shape torch.Size([960, 320]) from checkpoint, the shape in current model is torch.Size([1536, 512]).
size mismatch for decoders.layers.1.multihead_attn.in_proj_bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1536]).
size mismatch for decoders.layers.1.multihead_attn.out_proj.weight: copying a param with shape torch.Size([320, 320]) from checkpoint, the shape in current model is torch.Size([512, 512]).
size mismatch for decoders.layers.1.multihead_attn.out_proj.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.linear1.weight: copying a param with shape torch.Size([2048, 320]) from checkpoint, the shape in current model is torch.Size([2048, 512]).
size mismatch for decoders.layers.1.linear2.weight: copying a param with shape torch.Size([320, 2048]) from checkpoint, the shape in current model is torch.Size([512, 2048]).
size mismatch for decoders.layers.1.linear2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.norm1.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.norm1.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.norm2.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.norm2.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.norm3.weight: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for decoders.layers.1.norm3.bias: copying a param with shape torch.Size([320]) from checkpoint, the shape in current model is torch.Size([512]).
size mismatch for pe.pe: copying a param with shape torch.Size([768, 1, 320]) from checkpoint, the shape in current model is torch.Size([768, 1, 512]).
size mismatch for embd.weight: copying a param with shape torch.Size([19264, 320]) from checkpoint, the shape in current model is torch.Size([19264, 512]).
size mismatch for color_pred1.0.weight: copying a param with shape torch.Size([64, 320]) from checkpoint, the shape in current model is torch.Size([64, 512]).