Skip to content

Latest commit

 

History

History

mnist

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 

EasyTorch Example - MNIST Classification

Train

  • CPU
easytrain -c configs/mnist_cpu_cfg.py
  • GPU (1x)
easytrain -c configs/mnist_1x_cfg.py --devices 0

Validate

  • CPU
python validate.py -c configs/mnist_cpu_cfg.py
  • GPU (1x)
python validate.py -c configs/mnist_1x_cfg.py --devices 0