guychuk commited on
Commit
d7e37ea
·
verified ·
1 Parent(s): 8df7168

Upload 8 files

Browse files
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "CodeGenForCausalLM"
5
+ ],
6
+ "attn_pdrop": 0.0,
7
+ "bos_token_id": 1,
8
+ "embd_pdrop": 0.0,
9
+ "eos_token_id": 50256,
10
+ "gradient_checkpointing": false,
11
+ "initializer_range": 0.02,
12
+ "layer_norm_epsilon": 1e-05,
13
+ "model_type": "codegen",
14
+ "n_ctx": 2048,
15
+ "n_embd": 1024,
16
+ "n_head": 16,
17
+ "n_inner": null,
18
+ "n_layer": 20,
19
+ "n_positions": 2048,
20
+ "resid_pdrop": 0.0,
21
+ "rotary_dim": 32,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.1,
25
+ "summary_proj_to_labels": true,
26
+ "summary_type": "cls_index",
27
+ "summary_use_proj": true,
28
+ "task_specific_params": {
29
+ "text-generation": {
30
+ "do_sample": true,
31
+ "max_length": 50,
32
+ "temperature": 1.0
33
+ }
34
+ },
35
+ "tie_word_embeddings": false,
36
+ "tokenizer_class": "GPT2Tokenizer",
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.55.4",
39
+ "use_cache": true,
40
+ "vocab_size": 51200
41
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 50256,
5
+ "transformers_version": "4.55.4"
6
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5af9e810fe00bdd169fca4fe2a4cd03eb57d28e52a759e8b1cf8b20af7f5569
3
+ size 621314867
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:21ac788626a53d4c5d331c9d5cdab447e0c773fd0e9d8d95bbd942c5fbb018c5
3
+ size 1426902263
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0db1798478f6a6f7bda5704bc8d4a8ec83ed8bd0c9d780f3378b30ba41401878
3
+ size 14455
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51f7c991de3a3c468d9acec89cabc1677735d1ba747344862f4751a9438d6c48
3
+ size 1465
trainer_state.json ADDED
@@ -0,0 +1,139 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 50.0,
6
+ "eval_steps": 500,
7
+ "global_step": 150,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 3.4,
14
+ "grad_norm": 0.34349560737609863,
15
+ "learning_rate": 0.00047,
16
+ "loss": 1.5925,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 6.8,
21
+ "grad_norm": 0.007123686373233795,
22
+ "learning_rate": 0.00043666666666666664,
23
+ "loss": 0.0005,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 10.0,
28
+ "grad_norm": 0.0007117482600733638,
29
+ "learning_rate": 0.00040333333333333334,
30
+ "loss": 0.0,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 13.4,
35
+ "grad_norm": 0.00027794114430435,
36
+ "learning_rate": 0.00037,
37
+ "loss": 0.0,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 16.8,
42
+ "grad_norm": 0.00013230700278654695,
43
+ "learning_rate": 0.0003366666666666667,
44
+ "loss": 0.0,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 20.0,
49
+ "grad_norm": 9.377845708513632e-05,
50
+ "learning_rate": 0.00030333333333333335,
51
+ "loss": 0.0,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 23.4,
56
+ "grad_norm": 7.737267878837883e-05,
57
+ "learning_rate": 0.00027,
58
+ "loss": 0.0,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 26.8,
63
+ "grad_norm": 6.79552395013161e-05,
64
+ "learning_rate": 0.00023666666666666668,
65
+ "loss": 0.0,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 30.0,
70
+ "grad_norm": 6.188424595165998e-05,
71
+ "learning_rate": 0.00020333333333333333,
72
+ "loss": 0.0,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 33.4,
77
+ "grad_norm": 5.752248034696095e-05,
78
+ "learning_rate": 0.00017,
79
+ "loss": 0.0,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 36.8,
84
+ "grad_norm": 5.435854473034851e-05,
85
+ "learning_rate": 0.00013666666666666666,
86
+ "loss": 0.0,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 40.0,
91
+ "grad_norm": 5.203780528972857e-05,
92
+ "learning_rate": 0.00010333333333333333,
93
+ "loss": 0.0,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 43.4,
98
+ "grad_norm": 5.0433976866770536e-05,
99
+ "learning_rate": 7.000000000000001e-05,
100
+ "loss": 0.0,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 46.8,
105
+ "grad_norm": 4.946757690049708e-05,
106
+ "learning_rate": 3.6666666666666666e-05,
107
+ "loss": 0.0,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 50.0,
112
+ "grad_norm": 4.907431866740808e-05,
113
+ "learning_rate": 3.3333333333333337e-06,
114
+ "loss": 0.0,
115
+ "step": 150
116
+ }
117
+ ],
118
+ "logging_steps": 10,
119
+ "max_steps": 150,
120
+ "num_input_tokens_seen": 0,
121
+ "num_train_epochs": 50,
122
+ "save_steps": 500,
123
+ "stateful_callbacks": {
124
+ "TrainerControl": {
125
+ "args": {
126
+ "should_epoch_stop": false,
127
+ "should_evaluate": false,
128
+ "should_log": false,
129
+ "should_save": true,
130
+ "should_training_stop": true
131
+ },
132
+ "attributes": {}
133
+ }
134
+ },
135
+ "total_flos": 467379683328000.0,
136
+ "train_batch_size": 1,
137
+ "trial_name": null,
138
+ "trial_params": null
139
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68d11f9297b35a7a681108f1c001fb2043dd65cd6a0feb223f6b6f1a71d4ec13
3
+ size 5777