Yuy1024 / pytorch-demo

A small demo for training cnn with pytorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Description

A simple demo for training cnn with pytorch. It is a pytorch clone of tf-simple-example. It support training cnn with mnist, cifar10, cifar100 and svhn.

Requirement

  • python==3.6
  • pytorch>=1.0.0
  • torchvision
  • matplotlib (optional, for plot training logs)

Run

# train
python train.py

# eval
python eval.py log/resnet_20/checkpoint_69.pk

# plot training logs
python plot_log.py log/ -m true

Training Curve

training curve

About

A small demo for training cnn with pytorch.


Languages

Language:Python 100.0%