Model save
Browse files- README.md +40 -181
- all_results.json +9 -0
- config.json +49 -0
- generation_config.json +15 -0
- model.safetensors +3 -0
- tokenizer_config.json +0 -1
- train_results.json +9 -0
- trainer_state.json +455 -0
- training_args.bin +3 -0
README.md
CHANGED
@@ -1,199 +1,58 @@
|
|
1 |
---
|
|
|
2 |
library_name: transformers
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
4 |
---
|
5 |
|
6 |
-
# Model Card for
|
7 |
|
8 |
-
|
|
|
9 |
|
|
|
10 |
|
|
|
|
|
11 |
|
12 |
-
|
|
|
|
|
|
|
|
|
13 |
|
14 |
-
|
15 |
|
16 |
-
|
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 |
-
|
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 |
-
###
|
29 |
|
30 |
-
|
|
|
|
|
|
|
|
|
31 |
|
32 |
-
|
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 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
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
|