use python run.py --dataset *NAME* to select dataset for training Feel free to email me for any question.