up train.py