Fixed issue with EMA climbing decay instead of reducing it.

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

No files matched your search

+1 -1
View File
@@ -133,7 +133,7 @@ class ExponentialMovingAverage:
if param.dtype != torch.float32:
param_float = param_float.to(torch.float32)
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
if self.use_feedback:
+1 -1
View File
@@ -1 +1 @@
VERSION = "0.13.8"
VERSION = "0.13.9"