To run the code : python main_train.py --data ./data --max_layer 1 --epochs 1 For help : ``python main_train.py --help```