| model: |
| _target_: pyannote.audio.models.embedding.WeSpeakerResNet34 |
| sample_rate: 16000 |
| num_channels: 1 |
| num_mel_bins: 80 |
| frame_length: 25 |
| frame_shift: 10 |
| dither: 0.0 |
| window_type: hamming |
| use_energy: false |
|
|
| checkpoint: exp/cnceleb_v2/resnet/wespeaker/renset34_new_margin-LM/models/model_0.pt |
| data_type: shard |
| dataloader_args: |
| batch_size: 64 |
| drop_last: true |
| num_workers: 16 |
| pin_memory: false |
| prefetch_factor: 8 |
| dataset_args: |
| aug_prob: 0.6 |
| fbank_args: |
| dither: 1.0 |
| frame_length: 25 |
| frame_shift: 10 |
| num_mel_bins: 80 |
| num_frms: 600 |
| shuffle: true |
| shuffle_args: |
| shuffle_size: 2500 |
| spec_aug: false |
| spec_aug_args: |
| max_f: 8 |
| max_t: 10 |
| num_f_mask: 1 |
| num_t_mask: 1 |
| prob: 0.6 |
| speed_perturb: false |
| do_lm: true |
| exp_dir: exp/cnceleb_v2/resnet/wespeaker/renset34_new_margin-LM |
| gpus: |
| - 0 |
| - 1 |
| - 2 |
| - 3 |
| - 4 |
| - 5 |
| - 6 |
| - 7 |
| log_batch_interval: 100 |
| loss: CrossEntropyLoss |
| loss_args: {} |
| margin_scheduler: MarginScheduler |
| margin_update: |
| epoch_iter: 992 |
| final_margin: 0.5 |
| fix_start_epoch: 1 |
| increase_start_epoch: 1 |
| increase_type: exp |
| initial_margin: 0.5 |
| update_margin: true |
| model: ResNet34 |
| model_args: |
| embed_dim: 256 |
| feat_dim: 80 |
| pooling_func: TSTP |
| two_emb_layer: false |
| model_init: null |
| noise_data: data/cnceleb_wespeaker/musan/lmdb |
| num_avg: 1 |
| num_epochs: 5 |
| optimizer: SGD |
| optimizer_args: |
| lr: 0.0001 |
| momentum: 0.9 |
| nesterov: true |
| weight_decay: 0.0001 |
| projection_args: |
| easy_margin: false |
| embed_dim: 256 |
| num_class: 8379 |
| project_type: arc_margin |
| scale: 32.0 |
| reverb_data: data/cnceleb_wespeaker/rirs/lmdb |
| save_epoch_interval: 1 |
| scheduler: ExponentialDecrease |
| scheduler_args: |
| epoch_iter: 992 |
| final_lr: 2.5e-05 |
| initial_lr: 0.0001 |
| num_epochs: 5 |
| scale_ratio: 8.0 |
| warm_from_zero: true |
| warm_up_epoch: 1 |
| seed: 42 |
| train_data: data/cnceleb_wespeaker/cnceleb_train/shard.list |
| train_label: data/cnceleb_wespeaker/cnceleb_train/utt2spk |
|
|
