def build_callbacks_classifier(callbacks_config):
experiment_timing = ExperimentTiming()
model_checkpoints = ModelCheckpoint(**callbacks_config['model_checkpoint'])
lr_scheduler = ExponentialLRScheduler(**callbacks_config['lr_scheduler'])
validation_monitor = ValidationMonitor(**callbacks_config['validation_monitor'])
training_monitor = TrainingMonitor(**callbacks_config['training_monitor'])
neptune_monitor = NeptuneMonitor()
return CallbackList(
callbacks=[experiment_timing, model_checkpoints, lr_scheduler, training_monitor, validation_monitor,
neptune_monitor])
def build_callbacks_localizer(callbacks_config):
experiment_timing = ExperimentTiming()
model_checkpoints = ModelCheckpoint(**callbacks_config['model_checkpoint'])
lr_scheduler = ExponentialLRScheduler(**callbacks_config['lr_scheduler'])
training_monitor = TrainingMonitor(**callbacks_config['training_monitor'])
validation_monitor = ValidationMonitor(**callbacks_config['validation_monitor'])
neptune_monitor = NeptuneMonitorLocalizer(**callbacks_config['neptune_monitor'])
plot_bounding_box = PlotBoundingBoxPredictions(**callbacks_config['bounding_box_predictions'])
return CallbackList(
callbacks=[experiment_timing, model_checkpoints, lr_scheduler, training_monitor, validation_monitor,
neptune_monitor, plot_bounding_box])