My project is about image classification.
I’ve trained my model on kaggle and used
learn.export() to save it as
Now, I’ve tried to load it and use it to predict on a single image, in a new Kaggle notebook. But I get this error:
AttributeError: '_DenseLayer' object has no attribute 'memory_efficient'
This is my code:
!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"))
I used fast.ai 1.0.40 to train my model and for the testing part as well.
I tried going by this solution from a post, but I still have the same issue:
!pip install --upgrade torchvision