Closed LinLi-DL closed 4 years ago
您好, 我想用你训练好的模型再训练一下我自己的一个小数据集,但是直接报错了。Error: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv1.weight",.......
slover中的代码是这句: self.net_bone.base.load_state_dict(torch.load(self.config.resnet))
在网上找了一些办法,但是都没解决,请帮忙指导下,谢谢。
您好, 我想用你训练好的模型再训练一下我自己的一个小数据集,但是直接报错了。Error: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv1.weight",.......
slover中的代码是这句: self.net_bone.base.load_state_dict(torch.load(self.config.resnet))
在网上找了一些办法,但是都没解决,请帮忙指导下,谢谢。