1{2 "steps": 600,3 "best_val": -6.065922264944626,4 "wall_s": 168.56347799301147,5 "ex_per_s": 227.80735457767452,6 "peak_mem_mb": 86.469632,7 "n_params": 3200259,8 "device": "mps",9 "data_time_frac": 0.01094110779495412510}
main