Skip to content

Latest commit

 

History

History
 
 

mnist_rnn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

Example of MNIST using RNN

Motivation

Create pytorch example similar to Official Tensorflow Keras RNN example using MNIST here

pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py  # to specify GPU id to ex. 2