-
Notifications
You must be signed in to change notification settings - Fork 0
davidharket/FashionMNIST_Computer_Vision_Model
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
In this Jupyter notebook i have created three different PyTorch models for computer vision that have been tested on the opensource FashionMNIST dataset. The last model (model 2) was the most successfull one, reaching a 90% accuracy on the test predictions. This was a convolutional network based on the Tiny VVG architecture. The parameters should be saved to the state_dict, however, to achieve this i first set the learning rate of the optimizer (Stochastic gradient descent) to 0.1 for one round of training followed by another round of training with the learining rate 0.01. I used a basic cross entropy loss for the loss function.
About
Implementation of the tinyVGG architecture via PyTorch in Machine learning model trained on FashionMNIST dataset. 90% accuracy achieved (not saved to state_dict)
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published