Go to home
train.py