| { |
| "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 |
| } |
| } |
