TSN在pytorch1.0以上版本出现size mismatch for conv1_7x7_s2_bn.weight等错误
在pytorch1.0运行TSN时出现size mismatch错误RuntimeError: Error(s) in loading state_dict for BNInception: size mismatch for conv1_7x7_s2_bn.weight: copying a param with shape torch.Size([1, 64]) from checkpoint, the shape in current model is torch.Size([6
原创
2020-09-20 15:08:29 ·
2610 阅读 ·
5 评论