search

Harry-KIT / Image-Classification-on-small-datasets-in-Pytorch

This repo is special for those who want to start learning computer vision related tasks such as image classification.
36 stars 7 forks source link

Test model code #2

Closed tahirjhan closed 1 year ago

tahirjhan commented 1 year ago

Hi, Thanks for sharing your code. The code is well-written and easy to follow. However, I did not find the test code. How do you test your final model after training? Thanks.

Harry-KIT commented 1 year ago

Hi @tahirjhan. Thank you for your interest. I solved your problem. Please check test.py file