all_results.json

{
    "epoch": 7.0,
    "eval_gen_len": 33.9532,
    "eval_loss": 1.9251619577407837,
    "eval_mem_cpu_alloc_delta": 2314445767,
    "eval_mem_cpu_peaked_delta": 26998734,
    "eval_mem_gpu_alloc_delta": 0,
    "eval_mem_gpu_peaked_delta": 3355142144,
    "eval_rouge1": 26.4352,
    "eval_rouge2": 8.9293,
    "eval_rougeL": 21.2622,
    "eval_rougeLsum": 21.5518,
    "eval_runtime": 1054.0821,
    "eval_samples": 10358,
    "eval_samples_per_second": 9.827,
    "init_mem_cpu_alloc_delta": 177892,
    "init_mem_cpu_peaked_delta": 18282,
    "init_mem_gpu_alloc_delta": 1200707584,
    "init_mem_gpu_peaked_delta": 0,
    "test_gen_len": 33.803,
    "test_loss": 2.005567789077759,
    "test_mem_cpu_alloc_delta": 3104676530,
    "test_mem_cpu_peaked_delta": 36246285,
    "test_mem_gpu_alloc_delta": 0,
    "test_mem_gpu_peaked_delta": 3357975040,
    "test_rouge1": 26.0756,
    "test_rouge2": 8.4669,
    "test_rougeL": 20.8167,
    "test_rougeLsum": 21.0822,
    "test_runtime": 1405.8563,
    "test_samples": 13920,
    "test_samples_per_second": 9.901,
    "train_mem_cpu_alloc_delta": 9871729,
    "train_mem_cpu_peaked_delta": 1025220847,
    "train_mem_gpu_alloc_delta": 3609088512,
    "train_mem_gpu_peaked_delta": 8271511040,
    "train_runtime": 64951.4124,
    "train_samples": 266367,
    "train_samples_per_second": 0.897
}