From 658e647b3ffeb9cbe9f29b98cb4a0f29e71f6ad0 Mon Sep 17 00:00:00 2001
From: kalomaze <66376113+kalomaze@users.noreply.github.com>
Date: Sun, 23 Jul 2023 13:29:32 -0500
Subject: [PATCH] Experimental changes to get it 1:1 step count

---
 infer-web.py   | 2 --
 train/utils.py | 1 +
 2 files changed, 1 insertion(+), 2 deletions(-)

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