diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-12-07 22:54:04 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-12-07 22:54:04 +0100 |
| commit | 25b5d6983d51e0e791b96a76beb7e49f392cd9a8 (patch) | |
| tree | 526ba739714b3d040f7810c1a6be3ff0ba37fdb1 /src/training/trainer/callbacks/__init__.py | |
| parent | 5529e0fc9ca39e81fe0f08a54f257d32f0afe120 (diff) | |
Segmentation working!
Diffstat (limited to 'src/training/trainer/callbacks/__init__.py')
| -rw-r--r-- | src/training/trainer/callbacks/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/training/trainer/callbacks/__init__.py b/src/training/trainer/callbacks/__init__.py index e1bd858..95ec142 100644 --- a/src/training/trainer/callbacks/__init__.py +++ b/src/training/trainer/callbacks/__init__.py @@ -7,7 +7,7 @@ from .lr_schedulers import ( SWA, ) from .progress_bar import ProgressBar -from .wandb_callbacks import WandbCallback, WandbImageLogger +from .wandb_callbacks import WandbCallback, WandbImageLogger, WandbSegmentationLogger __all__ = [ "Callback", @@ -17,6 +17,7 @@ __all__ = [ "LRScheduler", "WandbCallback", "WandbImageLogger", + "WandbSegmentationLogger", "ProgressBar", "SWA", ] |