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
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