Upload 5 files

README.md

@@ -1,3 +1,3 @@
----
-license: mit
----
+---
+license: apache-2.0
+---

cnceleb_resnet34_LM.onnx

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

config.yaml

@@ -0,0 +1,97 @@
+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

model_5.pt

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

pytorch_model.bin

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