onPredictBatchEnd

open fun onPredictBatchEnd(batch: Int, batchSize: Int)
Content copied to clipboard

Called at the end of a batch during prediction phase.

Parameters

batch

index of batch within the current epoch.

batchSize

Number of samples in the current batch.