updated args

2026-09-09 11:32:07 +08:00 · 2019-06-26 17:49:58 -04:00
parent 1ee6d21db2
commit 71cd8f549d
2 changed files with 11 additions and 7 deletions
@@ -75,7 +75,8 @@ class ExampleModel(RootModule):
        # return loss_val, tqdm_dic

        output = OrderedDict({
-            'tng_loss': loss_val,
+            'loss': loss_val,
+            'tqdm_metrics': {}
        })
        return output