Oskar Holmström commited on
Commit
d14f47c
·
1 Parent(s): 744b905

First model version

Browse files
config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./models/OPT/opt-1.3b",
3
+ "_remove_final_layer_norm": false,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "relu",
6
+ "architectures": [
7
+ "OPTForCausalLM"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "bos_token_id": 2,
11
+ "do_layer_norm_before": true,
12
+ "dropout": 0.1,
13
+ "enable_bias": true,
14
+ "eos_token_id": 2,
15
+ "ffn_dim": 8192,
16
+ "hidden_size": 2048,
17
+ "init_std": 0.02,
18
+ "layer_norm_elementwise_affine": true,
19
+ "layerdrop": 0.0,
20
+ "max_position_embeddings": 2048,
21
+ "model_type": "opt",
22
+ "num_attention_heads": 32,
23
+ "num_hidden_layers": 24,
24
+ "pad_token_id": 1,
25
+ "prefix": "</s>",
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.26.0",
28
+ "use_cache": false,
29
+ "vocab_size": 50272,
30
+ "word_embed_proj_dim": 2048
31
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 2,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 1,
6
+ "transformers_version": "4.26.0"
7
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a99668e2f43d0bc2ea538dc56ee5b6a9a3193fb6ea44bd8378f119ba616da8e
3
+ size 10526299041
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02644fb7ade4ef5113bbe747c51459115e68eecf41f36a62a88a9f8c8be64afc
3
+ size 5263162257
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e0956de6390e858565f3e0e7eb7ec59770d049540a033a0a8ddd00bf2170575
3
+ size 14503
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc95b508acc43782ce53498e0ad90507a699c09b1452ef7ccf2e247b7d5df105
3
+ size 623
trainer_state.json ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.8036913731674553,
5
+ "global_step": 6000,
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.23,
12
+ "learning_rate": 7.78816199376947e-06,
13
+ "loss": 1.962,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.47,
18
+ "learning_rate": 1.557632398753894e-05,
19
+ "loss": 0.7774,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.7,
24
+ "learning_rate": 1.9626168224299065e-05,
25
+ "loss": 0.7326,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.93,
30
+ "learning_rate": 1.8760816891658016e-05,
31
+ "loss": 0.6961,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 1.17,
36
+ "learning_rate": 1.7895465559016964e-05,
37
+ "loss": 0.6348,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 1.4,
42
+ "learning_rate": 1.7030114226375908e-05,
43
+ "loss": 0.6088,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 1.64,
48
+ "learning_rate": 1.616476289373486e-05,
49
+ "loss": 0.5995,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 1.87,
54
+ "learning_rate": 1.5299411561093806e-05,
55
+ "loss": 0.6067,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 2.1,
60
+ "learning_rate": 1.4434060228452753e-05,
61
+ "loss": 0.5517,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 2.34,
66
+ "learning_rate": 1.3568708895811699e-05,
67
+ "loss": 0.4912,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 2.57,
72
+ "learning_rate": 1.2703357563170648e-05,
73
+ "loss": 0.4921,
74
+ "step": 5500
75
+ },
76
+ {
77
+ "epoch": 2.8,
78
+ "learning_rate": 1.1838006230529595e-05,
79
+ "loss": 0.4947,
80
+ "step": 6000
81
+ }
82
+ ],
83
+ "max_steps": 12840,
84
+ "num_train_epochs": 6,
85
+ "total_flos": 3.5783067431981875e+17,
86
+ "trial_name": null,
87
+ "trial_params": null
88
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eda971994eab2365d709b36816efa694f433c58c2849c619cbd44c66f87a171a
3
+ size 3503