modify training code
trained model