You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I trained the BasicVSR++ by using 'options/train/BasicVSRPP/train_BasicVSRPP_REDS.yml'.
Then running the inference code from 'inference/inference_basicvsrpp.py',
it occur following error when "model.load_state_dict(torch.load(args.model_path)['params'], strict=True)" :
Dear @LiangbinXie ,
I trained the BasicVSR++ by using 'options/train/BasicVSRPP/train_BasicVSRPP_REDS.yml'.
Then running the inference code from 'inference/inference_basicvsrpp.py',
it occur following error when "model.load_state_dict(torch.load(args.model_path)['params'], strict=True)" :
Error(s) in loading state_dict for BasicVSRPlusPlus:
Unexpected key(s) in state_dict: "deform_align.backward_1.weight", "deform_align.backward_1.bias", "deform_align.backward_1.conv_offset.0.weight", "deform_align.backward_1.conv_offset.0.bias", "deform_align.backward_1.conv_offset.2.weight", "deform_align.backward_1.conv_offset.2.bias", "deform_align.backward_1.conv_offset.4.weight", "deform_align.backward_1.conv_offset.4.bias", "deform_align.backward_1.conv_offset.6.weight", "deform_align.backward_1.conv_offset.6.bias", "deform_align.forward_1.weight", "deform_align.forward_1.bias", "deform_align.forward_1.conv_offset.0.weight", "deform_align.forward_1.conv_offset.0.bias", "deform_align.forward_1.conv_offset.2.weight", "deform_align.forward_1.conv_offset.2.bias", "deform_align.forward_1.conv_offset.4.weight", "deform_align.forward_1.conv_offset.4.bias", "deform_align.forward_1.conv_offset.6.weight", "deform_align.forward_1.conv_offset.6.bias", "deform_align.backward_2.weight", "deform_align.backward_2.bias", "deform_align.backward_2.conv_offset.0.weight", "deform_align.backward_2.conv_offset.0.bias", "deform_align.backward_2.conv_offset.2.weight", "deform_align.backward_2.conv_offset.2.bias", "deform_align.backward_2.conv_offset.4.weight", "deform_align.backward_2.conv_offset.4.bias", "deform_align.backward_2.conv_offset.6.weight", "deform_align.backward_2.conv_offset.6.bias", "deform_align.forward_2.weight", "deform_align.forward_2.bias", "deform_align.forward_2.conv_offset.0.weight", "deform_align.forward_2.conv_offset.0.bias", "deform_align.forward_2.conv_offset.2.weight", "deform_align.forward_2.conv_offset.2.bias", "deform_align.forward_2.conv_offset.4.weight", "deform_align.forward_2.conv_offset.4.bias", "deform_align.forward_2.conv_offset.6.weight", "deform_align.forward_2.conv_offset.6.bias".
It seem like deform_align weight unable load to the model, how to fix it ?
The text was updated successfully, but these errors were encountered: