README.md

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

For help : ``python main_train.py --help```