| { |
| "model_id": "m20", |
| "checkpoint": "/home/akshat/codemindv2/results/m20/final.pt", |
| "trained_step": 15258, |
| "total_params": 61920256, |
| "active_params": 61920256, |
| "flops_per_forward": 123840512.0, |
| "val_loss": 3.4625886261463164, |
| "val_perplexity": 31.899445466560927, |
| "bits_per_token": 4.99545943957982, |
| "ppl_tokens_evaluated": 655360, |
| "prefill_tok_per_sec": 158358.59954810305, |
| "inference_tok_per_sec": 178.26855692268356, |
| "inference_tok_per_sec_kvcache": 191.98823619444323, |
| "kvcache_speedup": 1.076960735581149, |
| "gen_tokens": 128, |
| "peak_infer_vram_gb": 0.385, |
| "peak_train_vram_gb": 4.062, |
| "closeness_answer_ppl": 23.819, |
| "closeness_hit_rate": 0.45, |
| "closeness_token_f1": 0.014, |
| "closeness_n": 20, |
| "closeness_items": [ |
| { |
| "prompt": "The capital of France is", |
| "answer": " Paris", |
| "generated": " the city of Paris.\n", |
| "answer_nll": 2.9506, |
| "answer_ppl": 19.117, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The capital of Japan is", |
| "answer": " Tokyo", |
| "generated": " the capital of the Republic of", |
| "answer_nll": 2.6959, |
| "answer_ppl": 14.818, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The largest planet in the solar system is", |
| "answer": " Jupiter", |
| "generated": " the Sun, which is the", |
| "answer_nll": 2.8111, |
| "answer_ppl": 16.628, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "Water is made of hydrogen and", |
| "answer": " oxygen", |
| "generated": " oxygen.\nThe hydrogen is", |
| "answer_nll": 0.7412, |
| "answer_ppl": 2.099, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The opposite of hot is", |
| "answer": " cold", |
| "generated": " the case for the two-", |
| "answer_nll": 5.2195, |
| "answer_ppl": 184.842, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The opposite of up is", |
| "answer": " down", |
| "generated": " the case of the two-", |
| "answer_nll": 7.8645, |
| "answer_ppl": 2603.199, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "Two plus two equals", |
| "answer": " four", |
| "generated": " one.\nThe first two", |
| "answer_nll": 3.4823, |
| "answer_ppl": 32.534, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The sun rises in the", |
| "answer": " east", |
| "generated": " sky, and the sun rises", |
| "answer_nll": 3.5493, |
| "answer_ppl": 34.79, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The first president of the United States was George", |
| "answer": " Washington", |
| "generated": " Washington.\nThe first president", |
| "answer_nll": 0.3028, |
| "answer_ppl": 1.354, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The speed of light is approximately 300,000 kilometers per", |
| "answer": " second", |
| "generated": " second.\nThe speed of", |
| "answer_nll": 0.3426, |
| "answer_ppl": 1.409, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "A triangle has three", |
| "answer": " sides", |
| "generated": " sides.\n- The triangle", |
| "answer_nll": 1.2375, |
| "answer_ppl": 3.447, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The chemical symbol for gold is", |
| "answer": " Au", |
| "generated": " gold.\nThe gold symbol", |
| "answer_nll": 9.1208, |
| "answer_ppl": 9143.113, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The freezing point of water in Celsius is", |
| "answer": " zero", |
| "generated": " about 1.5 degrees Celsius", |
| "answer_nll": 4.3002, |
| "answer_ppl": 73.711, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The author of Romeo and Juliet is William", |
| "answer": " Shakespeare", |
| "generated": " Shakespeare.\nThe author of", |
| "answer_nll": 0.1833, |
| "answer_ppl": 1.201, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The Earth revolves around the", |
| "answer": " Sun", |
| "generated": " sun.\nThe Earth is", |
| "answer_nll": 0.9723, |
| "answer_ppl": 2.644, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The plural of mouse is", |
| "answer": " mice", |
| "generated": " the plural of mouse.\n", |
| "answer_nll": 7.0402, |
| "answer_ppl": 1141.636, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "Photosynthesis occurs in the", |
| "answer": " leaves", |
| "generated": " form of photosynthesis.\n", |
| "answer_nll": 3.8898, |
| "answer_ppl": 48.901, |
| "hit": false, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The human heart pumps", |
| "answer": " blood", |
| "generated": " blood and oxygen to the brain", |
| "answer_nll": 1.0091, |
| "answer_ppl": 2.743, |
| "hit": true, |
| "token_f1": 0.286 |
| }, |
| { |
| "prompt": "The currency used in the United States is the", |
| "answer": " dollar", |
| "generated": " United States dollar.\nThe", |
| "answer_nll": 2.9926, |
| "answer_ppl": 19.938, |
| "hit": true, |
| "token_f1": 0.0 |
| }, |
| { |
| "prompt": "The tallest animal in the world is the", |
| "answer": " giraffe", |
| "generated": " elephant.\nThe elephant is", |
| "answer_nll": 2.7043, |
| "answer_ppl": 14.944, |
| "hit": false, |
| "token_f1": 0.0 |
| } |
| ], |
| "generation_samples": { |
| "The most important thing about sleep is": "The most important thing about sleep is to look at your own body and how you feel.\n- Try to get out of bed every night.\n- Do you feel sleepy all the time?\n- Do you feel like you are having a bad day, no matter what the time?\n- Do you feel as though you can", |
| "Scientists have discovered that": "Scientists have discovered that, in a recent study of about 30,000 participants from the Australian and European Union, the amount of sunlight in the atmosphere of the planet is about one-third of the sun\u2019s mass.\n\u201cWe are seeing significant changes in the intensity of light, but the effects this type", |
| "In the history of mathematics,": "In the history of mathematics, the term \"physics\" was used to describe the operations of many different fields.\nThe science of physics has become more and more popular in recent years, with the number of scientific journals running, and the number of experimental physicists working on them. There are numerous other fields, and their roles vary", |
| "Python code for prime number": "Python code for prime number of integers.\n- The integer values of the integers for integers are integers and integers.\n- When the integers are integers, the integers are integers and integers.\n- The integer values of the integers are integers and integers.\n- The integers are integers\n- The integers are integers\n-", |
| "The best way to learn a new language is": "The best way to learn a new language is to learn about its features and benefits. It is especially important to study the language in a way that is appropriate for your child's age.\nIt is very important to keep in mind that the best way to learn a language is to learn some basic elements. For example, learn to write a sentence" |
| } |
| } |