File size: 1,404 Bytes
a29b625 | 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 56 57 58 59 60 61 62 63 | {
"n_filters": 64,
"strides": [
8,
5,
4,
2
],
"dimension": 512,
"semantic_dimension": 768,
"bidirectional": true,
"dilation_base": 2,
"residual_kernel_size": 3,
"n_residual_layers": 1,
"lstm_layers": 2,
"activation": "ELU",
"codebook_size": 1024,
"n_q": 8,
"num_mels": 80,
"num_freq": 1025,
"n_fft": 1024,
"hop_size": 240,
"win_size": 1024,
"fmin": 0,
"fmax": 8000,
"fmax_for_loss": null,
"mel_loss_lambdas": [
45,
1,
1,
1
],
"recon_loss_lambda": 500,
"commitment_loss_lambda": 10,
"distill_loss_lambda": 120,
"distill_type": "d_axis",
"semantic_model_path": "/remote-home/share/models/hubert/ls_960/hubert-base-ls960",
"semantic_model_layer": "avg",
"train_files": "train_file_list.txt",
"valid_files": "dev_file_list.txt",
"results_folder": "Log/spt_base",
"sample_rate": 16000,
"batch_size": 8,
"epochs": 1000,
"learning_rate": 0.0001,
"intial_learning_rate": 0.0001,
"num_warmup_steps": 0,
"betas": [
0.9,
0.99
],
"adam_b2": 0.9,
"lr_decay": 0.98,
"seed": 1234,
"segment_size": 48000,
"wd": 0,
"num_workers": 8,
"log_steps": 100,
"stdout_steps": 10,
"save_model_steps": 1000,
"num_ckpt_keep": 6,
"showpiece_num": 8
} |