Upload 14 files

Ecapa_tdnn512_USphereface2_Inter_intra/config.yaml

@@ -0,0 +1,93 @@
+alpha_schedule:
+  alpha_final: 0.3
+  alpha_warmup_end: 140
+  alpha_warmup_start: 60
+data_type: shard
+dataloader_args:
+  batch_size: 512
+  drop_last: true
+  num_workers: 12
+  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
+  filter: true
+  filter_args:
+    max_num_frames: 800
+    min_num_frames: 100
+  frontend: fbank
+  num_frms: 200
+  resample_rate: 16000
+  sample_num_per_epoch: 0
+  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: true
+enable_amp: false
+exp_dir: exp/u_cube_plus/ecapa_u_cube_SphereFace2_ArcGuide_v3
+gpus:
+- 2
+- 3
+log_batch_interval: 100
+loss: CrossEntropyLoss
+loss_args: {}
+margin_scheduler: MarginScheduler
+margin_update:
+  epoch_iter: 1066
+  final_margin: 0.2
+  fix_start_epoch: 40
+  increase_start_epoch: 20
+  increase_type: exp
+  initial_margin: 0.0
+  update_margin: true
+model: ECAPA_TDNN_GLOB_c512
+model_args:
+  embed_dim: 192
+  feat_dim: 80
+  pooling_func: U_Cube_XI
+model_init: null
+noise_data: data/musan/lmdb
+num_avg: 10
+num_epochs: 150
+optimizer: SGD
+optimizer_args:
+  lr: 0.1
+  momentum: 0.9
+  nesterov: true
+  weight_decay: 0.0001
+projection_args:
+  do_lm: false
+  easy_margin: false
+  embed_dim: 192
+  lanbuda: 0.7
+  margin_type: C
+  num_class: 17982
+  project_type: sphereface2_uncertainty_arcguide
+  scale: 32
+  t: 3
+reverb_data: data/rirs/lmdb
+save_epoch_interval: 10
+scheduler: ExponentialDecrease
+scheduler_args:
+  epoch_iter: 1066
+  final_lr: 5.0e-05
+  initial_lr: 0.1
+  num_epochs: 150
+  scale_ratio: 16.0
+  warm_from_zero: true
+  warm_up_epoch: 6
+seed: 42
+train_data: data/vox2_dev/shard.list
+train_label: data/vox2_dev/utt2spk

Ecapa_tdnn512_USphereface2_Inter_intra/models/avg_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffd4697ddc12b28e325b311e1417bc27c8fbb47934e14e2905eb4854f4053bca
+size 40689419

Ecapa_tdnn_512_UAAMSoftmax_Inter_intra/config.yaml

@@ -0,0 +1,87 @@
+checkpoint: exp/u_cube_plus/ecapa_u_cube_inter_intra/models/model_140.pt
+data_type: shard
+dataloader_args:
+  batch_size: 512
+  drop_last: true
+  num_workers: 12
+  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
+  filter: true
+  filter_args:
+    max_num_frames: 800
+    min_num_frames: 100
+  frontend: fbank
+  num_frms: 200
+  resample_rate: 16000
+  sample_num_per_epoch: 0
+  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: true
+enable_amp: false
+exp_dir: exp/u_cube_plus/ecapa_u_cube_inter_intra
+gpus:
+- 2
+- 3
+log_batch_interval: 100
+loss: CrossEntropyLoss
+loss_args: {}
+margin_scheduler: MarginScheduler
+margin_update:
+  epoch_iter: 1066
+  final_margin: 0.2
+  fix_start_epoch: 40
+  increase_start_epoch: 20
+  increase_type: exp
+  initial_margin: 0.0
+  update_margin: true
+model: ECAPA_TDNN_GLOB_c512
+model_args:
+  embed_dim: 192
+  feat_dim: 80
+  pooling_func: U_Cube_XI
+model_init: null
+noise_data: data/musan/lmdb
+num_avg: 10
+num_epochs: 150
+optimizer: SGD
+optimizer_args:
+  lr: 0.1
+  momentum: 0.9
+  nesterov: true
+  weight_decay: 0.0001
+projection_args:
+  do_lm: false
+  easy_margin: false
+  embed_dim: 192
+  num_class: 17982
+  project_type: arc_margin_uncertainty_inter_intra
+  scale: 32
+reverb_data: data/rirs/lmdb
+save_epoch_interval: 10
+scheduler: ExponentialDecrease
+scheduler_args:
+  epoch_iter: 1066
+  final_lr: 5.0e-05
+  initial_lr: 0.1
+  num_epochs: 150
+  scale_ratio: 16.0
+  warm_from_zero: true
+  warm_up_epoch: 6
+seed: 42
+train_data: data/vox2_dev/shard.list
+train_label: data/vox2_dev/utt2spk

Ecapa_tdnn_512_UAAMSoftmax_Inter_intra/models/avg_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:922d93a63059dd4889bd6a1da14ecb0b0b97303a6f7d2dd1a391a8c2fd8b38bd
+size 40689163

