在Fastai中,可以使用learn.save
方法保存訓練好的模型,使用learn.load
方法加載已保存的模型。
保存模型的示例代碼如下:
learn.save('model_name')
加載模型的示例代碼如下:
learn.load('model_name')
其中,model_name
是保存和加載模型時指定的模型文件名。在保存模型時,會生成一個.pth
文件保存模型的參數,可以通過指定不同的文件名保存不同版本的模型。加載模型時,只需指定對應的模型文件名即可加載已保存的模型。