1{2 "steps": 12000,3 "best_val": -6.452572455371413,4 "wall_s": 4579.964827299118,5 "ex_per_s": 167.6868772926587,6 "peak_mem_mb": 402.962432,7 "n_params": 20718339,8 "device": "mps",9 "data_time_frac": 0.00415949293771662710}
013c8cc25081c2aa0aed28c2b71bcb00657ec293