From 11b4bc3fbc6202ccb43954d8a7ee360906fb7cf2 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 26 Jun 2019 18:21:17 -0400 Subject: [PATCH] removed self.model refs --- 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 dadc2ad25a..59fb6830a2 100644 --- a/docs/source/examples/example_model.py +++ b/docs/source/examples/example_model.py @@ -112,12 +112,12 @@ class ExampleModel(RootModule): """ val_loss_mean = 0 val_acc_mean = 0 - pdb.set_trace() for output in outputs: val_loss_mean += output['val_loss'] val_acc_mean += output['val_acc'] val_loss_mean /= len(outputs) + val_acc_mean /= len(outputs) tqdm_dic = {'val_loss': val_loss_mean.item(), 'val_acc': val_acc_mean.item()} return tqdm_dic