深度学习模型训练

出现问题原因:PyTorch版本环境不一致、torch.nn.DataParallel()关键字不匹配等,提出的解决办法是把strict参数赋False,如下:

解决办法:

FNet.load_state_dict(Fmodel_info['state_dict'])
# 改成:
FNet.load_state_dict(Fmodel_info['state_dict'], False)

其他的几个解决办法:

https://blog.csdn.net/Marmaladei/article/details/128773695

https://blog.csdn.net/Qz574662449/article/details/121633818

Pytorch总体问题解决方法:

https://zhuanlan.zhihu.com/p/365082327

猜你喜欢

转载自blog.csdn.net/qq_43681154/article/details/129647887