From fd1d3efd18d4cfc8a3a4db842369656a499f583a Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Fri, 9 Feb 2024 17:30:57 +0900
Subject: [PATCH] Fix colab for updated setting

---
 colab.ipynb    | 11 ++++++-----
 webui_train.py |  2 +-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/colab.ipynb b/colab.ipynb
index 370ca15..f5cab3e 100644
--- a/colab.ipynb
+++ b/colab.ipynb
@@ -4,7 +4,7 @@
       "cell_type": "markdown",
       "metadata": {},
       "source": [
-        "# Style-Bert-VITS2 (ver 2.1) のGoogle Colabでの学習\n",
+        "# Style-Bert-VITS2 (ver 2.2) のGoogle Colabでの学習\n",
         "\n",
         "Google Colab上でStyle-Bert-VITS2の学習を行うことができます。\n",
         "\n",
@@ -195,7 +195,7 @@
       "source": [
         "## 3. 学習の前処理\n",
         "\n",
-        "次に学習の前処理を行います。必要なパラメータをここで指定します。次のセルに設定等を入力して実行してください。"
+        "次に学習の前処理を行います。必要なパラメータをここで指定します。次のセルに設定等を入力して実行してください。「～～かどうか」は`True`もしくは`False`を指定してください。"
       ]
     },
     {
@@ -222,7 +222,7 @@
         "# 保存頻度。何ステップごとにモデルを保存するか。分からなければデフォルトのままで。\n",
         "save_every_steps = 1000\n",
         "\n",
-        "# 音声ファイルの音量を正規化するかどうか。`True`もしくは`False`\n",
+        "# 音声ファイルの音量を正規化するかどうか\n",
         "normalize = False\n",
         "\n",
         "# 音声ファイルの開始・終了にある無音区間を削除するかどうか\n",
@@ -255,7 +255,6 @@
         "    batch_size=batch_size,\n",
         "    epochs=epochs,\n",
         "    save_every_steps=save_every_steps,\n",
-        "    bf16_run=False,\n",
         "    num_processes=2,\n",
         "    normalize=normalize,\n",
         "    trim=trim,\n",
@@ -263,7 +262,9 @@
         "    freeze_JP_bert=False,\n",
         "    freeze_ZH_bert=False,\n",
         "    freeze_style=False,\n",
-        "    use_jp_extra=use_jp_extra\n",
+        "    use_jp_extra=use_jp_extra,\n",
+        "    val_per_lang=0,\n",
+        "    log_interval=200,\n",
         ")"
       ]
     },
diff --git a/webui_train.py b/webui_train.py
index f401580..3ea4bb4 100644
--- a/webui_train.py
+++ b/webui_train.py
@@ -176,7 +176,7 @@ def preprocess_text(model_name, use_jp_extra, val_per_lang):
         "--val-path",
         val_path,
         "--val-per-lang",
-        val_per_lang,
+        str(val_per_lang),
     ]
     if use_jp_extra:
         cmd.append("--use_jp_extra")