如何在Fastai中重用经过训练的模型进行预测?

时间:2020-07-22 06:06:55

标签: python deep-learning computer-vision google-colaboratory fast-ai

我已经使用fast.ai训练了一个模型以检测14个不同的类,为此我使用了google colab Gpu并使用fun 2> >(tee -a -i log) >> log 导出了模型 然后,我尝试在CPU上本地使用此模型进行预测,这是我正在使用的代码。

learner.export(path, "Plantvillage.pkl").

一切正常,但最后我发现了以下错误。

-> 535类型(自己)。名称,名称)) 536 537 def setattr (自身,名称,值):

AttributeError:“ ReLU”对象没有属性“阈值”

0 个答案:

没有答案