onTrainBatchEnd
open override fun onTrainBatchEnd(batch: Int, batchSize: Int, event: BatchTrainingEvent, logs: TrainingHistory)
Content copied to clipboard
Called at the end of a batch during training phase.
Parameters
batch
index of batch within the current epoch.
batchSize
Number of samples in the current batch.
event
Metric and loss values for this batch.
logs
training history, containing full information about previous epochs.