Thanks Zachary for your reply.
I am not fully familiar with pyTorch (unlike fastai). I loaded the model, but the data type is dict which has all the fastai model layers and info (is this accurate? [Solved] Using a fastai-trained model with plain Pytorch).
How can I make the model from this dictionary? and how can I make predictions using it?
In fastai everything was simple:
fastai_prediction = learn_nn.predict(data.iloc[0])
where learn_nn is the NN model. What is the equivalent process in pyTorch?
I think what you refer to as prediction tensor is equivalent of data.iloc[0] in this example, but I am stuck in one step before that.