fadel / pytorch_ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.

fadel/pytorch_ema Issues

Is the EMA used for training or just for tracking the EMA
Updated a year ago1
How to use it in pytorch-lightning?
Updated a year ago5
Running mean and running var use EMA?
Updated a year ago
is model.eval() necessary within or outside context manager
Updated 2 years ago1
Code does not work if some parameters are not trainable.
Closed 3 years ago3
Usage of weakref broke pickling/unpickling
Closed 3 years ago4
Whether the model with ema needs bo be fixed in the validation phase —— “model(with ema).eval()”?
Closed 3 years ago2
issues about add with torch.no_grad():
Closed 5 years ago1