wsstriving commited on
Commit
8a0abaf
1 Parent(s): ff7d15c

Upload 2 files

Browse files

add model files

Files changed (2) hide show
  1. avg_model.pt +3 -0
  2. config.yaml +77 -0
avg_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4c3cdaae5aba872ea888c904f5f4fd9fa41651ec072b1ebf40d611eb93e8d3e
3
+ size 38708805
config.yaml ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data_type: shard
2
+ dataloader_args:
3
+ batch_size: 128
4
+ drop_last: true
5
+ num_workers: 16
6
+ pin_memory: false
7
+ prefetch_factor: 8
8
+ dataset_args:
9
+ aug_prob: 0.6
10
+ fbank_args:
11
+ dither: 1.0
12
+ frame_length: 25
13
+ frame_shift: 10
14
+ num_mel_bins: 80
15
+ num_frms: 200
16
+ shuffle: true
17
+ shuffle_args:
18
+ shuffle_size: 2500
19
+ spec_aug: false
20
+ spec_aug_args:
21
+ max_f: 8
22
+ max_t: 10
23
+ num_f_mask: 1
24
+ num_t_mask: 1
25
+ prob: 0.6
26
+ speed_perturb: true
27
+ exp_dir: exp/ECAPA_TDNN_GLOB_c512-ASTP-emb192-fbank80-num_frms200-aug0.6-spTrue-saFalse-ArcMargin-SGD-epoch150
28
+ gpus:
29
+ - 0
30
+ - 1
31
+ log_batch_interval: 100
32
+ loss: CrossEntropyLoss
33
+ loss_args: {}
34
+ margin_scheduler: MarginScheduler
35
+ margin_update:
36
+ epoch_iter: 4265
37
+ final_margin: 0.2
38
+ fix_start_epoch: 40
39
+ increase_start_epoch: 20
40
+ increase_type: exp
41
+ initial_margin: 0.0
42
+ update_margin: true
43
+ model: ECAPA_TDNN_GLOB_c512
44
+ model_args:
45
+ embed_dim: 192
46
+ feat_dim: 80
47
+ pooling_func: ASTP
48
+ model_init: null
49
+ noise_data: data/musan/lmdb
50
+ num_avg: 10
51
+ num_epochs: 150
52
+ optimizer: SGD
53
+ optimizer_args:
54
+ lr: 0.1
55
+ momentum: 0.9
56
+ nesterov: true
57
+ weight_decay: 0.0001
58
+ projection_args:
59
+ easy_margin: false
60
+ embed_dim: 192
61
+ num_class: 17982
62
+ project_type: arc_margin
63
+ scale: 32.0
64
+ reverb_data: data/rirs/lmdb
65
+ save_epoch_interval: 5
66
+ scheduler: ExponentialDecrease
67
+ scheduler_args:
68
+ epoch_iter: 4265
69
+ final_lr: 5.0e-05
70
+ initial_lr: 0.1
71
+ num_epochs: 150
72
+ scale_ratio: 4.0
73
+ warm_from_zero: true
74
+ warm_up_epoch: 6
75
+ seed: 42
76
+ train_data: data/vox2_dev/shard.list
77
+ train_label: data/vox2_dev/utt2spk