Gemma Function Calling Adapter
Browse files- adapter_config.json +28 -0
- adapter_model.safetensors +3 -0
- all_results.json +7 -0
- special_tokens_map.json +34 -0
- tokenizer.model +3 -0
- tokenizer_config.json +72 -0
- train_results.json +7 -0
- trainer_log.jsonl +76 -0
- trainer_state.json +555 -0
- training_args.bin +3 -0
adapter_config.json
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "google/gemma-7b-it",
|
5 |
+
"bias": "none",
|
6 |
+
"fan_in_fan_out": false,
|
7 |
+
"inference_mode": true,
|
8 |
+
"init_lora_weights": true,
|
9 |
+
"layers_pattern": null,
|
10 |
+
"layers_to_transform": null,
|
11 |
+
"loftq_config": {},
|
12 |
+
"lora_alpha": 16.0,
|
13 |
+
"lora_dropout": 0.1,
|
14 |
+
"megatron_config": null,
|
15 |
+
"megatron_core": "megatron.core",
|
16 |
+
"modules_to_save": null,
|
17 |
+
"peft_type": "LORA",
|
18 |
+
"r": 8,
|
19 |
+
"rank_pattern": {},
|
20 |
+
"revision": null,
|
21 |
+
"target_modules": [
|
22 |
+
"v_proj",
|
23 |
+
"q_proj"
|
24 |
+
],
|
25 |
+
"task_type": "CAUSAL_LM",
|
26 |
+
"use_dora": false,
|
27 |
+
"use_rslora": false
|
28 |
+
}
|
adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8fae4636ee4392a569d9e6f0e1d54033a9543c427827223a666afa8645caffb6
|
3 |
+
size 12859872
|
all_results.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 1.0,
|
3 |
+
"train_loss": 0.5547979205449423,
|
4 |
+
"train_runtime": 23630.1849,
|
5 |
+
"train_samples_per_second": 0.127,
|
6 |
+
"train_steps_per_second": 0.016
|
7 |
+
}
|
special_tokens_map.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"additional_special_tokens": [
|
3 |
+
"<start_of_turn>",
|
4 |
+
"<end_of_turn>"
|
5 |
+
],
|
6 |
+
"bos_token": {
|
7 |
+
"content": "<bos>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false
|
12 |
+
},
|
13 |
+
"eos_token": {
|
14 |
+
"content": "<eos>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false
|
19 |
+
},
|
20 |
+
"pad_token": {
|
21 |
+
"content": "<pad>",
|
22 |
+
"lstrip": false,
|
23 |
+
"normalized": false,
|
24 |
+
"rstrip": false,
|
25 |
+
"single_word": false
|
26 |
+
},
|
27 |
+
"unk_token": {
|
28 |
+
"content": "<unk>",
|
29 |
+
"lstrip": false,
|
30 |
+
"normalized": false,
|
31 |
+
"rstrip": false,
|
32 |
+
"single_word": false
|
33 |
+
}
|
34 |
+
}
|
tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:61a7b147390c64585d6c3543dd6fc636906c9af3865a5548f27f31aee1d4c8e2
|
3 |
+
size 4241003
|
tokenizer_config.json
ADDED
@@ -0,0 +1,72 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": true,
|
3 |
+
"add_eos_token": false,
|
4 |
+
"added_tokens_decoder": {
|
5 |
+
"0": {
|
6 |
+
"content": "<pad>",
|
7 |
+
"lstrip": false,
|
8 |
+
"normalized": false,
|
9 |
+
"rstrip": false,
|
10 |
+
"single_word": false,
|
11 |
+
"special": true
|
12 |
+
},
|
13 |
+
"1": {
|
14 |
+
"content": "<eos>",
|
15 |
+
"lstrip": false,
|
16 |
+
"normalized": false,
|
17 |
+
"rstrip": false,
|
18 |
+
"single_word": false,
|
19 |
+
"special": true
|
20 |
+
},
|
21 |
+
"2": {
|
22 |
+
"content": "<bos>",
|
23 |
+
"lstrip": false,
|
24 |
+
"normalized": false,
|
25 |
+
"rstrip": false,
|
26 |
+
"single_word": false,
|
27 |
+
"special": true
|
28 |
+
},
|
29 |
+
"3": {
|
30 |
+
"content": "<unk>",
|
31 |
+
"lstrip": false,
|
32 |
+
"normalized": false,
|
33 |
+
"rstrip": false,
|
34 |
+
"single_word": false,
|
35 |
+
"special": true
|
36 |
+
},
|
37 |
+
"106": {
|
38 |
+
"content": "<start_of_turn>",
|
39 |
+
"lstrip": false,
|
40 |
+
"normalized": false,
|
41 |
+
"rstrip": false,
|
42 |
+
"single_word": false,
|
43 |
+
"special": true
|
44 |
+
},
|
45 |
+
"107": {
|
46 |
+
"content": "<end_of_turn>",
|
47 |
+
"lstrip": false,
|
48 |
+
"normalized": false,
|
49 |
+
"rstrip": false,
|
50 |
+
"single_word": false,
|
51 |
+
"special": true
|
52 |
+
}
|
53 |
+
},
|
54 |
+
"additional_special_tokens": [
|
55 |
+
"<start_of_turn>",
|
56 |
+
"<end_of_turn>"
|
57 |
+
],
|
58 |
+
"bos_token": "<bos>",
|
59 |
+
"chat_template": "{{ bos_token }}{% if messages[0]['role'] == 'system' %}{{ raise_exception('System role not supported') }}{% endif %}{% for message in messages %}{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}{% endif %}{% if (message['role'] == 'assistant') %}{% set role = 'model' %}{% else %}{% set role = message['role'] %}{% endif %}{{ '<start_of_turn>' + role + '\n' + message['content'] | trim + '<end_of_turn>\n' }}{% endfor %}{% if add_generation_prompt %}{{'<start_of_turn>model\n'}}{% endif %}",
|
60 |
+
"clean_up_tokenization_spaces": false,
|
61 |
+
"eos_token": "<eos>",
|
62 |
+
"legacy": null,
|
63 |
+
"model_max_length": 1000000000000000019884624838656,
|
64 |
+
"pad_token": "<pad>",
|
65 |
+
"padding_side": "right",
|
66 |
+
"sp_model_kwargs": {},
|
67 |
+
"spaces_between_special_tokens": false,
|
68 |
+
"split_special_tokens": false,
|
69 |
+
"tokenizer_class": "GemmaTokenizer",
|
70 |
+
"unk_token": "<unk>",
|
71 |
+
"use_default_system_prompt": false
|
72 |
+
}
|
train_results.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 1.0,
|
3 |
+
"train_loss": 0.5547979205449423,
|
4 |
+
"train_runtime": 23630.1849,
|
5 |
+
"train_samples_per_second": 0.127,
|
6 |
+
"train_steps_per_second": 0.016
|
7 |
+
}
|
trainer_log.jsonl
ADDED
@@ -0,0 +1,76 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{"current_steps": 5, "total_steps": 375, "loss": 1.3397, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.9985889393594345e-05, "epoch": 0.01, "percentage": 1.33, "elapsed_time": "0:05:42", "remaining_time": "7:02:16"}
|
2 |
+
{"current_steps": 10, "total_steps": 375, "loss": 1.2413, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.992859235371958e-05, "epoch": 0.03, "percentage": 2.67, "elapsed_time": "0:10:36", "remaining_time": "6:27:27"}
|
3 |
+
{"current_steps": 15, "total_steps": 375, "loss": 1.0141, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.9827327946335875e-05, "epoch": 0.04, "percentage": 4.0, "elapsed_time": "0:15:19", "remaining_time": "6:07:46"}
|
4 |
+
{"current_steps": 20, "total_steps": 375, "loss": 0.9878, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.968227477476554e-05, "epoch": 0.05, "percentage": 5.33, "elapsed_time": "0:19:19", "remaining_time": "5:42:55"}
|
5 |
+
{"current_steps": 25, "total_steps": 375, "loss": 1.0212, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.949368867399567e-05, "epoch": 0.07, "percentage": 6.67, "elapsed_time": "0:25:17", "remaining_time": "5:54:02"}
|
6 |
+
{"current_steps": 30, "total_steps": 375, "loss": 0.7766, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.9261902259453614e-05, "epoch": 0.08, "percentage": 8.0, "elapsed_time": "0:30:55", "remaining_time": "5:55:38"}
|
7 |
+
{"current_steps": 35, "total_steps": 375, "loss": 0.8701, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.898732434036244e-05, "epoch": 0.09, "percentage": 9.33, "elapsed_time": "0:36:49", "remaining_time": "5:57:40"}
|
8 |
+
{"current_steps": 40, "total_steps": 375, "loss": 0.6978, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.867043919871076e-05, "epoch": 0.11, "percentage": 10.67, "elapsed_time": "0:41:40", "remaining_time": "5:49:02"}
|
9 |
+
{"current_steps": 45, "total_steps": 375, "loss": 0.754, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.8311805735108894e-05, "epoch": 0.12, "percentage": 12.0, "elapsed_time": "0:47:13", "remaining_time": "5:46:19"}
|
10 |
+
{"current_steps": 50, "total_steps": 375, "loss": 0.6599, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.791205648303775e-05, "epoch": 0.13, "percentage": 13.33, "elapsed_time": "0:52:46", "remaining_time": "5:43:05"}
|
11 |
+
{"current_steps": 55, "total_steps": 375, "loss": 0.6316, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.747189649322894e-05, "epoch": 0.15, "percentage": 14.67, "elapsed_time": "0:58:09", "remaining_time": "5:38:25"}
|
12 |
+
{"current_steps": 60, "total_steps": 375, "loss": 0.7531, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.699210209014394e-05, "epoch": 0.16, "percentage": 16.0, "elapsed_time": "1:03:49", "remaining_time": "5:35:07"}
|
13 |
+
{"current_steps": 65, "total_steps": 375, "loss": 0.4528, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.6473519502745476e-05, "epoch": 0.17, "percentage": 17.33, "elapsed_time": "1:08:08", "remaining_time": "5:24:58"}
|
14 |
+
{"current_steps": 70, "total_steps": 375, "loss": 0.565, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.591706337197597e-05, "epoch": 0.19, "percentage": 18.67, "elapsed_time": "1:13:28", "remaining_time": "5:20:08"}
|
15 |
+
{"current_steps": 75, "total_steps": 375, "loss": 0.6038, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.532371513757564e-05, "epoch": 0.2, "percentage": 20.0, "elapsed_time": "1:18:30", "remaining_time": "5:14:02"}
|
16 |
+
{"current_steps": 80, "total_steps": 375, "loss": 0.6498, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.469452130708543e-05, "epoch": 0.21, "percentage": 21.33, "elapsed_time": "1:24:22", "remaining_time": "5:11:08"}
|
17 |
+
{"current_steps": 85, "total_steps": 375, "loss": 0.5324, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.4030591610087624e-05, "epoch": 0.23, "percentage": 22.67, "elapsed_time": "1:30:46", "remaining_time": "5:09:43"}
|
18 |
+
{"current_steps": 90, "total_steps": 375, "loss": 0.5354, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.3333097040939764e-05, "epoch": 0.24, "percentage": 24.0, "elapsed_time": "1:35:55", "remaining_time": "5:03:45"}
|
19 |
+
{"current_steps": 95, "total_steps": 375, "loss": 0.5865, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.260326779345393e-05, "epoch": 0.25, "percentage": 25.33, "elapsed_time": "1:40:49", "remaining_time": "4:57:08"}
|
20 |
+
{"current_steps": 100, "total_steps": 375, "loss": 0.5894, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.184239109116393e-05, "epoch": 0.27, "percentage": 26.67, "elapsed_time": "1:46:16", "remaining_time": "4:52:14"}
|
21 |
+
{"current_steps": 105, "total_steps": 375, "loss": 0.5148, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.105180891700746e-05, "epoch": 0.28, "percentage": 28.0, "elapsed_time": "1:51:06", "remaining_time": "4:45:41"}
|
22 |
+
{"current_steps": 110, "total_steps": 375, "loss": 0.437, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.023291564642711e-05, "epoch": 0.29, "percentage": 29.33, "elapsed_time": "1:56:24", "remaining_time": "4:40:26"}
|
23 |
+
{"current_steps": 115, "total_steps": 375, "loss": 0.5527, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.938715558806525e-05, "epoch": 0.31, "percentage": 30.67, "elapsed_time": "2:01:20", "remaining_time": "4:34:20"}
|
24 |
+
{"current_steps": 120, "total_steps": 375, "loss": 0.5582, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.851602043638994e-05, "epoch": 0.32, "percentage": 32.0, "elapsed_time": "2:05:34", "remaining_time": "4:26:49"}
|
25 |
+
{"current_steps": 125, "total_steps": 375, "loss": 0.528, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.7621046640744975e-05, "epoch": 0.33, "percentage": 33.33, "elapsed_time": "2:10:19", "remaining_time": "4:20:39"}
|
26 |
+
{"current_steps": 130, "total_steps": 375, "loss": 0.4583, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.670381269546429e-05, "epoch": 0.35, "percentage": 34.67, "elapsed_time": "2:15:35", "remaining_time": "4:15:32"}
|
27 |
+
{"current_steps": 135, "total_steps": 375, "loss": 0.4325, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.5765936355830356e-05, "epoch": 0.36, "percentage": 36.0, "elapsed_time": "2:20:09", "remaining_time": "4:09:10"}
|
28 |
+
{"current_steps": 140, "total_steps": 375, "loss": 0.5655, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.480907178478654e-05, "epoch": 0.37, "percentage": 37.33, "elapsed_time": "2:25:04", "remaining_time": "4:03:30"}
|
29 |
+
{"current_steps": 145, "total_steps": 375, "loss": 0.5477, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.383490663543635e-05, "epoch": 0.39, "percentage": 38.67, "elapsed_time": "2:31:56", "remaining_time": "4:01:01"}
|
30 |
+
{"current_steps": 150, "total_steps": 375, "loss": 0.673, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.2845159074474806e-05, "epoch": 0.4, "percentage": 40.0, "elapsed_time": "2:38:05", "remaining_time": "3:57:07"}
|
31 |
+
{"current_steps": 155, "total_steps": 375, "loss": 0.539, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.1841574751802076e-05, "epoch": 0.41, "percentage": 41.33, "elapsed_time": "2:43:34", "remaining_time": "3:52:09"}
|
32 |
+
{"current_steps": 160, "total_steps": 375, "loss": 0.511, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.082592372166412e-05, "epoch": 0.43, "percentage": 42.67, "elapsed_time": "2:48:15", "remaining_time": "3:46:05"}
|
33 |
+
{"current_steps": 165, "total_steps": 375, "loss": 0.402, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.9799997320750507e-05, "epoch": 0.44, "percentage": 44.0, "elapsed_time": "2:53:33", "remaining_time": "3:40:53"}
|
34 |
+
{"current_steps": 170, "total_steps": 375, "loss": 0.453, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.87656050087558e-05, "epoch": 0.45, "percentage": 45.33, "elapsed_time": "2:58:12", "remaining_time": "3:34:54"}
|
35 |
+
{"current_steps": 175, "total_steps": 375, "loss": 0.5608, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.7724571176976732e-05, "epoch": 0.47, "percentage": 46.67, "elapsed_time": "3:03:11", "remaining_time": "3:29:21"}
|
36 |
+
{"current_steps": 180, "total_steps": 375, "loss": 0.4598, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.667873193057407e-05, "epoch": 0.48, "percentage": 48.0, "elapsed_time": "3:07:52", "remaining_time": "3:23:31"}
|
37 |
+
{"current_steps": 185, "total_steps": 375, "loss": 0.4952, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.562993185017431e-05, "epoch": 0.49, "percentage": 49.33, "elapsed_time": "3:13:55", "remaining_time": "3:19:09"}
|
38 |
+
{"current_steps": 190, "total_steps": 375, "loss": 0.4087, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.4580020738523e-05, "epoch": 0.51, "percentage": 50.67, "elapsed_time": "3:19:01", "remaining_time": "3:13:47"}
|
39 |
+
{"current_steps": 195, "total_steps": 375, "loss": 0.5781, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.353085035792756e-05, "epoch": 0.52, "percentage": 52.0, "elapsed_time": "3:24:12", "remaining_time": "3:08:29"}
|
40 |
+
{"current_steps": 200, "total_steps": 375, "loss": 0.4127, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.2484271164243996e-05, "epoch": 0.53, "percentage": 53.33, "elapsed_time": "3:28:13", "remaining_time": "3:02:11"}
|
41 |
+
{"current_steps": 205, "total_steps": 375, "loss": 0.5682, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.1442129043167874e-05, "epoch": 0.55, "percentage": 54.67, "elapsed_time": "3:34:40", "remaining_time": "2:58:01"}
|
42 |
+
{"current_steps": 210, "total_steps": 375, "loss": 0.6413, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.0406262054585738e-05, "epoch": 0.56, "percentage": 56.0, "elapsed_time": "3:40:24", "remaining_time": "2:53:10"}
|
43 |
+
{"current_steps": 215, "total_steps": 375, "loss": 0.4034, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.937849719072931e-05, "epoch": 0.57, "percentage": 57.33, "elapsed_time": "3:45:14", "remaining_time": "2:47:37"}
|
44 |
+
{"current_steps": 220, "total_steps": 375, "loss": 0.5331, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.836064715384989e-05, "epoch": 0.59, "percentage": 58.67, "elapsed_time": "3:50:20", "remaining_time": "2:42:16"}
|
45 |
+
{"current_steps": 225, "total_steps": 375, "loss": 0.4308, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.7354507159096647e-05, "epoch": 0.6, "percentage": 60.0, "elapsed_time": "3:55:16", "remaining_time": "2:36:50"}
|
46 |
+
{"current_steps": 230, "total_steps": 375, "loss": 0.474, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.6361851768237324e-05, "epoch": 0.61, "percentage": 61.33, "elapsed_time": "4:00:20", "remaining_time": "2:31:31"}
|
47 |
+
{"current_steps": 235, "total_steps": 375, "loss": 0.4306, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.5384431759806083e-05, "epoch": 0.63, "percentage": 62.67, "elapsed_time": "4:06:17", "remaining_time": "2:26:43"}
|
48 |
+
{"current_steps": 240, "total_steps": 375, "loss": 0.5352, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.4423971041198556e-05, "epoch": 0.64, "percentage": 64.0, "elapsed_time": "4:11:43", "remaining_time": "2:21:35"}
|
49 |
+
{"current_steps": 245, "total_steps": 375, "loss": 0.436, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.3482163608160408e-05, "epoch": 0.65, "percentage": 65.33, "elapsed_time": "4:16:32", "remaining_time": "2:16:07"}
|
50 |
+
{"current_steps": 250, "total_steps": 375, "loss": 0.4172, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.2560670557032108e-05, "epoch": 0.67, "percentage": 66.67, "elapsed_time": "4:22:12", "remaining_time": "2:11:06"}
|
51 |
+
{"current_steps": 255, "total_steps": 375, "loss": 0.3712, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.1661117155019293e-05, "epoch": 0.68, "percentage": 68.0, "elapsed_time": "4:27:11", "remaining_time": "2:05:44"}
|
52 |
+
{"current_steps": 260, "total_steps": 375, "loss": 0.5507, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.0785089973656337e-05, "epoch": 0.69, "percentage": 69.33, "elapsed_time": "4:34:04", "remaining_time": "2:01:13"}
|
53 |
+
{"current_steps": 265, "total_steps": 375, "loss": 0.5309, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 9.934134090518593e-06, "epoch": 0.71, "percentage": 70.67, "elapsed_time": "4:39:03", "remaining_time": "1:55:49"}
|
54 |
+
{"current_steps": 270, "total_steps": 375, "loss": 0.4294, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 9.109750364118924e-06, "epoch": 0.72, "percentage": 72.0, "elapsed_time": "4:44:19", "remaining_time": "1:50:34"}
|
55 |
+
{"current_steps": 275, "total_steps": 375, "loss": 0.4236, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 8.313392786794833e-06, "epoch": 0.73, "percentage": 73.33, "elapsed_time": "4:49:21", "remaining_time": "1:45:13"}
|
56 |
+
{"current_steps": 280, "total_steps": 375, "loss": 0.555, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 7.546465920254974e-06, "epoch": 0.75, "percentage": 74.67, "elapsed_time": "4:56:01", "remaining_time": "1:40:26"}
|
57 |
+
{"current_steps": 285, "total_steps": 375, "loss": 0.5034, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 6.810322418308085e-06, "epoch": 0.76, "percentage": 76.0, "elapsed_time": "5:01:35", "remaining_time": "1:35:14"}
|
58 |
+
{"current_steps": 290, "total_steps": 375, "loss": 0.4616, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 6.106260641143546e-06, "epoch": 0.77, "percentage": 77.33, "elapsed_time": "5:06:33", "remaining_time": "1:29:51"}
|
59 |
+
{"current_steps": 295, "total_steps": 375, "loss": 0.456, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 5.435522365371376e-06, "epoch": 0.79, "percentage": 78.67, "elapsed_time": "5:12:41", "remaining_time": "1:24:47"}
|
60 |
+
{"current_steps": 300, "total_steps": 375, "loss": 0.4582, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.799290593860525e-06, "epoch": 0.8, "percentage": 80.0, "elapsed_time": "5:18:12", "remaining_time": "1:19:33"}
|
61 |
+
{"current_steps": 305, "total_steps": 375, "loss": 0.3826, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 4.198687469238297e-06, "epoch": 0.81, "percentage": 81.33, "elapsed_time": "5:22:16", "remaining_time": "1:13:57"}
|
62 |
+
{"current_steps": 310, "total_steps": 375, "loss": 0.4933, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.6347722947309843e-06, "epoch": 0.83, "percentage": 82.67, "elapsed_time": "5:27:52", "remaining_time": "1:08:44"}
|
63 |
+
{"current_steps": 315, "total_steps": 375, "loss": 0.5099, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.108539665836388e-06, "epoch": 0.84, "percentage": 84.0, "elapsed_time": "5:33:15", "remaining_time": "1:03:28"}
|
64 |
+
{"current_steps": 320, "total_steps": 375, "loss": 0.57, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.6209177161234445e-06, "epoch": 0.85, "percentage": 85.33, "elapsed_time": "5:39:26", "remaining_time": "0:58:20"}
|
65 |
+
{"current_steps": 325, "total_steps": 375, "loss": 0.4737, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.1727664802529216e-06, "epoch": 0.87, "percentage": 86.67, "elapsed_time": "5:45:04", "remaining_time": "0:53:05"}
|
66 |
+
{"current_steps": 330, "total_steps": 375, "loss": 0.4315, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.7648763771063837e-06, "epoch": 0.88, "percentage": 88.0, "elapsed_time": "5:49:47", "remaining_time": "0:47:41"}
|
67 |
+
{"current_steps": 335, "total_steps": 375, "loss": 0.4254, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.3979668156987425e-06, "epoch": 0.89, "percentage": 89.33, "elapsed_time": "5:54:41", "remaining_time": "0:42:21"}
|
68 |
+
{"current_steps": 340, "total_steps": 375, "loss": 0.3288, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.0726849263332256e-06, "epoch": 0.91, "percentage": 90.67, "elapsed_time": "5:58:37", "remaining_time": "0:36:54"}
|
69 |
+
{"current_steps": 345, "total_steps": 375, "loss": 0.2477, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 7.896044192366586e-07, "epoch": 0.92, "percentage": 92.0, "elapsed_time": "6:03:00", "remaining_time": "0:31:33"}
|
70 |
+
{"current_steps": 350, "total_steps": 375, "loss": 0.453, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 5.492245726881201e-07, "epoch": 0.93, "percentage": 93.33, "elapsed_time": "6:08:18", "remaining_time": "0:26:18"}
|
71 |
+
{"current_steps": 355, "total_steps": 375, "loss": 0.4833, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 3.51969352425624e-07, "epoch": 0.95, "percentage": 94.67, "elapsed_time": "6:14:04", "remaining_time": "0:21:04"}
|
72 |
+
{"current_steps": 360, "total_steps": 375, "loss": 0.5507, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 1.981866638839952e-07, "epoch": 0.96, "percentage": 96.0, "elapsed_time": "6:20:05", "remaining_time": "0:15:50"}
|
73 |
+
{"current_steps": 365, "total_steps": 375, "loss": 0.379, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 8.814773858275004e-08, "epoch": 0.97, "percentage": 97.33, "elapsed_time": "6:24:19", "remaining_time": "0:10:31"}
|
74 |
+
{"current_steps": 370, "total_steps": 375, "loss": 0.4705, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 2.2046655746280064e-08, "epoch": 0.99, "percentage": 98.67, "elapsed_time": "6:29:22", "remaining_time": "0:05:15"}
|
75 |
+
{"current_steps": 375, "total_steps": 375, "loss": 0.4502, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": 0.0, "epoch": 1.0, "percentage": 100.0, "elapsed_time": "6:33:50", "remaining_time": "0:00:00"}
|
76 |
+
{"current_steps": 375, "total_steps": 375, "loss": null, "eval_loss": null, "predict_loss": null, "reward": null, "learning_rate": null, "epoch": 1.0, "percentage": 100.0, "elapsed_time": "6:33:50", "remaining_time": "0:00:00"}
|
trainer_state.json
ADDED
@@ -0,0 +1,555 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 1.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 375,
|
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.01,
|
13 |
+
"grad_norm": 1.65970778465271,
|
14 |
+
"learning_rate": 4.9985889393594345e-05,
|
15 |
+
"loss": 1.3397,
|
16 |
+
"step": 5
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.03,
|
20 |
+
"grad_norm": 1.8339558839797974,
|
21 |
+
"learning_rate": 4.992859235371958e-05,
|
22 |
+
"loss": 1.2413,
|
23 |
+
"step": 10
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.04,
|
27 |
+
"grad_norm": 1.599440336227417,
|
28 |
+
"learning_rate": 4.9827327946335875e-05,
|
29 |
+
"loss": 1.0141,
|
30 |
+
"step": 15
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.05,
|
34 |
+
"grad_norm": 1.4671368598937988,
|
35 |
+
"learning_rate": 4.968227477476554e-05,
|
36 |
+
"loss": 0.9878,
|
37 |
+
"step": 20
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 0.07,
|
41 |
+
"grad_norm": 0.9982190728187561,
|
42 |
+
"learning_rate": 4.949368867399567e-05,
|
43 |
+
"loss": 1.0212,
|
44 |
+
"step": 25
|
45 |
+
},
|
46 |
+
{
|
47 |
+
"epoch": 0.08,
|
48 |
+
"grad_norm": 1.0809102058410645,
|
49 |
+
"learning_rate": 4.9261902259453614e-05,
|
50 |
+
"loss": 0.7766,
|
51 |
+
"step": 30
|
52 |
+
},
|
53 |
+
{
|
54 |
+
"epoch": 0.09,
|
55 |
+
"grad_norm": 0.9444558620452881,
|
56 |
+
"learning_rate": 4.898732434036244e-05,
|
57 |
+
"loss": 0.8701,
|
58 |
+
"step": 35
|
59 |
+
},
|
60 |
+
{
|
61 |
+
"epoch": 0.11,
|
62 |
+
"grad_norm": 1.0244667530059814,
|
63 |
+
"learning_rate": 4.867043919871076e-05,
|
64 |
+
"loss": 0.6978,
|
65 |
+
"step": 40
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.12,
|
69 |
+
"grad_norm": 0.7848422527313232,
|
70 |
+
"learning_rate": 4.8311805735108894e-05,
|
71 |
+
"loss": 0.754,
|
72 |
+
"step": 45
|
73 |
+
},
|
74 |
+
{
|
75 |
+
"epoch": 0.13,
|
76 |
+
"grad_norm": 0.9545549154281616,
|
77 |
+
"learning_rate": 4.791205648303775e-05,
|
78 |
+
"loss": 0.6599,
|
79 |
+
"step": 50
|
80 |
+
},
|
81 |
+
{
|
82 |
+
"epoch": 0.15,
|
83 |
+
"grad_norm": 0.6984307765960693,
|
84 |
+
"learning_rate": 4.747189649322894e-05,
|
85 |
+
"loss": 0.6316,
|
86 |
+
"step": 55
|
87 |
+
},
|
88 |
+
{
|
89 |
+
"epoch": 0.16,
|
90 |
+
"grad_norm": 0.7565098404884338,
|
91 |
+
"learning_rate": 4.699210209014394e-05,
|
92 |
+
"loss": 0.7531,
|
93 |
+
"step": 60
|
94 |
+
},
|
95 |
+
{
|
96 |
+
"epoch": 0.17,
|
97 |
+
"grad_norm": 0.5942355990409851,
|
98 |
+
"learning_rate": 4.6473519502745476e-05,
|
99 |
+
"loss": 0.4528,
|
100 |
+
"step": 65
|
101 |
+
},
|
102 |
+
{
|
103 |
+
"epoch": 0.19,
|
104 |
+
"grad_norm": 0.7522299885749817,
|
105 |
+
"learning_rate": 4.591706337197597e-05,
|
106 |
+
"loss": 0.565,
|
107 |
+
"step": 70
|
108 |
+
},
|
109 |
+
{
|
110 |
+
"epoch": 0.2,
|
111 |
+
"grad_norm": 0.8787449598312378,
|
112 |
+
"learning_rate": 4.532371513757564e-05,
|
113 |
+
"loss": 0.6038,
|
114 |
+
"step": 75
|
115 |
+
},
|
116 |
+
{
|
117 |
+
"epoch": 0.21,
|
118 |
+
"grad_norm": 0.6031142473220825,
|
119 |
+
"learning_rate": 4.469452130708543e-05,
|
120 |
+
"loss": 0.6498,
|
121 |
+
"step": 80
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.23,
|
125 |
+
"grad_norm": 0.838216245174408,
|
126 |
+
"learning_rate": 4.4030591610087624e-05,
|
127 |
+
"loss": 0.5324,
|
128 |
+
"step": 85
|
129 |
+
},
|
130 |
+
{
|
131 |
+
"epoch": 0.24,
|
132 |
+
"grad_norm": 1.0710391998291016,
|
133 |
+
"learning_rate": 4.3333097040939764e-05,
|
134 |
+
"loss": 0.5354,
|
135 |
+
"step": 90
|
136 |
+
},
|
137 |
+
{
|
138 |
+
"epoch": 0.25,
|
139 |
+
"grad_norm": 0.9150385856628418,
|
140 |
+
"learning_rate": 4.260326779345393e-05,
|
141 |
+
"loss": 0.5865,
|
142 |
+
"step": 95
|
143 |
+
},
|
144 |
+
{
|
145 |
+
"epoch": 0.27,
|
146 |
+
"grad_norm": 0.8779798746109009,
|
147 |
+
"learning_rate": 4.184239109116393e-05,
|
148 |
+
"loss": 0.5894,
|
149 |
+
"step": 100
|
150 |
+
},
|
151 |
+
{
|
152 |
+
"epoch": 0.28,
|
153 |
+
"grad_norm": 2.170987844467163,
|
154 |
+
"learning_rate": 4.105180891700746e-05,
|
155 |
+
"loss": 0.5148,
|
156 |
+
"step": 105
|
157 |
+
},
|
158 |
+
{
|
159 |
+
"epoch": 0.29,
|
160 |
+
"grad_norm": 0.8440865874290466,
|
161 |
+
"learning_rate": 4.023291564642711e-05,
|
162 |
+
"loss": 0.437,
|
163 |
+
"step": 110
|
164 |
+
},
|
165 |
+
{
|
166 |
+
"epoch": 0.31,
|
167 |
+
"grad_norm": 0.8862484693527222,
|
168 |
+
"learning_rate": 3.938715558806525e-05,
|
169 |
+
"loss": 0.5527,
|
170 |
+
"step": 115
|
171 |
+
},
|
172 |
+
{
|
173 |
+
"epoch": 0.32,
|
174 |
+
"grad_norm": 1.1455185413360596,
|
175 |
+
"learning_rate": 3.851602043638994e-05,
|
176 |
+
"loss": 0.5582,
|
177 |
+
"step": 120
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 0.33,
|
181 |
+
"grad_norm": 0.8740842938423157,
|
182 |
+
"learning_rate": 3.7621046640744975e-05,
|
183 |
+
"loss": 0.528,
|
184 |
+
"step": 125
|
185 |
+
},
|
186 |
+
{
|
187 |
+
"epoch": 0.35,
|
188 |
+
"grad_norm": 1.2668254375457764,
|
189 |
+
"learning_rate": 3.670381269546429e-05,
|
190 |
+
"loss": 0.4583,
|
191 |
+
"step": 130
|
192 |
+
},
|
193 |
+
{
|
194 |
+
"epoch": 0.36,
|
195 |
+
"grad_norm": 0.8675268292427063,
|
196 |
+
"learning_rate": 3.5765936355830356e-05,
|
197 |
+
"loss": 0.4325,
|
198 |
+
"step": 135
|
199 |
+
},
|
200 |
+
{
|
201 |
+
"epoch": 0.37,
|
202 |
+
"grad_norm": 0.9672185182571411,
|
203 |
+
"learning_rate": 3.480907178478654e-05,
|
204 |
+
"loss": 0.5655,
|
205 |
+
"step": 140
|
206 |
+
},
|
207 |
+
{
|
208 |
+
"epoch": 0.39,
|
209 |
+
"grad_norm": 1.1447197198867798,
|
210 |
+
"learning_rate": 3.383490663543635e-05,
|
211 |
+
"loss": 0.5477,
|
212 |
+
"step": 145
|
213 |
+
},
|
214 |
+
{
|
215 |
+
"epoch": 0.4,
|
216 |
+
"grad_norm": 0.7843021750450134,
|
217 |
+
"learning_rate": 3.2845159074474806e-05,
|
218 |
+
"loss": 0.673,
|
219 |
+
"step": 150
|
220 |
+
},
|
221 |
+
{
|
222 |
+
"epoch": 0.41,
|
223 |
+
"grad_norm": 0.8366529941558838,
|
224 |
+
"learning_rate": 3.1841574751802076e-05,
|
225 |
+
"loss": 0.539,
|
226 |
+
"step": 155
|
227 |
+
},
|
228 |
+
{
|
229 |
+
"epoch": 0.43,
|
230 |
+
"grad_norm": 1.1384434700012207,
|
231 |
+
"learning_rate": 3.082592372166412e-05,
|
232 |
+
"loss": 0.511,
|
233 |
+
"step": 160
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 0.44,
|
237 |
+
"grad_norm": 0.7654138803482056,
|
238 |
+
"learning_rate": 2.9799997320750507e-05,
|
239 |
+
"loss": 0.402,
|
240 |
+
"step": 165
|
241 |
+
},
|
242 |
+
{
|
243 |
+
"epoch": 0.45,
|
244 |
+
"grad_norm": 1.1874901056289673,
|
245 |
+
"learning_rate": 2.87656050087558e-05,
|
246 |
+
"loss": 0.453,
|
247 |
+
"step": 170
|
248 |
+
},
|
249 |
+
{
|
250 |
+
"epoch": 0.47,
|
251 |
+
"grad_norm": 0.9217435717582703,
|
252 |
+
"learning_rate": 2.7724571176976732e-05,
|
253 |
+
"loss": 0.5608,
|
254 |
+
"step": 175
|
255 |
+
},
|
256 |
+
{
|
257 |
+
"epoch": 0.48,
|
258 |
+
"grad_norm": 1.0420029163360596,
|
259 |
+
"learning_rate": 2.667873193057407e-05,
|
260 |
+
"loss": 0.4598,
|
261 |
+
"step": 180
|
262 |
+
},
|
263 |
+
{
|
264 |
+
"epoch": 0.49,
|
265 |
+
"grad_norm": 0.8808046579360962,
|
266 |
+
"learning_rate": 2.562993185017431e-05,
|
267 |
+
"loss": 0.4952,
|
268 |
+
"step": 185
|
269 |
+
},
|
270 |
+
{
|
271 |
+
"epoch": 0.51,
|
272 |
+
"grad_norm": 1.2101999521255493,
|
273 |
+
"learning_rate": 2.4580020738523e-05,
|
274 |
+
"loss": 0.4087,
|
275 |
+
"step": 190
|
276 |
+
},
|
277 |
+
{
|
278 |
+
"epoch": 0.52,
|
279 |
+
"grad_norm": 1.3266472816467285,
|
280 |
+
"learning_rate": 2.353085035792756e-05,
|
281 |
+
"loss": 0.5781,
|
282 |
+
"step": 195
|
283 |
+
},
|
284 |
+
{
|
285 |
+
"epoch": 0.53,
|
286 |
+
"grad_norm": 1.5473909378051758,
|
287 |
+
"learning_rate": 2.2484271164243996e-05,
|
288 |
+
"loss": 0.4127,
|
289 |
+
"step": 200
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 0.55,
|
293 |
+
"grad_norm": 1.1306591033935547,
|
294 |
+
"learning_rate": 2.1442129043167874e-05,
|
295 |
+
"loss": 0.5682,
|
296 |
+
"step": 205
|
297 |
+
},
|
298 |
+
{
|
299 |
+
"epoch": 0.56,
|
300 |
+
"grad_norm": 1.0642080307006836,
|
301 |
+
"learning_rate": 2.0406262054585738e-05,
|
302 |
+
"loss": 0.6413,
|
303 |
+
"step": 210
|
304 |
+
},
|
305 |
+
{
|
306 |
+
"epoch": 0.57,
|
307 |
+
"grad_norm": 0.9475000500679016,
|
308 |
+
"learning_rate": 1.937849719072931e-05,
|
309 |
+
"loss": 0.4034,
|
310 |
+
"step": 215
|
311 |
+
},
|
312 |
+
{
|
313 |
+
"epoch": 0.59,
|
314 |
+
"grad_norm": 1.3008495569229126,
|
315 |
+
"learning_rate": 1.836064715384989e-05,
|
316 |
+
"loss": 0.5331,
|
317 |
+
"step": 220
|
318 |
+
},
|
319 |
+
{
|
320 |
+
"epoch": 0.6,
|
321 |
+
"grad_norm": 1.1625583171844482,
|
322 |
+
"learning_rate": 1.7354507159096647e-05,
|
323 |
+
"loss": 0.4308,
|
324 |
+
"step": 225
|
325 |
+
},
|
326 |
+
{
|
327 |
+
"epoch": 0.61,
|
328 |
+
"grad_norm": 1.3643542528152466,
|
329 |
+
"learning_rate": 1.6361851768237324e-05,
|
330 |
+
"loss": 0.474,
|
331 |
+
"step": 230
|
332 |
+
},
|
333 |
+
{
|
334 |
+
"epoch": 0.63,
|
335 |
+
"grad_norm": 0.6814500093460083,
|
336 |
+
"learning_rate": 1.5384431759806083e-05,
|
337 |
+
"loss": 0.4306,
|
338 |
+
"step": 235
|
339 |
+
},
|
340 |
+
{
|
341 |
+
"epoch": 0.64,
|
342 |
+
"grad_norm": 1.1910830736160278,
|
343 |
+
"learning_rate": 1.4423971041198556e-05,
|
344 |
+
"loss": 0.5352,
|
345 |
+
"step": 240
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 0.65,
|
349 |
+
"grad_norm": 1.0804316997528076,
|
350 |
+
"learning_rate": 1.3482163608160408e-05,
|
351 |
+
"loss": 0.436,
|
352 |
+
"step": 245
|
353 |
+
},
|
354 |
+
{
|
355 |
+
"epoch": 0.67,
|
356 |
+
"grad_norm": 1.1781166791915894,
|
357 |
+
"learning_rate": 1.2560670557032108e-05,
|
358 |
+
"loss": 0.4172,
|
359 |
+
"step": 250
|
360 |
+
},
|
361 |
+
{
|
362 |
+
"epoch": 0.68,
|
363 |
+
"grad_norm": 1.1555372476577759,
|
364 |
+
"learning_rate": 1.1661117155019293e-05,
|
365 |
+
"loss": 0.3712,
|
366 |
+
"step": 255
|
367 |
+
},
|
368 |
+
{
|
369 |
+
"epoch": 0.69,
|
370 |
+
"grad_norm": 1.105285882949829,
|
371 |
+
"learning_rate": 1.0785089973656337e-05,
|
372 |
+
"loss": 0.5507,
|
373 |
+
"step": 260
|
374 |
+
},
|
375 |
+
{
|
376 |
+
"epoch": 0.71,
|
377 |
+
"grad_norm": 1.9189614057540894,
|
378 |
+
"learning_rate": 9.934134090518593e-06,
|
379 |
+
"loss": 0.5309,
|
380 |
+
"step": 265
|
381 |
+
},
|
382 |
+
{
|
383 |
+
"epoch": 0.72,
|
384 |
+
"grad_norm": 0.8669947981834412,
|
385 |
+
"learning_rate": 9.109750364118924e-06,
|
386 |
+
"loss": 0.4294,
|
387 |
+
"step": 270
|
388 |
+
},
|
389 |
+
{
|
390 |
+
"epoch": 0.73,
|
391 |
+
"grad_norm": 0.755587637424469,
|
392 |
+
"learning_rate": 8.313392786794833e-06,
|
393 |
+
"loss": 0.4236,
|
394 |
+
"step": 275
|
395 |
+
},
|
396 |
+
{
|
397 |
+
"epoch": 0.75,
|
398 |
+
"grad_norm": 1.1405256986618042,
|
399 |
+
"learning_rate": 7.546465920254974e-06,
|
400 |
+
"loss": 0.555,
|
401 |
+
"step": 280
|
402 |
+
},
|
403 |
+
{
|
404 |
+
"epoch": 0.76,
|
405 |
+
"grad_norm": 1.6737154722213745,
|
406 |
+
"learning_rate": 6.810322418308085e-06,
|
407 |
+
"loss": 0.5034,
|
408 |
+
"step": 285
|
409 |
+
},
|
410 |
+
{
|
411 |
+
"epoch": 0.77,
|
412 |
+
"grad_norm": 1.107371211051941,
|
413 |
+
"learning_rate": 6.106260641143546e-06,
|
414 |
+
"loss": 0.4616,
|
415 |
+
"step": 290
|
416 |
+
},
|
417 |
+
{
|
418 |
+
"epoch": 0.79,
|
419 |
+
"grad_norm": 1.2634700536727905,
|
420 |
+
"learning_rate": 5.435522365371376e-06,
|
421 |
+
"loss": 0.456,
|
422 |
+
"step": 295
|
423 |
+
},
|
424 |
+
{
|
425 |
+
"epoch": 0.8,
|
426 |
+
"grad_norm": 0.8122715950012207,
|
427 |
+
"learning_rate": 4.799290593860525e-06,
|
428 |
+
"loss": 0.4582,
|
429 |
+
"step": 300
|
430 |
+
},
|
431 |
+
{
|
432 |
+
"epoch": 0.81,
|
433 |
+
"grad_norm": 1.333025336265564,
|
434 |
+
"learning_rate": 4.198687469238297e-06,
|
435 |
+
"loss": 0.3826,
|
436 |
+
"step": 305
|
437 |
+
},
|
438 |
+
{
|
439 |
+
"epoch": 0.83,
|
440 |
+
"grad_norm": 1.5998661518096924,
|
441 |
+
"learning_rate": 3.6347722947309843e-06,
|
442 |
+
"loss": 0.4933,
|
443 |
+
"step": 310
|
444 |
+
},
|
445 |
+
{
|
446 |
+
"epoch": 0.84,
|
447 |
+
"grad_norm": 1.3631378412246704,
|
448 |
+
"learning_rate": 3.108539665836388e-06,
|
449 |
+
"loss": 0.5099,
|
450 |
+
"step": 315
|
451 |
+
},
|
452 |
+
{
|
453 |
+
"epoch": 0.85,
|
454 |
+
"grad_norm": 0.714860200881958,
|
455 |
+
"learning_rate": 2.6209177161234445e-06,
|
456 |
+
"loss": 0.57,
|
457 |
+
"step": 320
|
458 |
+
},
|
459 |
+
{
|
460 |
+
"epoch": 0.87,
|
461 |
+
"grad_norm": 0.5441424250602722,
|
462 |
+
"learning_rate": 2.1727664802529216e-06,
|
463 |
+
"loss": 0.4737,
|
464 |
+
"step": 325
|
465 |
+
},
|
466 |
+
{
|
467 |
+
"epoch": 0.88,
|
468 |
+
"grad_norm": 1.4750747680664062,
|
469 |
+
"learning_rate": 1.7648763771063837e-06,
|
470 |
+
"loss": 0.4315,
|
471 |
+
"step": 330
|
472 |
+
},
|
473 |
+
{
|
474 |
+
"epoch": 0.89,
|
475 |
+
"grad_norm": 1.6141853332519531,
|
476 |
+
"learning_rate": 1.3979668156987425e-06,
|
477 |
+
"loss": 0.4254,
|
478 |
+
"step": 335
|
479 |
+
},
|
480 |
+
{
|
481 |
+
"epoch": 0.91,
|
482 |
+
"grad_norm": 1.2416565418243408,
|
483 |
+
"learning_rate": 1.0726849263332256e-06,
|
484 |
+
"loss": 0.3288,
|
485 |
+
"step": 340
|
486 |
+
},
|
487 |
+
{
|
488 |
+
"epoch": 0.92,
|
489 |
+
"grad_norm": 1.2494091987609863,
|
490 |
+
"learning_rate": 7.896044192366586e-07,
|
491 |
+
"loss": 0.2477,
|
492 |
+
"step": 345
|
493 |
+
},
|
494 |
+
{
|
495 |
+
"epoch": 0.93,
|
496 |
+
"grad_norm": 0.9337052702903748,
|
497 |
+
"learning_rate": 5.492245726881201e-07,
|
498 |
+
"loss": 0.453,
|
499 |
+
"step": 350
|
500 |
+
},
|
501 |
+
{
|
502 |
+
"epoch": 0.95,
|
503 |
+
"grad_norm": 1.0654840469360352,
|
504 |
+
"learning_rate": 3.51969352425624e-07,
|
505 |
+
"loss": 0.4833,
|
506 |
+
"step": 355
|
507 |
+
},
|
508 |
+
{
|
509 |
+
"epoch": 0.96,
|
510 |
+
"grad_norm": 1.7972078323364258,
|
511 |
+
"learning_rate": 1.981866638839952e-07,
|
512 |
+
"loss": 0.5507,
|
513 |
+
"step": 360
|
514 |
+
},
|
515 |
+
{
|
516 |
+
"epoch": 0.97,
|
517 |
+
"grad_norm": 1.450646996498108,
|
518 |
+
"learning_rate": 8.814773858275004e-08,
|
519 |
+
"loss": 0.379,
|
520 |
+
"step": 365
|
521 |
+
},
|
522 |
+
{
|
523 |
+
"epoch": 0.99,
|
524 |
+
"grad_norm": 1.490728497505188,
|
525 |
+
"learning_rate": 2.2046655746280064e-08,
|
526 |
+
"loss": 0.4705,
|
527 |
+
"step": 370
|
528 |
+
},
|
529 |
+
{
|
530 |
+
"epoch": 1.0,
|
531 |
+
"grad_norm": 0.9804399609565735,
|
532 |
+
"learning_rate": 0.0,
|
533 |
+
"loss": 0.4502,
|
534 |
+
"step": 375
|
535 |
+
},
|
536 |
+
{
|
537 |
+
"epoch": 1.0,
|
538 |
+
"step": 375,
|
539 |
+
"total_flos": 6.701975198539776e+16,
|
540 |
+
"train_loss": 0.5547979205449423,
|
541 |
+
"train_runtime": 23630.1849,
|
542 |
+
"train_samples_per_second": 0.127,
|
543 |
+
"train_steps_per_second": 0.016
|
544 |
+
}
|
545 |
+
],
|
546 |
+
"logging_steps": 5,
|
547 |
+
"max_steps": 375,
|
548 |
+
"num_input_tokens_seen": 0,
|
549 |
+
"num_train_epochs": 1,
|
550 |
+
"save_steps": 10,
|
551 |
+
"total_flos": 6.701975198539776e+16,
|
552 |
+
"train_batch_size": 1,
|
553 |
+
"trial_name": null,
|
554 |
+
"trial_params": null
|
555 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c2b2a3419d3cb2cd66c9bbc4ca22151d520657e51fcc1826e0126e6e5734ac8d
|
3 |
+
size 5048
|