myshell-ai / MeloTTS

High-quality multi-lingual text-to-speech library by MyShell.ai. Support English, Spanish, French, Chinese, Japanese and Korean.
MIT License
4.87k stars 635 forks source link

Feature: Create a simple fine-tuning script #147

Open s-tweed opened 5 months ago

s-tweed commented 5 months ago

Is there a possibility of creating a simple to use (and modify) fine-tuning script? I have been trying to get the train.py script to work for quite a while on Windows and am having considerably difficulty, I think mostly because my environment is very different to yours (for example I am not distributing my training across multiple machines so gloo, torchrun, etc are not needed). I'm trying to fine-tine the model on a custom dataset so that I can change the voice to a specific male. A simplified script would be very helpful and appreciated!

Thanks, Spencer

RedBluePrinter commented 5 months ago

Hello @s-tweed i have made a command for you:

python -m torch.distributed.run --nproc_per_node=1 --master_port=10902 --master_addr=localhost train.py --c config.json --model Model-1

RedBluePrinter commented 5 months ago

Should work fine i Tested it On windows/linux(docker).