In this notebook we have understood the image classification task in pytorch.
Tuning the various hyper parameters of the model the accuracy metric was measured.
- Number of Epoch
- Activation Function (Relu, Leaky_Relu)
- Optimizer (Adam, SGD)
- Number of Hidden Layers (1,2,3)
- Number of Neurons in the Hidden Layer (25,50,100,256)
| Parameters | Accuracy |
|---|---|
| 15910 | 93.80% |
| 20175 | 93.90% |
| 39760 | 96.00% |
| 40785 | 96.30% |
| 84060 | 97.00% |
| 242762 | 97.40% |
Detailed model parameter monitoring sheet