在Fastai中，可以使用learn.save方法保存訓(xùn)練好的模型，使用learn.load方法加載已保存的模型。
保存模型的示例代碼如下：
learn.save('model_name')
加載模型的示例代碼如下：
learn.load('model_name')
其中，model_name是保存和加載模型時(shí)指定的模型文件名。在保存模型時(shí)，會(huì)生成一個(gè).pth文件保存模型的參數(shù)，可以通過(guò)指定不同的文件名保存不同版本的模型。加載模型時(shí)，只需指定對(duì)應(yīng)的模型文件名即可加載已保存的模型。
億速云公眾號(hào)
手機(jī)網(wǎng)站二維碼