From 73c104c80aa46d1568ece534a8d1e1bcf48a71c3 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 10:17:08 -0400
Subject: [PATCH] updated test docs

---
 pytorch_lightning/models/trainer.py | 3 +++
 tests/test_models.py                | 1 -
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 7c45f35a71..3b129d78ae 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -150,6 +150,9 @@ class Trainer(TrainerIO):
         self.use_ddp = False
         self.use_dp = False
 
+        # bookkeeping
+        self.avg_loss = 0
+
 
         # gpus come in as a string.
         # if gpus = -1 then use all available devices
diff --git a/tests/test_models.py b/tests/test_models.py
index e0b0a36e1a..6bac63cfd9 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -227,7 +227,6 @@ def test_amp_gpu_dp():
         experiment=get_exp(),
         max_nb_epochs=1,
         train_percent_check=0.4,
-        val_percent_check=0.4,
         gpus=[0, 1],
         distributed_backend='dp',
         use_amp=True