https://github.com/wenmuzhou/pytorch_mnist
learning pytorch with mnist dataset
https://github.com/wenmuzhou/pytorch_mnist
mnist pytorch pytorch-tutorials
Last synced: about 1 month ago
JSON representation
learning pytorch with mnist dataset
- Host: GitHub
- URL: https://github.com/wenmuzhou/pytorch_mnist
- Owner: WenmuZhou
- Created: 2018-04-16T13:30:36.000Z (about 8 years ago)
- Default Branch: master
- Last Pushed: 2018-09-21T04:51:56.000Z (over 7 years ago)
- Last Synced: 2024-12-28T16:35:06.285Z (over 1 year ago)
- Topics: mnist, pytorch, pytorch-tutorials
- Language: Python
- Homepage:
- Size: 453 KB
- Stars: 3
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# pytorch_mnist
learning pytorch with mnist dataset
# training
```sh
python3 train.py
```
# predict
```sh
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