Merit Kuldkepp commited on
Commit
8798144
·
1 Parent(s): 4351ae4

First model version

Browse files
Files changed (4) hide show
  1. config.json +60 -0
  2. generation_config.json +6 -0
  3. pytorch_model.bin +3 -0
  4. trainer_state.json +100 -0
config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/content/drive/MyDrive/results6/checkpoint-12460",
3
+ "architectures": [
4
+ "T5ForConditionalGeneration"
5
+ ],
6
+ "d_ff": 3072,
7
+ "d_kv": 64,
8
+ "d_model": 768,
9
+ "decoder_start_token_id": 0,
10
+ "dense_act_fn": "relu",
11
+ "dropout_rate": 0.1,
12
+ "eos_token_id": 1,
13
+ "feed_forward_proj": "relu",
14
+ "initializer_factor": 1.0,
15
+ "is_encoder_decoder": true,
16
+ "is_gated_act": false,
17
+ "layer_norm_epsilon": 1e-06,
18
+ "model_type": "t5",
19
+ "n_positions": 512,
20
+ "num_decoder_layers": 12,
21
+ "num_heads": 12,
22
+ "num_layers": 12,
23
+ "output_past": true,
24
+ "pad_token_id": 0,
25
+ "relative_attention_max_distance": 128,
26
+ "relative_attention_num_buckets": 32,
27
+ "task_specific_params": {
28
+ "summarization": {
29
+ "early_stopping": true,
30
+ "length_penalty": 2.0,
31
+ "max_length": 200,
32
+ "min_length": 30,
33
+ "no_repeat_ngram_size": 3,
34
+ "num_beams": 4,
35
+ "prefix": "summarize: "
36
+ },
37
+ "translation_en_to_de": {
38
+ "early_stopping": true,
39
+ "max_length": 300,
40
+ "num_beams": 4,
41
+ "prefix": "translate English to German: "
42
+ },
43
+ "translation_en_to_fr": {
44
+ "early_stopping": true,
45
+ "max_length": 300,
46
+ "num_beams": 4,
47
+ "prefix": "translate English to French: "
48
+ },
49
+ "translation_en_to_ro": {
50
+ "early_stopping": true,
51
+ "max_length": 300,
52
+ "num_beams": 4,
53
+ "prefix": "translate English to Romanian: "
54
+ }
55
+ },
56
+ "torch_dtype": "float32",
57
+ "transformers_version": "4.27.0",
58
+ "use_cache": true,
59
+ "vocab_size": 32128
60
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "decoder_start_token_id": 0,
3
+ "eos_token_id": 1,
4
+ "pad_token_id": 0,
5
+ "transformers_version": "4.27.0"
6
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f19c703280faab8db6875a6720b2083c60ea74facb7217a9af5c52ce4f013ca
3
+ size 891702929
trainer_state.json ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.2918661832809448,
3
+ "epoch": 5.0,
4
+ "global_step": 6230,
5
+ "is_hyper_param_search": false,
6
+ "is_local_process_zero": true,
7
+ "is_world_process_zero": true,
8
+ "log_history": [
9
+ {
10
+ "epoch": 4.08,
11
+ "learning_rate": 9.866238630283575e-06,
12
+ "loss": 0.8765,
13
+ "step": 500
14
+ },
15
+ {
16
+ "epoch": 4.16,
17
+ "learning_rate": 9.73247726056715e-06,
18
+ "loss": 0.8865,
19
+ "step": 1000
20
+ },
21
+ {
22
+ "epoch": 4.24,
23
+ "learning_rate": 9.598715890850723e-06,
24
+ "loss": 0.8513,
25
+ "step": 1500
26
+ },
27
+ {
28
+ "epoch": 4.32,
29
+ "learning_rate": 9.464954521134298e-06,
30
+ "loss": 0.8626,
31
+ "step": 2000
32
+ },
33
+ {
34
+ "epoch": 4.4,
35
+ "learning_rate": 9.331193151417872e-06,
36
+ "loss": 0.8445,
37
+ "step": 2500
38
+ },
39
+ {
40
+ "epoch": 4.48,
41
+ "learning_rate": 9.197431781701446e-06,
42
+ "loss": 0.8685,
43
+ "step": 3000
44
+ },
45
+ {
46
+ "epoch": 4.56,
47
+ "learning_rate": 9.06367041198502e-06,
48
+ "loss": 0.8664,
49
+ "step": 3500
50
+ },
51
+ {
52
+ "epoch": 4.64,
53
+ "learning_rate": 8.929909042268593e-06,
54
+ "loss": 0.8514,
55
+ "step": 4000
56
+ },
57
+ {
58
+ "epoch": 4.72,
59
+ "learning_rate": 8.796147672552167e-06,
60
+ "loss": 0.8476,
61
+ "step": 4500
62
+ },
63
+ {
64
+ "epoch": 4.8,
65
+ "learning_rate": 8.662386302835741e-06,
66
+ "loss": 0.8252,
67
+ "step": 5000
68
+ },
69
+ {
70
+ "epoch": 4.88,
71
+ "learning_rate": 8.528624933119315e-06,
72
+ "loss": 0.8658,
73
+ "step": 5500
74
+ },
75
+ {
76
+ "epoch": 4.96,
77
+ "learning_rate": 8.39486356340289e-06,
78
+ "loss": 0.8453,
79
+ "step": 6000
80
+ },
81
+ {
82
+ "epoch": 5.0,
83
+ "eval_gen_len": 18.796,
84
+ "eval_loss": 1.2918661832809448,
85
+ "eval_rouge1": 39.8436,
86
+ "eval_rouge2": 15.5632,
87
+ "eval_rougeL": 33.522,
88
+ "eval_rougeLsum": 35.2901,
89
+ "eval_runtime": 722.0768,
90
+ "eval_samples_per_second": 2.077,
91
+ "eval_steps_per_second": 1.039,
92
+ "step": 6230
93
+ }
94
+ ],
95
+ "max_steps": 37380,
96
+ "num_train_epochs": 6,
97
+ "total_flos": 1.48195611648e+16,
98
+ "trial_name": null,
99
+ "trial_params": null
100
+ }