diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-20 22:18:35 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-08-20 22:18:35 +0200 |
| commit | 1f459ba19422593de325983040e176f97cf4ffc0 (patch) | |
| tree | 89fef442d5dbe0c83253e9566d1762f0704f64e2 /src/training/trainer/callbacks/__init__.py | |
| parent | 95cbdf5bc1cc9639febda23c28d8f464c998b214 (diff) | |
A lot of stuff working :D. ResNet implemented!
Diffstat (limited to 'src/training/trainer/callbacks/__init__.py')
| -rw-r--r-- | src/training/trainer/callbacks/__init__.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/training/trainer/callbacks/__init__.py b/src/training/trainer/callbacks/__init__.py new file mode 100644 index 0000000..5942276 --- /dev/null +++ b/src/training/trainer/callbacks/__init__.py @@ -0,0 +1,21 @@ +"""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 .progress_bar import ProgressBar +from .wandb_callbacks import WandbCallback, WandbImageLogger + +__all__ = [ + "Callback", + "CallbackList", + "Checkpoint", + "EarlyStopping", + "WandbCallback", + "WandbImageLogger", + "CyclicLR", + "MultiStepLR", + "OneCycleLR", + "ProgressBar", + "ReduceLROnPlateau", + "StepLR", +] |