Pytorch 如何使用fastai保存的模型？

我在google colab中训练了我的模型，并在我的计算机中下载了.pkl文件。现在，我该如何使用它？如何加载.pkl文件以及是否需要安装fastai才能使其工作

如何加载.pkl文件

假设您已使用保存模型，则可以使用互补方法

我需要安装fastai才能工作吗

是的，如果以这种方式保存，则需要

fastai

。您还可以通过以下方式保存

learner

中包含的PyTorch模型本身：

torch.save(learner.model, "/path/to/model.pt") # or save it's state_dict, better option
model = torch.load("/path/to/model.pt")

---

无论哪种方式，您都需要这些库，因为

pickle

存储数据，但

class

定义和创建必须按代码提供。

它返回一个错误。”fastai.vision.learner没有属性“load”。我使用

learn.export

以.pkl格式导出文件。现在我似乎无法使用它。@RomanRegmi

fastai

version？这是最新的吗？我用的是google colab中的那个。所以，我不确定。然而，我试着在GoogleColab中使用保存的模型。我还是犯了个错误。一个主要版本已经过时了您的依赖项。刚刚检查了

colab

，它是

1.0.61

，而最新版本是

2.3.1

（根据）。请更新你的附件，它们真的过时了！