初始化项目,由ModelHub XC社区提供模型
Model: arnav-yadav/jailbreak-attacker-l2 Source: Original Platform
This commit is contained in:
210
checkpoint-150/README.md
Normal file
210
checkpoint-150/README.md
Normal file
@@ -0,0 +1,210 @@
|
||||
---
|
||||
base_model: unsloth/qwen2.5-1.5b-instruct-unsloth-bnb-4bit
|
||||
library_name: peft
|
||||
pipeline_tag: text-generation
|
||||
tags:
|
||||
- base_model:adapter:unsloth/qwen2.5-1.5b-instruct-unsloth-bnb-4bit
|
||||
- grpo
|
||||
- lora
|
||||
- transformers
|
||||
- trl
|
||||
- unsloth
|
||||
---
|
||||
|
||||
# Model Card for Model ID
|
||||
|
||||
<!-- Provide a quick summary of what the model is/does. -->
|
||||
|
||||
|
||||
|
||||
## Model Details
|
||||
|
||||
### Model Description
|
||||
|
||||
<!-- Provide a longer summary of what this model is. -->
|
||||
|
||||
|
||||
|
||||
- **Developed by:** [More Information Needed]
|
||||
- **Funded by [optional]:** [More Information Needed]
|
||||
- **Shared by [optional]:** [More Information Needed]
|
||||
- **Model type:** [More Information Needed]
|
||||
- **Language(s) (NLP):** [More Information Needed]
|
||||
- **License:** [More Information Needed]
|
||||
- **Finetuned from model [optional]:** [More Information Needed]
|
||||
|
||||
### Model Sources [optional]
|
||||
|
||||
<!-- Provide the basic links for the model. -->
|
||||
|
||||
- **Repository:** [More Information Needed]
|
||||
- **Paper [optional]:** [More Information Needed]
|
||||
- **Demo [optional]:** [More Information Needed]
|
||||
|
||||
## Uses
|
||||
|
||||
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
||||
|
||||
### Direct Use
|
||||
|
||||
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
### Downstream Use [optional]
|
||||
|
||||
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
### Out-of-Scope Use
|
||||
|
||||
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Bias, Risks, and Limitations
|
||||
|
||||
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
### Recommendations
|
||||
|
||||
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
||||
|
||||
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
||||
|
||||
## How to Get Started with the Model
|
||||
|
||||
Use the code below to get started with the model.
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Training Details
|
||||
|
||||
### Training Data
|
||||
|
||||
<!-- 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. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
### Training Procedure
|
||||
|
||||
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
||||
|
||||
#### Preprocessing [optional]
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
|
||||
#### Training Hyperparameters
|
||||
|
||||
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
||||
|
||||
#### Speeds, Sizes, Times [optional]
|
||||
|
||||
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Evaluation
|
||||
|
||||
<!-- This section describes the evaluation protocols and provides the results. -->
|
||||
|
||||
### Testing Data, Factors & Metrics
|
||||
|
||||
#### Testing Data
|
||||
|
||||
<!-- This should link to a Dataset Card if possible. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
#### Factors
|
||||
|
||||
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
#### Metrics
|
||||
|
||||
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
### Results
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
#### Summary
|
||||
|
||||
|
||||
|
||||
## Model Examination [optional]
|
||||
|
||||
<!-- Relevant interpretability work for the model goes here -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Environmental Impact
|
||||
|
||||
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
||||
|
||||
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).
|
||||
|
||||
- **Hardware Type:** [More Information Needed]
|
||||
- **Hours used:** [More Information Needed]
|
||||
- **Cloud Provider:** [More Information Needed]
|
||||
- **Compute Region:** [More Information Needed]
|
||||
- **Carbon Emitted:** [More Information Needed]
|
||||
|
||||
## Technical Specifications [optional]
|
||||
|
||||
### Model Architecture and Objective
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
### Compute Infrastructure
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
#### Hardware
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
#### Software
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Citation [optional]
|
||||
|
||||
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
||||
|
||||
**BibTeX:**
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
**APA:**
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Glossary [optional]
|
||||
|
||||
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## More Information [optional]
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Model Card Authors [optional]
|
||||
|
||||
[More Information Needed]
|
||||
|
||||
## Model Card Contact
|
||||
|
||||
[More Information Needed]
|
||||
### Framework versions
|
||||
|
||||
- PEFT 0.19.1
|
||||
49
checkpoint-150/adapter_config.json
Normal file
49
checkpoint-150/adapter_config.json
Normal file
@@ -0,0 +1,49 @@
|
||||
{
|
||||
"alora_invocation_tokens": null,
|
||||
"alpha_pattern": {},
|
||||
"arrow_config": null,
|
||||
"auto_mapping": {
|
||||
"base_model_class": "Qwen2ForCausalLM",
|
||||
"parent_library": "transformers.models.qwen2.modeling_qwen2",
|
||||
"unsloth_fixed": true
|
||||
},
|
||||
"base_model_name_or_path": "unsloth/qwen2.5-1.5b-instruct-unsloth-bnb-4bit",
|
||||
"bias": "none",
|
||||
"corda_config": null,
|
||||
"ensure_weight_tying": false,
|
||||
"eva_config": null,
|
||||
"exclude_modules": null,
|
||||
"fan_in_fan_out": false,
|
||||
"inference_mode": true,
|
||||
"init_lora_weights": true,
|
||||
"layer_replication": null,
|
||||
"layers_pattern": null,
|
||||
"layers_to_transform": null,
|
||||
"loftq_config": {},
|
||||
"lora_alpha": 16,
|
||||
"lora_bias": false,
|
||||
"lora_dropout": 0,
|
||||
"lora_ga_config": null,
|
||||
"megatron_config": null,
|
||||
"megatron_core": "megatron.core",
|
||||
"modules_to_save": null,
|
||||
"peft_type": "LORA",
|
||||
"peft_version": "0.19.1",
|
||||
"qalora_group_size": 16,
|
||||
"r": 16,
|
||||
"rank_pattern": {},
|
||||
"revision": null,
|
||||
"target_modules": [
|
||||
"q_proj",
|
||||
"k_proj",
|
||||
"v_proj",
|
||||
"o_proj"
|
||||
],
|
||||
"target_parameters": null,
|
||||
"task_type": "CAUSAL_LM",
|
||||
"trainable_token_indices": null,
|
||||
"use_bdlora": null,
|
||||
"use_dora": false,
|
||||
"use_qalora": false,
|
||||
"use_rslora": false
|
||||
}
|
||||
3
checkpoint-150/adapter_model.safetensors
Normal file
3
checkpoint-150/adapter_model.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:266eb8f6b5685d94013a7e3f7915eaf1b745d68327c70dca9737731fd1ffcf5c
|
||||
size 17462432
|
||||
54
checkpoint-150/chat_template.jinja
Normal file
54
checkpoint-150/chat_template.jinja
Normal file
@@ -0,0 +1,54 @@
|
||||
{%- if tools %}
|
||||
{{- '<|im_start|>system\n' }}
|
||||
{%- if messages[0]['role'] == 'system' %}
|
||||
{{- messages[0]['content'] }}
|
||||
{%- else %}
|
||||
{{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}
|
||||
{%- endif %}
|
||||
{{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
|
||||
{%- for tool in tools %}
|
||||
{{- "\n" }}
|
||||
{{- tool | tojson }}
|
||||
{%- endfor %}
|
||||
{{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
|
||||
{%- else %}
|
||||
{%- if messages[0]['role'] == 'system' %}
|
||||
{{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
|
||||
{%- else %}
|
||||
{{- '<|im_start|>system\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\n' }}
|
||||
{%- endif %}
|
||||
{%- endif %}
|
||||
{%- for message in messages %}
|
||||
{%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
|
||||
{{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
|
||||
{%- elif message.role == "assistant" %}
|
||||
{{- '<|im_start|>' + message.role }}
|
||||
{%- if message.content %}
|
||||
{{- '\n' + message.content }}
|
||||
{%- endif %}
|
||||
{%- for tool_call in message.tool_calls %}
|
||||
{%- if tool_call.function is defined %}
|
||||
{%- set tool_call = tool_call.function %}
|
||||
{%- endif %}
|
||||
{{- '\n<tool_call>\n{"name": "' }}
|
||||
{{- tool_call.name }}
|
||||
{{- '", "arguments": ' }}
|
||||
{{- tool_call.arguments | tojson }}
|
||||
{{- '}\n</tool_call>' }}
|
||||
{%- endfor %}
|
||||
{{- '<|im_end|>\n' }}
|
||||
{%- elif message.role == "tool" %}
|
||||
{%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
|
||||
{{- '<|im_start|>user' }}
|
||||
{%- endif %}
|
||||
{{- '\n<tool_response>\n' }}
|
||||
{{- message.content }}
|
||||
{{- '\n</tool_response>' }}
|
||||
{%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
|
||||
{{- '<|im_end|>\n' }}
|
||||
{%- endif %}
|
||||
{%- endif %}
|
||||
{%- endfor %}
|
||||
{%- if add_generation_prompt %}
|
||||
{{- '<|im_start|>assistant\n' }}
|
||||
{%- endif %}
|
||||
3
checkpoint-150/optimizer.pt
Normal file
3
checkpoint-150/optimizer.pt
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:2f18a375d42fddfe97539e8c114dfaeb434882c99c0f4d3d54c09b0f8fd3d090
|
||||
size 9117451
|
||||
3
checkpoint-150/rng_state.pth
Normal file
3
checkpoint-150/rng_state.pth
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:5611ccab0729462d7748ff6ecab8ec349d7a0ac4719c59e381dc9f57aac776a9
|
||||
size 14645
|
||||
3
checkpoint-150/scaler.pt
Normal file
3
checkpoint-150/scaler.pt
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:b7a646a858b71ab76099f4dca96d8db39ffde2f88258f5a4b082ff62947bcc11
|
||||
size 1383
|
||||
3
checkpoint-150/scheduler.pt
Normal file
3
checkpoint-150/scheduler.pt
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:8206307394a449e70d8cda749820dc5f2c93cd0fc51958acc50e593295c4ad40
|
||||
size 1465
|
||||
3
checkpoint-150/tokenizer.json
Normal file
3
checkpoint-150/tokenizer.json
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:af5891a15588546db1ac7f2baf8fa94835a51a85c032c39793a55bb048b47446
|
||||
size 11422523
|
||||
201
checkpoint-150/tokenizer_config.json
Normal file
201
checkpoint-150/tokenizer_config.json
Normal file
@@ -0,0 +1,201 @@
|
||||
{
|
||||
"add_prefix_space": false,
|
||||
"backend": "tokenizers",
|
||||
"bos_token": null,
|
||||
"clean_up_tokenization_spaces": false,
|
||||
"eos_token": "<|im_end|>",
|
||||
"errors": "replace",
|
||||
"is_local": false,
|
||||
"model_max_length": 32768,
|
||||
"pad_token": "<|PAD_TOKEN|>",
|
||||
"padding_side": "right",
|
||||
"split_special_tokens": false,
|
||||
"tokenizer_class": "Qwen2Tokenizer",
|
||||
"unk_token": null,
|
||||
"added_tokens_decoder": {
|
||||
"151643": {
|
||||
"content": "<|endoftext|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151644": {
|
||||
"content": "<|im_start|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151645": {
|
||||
"content": "<|im_end|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151646": {
|
||||
"content": "<|object_ref_start|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151647": {
|
||||
"content": "<|object_ref_end|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151648": {
|
||||
"content": "<|box_start|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151649": {
|
||||
"content": "<|box_end|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151650": {
|
||||
"content": "<|quad_start|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151651": {
|
||||
"content": "<|quad_end|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151652": {
|
||||
"content": "<|vision_start|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151653": {
|
||||
"content": "<|vision_end|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151654": {
|
||||
"content": "<|vision_pad|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151655": {
|
||||
"content": "<|image_pad|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151656": {
|
||||
"content": "<|video_pad|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
},
|
||||
"151657": {
|
||||
"content": "<tool_call>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151658": {
|
||||
"content": "</tool_call>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151659": {
|
||||
"content": "<|fim_prefix|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151660": {
|
||||
"content": "<|fim_middle|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151661": {
|
||||
"content": "<|fim_suffix|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151662": {
|
||||
"content": "<|fim_pad|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151663": {
|
||||
"content": "<|repo_name|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151664": {
|
||||
"content": "<|file_sep|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": false
|
||||
},
|
||||
"151665": {
|
||||
"content": "<|PAD_TOKEN|>",
|
||||
"single_word": false,
|
||||
"lstrip": false,
|
||||
"rstrip": false,
|
||||
"normalized": false,
|
||||
"special": true
|
||||
}
|
||||
}
|
||||
}
|
||||
439
checkpoint-150/trainer_state.json
Normal file
439
checkpoint-150/trainer_state.json
Normal file
@@ -0,0 +1,439 @@
|
||||
{
|
||||
"best_global_step": null,
|
||||
"best_metric": null,
|
||||
"best_model_checkpoint": null,
|
||||
"epoch": 0.9375,
|
||||
"eval_steps": 500,
|
||||
"global_step": 150,
|
||||
"is_hyper_param_search": false,
|
||||
"is_local_process_zero": true,
|
||||
"is_world_process_zero": true,
|
||||
"log_history": [
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.0625,
|
||||
"frac_reward_zero_std": 0.075,
|
||||
"grad_norm": 0.12458323687314987,
|
||||
"kl": 1.9381546007934957e-05,
|
||||
"learning_rate": 9.375000000000001e-06,
|
||||
"loss": 3.427267074584961e-08,
|
||||
"num_tokens": 57908.0,
|
||||
"reward": -1.4099999368190765,
|
||||
"reward_std": 0.7305359303951263,
|
||||
"rewards/reward_fn/mean": -1.409999930858612,
|
||||
"rewards/reward_fn/std": 0.8336655974388123,
|
||||
"step": 10
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.125,
|
||||
"frac_reward_zero_std": 0.05,
|
||||
"grad_norm": 0.13424909114837646,
|
||||
"kl": 3.758955572266132e-05,
|
||||
"learning_rate": 1.9791666666666665e-05,
|
||||
"loss": 2.0861625671386717e-08,
|
||||
"num_tokens": 115744.0,
|
||||
"reward": -1.5606249690055847,
|
||||
"reward_std": 0.6338691949844361,
|
||||
"rewards/reward_fn/mean": -1.5606249451637269,
|
||||
"rewards/reward_fn/std": 0.6913031578063965,
|
||||
"step": 20
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.1875,
|
||||
"frac_reward_zero_std": 0.025,
|
||||
"grad_norm": 0.12381561845541,
|
||||
"kl": 0.000246869730472099,
|
||||
"learning_rate": 3.0208333333333334e-05,
|
||||
"loss": 2.4400651454925537e-07,
|
||||
"num_tokens": 173524.0,
|
||||
"reward": -1.4581249356269836,
|
||||
"reward_std": 0.7430005550384522,
|
||||
"rewards/reward_fn/mean": -1.4581249117851258,
|
||||
"rewards/reward_fn/std": 0.7565619319677352,
|
||||
"step": 30
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.25,
|
||||
"frac_reward_zero_std": 0.025,
|
||||
"grad_norm": 0.14411121606826782,
|
||||
"kl": 0.0016107790230307727,
|
||||
"learning_rate": 4.0625000000000005e-05,
|
||||
"loss": 1.6100704669952392e-06,
|
||||
"num_tokens": 231416.0,
|
||||
"reward": -1.3331249535083771,
|
||||
"reward_std": 0.6972330152988434,
|
||||
"rewards/reward_fn/mean": -1.3331249415874482,
|
||||
"rewards/reward_fn/std": 0.766948527097702,
|
||||
"step": 40
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.3125,
|
||||
"frac_reward_zero_std": 0.175,
|
||||
"grad_norm": 0.14639103412628174,
|
||||
"kl": 0.007588638024753891,
|
||||
"learning_rate": 4.988425925925926e-05,
|
||||
"loss": 7.59512186050415e-06,
|
||||
"num_tokens": 289368.0,
|
||||
"reward": -1.0006249606609345,
|
||||
"reward_std": 0.6684236854314805,
|
||||
"rewards/reward_fn/mean": -1.0006249785423278,
|
||||
"rewards/reward_fn/std": 0.7909055948257446,
|
||||
"step": 50
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.375,
|
||||
"frac_reward_zero_std": 0.3,
|
||||
"grad_norm": 0.1545860469341278,
|
||||
"kl": 0.024985690240282566,
|
||||
"learning_rate": 4.8726851851851855e-05,
|
||||
"loss": 2.498328685760498e-05,
|
||||
"num_tokens": 346964.0,
|
||||
"reward": -0.6074999988079071,
|
||||
"reward_std": 0.5110155880451203,
|
||||
"rewards/reward_fn/mean": -0.6075000107288361,
|
||||
"rewards/reward_fn/std": 0.6012047231197357,
|
||||
"step": 60
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.4375,
|
||||
"frac_reward_zero_std": 0.725,
|
||||
"grad_norm": 0.12491069734096527,
|
||||
"kl": 0.042862199200317266,
|
||||
"learning_rate": 4.756944444444444e-05,
|
||||
"loss": 4.285752947907895e-05,
|
||||
"num_tokens": 404676.0,
|
||||
"reward": -0.41250000298023226,
|
||||
"reward_std": 0.19732050001621246,
|
||||
"rewards/reward_fn/mean": -0.41250000298023226,
|
||||
"rewards/reward_fn/std": 0.3099328398704529,
|
||||
"step": 70
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.5,
|
||||
"frac_reward_zero_std": 0.825,
|
||||
"grad_norm": 0.139762744307518,
|
||||
"kl": 0.05795831000432372,
|
||||
"learning_rate": 4.6412037037037034e-05,
|
||||
"loss": 5.795210599899292e-05,
|
||||
"num_tokens": 462404.0,
|
||||
"reward": -0.3443750023841858,
|
||||
"reward_std": 0.11552571803331375,
|
||||
"rewards/reward_fn/mean": -0.344375005364418,
|
||||
"rewards/reward_fn/std": 0.1971571922302246,
|
||||
"step": 80
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.5625,
|
||||
"frac_reward_zero_std": 0.9,
|
||||
"grad_norm": 0.0012635246384888887,
|
||||
"kl": 0.0758059567771852,
|
||||
"learning_rate": 4.525462962962963e-05,
|
||||
"loss": 7.580451783724129e-05,
|
||||
"num_tokens": 520292.0,
|
||||
"reward": -0.341250005364418,
|
||||
"reward_std": 0.06749999970197677,
|
||||
"rewards/reward_fn/mean": -0.34125000834465025,
|
||||
"rewards/reward_fn/std": 0.14239275455474854,
|
||||
"step": 90
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.625,
|
||||
"frac_reward_zero_std": 0.925,
|
||||
"grad_norm": 0.28265202045440674,
|
||||
"kl": 0.10205610543489456,
|
||||
"learning_rate": 4.4097222222222226e-05,
|
||||
"loss": 0.00010205432772636414,
|
||||
"num_tokens": 578164.0,
|
||||
"reward": -0.31062501221895217,
|
||||
"reward_std": 0.05374999940395355,
|
||||
"rewards/reward_fn/mean": -0.31062501221895217,
|
||||
"rewards/reward_fn/std": 0.0848927527666092,
|
||||
"step": 100
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.6875,
|
||||
"frac_reward_zero_std": 0.925,
|
||||
"grad_norm": 0.15435293316841125,
|
||||
"kl": 0.11740942746400833,
|
||||
"learning_rate": 4.293981481481482e-05,
|
||||
"loss": 0.00011740787886083126,
|
||||
"num_tokens": 635928.0,
|
||||
"reward": -0.3262500077486038,
|
||||
"reward_std": 0.052499999105930326,
|
||||
"rewards/reward_fn/mean": -0.3262500077486038,
|
||||
"rewards/reward_fn/std": 0.10499999821186065,
|
||||
"step": 110
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.75,
|
||||
"frac_reward_zero_std": 0.975,
|
||||
"grad_norm": 0.0017832271987572312,
|
||||
"kl": 0.1092877883464098,
|
||||
"learning_rate": 4.178240740740741e-05,
|
||||
"loss": 0.0001092874095775187,
|
||||
"num_tokens": 693776.0,
|
||||
"reward": -0.31125001013278963,
|
||||
"reward_std": 0.02249999940395355,
|
||||
"rewards/reward_fn/mean": -0.31125001013278963,
|
||||
"rewards/reward_fn/std": 0.04499999582767487,
|
||||
"step": 120
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.8125,
|
||||
"frac_reward_zero_std": 0.925,
|
||||
"grad_norm": 0.0019980475772172213,
|
||||
"kl": 0.1017705887556076,
|
||||
"learning_rate": 4.0625000000000005e-05,
|
||||
"loss": 0.00010176897048950196,
|
||||
"num_tokens": 751704.0,
|
||||
"reward": -0.3262500077486038,
|
||||
"reward_std": 0.05249999761581421,
|
||||
"rewards/reward_fn/mean": -0.3262500077486038,
|
||||
"rewards/reward_fn/std": 0.10499999821186065,
|
||||
"step": 130
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.875,
|
||||
"frac_reward_zero_std": 0.975,
|
||||
"grad_norm": 0.006963597144931555,
|
||||
"kl": 0.10655302237719297,
|
||||
"learning_rate": 3.94675925925926e-05,
|
||||
"loss": 0.00010655303485691547,
|
||||
"num_tokens": 809360.0,
|
||||
"reward": -0.31125001013278963,
|
||||
"reward_std": 0.02249999940395355,
|
||||
"rewards/reward_fn/mean": -0.31125001013278963,
|
||||
"rewards/reward_fn/std": 0.04499999582767487,
|
||||
"step": 140
|
||||
},
|
||||
{
|
||||
"clip_ratio/high_max": 0.0,
|
||||
"clip_ratio/high_mean": 0.0,
|
||||
"clip_ratio/low_mean": 0.0,
|
||||
"clip_ratio/low_min": 0.0,
|
||||
"clip_ratio/region_mean": 0.0,
|
||||
"completion_length": 256.0,
|
||||
"completions/clipped_ratio": 1.0,
|
||||
"completions/max_length": 256.0,
|
||||
"completions/max_terminated_length": 0.0,
|
||||
"completions/mean_length": 256.0,
|
||||
"completions/mean_terminated_length": 0.0,
|
||||
"completions/min_length": 256.0,
|
||||
"completions/min_terminated_length": 0.0,
|
||||
"epoch": 0.9375,
|
||||
"frac_reward_zero_std": 0.925,
|
||||
"grad_norm": 0.0015019988641142845,
|
||||
"kl": 0.10568692134693265,
|
||||
"learning_rate": 3.831018518518519e-05,
|
||||
"loss": 0.00010568510042503476,
|
||||
"num_tokens": 867440.0,
|
||||
"reward": -0.33375000655651094,
|
||||
"reward_std": 0.06749999821186066,
|
||||
"rewards/reward_fn/mean": -0.33375000655651094,
|
||||
"rewards/reward_fn/std": 0.1349999874830246,
|
||||
"step": 150
|
||||
}
|
||||
],
|
||||
"logging_steps": 10,
|
||||
"max_steps": 480,
|
||||
"num_input_tokens_seen": 867440,
|
||||
"num_train_epochs": 3,
|
||||
"save_steps": 50,
|
||||
"stateful_callbacks": {
|
||||
"TrainerControl": {
|
||||
"args": {
|
||||
"should_epoch_stop": false,
|
||||
"should_evaluate": false,
|
||||
"should_log": false,
|
||||
"should_save": true,
|
||||
"should_training_stop": false
|
||||
},
|
||||
"attributes": {}
|
||||
}
|
||||
},
|
||||
"total_flos": 0.0,
|
||||
"train_batch_size": 4,
|
||||
"trial_name": null,
|
||||
"trial_params": null
|
||||
}
|
||||
3
checkpoint-150/training_args.bin
Normal file
3
checkpoint-150/training_args.bin
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:9bb5eec9fa732effecc39928ed0f7eda205a0e16573a345963766968a1a8e3e2
|
||||
size 6737
|
||||
Reference in New Issue
Block a user