BasicSR icon indicating copy to clipboard operation
BasicSR copied to clipboard

使用torch.load 预训练模型为什么不行?

Open Yzs007 opened this issue 1 year ago • 0 comments

RuntimeError: Error(s) in loading state_dict for SAFMN: Missing key(s) in state_dict: "to_feat.weight", "to_feat.bias", "feats.0.norm1.weight", "feats.0.norm1.bias", "feats.0.norm2.weight", "feats.0.norm2.bias", "feats.0.safm.mfr.0.weight", "feats.0.safm.mfr.0.bias", "feats.0.safm.mfr.1.weight", "feats.0.safm.mfr.1.bias", "feats.0.safm.mfr.2.weight", "feats.0.safm.mfr.2.bias", "feats.0.safm.mfr.3.weight", "feats.0.safm.mfr.3.bias", "feats.0.safm.aggr.weight", Unexpected key(s) in state_dict: "params".

Yzs007 avatar Mar 16 '23 06:03 Yzs007