BleachNick commited on
Commit
6a668c6
·
verified ·
1 Parent(s): fa0ed43

Model save

Browse files
README.md CHANGED
@@ -1,199 +1,58 @@
1
  ---
 
2
  library_name: transformers
3
- tags: []
 
 
 
 
 
4
  ---
5
 
6
- # Model Card for Model ID
7
 
8
- <!-- Provide a quick summary of what the model is/does. -->
 
9
 
 
10
 
 
 
11
 
12
- ## Model Details
 
 
 
 
13
 
14
- ### Model Description
15
 
16
- <!-- Provide a longer summary of what this model is. -->
17
 
18
- This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
19
 
20
- - **Developed by:** [More Information Needed]
21
- - **Funded by [optional]:** [More Information Needed]
22
- - **Shared by [optional]:** [More Information Needed]
23
- - **Model type:** [More Information Needed]
24
- - **Language(s) (NLP):** [More Information Needed]
25
- - **License:** [More Information Needed]
26
- - **Finetuned from model [optional]:** [More Information Needed]
27
 
28
- ### Model Sources [optional]
29
 
30
- <!-- Provide the basic links for the model. -->
 
 
 
 
31
 
32
- - **Repository:** [More Information Needed]
33
- - **Paper [optional]:** [More Information Needed]
34
- - **Demo [optional]:** [More Information Needed]
35
 
36
- ## Uses
37
 
38
- <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
39
 
40
- ### Direct Use
41
-
42
- <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
43
-
44
- [More Information Needed]
45
-
46
- ### Downstream Use [optional]
47
-
48
- <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
49
-
50
- [More Information Needed]
51
-
52
- ### Out-of-Scope Use
53
-
54
- <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
55
-
56
- [More Information Needed]
57
-
58
- ## Bias, Risks, and Limitations
59
-
60
- <!-- This section is meant to convey both technical and sociotechnical limitations. -->
61
-
62
- [More Information Needed]
63
-
64
- ### Recommendations
65
-
66
- <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
67
-
68
- Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
69
-
70
- ## How to Get Started with the Model
71
-
72
- Use the code below to get started with the model.
73
-
74
- [More Information Needed]
75
-
76
- ## Training Details
77
-
78
- ### Training Data
79
-
80
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
-
82
- [More Information Needed]
83
-
84
- ### Training Procedure
85
-
86
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
87
-
88
- #### Preprocessing [optional]
89
-
90
- [More Information Needed]
91
-
92
-
93
- #### Training Hyperparameters
94
-
95
- - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
96
-
97
- #### Speeds, Sizes, Times [optional]
98
-
99
- <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
100
-
101
- [More Information Needed]
102
-
103
- ## Evaluation
104
-
105
- <!-- This section describes the evaluation protocols and provides the results. -->
106
-
107
- ### Testing Data, Factors & Metrics
108
-
109
- #### Testing Data
110
-
111
- <!-- This should link to a Dataset Card if possible. -->
112
-
113
- [More Information Needed]
114
-
115
- #### Factors
116
-
117
- <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
118
-
119
- [More Information Needed]
120
-
121
- #### Metrics
122
-
123
- <!-- These are the evaluation metrics being used, ideally with a description of why. -->
124
-
125
- [More Information Needed]
126
-
127
- ### Results
128
-
129
- [More Information Needed]
130
-
131
- #### Summary
132
-
133
-
134
-
135
- ## Model Examination [optional]
136
-
137
- <!-- Relevant interpretability work for the model goes here -->
138
-
139
- [More Information Needed]
140
-
141
- ## Environmental Impact
142
-
143
- <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
144
-
145
- Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
146
-
147
- - **Hardware Type:** [More Information Needed]
148
- - **Hours used:** [More Information Needed]
149
- - **Cloud Provider:** [More Information Needed]
150
- - **Compute Region:** [More Information Needed]
151
- - **Carbon Emitted:** [More Information Needed]
152
-
153
- ## Technical Specifications [optional]
154
-
155
- ### Model Architecture and Objective
156
-
157
- [More Information Needed]
158
-
159
- ### Compute Infrastructure
160
-
161
- [More Information Needed]
162
-
163
- #### Hardware
164
-
165
- [More Information Needed]
166
-
167
- #### Software
168
-
169
- [More Information Needed]
170
-
171
- ## Citation [optional]
172
-
173
- <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
174
-
175
- **BibTeX:**
176
-
177
- [More Information Needed]
178
-
179
- **APA:**
180
-
181
- [More Information Needed]
182
-
183
- ## Glossary [optional]
184
-
185
- <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
186
-
187
- [More Information Needed]
188
-
189
- ## More Information [optional]
190
-
191
- [More Information Needed]
192
-
193
- ## Model Card Authors [optional]
194
-
195
- [More Information Needed]
196
-
197
- ## Model Card Contact
198
-
199
- [More Information Needed]
 
