Experimental changes to get it 1:1 step count

This commit is contained in:
kalomaze
2023-07-23 13:29:32 -05:00
parent 62ab1778a8
commit 658e647b3f
2 changed files with 1 additions and 2 deletions

View File

@@ -948,8 +948,6 @@ def set_log_interval(exp_dir, batch_size12):
if wav_files:
sample_size = len(wav_files)
log_interval = math.ceil(sample_size / batch_size12)
if log_interval > 1:
log_interval += 1
return log_interval

View File

@@ -395,6 +395,7 @@ def get_hparams(init=True):
hparams.if_cache_data_in_gpu = args.if_cache_data_in_gpu
hparams.data.training_files = "%s/filelist.txt" % experiment_dir
args.log_interval += 1
hparams.train.log_interval = args.log_interval
# Update log_interval in the 'train' section of the config dictionary