Spaces:
Runtime error
Runtime error
File size: 1,392 Bytes
33e3a91 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
{
"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"
}
}
|