| { | |
| "step": 650, | |
| "val_loss": 1.0664211511611938, | |
| "mmlu_acc": 0.3623046875, | |
| "arc_easy_acc": 0.419921875, | |
| "gsm8k_acc": 0.03125, | |
| "humaneval_acc": 0.03125, | |
| "model_config": { | |
| "sequence_len": 2048, | |
| "vocab_size": 65536, | |
| "n_layer": 20, | |
| "n_head": 10, | |
| "n_kv_head": 10, | |
| "n_embd": 1280, | |
| "bos_token": "<|bos|>", | |
| "eos_token": "<|assistant_end|>", | |
| "pad_token": "<|assistant_end|>", | |
| "chat_template": "{% if messages[0]['role'] == 'system' %}<|bos|><|user_start|>{{ messages[0]['content'] }}\n\n{{ messages[1]['content'] }}<|user_end|>{% set messages = messages[2:] %}{% else %}<|bos|>{% endif %}{% for message in messages %}{% if loop.index0 % 2 == 0 %}<|user_start|>{{ message['content'] }}<|user_end|>{% else %}<|assistant_start|>{{ message['content'] }}<|assistant_end|>{% endif %}{% endfor %}" | |
| } | |
| } |