python - DenseNet - 新图像失败的预测

Question

我的项目是关于图像分类的。我已经在 kaggle 上训练了我的模型，learn.export()并将其保存为export.pkl.

现在，我尝试在新的 Kaggle 笔记本中加载它并使用它来预测单个图像。但我得到这个错误：

AttributeError: '_DenseLayer' object has no attribute 'memory_efficient'

这是我的代码：

!pip install fastai==1.0.40
from fastai.basic_train import *
from pathlib import Path
path = Path()
model_export = load_learner(path/'../input/trainingmodel')
from fastai.vision import * # open_image
cat, tensor, probs = model_export.predict(open_image(f"../input/test/imagepng"))

我使用 fast.ai 1.0.40 来训练我的模型和测试部分。我尝试从一个帖子中使用这个解决方案，但我仍然遇到同样的问题：

!pip install --upgrade torchvision

标签： pythontensorflowfast-aidensenet