{
 "steps": 600,
 "best_val": -6.065922264944626,
 "wall_s": 168.56347799301147,
 "ex_per_s": 227.80735457767452,
 "peak_mem_mb": 86.469632,
 "n_params": 3200259,
 "device": "mps",
 "data_time_frac": 0.010941107794954125
}