chengdongliang commited on
Commit
7d2f308
·
verified ·
1 Parent(s): e98c17f

upload model.pt (#1)

Browse files

- upload model.pt (37fdef963ea41030b17ff0a2e23a8bf799de71c7)

Files changed (2) hide show
  1. config.yaml +86 -0
  2. model_5.pt +3 -0
config.yaml ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ checkpoint: exp/cnceleb_v2/resnet/wespeaker/renset34_new_margin-LM/models/model_0.pt
2
+ data_type: shard
3
+ dataloader_args:
4
+ batch_size: 64
5
+ drop_last: true
6
+ num_workers: 16
7
+ pin_memory: false
8
+ prefetch_factor: 8
9
+ dataset_args:
10
+ aug_prob: 0.6
11
+ fbank_args:
12
+ dither: 1.0
13
+ frame_length: 25
14
+ frame_shift: 10
15
+ num_mel_bins: 80
16
+ num_frms: 600
17
+ shuffle: true
18
+ shuffle_args:
19
+ shuffle_size: 2500
20
+ spec_aug: false
21
+ spec_aug_args:
22
+ max_f: 8
23
+ max_t: 10
24
+ num_f_mask: 1
25
+ num_t_mask: 1
26
+ prob: 0.6
27
+ speed_perturb: false
28
+ do_lm: true
29
+ exp_dir: exp/cnceleb_v2/resnet/wespeaker/renset34_new_margin-LM
30
+ gpus:
31
+ - 0
32
+ - 1
33
+ - 2
34
+ - 3
35
+ - 4
36
+ - 5
37
+ - 6
38
+ - 7
39
+ log_batch_interval: 100
40
+ loss: CrossEntropyLoss
41
+ loss_args: {}
42
+ margin_scheduler: MarginScheduler
43
+ margin_update:
44
+ epoch_iter: 992
45
+ final_margin: 0.5
46
+ fix_start_epoch: 1
47
+ increase_start_epoch: 1
48
+ increase_type: exp
49
+ initial_margin: 0.5
50
+ update_margin: true
51
+ model: ResNet34
52
+ model_args:
53
+ embed_dim: 256
54
+ feat_dim: 80
55
+ pooling_func: TSTP
56
+ two_emb_layer: false
57
+ model_init: null
58
+ noise_data: data/cnceleb_wespeaker/musan/lmdb
59
+ num_avg: 1
60
+ num_epochs: 5
61
+ optimizer: SGD
62
+ optimizer_args:
63
+ lr: 0.0001
64
+ momentum: 0.9
65
+ nesterov: true
66
+ weight_decay: 0.0001
67
+ projection_args:
68
+ easy_margin: false
69
+ embed_dim: 256
70
+ num_class: 8379
71
+ project_type: arc_margin
72
+ scale: 32.0
73
+ reverb_data: data/cnceleb_wespeaker/rirs/lmdb
74
+ save_epoch_interval: 1
75
+ scheduler: ExponentialDecrease
76
+ scheduler_args:
77
+ epoch_iter: 992
78
+ final_lr: 2.5e-05
79
+ initial_lr: 0.0001
80
+ num_epochs: 5
81
+ scale_ratio: 8.0
82
+ warm_from_zero: true
83
+ warm_up_epoch: 1
84
+ seed: 42
85
+ train_data: data/cnceleb_wespeaker/cnceleb_train/shard.list
86
+ train_label: data/cnceleb_wespeaker/cnceleb_train/utt2spk
model_5.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fed1fbf1bb1e772f794e54f7dd8c5635485a0ccce799ecb429a9232d22596f7
3
+ size 35222731