onTrainEnd

open override fun onTrainEnd(logs: TrainingHistory)