【pytorch】保存模型

参考资料：Pytorch 保存模型与加载模型 - 知乎

这两种方法区别在于：

一个保存整个网络，一个只保存网络参数

只保存网络参数时，要重新加载网络之前，需要重新定义model

# 保存整个网络
torch.save(net, PATH) 
# 保存网络中的参数, 速度快，占空间少
torch.save(net.state_dict(),PATH)
#--------------------------------------------------
#针对上面一般的保存方法，加载的方法分别是：
model_dict=torch.load(PATH)
model_dict=model.load_state_dict(torch.load(PATH))