torch2coreml icon indicating copy to clipboard operation
torch2coreml copied to clipboard

How to use Trained model in Pytorch?

Open Karthik-89 opened this issue 6 years ago • 0 comments

Hi I have trained a model using images. Now I want to test it with a single image.How to use my model weights. Its saved as a pth file. I'm new to python and deep learning. I used to work on R. In R once the model is trained, model.predict / predict command will give us the predictions. How its done in Pytorch?

Karthik-89 avatar Apr 10 '18 07:04 Karthik-89