diff options
| author | Linux-cpp-lisp <1473644+Linux-cpp-lisp@users.noreply.github.com> | 2021-11-17 15:41:41 -0500 |
|---|---|---|
| committer | Linux-cpp-lisp <1473644+Linux-cpp-lisp@users.noreply.github.com> | 2021-11-17 15:41:41 -0500 |
| commit | 30a1306f2e5dc6aa91b4b64c2a3acd1bb3b0d7b6 (patch) | |
| tree | 1204efe7c93c741136829e80628b1b42ac603e8f /torch_ema/__init__.py | |
| parent | 3985995e523aa25dd3cff7e7984130eef90a4282 (diff) | |
handle non-trainable parameters
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] |