updated test docs

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

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