Modify training model