Hugging Face's logo

xiulinyang
/
GPT2-small-AR

PyTorch
gpt2
Model card Files
xet
 Community
xiulinyang commited on
Commit
798b6fe
·
1 Parent(s): 2af2716

Adding model checkpoints and config files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. checkpoint-0/config.json +38 -0
  2. checkpoint-0/pytorch_model.bin +3 -0
  3. checkpoint-0/special_tokens_map.json +1 -0
  4. checkpoint-0/tokenizer_config.json +1 -0
  5. checkpoint-0/training_args.bin +3 -0
  6. checkpoint-100/config.json +38 -0
  7. checkpoint-100/optimizer.pt +3 -0
  8. checkpoint-100/pytorch_model.bin +3 -0
  9. checkpoint-100/rng_state.pth +3 -0
  10. checkpoint-100/scaler.pt +3 -0
  11. checkpoint-100/scheduler.pt +3 -0
  12. checkpoint-100/special_tokens_map.json +1 -0
  13. checkpoint-100/tokenizer_config.json +1 -0
  14. checkpoint-100/trainer_state.json +34 -0
  15. checkpoint-100/training_args.bin +3 -0
  16. checkpoint-1000/config.json +38 -0
  17. checkpoint-1000/optimizer.pt +3 -0
  18. checkpoint-1000/pytorch_model.bin +3 -0
  19. checkpoint-1000/rng_state.pth +3 -0
  20. checkpoint-1000/scaler.pt +3 -0
  21. checkpoint-1000/scheduler.pt +3 -0
  22. checkpoint-1000/special_tokens_map.json +1 -0
  23. checkpoint-1000/tokenizer_config.json +1 -0
  24. checkpoint-1000/trainer_state.json +158 -0
  25. checkpoint-1000/training_args.bin +3 -0
  26. checkpoint-1100/config.json +38 -0
  27. checkpoint-1100/optimizer.pt +3 -0
  28. checkpoint-1100/pytorch_model.bin +3 -0
  29. checkpoint-1100/rng_state.pth +3 -0
  30. checkpoint-1100/scaler.pt +3 -0
  31. checkpoint-1100/scheduler.pt +3 -0
  32. checkpoint-1100/special_tokens_map.json +1 -0
  33. checkpoint-1100/tokenizer_config.json +1 -0
  34. checkpoint-1100/trainer_state.json +170 -0
  35. checkpoint-1100/training_args.bin +3 -0
  36. checkpoint-1200/config.json +38 -0
  37. checkpoint-1200/optimizer.pt +3 -0
  38. checkpoint-1200/pytorch_model.bin +3 -0
  39. checkpoint-1200/rng_state.pth +3 -0
  40. checkpoint-1200/scaler.pt +3 -0
  41. checkpoint-1200/scheduler.pt +3 -0
  42. checkpoint-1200/special_tokens_map.json +1 -0
  43. checkpoint-1200/tokenizer_config.json +1 -0
  44. checkpoint-1200/trainer_state.json +182 -0
  45. checkpoint-1200/training_args.bin +3 -0
  46. checkpoint-200/config.json +38 -0
  47. checkpoint-200/optimizer.pt +3 -0
  48. checkpoint-200/pytorch_model.bin +3 -0
  49. checkpoint-200/rng_state.pth +3 -0
  50. checkpoint-200/scaler.pt +3 -0
