Added guidance for training on local machine

This commit is contained in:
Plachta
2023-06-12 18:26:59 +08:00
parent 9398660323
commit 1d7e8fc637
2 changed files with 9 additions and 4 deletions
+6 -2
View File
@@ -1,6 +1,6 @@
# Train locally
### Build environment
0. Make sure you have Python>=3.6, <=3.8;
0. Make sure you have installed `Python>=3.6, <=3.8`, CMake & C/C++ compilers;
1. Clone this repository;
2. Run `pip install -r requirements`;
3. Install GPU version PyTorch: (Make sure you have CUDA 11.6 or 11.7 installed)
@@ -98,4 +98,8 @@
Do replace `{Maximum_epochs}` with your desired number of epochs. Empirically, 100 or more is recommended.
To view training progress, open a new terminal and `cd` to the project root directory, run `tensorboard --logdir="./OUTPUT_MODEL"`, then visit `localhost:6006` with your web browser.
12. After training is completed, you can use your model by running:
`python VC_inference.py --model_dir ./OUTPUT_MODEL/G_latest.pth --share True`
`python VC_inference.py --model_dir ./OUTPUT_MODEL/G_latest.pth --share True`
13. To clear all audio data, run:
```
rm -rf ./custom_character_voice/* ./video_data/* ./raw_audio/* ./denoised_audio/* ./segmented_character_voice/* long_character_anno.txt short_character_anno.txt
```
+3 -2
View File
@@ -26,8 +26,9 @@ Chinese & Japanese[![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F
## Fine-tuning
It's recommended to perform fine-tuning on [Google Colab](https://colab.research.google.com/drive/1pn1xnFfdLK63gVXDwV4zCXfVeo8c-I-0?usp=sharing)
because the original VITS has some dependencies that are difficult to configure.
See [LOCAL.md](https://github.com/Plachtaa/VITS-fast-fine-tuning/blob/main/LOCAL.md) for local training guide.
Alternatively, you can perform fine-tuning on [Google Colab](https://colab.research.google.com/drive/1pn1xnFfdLK63gVXDwV4zCXfVeo8c-I-0?usp=sharing)
### How long does it take?
1. Install dependencies (3 min)