From 0d31b9a22913bee60301d08bbb14c34c0e8c2d59 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 13 Aug 2019 11:38:35 -0400
Subject: [PATCH] updated readme

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 6e408e59..1c9fed0c 100644
--- a/README.md
+++ b/README.md
@@ -87,7 +87,7 @@ class CoolModel(pl.LightningModule):
         y_hat = self.forward(x)
         return {'loss': F.cross_entropy(y_hat, y)}
 
-    def validation_step(self, batch, batch_nb, dataloader_i):
+    def validation_step(self, batch, batch_nb):
         # OPTIONAL
         x, y = batch
         y_hat = self.forward(x)