From 310c8d7eb7d97d958c80655ba60e74fa9cc40404 Mon Sep 17 00:00:00 2001 From: Felix Draxler Date: Fri, 17 Mar 2023 16:15:23 +0100 Subject: [PATCH] Expose SkipBatch from .trainable --- src/lightning_trainable/__init__.py | 2 +- src/lightning_trainable/trainable/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lightning_trainable/__init__.py b/src/lightning_trainable/__init__.py index 23e0dc3..f8203f1 100644 --- a/src/lightning_trainable/__init__.py +++ b/src/lightning_trainable/__init__.py @@ -4,5 +4,5 @@ from . import modules from . import trainable -from .trainable import Trainable, TrainableHParams +from .trainable import Trainable, TrainableHParams, SkipBatch from .hparams import HParams diff --git a/src/lightning_trainable/trainable/__init__.py b/src/lightning_trainable/trainable/__init__.py index 369ce7e..ac8b179 100644 --- a/src/lightning_trainable/trainable/__init__.py +++ b/src/lightning_trainable/trainable/__init__.py @@ -1 +1 @@ -from .trainable import Trainable, TrainableHParams +from .trainable import Trainable, TrainableHParams, SkipBatch