__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: 6cf180fb591fc4eae3ab0eb5c4d0e6e364bfe526 (plain)

1
2
3

from .ema import ExponentialMovingAverage

__all__ = [ExponentialMovingAverage]