diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-03 23:33:34 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-03 23:33:34 +0200 |
| commit | 07dd14116fe1d8148fb614b160245287533620fc (patch) | |
| tree | 63395d88b17a14ad453c52889fcf541e6cbbdd3e /src/training/callbacks/__init__.py | |
| parent | 704451318eb6b0b600ab314cb5aabfac82416bda (diff) | |
Working Emnist lines dataset.
Diffstat (limited to 'src/training/callbacks/__init__.py')
| -rw-r--r-- | src/training/callbacks/__init__.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/src/training/callbacks/__init__.py b/src/training/callbacks/__init__.py index 868d739..fbcc285 100644 --- a/src/training/callbacks/__init__.py +++ b/src/training/callbacks/__init__.py @@ -1 +1,19 @@ -"""TBC.""" +"""The callback modules used in the training script.""" +from .base import Callback, CallbackList, Checkpoint +from .early_stopping import EarlyStopping +from .lr_schedulers import CyclicLR, MultiStepLR, OneCycleLR, ReduceLROnPlateau, StepLR +from .wandb_callbacks import WandbCallback, WandbImageLogger + +__all__ = [ + "Callback", + "CallbackList", + "Checkpoint", + "EarlyStopping", + "WandbCallback", + "WandbImageLogger", + "CyclicLR", + "MultiStepLR", + "OneCycleLR", + "ReduceLROnPlateau", + "StepLR", +] |