[ EDIT ] : this issue is solved (see thread [ SOLVED ] Get prediction from the ImageNet model without creating a databunch).
Hello @myltykritik.
I’m searching an answer in the forum about how to use a pretrained model without changing anything. For example, I would like to use resnet34 with the 1000 classes of ImageNet.
It is easy to get the weights (for example, model = models.resnet34(pretrained=True)), but how to use it after to get a prediction through model.predict(img)? I got the following error.
AttributeError: 'ResNet' object has no attribute 'predict'
I guess I need to build a learner with create_cnn(data, model) for example but for that I need a databunch data with my 1000 classes… and I do not want to download the 138 GB of Imagenet training dataset
Any advice?