如何使用 fastai 保存的模型？

Question

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

Answer 1

你需要使用：

from fastbook import load_learner

learn_inference = load_learner(filename)

见：https://docs.fast.ai/learner.html#load_learner 用于文档。

Answer 2

如何加载 .pkl 文件

假设您已使用 learner.save 保存模型，您可以使用互补的 learner.load 方法。

我是否需要安装 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 定义和创建必须以代码方式提供。