Skip to content

Commit

Permalink
Fine-Tuning section added.
Browse files Browse the repository at this point in the history
  • Loading branch information
jik876 committed Oct 30, 2020
1 parent 28ecbea commit 688af11
Showing 1 changed file with 22 additions and 6 deletions.
28 changes: 22 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,12 +58,28 @@ Details of each folder are as in follows:
|UNIVERSAL_V1|V1|Universal|No|


## Fine-Tuning
1. Generate mel-spectrograms in numpy format using [Tacotron2](https://github.com/NVIDIA/tacotron2) with teacher-forcing.<br/>
The file name of the generated mel-spectrogram should match the audio file and the extension should be `.npy`.<br/>
Example:
```
Audio File : LJ001-0001.wav
Mel-Spectrogram File : LJ001-0001.npy
```
2. Create `ft_dataset` folder and copy the generated mel-spectrogram files into it.<br/>
3. Run the following command.
```
python train.py --fine_tuning True --config config_v1.json
```
For other command line options, please refer to the training section.
## Inference from wav file
1. Make `test_files` directory and copy wav files into the directory.
2. Run the following command.
```
python inference.py --checkpoint_file [generator checkpoint file path]
```
```
python inference.py --checkpoint_file [generator checkpoint file path]
```
Generated wav files are saved in `generated_files` by default.<br>
You can change the path by adding `--output_dir` option.
Expand All @@ -73,9 +89,9 @@ You can change the path by adding `--output_dir` option.
You can generate mel-spectrograms using [Tacotron2](https://github.com/NVIDIA/tacotron2),
[Glow-TTS](https://github.com/jaywalnut310/glow-tts) and so forth.
2. Run the following command.
```
python inference_e2e.py --checkpoint_file [generator checkpoint file path]
```
```
python inference_e2e.py --checkpoint_file [generator checkpoint file path]
```
Generated wav files are saved in `generated_files_from_mel` by default.<br>
You can change the path by adding `--output_dir` option.
Expand Down

0 comments on commit 688af11

Please sign in to comment.