{
    "loss": [
        1.187568E+01,
        9.759526E+00
    ],
    "time": [
        9832.1,
        3407.1
    ],
    "memo info": [
        {
            "rank": 0,
            "allocated memory": 33781.49658203125,
            "max allocated memory": 33781.4970703125
        },
        {
            "rank": 4,
            "allocated memory": 37217.69384765625,
            "max allocated memory": 37217.6943359375
        }
    ],
    "warm_up": 1
}