added early epoch stopping hook

src/pytorch-lightning/models/trainer.py

@@ -267,7 +267,8 @@ class Trainer(TrainerIO):
                     break
 
                 # give model a chance to end epoch early
-                if self.model.should_stop_epoch():
+                if self.model.should_stop_epoch:
+                    self.model.should_stop_epoch = False
                     break
 
                 # ---------------

src/pytorch-lightning/root_module/hooks.py

@@ -19,5 +19,3 @@ class ModelHooks(torch.nn.Module):
     def on_post_performance_check(self):
         pass
 
-    def should_stop_epoch(self):
-        return False

src/pytorch-lightning/root_module/root_module.py

@@ -24,6 +24,7 @@ class RootModule(GradInformation, ModelIO, OptimizerConfig, ModelHooks):
         self.overfit = hparams.overfit
         self.gradient_clip = hparams.gradient_clip
         self.num = 2
+        self.should_stop_epoch = False
 
         # track if gpu was requested for checkpointing
         self.on_gpu = False