yeok commited on
Commit
55ad3fe
·
verified ·
1 Parent(s): bc40756

End of training

Browse files
Files changed (4) hide show
  1. README.md +5 -3
  2. all_results.json +5 -0
  3. config.json +29 -0
  4. eval_results.json +7 -0
README.md CHANGED
@@ -1,17 +1,19 @@
1
  ---
2
  base_model: Qwen/Qwen2.5-1.5B-Instruct
 
3
  library_name: transformers
4
- model_name: qwen-2.5-1.5B-instruct-sft-lora-countdown-search-5k
5
  tags:
6
  - generated_from_trainer
 
7
  - trl
8
  - sft
9
  licence: license
10
  ---
11
 
12
- # Model Card for qwen-2.5-1.5B-instruct-sft-lora-countdown-search-5k
13
 
14
- This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
 
1
  ---
2
  base_model: Qwen/Qwen2.5-1.5B-Instruct
3
+ datasets: MelinaLaimon/stream-of-search
4
  library_name: transformers
5
+ model_name: Qwen/Qwen2.5-1.5B-Instruct
6
  tags:
7
  - generated_from_trainer
8
+ - alignment-handbook
9
  - trl
10
  - sft
11
  licence: license
12
  ---
13
 
14
+ # Model Card for Qwen/Qwen2.5-1.5B-Instruct
15
 
16
+ This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct) on the [MelinaLaimon/stream-of-search](https://huggingface.co/datasets/MelinaLaimon/stream-of-search) dataset.
17
  It has been trained using [TRL](https://github.com/huggingface/trl).
18
 
19
  ## Quick start
all_results.json CHANGED
@@ -1,4 +1,9 @@
1
  {
 
 
 
 
 
2
  "total_flos": 2.4518665439700582e+17,
3
  "train_loss": 0.05413007871165959,
4
  "train_runtime": 8945.5939,
 
1
  {
2
+ "eval_loss": 0.04217606037855148,
3
+ "eval_runtime": 980.4186,
4
+ "eval_samples": 2000,
5
+ "eval_samples_per_second": 3.163,
6
+ "eval_steps_per_second": 1.582,
7
  "total_flos": 2.4518665439700582e+17,
8
  "train_loss": 0.05413007871165959,
9
  "train_runtime": 8945.5939,
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "architectures": [
4
+ "Qwen2ForCausalLM"
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
+ "initializer_range": 0.02,
12
+ "intermediate_size": 8960,
13
+ "max_position_embeddings": 32768,
14
+ "max_window_layers": 21,
15
+ "model_type": "qwen2",
16
+ "num_attention_heads": 12,
17
+ "num_hidden_layers": 28,
18
+ "num_key_value_heads": 2,
19
+ "rms_norm_eps": 1e-06,
20
+ "rope_scaling": null,
21
+ "rope_theta": 1000000.0,
22
+ "sliding_window": 32768,
23
+ "tie_word_embeddings": true,
24
+ "torch_dtype": "bfloat16",
25
+ "transformers_version": "4.50.0",
26
+ "use_cache": true,
27
+ "use_sliding_window": false,
28
+ "vocab_size": 151936
29
+ }
eval_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "eval_loss": 0.04217606037855148,
3
+ "eval_runtime": 980.4186,
4
+ "eval_samples": 2000,
5
+ "eval_samples_per_second": 3.163,
6
+ "eval_steps_per_second": 1.582
7
+ }