1
  ---
2
+ base_model: Qwen/Qwen2-VL-2B-Instruct
3
  library_name: transformers
4
+ model_name: Qwen2-VL-2B-Instruct-SFT-Clevr_CoGenT_TrainA_R1
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
  ---
11
 
12
+ # Model Card for Qwen2-VL-2B-Instruct-SFT-Clevr_CoGenT_TrainA_R1
13
 
14
+ This model is a fine-tuned version of [Qwen/Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
+ ## Quick start
18
 
19
+ ```python
20
+ from transformers import pipeline
21
 
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="BleachNick/Qwen2-VL-2B-Instruct-SFT-Clevr_CoGenT_TrainA_R1", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
 
28
+ ## Training procedure
29
 
30
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/haozhezhao/huggingface/runs/9hjq1oia)
31
 
 
32
 
33
+ This model was trained with SFT.
 
 
 
 
 
 
34
 
35
+ ### Framework versions
36
 
37
+ - TRL: 0.14.0
38
+ - Transformers: 4.49.0.dev0
39
+ - Pytorch: 2.5.1+cu121
40
+ - Datasets: 3.2.0
41
+ - Tokenizers: 0.21.0
42
 
43
+ ## Citations
 
 
44
 
 
45
 
 
46
 
47
+ Cite TRL as:
48
+
49
+ ```bibtex
50
+ @misc{vonwerra2022trl,
51
+ title = {{TRL: Transformer Reinforcement Learning}},
52
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
53
+ year = 2020,
54
+ journal = {GitHub repository},
55
+ publisher = {GitHub},
56
+ howpublished = {\url{https://github.com/huggingface/trl}}
57
+ }
58
+ ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.9991532599491956,
3
+ "total_flos": 8.380405739893555e+17,
4
+ "train_loss": 0.6345516301817813,
5
+ "train_runtime": 3216.2267,
6
+ "train_samples": 37773,
7
+ "train_samples_per_second": 11.745,
8
+ "train_steps_per_second": 0.092
9
+ }
config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "Qwen/Qwen2-VL-2B-Instruct",
3
+ "architectures": [
4
+ "Qwen2VLForConditionalGeneration"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "eos_token_id": 151645,
9
+ "hidden_act": "silu",
10
+ "hidden_size": 1536,
11
+ "image_token_id": 151655,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 8960,
14
+ "max_position_embeddings": 32768,
15
+ "max_window_layers": 28,
16
+ "model_type": "qwen2_vl",
17
+ "num_attention_heads": 12,
18
+ "num_hidden_layers": 28,
19
+ "num_key_value_heads": 2,
20
+ "rms_norm_eps": 1e-06,
21
+ "rope_scaling": {
22
+ "mrope_section": [
23
+ 16,
24
+ 24,
25
+ 24
26
+ ],
27
+ "rope_type": "default",
28
+ "type": "default"
29
+ },
30
+ "rope_theta": 1000000.0,
31
+ "sliding_window": 32768,
32
+ "tie_word_embeddings": true,
33
+ "torch_dtype": "bfloat16",
34
+ "transformers_version": "4.49.0.dev0",
35
+ "use_cache": false,
36
+ "use_sliding_window": false,
37
+ "video_token_id": 151656,
38
+ "vision_config": {
39
+ "hidden_size": 1536,
40
+ "in_chans": 3,
41
+ "model_type": "qwen2_vl",
42
+ "spatial_patch_size": 14,
43
+ "torch_dtype": "bfloat16"
44
+ },
45
+ "vision_end_token_id": 151653,
46
+ "vision_start_token_id": 151652,
47
+ "vision_token_id": 151654,
48
+ "vocab_size": 151936
49
+ }
generation_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attn_implementation": null,
3
+ "bos_token_id": 151643,
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 151645,
7
+ 151643
8
+ ],
9
+ "pad_token_id": 151643,
10
+ "temperature": 0.01,
11
+ "top_k": 1,
12
+ "top_p": 0.001,
13
+ "transformers_version": "4.49.0.dev0",
14
+ "use_cache": false
15
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e261fdc4bc23a8089af0f9990a21e4e755e0bb4f9af7a8368db050d82e8d910
3
+ size 4884798456
tokenizer_config.json CHANGED
@@ -138,7 +138,6 @@
138
  "model_max_length": 32768,
139
  "pad_token": "<|endoftext|>",
140
  "padding_side": "left",
141
- "processor_class": "Qwen2VLProcessor",
142
  "split_special_tokens": false,
143
  "tokenizer_class": "Qwen2Tokenizer",
144
  "unk_token": null
 
138
  "model_max_length": 32768,
139
  "pad_token": "<|endoftext|>",
140
  "padding_side": "left",
 
141
  "split_special_tokens": false,
142
  "tokenizer_class": "Qwen2Tokenizer",
143
  "unk_token": null
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.9991532599491956,
3
+ "total_flos": 8.380405739893555e+17,
4
+ "train_loss": 0.6345516301817813,
5
+ "train_runtime": 3216.2267,
6
+ "train_samples": 37773,
7
+ "train_samples_per_second": 11.745,
8
+ "train_steps_per_second": 0.092
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,455 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9991532599491956,
5
+ "eval_steps": 500,
6
+ "global_step": 295,
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.01693480101608806,
13
+ "grad_norm": 11.07681941986084,
14
+ "learning_rate": 3.3333333333333333e-06,
15
+ "loss": 1.8357,
16
+ "step": 5
17
+ },
18
+ {
19
+ "epoch": 0.03386960203217612,
20
+ "grad_norm": 4.368485450744629,
21
+ "learning_rate": 6.666666666666667e-06,
22
+ "loss": 1.6208,
23
+ "step": 10
24
+ },
25
+ {
26
+ "epoch": 0.05080440304826418,
27
+ "grad_norm": 2.597416877746582,
28
+ "learning_rate": 1e-05,
29
+ "loss": 1.2437,
30
+ "step": 15
31
+ },
32
+ {
33
+ "epoch": 0.06773920406435224,
34
+ "grad_norm": 2.159755229949951,
35
+ "learning_rate": 1.3333333333333333e-05,
36
+ "loss": 0.9804,
37
+ "step": 20
38
+ },
39
+ {
40
+ "epoch": 0.0846740050804403,
41
+ "grad_norm": 1.7200169563293457,
42
+ "learning_rate": 1.6666666666666667e-05,
43
+ "loss": 0.8607,
44
+ "step": 25
45
+ },
46
+ {
47
+ "epoch": 0.10160880609652836,
48
+ "grad_norm": 1.2486546039581299,
49
+ "learning_rate": 2e-05,
50
+ "loss": 0.7869,
51
+ "step": 30
52
+ },
53
+ {
54
+ "epoch": 0.11854360711261643,
55
+ "grad_norm": 1.2447178363800049,
56
+ "learning_rate": 1.9982437317643218e-05,
57
+ "loss": 0.7324,
58
+ "step": 35
59
+ },
60
+ {
61
+ "epoch": 0.1354784081287045,
62
+ "grad_norm": 1.129758358001709,
63
+ "learning_rate": 1.992981096013517e-05,
64
+ "loss": 0.7058,
65
+ "step": 40
66
+ },
67
+ {
68
+ "epoch": 0.15241320914479256,
69
+ "grad_norm": 0.9256328344345093,
70
+ "learning_rate": 1.984230577947597e-05,
71
+ "loss": 0.6845,
72
+ "step": 45
73
+ },
74
+ {
75
+ "epoch": 0.1693480101608806,
76
+ "grad_norm": 0.9811558723449707,
77
+ "learning_rate": 1.972022914080411e-05,
78
+ "loss": 0.6698,
79
+ "step": 50
80
+ },
81
+ {
82
+ "epoch": 0.18628281117696868,
83
+ "grad_norm": 1.0069135427474976,
84
+ "learning_rate": 1.9564009842765225e-05,
85
+ "loss": 0.6393,
86
+ "step": 55
87
+ },
88
+ {
89
+ "epoch": 0.20321761219305673,
90
+ "grad_norm": 0.9143001437187195,
91
+ "learning_rate": 1.9374196611341212e-05,
92
+ "loss": 0.6321,
93
+ "step": 60
94
+ },
95
+ {
96
+ "epoch": 0.2201524132091448,
97
+ "grad_norm": 0.9945001006126404,
98
+ "learning_rate": 1.9151456172430186e-05,
99
+ "loss": 0.6286,
100
+ "step": 65
101
+ },
102
+ {
103
+ "epoch": 0.23708721422523285,
104
+ "grad_norm": 0.9755650758743286,
105
+ "learning_rate": 1.8896570909947477e-05,
106
+ "loss": 0.6178,
107
+ "step": 70
108
+ },
109
+ {
110
+ "epoch": 0.2540220152413209,
111
+ "grad_norm": 0.9144276976585388,
112
+ "learning_rate": 1.8610436117673557e-05,
113
+ "loss": 0.6204,
114
+ "step": 75
115
+ },
116
+ {
117
+ "epoch": 0.270956816257409,
118
+ "grad_norm": 0.9252317547798157,
119
+ "learning_rate": 1.829405685450202e-05,
120
+ "loss": 0.6124,
121
+ "step": 80
122
+ },
123
+ {
124
+ "epoch": 0.28789161727349705,
125
+ "grad_norm": 0.7833272218704224,
126
+ "learning_rate": 1.7948544414133534e-05,
127
+ "loss": 0.5958,
128
+ "step": 85
129
+ },
130
+ {
131
+ "epoch": 0.3048264182895851,
132
+ "grad_norm": 0.8833147883415222,
133
+ "learning_rate": 1.7575112421616203e-05,
134
+ "loss": 0.6049,
135
+ "step": 90
136
+ },
137
+ {
138
+ "epoch": 0.32176121930567314,
139
+ "grad_norm": 0.8253676295280457,
140
+ "learning_rate": 1.717507257044331e-05,
141
+ "loss": 0.5905,
142
+ "step": 95
143
+ },
144
+ {
145
+ "epoch": 0.3386960203217612,
146
+ "grad_norm": 0.800396740436554,
147
+ "learning_rate": 1.6749830015182106e-05,
148
+ "loss": 0.5698,
149
+ "step": 100
150
+ },
151
+ {
152
+ "epoch": 0.3556308213378493,
153
+ "grad_norm": 0.9177567362785339,
154
+ "learning_rate": 1.6300878435817115e-05,
155
+ "loss": 0.5787,
156
+ "step": 105
157
+ },
158
+ {
159
+ "epoch": 0.37256562235393736,
160
+ "grad_norm": 0.8931817412376404,
161
+ "learning_rate": 1.5829794791144723e-05,
162
+ "loss": 0.5748,
163
+ "step": 110
164
+ },
165
+ {
166
+ "epoch": 0.3895004233700254,
167
+ "grad_norm": 1.0776097774505615,
168
+ "learning_rate": 1.533823377964791e-05,
169
+ "loss": 0.5693,
170
+ "step": 115
171
+ },
172
+ {
173
+ "epoch": 0.40643522438611346,
174
+ "grad_norm": 0.9571148157119751,
175
+ "learning_rate": 1.482792202730745e-05,
176
+ "loss": 0.5795,
177
+ "step": 120
178
+ },
179
+ {
180
+ "epoch": 0.42337002540220153,
181
+ "grad_norm": 0.9276224970817566,
182
+ "learning_rate": 1.4300652022765207e-05,
183
+ "loss": 0.5744,
184
+ "step": 125
185
+ },
186
+ {
187
+ "epoch": 0.4403048264182896,
188
+ "grad_norm": 0.8562670350074768,
189
+ "learning_rate": 1.3758275821142382e-05,
190
+ "loss": 0.574,
191
+ "step": 130
192
+ },
193
+ {
194
+ "epoch": 0.4572396274343776,
195
+ "grad_norm": 0.987328827381134,
196
+ "learning_rate": 1.3202698538628376e-05,
197
+ "loss": 0.5612,
198
+ "step": 135
199
+ },
200
+ {
201
+ "epoch": 0.4741744284504657,
202
+ "grad_norm": 0.7609542012214661,
203
+ "learning_rate": 1.2635871660690677e-05,
204
+ "loss": 0.5608,
205
+ "step": 140
206
+ },
207
+ {
208
+ "epoch": 0.4911092294665538,
209
+ "grad_norm": 0.752984881401062,
210
+ "learning_rate": 1.2059786187410984e-05,
211
+ "loss": 0.5462,
212
+ "step": 145
213
+ },
214
+ {
215
+ "epoch": 0.5080440304826418,
216
+ "grad_norm": 0.7577420473098755,
217
+ "learning_rate": 1.1476465640024814e-05,
218
+ "loss": 0.5565,
219
+ "step": 150
220
+ },
221
+ {
222
+ "epoch": 0.5249788314987299,
223
+ "grad_norm": 0.7515721321105957,
224
+ "learning_rate": 1.0887958953229349e-05,
225
+ "loss": 0.5475,
226
+ "step": 155
227
+ },
228
+ {
229
+ "epoch": 0.541913632514818,
230
+ "grad_norm": 0.7427639961242676,
231
+ "learning_rate": 1.0296333278225599e-05,
232
+ "loss": 0.5561,
233
+ "step": 160
234
+ },
235
+ {
236
+ "epoch": 0.558848433530906,
237
+ "grad_norm": 0.8810573816299438,
238
+ "learning_rate": 9.703666721774403e-06,
239
+ "loss": 0.5391,
240
+ "step": 165
241
+ },
242
+ {
243
+ "epoch": 0.5757832345469941,
244
+ "grad_norm": 0.7623007297515869,
245
+ "learning_rate": 9.112041046770653e-06,
246
+ "loss": 0.5504,
247
+ "step": 170
248
+ },
249
+ {
250
+ "epoch": 0.5927180355630821,
251
+ "grad_norm": 0.7749871015548706,
252
+ "learning_rate": 8.52353435997519e-06,
253
+ "loss": 0.5442,
254
+ "step": 175
255
+ },
256
+ {
257
+ "epoch": 0.6096528365791702,
258
+ "grad_norm": 0.731037974357605,
259
+ "learning_rate": 7.940213812589018e-06,
260
+ "loss": 0.5504,
261
+ "step": 180
262
+ },
263
+ {
264
+ "epoch": 0.6265876375952583,
265
+ "grad_norm": 0.6718469262123108,
266
+ "learning_rate": 7.364128339309326e-06,
267
+ "loss": 0.5461,
268
+ "step": 185
269
+ },
270
+ {
271
+ "epoch": 0.6435224386113463,
272
+ "grad_norm": 0.7169979214668274,
273
+ "learning_rate": 6.797301461371626e-06,
274
+ "loss": 0.5387,
275
+ "step": 190
276
+ },
277
+ {
278
+ "epoch": 0.6604572396274344,
279
+ "grad_norm": 0.6952066421508789,
280
+ "learning_rate": 6.241724178857621e-06,
281
+ "loss": 0.5391,
282
+ "step": 195
283
+ },
284
+ {
285
+ "epoch": 0.6773920406435224,
286
+ "grad_norm": 0.7324413657188416,
287
+ "learning_rate": 5.699347977234799e-06,
288
+ "loss": 0.5351,
289
+ "step": 200
290
+ },
291
+ {
292
+ "epoch": 0.6943268416596104,
293
+ "grad_norm": 0.7053231596946716,
294
+ "learning_rate": 5.172077972692553e-06,
295
+ "loss": 0.5393,
296
+ "step": 205
297
+ },
298
+ {
299
+ "epoch": 0.7112616426756986,
300
+ "grad_norm": 0.6486250758171082,
301
+ "learning_rate": 4.661766220352098e-06,
302
+ "loss": 0.5311,
303
+ "step": 210
304
+ },
305
+ {
306
+ "epoch": 0.7281964436917866,
307
+ "grad_norm": 0.6176902055740356,
308
+ "learning_rate": 4.170205208855281e-06,
309
+ "loss": 0.5342,
310
+ "step": 215
311
+ },
312
+ {
313
+ "epoch": 0.7451312447078747,
314
+ "grad_norm": 0.6902855634689331,
315
+ "learning_rate": 3.6991215641828903e-06,
316
+ "loss": 0.526,
317
+ "step": 220
318
+ },
319
+ {
320
+ "epoch": 0.7620660457239627,
321
+ "grad_norm": 0.6924082636833191,
322
+ "learning_rate": 3.250169984817897e-06,
323
+ "loss": 0.5393,
324
+ "step": 225
325
+ },
326
+ {
327
+ "epoch": 0.7790008467400508,
328
+ "grad_norm": 0.6927950978279114,
329
+ "learning_rate": 2.8249274295566863e-06,
330
+ "loss": 0.5372,
331
+ "step": 230
332
+ },
333
+ {
334
+ "epoch": 0.7959356477561389,
335
+ "grad_norm": 0.6544324159622192,
336
+ "learning_rate": 2.424887578383799e-06,
337
+ "loss": 0.53,
338
+ "step": 235
339
+ },
340
+ {
341
+ "epoch": 0.8128704487722269,
342
+ "grad_norm": 0.6218433976173401,
343
+ "learning_rate": 2.0514555858664663e-06,
344
+ "loss": 0.5383,
345
+ "step": 240
346
+ },
347
+ {
348
+ "epoch": 0.8298052497883149,
349
+ "grad_norm": 0.6293333172798157,
350
+ "learning_rate": 1.7059431454979825e-06,
351
+ "loss": 0.5242,
352
+ "step": 245
353
+ },
354
+ {
355
+ "epoch": 0.8467400508044031,
356
+ "grad_norm": 0.6629937291145325,
357
+ "learning_rate": 1.3895638823264447e-06,
358
+ "loss": 0.5283,
359
+ "step": 250
360
+ },
361
+ {
362
+ "epoch": 0.8636748518204911,
363
+ "grad_norm": 0.6638931632041931,
364
+ "learning_rate": 1.1034290900525279e-06,
365
+ "loss": 0.5346,
366
+ "step": 255
367
+ },
368
+ {
369
+ "epoch": 0.8806096528365792,
370
+ "grad_norm": 0.6154753565788269,
371
+ "learning_rate": 8.485438275698154e-07,
372
+ "loss": 0.5286,
373
+ "step": 260
374
+ },
375
+ {
376
+ "epoch": 0.8975444538526672,
377
+ "grad_norm": 0.5576867461204529,
378
+ "learning_rate": 6.258033886587911e-07,
379
+ "loss": 0.5197,
380
+ "step": 265
381
+ },
382
+ {
383
+ "epoch": 0.9144792548687553,
384
+ "grad_norm": 0.5925081372261047,
385
+ "learning_rate": 4.359901572347758e-07,
386
+ "loss": 0.5272,
387
+ "step": 270
388
+ },
389
+ {
390
+ "epoch": 0.9314140558848434,
391
+ "grad_norm": 0.8894816040992737,
392
+ "learning_rate": 2.7977085919589253e-07,
393
+ "loss": 0.5229,
394
+ "step": 275
395
+ },
396
+ {
397
+ "epoch": 0.9483488569009314,
398
+ "grad_norm": 0.613675057888031,
399
+ "learning_rate": 1.5769422052403172e-07,
400
+ "loss": 0.5336,
401
+ "step": 280
402
+ },
403
+ {
404
+ "epoch": 0.9652836579170194,
405
+ "grad_norm": 0.5702030658721924,
406
+ "learning_rate": 7.018903986483083e-08,
407
+ "loss": 0.5295,
408
+ "step": 285
409
+ },
410
+ {
411
+ "epoch": 0.9822184589331076,
412
+ "grad_norm": 0.6007622480392456,
413
+ "learning_rate": 1.7562682356786488e-08,
414
+ "loss": 0.5357,
415
+ "step": 290
416
+ },
417
+ {
418
+ "epoch": 0.9991532599491956,
419
+ "grad_norm": 0.5643032193183899,
420
+ "learning_rate": 0.0,
421
+ "loss": 0.5246,
422
+ "step": 295
423
+ },
424
+ {
425
+ "epoch": 0.9991532599491956,
426
+ "step": 295,
427
+ "total_flos": 8.380405739893555e+17,
428
+ "train_loss": 0.6345516301817813,
429
+ "train_runtime": 3216.2267,
430
+ "train_samples_per_second": 11.745,
431
+ "train_steps_per_second": 0.092
432
+ }
433
+ ],
434
+ "logging_steps": 5,
435
+ "max_steps": 295,
436
+ "num_input_tokens_seen": 0,
437
+ "num_train_epochs": 1,
438
+ "save_steps": 500,
439
+ "stateful_callbacks": {
440
+ "TrainerControl": {
441
+ "args": {
442
+ "should_epoch_stop": false,
443
+ "should_evaluate": false,
444
+ "should_log": false,
445
+ "should_save": false,
446
+ "should_training_stop": false
447
+ },
448
+ "attributes": {}
449
+ }
450
+ },
451
+ "total_flos": 8.380405739893555e+17,
452
+ "train_batch_size": 4,
453
+ "trial_name": null,
454
+ "trial_params": null
455
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3810921b32631cca3754c3b8840ae85e3f3f5d65d48e95c60d784ac85c960652
3
+ size 7224