checkpoint-0/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 0,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 0,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 768,
15
+ "n_head": 12,
16
+ "n_inner": null,
17
+ "n_layer": 12,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": true,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": true,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.2,
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": 1024
32
+ }
33
+ },
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.18.0",
36
+ "use_cache": false,
37
+ "vocab_size": 64000
38
+ }
checkpoint-0/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d9938f9842ab67a4ffdb8192f0ef1a92afea4592e39c9cede29e616f6bb617f
3
+ size 552615017
checkpoint-0/special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
checkpoint-0/tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokenizer_class": "PassthroughTokenizer"}
checkpoint-0/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f7a8813458a4fe4504a0399689972683329e3ecf180a5751fb9d319b2b9ab7f
3
+ size 3183
checkpoint-100/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 0,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 0,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 768,
15
+ "n_head": 12,
16
+ "n_inner": null,
17
+ "n_layer": 12,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": true,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": true,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.2,
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": 1024
32
+ }
33
+ },
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.18.0",
36
+ "use_cache": false,
37
+ "vocab_size": 64000
38
+ }
checkpoint-100/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31ce645e089b5c13d93ddc41dbb03b99ba2e8405370f6bd4a4261c4bafbfc1f8
3
+ size 1080040817
checkpoint-100/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:551a90329c4252446c053879f3fc1a8fe887474155c822fd52473a9aa7fde08b
3
+ size 552615017
checkpoint-100/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ce3d614cae87a2a2a3d14a905bff34ed78c43763648037d7c081ee15db8abcb
3
+ size 14567
checkpoint-100/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13a3423b2fe42f204bc8fe2c666ff379f9fd753a0f13613064a5e71e86b519e8
3
+ size 559
checkpoint-100/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75e9b9d31d11c624d89b0c04ad496adf4b5addd3e703848d2583972c703e8da6
3
+ size 623
checkpoint-100/special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
checkpoint-100/tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokenizer_class": "PassthroughTokenizer"}
checkpoint-100/trainer_state.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0183333333333335,
5
+ "global_step": 100,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 4.9999999999999996e-06,
13
+ "loss": 11.1259,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 1.02,
18
+ "learning_rate": 0.00025,
19
+ "loss": 8.8129,
20
+ "step": 50
21
+ },
22
+ {
23
+ "epoch": 3.02,
24
+ "learning_rate": 0.0005,
25
+ "loss": 7.1869,
26
+ "step": 100
27
+ }
28
+ ],
29
+ "max_steps": 1200,
30
+ "num_train_epochs": 9223372036854775807,
31
+ "total_flos": 2.7248578265088e+16,
32
+ "trial_name": null,
33
+ "trial_params": null
34
+ }
checkpoint-100/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f7a8813458a4fe4504a0399689972683329e3ecf180a5751fb9d319b2b9ab7f
3
+ size 3183
checkpoint-1000/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 0,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 0,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 768,
15
+ "n_head": 12,
16
+ "n_inner": null,
17
+ "n_layer": 12,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": true,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": true,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.2,
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": 1024
32
+ }
33
+ },
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.18.0",
36
+ "use_cache": false,
37
+ "vocab_size": 64000
38
+ }
checkpoint-1000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ba6e5a33b1e57f01eafb424853635a5b0c6634b4f4b1435ca6a6259c231b7a9
3
+ size 1080041009
checkpoint-1000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfcfe118da1abf34ff4818ee124f48b1e7dbd58bac92c185adcc2b9770d5c212
3
+ size 552615017
checkpoint-1000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90d9f5d49b3a45d37d36896a4a6e66be8fa8ccae1cf10d2d8456f2379f95276c
3
+ size 14567
checkpoint-1000/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f810fc7b695697c440d8985f6042b4ba23a9e1027604c265718b518ca29f1b2b
3
+ size 559
checkpoint-1000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0691206f4bd9ca409d6e7104087a4e0eb05df8f8f555a400f6ecc532edba52d8
3
+ size 623
checkpoint-1000/special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
checkpoint-1000/tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokenizer_class": "PassthroughTokenizer"}
checkpoint-1000/trainer_state.json ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 38.01,
5
+ "global_step": 1000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 4.9999999999999996e-06,
13
+ "loss": 11.1259,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 1.02,
18
+ "learning_rate": 0.00025,
19
+ "loss": 8.8129,
20
+ "step": 50
21
+ },
22
+ {
23
+ "epoch": 3.02,
24
+ "learning_rate": 0.0005,
25
+ "loss": 7.1869,
26
+ "step": 100
27
+ },
28
+ {
29
+ "epoch": 5.02,
30
+ "learning_rate": 0.0005833333333333333,
31
+ "loss": 6.5568,
32
+ "step": 150
33
+ },
34
+ {
35
+ "epoch": 7.01,
36
+ "learning_rate": 0.0005555555555555556,
37
+ "loss": 6.0674,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 9.01,
42
+ "learning_rate": 0.0005277777777777777,
43
+ "loss": 5.7599,
44
+ "step": 250
45
+ },
46
+ {
47
+ "epoch": 11.01,
48
+ "learning_rate": 0.0005,
49
+ "loss": 5.5578,
50
+ "step": 300
51
+ },
52
+ {
53
+ "epoch": 13.01,
54
+ "learning_rate": 0.00047222222222222224,
55
+ "loss": 5.3958,
56
+ "step": 350
57
+ },
58
+ {
59
+ "epoch": 15.01,
60
+ "learning_rate": 0.00044444444444444436,
61
+ "loss": 5.2326,
62
+ "step": 400
63
+ },
64
+ {
65
+ "epoch": 17.01,
66
+ "learning_rate": 0.00041666666666666664,
67
+ "loss": 5.0814,
68
+ "step": 450
69
+ },
70
+ {
71
+ "epoch": 19.0,
72
+ "learning_rate": 0.00038888888888888887,
73
+ "loss": 4.9099,
74
+ "step": 500
75
+ },
76
+ {
77
+ "epoch": 21.0,
78
+ "learning_rate": 0.0003611111111111111,
79
+ "loss": 4.7405,
80
+ "step": 550
81
+ },
82
+ {
83
+ "epoch": 23.0,
84
+ "learning_rate": 0.0003333333333333333,
85
+ "loss": 4.5977,
86
+ "step": 600
87
+ },
88
+ {
89
+ "epoch": 24.02,
90
+ "learning_rate": 0.00030555555555555555,
91
+ "loss": 4.4226,
92
+ "step": 650
93
+ },
94
+ {
95
+ "epoch": 26.02,
96
+ "learning_rate": 0.0002777777777777778,
97
+ "loss": 4.3734,
98
+ "step": 700
99
+ },
100
+ {
101
+ "epoch": 28.02,
102
+ "learning_rate": 0.00025,
103
+ "loss": 4.2801,
104
+ "step": 750
105
+ },
106
+ {
107
+ "epoch": 30.02,
108
+ "learning_rate": 0.00022222222222222218,
109
+ "loss": 4.195,
110
+ "step": 800
111
+ },
112
+ {
113
+ "epoch": 32.02,
114
+ "learning_rate": 0.00019444444444444443,
115
+ "loss": 4.117,
116
+ "step": 850
117
+ },
118
+ {
119
+ "epoch": 34.01,
120
+ "learning_rate": 0.00016666666666666666,
121
+ "loss": 4.0469,
122
+ "step": 900
123
+ },
124
+ {
125
+ "epoch": 36.01,
126
+ "learning_rate": 0.0001388888888888889,
127
+ "loss": 3.9843,
128
+ "step": 950
129
+ },
130
+ {
131
+ "epoch": 38.01,
132
+ "learning_rate": 0.00011111111111111109,
133
+ "loss": 3.9288,
134
+ "step": 1000
135
+ },
136
+ {
137
+ "epoch": 38.01,
138
+ "eval_loss": 5.421392917633057,
139
+ "eval_runtime": 5.2963,
140
+ "eval_samples_per_second": 16.238,
141
+ "eval_steps_per_second": 1.133,
142
+ "step": 1000
143
+ },
144
+ {
145
+ "epoch": 38.01,
146
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_loss": 5.421392917633057,
147
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_ppl": 226.1939727440261,
148
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_runtime": 5.2963,
149
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_samples_per_second": 16.238,
150
+ "step": 1000
151
+ }
152
+ ],
153
+ "max_steps": 1200,
154
+ "num_train_epochs": 9223372036854775807,
155
+ "total_flos": 2.73798513819648e+17,
156
+ "trial_name": null,
157
+ "trial_params": null
158
+ }
checkpoint-1000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f7a8813458a4fe4504a0399689972683329e3ecf180a5751fb9d319b2b9ab7f
3
+ size 3183
checkpoint-1100/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 0,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 0,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 768,
15
+ "n_head": 12,
16
+ "n_inner": null,
17
+ "n_layer": 12,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": true,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": true,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.2,
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": 1024
32
+ }
33
+ },
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.18.0",
36
+ "use_cache": false,
37
+ "vocab_size": 64000
38
+ }
checkpoint-1100/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:22b765021c39f3200331b8273c3a3e87fdd379831760b7a18d2ff8d7e9cb57a2
3
+ size 1080041009
checkpoint-1100/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fd5031aef73e64832ae02ddae426c9597d8ef7596b789ad6a9263f5066e31d2
3
+ size 552615017
checkpoint-1100/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f6b4ddc122460c50f83c387f129e0b43a75be90f55e4bcb04a6e8e73b75e529
3
+ size 14567
checkpoint-1100/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fb16c30b686aa43e110b0d33f9d46bf3127b7124542ca8dc34831233d4675a0
3
+ size 559
checkpoint-1100/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f1b7713e4bb40428f29080b7d08d4a52f779ac863737861e4724292b2cf6c59
3
+ size 623
checkpoint-1100/special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
checkpoint-1100/tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokenizer_class": "PassthroughTokenizer"}
checkpoint-1100/trainer_state.json ADDED
@@ -0,0 +1,170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 42.00666666666667,
5
+ "global_step": 1100,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 4.9999999999999996e-06,
13
+ "loss": 11.1259,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 1.02,
18
+ "learning_rate": 0.00025,
19
+ "loss": 8.8129,
20
+ "step": 50
21
+ },
22
+ {
23
+ "epoch": 3.02,
24
+ "learning_rate": 0.0005,
25
+ "loss": 7.1869,
26
+ "step": 100
27
+ },
28
+ {
29
+ "epoch": 5.02,
30
+ "learning_rate": 0.0005833333333333333,
31
+ "loss": 6.5568,
32
+ "step": 150
33
+ },
34
+ {
35
+ "epoch": 7.01,
36
+ "learning_rate": 0.0005555555555555556,
37
+ "loss": 6.0674,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 9.01,
42
+ "learning_rate": 0.0005277777777777777,
43
+ "loss": 5.7599,
44
+ "step": 250
45
+ },
46
+ {
47
+ "epoch": 11.01,
48
+ "learning_rate": 0.0005,
49
+ "loss": 5.5578,
50
+ "step": 300
51
+ },
52
+ {
53
+ "epoch": 13.01,
54
+ "learning_rate": 0.00047222222222222224,
55
+ "loss": 5.3958,
56
+ "step": 350
57
+ },
58
+ {
59
+ "epoch": 15.01,
60
+ "learning_rate": 0.00044444444444444436,
61
+ "loss": 5.2326,
62
+ "step": 400
63
+ },
64
+ {
65
+ "epoch": 17.01,
66
+ "learning_rate": 0.00041666666666666664,
67
+ "loss": 5.0814,
68
+ "step": 450
69
+ },
70
+ {
71
+ "epoch": 19.0,
72
+ "learning_rate": 0.00038888888888888887,
73
+ "loss": 4.9099,
74
+ "step": 500
75
+ },
76
+ {
77
+ "epoch": 21.0,
78
+ "learning_rate": 0.0003611111111111111,
79
+ "loss": 4.7405,
80
+ "step": 550
81
+ },
82
+ {
83
+ "epoch": 23.0,
84
+ "learning_rate": 0.0003333333333333333,
85
+ "loss": 4.5977,
86
+ "step": 600
87
+ },
88
+ {
89
+ "epoch": 24.02,
90
+ "learning_rate": 0.00030555555555555555,
91
+ "loss": 4.4226,
92
+ "step": 650
93
+ },
94
+ {
95
+ "epoch": 26.02,
96
+ "learning_rate": 0.0002777777777777778,
97
+ "loss": 4.3734,
98
+ "step": 700
99
+ },
100
+ {
101
+ "epoch": 28.02,
102
+ "learning_rate": 0.00025,
103
+ "loss": 4.2801,
104
+ "step": 750
105
+ },
106
+ {
107
+ "epoch": 30.02,
108
+ "learning_rate": 0.00022222222222222218,
109
+ "loss": 4.195,
110
+ "step": 800
111
+ },
112
+ {
113
+ "epoch": 32.02,
114
+ "learning_rate": 0.00019444444444444443,
115
+ "loss": 4.117,
116
+ "step": 850
117
+ },
118
+ {
119
+ "epoch": 34.01,
120
+ "learning_rate": 0.00016666666666666666,
121
+ "loss": 4.0469,
122
+ "step": 900
123
+ },
124
+ {
125
+ "epoch": 36.01,
126
+ "learning_rate": 0.0001388888888888889,
127
+ "loss": 3.9843,
128
+ "step": 950
129
+ },
130
+ {
131
+ "epoch": 38.01,
132
+ "learning_rate": 0.00011111111111111109,
133
+ "loss": 3.9288,
134
+ "step": 1000
135
+ },
136
+ {
137
+ "epoch": 38.01,
138
+ "eval_loss": 5.421392917633057,
139
+ "eval_runtime": 5.2963,
140
+ "eval_samples_per_second": 16.238,
141
+ "eval_steps_per_second": 1.133,
142
+ "step": 1000
143
+ },
144
+ {
145
+ "epoch": 38.01,
146
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_loss": 5.421392917633057,
147
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_ppl": 226.1939727440261,
148
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_runtime": 5.2963,
149
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_samples_per_second": 16.238,
150
+ "step": 1000
151
+ },
152
+ {
153
+ "epoch": 40.01,
154
+ "learning_rate": 8.333333333333333e-05,
155
+ "loss": 3.8811,
156
+ "step": 1050
157
+ },
158
+ {
159
+ "epoch": 42.01,
160
+ "learning_rate": 5.5555555555555545e-05,
161
+ "loss": 3.8399,
162
+ "step": 1100
163
+ }
164
+ ],
165
+ "max_steps": 1200,
166
+ "num_train_epochs": 9223372036854775807,
167
+ "total_flos": 3.01211183480832e+17,
168
+ "trial_name": null,
169
+ "trial_params": null
170
+ }
checkpoint-1100/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f7a8813458a4fe4504a0399689972683329e3ecf180a5751fb9d319b2b9ab7f
3
+ size 3183
checkpoint-1200/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 0,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 0,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 768,
15
+ "n_head": 12,
16
+ "n_inner": null,
17
+ "n_layer": 12,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": true,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": true,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.2,
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": 1024
32
+ }
33
+ },
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.18.0",
36
+ "use_cache": false,
37
+ "vocab_size": 64000
38
+ }
checkpoint-1200/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d33720fe57a7f0905d6415099c325a97b485dcd7e7102370d9d879b5b41e2683
3
+ size 1080041009
checkpoint-1200/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3804d0526cf824af676d65a388447f7c82a43e3c7f081b8e78d4618805efde40
3
+ size 552615017
checkpoint-1200/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:844447be1a207102c945ce5f979114648ce7ff240fdda57f96f05cff5a24f762
3
+ size 14567
checkpoint-1200/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19c7277eaca0850ae3e9b6790b3d002d820169cce0671185e672c28c8ae8e056
3
+ size 559
checkpoint-1200/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:935a8fb09a6e9698d9894853b05e181b3f56098deaaecddde08e55f06bf000c4
3
+ size 623
checkpoint-1200/special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
checkpoint-1200/tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tokenizer_class": "PassthroughTokenizer"}
checkpoint-1200/trainer_state.json ADDED
@@ -0,0 +1,182 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 46.00333333333333,
5
+ "global_step": 1200,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 4.9999999999999996e-06,
13
+ "loss": 11.1259,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 1.02,
18
+ "learning_rate": 0.00025,
19
+ "loss": 8.8129,
20
+ "step": 50
21
+ },
22
+ {
23
+ "epoch": 3.02,
24
+ "learning_rate": 0.0005,
25
+ "loss": 7.1869,
26
+ "step": 100
27
+ },
28
+ {
29
+ "epoch": 5.02,
30
+ "learning_rate": 0.0005833333333333333,
31
+ "loss": 6.5568,
32
+ "step": 150
33
+ },
34
+ {
35
+ "epoch": 7.01,
36
+ "learning_rate": 0.0005555555555555556,
37
+ "loss": 6.0674,
38
+ "step": 200
39
+ },
40
+ {
41
+ "epoch": 9.01,
42
+ "learning_rate": 0.0005277777777777777,
43
+ "loss": 5.7599,
44
+ "step": 250
45
+ },
46
+ {
47
+ "epoch": 11.01,
48
+ "learning_rate": 0.0005,
49
+ "loss": 5.5578,
50
+ "step": 300
51
+ },
52
+ {
53
+ "epoch": 13.01,
54
+ "learning_rate": 0.00047222222222222224,
55
+ "loss": 5.3958,
56
+ "step": 350
57
+ },
58
+ {
59
+ "epoch": 15.01,
60
+ "learning_rate": 0.00044444444444444436,
61
+ "loss": 5.2326,
62
+ "step": 400
63
+ },
64
+ {
65
+ "epoch": 17.01,
66
+ "learning_rate": 0.00041666666666666664,
67
+ "loss": 5.0814,
68
+ "step": 450
69
+ },
70
+ {
71
+ "epoch": 19.0,
72
+ "learning_rate": 0.00038888888888888887,
73
+ "loss": 4.9099,
74
+ "step": 500
75
+ },
76
+ {
77
+ "epoch": 21.0,
78
+ "learning_rate": 0.0003611111111111111,
79
+ "loss": 4.7405,
80
+ "step": 550
81
+ },
82
+ {
83
+ "epoch": 23.0,
84
+ "learning_rate": 0.0003333333333333333,
85
+ "loss": 4.5977,
86
+ "step": 600
87
+ },
88
+ {
89
+ "epoch": 24.02,
90
+ "learning_rate": 0.00030555555555555555,
91
+ "loss": 4.4226,
92
+ "step": 650
93
+ },
94
+ {
95
+ "epoch": 26.02,
96
+ "learning_rate": 0.0002777777777777778,
97
+ "loss": 4.3734,
98
+ "step": 700
99
+ },
100
+ {
101
+ "epoch": 28.02,
102
+ "learning_rate": 0.00025,
103
+ "loss": 4.2801,
104
+ "step": 750
105
+ },
106
+ {
107
+ "epoch": 30.02,
108
+ "learning_rate": 0.00022222222222222218,
109
+ "loss": 4.195,
110
+ "step": 800
111
+ },
112
+ {
113
+ "epoch": 32.02,
114
+ "learning_rate": 0.00019444444444444443,
115
+ "loss": 4.117,
116
+ "step": 850
117
+ },
118
+ {
119
+ "epoch": 34.01,
120
+ "learning_rate": 0.00016666666666666666,
121
+ "loss": 4.0469,
122
+ "step": 900
123
+ },
124
+ {
125
+ "epoch": 36.01,
126
+ "learning_rate": 0.0001388888888888889,
127
+ "loss": 3.9843,
128
+ "step": 950
129
+ },
130
+ {
131
+ "epoch": 38.01,
132
+ "learning_rate": 0.00011111111111111109,
133
+ "loss": 3.9288,
134
+ "step": 1000
135
+ },
136
+ {
137
+ "epoch": 38.01,
138
+ "eval_loss": 5.421392917633057,
139
+ "eval_runtime": 5.2963,
140
+ "eval_samples_per_second": 16.238,
141
+ "eval_steps_per_second": 1.133,
142
+ "step": 1000
143
+ },
144
+ {
145
+ "epoch": 38.01,
146
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_loss": 5.421392917633057,
147
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_ppl": 226.1939727440261,
148
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_runtime": 5.2963,
149
+ "eval_/scratch/ykyao/projects/multilingual-LM/training/multilingual_dataset.py_samples_per_second": 16.238,
150
+ "step": 1000
151
+ },
152
+ {
153
+ "epoch": 40.01,
154
+ "learning_rate": 8.333333333333333e-05,
155
+ "loss": 3.8811,
156
+ "step": 1050
157
+ },
158
+ {
159
+ "epoch": 42.01,
160
+ "learning_rate": 5.5555555555555545e-05,
161
+ "loss": 3.8399,
162
+ "step": 1100
163
+ },
164
+ {
165
+ "epoch": 44.01,
166
+ "learning_rate": 2.7777777777777772e-05,
167
+ "loss": 3.8059,
168
+ "step": 1150
169
+ },
170
+ {
171
+ "epoch": 46.0,
172
+ "learning_rate": 0.0,
173
+ "loss": 3.7808,
174
+ "step": 1200
175
+ }
176
+ ],
177
+ "max_steps": 1200,
178
+ "num_train_epochs": 9223372036854775807,
179
+ "total_flos": 3.28623853142016e+17,
180
+ "trial_name": null,
181
+ "trial_params": null
182
+ }
checkpoint-1200/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f7a8813458a4fe4504a0399689972683329e3ecf180a5751fb9d319b2b9ab7f
3
+ size 3183
checkpoint-200/config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_function": "gelu_new",
3
+ "architectures": [
4
+ "GPT2LMHeadModel"
5
+ ],
6
+ "attn_pdrop": 0.1,
7
+ "bos_token_id": 0,
8
+ "embd_pdrop": 0.1,
9
+ "eos_token_id": 0,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_epsilon": 1e-05,
12
+ "model_type": "gpt2",
13
+ "n_ctx": 1024,
14
+ "n_embd": 768,
15
+ "n_head": 12,
16
+ "n_inner": null,
17
+ "n_layer": 12,
18
+ "n_positions": 1024,
19
+ "reorder_and_upcast_attn": true,
20
+ "resid_pdrop": 0.1,
21
+ "scale_attn_by_inverse_layer_idx": true,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.2,
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": 1024
32
+ }
33
+ },
34
+ "torch_dtype": "float32",
35
+ "transformers_version": "4.18.0",
36
+ "use_cache": false,
37
+ "vocab_size": 64000
38
+ }
checkpoint-200/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93facc0aeb96c3523f43014c0d5a5cde08d4497c4988166f856083e4f8f78b5e
3
+ size 1080040817
checkpoint-200/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e1cef4e077c9f9d793269b5f8f8b4436db686fd0a6b234ad28b103893d06b08
3
+ size 552615017
checkpoint-200/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5761d23f2e2f936c1ff48ba6ab3e1000cfa77a0c280ef5bd98f37848f10a669c
3
+ size 14567
checkpoint-200/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb6982c29cd162f49aeb531674acf574eccd46a8f556bec596040d7c3b95200a
3
+ size 559