Skip to content

b-hahn/fashion_mnist_classifiers

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

fashion_mnist_classifiers

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.

About

Fashion MNIST Classification using deep neural networks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages