Skip to content

Files

Latest commit

9aad148 · Oct 27, 2022

History

History
This branch is 63 commits behind pytorch/examples:main.

mnist_rnn

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