chengdongliang commited on
Commit
f5a2018
1 Parent(s): c674d0c

upload model.pt

Browse files
Files changed (2) hide show
  1. avg_model.pt +3 -0
  2. config.yaml +84 -0
avg_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e44852f375b6e22de812f64dc8dad17d2634ed44b90098b5da6c36e0a076cc30
3
+ size 35219659
config.yaml ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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/cnceleb_v2/resnet/wespeaker/renset34_new_margin
28
+ gpus:
29
+ - 0
30
+ - 1
31
+ - 2
32
+ - 3
33
+ - 4
34
+ - 5
35
+ - 6
36
+ - 7
37
+ log_batch_interval: 100
38
+ loss: CrossEntropyLoss
39
+ loss_args: {}
40
+ margin_scheduler: MarginScheduler
41
+ margin_update:
42
+ epoch_iter: 496
43
+ final_margin: 0.2
44
+ fix_start_epoch: 40
45
+ increase_start_epoch: 20
46
+ increase_type: exp
47
+ initial_margin: 0.0
48
+ update_margin: true
49
+ model: ResNet34
50
+ model_args:
51
+ embed_dim: 256
52
+ feat_dim: 80
53
+ pooling_func: TSTP
54
+ two_emb_layer: false
55
+ model_init: null
56
+ noise_data: data/cnceleb_wespeaker/musan/lmdb
57
+ num_avg: 10
58
+ num_epochs: 150
59
+ optimizer: SGD
60
+ optimizer_args:
61
+ lr: 0.1
62
+ momentum: 0.9
63
+ nesterov: true
64
+ weight_decay: 0.0001
65
+ projection_args:
66
+ easy_margin: false
67
+ embed_dim: 256
68
+ num_class: 8379
69
+ project_type: arc_margin
70
+ scale: 32.0
71
+ reverb_data: data/cnceleb_wespeaker/rirs/lmdb
72
+ save_epoch_interval: 5
73
+ scheduler: ExponentialDecrease
74
+ scheduler_args:
75
+ epoch_iter: 496
76
+ final_lr: 5.0e-05
77
+ initial_lr: 0.1
78
+ num_epochs: 150
79
+ scale_ratio: 16.0
80
+ warm_from_zero: true
81
+ warm_up_epoch: 6
82
+ seed: 42
83
+ train_data: data/cnceleb_wespeaker/cnceleb_train/shard.list
84
+ train_label: data/cnceleb_wespeaker/cnceleb_train/utt2spk