本文共 376 字，大约阅读时间需要 1 分钟。

古月居 PyTorch入门：一起从零搭建神经网络 七、PyTorch车牌字符识别项目四

1、如何把训练好的模型打包成文件

我们在搭建好数据集和神经网络之后，每次运行如果都提前训练，那将会花费相当长的时间。所以我们就需要将训练好的模型参数进行保存。这样做的好处是不用每次都训练，可以直接调用训练好的文件包。

代码：

torch.save(cnn,'CNN.pth')#保存模型torch.save(cnn.state_dict(),'CNN.pth')#保存模型参数

cnn =CNN_h.CNN()cnn.load_state_dict(torch.load('CNN.pth'))#加载模型

代码全集：

2、结果测试

运行正确率大概在50%-80%。

转载地址：http://xnexi.baihongyu.com/