python main.py train