Fastai怎么保存和加载训练好的模型
lewis
2020-01-29
22次阅读
在Fastai中,可以使用learn.save方法保存训练好的模型,使用learn.load方法加载已保存的模型。
保存模型的示例代码如下:
learn.save('model_name')
加载模型的示例代码如下:
learn.load('model_name')
其中,model_name是保存和加载模型时指定的模型文件名。在保存模型时,会生成一个.pth文件保存模型的参数,可以通过指定不同的文件名保存不同版本的模型。加载模型时,只需指定对应的模型文件名即可加载已保存的模型。

发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。