1{"event": "train", "step": 20, "loss": 0.0003037545830011368, "grad_norm": 0.22581392526626587, "lr": 0.001, "ex_per_s": 402.9437064123233, "data_frac": 0.7259477082976826, "mem_mb": 5.659648, "copula_nll": 0.0003037545830011368, "elapsed_s": 1.6}2{"event": "train", "step": 40, "loss": -0.00032298461883328855, "grad_norm": 0.11710090190172195, "lr": 0.0001, "ex_per_s": 627.0390671526434, "data_frac": 0.5778939993116096, "mem_mb": 5.659648, "copula_nll": -0.00032298461883328855, "elapsed_s": 2.0}3{"event": "val", "step": 40, "val_nll": -0.00032298461883328855, "best": -0.00032298461883328855, "bad": 0, "mem_mb": 5.659648, "elapsed_s": 2.1}4{"event": "done", "steps": 40, "best_val": -0.00032298461883328855, "wall_s": 2.0973660945892334, "ex_per_s": 610.2892591341744, "peak_mem_mb": 5.659648, "n_params": 190342, "device": "mps", "data_time_frac": 0.5624569810290814, "elapsed_s": 2.1}5