search

AliaksandrSiarohin / motion-cosegmentation

Reference code for "Motion-supervised Co-Part Segmentation" paper
Other
653 stars 142 forks source link

Problems when training the model #39

Open Mumuwei opened 3 years ago

Mumuwei commented 3 years ago

When training the model, I encountered the following problems：

in logger.py line 35 module.statedict()['affine.weight'].copy(checkpoint['kp_detector']['jacobian.weight']) RuntimeError: The expanded size of the tensor (60) must match the existing size (40) at non-singleton dimension 0. Target sizes: [60, 35, 7, 7]. Tensor sizes: [40, 35, 7, 7]

zhan-xu commented 2 years ago

how did you solve the problem?

zhan-xu commented 2 years ago

i think we need to modify num_kp in first-order code to match num_segments in this code