diff options
Diffstat (limited to 'training/trainer/callbacks/__init__.py')
-rw-r--r-- | training/trainer/callbacks/__init__.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/training/trainer/callbacks/__init__.py b/training/trainer/callbacks/__init__.py new file mode 100644 index 0000000..80c4177 --- /dev/null +++ b/training/trainer/callbacks/__init__.py @@ -0,0 +1,29 @@ +"""The callback modules used in the training script.""" +from .base import Callback, CallbackList +from .checkpoint import Checkpoint +from .early_stopping import EarlyStopping +from .lr_schedulers import ( + LRScheduler, + SWA, +) +from .progress_bar import ProgressBar +from .wandb_callbacks import ( + WandbCallback, + WandbImageLogger, + WandbReconstructionLogger, + WandbSegmentationLogger, +) + +__all__ = [ + "Callback", + "CallbackList", + "Checkpoint", + "EarlyStopping", + "LRScheduler", + "WandbCallback", + "WandbImageLogger", + "WandbReconstructionLogger", + "WandbSegmentationLogger", + "ProgressBar", + "SWA", +] |