onPredictBatchBegin

open fun onPredictBatchBegin(batch: Int, batchSize: Int)

Called at the beginning of a batch during prediction phase.

Parameters

batch

index of batch.

batchSize

Number of samples in the current batch.