Task Oriented Model Training