Skip to content

Latest commit

 

History

History
29 lines (19 loc) · 352 Bytes

README.md

File metadata and controls

29 lines (19 loc) · 352 Bytes

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