Added --split option to train.py