prisma-ai / torch2coreml

Torch7 -> CoreML
MIT License
384 stars 54 forks source link

How to use Trained model in Pytorch? #24

Open Karthik-89 opened 6 years ago

Karthik-89 commented 6 years ago

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?