原文地址
分类目录——Pytorch
这里在我写的 Pytorch：一个简单的神经网络——分类 的基础上进行对模型的保存和提取操作。
为了检验保存的模型就是训练好的模型，先用训练好的模型做个测试
print(net(torch.tensor([2., 5.]))) # 用模型判断（2，5）这个点所属的类别
# tensor([0.9701, 0.0299], grad_fn=) # 数据0、1两个类的概率
Pytorch对模型的报讯提取有两种方式
torch.save(net, 'classnet.pkl')
这种方式将会保存整个模型，包括模型的结构和参数
# 保存
torch.save(n
