Billpai
commited on
Commit
·
0a17a23
1
Parent(s):
f03cd94
test
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- ckpts/svc/vocalist_l1_contentvec+whisper/args.json +0 -257
- ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/optimizer.bin +0 -3
- ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/pytorch_model.bin +0 -3
- ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/random_states_0.pkl +0 -3
- ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/singers.json +0 -17
- ckpts/svc/vocalist_l1_contentvec+whisper/log/vocalist_l1_contentvec+whisper/events.out.tfevents.1696052302.mmnewyardnodesz63219.120.0 +0 -3
- ckpts/svc/vocalist_l1_contentvec+whisper/log/vocalist_l1_contentvec+whisper/events.out.tfevents.1696052302.mmnewyardnodesz63219.120.1 +0 -3
- ckpts/svc/vocalist_l1_contentvec+whisper/singers.json +0 -17
- config/audioldm.json +0 -92
- config/autoencoderkl.json +0 -69
- config/base.json +0 -220
- config/comosvc.json +0 -216
- config/diffusion.json +0 -227
- config/fs2.json +0 -117
- config/transformer.json +0 -180
- config/tts.json +0 -23
- config/valle.json +0 -52
- config/vits.json +0 -101
- config/vocoder.json +0 -84
- egs/svc/MultipleContentsSVC/README.md +0 -153
- egs/svc/MultipleContentsSVC/exp_config.json +0 -126
- egs/svc/MultipleContentsSVC/run.sh +0 -1
- egs/svc/README.md +0 -34
- egs/svc/_template/run.sh +0 -150
- egs/vocoder/README.md +0 -23
- egs/vocoder/diffusion/README.md +0 -0
- egs/vocoder/diffusion/exp_config_base.json +0 -0
- egs/vocoder/gan/README.md +0 -224
- egs/vocoder/gan/_template/run.sh +0 -143
- egs/vocoder/gan/apnet/exp_config.json +0 -45
- egs/vocoder/gan/apnet/run.sh +0 -143
- egs/vocoder/gan/bigvgan/exp_config.json +0 -66
- egs/vocoder/gan/bigvgan/run.sh +0 -143
- egs/vocoder/gan/bigvgan_large/exp_config.json +0 -70
- egs/vocoder/gan/bigvgan_large/run.sh +0 -143
- egs/vocoder/gan/exp_config_base.json +0 -111
- egs/vocoder/gan/hifigan/exp_config.json +0 -59
- egs/vocoder/gan/hifigan/run.sh +0 -143
- egs/vocoder/gan/melgan/exp_config.json +0 -34
- egs/vocoder/gan/melgan/run.sh +0 -143
- egs/vocoder/gan/nsfhifigan/exp_config.json +0 -83
- egs/vocoder/gan/nsfhifigan/run.sh +0 -143
- egs/vocoder/gan/tfr_enhanced_hifigan/README.md +0 -185
- egs/vocoder/gan/tfr_enhanced_hifigan/exp_config.json +0 -118
- egs/vocoder/gan/tfr_enhanced_hifigan/run.sh +0 -145
- examples/chinese_female_recordings.wav +0 -3
- examples/chinese_male_seperated.wav +0 -3
- examples/english_female_seperated.wav +0 -3
- examples/english_male_recordings.wav +0 -3
- examples/output/.DS_Store +0 -0
ckpts/svc/vocalist_l1_contentvec+whisper/args.json
DELETED
|
@@ -1,257 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"task_type": "svc",
|
| 3 |
-
"dataset": [
|
| 4 |
-
"vocalist_l1",
|
| 5 |
-
],
|
| 6 |
-
"exp_name": "vocalist_l1_contentvec+whisper",
|
| 7 |
-
"inference": {
|
| 8 |
-
"diffusion": {
|
| 9 |
-
"scheduler": "pndm",
|
| 10 |
-
"scheduler_settings": {
|
| 11 |
-
"num_inference_timesteps": 1000,
|
| 12 |
-
},
|
| 13 |
-
},
|
| 14 |
-
},
|
| 15 |
-
"model": {
|
| 16 |
-
"condition_encoder": {
|
| 17 |
-
"content_encoder_dim": 384,
|
| 18 |
-
"contentvec_dim": 256,
|
| 19 |
-
"f0_max": 1100,
|
| 20 |
-
"f0_min": 50,
|
| 21 |
-
"input_loudness_dim": 1,
|
| 22 |
-
"input_melody_dim": 1,
|
| 23 |
-
"merge_mode": "add",
|
| 24 |
-
"mert_dim": 256,
|
| 25 |
-
"n_bins_loudness": 256,
|
| 26 |
-
"n_bins_melody": 256,
|
| 27 |
-
"output_content_dim": 384,
|
| 28 |
-
"output_loudness_dim": 384,
|
| 29 |
-
"output_melody_dim": 384,
|
| 30 |
-
"output_singer_dim": 384,
|
| 31 |
-
"pitch_max": 1100,
|
| 32 |
-
"pitch_min": 50,
|
| 33 |
-
"singer_table_size": 512,
|
| 34 |
-
"use_conformer_for_content_features": false,
|
| 35 |
-
"use_contentvec": true,
|
| 36 |
-
"use_log_f0": true,
|
| 37 |
-
"use_log_loudness": true,
|
| 38 |
-
"use_mert": false,
|
| 39 |
-
"use_singer_encoder": true,
|
| 40 |
-
"use_spkid": true,
|
| 41 |
-
"use_wenet": false,
|
| 42 |
-
"use_whisper": true,
|
| 43 |
-
"wenet_dim": 512,
|
| 44 |
-
"whisper_dim": 1024,
|
| 45 |
-
},
|
| 46 |
-
"diffusion": {
|
| 47 |
-
"bidilconv": {
|
| 48 |
-
"base_channel": 384,
|
| 49 |
-
"conditioner_size": 384,
|
| 50 |
-
"conv_kernel_size": 3,
|
| 51 |
-
"dilation_cycle_length": 4,
|
| 52 |
-
"n_res_block": 20,
|
| 53 |
-
},
|
| 54 |
-
"model_type": "bidilconv",
|
| 55 |
-
"scheduler": "ddpm",
|
| 56 |
-
"scheduler_settings": {
|
| 57 |
-
"beta_end": 0.02,
|
| 58 |
-
"beta_schedule": "linear",
|
| 59 |
-
"beta_start": 0.0001,
|
| 60 |
-
"num_train_timesteps": 1000,
|
| 61 |
-
},
|
| 62 |
-
"step_encoder": {
|
| 63 |
-
"activation": "SiLU",
|
| 64 |
-
"dim_hidden_layer": 512,
|
| 65 |
-
"dim_raw_embedding": 128,
|
| 66 |
-
"max_period": 10000,
|
| 67 |
-
"num_layer": 2,
|
| 68 |
-
},
|
| 69 |
-
"unet2d": {
|
| 70 |
-
"down_block_types": [
|
| 71 |
-
"CrossAttnDownBlock2D",
|
| 72 |
-
"CrossAttnDownBlock2D",
|
| 73 |
-
"CrossAttnDownBlock2D",
|
| 74 |
-
"DownBlock2D",
|
| 75 |
-
],
|
| 76 |
-
"in_channels": 1,
|
| 77 |
-
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
| 78 |
-
"only_cross_attention": false,
|
| 79 |
-
"out_channels": 1,
|
| 80 |
-
"up_block_types": [
|
| 81 |
-
"UpBlock2D",
|
| 82 |
-
"CrossAttnUpBlock2D",
|
| 83 |
-
"CrossAttnUpBlock2D",
|
| 84 |
-
"CrossAttnUpBlock2D",
|
| 85 |
-
],
|
| 86 |
-
},
|
| 87 |
-
},
|
| 88 |
-
},
|
| 89 |
-
"model_type": "DiffWaveNetSVC",
|
| 90 |
-
"preprocess": {
|
| 91 |
-
"audio_dir": "audios",
|
| 92 |
-
"bits": 8,
|
| 93 |
-
"content_feature_batch_size": 16,
|
| 94 |
-
"contentvec_batch_size": 1,
|
| 95 |
-
"contentvec_dir": "contentvec",
|
| 96 |
-
"contentvec_file": "pretrained/contentvec/checkpoint_best_legacy_500.pt",
|
| 97 |
-
"contentvec_frameshift": 0.02,
|
| 98 |
-
"contentvec_sample_rate": 16000,
|
| 99 |
-
"dur_dir": "durs",
|
| 100 |
-
"duration_dir": "duration",
|
| 101 |
-
"emo2id": "emo2id.json",
|
| 102 |
-
"energy_dir": "energys",
|
| 103 |
-
"extract_audio": false,
|
| 104 |
-
"extract_contentvec_feature": true,
|
| 105 |
-
"extract_energy": true,
|
| 106 |
-
"extract_label": false,
|
| 107 |
-
"extract_mcep": false,
|
| 108 |
-
"extract_mel": true,
|
| 109 |
-
"extract_mert_feature": false,
|
| 110 |
-
"extract_pitch": true,
|
| 111 |
-
"extract_uv": true,
|
| 112 |
-
"extract_wenet_feature": false,
|
| 113 |
-
"extract_whisper_feature": true,
|
| 114 |
-
"f0_max": 1100,
|
| 115 |
-
"f0_min": 50,
|
| 116 |
-
"file_lst": "file.lst",
|
| 117 |
-
"fmax": 12000,
|
| 118 |
-
"fmin": 0,
|
| 119 |
-
"hop_size": 256,
|
| 120 |
-
"is_label": true,
|
| 121 |
-
"is_mu_law": true,
|
| 122 |
-
"lab_dir": "labs",
|
| 123 |
-
"label_dir": "labels",
|
| 124 |
-
"mcep_dir": "mcep",
|
| 125 |
-
"mel_dir": "mels",
|
| 126 |
-
"mel_min_max_norm": true,
|
| 127 |
-
"mel_min_max_stats_dir": "mel_min_max_stats",
|
| 128 |
-
"mert_dir": "mert",
|
| 129 |
-
"mert_feature_layer": -1,
|
| 130 |
-
"mert_frameshit": 0.01333,
|
| 131 |
-
"mert_hop_size": 320,
|
| 132 |
-
"mert_model": "m-a-p/MERT-v1-330M",
|
| 133 |
-
"min_level_db": -115,
|
| 134 |
-
"mu_law_norm": false,
|
| 135 |
-
"n_fft": 1024,
|
| 136 |
-
"n_mel": 100,
|
| 137 |
-
"num_silent_frames": 8,
|
| 138 |
-
"num_workers": 8,
|
| 139 |
-
"phone_seq_file": "phone_seq_file",
|
| 140 |
-
"pin_memory": true,
|
| 141 |
-
"pitch_bin": 256,
|
| 142 |
-
"pitch_dir": "pitches",
|
| 143 |
-
"pitch_extractor": "crepe", // "parselmouth"
|
| 144 |
-
"pitch_max": 1100.0,
|
| 145 |
-
"pitch_min": 50.0,
|
| 146 |
-
"processed_dir": "ckpts/svc/vocalist_l1_contentvec+whisper/data",
|
| 147 |
-
"ref_level_db": 20,
|
| 148 |
-
"sample_rate": 24000,
|
| 149 |
-
"spk2id": "singers.json",
|
| 150 |
-
"train_file": "train.json",
|
| 151 |
-
"trim_fft_size": 512,
|
| 152 |
-
"trim_hop_size": 128,
|
| 153 |
-
"trim_silence": false,
|
| 154 |
-
"trim_top_db": 30,
|
| 155 |
-
"trimmed_wav_dir": "trimmed_wavs",
|
| 156 |
-
"use_audio": false,
|
| 157 |
-
"use_contentvec": true,
|
| 158 |
-
"use_dur": false,
|
| 159 |
-
"use_emoid": false,
|
| 160 |
-
"use_frame_duration": false,
|
| 161 |
-
"use_frame_energy": true,
|
| 162 |
-
"use_frame_pitch": true,
|
| 163 |
-
"use_lab": false,
|
| 164 |
-
"use_label": false,
|
| 165 |
-
"use_log_scale_energy": false,
|
| 166 |
-
"use_log_scale_pitch": false,
|
| 167 |
-
"use_mel": true,
|
| 168 |
-
"use_mert": false,
|
| 169 |
-
"use_min_max_norm_mel": true,
|
| 170 |
-
"use_one_hot": false,
|
| 171 |
-
"use_phn_seq": false,
|
| 172 |
-
"use_phone_duration": false,
|
| 173 |
-
"use_phone_energy": false,
|
| 174 |
-
"use_phone_pitch": false,
|
| 175 |
-
"use_spkid": true,
|
| 176 |
-
"use_uv": true,
|
| 177 |
-
"use_wav": false,
|
| 178 |
-
"use_wenet": false,
|
| 179 |
-
"use_whisper": true,
|
| 180 |
-
"utt2emo": "utt2emo",
|
| 181 |
-
"utt2spk": "utt2singer",
|
| 182 |
-
"uv_dir": "uvs",
|
| 183 |
-
"valid_file": "test.json",
|
| 184 |
-
"wav_dir": "wavs",
|
| 185 |
-
"wenet_batch_size": 1,
|
| 186 |
-
"wenet_config": "pretrained/wenet/20220506_u2pp_conformer_exp/train.yaml",
|
| 187 |
-
"wenet_dir": "wenet",
|
| 188 |
-
"wenet_downsample_rate": 4,
|
| 189 |
-
"wenet_frameshift": 0.01,
|
| 190 |
-
"wenet_model_path": "pretrained/wenet/20220506_u2pp_conformer_exp/final.pt",
|
| 191 |
-
"wenet_sample_rate": 16000,
|
| 192 |
-
"whisper_batch_size": 30,
|
| 193 |
-
"whisper_dir": "whisper",
|
| 194 |
-
"whisper_downsample_rate": 2,
|
| 195 |
-
"whisper_frameshift": 0.01,
|
| 196 |
-
"whisper_model": "medium",
|
| 197 |
-
"whisper_model_path": "pretrained/whisper/medium.pt",
|
| 198 |
-
"whisper_sample_rate": 16000,
|
| 199 |
-
"win_size": 1024,
|
| 200 |
-
},
|
| 201 |
-
"supported_model_type": [
|
| 202 |
-
"Fastspeech2",
|
| 203 |
-
"DiffSVC",
|
| 204 |
-
"Transformer",
|
| 205 |
-
"EDM",
|
| 206 |
-
"CD",
|
| 207 |
-
],
|
| 208 |
-
"train": {
|
| 209 |
-
"adamw": {
|
| 210 |
-
"lr": 0.0004,
|
| 211 |
-
},
|
| 212 |
-
"batch_size": 32,
|
| 213 |
-
"dataloader": {
|
| 214 |
-
"num_worker": 8,
|
| 215 |
-
"pin_memory": true,
|
| 216 |
-
},
|
| 217 |
-
"ddp": true,
|
| 218 |
-
"epochs": 50000,
|
| 219 |
-
"gradient_accumulation_step": 1,
|
| 220 |
-
"keep_checkpoint_max": 5,
|
| 221 |
-
"keep_last": [
|
| 222 |
-
5,
|
| 223 |
-
-1,
|
| 224 |
-
],
|
| 225 |
-
"max_epoch": -1,
|
| 226 |
-
"max_steps": 1000000,
|
| 227 |
-
"multi_speaker_training": false,
|
| 228 |
-
"optimizer": "AdamW",
|
| 229 |
-
"random_seed": 10086,
|
| 230 |
-
"reducelronplateau": {
|
| 231 |
-
"factor": 0.8,
|
| 232 |
-
"min_lr": 0.0001,
|
| 233 |
-
"patience": 10,
|
| 234 |
-
},
|
| 235 |
-
"run_eval": [
|
| 236 |
-
false,
|
| 237 |
-
true,
|
| 238 |
-
],
|
| 239 |
-
"sampler": {
|
| 240 |
-
"drop_last": true,
|
| 241 |
-
"holistic_shuffle": false,
|
| 242 |
-
},
|
| 243 |
-
"save_checkpoint_stride": [
|
| 244 |
-
3,
|
| 245 |
-
10,
|
| 246 |
-
],
|
| 247 |
-
"save_checkpoints_steps": 10000,
|
| 248 |
-
"save_summary_steps": 500,
|
| 249 |
-
"scheduler": "ReduceLROnPlateau",
|
| 250 |
-
"total_training_steps": 50000,
|
| 251 |
-
"tracker": [
|
| 252 |
-
"tensorboard",
|
| 253 |
-
],
|
| 254 |
-
"valid_interval": 10000,
|
| 255 |
-
},
|
| 256 |
-
"use_custom_dataset": true,
|
| 257 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/optimizer.bin
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:836af10b834c7aec9209eb19ce43559e6ef1e3a59bd6468e90cadbc9a18749ef
|
| 3 |
-
size 249512389
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/pytorch_model.bin
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:d54eed12bef331095fc367f196d07c5061d5cb72dd6fe0e1e4453b997bf1d68d
|
| 3 |
-
size 124755137
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/random_states_0.pkl
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:6798ddffadcd7d5405a77e667c674c474e4fef0cba817fdd300c7c985c1e82fe
|
| 3 |
-
size 14599
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/checkpoint/epoch-6852_step-0678447_loss-1.946773/singers.json
DELETED
|
@@ -1,17 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"vocalist_l1_Adele": 0,
|
| 3 |
-
"vocalist_l1_Beyonce": 1,
|
| 4 |
-
"vocalist_l1_BrunoMars": 2,
|
| 5 |
-
"vocalist_l1_JohnMayer": 3,
|
| 6 |
-
"vocalist_l1_MichaelJackson": 4,
|
| 7 |
-
"vocalist_l1_TaylorSwift": 5,
|
| 8 |
-
"vocalist_l1_张学友": 6,
|
| 9 |
-
"vocalist_l1_李健": 7,
|
| 10 |
-
"vocalist_l1_汪峰": 8,
|
| 11 |
-
"vocalist_l1_王菲": 9,
|
| 12 |
-
"vocalist_l1_石倚洁": 10,
|
| 13 |
-
"vocalist_l1_蔡琴": 11,
|
| 14 |
-
"vocalist_l1_那英": 12,
|
| 15 |
-
"vocalist_l1_陈奕迅": 13,
|
| 16 |
-
"vocalist_l1_陶喆": 14
|
| 17 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/log/vocalist_l1_contentvec+whisper/events.out.tfevents.1696052302.mmnewyardnodesz63219.120.0
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:d7f490fd0c97876e24bfc44413365ded7ff5d22c1c79f0dac0b754f3b32df76f
|
| 3 |
-
size 88
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/log/vocalist_l1_contentvec+whisper/events.out.tfevents.1696052302.mmnewyardnodesz63219.120.1
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:e01bcf2fa621ba563b70568c18fe0742d0f48cafae83a6e8beb0bb6d1f6d146d
|
| 3 |
-
size 77413046
|
|
|
|
|
|
|
|
|
|
|
|
ckpts/svc/vocalist_l1_contentvec+whisper/singers.json
DELETED
|
@@ -1,17 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"vocalist_l1_Adele": 0,
|
| 3 |
-
"vocalist_l1_Beyonce": 1,
|
| 4 |
-
"vocalist_l1_BrunoMars": 2,
|
| 5 |
-
"vocalist_l1_JohnMayer": 3,
|
| 6 |
-
"vocalist_l1_MichaelJackson": 4,
|
| 7 |
-
"vocalist_l1_TaylorSwift": 5,
|
| 8 |
-
"vocalist_l1_张学友": 6,
|
| 9 |
-
"vocalist_l1_李健": 7,
|
| 10 |
-
"vocalist_l1_汪峰": 8,
|
| 11 |
-
"vocalist_l1_王菲": 9,
|
| 12 |
-
"vocalist_l1_石倚洁": 10,
|
| 13 |
-
"vocalist_l1_蔡琴": 11,
|
| 14 |
-
"vocalist_l1_那英": 12,
|
| 15 |
-
"vocalist_l1_陈奕迅": 13,
|
| 16 |
-
"vocalist_l1_陶喆": 14
|
| 17 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/audioldm.json
DELETED
|
@@ -1,92 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/base.json",
|
| 3 |
-
"model_type": "AudioLDM",
|
| 4 |
-
"task_type": "tta",
|
| 5 |
-
"dataset": [
|
| 6 |
-
"AudioCaps"
|
| 7 |
-
],
|
| 8 |
-
"preprocess": {
|
| 9 |
-
// feature used for model training
|
| 10 |
-
"use_spkid": false,
|
| 11 |
-
"use_uv": false,
|
| 12 |
-
"use_frame_pitch": false,
|
| 13 |
-
"use_phone_pitch": false,
|
| 14 |
-
"use_frame_energy": false,
|
| 15 |
-
"use_phone_energy": false,
|
| 16 |
-
"use_mel": false,
|
| 17 |
-
"use_audio": false,
|
| 18 |
-
"use_label": false,
|
| 19 |
-
"use_one_hot": false,
|
| 20 |
-
"cond_mask_prob": 0.1
|
| 21 |
-
},
|
| 22 |
-
// model
|
| 23 |
-
"model": {
|
| 24 |
-
"audioldm": {
|
| 25 |
-
"image_size": 32,
|
| 26 |
-
"in_channels": 4,
|
| 27 |
-
"out_channels": 4,
|
| 28 |
-
"model_channels": 256,
|
| 29 |
-
"attention_resolutions": [
|
| 30 |
-
4,
|
| 31 |
-
2,
|
| 32 |
-
1
|
| 33 |
-
],
|
| 34 |
-
"num_res_blocks": 2,
|
| 35 |
-
"channel_mult": [
|
| 36 |
-
1,
|
| 37 |
-
2,
|
| 38 |
-
4
|
| 39 |
-
],
|
| 40 |
-
"num_heads": 8,
|
| 41 |
-
"use_spatial_transformer": true,
|
| 42 |
-
"transformer_depth": 1,
|
| 43 |
-
"context_dim": 768,
|
| 44 |
-
"use_checkpoint": true,
|
| 45 |
-
"legacy": false
|
| 46 |
-
},
|
| 47 |
-
"autoencoderkl": {
|
| 48 |
-
"ch": 128,
|
| 49 |
-
"ch_mult": [
|
| 50 |
-
1,
|
| 51 |
-
1,
|
| 52 |
-
2,
|
| 53 |
-
2,
|
| 54 |
-
4
|
| 55 |
-
],
|
| 56 |
-
"num_res_blocks": 2,
|
| 57 |
-
"in_channels": 1,
|
| 58 |
-
"z_channels": 4,
|
| 59 |
-
"out_ch": 1,
|
| 60 |
-
"double_z": true
|
| 61 |
-
},
|
| 62 |
-
"noise_scheduler": {
|
| 63 |
-
"num_train_timesteps": 1000,
|
| 64 |
-
"beta_start": 0.00085,
|
| 65 |
-
"beta_end": 0.012,
|
| 66 |
-
"beta_schedule": "scaled_linear",
|
| 67 |
-
"clip_sample": false,
|
| 68 |
-
"steps_offset": 1,
|
| 69 |
-
"set_alpha_to_one": false,
|
| 70 |
-
"skip_prk_steps": true,
|
| 71 |
-
"prediction_type": "epsilon"
|
| 72 |
-
}
|
| 73 |
-
},
|
| 74 |
-
// train
|
| 75 |
-
"train": {
|
| 76 |
-
"lronPlateau": {
|
| 77 |
-
"factor": 0.9,
|
| 78 |
-
"patience": 100,
|
| 79 |
-
"min_lr": 4.0e-5,
|
| 80 |
-
"verbose": true
|
| 81 |
-
},
|
| 82 |
-
"adam": {
|
| 83 |
-
"lr": 5.0e-5,
|
| 84 |
-
"betas": [
|
| 85 |
-
0.9,
|
| 86 |
-
0.999
|
| 87 |
-
],
|
| 88 |
-
"weight_decay": 1.0e-2,
|
| 89 |
-
"eps": 1.0e-8
|
| 90 |
-
}
|
| 91 |
-
}
|
| 92 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/autoencoderkl.json
DELETED
|
@@ -1,69 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/base.json",
|
| 3 |
-
"model_type": "AutoencoderKL",
|
| 4 |
-
"task_type": "tta",
|
| 5 |
-
"dataset": [
|
| 6 |
-
"AudioCaps"
|
| 7 |
-
],
|
| 8 |
-
"preprocess": {
|
| 9 |
-
// feature used for model training
|
| 10 |
-
"use_spkid": false,
|
| 11 |
-
"use_uv": false,
|
| 12 |
-
"use_frame_pitch": false,
|
| 13 |
-
"use_phone_pitch": false,
|
| 14 |
-
"use_frame_energy": false,
|
| 15 |
-
"use_phone_energy": false,
|
| 16 |
-
"use_mel": false,
|
| 17 |
-
"use_audio": false,
|
| 18 |
-
"use_label": false,
|
| 19 |
-
"use_one_hot": false
|
| 20 |
-
},
|
| 21 |
-
// model
|
| 22 |
-
"model": {
|
| 23 |
-
"autoencoderkl": {
|
| 24 |
-
"ch": 128,
|
| 25 |
-
"ch_mult": [
|
| 26 |
-
1,
|
| 27 |
-
1,
|
| 28 |
-
2,
|
| 29 |
-
2,
|
| 30 |
-
4
|
| 31 |
-
],
|
| 32 |
-
"num_res_blocks": 2,
|
| 33 |
-
"in_channels": 1,
|
| 34 |
-
"z_channels": 4,
|
| 35 |
-
"out_ch": 1,
|
| 36 |
-
"double_z": true
|
| 37 |
-
},
|
| 38 |
-
"loss": {
|
| 39 |
-
"kl_weight": 1e-8,
|
| 40 |
-
"disc_weight": 0.5,
|
| 41 |
-
"disc_factor": 1.0,
|
| 42 |
-
"logvar_init": 0.0,
|
| 43 |
-
"min_adapt_d_weight": 0.0,
|
| 44 |
-
"max_adapt_d_weight": 10.0,
|
| 45 |
-
"disc_start": 50001,
|
| 46 |
-
"disc_in_channels": 1,
|
| 47 |
-
"disc_num_layers": 3,
|
| 48 |
-
"use_actnorm": false
|
| 49 |
-
}
|
| 50 |
-
},
|
| 51 |
-
// train
|
| 52 |
-
"train": {
|
| 53 |
-
"lronPlateau": {
|
| 54 |
-
"factor": 0.9,
|
| 55 |
-
"patience": 100,
|
| 56 |
-
"min_lr": 4.0e-5,
|
| 57 |
-
"verbose": true
|
| 58 |
-
},
|
| 59 |
-
"adam": {
|
| 60 |
-
"lr": 4.0e-4,
|
| 61 |
-
"betas": [
|
| 62 |
-
0.9,
|
| 63 |
-
0.999
|
| 64 |
-
],
|
| 65 |
-
"weight_decay": 1.0e-2,
|
| 66 |
-
"eps": 1.0e-8
|
| 67 |
-
}
|
| 68 |
-
}
|
| 69 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/base.json
DELETED
|
@@ -1,220 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"supported_model_type": [
|
| 3 |
-
"GANVocoder",
|
| 4 |
-
"Fastspeech2",
|
| 5 |
-
"DiffSVC",
|
| 6 |
-
"Transformer",
|
| 7 |
-
"EDM",
|
| 8 |
-
"CD"
|
| 9 |
-
],
|
| 10 |
-
"task_type": "",
|
| 11 |
-
"dataset": [],
|
| 12 |
-
"use_custom_dataset": false,
|
| 13 |
-
"preprocess": {
|
| 14 |
-
"phone_extractor": "espeak", // "espeak, pypinyin, pypinyin_initials_finals, lexicon"
|
| 15 |
-
// trim audio silence
|
| 16 |
-
"data_augment": false,
|
| 17 |
-
"trim_silence": false,
|
| 18 |
-
"num_silent_frames": 8,
|
| 19 |
-
"trim_fft_size": 512, // fft size used in trimming
|
| 20 |
-
"trim_hop_size": 128, // hop size used in trimming
|
| 21 |
-
"trim_top_db": 30, // top db used in trimming sensitive to each dataset
|
| 22 |
-
// acoustic features
|
| 23 |
-
"extract_mel": false,
|
| 24 |
-
"mel_extract_mode": "",
|
| 25 |
-
"extract_linear_spec": false,
|
| 26 |
-
"extract_mcep": false,
|
| 27 |
-
"extract_pitch": false,
|
| 28 |
-
"extract_acoustic_token": false,
|
| 29 |
-
"pitch_remove_outlier": false,
|
| 30 |
-
"extract_uv": false,
|
| 31 |
-
"pitch_norm": false,
|
| 32 |
-
"extract_audio": false,
|
| 33 |
-
"extract_label": false,
|
| 34 |
-
"pitch_extractor": "parselmouth", // pyin, dio, pyworld, pyreaper, parselmouth, CWT (Continuous Wavelet Transform)
|
| 35 |
-
"extract_energy": false,
|
| 36 |
-
"energy_remove_outlier": false,
|
| 37 |
-
"energy_norm": false,
|
| 38 |
-
"energy_extract_mode": "from_mel",
|
| 39 |
-
"extract_duration": false,
|
| 40 |
-
"extract_amplitude_phase": false,
|
| 41 |
-
"mel_min_max_norm": false,
|
| 42 |
-
// lingusitic features
|
| 43 |
-
"extract_phone": false,
|
| 44 |
-
"lexicon_path": "./text/lexicon/librispeech-lexicon.txt",
|
| 45 |
-
// content features
|
| 46 |
-
"extract_whisper_feature": false,
|
| 47 |
-
"extract_contentvec_feature": false,
|
| 48 |
-
"extract_mert_feature": false,
|
| 49 |
-
"extract_wenet_feature": false,
|
| 50 |
-
// Settings for data preprocessing
|
| 51 |
-
"n_mel": 80,
|
| 52 |
-
"win_size": 480,
|
| 53 |
-
"hop_size": 120,
|
| 54 |
-
"sample_rate": 24000,
|
| 55 |
-
"n_fft": 1024,
|
| 56 |
-
"fmin": 0,
|
| 57 |
-
"fmax": 12000,
|
| 58 |
-
"min_level_db": -115,
|
| 59 |
-
"ref_level_db": 20,
|
| 60 |
-
"bits": 8,
|
| 61 |
-
// Directory names of processed data or extracted features
|
| 62 |
-
"processed_dir": "processed_data",
|
| 63 |
-
"trimmed_wav_dir": "trimmed_wavs", // directory name of silence trimed wav
|
| 64 |
-
"raw_data": "raw_data",
|
| 65 |
-
"phone_dir": "phones",
|
| 66 |
-
"wav_dir": "wavs", // directory name of processed wav (such as downsampled waveform)
|
| 67 |
-
"audio_dir": "audios",
|
| 68 |
-
"log_amplitude_dir": "log_amplitudes",
|
| 69 |
-
"phase_dir": "phases",
|
| 70 |
-
"real_dir": "reals",
|
| 71 |
-
"imaginary_dir": "imaginarys",
|
| 72 |
-
"label_dir": "labels",
|
| 73 |
-
"linear_dir": "linears",
|
| 74 |
-
"mel_dir": "mels", // directory name of extraced mel features
|
| 75 |
-
"mcep_dir": "mcep", // directory name of extraced mcep features
|
| 76 |
-
"dur_dir": "durs",
|
| 77 |
-
"symbols_dict": "symbols.dict",
|
| 78 |
-
"lab_dir": "labs", // directory name of extraced label features
|
| 79 |
-
"wenet_dir": "wenet", // directory name of extraced wenet features
|
| 80 |
-
"contentvec_dir": "contentvec", // directory name of extraced wenet features
|
| 81 |
-
"pitch_dir": "pitches", // directory name of extraced pitch features
|
| 82 |
-
"energy_dir": "energys", // directory name of extracted energy features
|
| 83 |
-
"phone_pitch_dir": "phone_pitches", // directory name of extraced pitch features
|
| 84 |
-
"phone_energy_dir": "phone_energys", // directory name of extracted energy features
|
| 85 |
-
"uv_dir": "uvs", // directory name of extracted unvoiced features
|
| 86 |
-
"duration_dir": "duration", // ground-truth duration file
|
| 87 |
-
"phone_seq_file": "phone_seq_file", // phoneme sequence file
|
| 88 |
-
"file_lst": "file.lst",
|
| 89 |
-
"train_file": "train.json", // training set, the json file contains detailed information about the dataset, including dataset name, utterance id, duration of the utterance
|
| 90 |
-
"valid_file": "valid.json", // validattion set
|
| 91 |
-
"spk2id": "spk2id.json", // used for multi-speaker dataset
|
| 92 |
-
"utt2spk": "utt2spk", // used for multi-speaker dataset
|
| 93 |
-
"emo2id": "emo2id.json", // used for multi-emotion dataset
|
| 94 |
-
"utt2emo": "utt2emo", // used for multi-emotion dataset
|
| 95 |
-
// Features used for model training
|
| 96 |
-
"use_text": false,
|
| 97 |
-
"use_phone": false,
|
| 98 |
-
"use_phn_seq": false,
|
| 99 |
-
"use_lab": false,
|
| 100 |
-
"use_linear": false,
|
| 101 |
-
"use_mel": false,
|
| 102 |
-
"use_min_max_norm_mel": false,
|
| 103 |
-
"use_wav": false,
|
| 104 |
-
"use_phone_pitch": false,
|
| 105 |
-
"use_log_scale_pitch": false,
|
| 106 |
-
"use_phone_energy": false,
|
| 107 |
-
"use_phone_duration": false,
|
| 108 |
-
"use_log_scale_energy": false,
|
| 109 |
-
"use_wenet": false,
|
| 110 |
-
"use_dur": false,
|
| 111 |
-
"use_spkid": false, // True: use speaker id for multi-speaker dataset
|
| 112 |
-
"use_emoid": false, // True: use emotion id for multi-emotion dataset
|
| 113 |
-
"use_frame_pitch": false,
|
| 114 |
-
"use_uv": false,
|
| 115 |
-
"use_frame_energy": false,
|
| 116 |
-
"use_frame_duration": false,
|
| 117 |
-
"use_audio": false,
|
| 118 |
-
"use_label": false,
|
| 119 |
-
"use_one_hot": false,
|
| 120 |
-
"use_amplitude_phase": false,
|
| 121 |
-
"data_augment": false,
|
| 122 |
-
"align_mel_duration": false
|
| 123 |
-
},
|
| 124 |
-
"train": {
|
| 125 |
-
"ddp": true,
|
| 126 |
-
"random_seed": 970227,
|
| 127 |
-
"batch_size": 16,
|
| 128 |
-
"max_steps": 1000000,
|
| 129 |
-
// Trackers
|
| 130 |
-
"tracker": [
|
| 131 |
-
"tensorboard"
|
| 132 |
-
// "wandb",
|
| 133 |
-
// "cometml",
|
| 134 |
-
// "mlflow",
|
| 135 |
-
],
|
| 136 |
-
"max_epoch": -1,
|
| 137 |
-
// -1 means no limit
|
| 138 |
-
"save_checkpoint_stride": [
|
| 139 |
-
5,
|
| 140 |
-
20
|
| 141 |
-
],
|
| 142 |
-
// unit is epoch
|
| 143 |
-
"keep_last": [
|
| 144 |
-
3,
|
| 145 |
-
-1
|
| 146 |
-
],
|
| 147 |
-
// -1 means infinite, if one number will broadcast
|
| 148 |
-
"run_eval": [
|
| 149 |
-
false,
|
| 150 |
-
true
|
| 151 |
-
],
|
| 152 |
-
// if one number will broadcast
|
| 153 |
-
// Fix the random seed
|
| 154 |
-
"random_seed": 10086,
|
| 155 |
-
// Optimizer
|
| 156 |
-
"optimizer": "AdamW",
|
| 157 |
-
"adamw": {
|
| 158 |
-
"lr": 4.0e-4
|
| 159 |
-
// nn model lr
|
| 160 |
-
},
|
| 161 |
-
// LR Scheduler
|
| 162 |
-
"scheduler": "ReduceLROnPlateau",
|
| 163 |
-
"reducelronplateau": {
|
| 164 |
-
"factor": 0.8,
|
| 165 |
-
"patience": 10,
|
| 166 |
-
// unit is epoch
|
| 167 |
-
"min_lr": 1.0e-4
|
| 168 |
-
},
|
| 169 |
-
// Batchsampler
|
| 170 |
-
"sampler": {
|
| 171 |
-
"holistic_shuffle": true,
|
| 172 |
-
"drop_last": true
|
| 173 |
-
},
|
| 174 |
-
// Dataloader
|
| 175 |
-
"dataloader": {
|
| 176 |
-
"num_worker": 32,
|
| 177 |
-
"pin_memory": true
|
| 178 |
-
},
|
| 179 |
-
"gradient_accumulation_step": 1,
|
| 180 |
-
"total_training_steps": 50000,
|
| 181 |
-
"save_summary_steps": 500,
|
| 182 |
-
"save_checkpoints_steps": 10000,
|
| 183 |
-
"valid_interval": 10000,
|
| 184 |
-
"keep_checkpoint_max": 5,
|
| 185 |
-
"multi_speaker_training": false, // True: train multi-speaker model; False: training single-speaker model;
|
| 186 |
-
"max_epoch": -1,
|
| 187 |
-
// -1 means no limit
|
| 188 |
-
"save_checkpoint_stride": [
|
| 189 |
-
5,
|
| 190 |
-
20
|
| 191 |
-
],
|
| 192 |
-
// unit is epoch
|
| 193 |
-
"keep_last": [
|
| 194 |
-
3,
|
| 195 |
-
-1
|
| 196 |
-
],
|
| 197 |
-
// -1 means infinite, if one number will broadcast
|
| 198 |
-
"run_eval": [
|
| 199 |
-
false,
|
| 200 |
-
true
|
| 201 |
-
],
|
| 202 |
-
// Batchsampler
|
| 203 |
-
"sampler": {
|
| 204 |
-
"holistic_shuffle": true,
|
| 205 |
-
"drop_last": true
|
| 206 |
-
},
|
| 207 |
-
// Dataloader
|
| 208 |
-
"dataloader": {
|
| 209 |
-
"num_worker": 32,
|
| 210 |
-
"pin_memory": true
|
| 211 |
-
},
|
| 212 |
-
// Trackers
|
| 213 |
-
"tracker": [
|
| 214 |
-
"tensorboard"
|
| 215 |
-
// "wandb",
|
| 216 |
-
// "cometml",
|
| 217 |
-
// "mlflow",
|
| 218 |
-
],
|
| 219 |
-
},
|
| 220 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/comosvc.json
DELETED
|
@@ -1,216 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/base.json",
|
| 3 |
-
"model_type": "DiffComoSVC",
|
| 4 |
-
"task_type": "svc",
|
| 5 |
-
"use_custom_dataset": false,
|
| 6 |
-
"preprocess": {
|
| 7 |
-
// data augmentations
|
| 8 |
-
"use_pitch_shift": false,
|
| 9 |
-
"use_formant_shift": false,
|
| 10 |
-
"use_time_stretch": false,
|
| 11 |
-
"use_equalizer": false,
|
| 12 |
-
// acoustic features
|
| 13 |
-
"extract_mel": true,
|
| 14 |
-
"mel_min_max_norm": true,
|
| 15 |
-
"extract_pitch": true,
|
| 16 |
-
"pitch_extractor": "parselmouth",
|
| 17 |
-
"extract_uv": true,
|
| 18 |
-
"extract_energy": true,
|
| 19 |
-
// content features
|
| 20 |
-
"extract_whisper_feature": false,
|
| 21 |
-
"whisper_sample_rate": 16000,
|
| 22 |
-
"extract_contentvec_feature": false,
|
| 23 |
-
"contentvec_sample_rate": 16000,
|
| 24 |
-
"extract_wenet_feature": false,
|
| 25 |
-
"wenet_sample_rate": 16000,
|
| 26 |
-
"extract_mert_feature": false,
|
| 27 |
-
"mert_sample_rate": 16000,
|
| 28 |
-
// Default config for whisper
|
| 29 |
-
"whisper_frameshift": 0.01,
|
| 30 |
-
"whisper_downsample_rate": 2,
|
| 31 |
-
// Default config for content vector
|
| 32 |
-
"contentvec_frameshift": 0.02,
|
| 33 |
-
// Default config for mert
|
| 34 |
-
"mert_model": "m-a-p/MERT-v1-330M",
|
| 35 |
-
"mert_feature_layer": -1,
|
| 36 |
-
"mert_hop_size": 320,
|
| 37 |
-
// 24k
|
| 38 |
-
"mert_frameshit": 0.01333,
|
| 39 |
-
// 10ms
|
| 40 |
-
"wenet_frameshift": 0.01,
|
| 41 |
-
// wenetspeech is 4, gigaspeech is 6
|
| 42 |
-
"wenet_downsample_rate": 4,
|
| 43 |
-
// Default config
|
| 44 |
-
"n_mel": 100,
|
| 45 |
-
"win_size": 1024,
|
| 46 |
-
// todo
|
| 47 |
-
"hop_size": 256,
|
| 48 |
-
"sample_rate": 24000,
|
| 49 |
-
"n_fft": 1024,
|
| 50 |
-
// todo
|
| 51 |
-
"fmin": 0,
|
| 52 |
-
"fmax": 12000,
|
| 53 |
-
// todo
|
| 54 |
-
"f0_min": 50,
|
| 55 |
-
// ~C2
|
| 56 |
-
"f0_max": 1100,
|
| 57 |
-
//1100, // ~C6(1100), ~G5(800)
|
| 58 |
-
"pitch_bin": 256,
|
| 59 |
-
"pitch_max": 1100.0,
|
| 60 |
-
"pitch_min": 50.0,
|
| 61 |
-
"is_label": true,
|
| 62 |
-
"is_mu_law": true,
|
| 63 |
-
"bits": 8,
|
| 64 |
-
"mel_min_max_stats_dir": "mel_min_max_stats",
|
| 65 |
-
"whisper_dir": "whisper",
|
| 66 |
-
"contentvec_dir": "contentvec",
|
| 67 |
-
"wenet_dir": "wenet",
|
| 68 |
-
"mert_dir": "mert",
|
| 69 |
-
// Extract content features using dataloader
|
| 70 |
-
"pin_memory": true,
|
| 71 |
-
"num_workers": 8,
|
| 72 |
-
"content_feature_batch_size": 16,
|
| 73 |
-
// Features used for model training
|
| 74 |
-
"use_mel": true,
|
| 75 |
-
"use_min_max_norm_mel": true,
|
| 76 |
-
"use_frame_pitch": true,
|
| 77 |
-
"use_uv": true,
|
| 78 |
-
"use_frame_energy": true,
|
| 79 |
-
"use_log_scale_pitch": false,
|
| 80 |
-
"use_log_scale_energy": false,
|
| 81 |
-
"use_spkid": true,
|
| 82 |
-
// Meta file
|
| 83 |
-
"train_file": "train.json",
|
| 84 |
-
"valid_file": "test.json",
|
| 85 |
-
"spk2id": "singers.json",
|
| 86 |
-
"utt2spk": "utt2singer"
|
| 87 |
-
},
|
| 88 |
-
"model": {
|
| 89 |
-
"teacher_model_path": "[Your Teacher Model Path].bin",
|
| 90 |
-
"condition_encoder": {
|
| 91 |
-
"merge_mode": "add",
|
| 92 |
-
"input_melody_dim": 1,
|
| 93 |
-
"use_log_f0": true,
|
| 94 |
-
"n_bins_melody": 256,
|
| 95 |
-
//# Quantization (0 for not quantization)
|
| 96 |
-
"output_melody_dim": 384,
|
| 97 |
-
"input_loudness_dim": 1,
|
| 98 |
-
"use_log_loudness": true,
|
| 99 |
-
"n_bins_loudness": 256,
|
| 100 |
-
"output_loudness_dim": 384,
|
| 101 |
-
"use_whisper": false,
|
| 102 |
-
"use_contentvec": false,
|
| 103 |
-
"use_wenet": false,
|
| 104 |
-
"use_mert": false,
|
| 105 |
-
"whisper_dim": 1024,
|
| 106 |
-
"contentvec_dim": 256,
|
| 107 |
-
"mert_dim": 256,
|
| 108 |
-
"wenet_dim": 512,
|
| 109 |
-
"content_encoder_dim": 384,
|
| 110 |
-
"output_singer_dim": 384,
|
| 111 |
-
"singer_table_size": 512,
|
| 112 |
-
"output_content_dim": 384,
|
| 113 |
-
"use_spkid": true
|
| 114 |
-
},
|
| 115 |
-
"comosvc": {
|
| 116 |
-
"distill": false,
|
| 117 |
-
// conformer encoder
|
| 118 |
-
"input_dim": 384,
|
| 119 |
-
"output_dim": 100,
|
| 120 |
-
"n_heads": 2,
|
| 121 |
-
"n_layers": 6,
|
| 122 |
-
"filter_channels": 512,
|
| 123 |
-
"dropout": 0.1,
|
| 124 |
-
// karras diffusion
|
| 125 |
-
"P_mean": -1.2,
|
| 126 |
-
"P_std": 1.2,
|
| 127 |
-
"sigma_data": 0.5,
|
| 128 |
-
"sigma_min": 0.002,
|
| 129 |
-
"sigma_max": 80,
|
| 130 |
-
"rho": 7,
|
| 131 |
-
"n_timesteps": 40,
|
| 132 |
-
},
|
| 133 |
-
"diffusion": {
|
| 134 |
-
// Diffusion steps encoder
|
| 135 |
-
"step_encoder": {
|
| 136 |
-
"dim_raw_embedding": 128,
|
| 137 |
-
"dim_hidden_layer": 512,
|
| 138 |
-
"activation": "SiLU",
|
| 139 |
-
"num_layer": 2,
|
| 140 |
-
"max_period": 10000
|
| 141 |
-
},
|
| 142 |
-
// Diffusion decoder
|
| 143 |
-
"model_type": "bidilconv",
|
| 144 |
-
// bidilconv, unet2d, TODO: unet1d
|
| 145 |
-
"bidilconv": {
|
| 146 |
-
"base_channel": 384,
|
| 147 |
-
"n_res_block": 20,
|
| 148 |
-
"conv_kernel_size": 3,
|
| 149 |
-
"dilation_cycle_length": 4,
|
| 150 |
-
// specially, 1 means no dilation
|
| 151 |
-
"conditioner_size": 100
|
| 152 |
-
}
|
| 153 |
-
},
|
| 154 |
-
},
|
| 155 |
-
"train": {
|
| 156 |
-
// Basic settings
|
| 157 |
-
"fast_steps": 0,
|
| 158 |
-
"batch_size": 32,
|
| 159 |
-
"gradient_accumulation_step": 1,
|
| 160 |
-
"max_epoch": -1,
|
| 161 |
-
// -1 means no limit
|
| 162 |
-
"save_checkpoint_stride": [
|
| 163 |
-
10,
|
| 164 |
-
100
|
| 165 |
-
],
|
| 166 |
-
// unit is epoch
|
| 167 |
-
"keep_last": [
|
| 168 |
-
3,
|
| 169 |
-
-1
|
| 170 |
-
],
|
| 171 |
-
// -1 means infinite, if one number will broadcast
|
| 172 |
-
"run_eval": [
|
| 173 |
-
false,
|
| 174 |
-
true
|
| 175 |
-
],
|
| 176 |
-
// if one number will broadcast
|
| 177 |
-
// Fix the random seed
|
| 178 |
-
"random_seed": 10086,
|
| 179 |
-
// Batchsampler
|
| 180 |
-
"sampler": {
|
| 181 |
-
"holistic_shuffle": true,
|
| 182 |
-
"drop_last": true
|
| 183 |
-
},
|
| 184 |
-
// Dataloader
|
| 185 |
-
"dataloader": {
|
| 186 |
-
"num_worker": 32,
|
| 187 |
-
"pin_memory": true
|
| 188 |
-
},
|
| 189 |
-
// Trackers
|
| 190 |
-
"tracker": [
|
| 191 |
-
"tensorboard"
|
| 192 |
-
// "wandb",
|
| 193 |
-
// "cometml",
|
| 194 |
-
// "mlflow",
|
| 195 |
-
],
|
| 196 |
-
// Optimizer
|
| 197 |
-
"optimizer": "AdamW",
|
| 198 |
-
"adamw": {
|
| 199 |
-
"lr": 4.0e-4
|
| 200 |
-
// nn model lr
|
| 201 |
-
},
|
| 202 |
-
// LR Scheduler
|
| 203 |
-
"scheduler": "ReduceLROnPlateau",
|
| 204 |
-
"reducelronplateau": {
|
| 205 |
-
"factor": 0.8,
|
| 206 |
-
"patience": 10,
|
| 207 |
-
// unit is epoch
|
| 208 |
-
"min_lr": 1.0e-4
|
| 209 |
-
}
|
| 210 |
-
},
|
| 211 |
-
"inference": {
|
| 212 |
-
"comosvc": {
|
| 213 |
-
"inference_steps": 40
|
| 214 |
-
}
|
| 215 |
-
}
|
| 216 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/diffusion.json
DELETED
|
@@ -1,227 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
// FIXME: THESE ARE LEGACY
|
| 3 |
-
"base_config": "config/base.json",
|
| 4 |
-
"model_type": "diffusion",
|
| 5 |
-
"task_type": "svc",
|
| 6 |
-
"use_custom_dataset": false,
|
| 7 |
-
"preprocess": {
|
| 8 |
-
// data augmentations
|
| 9 |
-
"use_pitch_shift": false,
|
| 10 |
-
"use_formant_shift": false,
|
| 11 |
-
"use_time_stretch": false,
|
| 12 |
-
"use_equalizer": false,
|
| 13 |
-
// acoustic features
|
| 14 |
-
"extract_mel": true,
|
| 15 |
-
"mel_min_max_norm": true,
|
| 16 |
-
"extract_pitch": true,
|
| 17 |
-
"pitch_extractor": "parselmouth",
|
| 18 |
-
"extract_uv": true,
|
| 19 |
-
"extract_energy": true,
|
| 20 |
-
// content features
|
| 21 |
-
"extract_whisper_feature": false,
|
| 22 |
-
"whisper_sample_rate": 16000,
|
| 23 |
-
"extract_contentvec_feature": false,
|
| 24 |
-
"contentvec_sample_rate": 16000,
|
| 25 |
-
"extract_wenet_feature": false,
|
| 26 |
-
"wenet_sample_rate": 16000,
|
| 27 |
-
"extract_mert_feature": false,
|
| 28 |
-
"mert_sample_rate": 16000,
|
| 29 |
-
// Default config for whisper
|
| 30 |
-
"whisper_frameshift": 0.01,
|
| 31 |
-
"whisper_downsample_rate": 2,
|
| 32 |
-
// Default config for content vector
|
| 33 |
-
"contentvec_frameshift": 0.02,
|
| 34 |
-
// Default config for mert
|
| 35 |
-
"mert_model": "m-a-p/MERT-v1-330M",
|
| 36 |
-
"mert_feature_layer": -1,
|
| 37 |
-
"mert_hop_size": 320,
|
| 38 |
-
// 24k
|
| 39 |
-
"mert_frameshit": 0.01333,
|
| 40 |
-
// 10ms
|
| 41 |
-
"wenet_frameshift": 0.01,
|
| 42 |
-
// wenetspeech is 4, gigaspeech is 6
|
| 43 |
-
"wenet_downsample_rate": 4,
|
| 44 |
-
// Default config
|
| 45 |
-
"n_mel": 100,
|
| 46 |
-
"win_size": 1024,
|
| 47 |
-
// todo
|
| 48 |
-
"hop_size": 256,
|
| 49 |
-
"sample_rate": 24000,
|
| 50 |
-
"n_fft": 1024,
|
| 51 |
-
// todo
|
| 52 |
-
"fmin": 0,
|
| 53 |
-
"fmax": 12000,
|
| 54 |
-
// todo
|
| 55 |
-
"f0_min": 50,
|
| 56 |
-
// ~C2
|
| 57 |
-
"f0_max": 1100,
|
| 58 |
-
//1100, // ~C6(1100), ~G5(800)
|
| 59 |
-
"pitch_bin": 256,
|
| 60 |
-
"pitch_max": 1100.0,
|
| 61 |
-
"pitch_min": 50.0,
|
| 62 |
-
"is_label": true,
|
| 63 |
-
"is_mu_law": true,
|
| 64 |
-
"bits": 8,
|
| 65 |
-
"mel_min_max_stats_dir": "mel_min_max_stats",
|
| 66 |
-
"whisper_dir": "whisper",
|
| 67 |
-
"contentvec_dir": "contentvec",
|
| 68 |
-
"wenet_dir": "wenet",
|
| 69 |
-
"mert_dir": "mert",
|
| 70 |
-
// Extract content features using dataloader
|
| 71 |
-
"pin_memory": true,
|
| 72 |
-
"num_workers": 8,
|
| 73 |
-
"content_feature_batch_size": 16,
|
| 74 |
-
// Features used for model training
|
| 75 |
-
"use_mel": true,
|
| 76 |
-
"use_min_max_norm_mel": true,
|
| 77 |
-
"use_frame_pitch": true,
|
| 78 |
-
"use_uv": true,
|
| 79 |
-
"use_frame_energy": true,
|
| 80 |
-
"use_log_scale_pitch": false,
|
| 81 |
-
"use_log_scale_energy": false,
|
| 82 |
-
"use_spkid": true,
|
| 83 |
-
// Meta file
|
| 84 |
-
"train_file": "train.json",
|
| 85 |
-
"valid_file": "test.json",
|
| 86 |
-
"spk2id": "singers.json",
|
| 87 |
-
"utt2spk": "utt2singer"
|
| 88 |
-
},
|
| 89 |
-
"model": {
|
| 90 |
-
"condition_encoder": {
|
| 91 |
-
"merge_mode": "add",
|
| 92 |
-
"input_melody_dim": 1,
|
| 93 |
-
"use_log_f0": true,
|
| 94 |
-
"n_bins_melody": 256,
|
| 95 |
-
//# Quantization (0 for not quantization)
|
| 96 |
-
"output_melody_dim": 384,
|
| 97 |
-
"input_loudness_dim": 1,
|
| 98 |
-
"use_log_loudness": true,
|
| 99 |
-
"n_bins_loudness": 256,
|
| 100 |
-
"output_loudness_dim": 384,
|
| 101 |
-
"use_whisper": false,
|
| 102 |
-
"use_contentvec": false,
|
| 103 |
-
"use_wenet": false,
|
| 104 |
-
"use_mert": false,
|
| 105 |
-
"whisper_dim": 1024,
|
| 106 |
-
"contentvec_dim": 256,
|
| 107 |
-
"mert_dim": 256,
|
| 108 |
-
"wenet_dim": 512,
|
| 109 |
-
"content_encoder_dim": 384,
|
| 110 |
-
"output_singer_dim": 384,
|
| 111 |
-
"singer_table_size": 512,
|
| 112 |
-
"output_content_dim": 384,
|
| 113 |
-
"use_spkid": true
|
| 114 |
-
},
|
| 115 |
-
// FIXME: FOLLOWING ARE NEW!!
|
| 116 |
-
"diffusion": {
|
| 117 |
-
"scheduler": "ddpm",
|
| 118 |
-
"scheduler_settings": {
|
| 119 |
-
"num_train_timesteps": 1000,
|
| 120 |
-
"beta_start": 1.0e-4,
|
| 121 |
-
"beta_end": 0.02,
|
| 122 |
-
"beta_schedule": "linear"
|
| 123 |
-
},
|
| 124 |
-
// Diffusion steps encoder
|
| 125 |
-
"step_encoder": {
|
| 126 |
-
"dim_raw_embedding": 128,
|
| 127 |
-
"dim_hidden_layer": 512,
|
| 128 |
-
"activation": "SiLU",
|
| 129 |
-
"num_layer": 2,
|
| 130 |
-
"max_period": 10000
|
| 131 |
-
},
|
| 132 |
-
// Diffusion decoder
|
| 133 |
-
"model_type": "bidilconv",
|
| 134 |
-
// bidilconv, unet2d, TODO: unet1d
|
| 135 |
-
"bidilconv": {
|
| 136 |
-
"base_channel": 384,
|
| 137 |
-
"n_res_block": 20,
|
| 138 |
-
"conv_kernel_size": 3,
|
| 139 |
-
"dilation_cycle_length": 4,
|
| 140 |
-
// specially, 1 means no dilation
|
| 141 |
-
"conditioner_size": 384
|
| 142 |
-
},
|
| 143 |
-
"unet2d": {
|
| 144 |
-
"in_channels": 1,
|
| 145 |
-
"out_channels": 1,
|
| 146 |
-
"down_block_types": [
|
| 147 |
-
"CrossAttnDownBlock2D",
|
| 148 |
-
"CrossAttnDownBlock2D",
|
| 149 |
-
"CrossAttnDownBlock2D",
|
| 150 |
-
"DownBlock2D"
|
| 151 |
-
],
|
| 152 |
-
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
| 153 |
-
"up_block_types": [
|
| 154 |
-
"UpBlock2D",
|
| 155 |
-
"CrossAttnUpBlock2D",
|
| 156 |
-
"CrossAttnUpBlock2D",
|
| 157 |
-
"CrossAttnUpBlock2D"
|
| 158 |
-
],
|
| 159 |
-
"only_cross_attention": false
|
| 160 |
-
}
|
| 161 |
-
}
|
| 162 |
-
},
|
| 163 |
-
// FIXME: FOLLOWING ARE NEW!!
|
| 164 |
-
"train": {
|
| 165 |
-
// Basic settings
|
| 166 |
-
"batch_size": 64,
|
| 167 |
-
"gradient_accumulation_step": 1,
|
| 168 |
-
"max_epoch": -1,
|
| 169 |
-
// -1 means no limit
|
| 170 |
-
"save_checkpoint_stride": [
|
| 171 |
-
5,
|
| 172 |
-
20
|
| 173 |
-
],
|
| 174 |
-
// unit is epoch
|
| 175 |
-
"keep_last": [
|
| 176 |
-
3,
|
| 177 |
-
-1
|
| 178 |
-
],
|
| 179 |
-
// -1 means infinite, if one number will broadcast
|
| 180 |
-
"run_eval": [
|
| 181 |
-
false,
|
| 182 |
-
true
|
| 183 |
-
],
|
| 184 |
-
// if one number will broadcast
|
| 185 |
-
// Fix the random seed
|
| 186 |
-
"random_seed": 10086,
|
| 187 |
-
// Batchsampler
|
| 188 |
-
"sampler": {
|
| 189 |
-
"holistic_shuffle": true,
|
| 190 |
-
"drop_last": true
|
| 191 |
-
},
|
| 192 |
-
// Dataloader
|
| 193 |
-
"dataloader": {
|
| 194 |
-
"num_worker": 32,
|
| 195 |
-
"pin_memory": true
|
| 196 |
-
},
|
| 197 |
-
// Trackers
|
| 198 |
-
"tracker": [
|
| 199 |
-
"tensorboard"
|
| 200 |
-
// "wandb",
|
| 201 |
-
// "cometml",
|
| 202 |
-
// "mlflow",
|
| 203 |
-
],
|
| 204 |
-
// Optimizer
|
| 205 |
-
"optimizer": "AdamW",
|
| 206 |
-
"adamw": {
|
| 207 |
-
"lr": 4.0e-4
|
| 208 |
-
// nn model lr
|
| 209 |
-
},
|
| 210 |
-
// LR Scheduler
|
| 211 |
-
"scheduler": "ReduceLROnPlateau",
|
| 212 |
-
"reducelronplateau": {
|
| 213 |
-
"factor": 0.8,
|
| 214 |
-
"patience": 10,
|
| 215 |
-
// unit is epoch
|
| 216 |
-
"min_lr": 1.0e-4
|
| 217 |
-
}
|
| 218 |
-
},
|
| 219 |
-
"inference": {
|
| 220 |
-
"diffusion": {
|
| 221 |
-
"scheduler": "pndm",
|
| 222 |
-
"scheduler_settings": {
|
| 223 |
-
"num_inference_timesteps": 1000
|
| 224 |
-
}
|
| 225 |
-
}
|
| 226 |
-
}
|
| 227 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/fs2.json
DELETED
|
@@ -1,117 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/tts.json",
|
| 3 |
-
"model_type": "FastSpeech2",
|
| 4 |
-
"task_type": "tts",
|
| 5 |
-
"dataset": ["LJSpeech"],
|
| 6 |
-
"preprocess": {
|
| 7 |
-
// acoustic features
|
| 8 |
-
"extract_audio": true,
|
| 9 |
-
"extract_mel": true,
|
| 10 |
-
"mel_extract_mode": "taco",
|
| 11 |
-
"mel_min_max_norm": false,
|
| 12 |
-
"extract_pitch": true,
|
| 13 |
-
"extract_uv": false,
|
| 14 |
-
"pitch_extractor": "dio",
|
| 15 |
-
"extract_energy": true,
|
| 16 |
-
"energy_extract_mode": "from_tacotron_stft",
|
| 17 |
-
"extract_duration": true,
|
| 18 |
-
"use_phone": true,
|
| 19 |
-
"pitch_norm": true,
|
| 20 |
-
"energy_norm": true,
|
| 21 |
-
"pitch_remove_outlier": true,
|
| 22 |
-
"energy_remove_outlier": true,
|
| 23 |
-
|
| 24 |
-
// Default config
|
| 25 |
-
"n_mel": 80,
|
| 26 |
-
"win_size": 1024, // todo
|
| 27 |
-
"hop_size": 256,
|
| 28 |
-
"sample_rate": 22050,
|
| 29 |
-
"n_fft": 1024, // todo
|
| 30 |
-
"fmin": 0,
|
| 31 |
-
"fmax": 8000, // todo
|
| 32 |
-
"raw_data": "raw_data",
|
| 33 |
-
"text_cleaners": ["english_cleaners"],
|
| 34 |
-
"f0_min": 71, // ~C2
|
| 35 |
-
"f0_max": 800, //1100, // ~C6(1100), ~G5(800)
|
| 36 |
-
"pitch_bin": 256,
|
| 37 |
-
"pitch_max": 1100.0,
|
| 38 |
-
"pitch_min": 50.0,
|
| 39 |
-
"is_label": true,
|
| 40 |
-
"is_mu_law": true,
|
| 41 |
-
"bits": 8,
|
| 42 |
-
|
| 43 |
-
"mel_min_max_stats_dir": "mel_min_max_stats",
|
| 44 |
-
"whisper_dir": "whisper",
|
| 45 |
-
"content_vector_dir": "content_vector",
|
| 46 |
-
"wenet_dir": "wenet",
|
| 47 |
-
"mert_dir": "mert",
|
| 48 |
-
"spk2id":"spk2id.json",
|
| 49 |
-
"utt2spk":"utt2spk",
|
| 50 |
-
|
| 51 |
-
// Features used for model training
|
| 52 |
-
"use_mel": true,
|
| 53 |
-
"use_min_max_norm_mel": false,
|
| 54 |
-
"use_frame_pitch": false,
|
| 55 |
-
"use_frame_energy": false,
|
| 56 |
-
"use_phone_pitch": true,
|
| 57 |
-
"use_phone_energy": true,
|
| 58 |
-
"use_log_scale_pitch": false,
|
| 59 |
-
"use_log_scale_energy": false,
|
| 60 |
-
"use_spkid": false,
|
| 61 |
-
"align_mel_duration": true,
|
| 62 |
-
"text_cleaners": ["english_cleaners"]
|
| 63 |
-
},
|
| 64 |
-
"model": {
|
| 65 |
-
// Settings for transformer
|
| 66 |
-
"transformer": {
|
| 67 |
-
"encoder_layer": 4,
|
| 68 |
-
"encoder_head": 2,
|
| 69 |
-
"encoder_hidden": 256,
|
| 70 |
-
"decoder_layer": 6,
|
| 71 |
-
"decoder_head": 2,
|
| 72 |
-
"decoder_hidden": 256,
|
| 73 |
-
"conv_filter_size": 1024,
|
| 74 |
-
"conv_kernel_size": [9, 1],
|
| 75 |
-
"encoder_dropout": 0.2,
|
| 76 |
-
"decoder_dropout": 0.2
|
| 77 |
-
},
|
| 78 |
-
|
| 79 |
-
// Settings for variance_predictor
|
| 80 |
-
"variance_predictor":{
|
| 81 |
-
"filter_size": 256,
|
| 82 |
-
"kernel_size": 3,
|
| 83 |
-
"dropout": 0.5
|
| 84 |
-
},
|
| 85 |
-
"variance_embedding":{
|
| 86 |
-
"pitch_quantization": "linear", // support 'linear' or 'log', 'log' is allowed only if the pitch values are not normalized during preprocessing
|
| 87 |
-
"energy_quantization": "linear", // support 'linear' or 'log', 'log' is allowed only if the energy values are not normalized during preprocessing
|
| 88 |
-
"n_bins": 256
|
| 89 |
-
},
|
| 90 |
-
"max_seq_len": 1000
|
| 91 |
-
},
|
| 92 |
-
"train":{
|
| 93 |
-
"batch_size": 16,
|
| 94 |
-
"sort_sample": true,
|
| 95 |
-
"drop_last": true,
|
| 96 |
-
"group_size": 4,
|
| 97 |
-
"grad_clip_thresh": 1.0,
|
| 98 |
-
"dataloader": {
|
| 99 |
-
"num_worker": 8,
|
| 100 |
-
"pin_memory": true
|
| 101 |
-
},
|
| 102 |
-
"lr_scheduler":{
|
| 103 |
-
"num_warmup": 4000
|
| 104 |
-
},
|
| 105 |
-
// LR Scheduler
|
| 106 |
-
"scheduler": "NoamLR",
|
| 107 |
-
// Optimizer
|
| 108 |
-
"optimizer": "Adam",
|
| 109 |
-
"adam": {
|
| 110 |
-
"lr": 0.0625,
|
| 111 |
-
"betas": [0.9, 0.98],
|
| 112 |
-
"eps": 0.000000001,
|
| 113 |
-
"weight_decay": 0.0
|
| 114 |
-
},
|
| 115 |
-
}
|
| 116 |
-
|
| 117 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/transformer.json
DELETED
|
@@ -1,180 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/base.json",
|
| 3 |
-
"model_type": "Transformer",
|
| 4 |
-
"task_type": "svc",
|
| 5 |
-
"use_custom_dataset": false,
|
| 6 |
-
"preprocess": {
|
| 7 |
-
// data augmentations
|
| 8 |
-
"use_pitch_shift": false,
|
| 9 |
-
"use_formant_shift": false,
|
| 10 |
-
"use_time_stretch": false,
|
| 11 |
-
"use_equalizer": false,
|
| 12 |
-
// acoustic features
|
| 13 |
-
"extract_mel": true,
|
| 14 |
-
"mel_min_max_norm": true,
|
| 15 |
-
"extract_pitch": true,
|
| 16 |
-
"pitch_extractor": "parselmouth",
|
| 17 |
-
"extract_uv": true,
|
| 18 |
-
"extract_energy": true,
|
| 19 |
-
// content features
|
| 20 |
-
"extract_whisper_feature": false,
|
| 21 |
-
"whisper_sample_rate": 16000,
|
| 22 |
-
"extract_contentvec_feature": false,
|
| 23 |
-
"contentvec_sample_rate": 16000,
|
| 24 |
-
"extract_wenet_feature": false,
|
| 25 |
-
"wenet_sample_rate": 16000,
|
| 26 |
-
"extract_mert_feature": false,
|
| 27 |
-
"mert_sample_rate": 16000,
|
| 28 |
-
// Default config for whisper
|
| 29 |
-
"whisper_frameshift": 0.01,
|
| 30 |
-
"whisper_downsample_rate": 2,
|
| 31 |
-
// Default config for content vector
|
| 32 |
-
"contentvec_frameshift": 0.02,
|
| 33 |
-
// Default config for mert
|
| 34 |
-
"mert_model": "m-a-p/MERT-v1-330M",
|
| 35 |
-
"mert_feature_layer": -1,
|
| 36 |
-
"mert_hop_size": 320,
|
| 37 |
-
// 24k
|
| 38 |
-
"mert_frameshit": 0.01333,
|
| 39 |
-
// 10ms
|
| 40 |
-
"wenet_frameshift": 0.01,
|
| 41 |
-
// wenetspeech is 4, gigaspeech is 6
|
| 42 |
-
"wenet_downsample_rate": 4,
|
| 43 |
-
// Default config
|
| 44 |
-
"n_mel": 100,
|
| 45 |
-
"win_size": 1024,
|
| 46 |
-
// todo
|
| 47 |
-
"hop_size": 256,
|
| 48 |
-
"sample_rate": 24000,
|
| 49 |
-
"n_fft": 1024,
|
| 50 |
-
// todo
|
| 51 |
-
"fmin": 0,
|
| 52 |
-
"fmax": 12000,
|
| 53 |
-
// todo
|
| 54 |
-
"f0_min": 50,
|
| 55 |
-
// ~C2
|
| 56 |
-
"f0_max": 1100,
|
| 57 |
-
//1100, // ~C6(1100), ~G5(800)
|
| 58 |
-
"pitch_bin": 256,
|
| 59 |
-
"pitch_max": 1100.0,
|
| 60 |
-
"pitch_min": 50.0,
|
| 61 |
-
"is_label": true,
|
| 62 |
-
"is_mu_law": true,
|
| 63 |
-
"bits": 8,
|
| 64 |
-
"mel_min_max_stats_dir": "mel_min_max_stats",
|
| 65 |
-
"whisper_dir": "whisper",
|
| 66 |
-
"contentvec_dir": "contentvec",
|
| 67 |
-
"wenet_dir": "wenet",
|
| 68 |
-
"mert_dir": "mert",
|
| 69 |
-
// Extract content features using dataloader
|
| 70 |
-
"pin_memory": true,
|
| 71 |
-
"num_workers": 8,
|
| 72 |
-
"content_feature_batch_size": 16,
|
| 73 |
-
// Features used for model training
|
| 74 |
-
"use_mel": true,
|
| 75 |
-
"use_min_max_norm_mel": true,
|
| 76 |
-
"use_frame_pitch": true,
|
| 77 |
-
"use_uv": true,
|
| 78 |
-
"use_frame_energy": true,
|
| 79 |
-
"use_log_scale_pitch": false,
|
| 80 |
-
"use_log_scale_energy": false,
|
| 81 |
-
"use_spkid": true,
|
| 82 |
-
// Meta file
|
| 83 |
-
"train_file": "train.json",
|
| 84 |
-
"valid_file": "test.json",
|
| 85 |
-
"spk2id": "singers.json",
|
| 86 |
-
"utt2spk": "utt2singer"
|
| 87 |
-
},
|
| 88 |
-
"model": {
|
| 89 |
-
"condition_encoder": {
|
| 90 |
-
"merge_mode": "add",
|
| 91 |
-
"input_melody_dim": 1,
|
| 92 |
-
"use_log_f0": true,
|
| 93 |
-
"n_bins_melody": 256,
|
| 94 |
-
//# Quantization (0 for not quantization)
|
| 95 |
-
"output_melody_dim": 384,
|
| 96 |
-
"input_loudness_dim": 1,
|
| 97 |
-
"use_log_loudness": true,
|
| 98 |
-
"n_bins_loudness": 256,
|
| 99 |
-
"output_loudness_dim": 384,
|
| 100 |
-
"use_whisper": false,
|
| 101 |
-
"use_contentvec": true,
|
| 102 |
-
"use_wenet": false,
|
| 103 |
-
"use_mert": false,
|
| 104 |
-
"whisper_dim": 1024,
|
| 105 |
-
"contentvec_dim": 256,
|
| 106 |
-
"mert_dim": 256,
|
| 107 |
-
"wenet_dim": 512,
|
| 108 |
-
"content_encoder_dim": 384,
|
| 109 |
-
"output_singer_dim": 384,
|
| 110 |
-
"singer_table_size": 512,
|
| 111 |
-
"output_content_dim": 384,
|
| 112 |
-
"use_spkid": true
|
| 113 |
-
},
|
| 114 |
-
"transformer": {
|
| 115 |
-
"type": "conformer",
|
| 116 |
-
// 'conformer' or 'transformer'
|
| 117 |
-
"input_dim": 384,
|
| 118 |
-
"output_dim": 100,
|
| 119 |
-
"n_heads": 2,
|
| 120 |
-
"n_layers": 6,
|
| 121 |
-
"filter_channels": 512,
|
| 122 |
-
"dropout": 0.1,
|
| 123 |
-
}
|
| 124 |
-
},
|
| 125 |
-
"train": {
|
| 126 |
-
// Basic settings
|
| 127 |
-
"batch_size": 64,
|
| 128 |
-
"gradient_accumulation_step": 1,
|
| 129 |
-
"max_epoch": -1,
|
| 130 |
-
// -1 means no limit
|
| 131 |
-
"save_checkpoint_stride": [
|
| 132 |
-
10,
|
| 133 |
-
100
|
| 134 |
-
],
|
| 135 |
-
// unit is epoch
|
| 136 |
-
"keep_last": [
|
| 137 |
-
3,
|
| 138 |
-
-1
|
| 139 |
-
],
|
| 140 |
-
// -1 means infinite, if one number will broadcast
|
| 141 |
-
"run_eval": [
|
| 142 |
-
false,
|
| 143 |
-
true
|
| 144 |
-
],
|
| 145 |
-
// if one number will broadcast
|
| 146 |
-
// Fix the random seed
|
| 147 |
-
"random_seed": 10086,
|
| 148 |
-
// Batchsampler
|
| 149 |
-
"sampler": {
|
| 150 |
-
"holistic_shuffle": true,
|
| 151 |
-
"drop_last": true
|
| 152 |
-
},
|
| 153 |
-
// Dataloader
|
| 154 |
-
"dataloader": {
|
| 155 |
-
"num_worker": 32,
|
| 156 |
-
"pin_memory": true
|
| 157 |
-
},
|
| 158 |
-
// Trackers
|
| 159 |
-
"tracker": [
|
| 160 |
-
"tensorboard"
|
| 161 |
-
// "wandb",
|
| 162 |
-
// "cometml",
|
| 163 |
-
// "mlflow",
|
| 164 |
-
],
|
| 165 |
-
// Optimizer
|
| 166 |
-
"optimizer": "AdamW",
|
| 167 |
-
"adamw": {
|
| 168 |
-
"lr": 4.0e-4
|
| 169 |
-
// nn model lr
|
| 170 |
-
},
|
| 171 |
-
// LR Scheduler
|
| 172 |
-
"scheduler": "ReduceLROnPlateau",
|
| 173 |
-
"reducelronplateau": {
|
| 174 |
-
"factor": 0.8,
|
| 175 |
-
"patience": 10,
|
| 176 |
-
// unit is epoch
|
| 177 |
-
"min_lr": 1.0e-4
|
| 178 |
-
}
|
| 179 |
-
}
|
| 180 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/tts.json
DELETED
|
@@ -1,23 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/base.json",
|
| 3 |
-
"supported_model_type": [
|
| 4 |
-
"Fastspeech2",
|
| 5 |
-
"VITS",
|
| 6 |
-
"VALLE",
|
| 7 |
-
],
|
| 8 |
-
"task_type": "tts",
|
| 9 |
-
"preprocess": {
|
| 10 |
-
"language": "en-us",
|
| 11 |
-
// linguistic features
|
| 12 |
-
"extract_phone": true,
|
| 13 |
-
"phone_extractor": "espeak", // "espeak, pypinyin, pypinyin_initials_finals, lexicon (only for language=en-us right now)"
|
| 14 |
-
"lexicon_path": "./text/lexicon/librispeech-lexicon.txt",
|
| 15 |
-
// Directory names of processed data or extracted features
|
| 16 |
-
"phone_dir": "phones",
|
| 17 |
-
"use_phone": true,
|
| 18 |
-
},
|
| 19 |
-
"model": {
|
| 20 |
-
"text_token_num": 512,
|
| 21 |
-
}
|
| 22 |
-
|
| 23 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/valle.json
DELETED
|
@@ -1,52 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/tts.json",
|
| 3 |
-
"model_type": "VALLE",
|
| 4 |
-
"task_type": "tts",
|
| 5 |
-
"dataset": [
|
| 6 |
-
"libritts"
|
| 7 |
-
],
|
| 8 |
-
"preprocess": {
|
| 9 |
-
"extract_phone": true,
|
| 10 |
-
"phone_extractor": "espeak", // phoneme extractor: espeak, pypinyin, pypinyin_initials_finals or lexicon
|
| 11 |
-
"extract_acoustic_token": true,
|
| 12 |
-
"acoustic_token_extractor": "Encodec", // acoustic token extractor: encodec, dac(todo)
|
| 13 |
-
"acoustic_token_dir": "acoutic_tokens",
|
| 14 |
-
"use_text": false,
|
| 15 |
-
"use_phone": true,
|
| 16 |
-
"use_acoustic_token": true,
|
| 17 |
-
"symbols_dict": "symbols.dict",
|
| 18 |
-
"min_duration": 0.5, // the duration lowerbound to filter the audio with duration < min_duration
|
| 19 |
-
"max_duration": 14, // the duration uperbound to filter the audio with duration > max_duration.
|
| 20 |
-
"sampling_rate": 24000,
|
| 21 |
-
},
|
| 22 |
-
"model": {
|
| 23 |
-
"text_token_num": 512,
|
| 24 |
-
"audio_token_num": 1024,
|
| 25 |
-
"decoder_dim": 1024, // embedding dimension of the decoder model
|
| 26 |
-
"nhead": 16, // number of attention heads in the decoder layers
|
| 27 |
-
"num_decoder_layers": 12, // number of decoder layers
|
| 28 |
-
"norm_first": true, // pre or post Normalization.
|
| 29 |
-
"add_prenet": false, // whether add PreNet after Inputs
|
| 30 |
-
"prefix_mode": 0, // mode for how to prefix VALL-E NAR Decoder, 0: no prefix, 1: 0 to random, 2: random to random, 4: chunk of pre or post utterance
|
| 31 |
-
"share_embedding": true, // share the parameters of the output projection layer with the parameters of the acoustic embedding
|
| 32 |
-
"nar_scale_factor": 1, // model scale factor which will be assigned different meanings in different models
|
| 33 |
-
"prepend_bos": false, // whether prepend <BOS> to the acoustic tokens -> AR Decoder inputs
|
| 34 |
-
"num_quantizers": 8, // numbert of the audio quantization layers
|
| 35 |
-
// "scaling_xformers": false, // Apply Reworked Conformer scaling on Transformers
|
| 36 |
-
},
|
| 37 |
-
"train": {
|
| 38 |
-
"ddp": false,
|
| 39 |
-
"train_stage": 1, // 0: train all modules, For VALL_E, support 1: AR Decoder 2: NAR Decoder(s)
|
| 40 |
-
"max_epoch": 20,
|
| 41 |
-
"optimizer": "ScaledAdam",
|
| 42 |
-
"scheduler": "Eden",
|
| 43 |
-
"warmup_steps": 200, // number of steps that affects how rapidly the learning rate decreases
|
| 44 |
-
"base_lr": 0.05, // base learning rate."
|
| 45 |
-
"valid_interval": 1000,
|
| 46 |
-
"log_epoch_step": 1000,
|
| 47 |
-
"save_checkpoint_stride": [
|
| 48 |
-
1,
|
| 49 |
-
1
|
| 50 |
-
]
|
| 51 |
-
}
|
| 52 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/vits.json
DELETED
|
@@ -1,101 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/tts.json",
|
| 3 |
-
"model_type": "VITS",
|
| 4 |
-
"task_type": "tts",
|
| 5 |
-
"preprocess": {
|
| 6 |
-
"extract_phone": true,
|
| 7 |
-
"extract_mel": true,
|
| 8 |
-
"n_mel": 80,
|
| 9 |
-
"fmin": 0,
|
| 10 |
-
"fmax": null,
|
| 11 |
-
"extract_linear_spec": true,
|
| 12 |
-
"extract_audio": true,
|
| 13 |
-
"use_linear": true,
|
| 14 |
-
"use_mel": true,
|
| 15 |
-
"use_audio": true,
|
| 16 |
-
"use_text": false,
|
| 17 |
-
"use_phone": true,
|
| 18 |
-
"lexicon_path": "./text/lexicon/librispeech-lexicon.txt",
|
| 19 |
-
"n_fft": 1024,
|
| 20 |
-
"win_size": 1024,
|
| 21 |
-
"hop_size": 256,
|
| 22 |
-
"segment_size": 8192,
|
| 23 |
-
"text_cleaners": [
|
| 24 |
-
"english_cleaners"
|
| 25 |
-
]
|
| 26 |
-
},
|
| 27 |
-
"model": {
|
| 28 |
-
"text_token_num": 512,
|
| 29 |
-
"inter_channels": 192,
|
| 30 |
-
"hidden_channels": 192,
|
| 31 |
-
"filter_channels": 768,
|
| 32 |
-
"n_heads": 2,
|
| 33 |
-
"n_layers": 6,
|
| 34 |
-
"kernel_size": 3,
|
| 35 |
-
"p_dropout": 0.1,
|
| 36 |
-
"resblock": "1",
|
| 37 |
-
"resblock_kernel_sizes": [
|
| 38 |
-
3,
|
| 39 |
-
7,
|
| 40 |
-
11
|
| 41 |
-
],
|
| 42 |
-
"resblock_dilation_sizes": [
|
| 43 |
-
[
|
| 44 |
-
1,
|
| 45 |
-
3,
|
| 46 |
-
5
|
| 47 |
-
],
|
| 48 |
-
[
|
| 49 |
-
1,
|
| 50 |
-
3,
|
| 51 |
-
5
|
| 52 |
-
],
|
| 53 |
-
[
|
| 54 |
-
1,
|
| 55 |
-
3,
|
| 56 |
-
5
|
| 57 |
-
]
|
| 58 |
-
],
|
| 59 |
-
"upsample_rates": [
|
| 60 |
-
8,
|
| 61 |
-
8,
|
| 62 |
-
2,
|
| 63 |
-
2
|
| 64 |
-
],
|
| 65 |
-
"upsample_initial_channel": 512,
|
| 66 |
-
"upsample_kernel_sizes": [
|
| 67 |
-
16,
|
| 68 |
-
16,
|
| 69 |
-
4,
|
| 70 |
-
4
|
| 71 |
-
],
|
| 72 |
-
"n_layers_q": 3,
|
| 73 |
-
"use_spectral_norm": false,
|
| 74 |
-
"n_speakers": 0, // number of speakers, while be automatically set if n_speakers is 0 and multi_speaker_training is true
|
| 75 |
-
"gin_channels": 256,
|
| 76 |
-
"use_sdp": true
|
| 77 |
-
},
|
| 78 |
-
"train": {
|
| 79 |
-
"fp16_run": true,
|
| 80 |
-
"learning_rate": 2e-4,
|
| 81 |
-
"betas": [
|
| 82 |
-
0.8,
|
| 83 |
-
0.99
|
| 84 |
-
],
|
| 85 |
-
"eps": 1e-9,
|
| 86 |
-
"batch_size": 16,
|
| 87 |
-
"lr_decay": 0.999875,
|
| 88 |
-
// "segment_size": 8192,
|
| 89 |
-
"init_lr_ratio": 1,
|
| 90 |
-
"warmup_epochs": 0,
|
| 91 |
-
"c_mel": 45,
|
| 92 |
-
"c_kl": 1.0,
|
| 93 |
-
"AdamW": {
|
| 94 |
-
"betas": [
|
| 95 |
-
0.8,
|
| 96 |
-
0.99
|
| 97 |
-
],
|
| 98 |
-
"eps": 1e-9,
|
| 99 |
-
}
|
| 100 |
-
}
|
| 101 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config/vocoder.json
DELETED
|
@@ -1,84 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/base.json",
|
| 3 |
-
"dataset": [
|
| 4 |
-
"LJSpeech",
|
| 5 |
-
"LibriTTS",
|
| 6 |
-
"opencpop",
|
| 7 |
-
"m4singer",
|
| 8 |
-
"svcc",
|
| 9 |
-
"svcceval",
|
| 10 |
-
"pjs",
|
| 11 |
-
"opensinger",
|
| 12 |
-
"popbutfy",
|
| 13 |
-
"nus48e",
|
| 14 |
-
"popcs",
|
| 15 |
-
"kising",
|
| 16 |
-
"csd",
|
| 17 |
-
"opera",
|
| 18 |
-
"vctk",
|
| 19 |
-
"lijian",
|
| 20 |
-
"cdmusiceval"
|
| 21 |
-
],
|
| 22 |
-
"task_type": "vocoder",
|
| 23 |
-
"preprocess": {
|
| 24 |
-
// acoustic features
|
| 25 |
-
"extract_mel": true,
|
| 26 |
-
"extract_pitch": false,
|
| 27 |
-
"extract_uv": false,
|
| 28 |
-
"extract_audio": true,
|
| 29 |
-
"extract_label": false,
|
| 30 |
-
"extract_one_hot": false,
|
| 31 |
-
"extract_amplitude_phase": false,
|
| 32 |
-
"pitch_extractor": "parselmouth",
|
| 33 |
-
// Settings for data preprocessing
|
| 34 |
-
"n_mel": 100,
|
| 35 |
-
"win_size": 1024,
|
| 36 |
-
"hop_size": 256,
|
| 37 |
-
"sample_rate": 24000,
|
| 38 |
-
"n_fft": 1024,
|
| 39 |
-
"fmin": 0,
|
| 40 |
-
"fmax": 12000,
|
| 41 |
-
"f0_min": 50,
|
| 42 |
-
"f0_max": 1100,
|
| 43 |
-
"pitch_bin": 256,
|
| 44 |
-
"pitch_max": 1100.0,
|
| 45 |
-
"pitch_min": 50.0,
|
| 46 |
-
"is_mu_law": false,
|
| 47 |
-
"bits": 8,
|
| 48 |
-
"cut_mel_frame": 32,
|
| 49 |
-
// Directory names of processed data or extracted features
|
| 50 |
-
"spk2id": "singers.json",
|
| 51 |
-
// Features used for model training
|
| 52 |
-
"use_mel": true,
|
| 53 |
-
"use_frame_pitch": false,
|
| 54 |
-
"use_uv": false,
|
| 55 |
-
"use_audio": true,
|
| 56 |
-
"use_label": false,
|
| 57 |
-
"use_one_hot": false,
|
| 58 |
-
"train_file": "train.json",
|
| 59 |
-
"valid_file": "test.json"
|
| 60 |
-
},
|
| 61 |
-
"train": {
|
| 62 |
-
"random_seed": 114514,
|
| 63 |
-
"batch_size": 64,
|
| 64 |
-
"gradient_accumulation_step": 1,
|
| 65 |
-
"max_epoch": 1000000,
|
| 66 |
-
"save_checkpoint_stride": [
|
| 67 |
-
20
|
| 68 |
-
],
|
| 69 |
-
"run_eval": [
|
| 70 |
-
true
|
| 71 |
-
],
|
| 72 |
-
"sampler": {
|
| 73 |
-
"holistic_shuffle": true,
|
| 74 |
-
"drop_last": true
|
| 75 |
-
},
|
| 76 |
-
"dataloader": {
|
| 77 |
-
"num_worker": 4,
|
| 78 |
-
"pin_memory": true
|
| 79 |
-
},
|
| 80 |
-
"tracker": [
|
| 81 |
-
"tensorboard"
|
| 82 |
-
],
|
| 83 |
-
}
|
| 84 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/svc/MultipleContentsSVC/README.md
DELETED
|
@@ -1,153 +0,0 @@
|
|
| 1 |
-
# Leveraging Content-based Features from Multiple Acoustic Models for Singing Voice Conversion
|
| 2 |
-
|
| 3 |
-
[](https://arxiv.org/abs/2310.11160)
|
| 4 |
-
[](https://www.zhangxueyao.com/data/MultipleContentsSVC/index.html)
|
| 5 |
-
|
| 6 |
-
<br>
|
| 7 |
-
<div align="center">
|
| 8 |
-
<img src="../../../imgs/svc/MultipleContentsSVC.png" width="85%">
|
| 9 |
-
</div>
|
| 10 |
-
<br>
|
| 11 |
-
|
| 12 |
-
This is the official implementation of the paper "[Leveraging Content-based Features from Multiple Acoustic Models for Singing Voice Conversion](https://arxiv.org/abs/2310.11160)" (NeurIPS 2023 Workshop on Machine Learning for Audio). Specially,
|
| 13 |
-
|
| 14 |
-
- The muptile content features are from [Whipser](https://github.com/wenet-e2e/wenet) and [ContentVec](https://github.com/auspicious3000/contentvec).
|
| 15 |
-
- The acoustic model is based on Bidirectional Non-Causal Dilated CNN (called `DiffWaveNetSVC` in Amphion), which is similar to [WaveNet](https://arxiv.org/pdf/1609.03499.pdf), [DiffWave](https://openreview.net/forum?id=a-xFK8Ymz5J), and [DiffSVC](https://ieeexplore.ieee.org/document/9688219).
|
| 16 |
-
- The vocoder is [BigVGAN](https://github.com/NVIDIA/BigVGAN) architecture and we fine-tuned it in over 120 hours singing voice data.
|
| 17 |
-
|
| 18 |
-
There are four stages in total:
|
| 19 |
-
|
| 20 |
-
1. Data preparation
|
| 21 |
-
2. Features extraction
|
| 22 |
-
3. Training
|
| 23 |
-
4. Inference/conversion
|
| 24 |
-
|
| 25 |
-
> **NOTE:** You need to run every command of this recipe in the `Amphion` root path:
|
| 26 |
-
> ```bash
|
| 27 |
-
> cd Amphion
|
| 28 |
-
> ```
|
| 29 |
-
|
| 30 |
-
## 1. Data Preparation
|
| 31 |
-
|
| 32 |
-
### Dataset Download
|
| 33 |
-
|
| 34 |
-
By default, we utilize the five datasets for training: M4Singer, Opencpop, OpenSinger, SVCC, and VCTK. How to download them is detailed [here](../../datasets/README.md).
|
| 35 |
-
|
| 36 |
-
### Configuration
|
| 37 |
-
|
| 38 |
-
Specify the dataset paths in `exp_config.json`. Note that you can change the `dataset` list to use your preferred datasets.
|
| 39 |
-
|
| 40 |
-
```json
|
| 41 |
-
"dataset": [
|
| 42 |
-
"m4singer",
|
| 43 |
-
"opencpop",
|
| 44 |
-
"opensinger",
|
| 45 |
-
"svcc",
|
| 46 |
-
"vctk"
|
| 47 |
-
],
|
| 48 |
-
"dataset_path": {
|
| 49 |
-
// TODO: Fill in your dataset path
|
| 50 |
-
"m4singer": "[M4Singer dataset path]",
|
| 51 |
-
"opencpop": "[Opencpop dataset path]",
|
| 52 |
-
"opensinger": "[OpenSinger dataset path]",
|
| 53 |
-
"svcc": "[SVCC dataset path]",
|
| 54 |
-
"vctk": "[VCTK dataset path]"
|
| 55 |
-
},
|
| 56 |
-
```
|
| 57 |
-
|
| 58 |
-
## 2. Features Extraction
|
| 59 |
-
|
| 60 |
-
### Content-based Pretrained Models Download
|
| 61 |
-
|
| 62 |
-
By default, we utilize the Whisper and ContentVec to extract content features. How to download them is detailed [here](../../../pretrained/README.md).
|
| 63 |
-
|
| 64 |
-
### Configuration
|
| 65 |
-
|
| 66 |
-
Specify the dataset path and the output path for saving the processed data and the training model in `exp_config.json`:
|
| 67 |
-
|
| 68 |
-
```json
|
| 69 |
-
// TODO: Fill in the output log path. The default value is "Amphion/ckpts/svc"
|
| 70 |
-
"log_dir": "ckpts/svc",
|
| 71 |
-
"preprocess": {
|
| 72 |
-
// TODO: Fill in the output data path. The default value is "Amphion/data"
|
| 73 |
-
"processed_dir": "data",
|
| 74 |
-
...
|
| 75 |
-
},
|
| 76 |
-
```
|
| 77 |
-
|
| 78 |
-
### Run
|
| 79 |
-
|
| 80 |
-
Run the `run.sh` as the preproces stage (set `--stage 1`).
|
| 81 |
-
|
| 82 |
-
```bash
|
| 83 |
-
sh egs/svc/MultipleContentsSVC/run.sh --stage 1
|
| 84 |
-
```
|
| 85 |
-
|
| 86 |
-
> **NOTE:** The `CUDA_VISIBLE_DEVICES` is set as `"0"` in default. You can change it when running `run.sh` by specifying such as `--gpu "1"`.
|
| 87 |
-
|
| 88 |
-
## 3. Training
|
| 89 |
-
|
| 90 |
-
### Configuration
|
| 91 |
-
|
| 92 |
-
We provide the default hyparameters in the `exp_config.json`. They can work on single NVIDIA-24g GPU. You can adjust them based on you GPU machines.
|
| 93 |
-
|
| 94 |
-
```json
|
| 95 |
-
"train": {
|
| 96 |
-
"batch_size": 32,
|
| 97 |
-
...
|
| 98 |
-
"adamw": {
|
| 99 |
-
"lr": 2.0e-4
|
| 100 |
-
},
|
| 101 |
-
...
|
| 102 |
-
}
|
| 103 |
-
```
|
| 104 |
-
|
| 105 |
-
### Run
|
| 106 |
-
|
| 107 |
-
Run the `run.sh` as the training stage (set `--stage 2`). Specify a experimental name to run the following command. The tensorboard logs and checkpoints will be saved in `Amphion/ckpts/svc/[YourExptName]`.
|
| 108 |
-
|
| 109 |
-
```bash
|
| 110 |
-
sh egs/svc/MultipleContentsSVC/run.sh --stage 2 --name [YourExptName]
|
| 111 |
-
```
|
| 112 |
-
|
| 113 |
-
> **NOTE:** The `CUDA_VISIBLE_DEVICES` is set as `"0"` in default. You can change it when running `run.sh` by specifying such as `--gpu "0,1,2,3"`.
|
| 114 |
-
|
| 115 |
-
## 4. Inference/Conversion
|
| 116 |
-
|
| 117 |
-
### Pretrained Vocoder Download
|
| 118 |
-
|
| 119 |
-
We fine-tune the official BigVGAN pretrained model with over 120 hours singing voice data. The benifits of fine-tuning has been investigated in our paper (see this [demo page](https://www.zhangxueyao.com/data/MultipleContentsSVC/vocoder.html)). The final pretrained singing voice vocoder is released [here](../../../pretrained/README.md#amphion-singing-bigvgan) (called `Amphion Singing BigVGAN`).
|
| 120 |
-
|
| 121 |
-
### Run
|
| 122 |
-
|
| 123 |
-
For inference/conversion, you need to specify the following configurations when running `run.sh`:
|
| 124 |
-
|
| 125 |
-
| Parameters | Description | Example |
|
| 126 |
-
| --------------------------------------------------- | ------------------------------------------------------------------------------------------------------------------------------------------ | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
|
| 127 |
-
| `--infer_expt_dir` | The experimental directory which contains `checkpoint` | `Amphion/ckpts/svc/[YourExptName]` |
|
| 128 |
-
| `--infer_output_dir` | The output directory to save inferred audios. | `Amphion/ckpts/svc/[YourExptName]/result` |
|
| 129 |
-
| `--infer_source_file` or `--infer_source_audio_dir` | The inference source (can be a json file or a dir). | The `infer_source_file` could be `Amphion/data/[YourDataset]/test.json`, and the `infer_source_audio_dir` is a folder which includes several audio files (*.wav, *.mp3 or *.flac). |
|
| 130 |
-
| `--infer_target_speaker` | The target speaker you want to convert into. You can refer to `Amphion/ckpts/svc/[YourExptName]/singers.json` to choose a trained speaker. | For opencpop dataset, the speaker name would be `opencpop_female1`. |
|
| 131 |
-
| `--infer_key_shift` | How many semitones you want to transpose. | `"autoshfit"` (by default), `3`, `-3`, etc. |
|
| 132 |
-
|
| 133 |
-
For example, if you want to make `opencpop_female1` sing the songs in the `[Your Audios Folder]`, just run:
|
| 134 |
-
|
| 135 |
-
```bash
|
| 136 |
-
sh egs/svc/MultipleContentsSVC/run.sh --stage 3 --gpu "0" \
|
| 137 |
-
--infer_expt_dir Amphion/ckpts/svc/[YourExptName] \
|
| 138 |
-
--infer_output_dir Amphion/ckpts/svc/[YourExptName]/result \
|
| 139 |
-
--infer_source_audio_dir [Your Audios Folder] \
|
| 140 |
-
--infer_target_speaker "opencpop_female1" \
|
| 141 |
-
--infer_key_shift "autoshift"
|
| 142 |
-
```
|
| 143 |
-
|
| 144 |
-
## Citations
|
| 145 |
-
|
| 146 |
-
```bibtex
|
| 147 |
-
@article{zhang2023leveraging,
|
| 148 |
-
title={Leveraging Content-based Features from Multiple Acoustic Models for Singing Voice Conversion},
|
| 149 |
-
author={Zhang, Xueyao and Gu, Yicheng and Chen, Haopeng and Fang, Zihao and Zou, Lexiao and Xue, Liumeng and Wu, Zhizheng},
|
| 150 |
-
journal={Machine Learning for Audio Worshop, NeurIPS 2023},
|
| 151 |
-
year={2023}
|
| 152 |
-
}
|
| 153 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/svc/MultipleContentsSVC/exp_config.json
DELETED
|
@@ -1,126 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/diffusion.json",
|
| 3 |
-
"model_type": "DiffWaveNetSVC",
|
| 4 |
-
"dataset": [
|
| 5 |
-
"m4singer",
|
| 6 |
-
"opencpop",
|
| 7 |
-
"opensinger",
|
| 8 |
-
"svcc",
|
| 9 |
-
"vctk"
|
| 10 |
-
],
|
| 11 |
-
"dataset_path": {
|
| 12 |
-
// TODO: Fill in your dataset path
|
| 13 |
-
"m4singer": "[M4Singer dataset path]",
|
| 14 |
-
"opencpop": "[Opencpop dataset path]",
|
| 15 |
-
"opensinger": "[OpenSinger dataset path]",
|
| 16 |
-
"svcc": "[SVCC dataset path]",
|
| 17 |
-
"vctk": "[VCTK dataset path]"
|
| 18 |
-
},
|
| 19 |
-
// TODO: Fill in the output log path. The default value is "Amphion/ckpts/svc"
|
| 20 |
-
"log_dir": "ckpts/svc",
|
| 21 |
-
"preprocess": {
|
| 22 |
-
// TODO: Fill in the output data path. The default value is "Amphion/data"
|
| 23 |
-
"processed_dir": "data",
|
| 24 |
-
// Config for features extraction
|
| 25 |
-
"extract_mel": true,
|
| 26 |
-
"extract_pitch": true,
|
| 27 |
-
"extract_energy": true,
|
| 28 |
-
"extract_whisper_feature": true,
|
| 29 |
-
"extract_contentvec_feature": true,
|
| 30 |
-
"extract_wenet_feature": false,
|
| 31 |
-
"whisper_batch_size": 30, // decrease it if your GPU is out of memory
|
| 32 |
-
"contentvec_batch_size": 1,
|
| 33 |
-
// Fill in the content-based pretrained model's path
|
| 34 |
-
"contentvec_file": "pretrained/contentvec/checkpoint_best_legacy_500.pt",
|
| 35 |
-
"wenet_model_path": "pretrained/wenet/20220506_u2pp_conformer_exp/final.pt",
|
| 36 |
-
"wenet_config": "pretrained/wenet/20220506_u2pp_conformer_exp/train.yaml",
|
| 37 |
-
"whisper_model": "medium",
|
| 38 |
-
"whisper_model_path": "pretrained/whisper/medium.pt",
|
| 39 |
-
// Config for features usage
|
| 40 |
-
"use_mel": true,
|
| 41 |
-
"use_min_max_norm_mel": true,
|
| 42 |
-
"use_frame_pitch": true,
|
| 43 |
-
"use_frame_energy": true,
|
| 44 |
-
"use_spkid": true,
|
| 45 |
-
"use_whisper": true,
|
| 46 |
-
"use_contentvec": true,
|
| 47 |
-
"use_wenet": false,
|
| 48 |
-
"n_mel": 100,
|
| 49 |
-
"sample_rate": 24000
|
| 50 |
-
},
|
| 51 |
-
"model": {
|
| 52 |
-
"condition_encoder": {
|
| 53 |
-
// Config for features usage
|
| 54 |
-
"use_whisper": true,
|
| 55 |
-
"use_contentvec": true,
|
| 56 |
-
"use_wenet": false,
|
| 57 |
-
"whisper_dim": 1024,
|
| 58 |
-
"contentvec_dim": 256,
|
| 59 |
-
"wenet_dim": 512,
|
| 60 |
-
"use_singer_encoder": false,
|
| 61 |
-
"pitch_min": 50,
|
| 62 |
-
"pitch_max": 1100
|
| 63 |
-
},
|
| 64 |
-
"diffusion": {
|
| 65 |
-
"scheduler": "ddpm",
|
| 66 |
-
"scheduler_settings": {
|
| 67 |
-
"num_train_timesteps": 1000,
|
| 68 |
-
"beta_start": 1.0e-4,
|
| 69 |
-
"beta_end": 0.02,
|
| 70 |
-
"beta_schedule": "linear"
|
| 71 |
-
},
|
| 72 |
-
// Diffusion steps encoder
|
| 73 |
-
"step_encoder": {
|
| 74 |
-
"dim_raw_embedding": 128,
|
| 75 |
-
"dim_hidden_layer": 512,
|
| 76 |
-
"activation": "SiLU",
|
| 77 |
-
"num_layer": 2,
|
| 78 |
-
"max_period": 10000
|
| 79 |
-
},
|
| 80 |
-
// Diffusion decoder
|
| 81 |
-
"model_type": "bidilconv",
|
| 82 |
-
// bidilconv, unet2d, TODO: unet1d
|
| 83 |
-
"bidilconv": {
|
| 84 |
-
"base_channel": 512,
|
| 85 |
-
"n_res_block": 40,
|
| 86 |
-
"conv_kernel_size": 3,
|
| 87 |
-
"dilation_cycle_length": 4,
|
| 88 |
-
// specially, 1 means no dilation
|
| 89 |
-
"conditioner_size": 384
|
| 90 |
-
}
|
| 91 |
-
}
|
| 92 |
-
},
|
| 93 |
-
"train": {
|
| 94 |
-
"batch_size": 32,
|
| 95 |
-
"gradient_accumulation_step": 1,
|
| 96 |
-
"max_epoch": -1, // -1 means no limit
|
| 97 |
-
"save_checkpoint_stride": [
|
| 98 |
-
3,
|
| 99 |
-
50
|
| 100 |
-
],
|
| 101 |
-
"keep_last": [
|
| 102 |
-
3,
|
| 103 |
-
2
|
| 104 |
-
],
|
| 105 |
-
"run_eval": [
|
| 106 |
-
true,
|
| 107 |
-
true
|
| 108 |
-
],
|
| 109 |
-
"adamw": {
|
| 110 |
-
"lr": 2.0e-4
|
| 111 |
-
},
|
| 112 |
-
"reducelronplateau": {
|
| 113 |
-
"factor": 0.8,
|
| 114 |
-
"patience": 30,
|
| 115 |
-
"min_lr": 1.0e-4
|
| 116 |
-
},
|
| 117 |
-
"dataloader": {
|
| 118 |
-
"num_worker": 8,
|
| 119 |
-
"pin_memory": true
|
| 120 |
-
},
|
| 121 |
-
"sampler": {
|
| 122 |
-
"holistic_shuffle": false,
|
| 123 |
-
"drop_last": true
|
| 124 |
-
}
|
| 125 |
-
}
|
| 126 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/svc/MultipleContentsSVC/run.sh
DELETED
|
@@ -1 +0,0 @@
|
|
| 1 |
-
../_template/run.sh
|
|
|
|
|
|
egs/svc/README.md
DELETED
|
@@ -1,34 +0,0 @@
|
|
| 1 |
-
# Amphion Singing Voice Conversion (SVC) Recipe
|
| 2 |
-
|
| 3 |
-
## Quick Start
|
| 4 |
-
|
| 5 |
-
We provide a **[beginner recipe](MultipleContentsSVC)** to demonstrate how to train a cutting edge SVC model. Specifically, it is also an official implementation of the paper "[Leveraging Content-based Features from Multiple Acoustic Models for Singing Voice Conversion](https://arxiv.org/abs/2310.11160)" (NeurIPS 2023 Workshop on Machine Learning for Audio). Some demos can be seen [here](https://www.zhangxueyao.com/data/MultipleContentsSVC/index.html).
|
| 6 |
-
|
| 7 |
-
## Supported Model Architectures
|
| 8 |
-
|
| 9 |
-
The main idea of SVC is to first disentangle the speaker-agnostic representations from the source audio, and then inject the desired speaker information to synthesize the target, which usually utilizes an acoustic decoder and a subsequent waveform synthesizer (vocoder):
|
| 10 |
-
|
| 11 |
-
<br>
|
| 12 |
-
<div align="center">
|
| 13 |
-
<img src="../../imgs/svc/pipeline.png" width="70%">
|
| 14 |
-
</div>
|
| 15 |
-
<br>
|
| 16 |
-
|
| 17 |
-
Until now, Amphion SVC has supported the following features and models:
|
| 18 |
-
|
| 19 |
-
- **Speaker-agnostic Representations**:
|
| 20 |
-
- Content Features: Sourcing from [WeNet](https://github.com/wenet-e2e/wenet), [Whisper](https://github.com/openai/whisper), and [ContentVec](https://github.com/auspicious3000/contentvec).
|
| 21 |
-
- Prosody Features: F0 and energy.
|
| 22 |
-
- **Speaker Embeddings**:
|
| 23 |
-
- Speaker Look-Up Table.
|
| 24 |
-
- Reference Encoder (👨💻 developing): It can be used for zero-shot SVC.
|
| 25 |
-
- **Acoustic Decoders**:
|
| 26 |
-
- Diffusion-based models:
|
| 27 |
-
- **[DiffWaveNetSVC](MultipleContentsSVC)**: The encoder is based on Bidirectional Non-Causal Dilated CNN, which is similar to [WaveNet](https://arxiv.org/pdf/1609.03499.pdf), [DiffWave](https://openreview.net/forum?id=a-xFK8Ymz5J), and [DiffSVC](https://ieeexplore.ieee.org/document/9688219).
|
| 28 |
-
- **[DiffComoSVC](DiffComoSVC)** (👨💻 developing): The diffusion framework is based on [Consistency Model](https://proceedings.mlr.press/v202/song23a.html). It can significantly accelerate the inference process of the diffusion model.
|
| 29 |
-
- Transformer-based models:
|
| 30 |
-
- **[TransformerSVC](TransformerSVC)**: Encoder-only and Non-autoregressive Transformer Architecture.
|
| 31 |
-
- VAE- and Flow-based models:
|
| 32 |
-
- **[VitsSVC]()** (👨💻 developing): It is designed as a [VITS](https://arxiv.org/abs/2106.06103)-like model whose textual input is replaced by the content features, which is similar to [so-vits-svc](https://github.com/svc-develop-team/so-vits-svc).
|
| 33 |
-
- **Waveform Synthesizers (Vocoders)**:
|
| 34 |
-
- The supported vocoders can be seen in [Amphion Vocoder Recipe](../vocoder/README.md).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/svc/_template/run.sh
DELETED
|
@@ -1,150 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $exp_dir)))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,resume_from_ckpt_path:,resume_type:,infer_expt_dir:,infer_output_dir:,infer_source_file:,infer_source_audio_dir:,infer_target_speaker:,infer_key_shift:,infer_vocoder_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--resume_from_ckpt_path) shift; resume_from_ckpt_path=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 37 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 39 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The inference source (can be a json file or a dir). For example, the source_file can be "[Your path to save processed data]/[YourDataset]/test.json", and the source_audio_dir can be "$work_dir/source_audio" which includes several audio files (*.wav, *.mp3 or *.flac).
|
| 41 |
-
--infer_source_file) shift; infer_source_file=$1 ; shift ;;
|
| 42 |
-
--infer_source_audio_dir) shift; infer_source_audio_dir=$1 ; shift ;;
|
| 43 |
-
# [Only for Inference] Specify the target speaker you want to convert into. You can refer to "[Your path to save logs and checkpoints]/[Your Expt Name]/singers.json". In this singer look-up table, you can see the usable speaker names (all the keys of the dictionary). For example, for opencpop dataset, the speaker name would be "opencpop_female1".
|
| 44 |
-
--infer_target_speaker) shift; infer_target_speaker=$1 ; shift ;;
|
| 45 |
-
# [Only for Inference] For advanced users, you can modify the trans_key parameters into an integer (which means the semitones you want to transpose). Its default value is "autoshift".
|
| 46 |
-
--infer_key_shift) shift; infer_key_shift=$1 ; shift ;;
|
| 47 |
-
# [Only for Inference] The vocoder dir. Its default value is Amphion/pretrained/bigvgan. See Amphion/pretrained/README.md to download the pretrained BigVGAN vocoders.
|
| 48 |
-
--infer_vocoder_dir) shift; infer_vocoder_dir=$1 ; shift ;;
|
| 49 |
-
|
| 50 |
-
--) shift ; break ;;
|
| 51 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 52 |
-
esac
|
| 53 |
-
done
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
### Value check ###
|
| 57 |
-
if [ -z "$running_stage" ]; then
|
| 58 |
-
echo "[Error] Please specify the running stage"
|
| 59 |
-
exit 1
|
| 60 |
-
fi
|
| 61 |
-
|
| 62 |
-
if [ -z "$exp_config" ]; then
|
| 63 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 64 |
-
fi
|
| 65 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 66 |
-
|
| 67 |
-
if [ -z "$gpu" ]; then
|
| 68 |
-
gpu="0"
|
| 69 |
-
fi
|
| 70 |
-
|
| 71 |
-
######## Features Extraction ###########
|
| 72 |
-
if [ $running_stage -eq 1 ]; then
|
| 73 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/svc/preprocess.py \
|
| 74 |
-
--config $exp_config \
|
| 75 |
-
--num_workers 4
|
| 76 |
-
fi
|
| 77 |
-
|
| 78 |
-
######## Training ###########
|
| 79 |
-
if [ $running_stage -eq 2 ]; then
|
| 80 |
-
if [ -z "$exp_name" ]; then
|
| 81 |
-
echo "[Error] Please specify the experiments name"
|
| 82 |
-
exit 1
|
| 83 |
-
fi
|
| 84 |
-
echo "Exprimental Name: $exp_name"
|
| 85 |
-
|
| 86 |
-
if [ "$resume" = true ]; then
|
| 87 |
-
echo "Automatically resume from the experimental dir..."
|
| 88 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/svc/train.py \
|
| 89 |
-
--config "$exp_config" \
|
| 90 |
-
--exp_name "$exp_name" \
|
| 91 |
-
--log_level info \
|
| 92 |
-
--resume
|
| 93 |
-
else
|
| 94 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/svc/train.py \
|
| 95 |
-
--config "$exp_config" \
|
| 96 |
-
--exp_name "$exp_name" \
|
| 97 |
-
--log_level info \
|
| 98 |
-
--resume_from_ckpt_path "$resume_from_ckpt_path" \
|
| 99 |
-
--resume_type "$resume_type"
|
| 100 |
-
fi
|
| 101 |
-
fi
|
| 102 |
-
|
| 103 |
-
######## Inference/Conversion ###########
|
| 104 |
-
if [ $running_stage -eq 3 ]; then
|
| 105 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 106 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 107 |
-
exit 1
|
| 108 |
-
fi
|
| 109 |
-
|
| 110 |
-
if [ -z "$infer_output_dir" ]; then
|
| 111 |
-
infer_output_dir="$expt_dir/result"
|
| 112 |
-
fi
|
| 113 |
-
|
| 114 |
-
if [ -z "$infer_source_file" ] && [ -z "$infer_source_audio_dir" ]; then
|
| 115 |
-
echo "[Error] Please specify the source file/dir. The inference source (can be a json file or a dir). For example, the source_file can be "[Your path to save processed data]/[YourDataset]/test.json", and the source_audio_dir should include several audio files (*.wav, *.mp3 or *.flac)."
|
| 116 |
-
exit 1
|
| 117 |
-
fi
|
| 118 |
-
|
| 119 |
-
if [ -z "$infer_source_file" ]; then
|
| 120 |
-
infer_source=$infer_source_audio_dir
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ -z "$infer_source_audio_dir" ]; then
|
| 124 |
-
infer_source=$infer_source_file
|
| 125 |
-
fi
|
| 126 |
-
|
| 127 |
-
if [ -z "$infer_target_speaker" ]; then
|
| 128 |
-
echo "[Error] Please specify the target speaker. You can refer to "[Your path to save logs and checkpoints]/[Your Expt Name]/singers.json". In this singer look-up table, you can see the usable speaker names (all the keys of the dictionary). For example, for opencpop dataset, the speaker name would be "opencpop_female1""
|
| 129 |
-
exit 1
|
| 130 |
-
fi
|
| 131 |
-
|
| 132 |
-
if [ -z "$infer_key_shift" ]; then
|
| 133 |
-
infer_key_shift="autoshift"
|
| 134 |
-
fi
|
| 135 |
-
|
| 136 |
-
if [ -z "$infer_vocoder_dir" ]; then
|
| 137 |
-
infer_vocoder_dir="$work_dir"/pretrained/bigvgan
|
| 138 |
-
echo "[Warning] You don't specify the infer_vocoder_dir. It is set $infer_vocoder_dir by default. Make sure that you have followed Amphoion/pretrained/README.md to download the pretrained BigVGAN vocoder checkpoint."
|
| 139 |
-
fi
|
| 140 |
-
|
| 141 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/svc/inference.py \
|
| 142 |
-
--config $exp_config \
|
| 143 |
-
--acoustics_dir $infer_expt_dir \
|
| 144 |
-
--vocoder_dir $infer_vocoder_dir \
|
| 145 |
-
--target_singer $infer_target_speaker \
|
| 146 |
-
--trans_key $infer_key_shift \
|
| 147 |
-
--source $infer_source \
|
| 148 |
-
--output_dir $infer_output_dir \
|
| 149 |
-
--log_level debug
|
| 150 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/README.md
DELETED
|
@@ -1,23 +0,0 @@
|
|
| 1 |
-
# Amphion Vocoder Recipe
|
| 2 |
-
|
| 3 |
-
## Quick Start
|
| 4 |
-
|
| 5 |
-
We provide a [**beginner recipe**](gan/tfr_enhanced_hifigan/README.md) to demonstrate how to train a high quality HiFi-GAN speech vocoder. Specially, it is also an official implementation of our paper "[Multi-Scale Sub-Band Constant-Q Transform Discriminator for High-Fidelity Vocoder](https://arxiv.org/abs/2311.14957)". Some demos can be seen [here](https://vocodexelysium.github.io/MS-SB-CQTD/).
|
| 6 |
-
|
| 7 |
-
## Supported Models
|
| 8 |
-
|
| 9 |
-
Neural vocoder generates audible waveforms from acoustic representations, which is one of the key parts for current audio generation systems. Until now, Amphion has supported various widely-used vocoders according to different vocoder types, including:
|
| 10 |
-
|
| 11 |
-
- **GAN-based vocoders**, which we have provided [**a unified recipe**](gan/README.md) :
|
| 12 |
-
- [MelGAN](https://arxiv.org/abs/1910.06711)
|
| 13 |
-
- [HiFi-GAN](https://arxiv.org/abs/2010.05646)
|
| 14 |
-
- [NSF-HiFiGAN](https://github.com/nii-yamagishilab/project-NN-Pytorch-scripts)
|
| 15 |
-
- [BigVGAN](https://arxiv.org/abs/2206.04658)
|
| 16 |
-
- [APNet](https://arxiv.org/abs/2305.07952)
|
| 17 |
-
- **Flow-based vocoders** (👨💻 developing):
|
| 18 |
-
- [WaveGlow](https://arxiv.org/abs/1811.00002)
|
| 19 |
-
- **Diffusion-based vocoders** (👨💻 developing):
|
| 20 |
-
- [Diffwave](https://arxiv.org/abs/2009.09761)
|
| 21 |
-
- **Auto-regressive based vocoders** (👨💻 developing):
|
| 22 |
-
- [WaveNet](https://arxiv.org/abs/1609.03499)
|
| 23 |
-
- [WaveRNN](https://arxiv.org/abs/1802.08435v1)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/diffusion/README.md
DELETED
|
File without changes
|
egs/vocoder/diffusion/exp_config_base.json
DELETED
|
File without changes
|
egs/vocoder/gan/README.md
DELETED
|
@@ -1,224 +0,0 @@
|
|
| 1 |
-
# Amphion GAN-based Vocoder Recipe
|
| 2 |
-
|
| 3 |
-
## Supported Model Architectures
|
| 4 |
-
|
| 5 |
-
GAN-based Vocoder consists of a generator and multiple discriminators, as illustrated below:
|
| 6 |
-
|
| 7 |
-
<br>
|
| 8 |
-
<div align="center">
|
| 9 |
-
<img src="../../../imgs/vocoder/gan/pipeline.png" width="40%">
|
| 10 |
-
</div>
|
| 11 |
-
<br>
|
| 12 |
-
|
| 13 |
-
Until now, Amphion GAN-based Vocoder has supported the following generators and discriminators.
|
| 14 |
-
|
| 15 |
-
- **Generators**
|
| 16 |
-
- [MelGAN](https://arxiv.org/abs/1910.06711)
|
| 17 |
-
- [HiFi-GAN](https://arxiv.org/abs/2010.05646)
|
| 18 |
-
- [NSF-HiFiGAN](https://github.com/nii-yamagishilab/project-NN-Pytorch-scripts)
|
| 19 |
-
- [BigVGAN](https://arxiv.org/abs/2206.04658)
|
| 20 |
-
- [APNet](https://arxiv.org/abs/2305.07952)
|
| 21 |
-
- **Discriminators**
|
| 22 |
-
- [Multi-Scale Discriminator](https://arxiv.org/abs/2010.05646)
|
| 23 |
-
- [Multi-Period Discriminator](https://arxiv.org/abs/2010.05646)
|
| 24 |
-
- [Multi-Resolution Discriminator](https://arxiv.org/abs/2011.09631)
|
| 25 |
-
- [Multi-Scale Short-Time Fourier Transform Discriminator](https://arxiv.org/abs/2210.13438)
|
| 26 |
-
- [**Multi-Scale Constant-Q Transfrom Discriminator (ours)**](https://arxiv.org/abs/2311.14957)
|
| 27 |
-
|
| 28 |
-
You can use any vocoder architecture with any dataset you want. There are four steps in total:
|
| 29 |
-
|
| 30 |
-
1. Data preparation
|
| 31 |
-
2. Feature extraction
|
| 32 |
-
3. Training
|
| 33 |
-
4. Inference
|
| 34 |
-
|
| 35 |
-
> **NOTE:** You need to run every command of this recipe in the `Amphion` root path:
|
| 36 |
-
> ```bash
|
| 37 |
-
> cd Amphion
|
| 38 |
-
> ```
|
| 39 |
-
|
| 40 |
-
## 1. Data Preparation
|
| 41 |
-
|
| 42 |
-
You can train the vocoder with any datasets. Amphion's supported open-source datasets are detailed [here](../../../datasets/README.md).
|
| 43 |
-
|
| 44 |
-
### Configuration
|
| 45 |
-
|
| 46 |
-
Specify the dataset path in `exp_config_base.json`. Note that you can change the `dataset` list to use your preferred datasets.
|
| 47 |
-
|
| 48 |
-
```json
|
| 49 |
-
"dataset": [
|
| 50 |
-
"csd",
|
| 51 |
-
"kising",
|
| 52 |
-
"m4singer",
|
| 53 |
-
"nus48e",
|
| 54 |
-
"opencpop",
|
| 55 |
-
"opensinger",
|
| 56 |
-
"opera",
|
| 57 |
-
"pjs",
|
| 58 |
-
"popbutfy",
|
| 59 |
-
"popcs",
|
| 60 |
-
"ljspeech",
|
| 61 |
-
"vctk",
|
| 62 |
-
"libritts",
|
| 63 |
-
],
|
| 64 |
-
"dataset_path": {
|
| 65 |
-
// TODO: Fill in your dataset path
|
| 66 |
-
"csd": "[dataset path]",
|
| 67 |
-
"kising": "[dataset path]",
|
| 68 |
-
"m4singer": "[dataset path]",
|
| 69 |
-
"nus48e": "[dataset path]",
|
| 70 |
-
"opencpop": "[dataset path]",
|
| 71 |
-
"opensinger": "[dataset path]",
|
| 72 |
-
"opera": "[dataset path]",
|
| 73 |
-
"pjs": "[dataset path]",
|
| 74 |
-
"popbutfy": "[dataset path]",
|
| 75 |
-
"popcs": "[dataset path]",
|
| 76 |
-
"ljspeech": "[dataset path]",
|
| 77 |
-
"vctk": "[dataset path]",
|
| 78 |
-
"libritts": "[dataset path]",
|
| 79 |
-
},
|
| 80 |
-
```
|
| 81 |
-
|
| 82 |
-
### 2. Feature Extraction
|
| 83 |
-
|
| 84 |
-
The needed features are speficied in the individual vocoder direction so it doesn't require any modification.
|
| 85 |
-
|
| 86 |
-
### Configuration
|
| 87 |
-
|
| 88 |
-
Specify the dataset path and the output path for saving the processed data and the training model in `exp_config_base.json`:
|
| 89 |
-
|
| 90 |
-
```json
|
| 91 |
-
// TODO: Fill in the output log path. The default value is "Amphion/ckpts/vocoder"
|
| 92 |
-
"log_dir": "ckpts/vocoder",
|
| 93 |
-
"preprocess": {
|
| 94 |
-
// TODO: Fill in the output data path. The default value is "Amphion/data"
|
| 95 |
-
"processed_dir": "data",
|
| 96 |
-
...
|
| 97 |
-
},
|
| 98 |
-
```
|
| 99 |
-
|
| 100 |
-
### Run
|
| 101 |
-
|
| 102 |
-
Run the `run.sh` as the preproces stage (set `--stage 1`).
|
| 103 |
-
|
| 104 |
-
```bash
|
| 105 |
-
sh egs/vocoder/gan/{vocoder_name}/run.sh --stage 1
|
| 106 |
-
```
|
| 107 |
-
|
| 108 |
-
> **NOTE:** The `CUDA_VISIBLE_DEVICES` is set as `"0"` in default. You can change it when running `run.sh` by specifying such as `--gpu "1"`.
|
| 109 |
-
|
| 110 |
-
## 3. Training
|
| 111 |
-
|
| 112 |
-
### Configuration
|
| 113 |
-
|
| 114 |
-
We provide the default hyparameters in the `exp_config_base.json`. They can work on single NVIDIA-24g GPU. You can adjust them based on you GPU machines.
|
| 115 |
-
|
| 116 |
-
```json
|
| 117 |
-
"train": {
|
| 118 |
-
"batch_size": 16,
|
| 119 |
-
"max_epoch": 1000000,
|
| 120 |
-
"save_checkpoint_stride": [20],
|
| 121 |
-
"adamw": {
|
| 122 |
-
"lr": 2.0e-4,
|
| 123 |
-
"adam_b1": 0.8,
|
| 124 |
-
"adam_b2": 0.99
|
| 125 |
-
},
|
| 126 |
-
"exponential_lr": {
|
| 127 |
-
"lr_decay": 0.999
|
| 128 |
-
},
|
| 129 |
-
}
|
| 130 |
-
```
|
| 131 |
-
|
| 132 |
-
You can also choose any amount of prefered discriminators for training in the `exp_config_base.json`.
|
| 133 |
-
|
| 134 |
-
```json
|
| 135 |
-
"discriminators": [
|
| 136 |
-
"msd",
|
| 137 |
-
"mpd",
|
| 138 |
-
"msstftd",
|
| 139 |
-
"mssbcqtd",
|
| 140 |
-
],
|
| 141 |
-
```
|
| 142 |
-
|
| 143 |
-
### Run
|
| 144 |
-
|
| 145 |
-
Run the `run.sh` as the training stage (set `--stage 2`). Specify a experimental name to run the following command. The tensorboard logs and checkpoints will be saved in `Amphion/ckpts/vocoder/[YourExptName]`.
|
| 146 |
-
|
| 147 |
-
```bash
|
| 148 |
-
sh egs/vocoder/gan/{vocoder_name}/run.sh --stage 2 --name [YourExptName]
|
| 149 |
-
```
|
| 150 |
-
|
| 151 |
-
> **NOTE:** The `CUDA_VISIBLE_DEVICES` is set as `"0"` in default. You can change it when running `run.sh` by specifying such as `--gpu "0,1,2,3"`.
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
## 4. Inference
|
| 155 |
-
|
| 156 |
-
### Run
|
| 157 |
-
|
| 158 |
-
Run the `run.sh` as the training stage (set `--stage 3`), we provide three different inference modes, including `infer_from_dataset`, `infer_from_feature`, `and infer_from_audio`.
|
| 159 |
-
|
| 160 |
-
```bash
|
| 161 |
-
sh egs/vocoder/gan/{vocoder_name}/run.sh --stage 3 \
|
| 162 |
-
--infer_mode [Your chosen inference mode] \
|
| 163 |
-
--infer_datasets [Datasets you want to inference, needed when infer_from_dataset] \
|
| 164 |
-
--infer_feature_dir [Your path to your predicted acoustic features, needed when infer_from_feature] \
|
| 165 |
-
--infer_audio_dir [Your path to your audio files, needed when infer_form_audio] \
|
| 166 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 167 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 168 |
-
```
|
| 169 |
-
|
| 170 |
-
#### a. Inference from Dataset
|
| 171 |
-
|
| 172 |
-
Run the `run.sh` with specified datasets, here is an example.
|
| 173 |
-
|
| 174 |
-
```bash
|
| 175 |
-
sh egs/vocoder/gan/{vocoder_name}/run.sh --stage 3 \
|
| 176 |
-
--infer_mode infer_from_dataset \
|
| 177 |
-
--infer_datasets "libritts vctk ljspeech" \
|
| 178 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 179 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 180 |
-
```
|
| 181 |
-
|
| 182 |
-
#### b. Inference from Features
|
| 183 |
-
|
| 184 |
-
If you want to inference from your generated acoustic features, you should first prepare your acoustic features into the following structure:
|
| 185 |
-
|
| 186 |
-
```plaintext
|
| 187 |
-
┣ {infer_feature_dir}
|
| 188 |
-
┃ ┣ mels
|
| 189 |
-
┃ ┃ ┣ sample1.npy
|
| 190 |
-
┃ ┃ ┣ sample2.npy
|
| 191 |
-
┃ ┣ f0s (required if you use NSF-HiFiGAN)
|
| 192 |
-
┃ ┃ ┣ sample1.npy
|
| 193 |
-
┃ ┃ ┣ sample2.npy
|
| 194 |
-
```
|
| 195 |
-
|
| 196 |
-
Then run the `run.sh` with specificed folder direction, here is an example.
|
| 197 |
-
|
| 198 |
-
```bash
|
| 199 |
-
sh egs/vocoder/gan/{vocoder_name}/run.sh --stage 3 \
|
| 200 |
-
--infer_mode infer_from_feature \
|
| 201 |
-
--infer_feature_dir [Your path to your predicted acoustic features] \
|
| 202 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 203 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 204 |
-
```
|
| 205 |
-
|
| 206 |
-
#### c. Inference from Audios
|
| 207 |
-
|
| 208 |
-
If you want to inference from audios for quick analysis synthesis, you should first prepare your audios into the following structure:
|
| 209 |
-
|
| 210 |
-
```plaintext
|
| 211 |
-
┣ audios
|
| 212 |
-
┃ ┣ sample1.wav
|
| 213 |
-
┃ ┣ sample2.wav
|
| 214 |
-
```
|
| 215 |
-
|
| 216 |
-
Then run the `run.sh` with specificed folder direction, here is an example.
|
| 217 |
-
|
| 218 |
-
```bash
|
| 219 |
-
sh egs/vocoder/gan/{vocoder_name}/run.sh --stage 3 \
|
| 220 |
-
--infer_mode infer_from_audio \
|
| 221 |
-
--infer_audio_dir [Your path to your audio files] \
|
| 222 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 223 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 224 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/_template/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/apnet/exp_config.json
DELETED
|
@@ -1,45 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"preprocess": {
|
| 4 |
-
// acoustic features
|
| 5 |
-
"extract_mel": true,
|
| 6 |
-
"extract_audio": true,
|
| 7 |
-
"extract_amplitude_phase": true,
|
| 8 |
-
|
| 9 |
-
// Features used for model training
|
| 10 |
-
"use_mel": true,
|
| 11 |
-
"use_audio": true,
|
| 12 |
-
"use_amplitude_phase": true
|
| 13 |
-
},
|
| 14 |
-
"model": {
|
| 15 |
-
"generator": "apnet",
|
| 16 |
-
"apnet": {
|
| 17 |
-
"ASP_channel": 512,
|
| 18 |
-
"ASP_resblock_kernel_sizes": [3,7,11],
|
| 19 |
-
"ASP_resblock_dilation_sizes": [[1,3,5], [1,3,5], [1,3,5]],
|
| 20 |
-
"ASP_input_conv_kernel_size": 7,
|
| 21 |
-
"ASP_output_conv_kernel_size": 7,
|
| 22 |
-
|
| 23 |
-
"PSP_channel": 512,
|
| 24 |
-
"PSP_resblock_kernel_sizes": [3,7,11],
|
| 25 |
-
"PSP_resblock_dilation_sizes": [[1,3,5], [1,3,5], [1,3,5]],
|
| 26 |
-
"PSP_input_conv_kernel_size": 7,
|
| 27 |
-
"PSP_output_R_conv_kernel_size": 7,
|
| 28 |
-
"PSP_output_I_conv_kernel_size": 7,
|
| 29 |
-
}
|
| 30 |
-
},
|
| 31 |
-
"train": {
|
| 32 |
-
"criterions": [
|
| 33 |
-
"feature",
|
| 34 |
-
"discriminator",
|
| 35 |
-
"generator",
|
| 36 |
-
"mel",
|
| 37 |
-
"phase",
|
| 38 |
-
"amplitude",
|
| 39 |
-
"consistency"
|
| 40 |
-
]
|
| 41 |
-
},
|
| 42 |
-
"inference": {
|
| 43 |
-
"batch_size": 1,
|
| 44 |
-
}
|
| 45 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/apnet/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/bigvgan/exp_config.json
DELETED
|
@@ -1,66 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"preprocess": {
|
| 4 |
-
// acoustic features
|
| 5 |
-
"extract_mel": true,
|
| 6 |
-
"extract_audio": true,
|
| 7 |
-
|
| 8 |
-
// Features used for model training
|
| 9 |
-
"use_mel": true,
|
| 10 |
-
"use_audio": true
|
| 11 |
-
},
|
| 12 |
-
"model": {
|
| 13 |
-
"generator": "bigvgan",
|
| 14 |
-
"bigvgan": {
|
| 15 |
-
"resblock": "1",
|
| 16 |
-
"activation": "snakebeta",
|
| 17 |
-
"snake_logscale": true,
|
| 18 |
-
"upsample_rates": [
|
| 19 |
-
8,
|
| 20 |
-
8,
|
| 21 |
-
2,
|
| 22 |
-
2,
|
| 23 |
-
],
|
| 24 |
-
"upsample_kernel_sizes": [
|
| 25 |
-
16,
|
| 26 |
-
16,
|
| 27 |
-
4,
|
| 28 |
-
4
|
| 29 |
-
],
|
| 30 |
-
"upsample_initial_channel": 512,
|
| 31 |
-
"resblock_kernel_sizes": [
|
| 32 |
-
3,
|
| 33 |
-
7,
|
| 34 |
-
11
|
| 35 |
-
],
|
| 36 |
-
"resblock_dilation_sizes": [
|
| 37 |
-
[
|
| 38 |
-
1,
|
| 39 |
-
3,
|
| 40 |
-
5
|
| 41 |
-
],
|
| 42 |
-
[
|
| 43 |
-
1,
|
| 44 |
-
3,
|
| 45 |
-
5
|
| 46 |
-
],
|
| 47 |
-
[
|
| 48 |
-
1,
|
| 49 |
-
3,
|
| 50 |
-
5
|
| 51 |
-
]
|
| 52 |
-
]
|
| 53 |
-
}
|
| 54 |
-
},
|
| 55 |
-
"train": {
|
| 56 |
-
"criterions": [
|
| 57 |
-
"feature",
|
| 58 |
-
"discriminator",
|
| 59 |
-
"generator",
|
| 60 |
-
"mel",
|
| 61 |
-
]
|
| 62 |
-
},
|
| 63 |
-
"inference": {
|
| 64 |
-
"batch_size": 1,
|
| 65 |
-
}
|
| 66 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/bigvgan/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/bigvgan_large/exp_config.json
DELETED
|
@@ -1,70 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"preprocess": {
|
| 4 |
-
// acoustic features
|
| 5 |
-
"extract_mel": true,
|
| 6 |
-
"extract_audio": true,
|
| 7 |
-
|
| 8 |
-
// Features used for model training
|
| 9 |
-
"use_mel": true,
|
| 10 |
-
"use_audio": true
|
| 11 |
-
},
|
| 12 |
-
"model": {
|
| 13 |
-
"generator": "bigvgan",
|
| 14 |
-
"bigvgan": {
|
| 15 |
-
"resblock": "1",
|
| 16 |
-
"activation": "snakebeta",
|
| 17 |
-
"snake_logscale": true,
|
| 18 |
-
"upsample_rates": [
|
| 19 |
-
4,
|
| 20 |
-
4,
|
| 21 |
-
2,
|
| 22 |
-
2,
|
| 23 |
-
2,
|
| 24 |
-
2
|
| 25 |
-
],
|
| 26 |
-
"upsample_kernel_sizes": [
|
| 27 |
-
8,
|
| 28 |
-
8,
|
| 29 |
-
4,
|
| 30 |
-
4,
|
| 31 |
-
4,
|
| 32 |
-
4
|
| 33 |
-
],
|
| 34 |
-
"upsample_initial_channel": 1536,
|
| 35 |
-
"resblock_kernel_sizes": [
|
| 36 |
-
3,
|
| 37 |
-
7,
|
| 38 |
-
11
|
| 39 |
-
],
|
| 40 |
-
"resblock_dilation_sizes": [
|
| 41 |
-
[
|
| 42 |
-
1,
|
| 43 |
-
3,
|
| 44 |
-
5
|
| 45 |
-
],
|
| 46 |
-
[
|
| 47 |
-
1,
|
| 48 |
-
3,
|
| 49 |
-
5
|
| 50 |
-
],
|
| 51 |
-
[
|
| 52 |
-
1,
|
| 53 |
-
3,
|
| 54 |
-
5
|
| 55 |
-
]
|
| 56 |
-
]
|
| 57 |
-
},
|
| 58 |
-
},
|
| 59 |
-
"train": {
|
| 60 |
-
"criterions": [
|
| 61 |
-
"feature",
|
| 62 |
-
"discriminator",
|
| 63 |
-
"generator",
|
| 64 |
-
"mel",
|
| 65 |
-
]
|
| 66 |
-
},
|
| 67 |
-
"inference": {
|
| 68 |
-
"batch_size": 1,
|
| 69 |
-
}
|
| 70 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/bigvgan_large/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/exp_config_base.json
DELETED
|
@@ -1,111 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "config/vocoder.json",
|
| 3 |
-
"model_type": "GANVocoder",
|
| 4 |
-
// TODO: Choose your needed datasets
|
| 5 |
-
"dataset": [
|
| 6 |
-
"csd",
|
| 7 |
-
"kising",
|
| 8 |
-
"m4singer",
|
| 9 |
-
"nus48e",
|
| 10 |
-
"opencpop",
|
| 11 |
-
"opensinger",
|
| 12 |
-
"opera",
|
| 13 |
-
"pjs",
|
| 14 |
-
"popbutfy",
|
| 15 |
-
"popcs",
|
| 16 |
-
"ljspeech",
|
| 17 |
-
"vctk",
|
| 18 |
-
"libritts",
|
| 19 |
-
],
|
| 20 |
-
"dataset_path": {
|
| 21 |
-
// TODO: Fill in your dataset path
|
| 22 |
-
"csd": "[dataset path]",
|
| 23 |
-
"kising": "[dataset path]",
|
| 24 |
-
"m4singer": "[dataset path]",
|
| 25 |
-
"nus48e": "[dataset path]",
|
| 26 |
-
"opencpop": "[dataset path]",
|
| 27 |
-
"opensinger": "[dataset path]",
|
| 28 |
-
"opera": "[dataset path]",
|
| 29 |
-
"pjs": "[dataset path]",
|
| 30 |
-
"popbutfy": "[dataset path]",
|
| 31 |
-
"popcs": "[dataset path]",
|
| 32 |
-
"ljspeech": "[dataset path]",
|
| 33 |
-
"vctk": "[dataset path]",
|
| 34 |
-
"libritts": "[dataset path]",
|
| 35 |
-
},
|
| 36 |
-
// TODO: Fill in the output log path
|
| 37 |
-
"log_dir": "ckpts/vocoder",
|
| 38 |
-
"preprocess": {
|
| 39 |
-
// Acoustic features
|
| 40 |
-
"extract_mel": true,
|
| 41 |
-
"extract_audio": true,
|
| 42 |
-
"extract_pitch": false,
|
| 43 |
-
"extract_uv": false,
|
| 44 |
-
"pitch_extractor": "parselmouth",
|
| 45 |
-
|
| 46 |
-
// Features used for model training
|
| 47 |
-
"use_mel": true,
|
| 48 |
-
"use_frame_pitch": false,
|
| 49 |
-
"use_uv": false,
|
| 50 |
-
"use_audio": true,
|
| 51 |
-
|
| 52 |
-
// TODO: Fill in the output data path
|
| 53 |
-
"processed_dir": "data/",
|
| 54 |
-
"n_mel": 100,
|
| 55 |
-
"sample_rate": 24000
|
| 56 |
-
},
|
| 57 |
-
"model": {
|
| 58 |
-
// TODO: Choose your needed discriminators
|
| 59 |
-
"discriminators": [
|
| 60 |
-
"msd",
|
| 61 |
-
"mpd",
|
| 62 |
-
"msstftd",
|
| 63 |
-
"mssbcqtd",
|
| 64 |
-
],
|
| 65 |
-
"mpd": {
|
| 66 |
-
"mpd_reshapes": [
|
| 67 |
-
2,
|
| 68 |
-
3,
|
| 69 |
-
5,
|
| 70 |
-
7,
|
| 71 |
-
11
|
| 72 |
-
],
|
| 73 |
-
"use_spectral_norm": false,
|
| 74 |
-
"discriminator_channel_mult_factor": 1
|
| 75 |
-
},
|
| 76 |
-
"mrd": {
|
| 77 |
-
"resolutions": [[1024, 120, 600], [2048, 240, 1200], [512, 50, 240]],
|
| 78 |
-
"use_spectral_norm": false,
|
| 79 |
-
"discriminator_channel_mult_factor": 1,
|
| 80 |
-
"mrd_override": false
|
| 81 |
-
},
|
| 82 |
-
"msstftd": {
|
| 83 |
-
"filters": 32
|
| 84 |
-
},
|
| 85 |
-
"mssbcqtd": {
|
| 86 |
-
hop_lengths: [512, 256, 256],
|
| 87 |
-
filters: 32,
|
| 88 |
-
max_filters: 1024,
|
| 89 |
-
filters_scale: 1,
|
| 90 |
-
dilations: [1, 2, 4],
|
| 91 |
-
in_channels: 1,
|
| 92 |
-
out_channels: 1,
|
| 93 |
-
n_octaves: [9, 9, 9],
|
| 94 |
-
bins_per_octaves: [24, 36, 48]
|
| 95 |
-
},
|
| 96 |
-
},
|
| 97 |
-
"train": {
|
| 98 |
-
// TODO: Choose a suitable batch size, training epoch, and save stride
|
| 99 |
-
"batch_size": 32,
|
| 100 |
-
"max_epoch": 1000000,
|
| 101 |
-
"save_checkpoint_stride": [20],
|
| 102 |
-
"adamw": {
|
| 103 |
-
"lr": 2.0e-4,
|
| 104 |
-
"adam_b1": 0.8,
|
| 105 |
-
"adam_b2": 0.99
|
| 106 |
-
},
|
| 107 |
-
"exponential_lr": {
|
| 108 |
-
"lr_decay": 0.999
|
| 109 |
-
},
|
| 110 |
-
}
|
| 111 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/hifigan/exp_config.json
DELETED
|
@@ -1,59 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"preprocess": {
|
| 4 |
-
// acoustic features
|
| 5 |
-
"extract_mel": true,
|
| 6 |
-
"extract_audio": true,
|
| 7 |
-
|
| 8 |
-
// Features used for model training
|
| 9 |
-
"use_mel": true,
|
| 10 |
-
"use_audio": true
|
| 11 |
-
},
|
| 12 |
-
"model": {
|
| 13 |
-
"generator": "hifigan",
|
| 14 |
-
"hifigan": {
|
| 15 |
-
"resblock": "2",
|
| 16 |
-
"upsample_rates": [
|
| 17 |
-
8,
|
| 18 |
-
8,
|
| 19 |
-
4
|
| 20 |
-
],
|
| 21 |
-
"upsample_kernel_sizes": [
|
| 22 |
-
16,
|
| 23 |
-
16,
|
| 24 |
-
8
|
| 25 |
-
],
|
| 26 |
-
"upsample_initial_channel": 256,
|
| 27 |
-
"resblock_kernel_sizes": [
|
| 28 |
-
3,
|
| 29 |
-
5,
|
| 30 |
-
7
|
| 31 |
-
],
|
| 32 |
-
"resblock_dilation_sizes": [
|
| 33 |
-
[
|
| 34 |
-
1,
|
| 35 |
-
2
|
| 36 |
-
],
|
| 37 |
-
[
|
| 38 |
-
2,
|
| 39 |
-
6
|
| 40 |
-
],
|
| 41 |
-
[
|
| 42 |
-
3,
|
| 43 |
-
12
|
| 44 |
-
]
|
| 45 |
-
]
|
| 46 |
-
}
|
| 47 |
-
},
|
| 48 |
-
"train": {
|
| 49 |
-
"criterions": [
|
| 50 |
-
"feature",
|
| 51 |
-
"discriminator",
|
| 52 |
-
"generator",
|
| 53 |
-
"mel",
|
| 54 |
-
]
|
| 55 |
-
},
|
| 56 |
-
"inference": {
|
| 57 |
-
"batch_size": 1,
|
| 58 |
-
}
|
| 59 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/hifigan/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/melgan/exp_config.json
DELETED
|
@@ -1,34 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"preprocess": {
|
| 4 |
-
// acoustic features
|
| 5 |
-
"extract_mel": true,
|
| 6 |
-
"extract_audio": true,
|
| 7 |
-
|
| 8 |
-
// Features used for model training
|
| 9 |
-
"use_mel": true,
|
| 10 |
-
"use_audio": true
|
| 11 |
-
},
|
| 12 |
-
"model": {
|
| 13 |
-
"generator": "melgan",
|
| 14 |
-
"melgan": {
|
| 15 |
-
"ratios": [8, 8, 2, 2],
|
| 16 |
-
"ngf": 32,
|
| 17 |
-
"n_residual_layers": 3,
|
| 18 |
-
"num_D": 3,
|
| 19 |
-
"ndf": 16,
|
| 20 |
-
"n_layers": 4,
|
| 21 |
-
"downsampling_factor": 4
|
| 22 |
-
},
|
| 23 |
-
},
|
| 24 |
-
"train": {
|
| 25 |
-
"criterions": [
|
| 26 |
-
"feature",
|
| 27 |
-
"discriminator",
|
| 28 |
-
"generator",
|
| 29 |
-
]
|
| 30 |
-
},
|
| 31 |
-
"inference": {
|
| 32 |
-
"batch_size": 1,
|
| 33 |
-
}
|
| 34 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/melgan/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/nsfhifigan/exp_config.json
DELETED
|
@@ -1,83 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"preprocess": {
|
| 4 |
-
// acoustic features
|
| 5 |
-
"extract_mel": true,
|
| 6 |
-
"extract_audio": true,
|
| 7 |
-
"extract_pitch": true,
|
| 8 |
-
|
| 9 |
-
// Features used for model training
|
| 10 |
-
"use_mel": true,
|
| 11 |
-
"use_audio": true,
|
| 12 |
-
"use_frame_pitch": true
|
| 13 |
-
},
|
| 14 |
-
"model": {
|
| 15 |
-
"generator": "nsfhifigan",
|
| 16 |
-
"nsfhifigan": {
|
| 17 |
-
"resblock": "1",
|
| 18 |
-
"harmonic_num": 8,
|
| 19 |
-
"upsample_rates": [
|
| 20 |
-
8,
|
| 21 |
-
4,
|
| 22 |
-
2,
|
| 23 |
-
2,
|
| 24 |
-
2
|
| 25 |
-
],
|
| 26 |
-
"upsample_kernel_sizes": [
|
| 27 |
-
16,
|
| 28 |
-
8,
|
| 29 |
-
4,
|
| 30 |
-
4,
|
| 31 |
-
4
|
| 32 |
-
],
|
| 33 |
-
"upsample_initial_channel": 768,
|
| 34 |
-
"resblock_kernel_sizes": [
|
| 35 |
-
3,
|
| 36 |
-
7,
|
| 37 |
-
11
|
| 38 |
-
],
|
| 39 |
-
"resblock_dilation_sizes": [
|
| 40 |
-
[
|
| 41 |
-
1,
|
| 42 |
-
3,
|
| 43 |
-
5
|
| 44 |
-
],
|
| 45 |
-
[
|
| 46 |
-
1,
|
| 47 |
-
3,
|
| 48 |
-
5
|
| 49 |
-
],
|
| 50 |
-
[
|
| 51 |
-
1,
|
| 52 |
-
3,
|
| 53 |
-
5
|
| 54 |
-
]
|
| 55 |
-
]
|
| 56 |
-
},
|
| 57 |
-
"mpd": {
|
| 58 |
-
"mpd_reshapes": [
|
| 59 |
-
2,
|
| 60 |
-
3,
|
| 61 |
-
5,
|
| 62 |
-
7,
|
| 63 |
-
11,
|
| 64 |
-
17,
|
| 65 |
-
23,
|
| 66 |
-
37
|
| 67 |
-
],
|
| 68 |
-
"use_spectral_norm": false,
|
| 69 |
-
"discriminator_channel_multi": 1
|
| 70 |
-
}
|
| 71 |
-
},
|
| 72 |
-
"train": {
|
| 73 |
-
"criterions": [
|
| 74 |
-
"feature",
|
| 75 |
-
"discriminator",
|
| 76 |
-
"generator",
|
| 77 |
-
"mel",
|
| 78 |
-
]
|
| 79 |
-
},
|
| 80 |
-
"inference": {
|
| 81 |
-
"batch_size": 1,
|
| 82 |
-
}
|
| 83 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/nsfhifigan/run.sh
DELETED
|
@@ -1,143 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 114 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 115 |
-
--config $exp_config \
|
| 116 |
-
--infer_mode $infer_mode \
|
| 117 |
-
--infer_datasets $infer_datasets \
|
| 118 |
-
--vocoder_dir $infer_expt_dir \
|
| 119 |
-
--output_dir $infer_output_dir \
|
| 120 |
-
--log_level debug
|
| 121 |
-
fi
|
| 122 |
-
|
| 123 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 124 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 125 |
-
--config $exp_config \
|
| 126 |
-
--infer_mode $infer_mode \
|
| 127 |
-
--feature_folder $infer_feature_dir \
|
| 128 |
-
--vocoder_dir $infer_expt_dir \
|
| 129 |
-
--output_dir $infer_output_dir \
|
| 130 |
-
--log_level debug
|
| 131 |
-
fi
|
| 132 |
-
|
| 133 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 134 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 135 |
-
--config $exp_config \
|
| 136 |
-
--infer_mode $infer_mode \
|
| 137 |
-
--audio_folder $infer_audio_dir \
|
| 138 |
-
--vocoder_dir $infer_expt_dir \
|
| 139 |
-
--output_dir $infer_output_dir \
|
| 140 |
-
--log_level debug
|
| 141 |
-
fi
|
| 142 |
-
|
| 143 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/tfr_enhanced_hifigan/README.md
DELETED
|
@@ -1,185 +0,0 @@
|
|
| 1 |
-
# Multi-Scale Sub-Band Constant-Q Transform Discriminator for High-Fedility Vocoder
|
| 2 |
-
|
| 3 |
-
[](https://arxiv.org/abs/2311.14957)
|
| 4 |
-
[](https://vocodexelysium.github.io/MS-SB-CQTD/)
|
| 5 |
-
|
| 6 |
-
<br>
|
| 7 |
-
<div align="center">
|
| 8 |
-
<img src="../../../../imgs/vocoder/gan/MSSBCQTD.png" width="80%">
|
| 9 |
-
</div>
|
| 10 |
-
<br>
|
| 11 |
-
|
| 12 |
-
This is the official implementation of the paper "[Multi-Scale Sub-Band Constant-Q Transform Discriminator for High-Fidelity Vocoder](https://arxiv.org/abs/2311.14957)". In this recipe, we will illustrate how to train a high quality HiFi-GAN on LibriTTS, VCTK and LJSpeech via utilizing multiple Time-Frequency-Representation-based Discriminators.
|
| 13 |
-
|
| 14 |
-
There are four stages in total:
|
| 15 |
-
|
| 16 |
-
1. Data preparation
|
| 17 |
-
2. Feature extraction
|
| 18 |
-
3. Training
|
| 19 |
-
4. Inference
|
| 20 |
-
|
| 21 |
-
> **NOTE:** You need to run every command of this recipe in the `Amphion` root path:
|
| 22 |
-
> ```bash
|
| 23 |
-
> cd Amphion
|
| 24 |
-
> ```
|
| 25 |
-
|
| 26 |
-
## 1. Data Preparation
|
| 27 |
-
|
| 28 |
-
### Dataset Download
|
| 29 |
-
|
| 30 |
-
By default, we utilize the three datasets for training: LibriTTS, VCTK and LJSpeech. How to download them is detailed in [here](../../../datasets/README.md).
|
| 31 |
-
|
| 32 |
-
### Configuration
|
| 33 |
-
|
| 34 |
-
Specify the dataset path in `exp_config.json`. Note that you can change the `dataset` list to use your preferred datasets.
|
| 35 |
-
|
| 36 |
-
```json
|
| 37 |
-
"dataset": [
|
| 38 |
-
"ljspeech",
|
| 39 |
-
"vctk",
|
| 40 |
-
"libritts",
|
| 41 |
-
],
|
| 42 |
-
"dataset_path": {
|
| 43 |
-
// TODO: Fill in your dataset path
|
| 44 |
-
"ljspeech": "[LJSpeech dataset path]",
|
| 45 |
-
"vctk": "[VCTK dataset path]",
|
| 46 |
-
"libritts": "[LibriTTS dataset path]",
|
| 47 |
-
},
|
| 48 |
-
```
|
| 49 |
-
|
| 50 |
-
## 2. Features Extraction
|
| 51 |
-
|
| 52 |
-
For HiFiGAN, only the Mel-Spectrogram and the Output Audio are needed for training.
|
| 53 |
-
|
| 54 |
-
### Configuration
|
| 55 |
-
|
| 56 |
-
Specify the dataset path and the output path for saving the processed data and the training model in `exp_config.json`:
|
| 57 |
-
|
| 58 |
-
```json
|
| 59 |
-
// TODO: Fill in the output log path. The default value is "Amphion/ckpts/vocoder"
|
| 60 |
-
"log_dir": "ckpts/vocoder",
|
| 61 |
-
"preprocess": {
|
| 62 |
-
// TODO: Fill in the output data path. The default value is "Amphion/data"
|
| 63 |
-
"processed_dir": "data",
|
| 64 |
-
...
|
| 65 |
-
},
|
| 66 |
-
```
|
| 67 |
-
|
| 68 |
-
### Run
|
| 69 |
-
|
| 70 |
-
Run the `run.sh` as the preproces stage (set `--stage 1`).
|
| 71 |
-
|
| 72 |
-
```bash
|
| 73 |
-
sh egs/vocoder/gan/tfr_enhanced_hifigan/run.sh --stage 1
|
| 74 |
-
```
|
| 75 |
-
|
| 76 |
-
> **NOTE:** The `CUDA_VISIBLE_DEVICES` is set as `"0"` in default. You can change it when running `run.sh` by specifying such as `--gpu "1"`.
|
| 77 |
-
|
| 78 |
-
## 3. Training
|
| 79 |
-
|
| 80 |
-
### Configuration
|
| 81 |
-
|
| 82 |
-
We provide the default hyparameters in the `exp_config.json`. They can work on single NVIDIA-24g GPU. You can adjust them based on you GPU machines.
|
| 83 |
-
|
| 84 |
-
```json
|
| 85 |
-
"train": {
|
| 86 |
-
"batch_size": 32,
|
| 87 |
-
...
|
| 88 |
-
}
|
| 89 |
-
```
|
| 90 |
-
|
| 91 |
-
### Run
|
| 92 |
-
|
| 93 |
-
Run the `run.sh` as the training stage (set `--stage 2`). Specify a experimental name to run the following command. The tensorboard logs and checkpoints will be saved in `Amphion/ckpts/vocoder/[YourExptName]`.
|
| 94 |
-
|
| 95 |
-
```bash
|
| 96 |
-
sh egs/vocoder/gan/tfr_enhanced_hifigan/run.sh --stage 2 --name [YourExptName]
|
| 97 |
-
```
|
| 98 |
-
|
| 99 |
-
> **NOTE:** The `CUDA_VISIBLE_DEVICES` is set as `"0"` in default. You can change it when running `run.sh` by specifying such as `--gpu "0,1,2,3"`.
|
| 100 |
-
|
| 101 |
-
## 4. Inference
|
| 102 |
-
|
| 103 |
-
### Pretrained Vocoder Download
|
| 104 |
-
|
| 105 |
-
We trained a HiFiGAN checkpoint with around 685 hours Speech data. The final pretrained checkpoint is released [here](../../../../pretrained/hifigan/README.md).
|
| 106 |
-
|
| 107 |
-
### Run
|
| 108 |
-
|
| 109 |
-
Run the `run.sh` as the training stage (set `--stage 3`), we provide three different inference modes, including `infer_from_dataset`, `infer_from_feature`, `and infer_from audio`.
|
| 110 |
-
|
| 111 |
-
```bash
|
| 112 |
-
sh egs/vocoder/gan/tfr_enhanced_hifigan/run.sh --stage 3 \
|
| 113 |
-
--infer_mode [Your chosen inference mode] \
|
| 114 |
-
--infer_datasets [Datasets you want to inference, needed when infer_from_dataset] \
|
| 115 |
-
--infer_feature_dir [Your path to your predicted acoustic features, needed when infer_from_feature] \
|
| 116 |
-
--infer_audio_dir [Your path to your audio files, needed when infer_form_audio] \
|
| 117 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 118 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 119 |
-
```
|
| 120 |
-
|
| 121 |
-
#### a. Inference from Dataset
|
| 122 |
-
|
| 123 |
-
Run the `run.sh` with specified datasets, here is an example.
|
| 124 |
-
|
| 125 |
-
```bash
|
| 126 |
-
sh egs/vocoder/gan/tfr_enhanced_hifigan/run.sh --stage 3 \
|
| 127 |
-
--infer_mode infer_from_dataset \
|
| 128 |
-
--infer_datasets "libritts vctk ljspeech" \
|
| 129 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 130 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 131 |
-
```
|
| 132 |
-
|
| 133 |
-
#### b. Inference from Features
|
| 134 |
-
|
| 135 |
-
If you want to inference from your generated acoustic features, you should first prepare your acoustic features into the following structure:
|
| 136 |
-
|
| 137 |
-
```plaintext
|
| 138 |
-
┣ {infer_feature_dir}
|
| 139 |
-
┃ ┣ mels
|
| 140 |
-
┃ ┃ ┣ sample1.npy
|
| 141 |
-
┃ ┃ ┣ sample2.npy
|
| 142 |
-
```
|
| 143 |
-
|
| 144 |
-
Then run the `run.sh` with specificed folder direction, here is an example.
|
| 145 |
-
|
| 146 |
-
```bash
|
| 147 |
-
sh egs/vocoder/gan/tfr_enhanced_hifigan/run.sh --stage 3 \
|
| 148 |
-
--infer_mode infer_from_feature \
|
| 149 |
-
--infer_feature_dir [Your path to your predicted acoustic features] \
|
| 150 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 151 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 152 |
-
```
|
| 153 |
-
|
| 154 |
-
#### c. Inference from Audios
|
| 155 |
-
|
| 156 |
-
If you want to inference from audios for quick analysis synthesis, you should first prepare your audios into the following structure:
|
| 157 |
-
|
| 158 |
-
```plaintext
|
| 159 |
-
┣ audios
|
| 160 |
-
┃ ┣ sample1.wav
|
| 161 |
-
┃ ┣ sample2.wav
|
| 162 |
-
```
|
| 163 |
-
|
| 164 |
-
Then run the `run.sh` with specificed folder direction, here is an example.
|
| 165 |
-
|
| 166 |
-
```bash
|
| 167 |
-
sh egs/vocoder/gan/tfr_enhanced_hifigan/run.sh --stage 3 \
|
| 168 |
-
--infer_mode infer_from_audio \
|
| 169 |
-
--infer_audio_dir [Your path to your audio files] \
|
| 170 |
-
--infer_expt_dir Amphion/ckpts/vocoder/[YourExptName] \
|
| 171 |
-
--infer_output_dir Amphion/ckpts/vocoder/[YourExptName]/result \
|
| 172 |
-
```
|
| 173 |
-
|
| 174 |
-
## Citations
|
| 175 |
-
|
| 176 |
-
```bibtex
|
| 177 |
-
@misc{gu2023cqt,
|
| 178 |
-
title={Multi-Scale Sub-Band Constant-Q Transform Discriminator for High-Fidelity Vocoder},
|
| 179 |
-
author={Yicheng Gu and Xueyao Zhang and Liumeng Xue and Zhizheng Wu},
|
| 180 |
-
year={2023},
|
| 181 |
-
eprint={2311.14957},
|
| 182 |
-
archivePrefix={arXiv},
|
| 183 |
-
primaryClass={cs.SD}
|
| 184 |
-
}
|
| 185 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/tfr_enhanced_hifigan/exp_config.json
DELETED
|
@@ -1,118 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"base_config": "egs/vocoder/gan/exp_config_base.json",
|
| 3 |
-
"model_type": "GANVocoder",
|
| 4 |
-
"dataset": [
|
| 5 |
-
"ljspeech",
|
| 6 |
-
"vctk",
|
| 7 |
-
"libritts",
|
| 8 |
-
],
|
| 9 |
-
"dataset_path": {
|
| 10 |
-
// TODO: Fill in your dataset path
|
| 11 |
-
"ljspeech": "[dataset path]",
|
| 12 |
-
"vctk": "[dataset path]",
|
| 13 |
-
"libritts": "[dataset path]",
|
| 14 |
-
},
|
| 15 |
-
// TODO: Fill in the output log path. The default value is "Amphion/ckpts/vocoder"
|
| 16 |
-
"log_dir": "ckpts/vocoder",
|
| 17 |
-
"preprocess": {
|
| 18 |
-
// TODO: Fill in the output data path. The default value is "Amphion/data"
|
| 19 |
-
"processed_dir": "data",
|
| 20 |
-
// acoustic features
|
| 21 |
-
"extract_mel": true,
|
| 22 |
-
"extract_audio": true,
|
| 23 |
-
"extract_pitch": false,
|
| 24 |
-
"extract_uv": false,
|
| 25 |
-
"extract_amplitude_phase": false,
|
| 26 |
-
"pitch_extractor": "parselmouth",
|
| 27 |
-
// Features used for model training
|
| 28 |
-
"use_mel": true,
|
| 29 |
-
"use_frame_pitch": false,
|
| 30 |
-
"use_uv": false,
|
| 31 |
-
"use_audio": true,
|
| 32 |
-
"n_mel": 100,
|
| 33 |
-
"sample_rate": 24000
|
| 34 |
-
},
|
| 35 |
-
"model": {
|
| 36 |
-
"generator": "hifigan",
|
| 37 |
-
"discriminators": [
|
| 38 |
-
"msd",
|
| 39 |
-
"mpd",
|
| 40 |
-
"mssbcqtd",
|
| 41 |
-
"msstftd",
|
| 42 |
-
],
|
| 43 |
-
"hifigan": {
|
| 44 |
-
"resblock": "1",
|
| 45 |
-
"upsample_rates": [
|
| 46 |
-
8,
|
| 47 |
-
4,
|
| 48 |
-
2,
|
| 49 |
-
2,
|
| 50 |
-
2
|
| 51 |
-
],
|
| 52 |
-
"upsample_kernel_sizes": [
|
| 53 |
-
16,
|
| 54 |
-
8,
|
| 55 |
-
4,
|
| 56 |
-
4,
|
| 57 |
-
4
|
| 58 |
-
],
|
| 59 |
-
"upsample_initial_channel": 768,
|
| 60 |
-
"resblock_kernel_sizes": [
|
| 61 |
-
3,
|
| 62 |
-
5,
|
| 63 |
-
7
|
| 64 |
-
],
|
| 65 |
-
"resblock_dilation_sizes": [
|
| 66 |
-
[
|
| 67 |
-
1,
|
| 68 |
-
3,
|
| 69 |
-
5
|
| 70 |
-
],
|
| 71 |
-
[
|
| 72 |
-
1,
|
| 73 |
-
3,
|
| 74 |
-
5
|
| 75 |
-
],
|
| 76 |
-
[
|
| 77 |
-
1,
|
| 78 |
-
3,
|
| 79 |
-
5
|
| 80 |
-
]
|
| 81 |
-
]
|
| 82 |
-
},
|
| 83 |
-
"mpd": {
|
| 84 |
-
"mpd_reshapes": [
|
| 85 |
-
2,
|
| 86 |
-
3,
|
| 87 |
-
5,
|
| 88 |
-
7,
|
| 89 |
-
11,
|
| 90 |
-
17,
|
| 91 |
-
23,
|
| 92 |
-
37
|
| 93 |
-
],
|
| 94 |
-
"use_spectral_norm": false,
|
| 95 |
-
"discriminator_channel_multi": 1
|
| 96 |
-
}
|
| 97 |
-
},
|
| 98 |
-
"train": {
|
| 99 |
-
"batch_size": 16,
|
| 100 |
-
"adamw": {
|
| 101 |
-
"lr": 2.0e-4,
|
| 102 |
-
"adam_b1": 0.8,
|
| 103 |
-
"adam_b2": 0.99
|
| 104 |
-
},
|
| 105 |
-
"exponential_lr": {
|
| 106 |
-
"lr_decay": 0.999
|
| 107 |
-
},
|
| 108 |
-
"criterions": [
|
| 109 |
-
"feature",
|
| 110 |
-
"discriminator",
|
| 111 |
-
"generator",
|
| 112 |
-
"mel",
|
| 113 |
-
]
|
| 114 |
-
},
|
| 115 |
-
"inference": {
|
| 116 |
-
"batch_size": 1,
|
| 117 |
-
}
|
| 118 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
egs/vocoder/gan/tfr_enhanced_hifigan/run.sh
DELETED
|
@@ -1,145 +0,0 @@
|
|
| 1 |
-
# Copyright (c) 2023 Amphion.
|
| 2 |
-
#
|
| 3 |
-
# This source code is licensed under the MIT license found in the
|
| 4 |
-
# LICENSE file in the root directory of this source tree.
|
| 5 |
-
|
| 6 |
-
######## Build Experiment Environment ###########
|
| 7 |
-
exp_dir=$(cd `dirname $0`; pwd)
|
| 8 |
-
work_dir=$(dirname $(dirname $(dirname $(dirname $exp_dir))))
|
| 9 |
-
|
| 10 |
-
export WORK_DIR=$work_dir
|
| 11 |
-
export PYTHONPATH=$work_dir
|
| 12 |
-
export PYTHONIOENCODING=UTF-8
|
| 13 |
-
|
| 14 |
-
######## Parse the Given Parameters from the Commond ###########
|
| 15 |
-
options=$(getopt -o c:n:s --long gpu:,config:,name:,stage:,resume:,checkpoint:,resume_type:,infer_mode:,infer_datasets:,infer_feature_dir:,infer_audio_dir:,infer_expt_dir:,infer_output_dir: -- "$@")
|
| 16 |
-
eval set -- "$options"
|
| 17 |
-
|
| 18 |
-
while true; do
|
| 19 |
-
case $1 in
|
| 20 |
-
# Experimental Configuration File
|
| 21 |
-
-c | --config) shift; exp_config=$1 ; shift ;;
|
| 22 |
-
# Experimental Name
|
| 23 |
-
-n | --name) shift; exp_name=$1 ; shift ;;
|
| 24 |
-
# Running Stage
|
| 25 |
-
-s | --stage) shift; running_stage=$1 ; shift ;;
|
| 26 |
-
# Visible GPU machines. The default value is "0".
|
| 27 |
-
--gpu) shift; gpu=$1 ; shift ;;
|
| 28 |
-
|
| 29 |
-
# [Only for Training] Resume configuration
|
| 30 |
-
--resume) shift; resume=$1 ; shift ;;
|
| 31 |
-
# [Only for Training] The specific checkpoint path that you want to resume from.
|
| 32 |
-
--checkpoint) shift; cehckpoint=$1 ; shift ;;
|
| 33 |
-
# [Only for Training] `resume` for loading all the things (including model weights, optimizer, scheduler, and random states). `finetune` for loading only the model weights.
|
| 34 |
-
--resume_type) shift; resume_type=$1 ; shift ;;
|
| 35 |
-
|
| 36 |
-
# [Only for Inference] The inference mode
|
| 37 |
-
--infer_mode) shift; infer_mode=$1 ; shift ;;
|
| 38 |
-
# [Only for Inference] The inferenced datasets
|
| 39 |
-
--infer_datasets) shift; infer_datasets=$1 ; shift ;;
|
| 40 |
-
# [Only for Inference] The feature dir for inference
|
| 41 |
-
--infer_feature_dir) shift; infer_feature_dir=$1 ; shift ;;
|
| 42 |
-
# [Only for Inference] The audio dir for inference
|
| 43 |
-
--infer_audio_dir) shift; infer_audio_dir=$1 ; shift ;;
|
| 44 |
-
# [Only for Inference] The experiment dir. The value is like "[Your path to save logs and checkpoints]/[YourExptName]"
|
| 45 |
-
--infer_expt_dir) shift; infer_expt_dir=$1 ; shift ;;
|
| 46 |
-
# [Only for Inference] The output dir to save inferred audios. Its default value is "$expt_dir/result"
|
| 47 |
-
--infer_output_dir) shift; infer_output_dir=$1 ; shift ;;
|
| 48 |
-
|
| 49 |
-
--) shift ; break ;;
|
| 50 |
-
*) echo "Invalid option: $1" exit 1 ;;
|
| 51 |
-
esac
|
| 52 |
-
done
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
### Value check ###
|
| 56 |
-
if [ -z "$running_stage" ]; then
|
| 57 |
-
echo "[Error] Please specify the running stage"
|
| 58 |
-
exit 1
|
| 59 |
-
fi
|
| 60 |
-
|
| 61 |
-
if [ -z "$exp_config" ]; then
|
| 62 |
-
exp_config="${exp_dir}"/exp_config.json
|
| 63 |
-
fi
|
| 64 |
-
echo "Exprimental Configuration File: $exp_config"
|
| 65 |
-
|
| 66 |
-
if [ -z "$gpu" ]; then
|
| 67 |
-
gpu="0"
|
| 68 |
-
fi
|
| 69 |
-
|
| 70 |
-
######## Features Extraction ###########
|
| 71 |
-
if [ $running_stage -eq 1 ]; then
|
| 72 |
-
CUDA_VISIBLE_DEVICES=$gpu python "${work_dir}"/bins/vocoder/preprocess.py \
|
| 73 |
-
--config $exp_config \
|
| 74 |
-
--num_workers 8
|
| 75 |
-
fi
|
| 76 |
-
|
| 77 |
-
######## Training ###########
|
| 78 |
-
if [ $running_stage -eq 2 ]; then
|
| 79 |
-
if [ -z "$exp_name" ]; then
|
| 80 |
-
echo "[Error] Please specify the experiments name"
|
| 81 |
-
exit 1
|
| 82 |
-
fi
|
| 83 |
-
echo "Exprimental Name: $exp_name"
|
| 84 |
-
|
| 85 |
-
if [ "$resume" = true ]; then
|
| 86 |
-
echo "Automatically resume from the experimental dir..."
|
| 87 |
-
CUDA_VISIBLE_DEVICES="$gpu" accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 88 |
-
--config "$exp_config" \
|
| 89 |
-
--exp_name "$exp_name" \
|
| 90 |
-
--log_level info \
|
| 91 |
-
--resume
|
| 92 |
-
else
|
| 93 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "${work_dir}"/bins/vocoder/train.py \
|
| 94 |
-
--config "$exp_config" \
|
| 95 |
-
--exp_name "$exp_name" \
|
| 96 |
-
--log_level info \
|
| 97 |
-
--checkpoint "$checkpoint" \
|
| 98 |
-
--resume_type "$resume_type"
|
| 99 |
-
fi
|
| 100 |
-
fi
|
| 101 |
-
|
| 102 |
-
######## Inference/Conversion ###########
|
| 103 |
-
if [ $running_stage -eq 3 ]; then
|
| 104 |
-
if [ -z "$infer_expt_dir" ]; then
|
| 105 |
-
echo "[Error] Please specify the experimental directionary. The value is like [Your path to save logs and checkpoints]/[YourExptName]"
|
| 106 |
-
exit 1
|
| 107 |
-
fi
|
| 108 |
-
|
| 109 |
-
if [ -z "$infer_output_dir" ]; then
|
| 110 |
-
infer_output_dir="$infer_expt_dir/result"
|
| 111 |
-
fi
|
| 112 |
-
|
| 113 |
-
echo $infer_datasets
|
| 114 |
-
|
| 115 |
-
if [ $infer_mode = "infer_from_dataset" ]; then
|
| 116 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 117 |
-
--config $exp_config \
|
| 118 |
-
--infer_mode $infer_mode \
|
| 119 |
-
--infer_datasets $infer_datasets \
|
| 120 |
-
--vocoder_dir $infer_expt_dir \
|
| 121 |
-
--output_dir $infer_output_dir \
|
| 122 |
-
--log_level debug
|
| 123 |
-
fi
|
| 124 |
-
|
| 125 |
-
if [ $infer_mode = "infer_from_feature" ]; then
|
| 126 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 127 |
-
--config $exp_config \
|
| 128 |
-
--infer_mode $infer_mode \
|
| 129 |
-
--feature_folder $infer_feature_dir \
|
| 130 |
-
--vocoder_dir $infer_expt_dir \
|
| 131 |
-
--output_dir $infer_output_dir \
|
| 132 |
-
--log_level debug
|
| 133 |
-
fi
|
| 134 |
-
|
| 135 |
-
if [ $infer_mode = "infer_from_audio" ]; then
|
| 136 |
-
CUDA_VISIBLE_DEVICES=$gpu accelerate launch "$work_dir"/bins/vocoder/inference.py \
|
| 137 |
-
--config $exp_config \
|
| 138 |
-
--infer_mode $infer_mode \
|
| 139 |
-
--audio_folder $infer_audio_dir \
|
| 140 |
-
--vocoder_dir $infer_expt_dir \
|
| 141 |
-
--output_dir $infer_output_dir \
|
| 142 |
-
--log_level debug
|
| 143 |
-
fi
|
| 144 |
-
|
| 145 |
-
fi
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
examples/chinese_female_recordings.wav
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:f710270fe3857211c55aaa1f813e310e68855ff9eabaf5b249537a2d4277cc30
|
| 3 |
-
size 448928
|
|
|
|
|
|
|
|
|
|
|
|
examples/chinese_male_seperated.wav
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:009077a677b23bff3154078930e6c624d218eb0acbe78990bec88f6bf5a6e5de
|
| 3 |
-
size 480044
|
|
|
|
|
|
|
|
|
|
|
|
examples/english_female_seperated.wav
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:87e75863ffb4e597467a825d019217e73d64dce1e9635de60a32559ffcb97cf4
|
| 3 |
-
size 1509584
|
|
|
|
|
|
|
|
|
|
|
|
examples/english_male_recordings.wav
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:e14ebf1c554ebb25e5169b4bcda36a685538e94c531f303339bad91ff93a2288
|
| 3 |
-
size 251948
|
|
|
|
|
|
|
|
|
|
|
|
examples/output/.DS_Store
DELETED
|
Binary file (6.15 kB)
|
|
|