pytorch_classifiers
pytorch_classifiers copied to clipboard
Almost any Image classification problem using pytorch
![image](https://user-images.githubusercontent.com/16069257/119592689-e9bbff00-be13-11eb-9284-21f9b6d2d518.png) I was able to run Makefile.train to train, and save some models to the Models/ folder. But when I run Makefile.predict to load those models to test, the errors...
Stylized the file. Improved the sentence structure. And made the overall look and feel of the readme file better.
Even after training for many epoch accuracy is remaining 0. Epoch 50/199 ---------- 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3378/3378 [20:02
Add multilayer to the model after the hidden layer, something like this ![1558435010957](https://user-images.githubusercontent.com/36812649/59367063-fcd58180-8d58-11e9-8fbb-57753fc738cc.png)
This fix can help pytorch_classifiers run on 0.4.1 version.
It will be nice if we know the complexity and running time to train this models.