保存模型
虽然没有明白是什么原因,但是找到了解决方法:
将torch.save(model, path) # 直接保存整个模型方法改为torch.save(model.state_dict(), path) # 保存模型的参数
载入模型
相应的,载入模型时将model = torch.load(path) # 直接加载模型方法改为
model = Model() # 先初始化一个模型
model.load_state_dict(torch.load(path)) # 再加载模型参数
————————————————
版权声明:本文为CSDN博主「我才是一卓」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/weixin_43667077/article/details/108961672