{ | |
"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 %}" | |
} | |
} |