| clip_grad: 5.0 | |
| config: confs/whole_utt/dae-bsrnn.yaml | |
| data_type: shard | |
| dataloader_args: | |
| batch_size: 1 | |
| drop_last: true | |
| num_workers: 1 | |
| pin_memory: true | |
| prefetch_factor: 6 | |
| dataset_args: | |
| SSA_enroll_prob: 0 | |
| additional: | |
| train: | |
| data_list: data/text_cue/dump/ready/train-100.jsonl | |
| data_type: TC-ASR | |
| val: | |
| data_list: data/text_cue/dump/ready/dev.jsonl | |
| data_type: TC-ASR | |
| chunk_len: 48000 | |
| fbank_args: | |
| dither: 1.0 | |
| frame_length: 25 | |
| frame_shift: 10 | |
| num_mel_bins: 80 | |
| noise_enroll_prob: 0 | |
| noise_lmdb_file: ./data/musan/lmdb | |
| noise_prob: 0 | |
| resample_rate: 16000 | |
| reverb_enroll_prob: 0 | |
| sample_num_per_epoch: 0 | |
| shuffle: false | |
| shuffle_args: | |
| shuffle_size: 2500 | |
| speaker_feat: true | |
| specaug_enroll_prob: 0 | |
| enable_amp: false | |
| exp_dir: exp/whole_utt/dae-bsrnn | |
| gpus: | |
| - 0 | |
| - 1 | |
| - 2 | |
| - 3 | |
| - 4 | |
| - 5 | |
| - 6 | |
| - 7 | |
| log_batch_interval: 100 | |
| loss: SISDR | |
| loss_args: {} | |
| model: | |
| tse_model: DAE-BSRNN | |
| model_args: | |
| tse_model: | |
| cue_emb_dim: 128 | |
| asr_encoder: | |
| model_name: 'TCASREncoder' | |
| checkpoint_path: exp/kce/epoch_149.pt #'checkpoint directory to the kce encoder', e.g. Pretrained/mix-clean/kce/epoch_149.pt | |
| fnn: | |
| input_dim: 256 | |
| num_layer: 3 | |
| output_dim: 128 | |
| multi_task: false | |
| spksInTrain: 251 | |
| cue_fuse: | |
| fuse_type: FiLM | |
| multi_fuse: false | |
| feature_dim: 128 | |
| multi_task: false | |
| num_repeat: 6 | |
| spk_optim_head: asr | |
| sr: 16000 | |
| stride: 128 | |
| win: 512 | |
| model_init: | |
| discriminator: null | |
| spk_model: null | |
| tse_model: null | |
| num_avg: 10 | |
| num_epochs: 150 | |
| optimizer: | |
| tse_model: Adam | |
| optimizer_args: | |
| tse_model: | |
| lr: 0.001 | |
| weight_decay: 0.0001 | |
| save_epoch_interval: 1 | |
| scheduler: | |
| tse_model: ExponentialDecrease | |
| scheduler_args: | |
| scale_ratio: 1.0 | |
| tse_model: | |
| epoch_iter: 1737 | |
| final_lr: 2.5e-05 | |
| initial_lr: 0.001 | |
| num_epochs: 150 | |
| warm_from_zero: false | |
| warm_up_epoch: 0 | |
| seed: 42 | |
| train_data: data/clean/train-100/shard.list | |
| train_spk2utt: data/clean/train-100/spk2enroll.json | |
| train_utt2spk: data/clean/train-100/single.utt2spk | |
| val_data: data/clean/dev/shard.list | |
| val_spk1_enroll: data/clean/dev/spk1.enroll | |
| val_spk2_enroll: data/clean/dev/spk2.enroll | |
| val_spk2utt: data/clean/dev/single.wav.scp | |
| whole_utt: true | |