This repo contains code to train four different neural network architectures for the purposes of classification on the Fashion MNIST dataset. To get started, install pytorch, torchvision and wandb. Then, simple specify the desired networks in the main function in train.py and run the file. Training progress is logged using wandb by default - if this isn't desired, run 'wandb off' in the console prior to training.
-
Notifications
You must be signed in to change notification settings - Fork 0
b-hahn/fashion_mnist_classifiers
About
Fashion MNIST Classification using deep neural networks.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published