mingye94 commited on
Commit
f7888a9
·
verified ·
1 Parent(s): 8e1465d

End of training

Browse files
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: llama3
3
- base_model: meta-llama/Meta-Llama-3-8B-Instruct
4
  tags:
5
  - trl
6
  - reward-trainer
@@ -17,10 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # rm_llama3_8B_helpsteer2
19
 
20
- This model is a fine-tuned version of [meta-llama/Meta-Llama-3-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.6931
23
- - Accuracy: 1.0
24
 
25
  ## Model description
26
 
@@ -39,25 +39,31 @@ More information needed
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
- - learning_rate: 1e-05
43
  - train_batch_size: 32
44
  - eval_batch_size: 8
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 10
49
- - num_epochs: 1
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:------:|:----:|:---------------:|:--------:|
55
- | 0.6931 | 0.1572 | 50 | 0.6931 | 1.0 |
56
- | 0.6931 | 0.3145 | 100 | 0.6931 | 1.0 |
57
- | 0.6931 | 0.4717 | 150 | 0.6931 | 1.0 |
58
- | 0.6931 | 0.6289 | 200 | 0.6931 | 1.0 |
59
- | 0.6931 | 0.7862 | 250 | 0.6931 | 1.0 |
60
- | 0.6931 | 0.9434 | 300 | 0.6931 | 1.0 |
 
 
 
 
 
 
61
 
62
 
63
  ### Framework versions
 
1
  ---
2
  license: llama3
3
+ base_model: meta-llama/Meta-Llama-3-8B
4
  tags:
5
  - trl
6
  - reward-trainer
 
17
 
18
  # rm_llama3_8B_helpsteer2
19
 
20
+ This model is a fine-tuned version of [meta-llama/Meta-Llama-3-8B](https://huggingface.co/meta-llama/Meta-Llama-3-8B) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.6567
23
+ - Accuracy: 0.6416
24
 
25
  ## Model description
26
 
 
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
+ - learning_rate: 5e-06
43
  - train_batch_size: 32
44
  - eval_batch_size: 8
45
  - seed: 42
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 10
49
+ - num_epochs: 2
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:------:|:----:|:---------------:|:--------:|
55
+ | 0.7676 | 0.1572 | 50 | 0.7708 | 0.5183 |
56
+ | 0.7583 | 0.3145 | 100 | 0.7717 | 0.6012 |
57
+ | 0.7383 | 0.4717 | 150 | 0.7041 | 0.6031 |
58
+ | 0.6935 | 0.6289 | 200 | 0.6895 | 0.5954 |
59
+ | 0.662 | 0.7862 | 250 | 0.6750 | 0.6262 |
60
+ | 0.6563 | 0.9434 | 300 | 0.6657 | 0.6339 |
61
+ | 0.5923 | 1.1006 | 350 | 0.6492 | 0.6493 |
62
+ | 0.5208 | 1.2579 | 400 | 0.6549 | 0.6416 |
63
+ | 0.5183 | 1.4151 | 450 | 0.6575 | 0.6301 |
64
+ | 0.5396 | 1.5723 | 500 | 0.6525 | 0.6281 |
65
+ | 0.5457 | 1.7296 | 550 | 0.6554 | 0.6320 |
66
+ | 0.5312 | 1.8868 | 600 | 0.6567 | 0.6416 |
67
 
68
 
69
  ### Framework versions
config.json CHANGED
@@ -1,12 +1,12 @@
1
  {
2
- "_name_or_path": "meta-llama/Meta-Llama-3-8B-Instruct",
3
  "architectures": [
4
  "LlamaForSequenceClassification"
5
  ],
6
  "attention_bias": false,
7
  "attention_dropout": 0.0,
8
  "bos_token_id": 128000,
9
- "eos_token_id": 128009,
10
  "hidden_act": "silu",
11
  "hidden_size": 4096,
12
  "id2label": {
@@ -23,7 +23,7 @@
23
  "num_attention_heads": 32,
24
  "num_hidden_layers": 32,
25
  "num_key_value_heads": 8,
26
- "pad_token_id": 128009,
27
  "pretraining_tp": 1,
28
  "rms_norm_eps": 1e-05,
29
  "rope_scaling": null,
 
1
  {
2
+ "_name_or_path": "meta-llama/Meta-Llama-3-8B",
3
  "architectures": [
4
  "LlamaForSequenceClassification"
5
  ],
6
  "attention_bias": false,
7
  "attention_dropout": 0.0,
8
  "bos_token_id": 128000,
9
+ "eos_token_id": 128001,
10
  "hidden_act": "silu",
11
  "hidden_size": 4096,
12
  "id2label": {
 
23
  "num_attention_heads": 32,
24
  "num_hidden_layers": 32,
25
  "num_key_value_heads": 8,
26
+ "pad_token_id": 128001,
27
  "pretraining_tp": 1,
28
  "rms_norm_eps": 1e-05,
29
  "rope_scaling": null,
model-00001-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d8cf9c4d0dd972e1a2131bfe656235ee98221679711a3beef6d46dadf0f20b5c
3
  size 4976698672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38723c28147bd62782695ea8aa5634dec5cf968ebc1b975f3661967f3e14821d
3
  size 4976698672
model-00002-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8d4782b4a69ef03845159ce1a15e272aadaaf134dc138d68f616098e8531729c
3
  size 4999802720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4b83e056f9d48e037562b74e9a7e76b6309b782602a7a501285c396391f082e
3
  size 4999802720
model-00003-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3acdd690e65c24f42a24581b8467af98bd3ca357444580f8012aacd2bd607921
3
  size 4915916176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:174796219072455f38a041a0101d60f0910b003f334ef50c672844511fa5bdbb
3
  size 4915916176
model-00004-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1873e93f6908ba2b51a61b160afda92887b5aa02196241bf8d0cafa109939678
3
  size 117473824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:813d40d751032d9ddcc54435aa3b2dc2d6f011615bfd54d5d54727e15f39463d
3
  size 117473824
special_tokens_map.json CHANGED
@@ -7,11 +7,11 @@
7
  "single_word": false
8
  },
9
  "eos_token": {
10
- "content": "<|eot_id|>",
11
  "lstrip": false,
12
  "normalized": false,
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
- "pad_token": "<|eot_id|>"
17
  }
 
7
  "single_word": false
8
  },
9
  "eos_token": {
10
+ "content": "<|end_of_text|>",
11
  "lstrip": false,
12
  "normalized": false,
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
+ "pad_token": "<|end_of_text|>"
17
  }
tokenizer.json CHANGED
@@ -12,9 +12,9 @@
12
  },
13
  "direction": "Right",
14
  "pad_to_multiple_of": null,
15
- "pad_id": 128009,
16
  "pad_type_id": 0,
17
- "pad_token": "<|eot_id|>"
18
  },
19
  "added_tokens": [
20
  {
 
12
  },
13
  "direction": "Right",
14
  "pad_to_multiple_of": null,
15
+ "pad_id": 128001,
16
  "pad_type_id": 0,
17
+ "pad_token": "<|end_of_text|>"
18
  },
19
  "added_tokens": [
20
  {
tokenizer_config.json CHANGED
@@ -2050,14 +2050,13 @@
2050
  }
2051
  },
2052
  "bos_token": "<|begin_of_text|>",
2053
- "chat_template": "{% set loop_messages = messages %}{% for message in loop_messages %}{% set content = '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n'+ message['content'] | trim + '<|eot_id|>' %}{% if loop.index0 == 0 %}{% set content = bos_token + content %}{% endif %}{{ content }}{% endfor %}{% if add_generation_prompt %}{{ '<|start_header_id|>assistant<|end_header_id|>\n\n' }}{% endif %}",
2054
  "clean_up_tokenization_spaces": true,
2055
- "eos_token": "<|eot_id|>",
2056
  "model_input_names": [
2057
  "input_ids",
2058
  "attention_mask"
2059
  ],
2060
  "model_max_length": 1000000000000000019884624838656,
2061
- "pad_token": "<|eot_id|>",
2062
  "tokenizer_class": "PreTrainedTokenizerFast"
2063
  }
 
2050
  }
2051
  },
2052
  "bos_token": "<|begin_of_text|>",
 
2053
  "clean_up_tokenization_spaces": true,
2054
+ "eos_token": "<|end_of_text|>",
2055
  "model_input_names": [
2056
  "input_ids",
2057
  "attention_mask"
2058
  ],
2059
  "model_max_length": 1000000000000000019884624838656,
2060
+ "pad_token": "<|end_of_text|>",
2061
  "tokenizer_class": "PreTrainedTokenizerFast"
2062
  }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:525af7619e3cfdeeb8faaeac9fe2216b9d22df5cec77dfb4a27e14d458e97496
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c41aaba649b0a4a3ea56100c58e005e244fdee9a1842123465f7c9c5439db1a
3
  size 5304