torch-ema

PyTorch library for computing moving averages of model parameters.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install torch-ema

Dependencies

Releases

Version Released Bookworm
Python 3.11
Trixie
Python 3.13
Files
0.3 2021-11-17
0.2 2021-11-17

Issues with this package?

Page last updated 2026-09-03 09:22:04 UTC