removed self.model refs

2019-06-26 18:21:17 -04:00 · 2019-06-26 18:21:17 -04:00 · 11b4bc3fbc
parent 787f523a71
commit 11b4bc3fbc
1 changed files with 1 additions and 1 deletions
--- 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