xshubhamx commited on
Commit
396490f
1 Parent(s): 8116e6c

Upload folder using huggingface_hub

Browse files
training_checkpoints/checkpoint-3215/config.json ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "xlnet/xlnet-base-cased",
3
+ "architectures": [
4
+ "XLNetForSequenceClassification"
5
+ ],
6
+ "attn_type": "bi",
7
+ "bi_data": false,
8
+ "bos_token_id": 1,
9
+ "clamp_len": -1,
10
+ "d_head": 64,
11
+ "d_inner": 3072,
12
+ "d_model": 768,
13
+ "dropout": 0.1,
14
+ "end_n_top": 5,
15
+ "eos_token_id": 2,
16
+ "ff_activation": "gelu",
17
+ "id2label": {
18
+ "0": "Issue",
19
+ "1": "Court Discourse",
20
+ "2": "Conclusion",
21
+ "3": "Precedent Analysis",
22
+ "4": "Section Analysis",
23
+ "5": "Argument by Petitioner",
24
+ "6": "Fact",
25
+ "7": "Argument by Respondent",
26
+ "8": "Ratio",
27
+ "9": "Appellant",
28
+ "10": "Respondent",
29
+ "11": "Argument by Appellant",
30
+ "12": "Petitioner",
31
+ "13": "Judge",
32
+ "14": "Argument by Defendant"
33
+ },
34
+ "initializer_range": 0.02,
35
+ "label2id": {
36
+ "Appellant": 9,
37
+ "Argument by Appellant": 11,
38
+ "Argument by Defendant": 14,
39
+ "Argument by Petitioner": 5,
40
+ "Argument by Respondent": 7,
41
+ "Conclusion": 2,
42
+ "Court Discourse": 1,
43
+ "Fact": 6,
44
+ "Issue": 0,
45
+ "Judge": 13,
46
+ "Petitioner": 12,
47
+ "Precedent Analysis": 3,
48
+ "Ratio": 8,
49
+ "Respondent": 10,
50
+ "Section Analysis": 4
51
+ },
52
+ "layer_norm_eps": 1e-12,
53
+ "mem_len": null,
54
+ "model_type": "xlnet",
55
+ "n_head": 12,
56
+ "n_layer": 12,
57
+ "pad_token_id": 5,
58
+ "problem_type": "single_label_classification",
59
+ "reuse_len": null,
60
+ "same_length": false,
61
+ "start_n_top": 5,
62
+ "summary_activation": "tanh",
63
+ "summary_last_dropout": 0.1,
64
+ "summary_type": "last",
65
+ "summary_use_proj": true,
66
+ "task_specific_params": {
67
+ "text-generation": {
68
+ "do_sample": true,
69
+ "max_length": 250
70
+ }
71
+ },
72
+ "torch_dtype": "float32",
73
+ "transformers_version": "4.35.2",
74
+ "untie_r": true,
75
+ "use_mems_eval": true,
76
+ "use_mems_train": false,
77
+ "vocab_size": 32000
78
+ }
training_checkpoints/checkpoint-3215/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b7c066838885d9896d751b2ec2eba3ba68f036d286d1f1b86b5bf941be0bc4b
3
+ size 469304588
training_checkpoints/checkpoint-3215/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c7f836336d455e001744f54aa88162effcdfd0361ddc54748fdf79fd901dfa0
3
+ size 938733178
training_checkpoints/checkpoint-3215/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f23bf911970adc5a8f5d56125ebef106ab1b497d6bcad5862a322d0af88b9a2
3
+ size 14244
training_checkpoints/checkpoint-3215/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6ad414b8df0b8096a839a3f77f00939aed1f42c82150b9b022b259d098d77cf
3
+ size 1064
training_checkpoints/checkpoint-3215/special_tokens_map.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<eop>",
4
+ "<eod>"
5
+ ],
6
+ "bos_token": "<s>",
7
+ "cls_token": "<cls>",
8
+ "eos_token": "</s>",
9
+ "mask_token": {
10
+ "content": "<mask>",
11
+ "lstrip": true,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "<pad>",
17
+ "sep_token": "<sep>",
18
+ "unk_token": "<unk>"
19
+ }
training_checkpoints/checkpoint-3215/spiece.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f8c1c0bc2854d1af911a8550288c1258af5ba50277f3a5c829b98eb86fc5646
3
+ size 798011
training_checkpoints/checkpoint-3215/tokenizer_config.json ADDED
@@ -0,0 +1,94 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<unk>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<s>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<cls>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "<sep>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "5": {
44
+ "content": "<pad>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "6": {
52
+ "content": "<mask>",
53
+ "lstrip": true,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "7": {
60
+ "content": "<eod>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "8": {
68
+ "content": "<eop>",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ }
75
+ },
76
+ "additional_special_tokens": [
77
+ "<eop>",
78
+ "<eod>"
79
+ ],
80
+ "bos_token": "<s>",
81
+ "clean_up_tokenization_spaces": true,
82
+ "cls_token": "<cls>",
83
+ "do_lower_case": false,
84
+ "eos_token": "</s>",
85
+ "keep_accents": false,
86
+ "mask_token": "<mask>",
87
+ "model_max_length": 1000000000000000019884624838656,
88
+ "pad_token": "<pad>",
89
+ "remove_space": true,
90
+ "sep_token": "<sep>",
91
+ "sp_model_kwargs": {},
92
+ "tokenizer_class": "XLNetTokenizer",
93
+ "unk_token": "<unk>"
94
+ }
training_checkpoints/checkpoint-3215/trainer_state.json ADDED
@@ -0,0 +1,165 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.722931923188587,
3
+ "best_model_checkpoint": "xlnet-base-cased/checkpoint-1929",
4
+ "epoch": 5.0,
5
+ "eval_steps": 500,
6
+ "global_step": 3215,
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.78,
13
+ "learning_rate": 4.871695178849145e-05,
14
+ "loss": 1.2613,
15
+ "step": 500
16
+ },
17
+ {
18
+ "epoch": 1.0,
19
+ "eval_accuracy": 0.7676219984508134,
20
+ "eval_f1_macro": 0.4818756771742453,
21
+ "eval_f1_micro": 0.7676219984508135,
22
+ "eval_f1_weighted": 0.7524385212708887,
23
+ "eval_loss": 0.775787353515625,
24
+ "eval_macro_fpr": 0.022019799606126652,
25
+ "eval_macro_sensitivity": 0.5128622254580788,
26
+ "eval_macro_specificity": 0.9823717840367622,
27
+ "eval_precision": 0.7672637518290021,
28
+ "eval_precision_macro": 0.5269420024270713,
29
+ "eval_recall": 0.7676219984508134,
30
+ "eval_recall_macro": 0.5128622254580788,
31
+ "eval_runtime": 65.0712,
32
+ "eval_samples_per_second": 19.84,
33
+ "eval_steps_per_second": 2.49,
34
+ "eval_weighted_fpr": 0.021165514321998025,
35
+ "eval_weighted_sensitivity": 0.7676219984508134,
36
+ "eval_weighted_specificity": 0.9679547621006213,
37
+ "step": 643
38
+ },
39
+ {
40
+ "epoch": 1.56,
41
+ "learning_rate": 4.7420943494038366e-05,
42
+ "loss": 0.7364,
43
+ "step": 1000
44
+ },
45
+ {
46
+ "epoch": 2.0,
47
+ "eval_accuracy": 0.8071262587141751,
48
+ "eval_f1_macro": 0.701878478303997,
49
+ "eval_f1_micro": 0.8071262587141751,
50
+ "eval_f1_weighted": 0.8012788916547595,
51
+ "eval_loss": 0.6754755973815918,
52
+ "eval_macro_fpr": 0.01739858713969909,
53
+ "eval_macro_sensitivity": 0.6971761972643127,
54
+ "eval_macro_specificity": 0.9854830758512745,
55
+ "eval_precision": 0.8087721671110009,
56
+ "eval_precision_macro": 0.7425135911700289,
57
+ "eval_recall": 0.8071262587141751,
58
+ "eval_recall_macro": 0.6971761972643127,
59
+ "eval_runtime": 86.774,
60
+ "eval_samples_per_second": 14.878,
61
+ "eval_steps_per_second": 1.867,
62
+ "eval_weighted_fpr": 0.016782368403316036,
63
+ "eval_weighted_sensitivity": 0.8071262587141751,
64
+ "eval_weighted_specificity": 0.9751198790549438,
65
+ "step": 1286
66
+ },
67
+ {
68
+ "epoch": 2.33,
69
+ "learning_rate": 4.612493519958528e-05,
70
+ "loss": 0.6021,
71
+ "step": 1500
72
+ },
73
+ {
74
+ "epoch": 3.0,
75
+ "eval_accuracy": 0.8063516653756778,
76
+ "eval_f1_macro": 0.722931923188587,
77
+ "eval_f1_micro": 0.8063516653756778,
78
+ "eval_f1_weighted": 0.8013697737227732,
79
+ "eval_loss": 0.8442708253860474,
80
+ "eval_macro_fpr": 0.01757802934678216,
81
+ "eval_macro_sensitivity": 0.7262014539622089,
82
+ "eval_macro_specificity": 0.9852110484486459,
83
+ "eval_precision": 0.8016312862866737,
84
+ "eval_precision_macro": 0.7269969408457856,
85
+ "eval_recall": 0.8063516653756778,
86
+ "eval_recall_macro": 0.7262014539622089,
87
+ "eval_runtime": 89.8454,
88
+ "eval_samples_per_second": 14.369,
89
+ "eval_steps_per_second": 1.803,
90
+ "eval_weighted_fpr": 0.01686454398273071,
91
+ "eval_weighted_sensitivity": 0.8063516653756778,
92
+ "eval_weighted_specificity": 0.9718140613540133,
93
+ "step": 1929
94
+ },
95
+ {
96
+ "epoch": 3.11,
97
+ "learning_rate": 4.4828926905132196e-05,
98
+ "loss": 0.5435,
99
+ "step": 2000
100
+ },
101
+ {
102
+ "epoch": 3.89,
103
+ "learning_rate": 4.353551062726802e-05,
104
+ "loss": 0.4361,
105
+ "step": 2500
106
+ },
107
+ {
108
+ "epoch": 4.0,
109
+ "eval_accuracy": 0.8001549186676995,
110
+ "eval_f1_macro": 0.7051095861561788,
111
+ "eval_f1_micro": 0.8001549186676995,
112
+ "eval_f1_weighted": 0.7970808578587494,
113
+ "eval_loss": 0.8849796056747437,
114
+ "eval_macro_fpr": 0.017973543700983616,
115
+ "eval_macro_sensitivity": 0.704754303632947,
116
+ "eval_macro_specificity": 0.984883284344937,
117
+ "eval_precision": 0.8001040615771999,
118
+ "eval_precision_macro": 0.716655759638538,
119
+ "eval_recall": 0.8001549186676995,
120
+ "eval_recall_macro": 0.704754303632947,
121
+ "eval_runtime": 88.029,
122
+ "eval_samples_per_second": 14.666,
123
+ "eval_steps_per_second": 1.84,
124
+ "eval_weighted_fpr": 0.017527173913043478,
125
+ "eval_weighted_sensitivity": 0.8001549186676995,
126
+ "eval_weighted_specificity": 0.9730943465063553,
127
+ "step": 2572
128
+ },
129
+ {
130
+ "epoch": 4.67,
131
+ "learning_rate": 4.223950233281493e-05,
132
+ "loss": 0.3359,
133
+ "step": 3000
134
+ },
135
+ {
136
+ "epoch": 5.0,
137
+ "eval_accuracy": 0.801704105344694,
138
+ "eval_f1_macro": 0.6458763793901992,
139
+ "eval_f1_micro": 0.801704105344694,
140
+ "eval_f1_weighted": 0.7961730342415405,
141
+ "eval_loss": 1.126444935798645,
142
+ "eval_macro_fpr": 0.01805997413754306,
143
+ "eval_macro_sensitivity": 0.6681366041180199,
144
+ "eval_macro_specificity": 0.9849908327913178,
145
+ "eval_precision": 0.7981459541145973,
146
+ "eval_precision_macro": 0.6531140613829887,
147
+ "eval_recall": 0.801704105344694,
148
+ "eval_recall_macro": 0.6681366041180199,
149
+ "eval_runtime": 90.8127,
150
+ "eval_samples_per_second": 14.216,
151
+ "eval_steps_per_second": 1.784,
152
+ "eval_weighted_fpr": 0.0173606401736064,
153
+ "eval_weighted_sensitivity": 0.801704105344694,
154
+ "eval_weighted_specificity": 0.973158386525071,
155
+ "step": 3215
156
+ }
157
+ ],
158
+ "logging_steps": 500,
159
+ "max_steps": 19290,
160
+ "num_train_epochs": 30,
161
+ "save_steps": 500,
162
+ "total_flos": 7319363341731840.0,
163
+ "trial_name": null,
164
+ "trial_params": null
165
+ }
training_checkpoints/checkpoint-3215/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec20e9f8adfcfee3337ee75e60d11aa0228a2bd24494baa212e8cd20722381d4
3
+ size 4600