fix loss info and typo (Closes GH-22).

2021-12-09 15:11:06 +08:00
parent 0d869161c2
commit 49cbc44069
16 changed files with 37 additions and 36 deletions
@@ -59,7 +59,7 @@ class LossAggregator():
                loss, info = loss_func(**v)
                for name, value in info.items():
                    loss_info['scalar/%s/%s' % (k, name)] = value
-                loss = loss.mean() * loss_func.loss_term_weights
+                loss = loss.mean() * loss_func.loss_term_weight
                loss_sum += loss

            else: