Unexpected and missing keys

[guist]

Hello,

I am trying to run the testing script for raster images. I saved the pre-trained checkpoint in the models/ directory and the R-50.pth file in the weights/ directory. Then I typed the command line:

python trainer.py test --config configs/train.yaml --ckpt_path models/line_complete-checkpoint_999-precision\=0.944-recall\=0.934-f1\=0.938.ckpt --trainer.devices 1

But I got the error below. Could you please tell me what went wrong?

Thank you!

Best regards, Guillaume

raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for Trainer: Missing key(s) in state_dict: "backbone2d.0.fpn_lateral2.weight", "backbone2d.0.fpn_lateral2.norm.weight", "backbone2d.0.fpn_lateral2.norm.bias", "backbone2d.0.fpn_lateral2.norm.running_mean", "backbone2d.0.fpn_lateral2.norm.running_var", "backbone2d.0.fpn_output2.weight", "backbone2d.0.fpn_output2.norm.weight", "backbone2d.0.fpn_output2.norm.bias", "backbone2d.0.fpn_output2.norm.running_mean", "backbone2d.0.fpn_output2.norm.running_var", "backbone2d.0.fpn_lateral3.weight", "backbone2d.0.fpn_lateral3.norm.weight", "backbone2d.0.fpn_lateral3.norm.bias", "backbone2d.0.fpn_lateral3.norm.running_mean", "backbone2d.0.fpn_lateral3.norm.running_var", "backbone2d.0.fpn_output3.weight", "backbone2d.0.fpn_output3.norm.weight", "backbone2d.0.fpn_output3.norm.bias", "backbone2d.0.fpn_output3.norm.running_mean", "backbone2d.0.fpn_output3.norm.running_var", "backbone2d.0.fpn_lateral4.weight", "backbone2d.0.fpn_lateral4.norm.weight", "backbone2d.0.fpn_lateral4.norm.bias", "backbone2d.0.fpn_lateral4.norm.running_mean", "backbone2d.0.fpn_lateral4.norm.running_var", "backbone2d.0.fpn_output4.weight", "backbone2d.0.fpn_output4.norm.weight", "backbone2d.0.fpn_output4.norm.bias", "backbone2d.0.fpn_output4.norm.running_mean", "backbone2d.0.fpn_output4.norm.running_var", "backbone2d.0.fpn_lateral5.weight", "backbone2d.0.fpn_lateral5.norm.weight", "backbone2d.0.fpn_lateral5.norm.bias", "backbone2d.0.fpn_lateral5.norm.running_mean", "backbone2d.0.fpn_lateral5.norm.running_var", "backbone2d.0.fpn_output5.weight", "backbone2d.0.fpn_output5.norm.weight", "backbone2d.0.fpn_output5.norm.bias", "backbone2d.0.fpn_output5.norm.running_mean", "backbone2d.0.fpn_output5.norm.running_var", "backbone2d.0.bottom_up.stem.conv1.weight", "backbone2d.0.bottom_up.stem.conv1.norm.weight", "backbone2d.0.bottom_up.stem.conv1.norm.bias", "backbone2d.0.bottom_up.stem.conv1.norm.running_mean", "backbone2d.0.bottom_up.stem.conv1.norm.running_var", "backbone2d.0.bottom_up.res2.0.shortcut.weight", "backbone2d.0.bottom_up.res2.0.shortcut.norm.weight", "backbone2d.0.bottom_up.res2.0.shortcut.norm.bias", "backbone2d.0.bottom_up.res2.0.shortcut.norm.running_mean", "backbone2d.0.bottom_up.res2.0.shortcut.norm.running_var", "backbone2d.0.bottom_up.res2.0.conv1.weight", "backbone2d.0.bottom_up.res2.0.conv1.norm.weight", "backbone2d.0.bottom_up.res2.0.conv1.norm.bias", "backbone2d.0.bottom_up.res2.0.conv1.norm.running_mean", "backbone2d.0.bottom_up.res2.0.conv1.norm.running_var", "backbone2d.0.bottom_up.res2.0.conv2.weight", "backbone2d.0.bottom_up.res2.0.conv2.norm.weight", "backbone2d.0.bottom_up.res2.0.conv2.norm.bias", "backbone2d.0.bottom_up.res2.0.conv2.norm.running_mean", "backbone2d.0.bottom_up.res2.0.conv2.norm.running_var", "backbone2d.0.bottom_up.res2.0.conv3.weight", "backbone2d.0.bottom_up.res2.0.conv3.norm.weight", "backbone2d.0.bottom_up.res2.0.conv3.norm.bias", "backbone2d.0.bottom_up.res2.0.conv3.norm.running_mean", "backbone2d.0.bottom_up.res2.0.conv3.norm.running_var", "backbone2d.0.bottom_up.res2.1.conv1.weight", "backbone2d.0.bottom_up.res2.1.conv1.norm.weight", "backbone2d.0.bottom_up.res2.1.conv1.norm.bias", "backbone2d.0.bottom_up.res2.1.conv1.norm.running_mean", "backbone2d.0.bottom_up.res2.1.conv1.norm.running_var", "backbone2d.0.bottom_up.res2.1.conv2.weight", "backbone2d.0.bottom_up.res2.1.conv2.norm.weight", "backbone2d.0.bottom_up.res2.1.conv2.norm.bias", "backbone2d.0.bottom_up.res2.1.conv2.norm.running_mean", "backbone2d.0.bottom_up.res2.1.conv2.norm.running_var", "backbone2d.0.bottom_up.res2.1.conv3.weight", "backbone2d.0.bottom_up.res2.1.conv3.norm.weight", "backbone2d.0.bottom_up.res2.1.conv3.norm.bias", "backbone2d.0.bottom_up.res2.1.conv3.norm.running_mean", "backbone2d.0.bottom_up.res2.1.conv3.norm.running_var", "backbone2d.0.bottom_up.res2.2.conv1.weight", "backbone2d.0.bottom_up.res2.2.conv1.norm.weight", "backbone2d.0.bottom_up.res2.2.conv1.norm.bias", "backbone2d.0.bottom_up.res2.2.conv1.norm.running_mean", "backbone2d.0.bottom_up.res2.2.conv1.norm.running_var", "backbone2d.0.bottom_up.res2.2.conv2.weight", "backbone2d.0.bottom_up.res2.2.conv2.norm.weight", "backbone2d.0.bottom_up.res2.2.conv2.norm.bias", "backbone2d.0.bottom_up.res2.2.conv2.norm.running_mean", "backbone2d.0.bottom_up.res2.2.conv2.norm.running_var", "backbone2d.0.bottom_up.res2.2.conv3.weight", "backbone2d.0.bottom_up.res2.2.conv3.norm.weight", "backbone2d.0.bottom_up.res2.2.conv3.norm.bias", "backbone2d.0.bottom_up.res2.2.conv3.norm.running_mean", "backbone2d.0.bottom_up.res2.2.conv3.norm.running_var", "backbone2d.0.bottom_up.res3.0.shortcut.weight", "backbone2d.0.bottom_up.res3.0.shortcut.norm.weight", "backbone2d.0.bottom_up.res3.0.shortcut.norm.bias", "backbone2d.0.bottom_up.res3.0.shortcut.norm.running_mean", "backbone2d.0.bottom_up.res3.0.shortcut.norm.running_var", "backbone2d.0.bottom_up.res3.0.conv1.weight", "backbone2d.0.bottom_up.res3.0.conv1.norm.weight", "backbone2d.0.bottom_up.res3.0.conv1.norm.bias", "backbone2d.0.bottom_up.res3.0.conv1.norm.running_mean", "backbone2d.0.bottom_up.res3.0.conv1.norm.running_var", "backbone2d.0.bottom_up.res3.0.conv2.weight", "backbone2d.0.bottom_up.res3.0.conv2.norm.weight", "backbone2d.0.bottom_up.res3.0.conv2.norm.bias", "backbone2d.0.bottom_up.res3.0.conv2.norm.running_mean", "backbone2d.0.bottom_up.res3.0.conv2.norm.running_var", "backbone2d.0.bottom_up.res3.0.conv3.weight", "backbone2d.0.bottom_up.res3.0.conv3.norm.weight", "backbone2d.0.bottom_up.res3.0.conv3.norm.bias", "backbone2d.0.bottom_up.res3.0.conv3.norm.running_mean", "backbone2d.0.bottom_up.res3.0.conv3.norm.running_var", "backbone2d.0.bottom_up.res3.1.conv1.weight", "backbone2d.0.bottom_up.res3.1.conv1.norm.weight", "backbone2d.0.bottom_up.res3.1.conv1.norm.bias", "backbone2d.0.bottom_up.res3.1.conv1.norm.running_mean", "backbone2d.0.bottom_up.res3.1.conv1.norm.running_var", "backbone2d.0.bottom_up.res3.1.conv2.weight", "backbone2d.0.bottom_up.res3.1.conv2.norm.weight", "backbone2d.0.bottom_up.res3.1.conv2.norm.bias", "backbone2d.0.bottom_up.res3.1.conv2.norm.running_mean", "backbone2d.0.bottom_up.res3.1.conv2.norm.running_var", "backbone2d.0.bottom_up.res3.1.conv3.weight", "backbone2d.0.bottom_up.res3.1.conv3.norm.weight", "backbone2d.0.bottom_up.res3.1.conv3.norm.bias", "backbone2d.0.bottom_up.res3.1.conv3.norm.running_mean", "backbone2d.0.bottom_up.res3.1.conv3.norm.running_var", "backbone2d.0.bottom_up.res3.2.conv1.weight", "backbone2d.0.bottom_up.res3.2.conv1.norm.weight", "backbone2d.0.bottom_up.res3.2.conv1.norm.bias", "backbone2d.0.bottom_up.res3.2.conv1.norm.running_mean", "backbone2d.0.bottom_up.res3.2.conv1.norm.running_var", "backbone2d.0.bottom_up.res3.2.conv2.weight", "backbone2d.0.bottom_up.res3.2.conv2.norm.weight", "backbone2d.0.bottom_up.res3.2.conv2.norm.bias", "backbone2d.0.bottom_up.res3.2.conv2.norm.running_mean", "backbone2d.0.bottom_up.res3.2.conv2.norm.running_var", "backbone2d.0.bottom_up.res3.2.conv3.weight", "backbone2d.0.bottom_up.res3.2.conv3.norm.weight", "backbone2d.0.bottom_up.res3.2.conv3.norm.bias", "backbone2d.0.bottom_up.res3.2.conv3.norm.running_mean", "backbone2d.0.bottom_up.res3.2.conv3.norm.running_var", "backbone2d.0.bottom_up.res3.3.conv1.weight", "backbone2d.0.bottom_up.res3.3.conv1.norm.weight", "backbone2d.0.bottom_up.res3.3.conv1.norm.bias", "backbone2d.0.bottom_up.res3.3.conv1.norm.running_mean", "backbone2d.0.bottom_up.res3.3.conv1.norm.running_var", "backbone2d.0.bottom_up.res3.3.conv2.weight", "backbone2d.0.bottom_up.res3.3.conv2.norm.weight", "backbone2d.0.bottom_up.res3.3.conv2.norm.bias", "backbone2d.0.bottom_up.res3.3.conv2.norm.running_mean", "backbone2d.0.bottom_up.res3.3.conv2.norm.running_var", "backbone2d.0.bottom_up.res3.3.conv3.weight", "backbone2d.0.bottom_up.res3.3.conv3.norm.weight", "backbone2d.0.bottom_up.res3.3.conv3.norm.bias", "backbone2d.0.bottom_up.res3.3.conv3.norm.running_mean", "backbone2d.0.bottom_up.res3.3.conv3.norm.running_var", "backbone2d.0.bottom_up.res4.0.shortcut.weight", "backbone2d.0.bottom_up.res4.0.shortcut.norm.weight", "backbone2d.0.bottom_up.res4.0.shortcut.norm.bias", "backbone2d.0.bottom_up.res4.0.shortcut.norm.running_mean", "backbone2d.0.bottom_up.res4.0.shortcut.norm.running_var", "backbone2d.0.bottom_up.res4.0.conv1.weight", "backbone2d.0.bottom_up.res4.0.conv1.norm.weight", "backbone2d.0.bottom_up.res4.0.conv1.norm.bias", "backbone2d.0.bottom_up.res4.0.conv1.norm.running_mean", "backbone2d.0.bottom_up.res4.0.conv1.norm.running_var", "backbone2d.0.bottom_up.res4.0.conv2.weight", "backbone2d.0.bottom_up.res4.0.conv2.norm.weight", "backbone2d.0.bottom_up.res4.0.conv2.norm.bias", "backbone2d.0.bottom_up.res4.0.conv2.norm.running_mean", "backbone2d.0.bottom_up.res4.0.conv2.norm.running_var", "backbone2d.0.bottom_up.res4.0.conv3.weight", "backbone2d.0.bottom_up.res4.0.conv3.norm.weight", "backbone2d.0.bottom_up.res4.0.conv3.norm.bias", "backbone2d.0.bottom_up.res4.0.conv3.norm.running_mean", "backbone2d.0.bottom_up.res4.0.conv3.norm.running_var", "backbone2d.0.bottom_up.res4.1.conv1.weight", "backbone2d.0.bottom_up.res4.1.conv1.norm.weight", "backbone2d.0.bottom_up.res4.1.conv1.norm.bias", "backbone2d.0.bottom_up.res4.1.conv1.norm.running_mean", "backbone2d.0.bottom_up.res4.1.conv1.norm.running_var", "backbone2d.0.bottom_up.res4.1.conv2.weight", "backbone2d.0.bottom_up.res4.1.conv2.norm.weight", "backbone2d.0.bottom_up.res4.1.conv2.norm.bias", "backbone2d.0.bottom_up.res4.1.conv2.norm.running_mean", "backbone2d.0.bottom_up.res4.1.conv2.norm.running_var", "backbone2d.0.bottom_up.res4.1.conv3.weight", "backbone2d.0.bottom_up.res4.1.conv3.norm.weight", "backbone2d.0.bottom_up.res4.1.conv3.norm.bias", "backbone2d.0.bottom_up.res4.1.conv3.norm.running_mean", "backbone2d.0.bottom_up.res4.1.conv3.norm.running_var", "backbone2d.0.bottom_up.res4.2.conv1.weight", "backbone2d.0.bottom_up.res4.2.conv1.norm.weight", "backbone2d.0.bottom_up.res4.2.conv1.norm.bias", "backbone2d.0.bottom_up.res4.2.conv1.norm.running_mean", "backbone2d.0.bottom_up.res4.2.conv1.norm.running_var", "backbone2d.0.bottom_up.res4.2.conv2.weight", "backbone2d.0.bottom_up.res4.2.conv2.norm.weight", "backbone2d.0.bottom_up.res4.2.conv2.norm.bias", "backbone2d.0.bottom_up.res4.2.conv2.norm.running_mean", "backbone2d.0.bottom_up.res4.2.conv2.norm.running_var", "backbone2d.0.bottom_up.res4.2.conv3.weight", "backbone2d.0.bottom_up.res4.2.conv3.norm.weight", "backbone2d.0.bottom_up.res4.2.conv3.norm.bias", "backbone2d.0.bottom_up.res4.2.conv3.norm.running_mean", "backbone2d.0.bottom_up.res4.2.conv3.norm.running_var", "backbone2d.0.bottom_up.res4.3.conv1.weight", "backbone2d.0.bottom_up.res4.3.conv1.norm.weight", "backbone2d.0.bottom_up.res4.3.conv1.norm.bias", "backbone2d.0.bottom_up.res4.3.conv1.norm.running_mean", "backbone2d.0.bottom_up.res4.3.conv1.norm.running_var", "backbone2d.0.bottom_up.res4.3.conv2.weight", "backbone2d.0.bottom_up.res4.3.conv2.norm.weight", "backbone2d.0.bottom_up.res4.3.conv2.norm.bias", "backbone2d.0.bottom_up.res4.3.conv2.norm.running_mean", "backbone2d.0.bottom_up.res4.3.conv2.norm.running_var", "backbone2d.0.bottom_up.res4.3.conv3.weight", "backbone2d.0.bottom_up.res4.3.conv3.norm.weight", "backbone2d.0.bottom_up.res4.3.conv3.norm.bias", "backbone2d.0.bottom_up.res4.3.conv3.norm.running_mean", "backbone2d.0.bottom_up.res4.3.conv3.norm.running_var", "backbone2d.0.bottom_up.res4.4.conv1.weight", "backbone2d.0.bottom_up.res4.4.conv1.norm.weight", "backbone2d.0.bottom_up.res4.4.conv1.norm.bias", "backbone2d.0.bottom_up.res4.4.conv1.norm.running_mean", "backbone2d.0.bottom_up.res4.4.conv1.norm.running_var", "backbone2d.0.bottom_up.res4.4.conv2.weight", "backbone2d.0.bottom_up.res4.4.conv2.norm.weight", "backbone2d.0.bottom_up.res4.4.conv2.norm.bias", "backbone2d.0.bottom_up.res4.4.conv2.norm.running_mean", "backbone2d.0.bottom_up.res4.4.conv2.norm.running_var", "backbone2d.0.bottom_up.res4.4.conv3.weight", "backbone2d.0.bottom_up.res4.4.conv3.norm.weight", "backbone2d.0.bottom_up.res4.4.conv3.norm.bias", "backbone2d.0.bottom_up.res4.4.conv3.norm.running_mean", "backbone2d.0.bottom_up.res4.4.conv3.norm.running_var", "backbone2d.0.bottom_up.res4.5.conv1.weight", "backbone2d.0.bottom_up.res4.5.conv1.norm.weight", "backbone2d.0.bottom_up.res4.5.conv1.norm.bias", "backbone2d.0.bottom_up.res4.5.conv1.norm.running_mean", "backbone2d.0.bottom_up.res4.5.conv1.norm.running_var", "backbone2d.0.bottom_up.res4.5.conv2.weight", "backbone2d.0.bottom_up.res4.5.conv2.norm.weight", "backbone2d.0.bottom_up.res4.5.conv2.norm.bias", "backbone2d.0.bottom_up.res4.5.conv2.norm.running_mean", "backbone2d.0.bottom_up.res4.5.conv2.norm.running_var", "backbone2d.0.bottom_up.res4.5.conv3.weight", "backbone2d.0.bottom_up.res4.5.conv3.norm.weight", "backbone2d.0.bottom_up.res4.5.conv3.norm.bias", "backbone2d.0.bottom_up.res4.5.conv3.norm.running_mean", "backbone2d.0.bottom_up.res4.5.conv3.norm.running_var", "backbone2d.0.bottom_up.res5.0.shortcut.weight", "backbone2d.0.bottom_up.res5.0.shortcut.norm.weight", "backbone2d.0.bottom_up.res5.0.shortcut.norm.bias", "backbone2d.0.bottom_up.res5.0.shortcut.norm.running_mean", "backbone2d.0.bottom_up.res5.0.shortcut.norm.running_var", "backbone2d.0.bottom_up.res5.0.conv1.weight", "backbone2d.0.bottom_up.res5.0.conv1.norm.weight", "backbone2d.0.bottom_up.res5.0.conv1.norm.bias", "backbone2d.0.bottom_up.res5.0.conv1.norm.running_mean", "backbone2d.0.bottom_up.res5.0.conv1.norm.running_var", "backbone2d.0.bottom_up.res5.0.conv2.weight", "backbone2d.0.bottom_up.res5.0.conv2.norm.weight", "backbone2d.0.bottom_up.res5.0.conv2.norm.bias", "backbone2d.0.bottom_up.res5.0.conv2.norm.running_mean", "backbone2d.0.bottom_up.res5.0.conv2.norm.running_var", "backbone2d.0.bottom_up.res5.0.conv3.weight", "backbone2d.0.bottom_up.res5.0.conv3.norm.weight", "backbone2d.0.bottom_up.res5.0.conv3.norm.bias", "backbone2d.0.bottom_up.res5.0.conv3.norm.running_mean", "backbone2d.0.bottom_up.res5.0.conv3.norm.running_var", "backbone2d.0.bottom_up.res5.1.conv1.weight", "backbone2d.0.bottom_up.res5.1.conv1.norm.weight", "backbone2d.0.bottom_up.res5.1.conv1.norm.bias", "backbone2d.0.bottom_up.res5.1.conv1.norm.running_mean", "backbone2d.0.bottom_up.res5.1.conv1.norm.running_var", "backbone2d.0.bottom_up.res5.1.conv2.weight", "backbone2d.0.bottom_up.res5.1.conv2.norm.weight", "backbone2d.0.bottom_up.res5.1.conv2.norm.bias", "backbone2d.0.bottom_up.res5.1.conv2.norm.running_mean", "backbone2d.0.bottom_up.res5.1.conv2.norm.running_var", "backbone2d.0.bottom_up.res5.1.conv3.weight", "backbone2d.0.bottom_up.res5.1.conv3.norm.weight", "backbone2d.0.bottom_up.res5.1.conv3.norm.bias", "backbone2d.0.bottom_up.res5.1.conv3.norm.running_mean", "backbone2d.0.bottom_up.res5.1.conv3.norm.running_var", "backbone2d.0.bottom_up.res5.2.conv1.weight", "backbone2d.0.bottom_up.res5.2.conv1.norm.weight", "backbone2d.0.bottom_up.res5.2.conv1.norm.bias", "backbone2d.0.bottom_up.res5.2.conv1.norm.running_mean", "backbone2d.0.bottom_up.res5.2.conv1.norm.running_var", "backbone2d.0.bottom_up.res5.2.conv2.weight", "backbone2d.0.bottom_up.res5.2.conv2.norm.weight", "backbone2d.0.bottom_up.res5.2.conv2.norm.bias", "backbone2d.0.bottom_up.res5.2.conv2.norm.running_mean", "backbone2d.0.bottom_up.res5.2.conv2.norm.running_var", "backbone2d.0.bottom_up.res5.2.conv3.weight", "backbone2d.0.bottom_up.res5.2.conv3.norm.weight", "backbone2d.0.bottom_up.res5.2.conv3.norm.bias", "backbone2d.0.bottom_up.res5.2.conv3.norm.running_mean", "backbone2d.0.bottom_up.res5.2.conv3.norm.running_var", "backbone2d.1.p2.0.weight", "backbone2d.1.p2.0.norm.weight", "backbone2d.1.p2.0.norm.bias", "backbone2d.1.p2.0.norm.running_mean", "backbone2d.1.p2.0.norm.running_var", "backbone2d.1.p3.0.weight", "backbone2d.1.p3.0.norm.weight", "backbone2d.1.p3.0.norm.bias", "backbone2d.1.p3.0.norm.running_mean", "backbone2d.1.p3.0.norm.running_var", "backbone2d.1.p4.0.weight", "backbone2d.1.p4.0.norm.weight", "backbone2d.1.p4.0.norm.bias", "backbone2d.1.p4.0.norm.running_mean", "backbone2d.1.p4.0.norm.running_var", "backbone2d.1.p4.2.weight", "backbone2d.1.p4.2.norm.weight", "backbone2d.1.p4.2.norm.bias", "backbone2d.1.p4.2.norm.running_mean", "backbone2d.1.p4.2.norm.running_var", "backbone2d.1.p5.0.weight", "backbone2d.1.p5.0.norm.weight", "backbone2d.1.p5.0.norm.bias", "backbone2d.1.p5.0.norm.running_mean", "backbone2d.1.p5.0.norm.running_var", "backbone2d.1.p5.2.weight", "backbone2d.1.p5.2.norm.weight", "backbone2d.1.p5.2.norm.bias", "backbone2d.1.p5.2.norm.running_mean", "backbone2d.1.p5.2.norm.running_var", "backbone2d.1.p5.4.weight", "backbone2d.1.p5.4.norm.weight", "backbone2d.1.p5.4.norm.bias", "backbone2d.1.p5.4.norm.running_mean", "backbone2d.1.p5.4.norm.running_var", "backbone3d.layers.0.0.conv1.weight", "backbone3d.layers.0.0.bn1.weight", "backbone3d.layers.0.0.bn1.bias", "backbone3d.layers.0.0.bn1.running_mean", "backbone3d.layers.0.0.bn1.running_var", "backbone3d.layers.0.0.conv2.weight", "backbone3d.layers.0.0.bn2.weight", "backbone3d.layers.0.0.bn2.bias", "backbone3d.layers.0.0.bn2.running_mean", "backbone3d.layers.0.0.bn2.running_var", "backbone3d.layers.0.0.downsample.0.weight", "backbone3d.layers.0.0.downsample.1.weight", "backbone3d.layers.0.0.downsample.1.bias", "backbone3d.layers.0.0.downsample.1.running_mean", "backbone3d.layers.0.0.downsample.1.running_var", "backbone3d.layers.1.0.weight", "backbone3d.layers.1.1.weight", "backbone3d.layers.1.1.bias", "backbone3d.layers.1.1.running_mean", "backbone3d.layers.1.1.running_var", "backbone3d.layers.1.4.conv1.weight", "backbone3d.layers.1.4.bn1.weight", "backbone3d.layers.1.4.bn1.bias", "backbone3d.layers.1.4.bn1.running_mean", "backbone3d.layers.1.4.bn1.running_var", "backbone3d.layers.1.4.conv2.weight", "backbone3d.layers.1.4.bn2.weight", "backbone3d.layers.1.4.bn2.bias", "backbone3d.layers.1.4.bn2.running_mean", "backbone3d.layers.1.4.bn2.running_var", "backbone3d.layers.1.5.conv1.weight", "backbone3d.layers.1.5.bn1.weight", "backbone3d.layers.1.5.bn1.bias", "backbone3d.layers.1.5.bn1.running_mean", "backbone3d.layers.1.5.bn1.running_var", "backbone3d.layers.1.5.conv2.weight", "backbone3d.layers.1.5.bn2.weight", "backbone3d.layers.1.5.bn2.bias", "backbone3d.layers.1.5.bn2.running_mean", "backbone3d.layers.1.5.bn2.running_var", "backbone3d.layers.2.0.weight", "backbone3d.layers.2.1.weight", "backbone3d.layers.2.1.bias", "backbone3d.layers.2.1.running_mean", "backbone3d.layers.2.1.running_var", "backbone3d.layers.2.4.conv1.weight", "backbone3d.layers.2.4.bn1.weight", "backbone3d.layers.2.4.bn1.bias", "backbone3d.layers.2.4.bn1.running_mean", "backbone3d.layers.2.4.bn1.running_var", "backbone3d.layers.2.4.conv2.weight", "backbone3d.layers.2.4.bn2.weight", "backbone3d.layers.2.4.bn2.bias", "backbone3d.layers.2.4.bn2.running_mean", "backbone3d.layers.2.4.bn2.running_var", "backbone3d.layers.2.5.conv1.weight", "backbone3d.layers.2.5.bn1.weight", "backbone3d.layers.2.5.bn1.bias", "backbone3d.layers.2.5.bn1.running_mean", "backbone3d.layers.2.5.bn1.running_var", "backbone3d.layers.2.5.conv2.weight", "backbone3d.layers.2.5.bn2.weight", "backbone3d.layers.2.5.bn2.bias", "backbone3d.layers.2.5.bn2.running_mean", "backbone3d.layers.2.5.bn2.running_var", "backbone3d.layers.2.6.conv1.weight", "backbone3d.layers.2.6.bn1.weight", "backbone3d.layers.2.6.bn1.bias", "backbone3d.layers.2.6.bn1.running_mean", "backbone3d.layers.2.6.bn1.running_var", "backbone3d.layers.2.6.conv2.weight", "backbone3d.layers.2.6.bn2.weight", "backbone3d.layers.2.6.bn2.bias", "backbone3d.layers.2.6.bn2.running_mean", "backbone3d.layers.2.6.bn2.running_var", "backbone3d.layers.3.0.weight", "backbone3d.layers.3.1.weight", "backbone3d.layers.3.1.bias", "backbone3d.layers.3.1.running_mean", "backbone3d.layers.3.1.running_var", "backbone3d.layers.3.4.conv1.weight", "backbone3d.layers.3.4.bn1.weight", "backbone3d.layers.3.4.bn1.bias", "backbone3d.layers.3.4.bn1.running_mean", "backbone3d.layers.3.4.bn1.running_var", "backbone3d.layers.3.4.conv2.weight", "backbone3d.layers.3.4.bn2.weight", "backbone3d.layers.3.4.bn2.bias", "backbone3d.layers.3.4.bn2.running_mean", "backbone3d.layers.3.4.bn2.running_var", "backbone3d.layers.3.5.conv1.weight", "backbone3d.layers.3.5.bn1.weight", "backbone3d.layers.3.5.bn1.bias", "backbone3d.layers.3.5.bn1.running_mean", "backbone3d.layers.3.5.bn1.running_var", "backbone3d.layers.3.5.conv2.weight", "backbone3d.layers.3.5.bn2.weight", "backbone3d.layers.3.5.bn2.bias", "backbone3d.layers.3.5.bn2.running_mean", "backbone3d.layers.3.5.bn2.running_var", "backbone3d.layers.3.6.conv1.weight", "backbone3d.layers.3.6.bn1.weight", "backbone3d.layers.3.6.bn1.bias", "backbone3d.layers.3.6.bn1.running_mean", "backbone3d.layers.3.6.bn1.running_var", "backbone3d.layers.3.6.conv2.weight", "backbone3d.layers.3.6.bn2.weight", "backbone3d.layers.3.6.bn2.bias", "backbone3d.layers.3.6.bn2.running_mean", "backbone3d.layers.3.6.bn2.running_var", "backbone3d.layers.3.7.conv1.weight", "backbone3d.layers.3.7.bn1.weight", "backbone3d.layers.3.7.bn1.bias", "backbone3d.layers.3.7.bn1.running_mean", "backbone3d.layers.3.7.bn1.running_var", "backbone3d.layers.3.7.conv2.weight", "backbone3d.layers.3.7.bn2.weight", "backbone3d.layers.3.7.bn2.bias", "backbone3d.layers.3.7.bn2.running_mean", "backbone3d.layers.3.7.bn2.running_var", "backbone3d.layers.4.0.weight", "backbone3d.layers.4.1.weight", "backbone3d.layers.4.1.bias", "backbone3d.layers.4.1.running_mean", "backbone3d.layers.4.1.running_var", "backbone3d.layers.4.4.conv1.weight", "backbone3d.layers.4.4.bn1.weight", "backbone3d.layers.4.4.bn1.bias", "backbone3d.layers.4.4.bn1.running_mean", "backbone3d.layers.4.4.bn1.running_var", "backbone3d.layers.4.4.conv2.weight", "backbone3d.layers.4.4.bn2.weight", "backbone3d.layers.4.4.bn2.bias", "backbone3d.layers.4.4.bn2.running_mean", "backbone3d.layers.4.4.bn2.running_var", "backbone3d.layers.4.5.conv1.weight", "backbone3d.layers.4.5.bn1.weight", "backbone3d.layers.4.5.bn1.bias", "backbone3d.layers.4.5.bn1.running_mean", "backbone3d.layers.4.5.bn1.running_var", "backbone3d.layers.4.5.conv2.weight", "backbone3d.layers.4.5.bn2.weight", "backbone3d.layers.4.5.bn2.bias", "backbone3d.layers.4.5.bn2.running_mean", "backbone3d.layers.4.5.bn2.running_var", "backbone3d.layers.4.6.conv1.weight", "backbone3d.layers.4.6.bn1.weight", "backbone3d.layers.4.6.bn1.bias", "backbone3d.layers.4.6.bn1.running_mean", "backbone3d.layers.4.6.bn1.running_var", "backbone3d.layers.4.6.conv2.weight", "backbone3d.layers.4.6.bn2.weight", "backbone3d.layers.4.6.bn2.bias", "backbone3d.layers.4.6.bn2.running_mean", "backbone3d.layers.4.6.bn2.running_var", "backbone3d.layers.4.7.conv1.weight", "backbone3d.layers.4.7.bn1.weight", "backbone3d.layers.4.7.bn1.bias", "backbone3d.layers.4.7.bn1.running_mean", "backbone3d.layers.4.7.bn1.running_var", "backbone3d.layers.4.7.conv2.weight", "backbone3d.layers.4.7.bn2.weight", "backbone3d.layers.4.7.bn2.bias", "backbone3d.layers.4.7.bn2.running_mean", "backbone3d.layers.4.7.bn2.running_var", "backbone3d.layers.4.8.conv1.weight", "backbone3d.layers.4.8.bn1.weight", "backbone3d.layers.4.8.bn1.bias", "backbone3d.layers.4.8.bn1.running_mean", "backbone3d.layers.4.8.bn1.running_var", "backbone3d.layers.4.8.conv2.weight", "backbone3d.layers.4.8.bn2.weight", "backbone3d.layers.4.8.bn2.bias", "backbone3d.layers.4.8.bn2.running_mean", "backbone3d.layers.4.8.bn2.running_var", "heads3d.pos_encoder.x_embed.weight", "heads3d.pos_encoder.y_embed.weight", "heads3d.pos_encoder.z_embed.weight", "heads3d.value_embedding.weight", "heads3d.query_coord_embedding.weight", "heads3d.query_pos_embedding.weight", "heads3d.linear.weight", "heads3d.linear.bias", "heads3d.decoder.layers.0.self_attn.in_proj_weight", "heads3d.decoder.layers.0.self_attn.in_proj_bias", "heads3d.decoder.layers.0.self_attn.out_proj.weight", "heads3d.decoder.layers.0.self_attn.out_proj.bias", "heads3d.decoder.layers.0.multihead_attn.in_proj_weight", "heads3d.decoder.layers.0.multihead_attn.in_proj_bias", "heads3d.decoder.layers.0.multihead_attn.out_proj.weight", "heads3d.decoder.layers.0.multihead_attn.out_proj.bias", "heads3d.decoder.layers.0.linear1.weight", "heads3d.decoder.layers.0.linear1.bias", "heads3d.decoder.layers.0.linear2.weight", "heads3d.decoder.layers.0.linear2.bias", "heads3d.decoder.layers.0.norm1.weight", "heads3d.decoder.layers.0.norm1.bias", "heads3d.decoder.layers.0.norm2.weight", "heads3d.decoder.layers.0.norm2.bias", "heads3d.decoder.layers.0.norm3.weight", "heads3d.decoder.layers.0.norm3.bias", "heads3d.decoder.layers.1.self_attn.in_proj_weight", "heads3d.decoder.layers.1.self_attn.in_proj_bias", "heads3d.decoder.layers.1.self_attn.out_proj.weight", "heads3d.decoder.layers.1.self_attn.out_proj.bias", "heads3d.decoder.layers.1.multihead_attn.in_proj_weight", "heads3d.decoder.layers.1.multihead_attn.in_proj_bias", "heads3d.decoder.layers.1.multihead_attn.out_proj.weight", "heads3d.decoder.layers.1.multihead_attn.out_proj.bias", "heads3d.decoder.layers.1.linear1.weight", "heads3d.decoder.layers.1.linear1.bias", "heads3d.decoder.layers.1.linear2.weight", "heads3d.decoder.layers.1.linear2.bias", "heads3d.decoder.layers.1.norm1.weight", "heads3d.decoder.layers.1.norm1.bias", "heads3d.decoder.layers.1.norm2.weight", "heads3d.decoder.layers.1.norm2.bias", "heads3d.decoder.layers.1.norm3.weight", "heads3d.decoder.layers.1.norm3.bias", "heads3d.decoder.layers.2.self_attn.in_proj_weight", "heads3d.decoder.layers.2.self_attn.in_proj_bias", "heads3d.decoder.layers.2.self_attn.out_proj.weight", "heads3d.decoder.layers.2.self_attn.out_proj.bias", "heads3d.decoder.layers.2.multihead_attn.in_proj_weight", "heads3d.decoder.layers.2.multihead_attn.in_proj_bias", "heads3d.decoder.layers.2.multihead_attn.out_proj.weight", "heads3d.decoder.layers.2.multihead_attn.out_proj.bias", "heads3d.decoder.layers.2.linear1.weight", "heads3d.decoder.layers.2.linear1.bias", "heads3d.decoder.layers.2.linear2.weight", "heads3d.decoder.layers.2.linear2.bias", "heads3d.decoder.layers.2.norm1.weight", "heads3d.decoder.layers.2.norm1.bias", "heads3d.decoder.layers.2.norm2.weight", "heads3d.decoder.layers.2.norm2.bias", "heads3d.decoder.layers.2.norm3.weight", "heads3d.decoder.layers.2.norm3.bias", "heads3d.decoder.layers.3.self_attn.in_proj_weight", "heads3d.decoder.layers.3.self_attn.in_proj_bias", "heads3d.decoder.layers.3.self_attn.out_proj.weight", "heads3d.decoder.layers.3.self_attn.out_proj.bias", "heads3d.decoder.layers.3.multihead_attn.in_proj_weight", "heads3d.decoder.layers.3.multihead_attn.in_proj_bias", "heads3d.decoder.layers.3.multihead_attn.out_proj.weight", "heads3d.decoder.layers.3.multihead_attn.out_proj.bias", "heads3d.decoder.layers.3.linear1.weight", "heads3d.decoder.layers.3.linear1.bias", "heads3d.decoder.layers.3.linear2.weight", "heads3d.decoder.layers.3.linear2.bias", "heads3d.decoder.layers.3.norm1.weight", "heads3d.decoder.layers.3.norm1.bias", "heads3d.decoder.layers.3.norm2.weight", "heads3d.decoder.layers.3.norm2.bias", "heads3d.decoder.layers.3.norm3.weight", "heads3d.decoder.layers.3.norm3.bias", "heads3d.decoder.layers.4.self_attn.in_proj_weight", "heads3d.decoder.layers.4.self_attn.in_proj_bias", "heads3d.decoder.layers.4.self_attn.out_proj.weight", "heads3d.decoder.layers.4.self_attn.out_proj.bias", "heads3d.decoder.layers.4.multihead_attn.in_proj_weight", "heads3d.decoder.layers.4.multihead_attn.in_proj_bias", "heads3d.decoder.layers.4.multihead_attn.out_proj.weight", "heads3d.decoder.layers.4.multihead_attn.out_proj.bias", "heads3d.decoder.layers.4.linear1.weight", "heads3d.decoder.layers.4.linear1.bias", "heads3d.decoder.layers.4.linear2.weight", "heads3d.decoder.layers.4.linear2.bias", "heads3d.decoder.layers.4.norm1.weight", "heads3d.decoder.layers.4.norm1.bias", "heads3d.decoder.layers.4.norm2.weight", "heads3d.decoder.layers.4.norm2.bias", "heads3d.decoder.layers.4.norm3.weight", "heads3d.decoder.layers.4.norm3.bias", "heads3d.decoder.layers.5.self_attn.in_proj_weight", "heads3d.decoder.layers.5.self_attn.in_proj_bias", "heads3d.decoder.layers.5.self_attn.out_proj.weight", "heads3d.decoder.layers.5.self_attn.out_proj.bias", "heads3d.decoder.layers.5.multihead_attn.in_proj_weight", "heads3d.decoder.layers.5.multihead_attn.in_proj_bias", "heads3d.decoder.layers.5.multihead_attn.out_proj.weight", "heads3d.decoder.layers.5.multihead_attn.out_proj.bias", "heads3d.decoder.layers.5.linear1.weight", "heads3d.decoder.layers.5.linear1.bias", "heads3d.decoder.layers.5.linear2.weight", "heads3d.decoder.layers.5.linear2.bias", "heads3d.decoder.layers.5.norm1.weight", "heads3d.decoder.layers.5.norm1.bias", "heads3d.decoder.layers.5.norm2.weight", "heads3d.decoder.layers.5.norm2.bias", "heads3d.decoder.layers.5.norm3.weight", "heads3d.decoder.layers.5.norm3.bias", "heads3d.decoder.norm.weight", "heads3d.decoder.norm.bias", "heads3d.vocab_head.weight", "heads3d.vocab_head.bias", "heads3d.pointer_head.weight", "heads3d.pointer_head.bias", "heads3d.switch_head.weight", "heads3d.switch_head.bias". Unexpected key(s) in state_dict: "model.input_embeddings.input_value.weight", "model.input_embeddings.input_pos.weight", "model.input_embeddings.input_coord.weight", "model.input_embeddings.input_view.weight", "model.input_embeddings.input_type.weight", "model.query_coord_embedding.weight", "model.query_pos_embedding.weight", "model.encoder.layers.0.self_attn.in_proj_weight", "model.encoder.layers.0.self_attn.in_proj_bias", "model.encoder.layers.0.self_attn.out_proj.weight", "model.encoder.layers.0.self_attn.out_proj.bias", "model.encoder.layers.0.linear1.weight", "model.encoder.layers.0.linear1.bias", "model.encoder.layers.0.linear2.weight", "model.encoder.layers.0.linear2.bias", "model.encoder.layers.0.norm1.weight", "model.encoder.layers.0.norm1.bias", "model.encoder.layers.0.norm2.weight", "model.encoder.layers.0.norm2.bias", "model.encoder.layers.1.self_attn.in_proj_weight", "model.encoder.layers.1.self_attn.in_proj_bias", "model.encoder.layers.1.self_attn.out_proj.weight", "model.encoder.layers.1.self_attn.out_proj.bias", "model.encoder.layers.1.linear1.weight", "model.encoder.layers.1.linear1.bias", "model.encoder.layers.1.linear2.weight", "model.encoder.layers.1.linear2.bias", "model.encoder.layers.1.norm1.weight", "model.encoder.layers.1.norm1.bias", "model.encoder.layers.1.norm2.weight", "model.encoder.layers.1.norm2.bias", "model.encoder.layers.2.self_attn.in_proj_weight", "model.encoder.layers.2.self_attn.in_proj_bias", "model.encoder.layers.2.self_attn.out_proj.weight", "model.encoder.layers.2.self_attn.out_proj.bias", "model.encoder.layers.2.linear1.weight", "model.encoder.layers.2.linear1.bias", "model.encoder.layers.2.linear2.weight", "model.encoder.layers.2.linear2.bias", "model.encoder.layers.2.norm1.weight", "model.encoder.layers.2.norm1.bias", "model.encoder.layers.2.norm2.weight", "model.encoder.layers.2.norm2.bias", "model.encoder.layers.3.self_attn.in_proj_weight", "model.encoder.layers.3.self_attn.in_proj_bias", "model.encoder.layers.3.self_attn.out_proj.weight", "model.encoder.layers.3.self_attn.out_proj.bias", "model.encoder.layers.3.linear1.weight", "model.encoder.layers.3.linear1.bias", "model.encoder.layers.3.linear2.weight", "model.encoder.layers.3.linear2.bias", "model.encoder.layers.3.norm1.weight", "model.encoder.layers.3.norm1.bias", "model.encoder.layers.3.norm2.weight", "model.encoder.layers.3.norm2.bias", "model.encoder.layers.4.self_attn.in_proj_weight", "model.encoder.layers.4.self_attn.in_proj_bias", "model.encoder.layers.4.self_attn.out_proj.weight", "model.encoder.layers.4.self_attn.out_proj.bias", "model.encoder.layers.4.linear1.weight", "model.encoder.layers.4.linear1.bias", "model.encoder.layers.4.linear2.weight", "model.encoder.layers.4.linear2.bias", "model.encoder.layers.4.norm1.weight", "model.encoder.layers.4.norm1.bias", "model.encoder.layers.4.norm2.weight", "model.encoder.layers.4.norm2.bias", "model.encoder.layers.5.self_attn.in_proj_weight", "model.encoder.layers.5.self_attn.in_proj_bias", "model.encoder.layers.5.self_attn.out_proj.weight", "model.encoder.layers.5.self_attn.out_proj.bias", "model.encoder.layers.5.linear1.weight", "model.encoder.layers.5.linear1.bias", "model.encoder.layers.5.linear2.weight", "model.encoder.layers.5.linear2.bias", "model.encoder.layers.5.norm1.weight", "model.encoder.layers.5.norm1.bias", "model.encoder.layers.5.norm2.weight", "model.encoder.layers.5.norm2.bias", "model.encoder.norm.weight", "model.encoder.norm.bias", "model.decoder.layers.0.self_attn.in_proj_weight", "model.decoder.layers.0.self_attn.in_proj_bias", "model.decoder.layers.0.self_attn.out_proj.weight", "model.decoder.layers.0.self_attn.out_proj.bias", "model.decoder.layers.0.multihead_attn.in_proj_weight", "model.decoder.layers.0.multihead_attn.in_proj_bias", "model.decoder.layers.0.multihead_attn.out_proj.weight", "model.decoder.layers.0.multihead_attn.out_proj.bias", "model.decoder.layers.0.linear1.weight", "model.decoder.layers.0.linear1.bias", "model.decoder.layers.0.linear2.weight", "model.decoder.layers.0.linear2.bias", "model.decoder.layers.0.norm1.weight", "model.decoder.layers.0.norm1.bias", "model.decoder.layers.0.norm2.weight", "model.decoder.layers.0.norm2.bias", "model.decoder.layers.0.norm3.weight", "model.decoder.layers.0.norm3.bias", "model.decoder.layers.1.self_attn.in_proj_weight", "model.decoder.layers.1.self_attn.in_proj_bias", "model.decoder.layers.1.self_attn.out_proj.weight", "model.decoder.layers.1.self_attn.out_proj.bias", "model.decoder.layers.1.multihead_attn.in_proj_weight", "model.decoder.layers.1.multihead_attn.in_proj_bias", "model.decoder.layers.1.multihead_attn.out_proj.weight", "model.decoder.layers.1.multihead_attn.out_proj.bias", "model.decoder.layers.1.linear1.weight", "model.decoder.layers.1.linear1.bias", "model.decoder.layers.1.linear2.weight", "model.decoder.layers.1.linear2.bias", "model.decoder.layers.1.norm1.weight", "model.decoder.layers.1.norm1.bias", "model.decoder.layers.1.norm2.weight", "model.decoder.layers.1.norm2.bias", "model.decoder.layers.1.norm3.weight", "model.decoder.layers.1.norm3.bias", "model.decoder.layers.2.self_attn.in_proj_weight", "model.decoder.layers.2.self_attn.in_proj_bias", "model.decoder.layers.2.self_attn.out_proj.weight", "model.decoder.layers.2.self_attn.out_proj.bias", "model.decoder.layers.2.multihead_attn.in_proj_weight", "model.decoder.layers.2.multihead_attn.in_proj_bias", "model.decoder.layers.2.multihead_attn.out_proj.weight", "model.decoder.layers.2.multihead_attn.out_proj.bias", "model.decoder.layers.2.linear1.weight", "model.decoder.layers.2.linear1.bias", "model.decoder.layers.2.linear2.weight", "model.decoder.layers.2.linear2.bias", "model.decoder.layers.2.norm1.weight", "model.decoder.layers.2.norm1.bias", "model.decoder.layers.2.norm2.weight", "model.decoder.layers.2.norm2.bias", "model.decoder.layers.2.norm3.weight", "model.decoder.layers.2.norm3.bias", "model.decoder.layers.3.self_attn.in_proj_weight", "model.decoder.layers.3.self_attn.in_proj_bias", "model.decoder.layers.3.self_attn.out_proj.weight", "model.decoder.layers.3.self_attn.out_proj.bias", "model.decoder.layers.3.multihead_attn.in_proj_weight", "model.decoder.layers.3.multihead_attn.in_proj_bias", "model.decoder.layers.3.multihead_attn.out_proj.weight", "model.decoder.layers.3.multihead_attn.out_proj.bias", "model.decoder.layers.3.linear1.weight", "model.decoder.layers.3.linear1.bias", "model.decoder.layers.3.linear2.weight", "model.decoder.layers.3.linear2.bias", "model.decoder.layers.3.norm1.weight", "model.decoder.layers.3.norm1.bias", "model.decoder.layers.3.norm2.weight", "model.decoder.layers.3.norm2.bias", "model.decoder.layers.3.norm3.weight", "model.decoder.layers.3.norm3.bias", "model.decoder.layers.4.self_attn.in_proj_weight", "model.decoder.layers.4.self_attn.in_proj_bias", "model.decoder.layers.4.self_attn.out_proj.weight", "model.decoder.layers.4.self_attn.out_proj.bias", "model.decoder.layers.4.multihead_attn.in_proj_weight", "model.decoder.layers.4.multihead_attn.in_proj_bias", "model.decoder.layers.4.multihead_attn.out_proj.weight", "model.decoder.layers.4.multihead_attn.out_proj.bias", "model.decoder.layers.4.linear1.weight", "model.decoder.layers.4.linear1.bias", "model.decoder.layers.4.linear2.weight", "model.decoder.layers.4.linear2.bias", "model.decoder.layers.4.norm1.weight", "model.decoder.layers.4.norm1.bias", "model.decoder.layers.4.norm2.weight", "model.decoder.layers.4.norm2.bias", "model.decoder.layers.4.norm3.weight", "model.decoder.layers.4.norm3.bias", "model.decoder.layers.5.self_attn.in_proj_weight", "model.decoder.layers.5.self_attn.in_proj_bias", "model.decoder.layers.5.self_attn.out_proj.weight", "model.decoder.layers.5.self_attn.out_proj.bias", "model.decoder.layers.5.multihead_attn.in_proj_weight", "model.decoder.layers.5.multihead_attn.in_proj_bias", "model.decoder.layers.5.multihead_attn.out_proj.weight", "model.decoder.layers.5.multihead_attn.out_proj.bias", "model.decoder.layers.5.linear1.weight", "model.decoder.layers.5.linear1.bias", "model.decoder.layers.5.linear2.weight", "model.decoder.layers.5.linear2.bias", "model.decoder.layers.5.norm1.weight", "model.decoder.layers.5.norm1.bias", "model.decoder.layers.5.norm2.weight", "model.decoder.layers.5.norm2.bias", "model.decoder.layers.5.norm3.weight", "model.decoder.layers.5.norm3.bias", "model.decoder.norm.weight", "model.decoder.norm.bias", "model.vocab_head.weight", "model.vocab_head.bias", "model.pointer_head.weight", "model.pointer_head.bias", "model.switch_head.weight", "model.switch_head.bias".

[bertjiazheng]

Hi,

Thanks for your interest in our work. The pre-trained model for the raster image model is incorrect. I have already updated the model link. Could you please check it and let me know if it's working now?

Best, Jia

[guist]

Hi Jia,

Thanks, the model could be loaded now.

Best, Guillaume