hosa.callbacks.early_stopping.EarlyStoppingAtMinLoss.on_predict_batch_end

EarlyStoppingAtMinLoss.on_predict_batch_end(batch, logs=None)

Called at the end of a batch in predict methods.

Subclasses should override for any actions to run.

Note that if the steps_per_execution argument to compile in tf.keras.Model is set to N, this method will only be called every N batches.

Parameters

• batch – Integer, index of batch within the current epoch.

• logs – Dict. Aggregated metric results up until this batch.