remove the retrain part of sample code