From 1f459ba19422593de325983040e176f97cf4ffc0 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Thu, 20 Aug 2020 22:18:35 +0200
Subject: A lot of stuff working :D. ResNet implemented!

---
 src/training/trainer/callbacks/__init__.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 src/training/trainer/callbacks/__init__.py

(limited to 'src/training/trainer/callbacks/__init__.py')

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",
+]
-- 
cgit v1.2.3-70-g09d2