From 9fc66026f1c6031eba3bbad3439dc15e4b2bd3f7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Pr=C3=A6sius?= <sebastian.devel@gmail.com>
Date: Thu, 22 Aug 2019 23:44:06 +0200
Subject: [PATCH] train = False in test_dataloader (#162)

A small change to the CoolModel example.
Now test_dataloader returns the MNIST test dataset.
---
 docs/LightningModule/RequiredTrainerInterface.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md
index d7863c7c4e..a369841051 100644
--- a/docs/LightningModule/RequiredTrainerInterface.md
+++ b/docs/LightningModule/RequiredTrainerInterface.md
@@ -80,7 +80,7 @@ class CoolModel(pl.LightningModule):
     @pl.data_loader
     def test_dataloader(self):
         # OPTIONAL
-        return DataLoader(MNIST(os.getcwd(), train=True, download=True, transform=transforms.ToTensor()), batch_size=32)
+        return DataLoader(MNIST(os.getcwd(), train=False, download=True, transform=transforms.ToTensor()), batch_size=32)
 ```
 ---   
 ### How do these methods fit into the broader training?