upload model.pt

avg_model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:07abeeb5150441995b51ea65c9ccc8feed78b33040012f1d2fad29a0e4f5b8d7
+oid sha256:3ced11d16c68ee4d1b9b8aea2494f56516d7cc07782d451fb437480ad00a6ac7
 size 66210861

config.yaml

@@ -1,6 +1,7 @@
+checkpoint: exp/CAMPPlus-TSTP-emb512-fbank80-num_frms200-aug0.6-spTrue-saFalse-ArcMargin-SGD-epoch150-LM/models/model_0.pt
 data_type: shard
 dataloader_args:
-  batch_size: 128
+  batch_size: 64
   drop_last: true
   num_workers: 16
   pin_memory: false
@@ -12,7 +13,7 @@ dataset_args:
     frame_length: 25
     frame_shift: 10
     num_mel_bins: 80
-  num_frms: 200
+  num_frms: 600
   resample_rate: 16000
   shuffle: true
   shuffle_args:
@@ -24,9 +25,10 @@ dataset_args:
     num_f_mask: 1
     num_t_mask: 1
     prob: 0.6
-  speed_perturb: true
+  speed_perturb: false
+do_lm: true
 enable_amp: false
-exp_dir: exp/CAMPPlus-TSTP-emb512-fbank80-num_frms200-aug0.6-spTrue-saFalse-ArcMargin-SGD-epoch150
+exp_dir: exp/CAMPPlus-TSTP-emb512-fbank80-num_frms200-aug0.6-spTrue-saFalse-ArcMargin-SGD-epoch150-LM
 gpus:
 - 0
 - 1
@@ -35,12 +37,12 @@ loss: CrossEntropyLoss
 loss_args: {}
 margin_scheduler: MarginScheduler
 margin_update:
-  epoch_iter: 4265
-  final_margin: 0.2
-  fix_start_epoch: 40
-  increase_start_epoch: 20
+  epoch_iter: 8531
+  final_margin: 0.5
+  fix_start_epoch: 1
+  increase_start_epoch: 1
   increase_type: exp
-  initial_margin: 0.0
+  initial_margin: 0.5
   update_margin: true
 model: CAMPPlus
 model_args:
@@ -49,32 +51,32 @@ model_args:
   pooling_func: TSTP
 model_init: null
 noise_data: data/musan/lmdb
-num_avg: 10
-num_epochs: 150
+num_avg: 1
+num_epochs: 5
 optimizer: SGD
 optimizer_args:
-  lr: 0.1
+  lr: 0.0001
   momentum: 0.9
   nesterov: true
   weight_decay: 0.0001
 projection_args:
-  do_lm: false
+  do_lm: true
   easy_margin: false
   embed_dim: 512
   num_class: 17982
   project_type: arc_margin
   scale: 32.0
 reverb_data: data/rirs/lmdb
-save_epoch_interval: 5
+save_epoch_interval: 1
 scheduler: ExponentialDecrease
 scheduler_args:
-  epoch_iter: 4265
-  final_lr: 5.0e-05
-  initial_lr: 0.1
-  num_epochs: 150
-  scale_ratio: 4.0
+  epoch_iter: 8531
+  final_lr: 2.5e-05
+  initial_lr: 0.0001
+  num_epochs: 5
+  scale_ratio: 2.0
   warm_from_zero: true
-  warm_up_epoch: 6
+  warm_up_epoch: 1
 seed: 42
 train_data: data/vox2_dev/shard.list
 train_label: data/vox2_dev/utt2spk