added some annotation on trainmodelthread