evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed issue with EMA climbing decay instead of reducing it.

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-09-12 21:00:31 -06:00
1 parent 547593449b
commit 881143cbe8
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

toolkit/ema.py
+1 -1
View File
@@ -133,7 +133,7 @@ class ExponentialMovingAverage:
if param.dtype != torch.float32: if param.dtype != torch.float32:
param_float = param_float.to(torch.float32) param_float = param_float.to(torch.float32)
gap = (s_param_float - param_float) gap = (s_param_float - param_float)
s_param_float.add_(gap * one_minus_decay) s_param_float.sub_(gap * one_minus_decay)
update_param = False update_param = False
if self.use_feedback: if self.use_feedback:
version.py
+1 -1
View File
@@ -1 +1 @@
VERSION = "0.13.8" VERSION = "0.13.9"