Giters

qiuqiangkong / pytorch_study

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Mnist classification with pytorch.

Examples of basic pytorch usage methods.

Author: Qiuqiang Kong (q.kong@surrey.ac.uk)

Run

CUDA_VISIBLE_DEVICES=0 python mnist_dnn_pt.py train

CUDA_VISIBLE_DEVICES=0 python main_dnn_pt_torch_generator.py train

CUDA_VISIBLE_DEVICES=0 python mnist_cnn_pt.py train

About

Languages

Language:Python 100.0%