Ecapa_tdnn_512_UAMSoftmax_Inter_intra/config.yaml

@@ -0,0 +1,87 @@
+checkpoint: exp/u_cube_plus/ecapa_u_cube_AM_inter_intra/models/model_149.pt
+data_type: shard
+dataloader_args:
+  batch_size: 512
+  drop_last: true
+  num_workers: 12
+  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
+  filter: true
+  filter_args:
+    max_num_frames: 800
+    min_num_frames: 100
+  frontend: fbank
+  num_frms: 200
+  resample_rate: 16000
+  sample_num_per_epoch: 0
+  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: true
+enable_amp: false
+exp_dir: exp/u_cube_plus/ecapa_u_cube_AM_inter_intra
+gpus:
+- 2
+- 3
+log_batch_interval: 100
+loss: CrossEntropyLoss
+loss_args: {}
+margin_scheduler: MarginScheduler
+margin_update:
+  epoch_iter: 1066
+  final_margin: 0.2
+  fix_start_epoch: 40
+  increase_start_epoch: 20
+  increase_type: exp
+  initial_margin: 0.0
+  update_margin: true
+model: ECAPA_TDNN_GLOB_c512
+model_args:
+  embed_dim: 192
+  feat_dim: 80
+  pooling_func: U_Cube_XI
+model_init: null
+noise_data: data/musan/lmdb
+num_avg: 10
+num_epochs: 150
+optimizer: SGD
+optimizer_args:
+  lr: 0.1
+  momentum: 0.9
+  nesterov: true
+  weight_decay: 0.0001
+projection_args:
+  do_lm: false
+  easy_margin: false
+  embed_dim: 192
+  num_class: 17982
+  project_type: add_margin_uncertainty_inter_intra
+  scale: 32
+reverb_data: data/rirs/lmdb
+save_epoch_interval: 10
+scheduler: ExponentialDecrease
+scheduler_args:
+  epoch_iter: 1066
+  final_lr: 5.0e-05
+  initial_lr: 0.1
+  num_epochs: 150
+  scale_ratio: 16.0
+  warm_from_zero: true
+  warm_up_epoch: 6
+seed: 42
+train_data: data/vox2_dev/shard.list
+train_label: data/vox2_dev/utt2spk

Ecapa_tdnn_512_UAMSoftmax_Inter_intra/models/avg_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c81468917d59141394a0831f5876b68492e952546791006bb1e4198e380c9fb
+size 40689163

RedimNetB2_UAAMSoftmax_Inter_intra/config.yaml

@@ -0,0 +1,86 @@
+data_type: shard
+dataloader_args:
+  batch_size: 128
+  drop_last: true
+  num_workers: 12
+  pin_memory: false
+  prefetch_factor: 4
+dataset_args:
+  aug_prob: 0.6
+  fbank_args:
+    dither: 1.0
+    frame_length: 25
+    frame_shift: 10
+    num_mel_bins: 72
+  filter: true
+  filter_args:
+    max_num_frames: 800
+    min_num_frames: 100
+  num_frms: 200
+  resample_rate: 16000
+  sample_num_per_epoch: 0
+  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: true
+enable_amp: false
+exp_dir: exp/redimnet_u_cube_UAAM_inter_intra
+gpus:
+- 2
+- 3
+log_batch_interval: 100
+loss: CrossEntropyLoss
+loss_args: {}
+margin_scheduler: MarginScheduler
+margin_update:
+  epoch_iter: 4265
+  final_margin: 0.2
+  fix_start_epoch: 40
+  increase_start_epoch: 20
+  increase_type: exp
+  initial_margin: 0.0
+  update_margin: true
+model: ReDimNetB2
+model_args:
+  embed_dim: 192
+  feat_dim: 72
+  pooling_func: U_Cube_XI
+  two_emb_layer: false
+model_init: null
+noise_data: data/musan/lmdb
+num_avg: 10
+num_epochs: 120
+optimizer: SGD
+optimizer_args:
+  lr: 0.1
+  momentum: 0.9
+  nesterov: true
+  weight_decay: 2.0e-05
+projection_args:
+  do_lm: false
+  easy_margin: false
+  embed_dim: 192
+  num_class: 17982
+  project_type: arc_margin_uncertainty_inter_intra
+  scale: 32.0
+reverb_data: data/rirs/lmdb
+save_epoch_interval: 5
+scheduler: ExponentialDecrease
+scheduler_args:
+  epoch_iter: 4265
+  final_lr: 5.0e-05
+  initial_lr: 0.1
+  num_epochs: 120
+  scale_ratio: 4.0
+  warm_from_zero: true
+  warm_up_epoch: 6
+seed: 42
+train_data: data/vox2_dev/shard.list
+train_label: data/vox2_dev/utt2spk

