mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Update README.md
This commit is contained in:
parent
4cac6b76d9
commit
e491663a76
@ -12,4 +12,26 @@ This recipe provides a VITS model trained on the VCTK dataset.
|
|||||||
|
|
||||||
Pretrained model can be found [here](https://huggingface.co/zrjin/icefall-tts-vctk-vits-2023-12-05), note that this model was pretrained on the Edinburgh DataShare VCTK dataset.
|
Pretrained model can be found [here](https://huggingface.co/zrjin/icefall-tts-vctk-vits-2023-12-05), note that this model was pretrained on the Edinburgh DataShare VCTK dataset.
|
||||||
|
|
||||||
For tutorial and more details, please refer to the [VITS documentation](https://k2-fsa.github.io/icefall/recipes/TTS/vctk/vits.html).
|
For tutorial and more details, please refer to the [VITS documentation](https://k2-fsa.github.io/icefall/recipes/TTS/vctk/vits.html).
|
||||||
|
|
||||||
|
The training command is given below:
|
||||||
|
```
|
||||||
|
export CUDA_VISIBLE_DEVICES="0,1,2,3"
|
||||||
|
./vits/train.py \
|
||||||
|
--world-size 4 \
|
||||||
|
--num-epochs 1000 \
|
||||||
|
--start-epoch 1 \
|
||||||
|
--use-fp16 1 \
|
||||||
|
--exp-dir vits/exp \
|
||||||
|
--tokens data/tokens.txt
|
||||||
|
--max-duration 350
|
||||||
|
```
|
||||||
|
|
||||||
|
To inference, use:
|
||||||
|
```
|
||||||
|
./vits/infer.py \
|
||||||
|
--epoch 1000 \
|
||||||
|
--exp-dir vits/exp \
|
||||||
|
--tokens data/tokens.txt \
|
||||||
|
--max-duration 500
|
||||||
|
```
|
Loading…
x
Reference in New Issue
Block a user