refactor main.py, add model&additionalTrainset