diff --git a/infer-web.py b/infer-web.py index 6155818..39149d7 100644 --- a/infer-web.py +++ b/infer-web.py @@ -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 diff --git a/train/utils.py b/train/utils.py index 783f251..657940f 100644 --- a/train/utils.py +++ b/train/utils.py @@ -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