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.
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.