Error when load the trained model of BasicVSR++
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 ?