Skip to content

Latest commit

 

History

History
10 lines (8 loc) · 310 Bytes

README.md

File metadata and controls

10 lines (8 loc) · 310 Bytes

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