add cleanupTraining
add cleanupTraining