From d1e8a02fd4a50d61d3101db8a9ae870201bde194 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Thu, 29 Jan 2026 03:01:14 +0100 Subject: [PATCH] training: fix epoch size calculation --- src/eynollah/training/train.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/eynollah/training/train.py b/src/eynollah/training/train.py index 9c638ea..1e2ab3e 100644 --- a/src/eynollah/training/train.py +++ b/src/eynollah/training/train.py @@ -394,17 +394,16 @@ def run(_config, if save_interval: callbacks.append(SaveWeightsAfterSteps(save_interval, dir_output, _config)) - _log.info("training on %d batches in %d epochs", - len(os.listdir(dir_flow_train_imgs)) // n_batch - 1, - n_epochs) - _log.info("validating on %d batches", - len(os.listdir(dir_flow_eval_imgs)) // n_batch - 1) + steps_train = len(os.listdir(dir_flow_train_imgs)) // n_batch # - 1 + steps_val = len(os.listdir(dir_flow_eval_imgs)) // n_batch + _log.info("training on %d batches in %d epochs", steps_train, n_epochs) + _log.info("validating on %d batches", steps_val) model.fit( train_gen, - steps_per_epoch=len(os.listdir(dir_flow_train_imgs)) // n_batch - 1, + steps_per_epoch=steps_train, validation_data=val_gen, #validation_steps=1, # rs: only one batch?? - validation_steps=len(os.listdir(dir_flow_eval_imgs)) // n_batch - 1, + validation_steps=steps_val, epochs=n_epochs, callbacks=callbacks)