RedimNetB2_UAAMSoftmax_Inter_intra/models/avg_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7107d1ea2bfd2308e3d5f9b46983ea381a568e5008c65f19787cf921116aa1c7
+size 35893643

RedimNetB2_USphereface2_Inter_intra/config.yaml

@@ -0,0 +1,86 @@
+data_type: shard
+dataloader_args:
+  batch_size: 128
+  drop_last: true
+  num_workers: 12
+  pin_memory: false
+  prefetch_factor: 4
+dataset_args:
+  aug_prob: 0.6
+  fbank_args:
+    dither: 1.0
+    frame_length: 25
+    frame_shift: 10
+    num_mel_bins: 72
+  filter: true
+  filter_args:
+    max_num_frames: 800
+    min_num_frames: 100
+  num_frms: 200
+  resample_rate: 16000
+  sample_num_per_epoch: 0
+  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: true
+enable_amp: false
+exp_dir: exp/redimnet_u_cube_sphereface2_uncertainty_arcguide
+gpus:
+- 2
+- 3
+log_batch_interval: 100
+loss: CrossEntropyLoss
+loss_args: {}
+margin_scheduler: MarginScheduler
+margin_update:
+  epoch_iter: 4265
+  final_margin: 0.2
+  fix_start_epoch: 40
+  increase_start_epoch: 20
+  increase_type: exp
+  initial_margin: 0.0
+  update_margin: true
+model: ReDimNetB2
+model_args:
+  embed_dim: 192
+  feat_dim: 72
+  pooling_func: U_Cube_XI
+  two_emb_layer: false
+model_init: null
+noise_data: data/musan/lmdb
+num_avg: 10
+num_epochs: 120
+optimizer: SGD
+optimizer_args:
+  lr: 0.1
+  momentum: 0.9
+  nesterov: true
+  weight_decay: 2.0e-05
+projection_args:
+  do_lm: false
+  easy_margin: false
+  embed_dim: 192
+  num_class: 17982
+  project_type: sphereface2_uncertainty_arcguide
+  scale: 32.0
+reverb_data: data/rirs/lmdb
+save_epoch_interval: 5
+scheduler: ExponentialDecrease
+scheduler_args:
+  epoch_iter: 4265
+  final_lr: 5.0e-05
+  initial_lr: 0.1
+  num_epochs: 120
+  scale_ratio: 4.0
+  warm_from_zero: true
+  warm_up_epoch: 6
+seed: 42
+train_data: data/vox2_dev/shard.list
+train_label: data/vox2_dev/utt2spk

RedimNetB2_USphereface2_Inter_intra/models/avg_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:26cb1f7afbf8190e992c4ef0e20c050a2f6705e1d24da941b9fa7250c25274f6
+size 35893899

ResNet34_UAAMSoftmax_Inter_intra/config.yaml

@@ -0,0 +1,86 @@
+data_type: shard
+dataloader_args:
+  batch_size: 256
+  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
+  filter: true
+  filter_args:
+    max_num_frames: 800
+    min_num_frames: 100
+  num_frms: 200
+  resample_rate: 16000
+  sample_num_per_epoch: 0
+  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: true
+enable_amp: false
+exp_dir: exp/ResNet34_UAAM_inter_intra
+gpus:
+- 0
+- 1
+log_batch_interval: 100
+loss: CrossEntropyLoss
+loss_args: {}
+margin_scheduler: MarginScheduler
+margin_update:
+  epoch_iter: 2132
+  final_margin: 0.2
+  fix_start_epoch: 40
+  increase_start_epoch: 20
+  increase_type: exp
+  initial_margin: 0.0
+  update_margin: true
+model: ResNet34
+model_args:
+  embed_dim: 256
+  feat_dim: 80
+  pooling_func: U_Cube_XI
+  two_emb_layer: false
+model_init: null
+noise_data: data/musan/lmdb
+num_avg: 10
+num_epochs: 150
+optimizer: SGD
+optimizer_args:
+  lr: 0.1
+  momentum: 0.9
+  nesterov: true
+  weight_decay: 0.0001
+projection_args:
+  do_lm: false
+  easy_margin: false
+  embed_dim: 256
+  num_class: 17982
+  project_type: arc_margin_uncertainty_inter_intra
+  scale: 32.0
+reverb_data: data/rirs/lmdb
+save_epoch_interval: 5
+scheduler: ExponentialDecrease
+scheduler_args:
+  epoch_iter: 2132
+  final_lr: 5.0e-05
+  initial_lr: 0.1
+  num_epochs: 150
+  scale_ratio: 8.0
+  warm_from_zero: true
+  warm_up_epoch: 6
+seed: 42
+train_data: data/vox2_dev/shard.list
+train_label: data/vox2_dev/utt2spk

ResNet34_UAAMSoftmax_Inter_intra/models/avg_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:748252ac46c828bff680081187d5e55fb34618c0b26f6939dd0bfa0cc09dfb16
+size 50879179