a2-task2-mars / metrics.json
FrenchKnuckles's picture
Upload metrics.json with huggingface_hub
44170fc verified
Raw History Blame Contribute Delete
5.07 kB
{
"name": "mars",
"optimizer": "mars",
"status": "ok",
"diverged_at_step": null,
"tokens_seen": 37806080,
"steps": 4615,
"total_steps": 4615,
"token_frac": 1.0,
"train_tokens_per_epoch": 37813248,
"final_val_loss": 4.083916366833703,
"final_bleu": 3.121595483131984,
"best_val_loss": 4.083916366833703,
"train_time_s": 1512.0,
"wall_time_s": 1655.0,
"sec_per_step": 0.32762018043336966,
"opt_step_ms": 41.85049692734608,
"optimizer_state_bytes": 500379920,
"peak_mem_gb": 3.12,
"total_params": 41698304,
"hparams": {
"name": "mars",
"lr": 0.002,
"betas": [
0.9,
0.95
],
"eps": 1e-08,
"weight_decay": 0.1,
"gamma": 0.025,
"warmup_steps": 300,
"schedule": "cosine",
"min_lr_frac": 0.1,
"batch_tokens": 8192,
"grad_clip": 1.0
},
"history": [
{
"val_loss": 5.5644740222707245,
"val_ppl": 260.98789378221204,
"bleu": 1.3933804109520285,
"bleu_human": 0.33815425204965927,
"bleu_n": 497,
"fraction": 0.1,
"step": 462,
"tokens_seen": 3784704,
"train_time_s": 151.0,
"wall_time_s": 164.6
},
{
"val_loss": 5.113784666392663,
"val_ppl": 166.29854990254802,
"bleu": 2.6254619253922757,
"bleu_human": 0.537738897277919,
"bleu_n": 497,
"fraction": 0.2,
"step": 923,
"tokens_seen": 7561216,
"train_time_s": 302.1,
"wall_time_s": 329.7
},
{
"val_loss": 4.862805895018096,
"val_ppl": 129.38673885545705,
"bleu": 2.1623057665899803,
"bleu_human": 0.3577395786141275,
"bleu_n": 497,
"fraction": 0.3,
"step": 1385,
"tokens_seen": 11345920,
"train_time_s": 453.6,
"wall_time_s": 495.5
},
{
"val_loss": 4.689373418502267,
"val_ppl": 108.78499578263424,
"bleu": 3.088920894442003,
"bleu_human": 0.6577046757389468,
"bleu_n": 497,
"fraction": 0.4,
"step": 1846,
"tokens_seen": 15122432,
"train_time_s": 604.6,
"wall_time_s": 660.9
},
{
"val_loss": 4.5270724500977835,
"val_ppl": 92.48740287394348,
"bleu": 2.8014124634589574,
"bleu_human": 0.558930500151262,
"bleu_n": 497,
"fraction": 0.5,
"step": 2308,
"tokens_seen": 18907136,
"train_time_s": 755.9,
"wall_time_s": 826.6
},
{
"val_loss": 4.389742649881667,
"val_ppl": 80.6196688292514,
"bleu": 2.9227675880427477,
"bleu_human": 0.49167359125807264,
"bleu_n": 497,
"fraction": 0.6,
"step": 2770,
"tokens_seen": 22691840,
"train_time_s": 907.2,
"wall_time_s": 992.5
},
{
"val_loss": 4.282210729879905,
"val_ppl": 72.40032076805875,
"bleu": 2.8767315006059877,
"bleu_human": 0.4865830846335313,
"bleu_n": 497,
"fraction": 0.7,
"step": 3231,
"tokens_seen": 26468352,
"train_time_s": 1058.2,
"wall_time_s": 1157.8
},
{
"val_loss": 4.188797038677841,
"val_ppl": 65.9434158486703,
"bleu": 2.796587438315598,
"bleu_human": 0.5671621194475668,
"bleu_n": 497,
"fraction": 0.8,
"step": 3693,
"tokens_seen": 30253056,
"train_time_s": 1209.6,
"wall_time_s": 1323.6
},
{
"val_loss": 4.122510206694202,
"val_ppl": 61.713962795464134,
"bleu": 2.8315733169687958,
"bleu_human": 0.5846257376730375,
"bleu_n": 497,
"fraction": 0.9,
"step": 4154,
"tokens_seen": 34029568,
"train_time_s": 1360.7,
"wall_time_s": 1489.3
},
{
"val_loss": 4.083916366833703,
"val_ppl": 59.37755938359462,
"bleu": 3.121595483131984,
"bleu_human": 0.5471262061546667,
"bleu_n": 497,
"fraction": 1.0,
"step": 4615,
"tokens_seen": 37806080,
"train_time_s": 1512.0,
"wall_time_s": 1655.0
}
],
"data_meta": {
"n_sources": 8290,
"n_train_sources": 7627,
"n_val_sources": 166,
"n_test_sources": 497,
"train_docs": 61016,
"val_docs": 1328,
"train_tokens": 37813248,
"val_tokens": 833536,
"reference_authors": [
"chunk_2",
"Meta-Llama-3-70B",
"Meta-Llama-3-70B-Instruct",
"Meta-Llama-3-8B",
"Meta-Llama-3-8B-Instruct",
"gpt-4o-2024-08-06",
"gpt-4o-mini-2024-07-18"
],
"tokens_per_word": 1.2684311662630021
},
"wandb_url": "https://wandb.ai/frenchknuckles-iiit-hyderabad/anlp-a2/runs/dec1586b",
"meta": {
"git_commit": "6a52970-dirty",
"host": "HP-Victus-16",
"platform": "Windows-11-10.0.26200-SP0",
"python": "3.13.12",
"torch": "2.14.0+cu126",
"cuda": "12.6",
"started_at": "2026-10-03T03:35:29+05:30",
"gpu": "NVIDIA GeForce RTX 3050 6GB Laptop GPU",
"gpu_memory_gb": 6.44
}
}