a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Don't accumulate losses for metrics when the loss isn't a tensor

Browse Source
This commit is contained in:
James Betker 2020-10-03 11:03:55 -06:00
parent 19a4075e1e
commit 8197fd646f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/steps/steps.py
View File

@ -152,6 +152,7 @@ class ConfigurableStep(Module):
l = loss(self.training_net, local_state) l = loss(self.training_net, local_state)
total_loss += l * self.weights[loss_name] total_loss += l * self.weights[loss_name]
# Record metrics. # Record metrics.
if isinstance(l, torch.Tensor):
self.loss_accumulator.add_loss(loss_name, l) self.loss_accumulator.add_loss(loss_name, l)
for n, v in loss.extra_metrics(): for n, v in loss.extra_metrics():
self.loss_accumulator.add_loss("%s_%s" % (loss_name, n), v) self.loss_accumulator.add_loss("%s_%s" % (loss_name, n), v)