wsstriving commited on
Commit
4d9c929
1 Parent(s): 83ab792

Upload 2 files

Browse files

upload model and config file

Files changed (2) hide show
  1. avg_model.pt +3 -0
  2. config.yaml +78 -0
avg_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7b8447ea84345e26432461b4dda224d1af34a3985a438268e0dfbbde059b1ce
3
+ size 133998630
config.yaml ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data_type: shard
2
+ dataloader_args:
3
+ batch_size: 32
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/ResNet293-TSTP-emb256-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: 17062
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: ResNet293
44
+ model_args:
45
+ embed_dim: 256
46
+ feat_dim: 80
47
+ pooling_func: TSTP
48
+ two_emb_layer: false
49
+ model_init: null
50
+ noise_data: data/musan/lmdb
51
+ num_avg: 2
52
+ num_epochs: 150
53
+ optimizer: SGD
54
+ optimizer_args:
55
+ lr: 0.1
56
+ momentum: 0.9
57
+ nesterov: true
58
+ weight_decay: 0.0001
59
+ projection_args:
60
+ easy_margin: false
61
+ embed_dim: 256
62
+ num_class: 17982
63
+ project_type: arc_margin
64
+ scale: 32.0
65
+ reverb_data: data/rirs/lmdb
66
+ save_epoch_interval: 5
67
+ scheduler: ExponentialDecrease
68
+ scheduler_args:
69
+ epoch_iter: 17062
70
+ final_lr: 5.0e-05
71
+ initial_lr: 0.1
72
+ num_epochs: 150
73
+ scale_ratio: 1.0
74
+ warm_from_zero: true
75
+ warm_up_epoch: 6
76
+ seed: 42
77
+ train_data: data/vox2_dev/shard.list
78
+ train_label: data/vox2_dev/utt2spk