Fix colab for updated setting

This commit is contained in:
litagin02
2024-02-09 17:30:57 +09:00
parent 527908626b
commit fd1d3efd18
2 changed files with 7 additions and 6 deletions

View File

@@ -4,7 +4,7 @@
"cell_type": "markdown", "cell_type": "markdown",
"metadata": {}, "metadata": {},
"source": [ "source": [
"# Style-Bert-VITS2 (ver 2.1) のGoogle Colabでの学習\n", "# Style-Bert-VITS2 (ver 2.2) のGoogle Colabでの学習\n",
"\n", "\n",
"Google Colab上でStyle-Bert-VITS2の学習を行うことができます。\n", "Google Colab上でStyle-Bert-VITS2の学習を行うことができます。\n",
"\n", "\n",
@@ -195,7 +195,7 @@
"source": [ "source": [
"## 3. 学習の前処理\n", "## 3. 学習の前処理\n",
"\n", "\n",
"次に学習の前処理を行います。必要なパラメータをここで指定します。次のセルに設定等を入力して実行してください。" "次に学習の前処理を行います。必要なパラメータをここで指定します。次のセルに設定等を入力して実行してください。「~~かどうか」は`True`もしくは`False`を指定してください。"
] ]
}, },
{ {
@@ -222,7 +222,7 @@
"# 保存頻度。何ステップごとにモデルを保存するか。分からなければデフォルトのままで。\n", "# 保存頻度。何ステップごとにモデルを保存するか。分からなければデフォルトのままで。\n",
"save_every_steps = 1000\n", "save_every_steps = 1000\n",
"\n", "\n",
"# 音声ファイルの音量を正規化するかどうか。`True`もしくは`False`\n", "# 音声ファイルの音量を正規化するかどうか\n",
"normalize = False\n", "normalize = False\n",
"\n", "\n",
"# 音声ファイルの開始・終了にある無音区間を削除するかどうか\n", "# 音声ファイルの開始・終了にある無音区間を削除するかどうか\n",
@@ -255,7 +255,6 @@
" batch_size=batch_size,\n", " batch_size=batch_size,\n",
" epochs=epochs,\n", " epochs=epochs,\n",
" save_every_steps=save_every_steps,\n", " save_every_steps=save_every_steps,\n",
" bf16_run=False,\n",
" num_processes=2,\n", " num_processes=2,\n",
" normalize=normalize,\n", " normalize=normalize,\n",
" trim=trim,\n", " trim=trim,\n",
@@ -263,7 +262,9 @@
" freeze_JP_bert=False,\n", " freeze_JP_bert=False,\n",
" freeze_ZH_bert=False,\n", " freeze_ZH_bert=False,\n",
" freeze_style=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",
")" ")"
] ]
}, },

View File

@@ -176,7 +176,7 @@ def preprocess_text(model_name, use_jp_extra, val_per_lang):
"--val-path", "--val-path",
val_path, val_path,
"--val-per-lang", "--val-per-lang",
val_per_lang, str(val_per_lang),
] ]
if use_jp_extra: if use_jp_extra:
cmd.append("--use_jp_extra") cmd.append("--use_jp_extra")