diff options
Diffstat (limited to 'torch_ema/__init__.py')
| -rw-r--r-- | torch_ema/__init__.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/torch_ema/__init__.py b/torch_ema/__init__.py index 9732013..6cf180f 100644 --- a/torch_ema/__init__.py +++ b/torch_ema/__init__.py @@ -1 +1,3 @@ -from .ema import * +from .ema import ExponentialMovingAverage + +__all__ = [ExponentialMovingAverage] |