BasicSR icon indicating copy to clipboard operation
BasicSR copied to clipboard

Error when load the trained model of BasicVSR++

Open vincentweisen1357 opened this issue 1 year ago • 0 comments

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 ?

vincentweisen1357 avatar Aug 30 '24 03:08 vincentweisen1357