如何加载FastAI导出的pkl文件？

Question

我已经使用 Google Colab 的免费 GPU 训练了我的 resnet 模型，因为我的笔记本电脑没有 GPU。 你可以在这里找到 - https://colab.research.google.com/drive/1FjwI26-fv1h3w5hlg-M5Fl_2nvWBvFzq?usp=sharing

现在，终于可以使用我创建的模型进行预测了。但我面临的主要问题是模型的加载。当然，我会将

pkl

或

pth

文件加载到我的本地环境中，并对其调用

predict()

方法，但显然，为了加载模型，您需要

Learner

类本身的对象。

我的情况应该是

cnn_learner

类的对象。为了制作该类的对象，我需要定义所有内容 -

ImageDataLoaders

并加载图像，只有这样，我才能通过

cnn_learner

制作

model = cnn_learner(dls, resnet18, metrics=error_rate

类的对象

dls

将是

ImageDataLoaders

的对象

我该如何解决这个问题？因为老实说，这对我来说似乎有点违反直觉。即使是 fastai 文档也表明我在导出模型时需要所有代码才能加载模型。但导出经过全面训练的模型的主要目的是在我希望的任何环境中使用它，因为它经过全面训练，我只需调用

predict()

方法就可以得到我的预测！

此外，我已经使用

pkl

以

model.export

格式保存模型，使用

pth

以

model.save

格式保存模型

如果有人可以帮助我，我将不胜感激。

非常感谢！

什么都试过了。 浏览了文档并自己试验了文档代码，但运气不佳。