| checkpoint: exp/redimnet2/expB6_v2/-LM/models/model_0.pt | |
| data_type: shard | |
| dataloader_args: | |
| batch_size: 16 | |
| drop_last: true | |
| num_workers: 4 | |
| pin_memory: false | |
| prefetch_factor: 4 | |
| dataset_args: | |
| aug_prob: 0.0 | |
| cmvn: false | |
| filter: false | |
| frontend: tfmel | |
| num_frms: 600 | |
| resample_rate: 16000 | |
| sample_num_per_epoch: 0 | |
| shuffle: true | |
| shuffle_args: | |
| shuffle_size: 2500 | |
| spec_aug: false | |
| speed_perturb: false | |
| tfmel_args: | |
| do_preemph: true | |
| do_spec_aug: false | |
| eps: 1.0e-08 | |
| f_max: 7600 | |
| f_min: 20 | |
| hop_length: 160 | |
| n_fft: 512 | |
| n_mels: 72 | |
| norm_signal: true | |
| win_length: 400 | |
| enable_amp: true | |
| exp_dir: exp/redimnet2/expB6_v2/-LM | |
| gpus: | |
| - 2 | |
| - 3 | |
| grad_accum_steps: 8 | |
| log_batch_interval: 100 | |
| loss: CrossEntropyLoss | |
| loss_args: {} | |
| margin_scheduler: MarginScheduler | |
| margin_update: | |
| epoch_iter: 34125 | |
| final_margin: 0.3 | |
| fix_start_epoch: 0 | |
| increase_start_epoch: 0 | |
| increase_type: exp | |
| initial_margin: 0.3 | |
| update_margin: false | |
| model: ReDimNet2Wrap | |
| model_args: | |
| C: 64 | |
| F: 72 | |
| block_1d_type: conv+att | |
| block_2d_type: basic_resnet | |
| causal: none | |
| compress_tconvs: true | |
| emb_bn: false | |
| embed_dim: 192 | |
| feat_dim: 72 | |
| fm_weigthing_type: NC | |
| global_context_att: true | |
| group_divisor: 1 | |
| hop_length: 160 | |
| out_channels: 224 | |
| pooling_func: ASTP | |
| return_2d_output: true | |
| spec: null | |
| spec_in_channels: 1 | |
| stages_setup: | |
| - - - 1 | |
| - 1 | |
| - 3 | |
| - 3 | |
| - - - 3 | |
| - 3 | |
| - 64 | |
| - - - 2 | |
| - 1 | |
| - 4 | |
| - 2 | |
| - - - 3 | |
| - 3 | |
| - 64 | |
| - - - 1 | |
| - 2 | |
| - 5 | |
| - 2 | |
| - - - 3 | |
| - 3 | |
| - 48 | |
| - - - 2 | |
| - 1 | |
| - 5 | |
| - 1 | |
| - - - 3 | |
| - 3 | |
| - 48 | |
| - - - 1 | |
| - 2 | |
| - 4 | |
| - 0.75 | |
| - - - 3 | |
| - 3 | |
| - 32 | |
| - - - 2 | |
| - 1 | |
| - 3 | |
| - 0.5 | |
| - - - 3 | |
| - 3 | |
| - 24 | |
| model_init: null | |
| noise_data: data/voxceleb/musan/lmdb | |
| num_avg: 1 | |
| num_epochs: 5 | |
| optimizer: SGD | |
| optimizer_args: | |
| lr: 0.0001 | |
| momentum: 0.9 | |
| nesterov: true | |
| weight_decay: 2.0e-05 | |
| projection_args: | |
| do_lm: false | |
| easy_margin: false | |
| embed_dim: 192 | |
| lanbuda: 0.7 | |
| margin_type: C | |
| num_class: 5994 | |
| project_type: sphereface2 | |
| scale: 32.0 | |
| t: 3 | |
| reverb_data: data/voxceleb/rirs/lmdb | |
| save_epoch_interval: 1 | |
| scheduler: ExponentialDecrease | |
| scheduler_args: | |
| epoch_iter: 34125 | |
| final_lr: 1.0e-05 | |
| initial_lr: 0.0001 | |
| num_epochs: 5 | |
| scale_ratio: 4.0 | |
| warm_from_zero: false | |
| warm_up_epoch: 0 | |
| seed: 42 | |
| train_data: data/voxceleb/vox2_dev/shard.list | |
| train_label: data/voxceleb/vox2_dev/utt2spk | |