From 51305697c1d075f236a5298d1a11e0b4e6a41220 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 25 Jun 2019 20:21:11 -0400 Subject: [PATCH] updated args --- docs/source/examples/example_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/examples/example_model.py b/docs/source/examples/example_model.py index 8f319a27c8..fe45ae5091 100644 --- a/docs/source/examples/example_model.py +++ b/docs/source/examples/example_model.py @@ -116,7 +116,7 @@ class ExampleModel(RootModule): accs.append(output['val_acc']) val_loss_mean /= len(outputs) - tqdm_dic = {'val_loss': val_loss_mean, 'val_acc': np.mean(accs)} + tqdm_dic = {'val_loss': val_loss_mean, 'val_acc': torch.mean(accs).item()} return tqdm_dic def update_tng_log_metrics(self, logs):