From 7a366688708a0c9fe9af270bab623411a686e5de Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sun, 12 Jun 2022 21:11:34 -0600
Subject: [PATCH] whoops!
---
codes/trainer/ExtensibleTrainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/codes/trainer/ExtensibleTrainer.py b/codes/trainer/ExtensibleTrainer.py
index 90ed33c2..40564d6b 100644
--- a/codes/trainer/ExtensibleTrainer.py
+++ b/codes/trainer/ExtensibleTrainer.py
@@ -402,7 +402,7 @@ class ExtensibleTrainer(BaseModel):
net.module.after_step(it)
if self.do_emas:
# When the EMA is on the CPU, only update every 10 steps to save processing time.
- if self.ema_on_cpu and step % 5 != 0:
+ if self.ema_on_cpu and it % 10 != 0:
continue
ema_params = self.emas[name].parameters()
net_params = net.parameters()