quim-motger commited on
Commit
d435972
1 Parent(s): 655010e

Upload 8 files

Browse files
config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "data/further_pretraining/xlnet-large-cased/checkpoint-42354",
3
+ "architectures": [
4
+ "XLNetLMHeadModel"
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": 4096,
12
+ "d_model": 1024,
13
+ "dropout": 0.1,
14
+ "end_n_top": 5,
15
+ "eos_token_id": 2,
16
+ "ff_activation": "gelu",
17
+ "initializer_range": 0.02,
18
+ "layer_norm_eps": 1e-12,
19
+ "mem_len": null,
20
+ "model_type": "xlnet",
21
+ "n_head": 16,
22
+ "n_layer": 24,
23
+ "pad_token_id": 5,
24
+ "reuse_len": null,
25
+ "same_length": false,
26
+ "start_n_top": 5,
27
+ "summary_activation": "tanh",
28
+ "summary_last_dropout": 0.1,
29
+ "summary_type": "last",
30
+ "summary_use_proj": true,
31
+ "task_specific_params": {
32
+ "text-generation": {
33
+ "do_sample": true,
34
+ "max_length": 250
35
+ }
36
+ },
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.37.2",
39
+ "untie_r": true,
40
+ "use_mems_eval": true,
41
+ "use_mems_train": false,
42
+ "vocab_size": 32000
43
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 2,
5
+ "pad_token_id": 5,
6
+ "transformers_version": "4.37.2"
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86ba3e8d38edc5e430a564acb7229761dc5ac70ee5c86a32505a01ef3f6ea834
3
+ size 1441248632
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ed847e330aa92c74b3f96f0fa64708c1d1aca4ce37bc60622abbea3539ad308c
3
+ size 2882752227
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9baa22c0a5f56036cc8b7a2f34fc4ae033a88859cc813c28d7171e45feb7c0d
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58ddc11590b44292b46c83821a904f319a165e36d3b4d30a50f0bd644c35745a
3
+ size 1064
trainer_state.json ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.00019271760538686067,
3
+ "best_model_checkpoint": "data/further_pretraining/xlnet-large-cased/checkpoint-4706",
4
+ "epoch": 1.0,
5
+ "eval_steps": 500,
6
+ "global_step": 4706,
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.11,
13
+ "learning_rate": 2.6812579685507864e-05,
14
+ "loss": 0.0002,
15
+ "step": 500
16
+ },
17
+ {
18
+ "epoch": 0.21,
19
+ "learning_rate": 2.3625159371015725e-05,
20
+ "loss": 0.0001,
21
+ "step": 1000
22
+ },
23
+ {
24
+ "epoch": 0.32,
25
+ "learning_rate": 2.043773905652359e-05,
26
+ "loss": 0.0001,
27
+ "step": 1500
28
+ },
29
+ {
30
+ "epoch": 0.42,
31
+ "learning_rate": 1.7250318742031452e-05,
32
+ "loss": 0.0077,
33
+ "step": 2000
34
+ },
35
+ {
36
+ "epoch": 0.53,
37
+ "learning_rate": 1.4062898427539311e-05,
38
+ "loss": 0.0006,
39
+ "step": 2500
40
+ },
41
+ {
42
+ "epoch": 0.64,
43
+ "learning_rate": 1.0875478113047175e-05,
44
+ "loss": 0.0006,
45
+ "step": 3000
46
+ },
47
+ {
48
+ "epoch": 0.74,
49
+ "learning_rate": 7.688057798555037e-06,
50
+ "loss": 0.0,
51
+ "step": 3500
52
+ },
53
+ {
54
+ "epoch": 0.85,
55
+ "learning_rate": 4.500637484062899e-06,
56
+ "loss": 0.0003,
57
+ "step": 4000
58
+ },
59
+ {
60
+ "epoch": 0.96,
61
+ "learning_rate": 1.3132171695707608e-06,
62
+ "loss": 0.0,
63
+ "step": 4500
64
+ },
65
+ {
66
+ "epoch": 1.0,
67
+ "eval_loss": 0.00019271760538686067,
68
+ "eval_runtime": 829.7336,
69
+ "eval_samples_per_second": 90.73,
70
+ "eval_steps_per_second": 5.672,
71
+ "step": 4706
72
+ }
73
+ ],
74
+ "logging_steps": 500,
75
+ "max_steps": 4706,
76
+ "num_input_tokens_seen": 0,
77
+ "num_train_epochs": 1,
78
+ "save_steps": 500,
79
+ "total_flos": 1.89368250236928e+16,
80
+ "train_batch_size": 16,
81
+ "trial_name": null,
82
+ "trial_params": null
83
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5138dad5d0f337f45af836663cb38b59f101b69dad77ac7e70b54b4025dd87c
3
+ size 4728