|
{ |
|
"wandb": { |
|
"mode": "disabled", |
|
"checkpoint": "checkpoint_name", |
|
"fork_checkpoint": false, |
|
"project": "project_name", |
|
"wandb_prefix": "wandb_prefix", |
|
"dir": "wandb_dir" |
|
}, |
|
"local": { |
|
"checkpoint_model": null, |
|
"checkpoint_discriminator": null |
|
}, |
|
"checkpoint_dir": "path/to/checkpoints", |
|
"seed": 123, |
|
"learning_rate": 0.0001, |
|
"dataset": "Jamendo, LibriTTS", |
|
"train_datafile": "data/train.txt", |
|
"validation_datafile": "data/val_short.txt", |
|
"n_epochs": 500000, |
|
"step_checkpoint": 10000, |
|
"step_media_log": 2000, |
|
"batch_grad_log": 2000, |
|
"batch_size": 32, |
|
"sample_rate": 44100, |
|
"exp_gamma": 0.9995, |
|
"adam_b1": 0.8, |
|
"adam_b2": 0.99, |
|
"segment_size": 16384, |
|
"segment_size_val": 262144, |
|
"n_cache_reuse": 30, |
|
"num_workers": 4, |
|
"prefetch_factor": 2, |
|
"use_discriminator": true, |
|
"unfreeze": { |
|
"steps": 100000, |
|
"loss_multiplier": { |
|
"loss_z": 0, |
|
"loss_ms_mel": 15, |
|
"loss_ms_stft": 1, |
|
"loss_adv_gen": 1, |
|
"loss_adv_feat": 2, |
|
"loss_waveform": 1 |
|
} |
|
}, |
|
"model": { |
|
"latent_dim": 1024, |
|
"n_codebooks": 9, |
|
"codebook_dim": 8, |
|
"codebook_size": 1024, |
|
"n_resblocks": 6, |
|
"initial_out_channels": 1024, |
|
"intermediate_dim": 1152, |
|
"resblock_type": "AMP", |
|
"resblock_kernel_sizes": [3, 3, 7, 7, 11, 11], |
|
"resblock_dilations": [[1, 3, 5], [1, 3, 5], [1, 3, 5], [1, 3, 5], [1, 3, 5], [1, 3, 5]], |
|
"predict_type": "z", |
|
"activation": "snake" |
|
}, |
|
"disc": { |
|
"resolutions": [[1024, 120, 600], [2048, 240, 1200], [512, 50, 240]], |
|
"periods": [2, 3, 5, 7, 11], |
|
"discriminator_channel_mult": 1, |
|
"use_spectral_norm": false |
|
}, |
|
"mel": { |
|
"n_fft": 1024, |
|
"win_length": 1024, |
|
"hop_length": 256, |
|
"f_min": 0, |
|
"f_max": null, |
|
"n_mels": 128 |
|
}, |
|
"loss_multiplier": { |
|
"loss_z": 15, |
|
"loss_ms_mel": 15, |
|
"loss_ms_stft": 1, |
|
"loss_adv_gen": 1, |
|
"loss_adv_feat": 2, |
|
"loss_waveform": 1 |
|
}, |
|
"backend": { |
|
"master_port": 12359 |
|
} |
|
} |