if return -1 from a hook that loop stopps

pytorch_lightning/models/trainer.py

@@ -327,7 +327,7 @@ class Trainer(TrainerIO):
 
         # hook
         if self.__is_function_implemented('on_batch_start'):
-            response = self.model.on_batch_start()
+            response = self.model.on_batch_start(data_batch)
             if response == -1:
                 return
 

pytorch_lightning/root_module/hooks.py

@@ -1,7 +1,7 @@
 import torch
 
 class ModelHooks(torch.nn.Module):
-    def on_batch_start(self):
+    def on_batch_start(self, data_batch):
         pass
 
     def on_batch_end(self):