fadel/pytorch_ema
Tiny PyTorch library for maintaining a moving average of a collection of parameters.
PythonMIT
Issues
- 2
- 1
- 5
How to use it in pytorch-lightning?
#12 opened by Devoe-97 - 0
Running mean and running var use EMA?
#13 opened by zhaohui-yang - 1
- 3
- 4
Usage of weakref broke pickling/unpickling
#10 opened by toshas - 2
Whether the model with ema needs bo be fixed in the validation phase —— “model(with ema).eval()”?
#3 opened by TIanCat - 1
issues about add with torch.no_grad():
#1 opened by liang233