How to use train model for predict unseen data?

1

Hi I am new here and after creating my first model I dont know how to use it.Everywhere I am finding only tutorial on train data but nor sure how to use my trained model.

2

You can find the code here. You need to first define your model, then create an object and load your trained model:

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()

then you can feed your data into model.