From 2e6c87c8e35916031a7abd84bd24ed6f6c505f67 Mon Sep 17 00:00:00 2001
From: Vadim Bereznyuk <kuynzereb@gmail.com>
Date: Thu, 24 Oct 2019 14:47:08 +0300
Subject: [PATCH] Set total number of batches in progress bar while testing

---
 pytorch_lightning/trainer/trainer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
index cbe9af374a120..a5beecc4b2b9f 100644
--- a/pytorch_lightning/trainer/trainer.py
+++ b/pytorch_lightning/trainer/trainer.py
@@ -432,6 +432,9 @@ def run_pretrain_routine(self, model):
 
         # when testing requested only run test and return
         if self.testing:
+            if self.show_progress_bar:
+                self.progress_bar.reset(self.nb_test_batches)
+
             self.run_evaluation(test=True)
             return