pip install -r requirements.txt
python main.py
# CUDA_VISIBLE_DEVICES=2 python main.py # to specify GPU id to ex. 2
martinzwm / cs197-lec16-starter Goto Github PK
View Code? Open in Web Editor NEWThis project forked from rajpurkar/cs197-lec14-15-starter