stouputils.data_science.models.keras_utils.callbacks package

Custom callbacks for Keras models.

Features:

• Learning rate finder callback for finding the optimal learning rate

• Warmup scheduler callback for warmup training

• Progressive unfreezing callback for unfreezing layers during training (incompatible with model.fit(), need a custom training loop)

• Tqdm progress bar callback for better training visualization

• Model checkpoint callback that only starts checkpointing after a given number of epochs

Submodules

• stouputils.data_science.models.keras_utils.callbacks.colored_progress_bar module
  • ColoredProgressBar
    • ColoredProgressBar.desc
    • ColoredProgressBar.track_epochs
    • ColoredProgressBar.show_lr
    • ColoredProgressBar.latest_val_loss
    • ColoredProgressBar.latest_lr
    • ColoredProgressBar.batch_count
    • ColoredProgressBar.update_frequency
    • ColoredProgressBar.color
    • ColoredProgressBar.pbar
    • ColoredProgressBar.epochs
    • ColoredProgressBar.steps
    • ColoredProgressBar.total
    • ColoredProgressBar.params
    • ColoredProgressBar.on_train_begin()
    • ColoredProgressBar.on_batch_end()
    • ColoredProgressBar.on_epoch_end()
    • ColoredProgressBar.on_train_end()
• stouputils.data_science.models.keras_utils.callbacks.learning_rate_finder module
  • LearningRateFinder
    • LearningRateFinder.min_lr
    • LearningRateFinder.max_lr
    • LearningRateFinder.total_updates
    • LearningRateFinder.update_per_epoch
    • LearningRateFinder.update_interval
    • LearningRateFinder.lr_mult
    • LearningRateFinder.learning_rates
    • LearningRateFinder.losses
    • LearningRateFinder.best_lr
    • LearningRateFinder.best_loss
    • LearningRateFinder.initial_weights
    • LearningRateFinder.on_train_begin()
    • LearningRateFinder._update_lr_and_track_metrics()
    • LearningRateFinder.on_batch_end()
    • LearningRateFinder.on_epoch_end()
    • LearningRateFinder.on_train_end()
• stouputils.data_science.models.keras_utils.callbacks.model_checkpoint_v2 module
  • ModelCheckpointV2
• stouputils.data_science.models.keras_utils.callbacks.progressive_unfreezing module
  • ProgressiveUnfreezing
    • ProgressiveUnfreezing.base_model
    • ProgressiveUnfreezing.steps_per_epoch
    • ProgressiveUnfreezing.epochs
    • ProgressiveUnfreezing.reset_weights
    • ProgressiveUnfreezing.reset_optimizer_function
    • ProgressiveUnfreezing.update_per_epoch
    • ProgressiveUnfreezing.update_interval
    • ProgressiveUnfreezing.progressive_freeze
    • ProgressiveUnfreezing.total_steps
    • ProgressiveUnfreezing.fraction_unfrozen
    • ProgressiveUnfreezing.losses
    • ProgressiveUnfreezing._all_layers
    • ProgressiveUnfreezing._initial_trainable
    • ProgressiveUnfreezing._initial_weights
    • ProgressiveUnfreezing._last_update_step
    • ProgressiveUnfreezing.on_train_begin()
    • ProgressiveUnfreezing._update_layers()
    • ProgressiveUnfreezing._track_loss()
    • ProgressiveUnfreezing.on_batch_begin()
    • ProgressiveUnfreezing.on_batch_end()
    • ProgressiveUnfreezing.on_epoch_begin()
    • ProgressiveUnfreezing.on_epoch_end()
    • ProgressiveUnfreezing.on_train_end()
    • ProgressiveUnfreezing.get_results()
• stouputils.data_science.models.keras_utils.callbacks.warmup_scheduler module
  • WarmupScheduler
    • WarmupScheduler.warmup_epochs
    • WarmupScheduler.initial_lr
    • WarmupScheduler.target_lr
    • WarmupScheduler.on_epoch_begin()