Pytorch:保存和提取模型

原文地址 分类目录——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