onPredictBatchEnd

open override fun onPredictBatchEnd(batch: Int, batchSize: Int)