diff --git a/README.md b/README.md index 8a13a0c..deb2bad 100644 --- a/README.md +++ b/README.md @@ -24,6 +24,12 @@ mfa model download dictionary english_us_arpa mfa model download g2p english_us_arpa ``` +Other dependencies needed before training: +```bash +pip install torch +micromamba install -c pytorch torchaudio numpy scikit-learn tqdm tensorboard matplotlib +``` + Dataset Download is now integrated in the training script. ```python training/train.py --config training/recipes/tcn_config.toml```