Non-matching tensors - Githubissues

AliaksandrSiarohin / motion-cosegmentation

Reference code for "Motion-supervised Co-Part Segmentation" paper

Other

653 stars 142 forks source link

Non-matching tensors #47

Closed ErikDz closed 3 years ago

ErikDz commented 3 years ago

module.state_dict()['affine.weight'].copy_(checkpoint['kp_detector']['jacobian.weight'])
RuntimeError: The size of tensor a (60) must match the size of tensor b (40) at non-singleton dimension 0

AliaksandrSiarohin commented 3 years ago

You should use config with 10 regions.