{ "loss": [ 3.338397E+00, 4.320688E+00, 2.540753E+00 ], "time": [ 48320.8, 6582.2, 6607.6 ], "memo info": [ { "rank": 0, "allocated memory": 16409.380859375, "max allocated memory": 17661.94970703125 } ], "warm_up": 1 }