diff options
Diffstat (limited to 'src/training/callbacks/__init__.py')
-rw-r--r-- | src/training/callbacks/__init__.py | 19 |
1 files changed, 0 insertions, 19 deletions
diff --git a/src/training/callbacks/__init__.py b/src/training/callbacks/__init__.py deleted file mode 100644 index fbcc285..0000000 --- a/src/training/callbacks/__init__.py +++ /dev/null @@ -1,19 +0,0 @@ -"""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", -] |