early epoch stopping

2026-09-09 11:32:07 +08:00 · 2019-04-23 08:27:27 -04:00
parent 1961a6abb2
commit ffd6dc678c
1 changed files with 1 additions and 1 deletions
@@ -326,7 +326,7 @@ class Trainer(TrainerIO):

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

        # hook
        if self.__is_function_implemented('on_batch_start'):