{
    "loss": [
        13.7243,
        10.3488,
        8.6670,
        8.5412,
        8.5662
    ],
    "time": [
        6125.3,
        1149.7,
        689.6,
        670.6,
        696.7
    ],
    "memo info": [
        {
            "rank": 0,
            "allocated memory": 21305.24462890625,
            "max allocated memory": 35111.3681640625
        }
    ],
    "warm_up": 2
}