{ "loss": [ 4.121116E-01, 1.235786E+00, 2.368348E+00 ], "time": [ 15836.4, 11196.4, 11230.8 ], "memo info": [ { "rank": 0, "allocated memory": 11617.18212890625, "max allocated memory": 22532.8046875 } ], "warm_up": 1 }