deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

whoops!

Browse Source
This commit is contained in:
James Betker 2022-06-12 21:11:34 -06:00
parent efabcf5008
commit 7a36668870
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/trainer/ExtensibleTrainer.py
View File

@ -402,7 +402,7 @@ class ExtensibleTrainer(BaseModel):
net.module.after_step(it) net.module.after_step(it)
if self.do_emas: if self.do_emas:
# When the EMA is on the CPU, only update every 10 steps to save processing time. # 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 continue
ema_params = self.emas[name].parameters() ema_params = self.emas[name].parameters()
net_params = net.parameters() net_params = net.parameters()