callbacks.rst

torchbearer.callbacks

Base Classes

torchbearer.bases.Callback

torchbearer.callbacks.callbacks

Imaging

Main Classes

torchbearer.callbacks.imaging.imaging

Deep Inside Convolutional Networks

torchbearer.callbacks.imaging.inside_cnns

Model Checkpointers

torchbearer.callbacks.checkpointers

Logging

torchbearer.callbacks.csv_logger

torchbearer.callbacks.printer

Tensorboard, Visdom and Others

torchbearer.callbacks.tensor_board

torchbearer.callbacks.live_loss_plot.LiveLossPlot

Early Stopping

torchbearer.callbacks.early_stopping

torchbearer.callbacks.terminate_on_nan

Gradient Clipping

torchbearer.callbacks.gradient_clipping

Learning Rate Schedulers

torchbearer.callbacks.torch_scheduler

Weight Decay

torchbearer.callbacks.weight_decay

Weight / Bias Initialisation

torchbearer.callbacks.init

Regularisers

torchbearer.callbacks.cutout.Cutout

torchbearer.callbacks.cutout.RandomErase

torchbearer.callbacks.cutout.CutMix

torchbearer.callbacks.mixup.Mixup

torchbearer.callbacks.between_class.BCPlus

torchbearer.callbacks.sample_pairing.SamplePairing

torchbearer.callbacks.label_smoothing.LabelSmoothingRegularisation

Unpack State

torchbearer.callbacks.unpack_state

Decorators

Main

The main callback decorators simply take a function and bind it to a callback point, returning the result.

torchbearer.callbacks.decorators.on_init

torchbearer.callbacks.decorators.on_start

torchbearer.callbacks.decorators.on_start_epoch

torchbearer.callbacks.decorators.on_start_training

torchbearer.callbacks.decorators.on_sample

torchbearer.callbacks.decorators.on_forward

torchbearer.callbacks.decorators.on_criterion

torchbearer.callbacks.decorators.on_backward

torchbearer.callbacks.decorators.on_step_training

torchbearer.callbacks.decorators.on_end_training

torchbearer.callbacks.decorators.on_start_validation

torchbearer.callbacks.decorators.on_sample_validation

torchbearer.callbacks.decorators.on_forward_validation

torchbearer.callbacks.decorators.on_criterion_validation

torchbearer.callbacks.decorators.on_step_validation

torchbearer.callbacks.decorators.on_end_validation

torchbearer.callbacks.decorators.on_end_epoch

torchbearer.callbacks.decorators.on_checkpoint

torchbearer.callbacks.decorators.on_end

Utility

Alongside the base callback decorators that simply bind a function to a callback point, Torchbearer has a number of utility decorators that help simplify callback construction.

torchbearer.callbacks.decorators