setCustomTrainLoop