From 3f657d8ede0f40463dca792a4c1b6a57fb37146a Mon Sep 17 00:00:00 2001 From: ZDisket <30500847+ZDisket@users.noreply.github.com> Date: Sat, 18 Mar 2023 20:37:34 -0300 Subject: [PATCH] add train, check notebook --- neo-tts/tacotron2-cac/README.md | 8 +- neo-tts/tacotron2-cac/check_model.py | 73 ++ neo-tts/tacotron2-cac/train.py | 26 +- neo-tts/tacotron2-cac/training.ipynb | 1311 ++++++++++++++++++++++++++ 4 files changed, 1416 insertions(+), 2 deletions(-) create mode 100644 neo-tts/tacotron2-cac/check_model.py create mode 100644 neo-tts/tacotron2-cac/training.ipynb diff --git a/neo-tts/tacotron2-cac/README.md b/neo-tts/tacotron2-cac/README.md index dd00d83..f132bb2 100644 --- a/neo-tts/tacotron2-cac/README.md +++ b/neo-tts/tacotron2-cac/README.md @@ -72,6 +72,7 @@ Train your model with ``` python3 train.py --dataset DATASET ``` +Check `training.ipynb` or the up-to-date [Colab Notebook](https://colab.research.google.com/drive/1ra5DRIyohXSNiT-3miBWkTZbaGDTczOJ?usp=sharing). # TensorBoard @@ -86,7 +87,12 @@ Use python3 infer_arpa.py --mode single --dataset SingleSpk22 --checkpoint tacotron2.pth --out-path MyAud --text "there are no faces" ``` Or check `inference.ipynb` for more lean, hands-on notebook version - +# Checking model +Use +``` +python3 check_model.py --dataset SingleSpk22 --checkpoint tacotron2.pth +``` +This script will try to load a checkpoint to CPU. If the file doesn't exist, it will exit with `2` return code. If it does but there's an error loading, it will exit with `1`. Otherwise, if the model is loaded well, it will exit with `0` to serve TensorBoard on your localhost.