we should modify the main_train to convert and load in the same script to reduce the number of steps.