mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only log the name of the first network in the total_loss training set

Browse Source
This commit is contained in:
James Betker 2020-09-12 16:07:09 -06:00
parent fb595e72a4
commit 5b85f891af
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/models/steps/steps.py
View File

@ -133,7 +133,7 @@ class ConfigurableStep(Module):
self.loss_accumulator.add_loss(loss_name, l)
for n, v in loss.extra_metrics():
self.loss_accumulator.add_loss("%s_%s" % (loss_name, n), v)
self.loss_accumulator.add_loss("%s_total" % (self.step_opt['training'],), total_loss)
self.loss_accumulator.add_loss("%s_total" % (self.step_opt['training'][0],), total_loss)
# Scale the loss down by the accumulation factor.
total_loss = total_loss / self.env['mega_batch_factor']