解决Pytorch参数不匹配报错RuntimeError: Error(s) in loading state_dict for ResNet:size mismatch for fc.weight
解决Pytorch参数不匹配报错RuntimeError: Error(s) in loading state_dict for ResNet:size mismatch for fc.weight: copying a param with shape torch.Size([2, 512]) from checkpoint, the shape in current model is torch.Size([100, 512]).的问题
原创
2024-06-18 16:15:22 ·
471 阅读 ·
0 评论