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.train: add verbose=1 consistently

Browse source
This commit is contained in:
Robert Sachunsky 2026-02-17 17:24:07 +01:00
parent 9b66867c21
commit 7bef8fa95a
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -471,6 +471,7 @@ def run(_config,
model.fit( model.fit(
train_gen.prefetch(tf.data.AUTOTUNE), # .repeat()?? train_gen.prefetch(tf.data.AUTOTUNE), # .repeat()??
validation_data=val_gen.prefetch(tf.data.AUTOTUNE), validation_data=val_gen.prefetch(tf.data.AUTOTUNE),
verbose=1,
epochs=n_epochs, epochs=n_epochs,
callbacks=callbacks, callbacks=callbacks,
initial_epoch=index_start) initial_epoch=index_start)
@ -544,6 +545,7 @@ def run(_config,
model.fit( model.fit(
train_ds, train_ds,
#validation_data=test_ds, #validation_data=test_ds,
verbose=1,
epochs=n_epochs, epochs=n_epochs,
callbacks=callbacks, callbacks=callbacks,
initial_epoch=index_start) initial_epoch=index_start)