updated args

2019-06-25 20:22:21 -04:00 · 2019-06-25 20:22:21 -04:00 · f49c2f4c25
parent 51305697c1
commit f49c2f4c25
1 changed files with 3 additions and 3 deletions
--- a/docs/source/examples/example_model.py
+++ b/docs/source/examples/example_model.py
@ -110,13 +110,13 @@ class ExampleModel(RootModule):
        :return:
        """
        val_loss_mean = 0
-        accs = []
+        val_acc_mean = 0
        for output in outputs:
            val_loss_mean += output['val_loss']
-            accs.append(output['val_acc'])
+            val_acc_mean += output['val_acc']

        val_loss_mean /= len(outputs)
-        tqdm_dic = {'val_loss': val_loss_mean, 'val_acc': torch.mean(accs).item()}
+        tqdm_dic = {'val_loss': val_loss_mean.item(), 'val_acc': val_acc_mean.item()}
        return tqdm_dic

    def update_tng_log_metrics(self, logs):