__init__.py « torch_ema - pytorch_ema - Tiny PyTorch library for maintaining a moving average of a collection of parameters.

about summary refs log tree commit diff

path: root/torch_ema/__init__.py

blob: 91c6bbf704fcf1315f0e75d9ed95702134c5abfa (about) (plain)

from ._version import __version__
from .ema import ExponentialMovingAverage

__all__ = [__version__, ExponentialMovingAverage]