release v0.1.dev15

early epoch stopping
2026-09-12 12:40:20 +08:00 · 2019-04-23 09:08:06 -04:00 · 2019-04-23 08:57:58 -04:00 · 2019-04-23 08:46:20 -04:00 · 2019-04-23 08:27:27 -04:00 · 2019-04-23 08:26:48 -04:00
2 changed files with 13 additions and 5 deletions
@@ -270,13 +270,14 @@ class Trainer(TrainerIO):
                # ---------------
                # RUN TRAIN STEP
                # ---------------
-                self.__run_tng_batch(data_batch)
+                batch_result = self.__run_tng_batch(data_batch)
+                early_stop_epoch = batch_result == -1

                # ---------------
                # RUN VAL STEP
                # ---------------
                is_val_check_batch = (batch_nb + 1) % self.val_check_batch == 0
-                if self.fast_dev_run or is_val_check_batch:
+                if self.fast_dev_run or is_val_check_batch or early_stop_epoch:
                    self.__run_validation()

                # when batch should be saved
@@ -308,6 +309,10 @@ class Trainer(TrainerIO):
                if self.__is_function_implemented('on_batch_end'):
                    self.model.on_batch_end()

+                # end epoch early
+                if early_stop_epoch:
+                    break
+
            # hook
            if self.__is_function_implemented('on_epoch_end'):
                self.model.on_epoch_end()
@@ -322,15 +327,16 @@ class Trainer(TrainerIO):
                if stop:
                    return

+
    def __run_tng_batch(self, data_batch):
        if data_batch is None:
-            return
+            return 0

        # hook
        if self.__is_function_implemented('on_batch_start'):
            response = self.model.on_batch_start(data_batch)
            if response == -1:
-                return
+                return -1

        if self.enable_tqdm:
            self.prog_bar.update(1)
@@ -372,6 +378,8 @@ class Trainer(TrainerIO):
        if self.__is_function_implemented('on_batch_end'):
            self.model.on_batch_end()

+        return 0
+
    def __run_validation(self):
        # decide if can check epochs
        can_check_epoch = (self.current_epoch + 1) % self.check_val_every_n_epoch == 0
@@ -7,7 +7,7 @@ from setuptools import setup, find_packages
 # http://blog.ionelmc.ro/2014/05/25/python-packaging/
 setup(
    name="pytorch-lightning",
-    version='0.1.dev14',
+    version='0.1.dev15',
    description="The Keras for ML researchers using PyTorch",
    author="William Falcon",
    author_email="waf2107@columbia.edu",
Author	SHA1	Message	Date
William Falcon	0637d8e7a5	release v0.1.dev15	2019-04-23 09:08:06 -04:00
William Falcon	2514f62913	early epoch stopping	2019-04-23 08:57:58 -04:00
William Falcon	95aee7ff96	early epoch stopping	2019-04-23 08:46:20 -04:00
William Falcon	ffd6dc678c	early epoch stopping	2019-04-23 08:27:27 -04:00
William Falcon	1961a6abb2	early epoch stopping	2019-04-23 08:26:48 -04:00