I am doing it like above, but if I have to take my saved model to someplace else and make predictions out of it, do I need to take the training data with it too ? as I understand from
ImageClassifierData.from_paths(PATH, ...)
What I am trying to do is, export the model I built in a format which I will be able to use somewhere else. Like we use the ResNet34 and similar ImageNet models to train our network.
you should be able to load the state directly into the model without building a Learner first (I haven’t tried it yet, but i dont see why it wouldn’t work)
from torch_imports import load_model
load_model(resnet34, path_to_your_saved_model)