Create main_train.py

Main file to train the data