Fix premature abort warning message

mittagessen · Nov 28, 2023 · ba4c81f · ba4c81f
1 parent 8ed41dc
commit ba4c81f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/kraken/ketos/pretrain.py b/kraken/ketos/pretrain.py
@@ -290,7 +290,7 @@ def pretrain(ctx, batch_size, pad, output, spec, load, freq, quit, epochs,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)
 

diff --git a/kraken/ketos/recognition.py b/kraken/ketos/recognition.py
@@ -324,7 +324,7 @@ def train(ctx, batch_size, pad, output, spec, append, load, freq, quit, epochs,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)
 

diff --git a/kraken/ketos/ro.py b/kraken/ketos/ro.py
@@ -260,7 +260,7 @@ def rotrain(ctx, batch_size, output, load, freq, quit, epochs, min_epochs, lag,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)
 

diff --git a/kraken/ketos/segmentation.py b/kraken/ketos/segmentation.py
@@ -357,7 +357,7 @@ def segtrain(ctx, output, spec, line_width, pad, load, freq, quit, epochs,
         logger.warning('Model did not improve during training.')
         ctx.exit(1)
 
-    if not model.epoch:
+    if not model.current_epoch:
         logger.warning('Training aborted before end of first epoch.')
         ctx.exit(1)