WenmuZhou / pytorch_mnist

learning pytorch with mnist dataset

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch_mnist

learning pytorch with mnist dataset

training

python3 train.py

predict

python3 predict.py -f /data/datasets/mnist/test/0/0_1.png -m AlexNet.pkl

the code Calculate 1000 times to average, if you only need to calculate once, comment it out in the code.

requirements

  • pytorch==0.4
  • torchvision
  • time
  • tensorboardX
  • cv2
  • numpy
ezoic increase your site revenue

About

learning pytorch with mnist dataset


Languages

Language:Python 100.0%