|
{ |
|
"network_config": { |
|
"channels_input": 1, |
|
"channels_output": 1, |
|
"channels_H": 64, |
|
"max_H": 768, |
|
"encoder_n_layers": 8, |
|
"kernel_size": 4, |
|
"stride": 2, |
|
"tsfm_n_layers": 5, |
|
"tsfm_n_head": 8, |
|
"tsfm_d_model": 512, |
|
"tsfm_d_inner": 2048 |
|
}, |
|
"train_config": { |
|
"exp_path": "DNS-large-full", |
|
"log":{ |
|
"directory": "./exp", |
|
"ckpt_iter": "max", |
|
"iters_per_ckpt": 10000, |
|
"iters_per_valid": 500 |
|
}, |
|
"optimization":{ |
|
"n_iters": 250000, |
|
"learning_rate": 2e-4, |
|
"batch_size_per_gpu": 8 |
|
}, |
|
"loss_config":{ |
|
"ell_p": 1, |
|
"ell_p_lambda": 1, |
|
"stft_lambda": 1, |
|
"stft_config":{ |
|
"sc_lambda": 0.5, |
|
"mag_lambda": 0.5, |
|
"band": "full", |
|
"hop_sizes": [50, 120, 240], |
|
"win_lengths": [240, 600, 1200], |
|
"fft_sizes": [512, 1024, 2048] |
|
} |
|
} |
|
}, |
|
"trainset_config": { |
|
"root": "./dns", |
|
"crop_length_sec": 10, |
|
"sample_rate": 16000 |
|
}, |
|
"gen_config":{ |
|
"output_directory": "./exp" |
|
}, |
|
"dist_config": { |
|
"dist_backend": "nccl", |
|
"dist_url": "tcp://localhost:54321" |
|
} |
|
} |
|
|