Skip to content

deep models for small image classification datasets

Notifications You must be signed in to change notification settings

falcondai/mnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 

Repository files navigation

mnist

Supervised learning with feedforward deep models on small standard datasets. The main feature is the modular model import with model-specific arguments.

usage

python train.py -m cnn --m-activation sigmoid

TODO

  • add dataset CIFAR-10
  • add dataset SVHN
  • add ResNet-like model
  • add DenseNet-like model

author

Falcon Dai ([email protected])