Improve WebUI and add loudness normalization

2023-12-29 18:43:14 +09:00
parent b0fbcc4667
commit fc2bde673a
7 changed files with 245 additions and 148 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -658,8 +658,9 @@ def train_and_evaluate(
                )

        global_step += 1
-    gc.collect()
-    torch.cuda.empty_cache()
+    # 本家ではこれをスピードアップのために消すと書かれていたので、一応消してみる
+    # gc.collect()
+    # torch.cuda.empty_cache()
    if rank == 0:
        logger.info(f"====> Epoch: {epoch}, step: {global_step}")