pytorch读取与保存网络模型
方法一：
只保存网络模型的参数，而不保存网络的结构
保存：
torch.save(model.state_dict(), 'model_params.pth') tor
阅读背景：
pytorch读取与保存网络模型
来源：互联网
分享到：
