Minor fixes.

This commit is contained in:
Mateo Cedillo
2023-06-03 11:33:23 -05:00
parent b18c1fcae4
commit 385adce106
@@ -5,7 +5,7 @@
"colab": {
"provenance": [],
"gpuType": "T4",
"authorship_tag": "ABX9TyNiXaJjWsxcoKeHr7HeL2eD",
"authorship_tag": "ABX9TyN5orawFZNYEhhbtJPe+1fd",
"include_colab_link": true
},
"kernelspec": {
@@ -250,7 +250,11 @@
"dataset_format = \"ljspeech\" #@param [\"ljspeech\", \"mycroft\"]\n",
"#@markdown ---\n",
"#@markdown ### Is this a single speaker dataset? Otherwise, uncheck:\n",
"single-speaker = True #@param {type:\"boolean\"}\n",
"single_speaker = True #@param {type:\"boolean\"}\n",
"if single_speaker:\n",
" force_sp = \" --single-speaker\"\n",
"else:\n",
" force_sp = \"\"\n",
"#@markdown ---\n",
"#@markdown ### Select the sample rate of the dataset\n",
"sample_rate = \"16000\" #@param [\"16000\", \"22050\"]\n",
@@ -261,8 +265,8 @@
" --input-dir /content/dataset \\\n",
" --output-dir {output_dir} \\\n",
" --dataset-format {dataset_format} \\\n",
" --sample-rate {sample_rate}|\n",
" --single-speaker {single-speaker}"
" --sample-rate {sample_rate} \\\n",
" {force_sp}"
],
"metadata": {
"cellView": "form",
@@ -319,14 +323,14 @@
"#@markdown * high - 22.05Khz audio, 28-32M params\n",
"quality = \"x-low\" #@param [\"high\", \"x-low\", \"medium\"]\n",
"#@markdown ---\n",
"#@markdown ### For how many steps to save training checkpoints?\n",
"#@markdown ### For how many epochs to save training checkpoints?\n",
"checkpoint_epochs = 25 #@param {type:\"integer\"}\n",
"#@markdown ---\n",
"#@markdown ### Step interval to generate model samples\n",
"log_every_n_steps = 250 #@param {type:\"integer\"}\n",
"#@markdown ---\n",
"#@markdown ### training epochs\n",
"max_epochs = 5000 #@param {type:\"integer\"}\n",
"max_epochs = 10000 #@param {type:\"integer\"}\n",
"#@markdown ---"
],
"metadata": {
@@ -356,6 +360,8 @@
"cell_type": "code",
"source": [
"#@markdown ## 6. Train\n",
"\n",
"#@markdown **Note: Remember to empty the trash of your Drive from time to time to avoid a lot of space consumption when saving the models.**\n",
"!python -m piper_train \\\n",
" --dataset-dir \"{output_dir}\" \\\n",
" --accelerator 'gpu' \\\n",