mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-14 12:32:20 +00:00
added pre-trained model
This commit is contained in:
parent
fe1498a6a5
commit
9f79e21135
@ -28,12 +28,14 @@ For more information, refer to the paper "LibriTTS: A Corpus Derived from LibriS
|
|||||||
|
|
||||||
This recipe provides a VITS model trained on the LibriTTS dataset.
|
This recipe provides a VITS model trained on the LibriTTS dataset.
|
||||||
|
|
||||||
|
Pretrained model can be found [here](https://huggingface.co/zrjin/icefall-tts-libritts-vits-2024-10-30).
|
||||||
|
|
||||||
The training command is given below:
|
The training command is given below:
|
||||||
```
|
```
|
||||||
export CUDA_VISIBLE_DEVICES=0,1,2,3
|
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
|
||||||
./vits/train.py \
|
./vits/train.py \
|
||||||
--world-size 4 \
|
--world-size 4 \
|
||||||
--num-epochs 1000 \
|
--num-epochs 400 \
|
||||||
--start-epoch 1 \
|
--start-epoch 1 \
|
||||||
--use-fp16 1 \
|
--use-fp16 1 \
|
||||||
--exp-dir vits/exp \
|
--exp-dir vits/exp \
|
||||||
@ -44,6 +46,6 @@ To inference, use:
|
|||||||
```
|
```
|
||||||
./vits/infer.py \
|
./vits/infer.py \
|
||||||
--exp-dir vits/exp \
|
--exp-dir vits/exp \
|
||||||
--epoch 1000 \
|
--epoch 400 \
|
||||||
--tokens data/tokens.txt
|
--tokens data/tokens.txt
|
||||||
```
|
```
|
||||||
|
Loading…
x
Reference in New Issue
Block a user