prachuryyaIITG commited on
Commit
337302e
·
verified ·
1 Parent(s): d41f398

Upload 6 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
config.json ADDED
@@ -0,0 +1,174 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "FacebookAI/xlm-roberta-large",
3
+ "architectures": [
4
+ "XLMRobertaForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 1024,
13
+ "id2label": {
14
+ "0": "B-AerospaceManufacturer",
15
+ "1": "B-AnatomicalStructure",
16
+ "2": "B-ArtWork",
17
+ "3": "B-Artist",
18
+ "4": "B-Athlete",
19
+ "5": "B-CarManufacturer",
20
+ "6": "B-Cleric",
21
+ "7": "B-Clothing",
22
+ "8": "B-Disease",
23
+ "9": "B-Drink",
24
+ "10": "B-Facility",
25
+ "11": "B-Food",
26
+ "12": "B-HumanSettlement",
27
+ "13": "B-MedicalProcedure",
28
+ "14": "B-Medication/Vaccine",
29
+ "15": "B-MusicalGRP",
30
+ "16": "B-MusicalWork",
31
+ "17": "B-ORG",
32
+ "18": "B-OtherLOC",
33
+ "19": "B-OtherPER",
34
+ "20": "B-OtherPROD",
35
+ "21": "B-Politician",
36
+ "22": "B-PrivateCORP",
37
+ "23": "B-PrivateCorp",
38
+ "24": "B-PublicCORP",
39
+ "25": "B-PublicCorp",
40
+ "26": "B-Scientist",
41
+ "27": "B-Software",
42
+ "28": "B-SportsGRP",
43
+ "29": "B-SportsManager",
44
+ "30": "B-Station",
45
+ "31": "B-Symptom",
46
+ "32": "B-Vehicle",
47
+ "33": "B-VisualWork",
48
+ "34": "B-WrittenWork",
49
+ "35": "I-AerospaceManufacturer",
50
+ "36": "I-AnatomicalStructure",
51
+ "37": "I-ArtWork",
52
+ "38": "I-Artist",
53
+ "39": "I-Athlete",
54
+ "40": "I-CarManufacturer",
55
+ "41": "I-Cleric",
56
+ "42": "I-Clothing",
57
+ "43": "I-Disease",
58
+ "44": "I-Drink",
59
+ "45": "I-Facility",
60
+ "46": "I-Food",
61
+ "47": "I-HumanSettlement",
62
+ "48": "I-MedicalProcedure",
63
+ "49": "I-Medication/Vaccine",
64
+ "50": "I-MusicalGRP",
65
+ "51": "I-MusicalWork",
66
+ "52": "I-ORG",
67
+ "53": "I-OtherLOC",
68
+ "54": "I-OtherPER",
69
+ "55": "I-OtherPROD",
70
+ "56": "I-Politician",
71
+ "57": "I-PrivateCORP",
72
+ "58": "I-PrivateCorp",
73
+ "59": "I-PublicCORP",
74
+ "60": "I-PublicCorp",
75
+ "61": "I-Scientist",
76
+ "62": "I-Software",
77
+ "63": "I-SportsGRP",
78
+ "64": "I-SportsManager",
79
+ "65": "I-Station",
80
+ "66": "I-Symptom",
81
+ "67": "I-Vehicle",
82
+ "68": "I-VisualWork",
83
+ "69": "I-WrittenWork",
84
+ "70": "O"
85
+ },
86
+ "initializer_range": 0.02,
87
+ "intermediate_size": 4096,
88
+ "label2id": {
89
+ "B-AerospaceManufacturer": 0,
90
+ "B-AnatomicalStructure": 1,
91
+ "B-ArtWork": 2,
92
+ "B-Artist": 3,
93
+ "B-Athlete": 4,
94
+ "B-CarManufacturer": 5,
95
+ "B-Cleric": 6,
96
+ "B-Clothing": 7,
97
+ "B-Disease": 8,
98
+ "B-Drink": 9,
99
+ "B-Facility": 10,
100
+ "B-Food": 11,
101
+ "B-HumanSettlement": 12,
102
+ "B-MedicalProcedure": 13,
103
+ "B-Medication/Vaccine": 14,
104
+ "B-MusicalGRP": 15,
105
+ "B-MusicalWork": 16,
106
+ "B-ORG": 17,
107
+ "B-OtherLOC": 18,
108
+ "B-OtherPER": 19,
109
+ "B-OtherPROD": 20,
110
+ "B-Politician": 21,
111
+ "B-PrivateCORP": 22,
112
+ "B-PrivateCorp": 23,
113
+ "B-PublicCORP": 24,
114
+ "B-PublicCorp": 25,
115
+ "B-Scientist": 26,
116
+ "B-Software": 27,
117
+ "B-SportsGRP": 28,
118
+ "B-SportsManager": 29,
119
+ "B-Station": 30,
120
+ "B-Symptom": 31,
121
+ "B-Vehicle": 32,
122
+ "B-VisualWork": 33,
123
+ "B-WrittenWork": 34,
124
+ "I-AerospaceManufacturer": 35,
125
+ "I-AnatomicalStructure": 36,
126
+ "I-ArtWork": 37,
127
+ "I-Artist": 38,
128
+ "I-Athlete": 39,
129
+ "I-CarManufacturer": 40,
130
+ "I-Cleric": 41,
131
+ "I-Clothing": 42,
132
+ "I-Disease": 43,
133
+ "I-Drink": 44,
134
+ "I-Facility": 45,
135
+ "I-Food": 46,
136
+ "I-HumanSettlement": 47,
137
+ "I-MedicalProcedure": 48,
138
+ "I-Medication/Vaccine": 49,
139
+ "I-MusicalGRP": 50,
140
+ "I-MusicalWork": 51,
141
+ "I-ORG": 52,
142
+ "I-OtherLOC": 53,
143
+ "I-OtherPER": 54,
144
+ "I-OtherPROD": 55,
145
+ "I-Politician": 56,
146
+ "I-PrivateCORP": 57,
147
+ "I-PrivateCorp": 58,
148
+ "I-PublicCORP": 59,
149
+ "I-PublicCorp": 60,
150
+ "I-Scientist": 61,
151
+ "I-Software": 62,
152
+ "I-SportsGRP": 63,
153
+ "I-SportsManager": 64,
154
+ "I-Station": 65,
155
+ "I-Symptom": 66,
156
+ "I-Vehicle": 67,
157
+ "I-VisualWork": 68,
158
+ "I-WrittenWork": 69,
159
+ "O": 70
160
+ },
161
+ "layer_norm_eps": 1e-05,
162
+ "max_position_embeddings": 514,
163
+ "model_type": "xlm-roberta",
164
+ "num_attention_heads": 16,
165
+ "num_hidden_layers": 24,
166
+ "output_past": true,
167
+ "pad_token_id": 1,
168
+ "position_embedding_type": "absolute",
169
+ "torch_dtype": "float32",
170
+ "transformers_version": "4.48.3",
171
+ "type_vocab_size": 1,
172
+ "use_cache": true,
173
+ "vocab_size": 250002
174
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7af5b43d3b9d3d561e38bc7eb070d52c681654e4505258d3bac20909453873dd
3
+ size 2235702964
special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "cls_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "mask_token": {
6
+ "content": "<mask>",
7
+ "lstrip": true,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "</s>",
14
+ "unk_token": "<unk>"
15
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8373f9cd3d27591e1924426bcc1c8799bc5a9affc4fc857982c5d66668dd1f41
3
+ size 17082832
tokenizer_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "250001": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "bos_token": "<s>",
46
+ "clean_up_tokenization_spaces": false,
47
+ "cls_token": "<s>",
48
+ "eos_token": "</s>",
49
+ "extra_special_tokens": {},
50
+ "mask_token": "<mask>",
51
+ "model_max_length": 512,
52
+ "pad_token": "<pad>",
53
+ "sep_token": "</s>",
54
+ "tokenizer_class": "XLMRobertaTokenizer",
55
+ "unk_token": "<unk>"
56
+ }
trainer_state.json ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.6493010279892045,
3
+ "best_model_checkpoint": "./models/as_TrA50_XLM/checkpoint-4986",
4
+ "epoch": 6.0,
5
+ "eval_steps": 500,
6
+ "global_step": 4986,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.9987966305655837,
13
+ "grad_norm": 4.042017936706543,
14
+ "learning_rate": 4.1676694745286805e-05,
15
+ "loss": 0.7179,
16
+ "step": 830
17
+ },
18
+ {
19
+ "epoch": 1.0,
20
+ "eval_f1": 0.5016971754152019,
21
+ "eval_loss": 0.3967933654785156,
22
+ "eval_precision": 0.48018796774380695,
23
+ "eval_recall": 0.525223681705692,
24
+ "eval_runtime": 36.5565,
25
+ "eval_samples_per_second": 303.339,
26
+ "eval_steps_per_second": 4.76,
27
+ "step": 831
28
+ },
29
+ {
30
+ "epoch": 1.9975932611311673,
31
+ "grad_norm": 3.9885966777801514,
32
+ "learning_rate": 3.335338949057361e-05,
33
+ "loss": 0.4331,
34
+ "step": 1660
35
+ },
36
+ {
37
+ "epoch": 2.0,
38
+ "eval_f1": 0.5287363073037502,
39
+ "eval_loss": 0.36940571665763855,
40
+ "eval_precision": 0.4921277061010278,
41
+ "eval_recall": 0.5712291389047528,
42
+ "eval_runtime": 36.6256,
43
+ "eval_samples_per_second": 302.766,
44
+ "eval_steps_per_second": 4.751,
45
+ "step": 1662
46
+ },
47
+ {
48
+ "epoch": 2.996389891696751,
49
+ "grad_norm": 3.235121250152588,
50
+ "learning_rate": 2.5030084235860413e-05,
51
+ "loss": 0.3391,
52
+ "step": 2490
53
+ },
54
+ {
55
+ "epoch": 3.0,
56
+ "eval_f1": 0.607596501529808,
57
+ "eval_loss": 0.289639949798584,
58
+ "eval_precision": 0.5922400289191468,
59
+ "eval_recall": 0.623770543816232,
60
+ "eval_runtime": 36.3105,
61
+ "eval_samples_per_second": 305.393,
62
+ "eval_steps_per_second": 4.792,
63
+ "step": 2493
64
+ },
65
+ {
66
+ "epoch": 3.9951865222623346,
67
+ "grad_norm": 5.095285415649414,
68
+ "learning_rate": 1.6706778981147215e-05,
69
+ "loss": 0.2637,
70
+ "step": 3320
71
+ },
72
+ {
73
+ "epoch": 4.0,
74
+ "eval_f1": 0.6188354585957905,
75
+ "eval_loss": 0.2840094566345215,
76
+ "eval_precision": 0.5999046824734898,
77
+ "eval_recall": 0.638999936544197,
78
+ "eval_runtime": 36.309,
79
+ "eval_samples_per_second": 305.407,
80
+ "eval_steps_per_second": 4.792,
81
+ "step": 3324
82
+ },
83
+ {
84
+ "epoch": 4.993983152827918,
85
+ "grad_norm": 3.0477702617645264,
86
+ "learning_rate": 8.383473726434016e-06,
87
+ "loss": 0.2006,
88
+ "step": 4150
89
+ },
90
+ {
91
+ "epoch": 5.0,
92
+ "eval_f1": 0.6398084151812781,
93
+ "eval_loss": 0.2787846028804779,
94
+ "eval_precision": 0.6125137848975565,
95
+ "eval_recall": 0.6696490894092265,
96
+ "eval_runtime": 36.4904,
97
+ "eval_samples_per_second": 303.888,
98
+ "eval_steps_per_second": 4.768,
99
+ "step": 4155
100
+ },
101
+ {
102
+ "epoch": 5.992779783393502,
103
+ "grad_norm": 3.478201150894165,
104
+ "learning_rate": 6.01684717208183e-08,
105
+ "loss": 0.1535,
106
+ "step": 4980
107
+ },
108
+ {
109
+ "epoch": 6.0,
110
+ "eval_f1": 0.6493010279892045,
111
+ "eval_loss": 0.27660489082336426,
112
+ "eval_precision": 0.6217911487977698,
113
+ "eval_recall": 0.6793578272733042,
114
+ "eval_runtime": 36.2487,
115
+ "eval_samples_per_second": 305.914,
116
+ "eval_steps_per_second": 4.8,
117
+ "step": 4986
118
+ }
119
+ ],
120
+ "logging_steps": 830,
121
+ "max_steps": 4986,
122
+ "num_input_tokens_seen": 0,
123
+ "num_train_epochs": 6,
124
+ "save_steps": 500,
125
+ "stateful_callbacks": {
126
+ "TrainerControl": {
127
+ "args": {
128
+ "should_epoch_stop": false,
129
+ "should_evaluate": false,
130
+ "should_log": false,
131
+ "should_save": true,
132
+ "should_training_stop": true
133
+ },
134
+ "attributes": {}
135
+ }
136
+ },
137
+ "total_flos": 4.58906727159972e+16,
138
+ "train_batch_size": 64,
139
+ "trial_name": null,
140
+ "trial_params": null
141
+ }