all_results.json

{
    "epoch": 3.0,
    "eval_accuracy": 0.9084,
    "eval_loss": 0.28339844942092896,
    "eval_mem_cpu_alloc_delta": 1593344,
    "eval_mem_cpu_peaked_delta": 294912,
    "eval_mem_gpu_alloc_delta": 0,
    "eval_mem_gpu_peaked_delta": 94704128,
    "eval_runtime": 116.0156,
    "eval_samples": 10000,
    "eval_samples_per_second": 86.195,
    "init_mem_cpu_alloc_delta": 1990225920,
    "init_mem_cpu_peaked_delta": 378310656,
    "init_mem_gpu_alloc_delta": 499890176,
    "init_mem_gpu_peaked_delta": 0,
    "train_mem_cpu_alloc_delta": 879861760,
    "train_mem_cpu_peaked_delta": 80719872,
    "train_mem_gpu_alloc_delta": 1509100544,
    "train_mem_gpu_peaked_delta": 18135112704,
    "train_runtime": 16177.7165,
    "train_samples": 550152,
    "train_samples_per_second": 0.797
}