RMC-16 commited on
Commit
5ac94bd
·
1 Parent(s): 86efd91

Upload 14 files

Browse files
config.json ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/wav2vec2-base",
3
+ "activation_dropout": 0.0,
4
+ "adapter_kernel_size": 3,
5
+ "adapter_stride": 2,
6
+ "add_adapter": false,
7
+ "apply_spec_augment": true,
8
+ "architectures": [
9
+ "Wav2Vec2ForCTC"
10
+ ],
11
+ "attention_dropout": 0.1,
12
+ "bos_token_id": 1,
13
+ "classifier_proj_size": 256,
14
+ "codevector_dim": 256,
15
+ "contrastive_logits_temperature": 0.1,
16
+ "conv_bias": false,
17
+ "conv_dim": [
18
+ 512,
19
+ 512,
20
+ 512,
21
+ 512,
22
+ 512,
23
+ 512,
24
+ 512
25
+ ],
26
+ "conv_kernel": [
27
+ 10,
28
+ 3,
29
+ 3,
30
+ 3,
31
+ 3,
32
+ 2,
33
+ 2
34
+ ],
35
+ "conv_stride": [
36
+ 5,
37
+ 2,
38
+ 2,
39
+ 2,
40
+ 2,
41
+ 2,
42
+ 2
43
+ ],
44
+ "ctc_loss_reduction": "mean",
45
+ "ctc_zero_infinity": false,
46
+ "diversity_loss_weight": 0.1,
47
+ "do_stable_layer_norm": false,
48
+ "eos_token_id": 2,
49
+ "feat_extract_activation": "gelu",
50
+ "feat_extract_norm": "group",
51
+ "feat_proj_dropout": 0.1,
52
+ "feat_quantizer_dropout": 0.0,
53
+ "final_dropout": 0.0,
54
+ "freeze_feat_extract_train": true,
55
+ "hidden_act": "gelu",
56
+ "hidden_dropout": 0.1,
57
+ "hidden_size": 768,
58
+ "initializer_range": 0.02,
59
+ "intermediate_size": 3072,
60
+ "layer_norm_eps": 1e-05,
61
+ "layerdrop": 0.0,
62
+ "mask_channel_length": 10,
63
+ "mask_channel_min_space": 1,
64
+ "mask_channel_other": 0.0,
65
+ "mask_channel_prob": 0.0,
66
+ "mask_channel_selection": "static",
67
+ "mask_feature_length": 10,
68
+ "mask_feature_min_masks": 0,
69
+ "mask_feature_prob": 0.0,
70
+ "mask_time_length": 10,
71
+ "mask_time_min_masks": 2,
72
+ "mask_time_min_space": 1,
73
+ "mask_time_other": 0.0,
74
+ "mask_time_prob": 0.05,
75
+ "mask_time_selection": "static",
76
+ "model_type": "wav2vec2",
77
+ "no_mask_channel_overlap": false,
78
+ "no_mask_time_overlap": false,
79
+ "num_adapter_layers": 3,
80
+ "num_attention_heads": 12,
81
+ "num_codevector_groups": 2,
82
+ "num_codevectors_per_group": 320,
83
+ "num_conv_pos_embedding_groups": 16,
84
+ "num_conv_pos_embeddings": 128,
85
+ "num_feat_extract_layers": 7,
86
+ "num_hidden_layers": 12,
87
+ "num_negatives": 100,
88
+ "output_hidden_size": 768,
89
+ "pad_token_id": 29,
90
+ "proj_codevector_dim": 256,
91
+ "tdnn_dilation": [
92
+ 1,
93
+ 2,
94
+ 3,
95
+ 1,
96
+ 1
97
+ ],
98
+ "tdnn_dim": [
99
+ 512,
100
+ 512,
101
+ 512,
102
+ 512,
103
+ 1500
104
+ ],
105
+ "tdnn_kernel": [
106
+ 5,
107
+ 3,
108
+ 3,
109
+ 1,
110
+ 1
111
+ ],
112
+ "torch_dtype": "float32",
113
+ "transformers_version": "4.17.0",
114
+ "use_weighted_layer_sum": false,
115
+ "vocab_size": 32,
116
+ "xvector_output_dim": 512
117
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f14ce9ba167d8790eca86e334a189e6a8da7f57a9d69084e945a9a28a58cdb92
3
+ size 721686661
preprocessor_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_extractor_type": "Wav2Vec2FeatureExtractor",
4
+ "feature_size": 1,
5
+ "padding_side": "right",
6
+ "padding_value": 0.0,
7
+ "return_attention_mask": false,
8
+ "sampling_rate": 16000
9
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6dab79080da0dd1dd31dd057e5921c8692cf5e3e10681cb32bd1e637dbafe576
3
+ size 377673441
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d802c5027bae16abf2bb75f06d8afa24df58a944d6293ef1acd06cc4920bf0ca
3
+ size 14639
runs/Apr17_11-57-35_a91d1ecb82dc/1681732750.2470453/events.out.tfevents.1681732750.a91d1ecb82dc.827.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:784354f02939db777bb12ccd3707bebaa6ad2c003267e3b9e476aba17dc92e3c
3
+ size 4843
runs/Apr17_11-57-35_a91d1ecb82dc/events.out.tfevents.1681732750.a91d1ecb82dc.827.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:54b11256c87c46bae46e3ddc64f86269d64298cee291aa57c4338b85572876a0
3
+ size 6003
runs/Apr17_12-10-32_a91d1ecb82dc/1681733513.3772361/events.out.tfevents.1681733513.a91d1ecb82dc.827.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60fe7642cff4d61d7242f5d78184c2925b28c48e7079baa03a16c54062ba27be
3
+ size 4843
runs/Apr17_12-10-32_a91d1ecb82dc/events.out.tfevents.1681733513.a91d1ecb82dc.827.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fa53d4025d36b8a2cf002e84c2cb57e83c6523c7260a1381e09125004d8c1e6
3
+ size 5906
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aaa718bc2dd47d51a5896106079a3b980a9d366522eabc323069c7ac050ae4eb
3
+ size 557
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23d426220e98e578b8a011c0083dc54e8b291000cda6e8064f2e540761de0aa0
3
+ size 627
trainer_state.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 20.080321285140563,
5
+ "global_step": 5000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 20.08,
12
+ "learning_rate": 3.8207109737248844e-05,
13
+ "loss": 0.1145,
14
+ "step": 5000
15
+ },
16
+ {
17
+ "epoch": 20.08,
18
+ "eval_loss": 0.504927933216095,
19
+ "eval_runtime": 57.7307,
20
+ "eval_samples_per_second": 29.101,
21
+ "eval_steps_per_second": 3.638,
22
+ "eval_wer": 0.3442905382123906,
23
+ "step": 5000
24
+ }
25
+ ],
26
+ "max_steps": 7470,
27
+ "num_train_epochs": 30,
28
+ "total_flos": 2.062236601020465e+18,
29
+ "trial_name": null,
30
+ "trial_params": null
31
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be1a650e69d7e11101e98535c2c25e87aa72d510d8309853656f21c398d1a9a8
3
+ size 3067
vocab.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"h": 0, "i": 1, "y": 2, "v": 3, "z": 4, "t": 5, "x": 6, "d": 7, "r": 8, "n": 9, "'": 10, "f": 11, "p": 12, "l": 13, "k": 14, "o": 15, "e": 16, "b": 18, "w": 19, "s": 20, "j": 21, "g": 22, "u": 23, "c": 24, "m": 25, "q": 26, "a": 27, "|": 17, "[UNK]": 28, "[PAD]": 29}