TimeStopping

class TimeStopping(seconds: Int, verbose: Boolean) : Callback

Callback for stopping training of a model when a specified amount of time has passed.

This callback could be used to limit the training time of a model to a specified maximum duration. When the specified amount of time passes, the training of the model is stopped (if it has not already been finished or stopped) at the end of current epoch.

Constructors

TimeStopping

fun TimeStopping(seconds: Int = 86400, verbose: Boolean = false)

Functions

onEpochBegin

open fun onEpochBegin(epoch: Int, logs: TrainingHistory)

Called at the start of an epoch during training phase.

onEpochEnd

open override fun onEpochEnd(epoch: Int, event: EpochTrainingEvent, logs: TrainingHistory)

Called at the end of an epoch during training phase.

onPredictBatchBegin

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

Called at the beginning of a batch during prediction phase.

onPredictBatchEnd

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

Called at the end of a batch during prediction phase.

onPredictBegin

open fun onPredictBegin()

Called at the beginning of prediction.

onPredictEnd

open fun onPredictEnd()

Called at the end of prediction.

onTestBatchBegin

open fun onTestBatchBegin(batch: Int, batchSize: Int, logs: History)

Called at the beginning of a batch during evaluation phase. Also called at the beginning of a validation batch during validation phase, if validation data is provided.

onTestBatchEnd

open fun onTestBatchEnd(batch: Int, batchSize: Int, event: BatchEvent?, logs: History)

Called at the end of a batch during evaluation phase. Also called at the end of a validation batch during validation phase, if validation data is provided.

onTestBegin

open fun onTestBegin()

Called at the beginning of evaluation or validation.

onTestEnd

open fun onTestEnd(logs: History)

Called at the end of evaluation or validation.

onTrainBatchBegin

open fun onTrainBatchBegin(batch: Int, batchSize: Int, logs: TrainingHistory)

Called at the beginning of a batch during training phase.

onTrainBatchEnd

open fun onTrainBatchEnd(batch: Int, batchSize: Int, event: BatchTrainingEvent, logs: TrainingHistory)

Called at the end of a batch during training phase.

onTrainBegin

open override fun onTrainBegin()

Called at the beginning of training.

onTrainEnd

open override fun onTrainEnd(logs: TrainingHistory)

Called at the end of training. This method is empty. Extend this class to handle this event.