auracaster/piper
3
0
Fork 0
mirror of https://github.com/pstrueb/piper.git synced 2026-05-31 17:07:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed training notebook.

Browse Source
This commit is contained in:
Mateo Cedillo
2023-07-03 22:04:51 -05:00
parent c830cdd3e1
commit 48f1d9237d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/piper_multilingual_training_notebook.ipynb
+1 -1
View File
@@ -300,7 +300,7 @@
"if action == \"Continue training\":\n", "if action == \"Continue training\":\n",
" if os.path.exists(f\"{output_dir}/lightning_logs/version_0/checkpoints/last.ckpt\"):\n", " if os.path.exists(f\"{output_dir}/lightning_logs/version_0/checkpoints/last.ckpt\"):\n",
" ft_command = f'--resume_from_checkpoint \"{output_dir}/lightning_logs/version_0/checkpoints/last.ckpt\" '\n", " ft_command = f'--resume_from_checkpoint \"{output_dir}/lightning_logs/version_0/checkpoints/last.ckpt\" '\n",
" print(f\"Continuing {model_name}'s training at: {output_dir}/lightning_logs/version_0/checkpoints/last.ckpt}\")\n", " print(f\"Continuing {model_name}'s training at: {output_dir}/lightning_logs/version_0/checkpoints/last.ckpt\")\n",
" else:\n", " else:\n",
" raise Exception(\"Training cannot be continued as there is no checkpoint to continue at.\")\n", " raise Exception(\"Training cannot be continued as there is no checkpoint to continue at.\")\n",
"elif action == \"finetune\":\n", "elif action == \"finetune\":\n",