size mismatch for garPcapsLayers.1.components: copying a param with shape torch.Size([64, 12774]) from checkpoint, the shape in current model is torch.Size([64, 8454]).

[Shakhbozboy]

~/BCNet/code$ python infer.py --inputs ../images --save-root ../recs
Traceback (most recent call last):
File "infer.py", line 117, in
net.load_state_dict(tor, True)
File "/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1224, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for ImageReconstructModel:
Missing key(s) in state_dict: "skinWsNet.ress.0.conv1.att_l", "skinWsNet.ress.0.conv1.att_r", "skinWsNet.ress.0.conv1.lin_l.weight", "skinWsNet.ress.0.conv1.lin_r.weight", "skinWsNet.ress.0.conv2.att_l", "skinWsNet.ress.0.conv2.att_r", "skinWsNet.ress.0.conv2.lin_l.weight", "skinWsNet.ress.0.conv2.lin_r.weight", "skinWsNet.ress.1.conv1.att_l", "skinWsNet.ress.1.conv1.att_r", "skinWsNet.ress.1.conv1.lin_l.weight", "skinWsNet.ress.1.conv1.lin_r.weight", "skinWsNet.ress.1.conv2.att_l", "skinWsNet.ress.1.conv2.att_r", "skinWsNet.ress.1.conv2.lin_l.weight", "skinWsNet.ress.1.conv2.lin_r.weight", "skinWsNet.ress.2.conv1.att_l", "skinWsNet.ress.2.conv1.att_r", "skinWsNet.ress.2.conv1.lin_l.weight", "skinWsNet.ress.2.conv1.lin_r.weight", "skinWsNet.ress.2.conv2.att_l", "skinWsNet.ress.2.conv2.att_r", "skinWsNet.ress.2.conv2.lin_l.weight", "skinWsNet.ress.2.conv2.lin_r.weight", "skinWsNet.ress.3.conv1.att_l", "skinWsNet.ress.3.conv1.att_r", "skinWsNet.ress.3.conv1.lin_l.weight", "skinWsNet.ress.3.conv1.lin_r.weight", "skinWsNet.ress.3.conv2.att_l", "skinWsNet.ress.3.conv2.att_r", "skinWsNet.ress.3.conv2.lin_l.weight", "skinWsNet.ress.3.conv2.lin_r.weight".
Unexpected key(s) in state_dict: "skinWsNet.ress.0.conv1.lin.weight", "skinWsNet.ress.0.conv1.att_i", "skinWsNet.ress.0.conv1.att_j", "skinWsNet.ress.0.conv2.lin.weight", "skinWsNet.ress.0.conv2.att_i", "skinWsNet.ress.0.conv2.att_j", "skinWsNet.ress.1.conv1.lin.weight", "skinWsNet.ress.1.conv1.att_i", "skinWsNet.ress.1.conv1.att_j", "skinWsNet.ress.1.conv2.lin.weight", "skinWsNet.ress.1.conv2.att_i", "skinWsNet.ress.1.conv2.att_j", "skinWsNet.ress.2.conv1.lin.weight", "skinWsNet.ress.2.conv1.att_i", "skinWsNet.ress.2.conv1.att_j", "skinWsNet.ress.2.conv2.lin.weight", "skinWsNet.ress.2.conv2.att_i", "skinWsNet.ress.2.conv2.att_j", "skinWsNet.ress.3.conv1.lin.weight", "skinWsNet.ress.3.conv1.att_i", "skinWsNet.ress.3.conv1.att_j", "skinWsNet.ress.3.conv2.lin.weight", "skinWsNet.ress.3.conv2.att_i", "skinWsNet.ress.3.conv2.att_j".
size mismatch for garPcapsLayers.1.mean: copying a param with shape torch.Size([12774]) from checkpoint, the shape in current model is torch.Size([8454]).
size mismatch for garPcapsLayers.1.components: copying a param with shape torch.Size([64, 12774]) from checkpoint, the shape in current model is torch.Size([64, 8454]).
size mismatch for garPcapsLayers.1.edge_index: copying a param with shape torch.Size([2, 25224]) from checkpoint, the shape in current model is torch.Size([2, 16572]).
size mismatch for garPcapsLayers.1.face_index: copying a param with shape torch.Size([8352, 3]) from checkpoint, the shape in current model is torch.Size([5468, 3]).
size mismatch for garPcapsLayers.1.vf_findex: copying a param with shape torch.Size([25056]) from checkpoint, the shape in current model is torch.Size([16404]).
size mismatch for garPcapsLayers.1.vf_vindex: copying a param with shape torch.Size([25056]) from checkpoint, the shape in current model is torch.Size([16404]).
size mismatch for garDisplacementLayers.1.res1.conv1.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.res1.conv2.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.ress.0.conv1.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.ress.0.conv2.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.ress.1.conv1.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.ress.1.conv2.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.ress.2.conv1.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.1.ress.2.conv2.spiral_adj: copying a param with shape torch.Size([1, 4259, 20]) from checkpoint, the shape in current model is torch.Size([1, 2819, 20]).
size mismatch for garDisplacementLayers.5.res1.conv1.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.res1.conv2.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.ress.0.conv1.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.ress.0.conv2.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.ress.1.conv1.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.ress.1.conv2.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.ress.2.conv1.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).
size mismatch for garDisplacementLayers.5.ress.2.conv2.spiral_adj: copying a param with shape torch.Size([1, 2819, 20]) from checkpoint, the shape in current model is torch.Size([1, 5405, 20]).

[jby1993]

maybe you can check #4