Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
大约 1 年 前同步成功

通知 115
Star 4999
Fork 1114
P
PaddleClas
提交 c32e2b09 编写于 作者: Y Yang Nie 提交者： Tingquan Gao
浏览文件
操作

Revert "Speedup EMA" 

This reverts commit 35fc732dadac4761852b18512b5c5df8785e36df.
上级 001cdb09
显示空白变更内容
内联 并排
Showing with 4 addition and 7 deletion
ppcls/utils/ema.py
浏览文件 @ c32e2b09
...@@ -32,14 +32,11 @@ class ExponentialMovingAverage(): ...@@ -32,14 +32,11 @@ class ExponentialMovingAverage():
@paddle.no_grad() @paddle.no_grad()
def _update(self, model, update_fn): def _update(self, model, update_fn):
for ema_v, model_v in zip(self.module.state_dict().values(), for ema_v, model_v in zip(self.module.state_dict().values(), model.state_dict().values()):
model.state_dict().values()): ema_v.set_value(update_fn(ema_v, model_v))
ema_v.set_value(update_fn(ema_v.numpy(), model_v.numpy()))
def update(self, model): def update(self, model):
self._update( self._update(model, update_fn=lambda e, m: self.decay * e + (1. - self.decay) * m)
model,
update_fn=lambda e, m: self.decay * e + (1. - self.decay) * m)
def set(self, model): def set(self, model):
self._update(model, update_fn=lambda e, m: m) self._update(model, update_fn=lambda e, m: m)
\ No newline at end of file
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或