├── .github └── workflows │ └── tests.yml ├── .gitignore ├── CHANGELOG.md ├── LICENSE ├── README.md ├── setup.cfg ├── setup.py ├── tests ├── test_ema.py └── test_state_dict.py └── torch_ema ├── __init__.py ├── _version.py └── ema.py /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/README.md -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/tests/test_ema.py -------------------------------------------------------------------------------- /tests/test_state_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/tests/test_state_dict.py -------------------------------------------------------------------------------- /torch_ema/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/torch_ema/__init__.py -------------------------------------------------------------------------------- /torch_ema/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.3" 2 | -------------------------------------------------------------------------------- /torch_ema/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fadel/pytorch_ema/HEAD/torch_ema/ema.py --------------------------------------------------------------------------------