qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-02-20 16:32:03 +01:00
Code Issues Projects Releases Wiki Activity

training: fix epoch size calculation

Browse source
This commit is contained in:
Robert Sachunsky 2026-01-29 03:01:14 +01:00
parent 29a0f19cee
commit d1e8a02fd4
1 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/eynollah/training/train.py
View file

@ -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)