torch.save() 函数讲解 保存不同类型的model，model.state_dict()格式

最近做项目在加载molde的时候始终无法加载，报错缺少文件。后来仔细查看才发现是model保存时候的问题。
torch.save(model,path) 会将model的参数、框架都保存到路径path中，但是在加载model的时候可能会因为包版本的不同报错，所以当保存所有模型参数时，需要将模型构造相关代码文件放在相同路径，否则在load的时候无法索引到model的框架。
建议：使用state_dict()模式保存model，torch.save(model.state_dict(),path)，这样保存为字典模式，可以直接load。