gaia.callbacks package

GAIA Training Callbacks

class gaia.callbacks.CallbackManager(callbacks=None)

Bases: object

Manages training callbacks for GAIA trainer

__init__(callbacks=None)

on_train_begin(logs=None)

Called at the beginning of training

on_train_end(logs=None)

Called at the end of training

on_train_error(logs=None, error=None)

Called when training encounters an error

on_epoch_begin(epoch, logs=None)

Called at the beginning of each epoch

on_epoch_end(epoch, logs=None)

Called at the end of each epoch

on_batch_begin(batch, logs=None)

Called at the beginning of each batch

on_batch_end(batch, logs=None)

Called at the end of each batch

class gaia.callbacks.EarlyStopping(patience=10, min_delta=0.0, monitor='val_loss')

Bases: object

Early stopping callback

__init__(patience=10, min_delta=0.0, monitor='val_loss')

on_epoch_end(epoch, logs=None)