| { |
| "step": 50000, |
| "weights_source": "ema", |
| "loss": 0.3709731566905975, |
| "edge_loss": 0.19136443853378296, |
| "diag_loss": 0.1796087145805359, |
| "jvp_rms": 2.2817168676853177, |
| "examples": 5120, |
| "loss_nfe_1": 0.39239816665649413, |
| "edge_loss_nfe_1": 0.21810964345932007, |
| "diag_loss_nfe_1": 0.1742885172367096, |
| "jvp_rms_nfe_1": 2.700194501876831, |
| "loss_nfe_2": 0.3679115891456604, |
| "edge_loss_nfe_2": 0.18770996630191802, |
| "diag_loss_nfe_2": 0.18020162284374236, |
| "jvp_rms_nfe_2": 2.536956226825714, |
| "loss_nfe_3": 0.35463637113571167, |
| "edge_loss_nfe_3": 0.1791027843952179, |
| "diag_loss_nfe_3": 0.17553358674049377, |
| "jvp_rms_nfe_3": 2.0643750607967375, |
| "loss_nfe_4": 0.38753209710121156, |
| "edge_loss_nfe_4": 0.19491184055805205, |
| "diag_loss_nfe_4": 0.19262024462223054, |
| "jvp_rms_nfe_4": 2.4871247947216033, |
| "loss_nfe_5": 0.3523875594139099, |
| "edge_loss_nfe_5": 0.17698795795440675, |
| "diag_loss_nfe_5": 0.17539960145950317, |
| "jvp_rms_nfe_5": 1.6199337542057037, |
| "raw_loss": 0.37272677540779114, |
| "raw_edge_loss": 0.19294691383838652, |
| "raw_diag_loss": 0.17977986514568328, |
| "raw_jvp_rms": 2.4338310968875883, |
| "raw_loss_nfe_1": 0.39853897094726565, |
| "raw_edge_loss_nfe_1": 0.2240687519311905, |
| "raw_diag_loss_nfe_1": 0.1744702249765396, |
| "raw_jvp_rms_nfe_1": 2.9251581192016602, |
| "raw_loss_nfe_2": 0.3692192077636719, |
| "raw_edge_loss_nfe_2": 0.18882804214954377, |
| "raw_diag_loss_nfe_2": 0.18039117157459258, |
| "raw_jvp_rms_nfe_2": 2.7281934142112734, |
| "raw_loss_nfe_3": 0.35528815984725953, |
| "raw_edge_loss_nfe_3": 0.1796249717473984, |
| "raw_diag_loss_nfe_3": 0.17566319108009337, |
| "raw_jvp_rms_nfe_3": 2.1835286557674407, |
| "raw_loss_nfe_4": 0.3877676367759705, |
| "raw_edge_loss_nfe_4": 0.19502896666526795, |
| "raw_diag_loss_nfe_4": 0.1927386701107025, |
| "raw_jvp_rms_nfe_4": 2.6403229236602783, |
| "raw_loss_nfe_5": 0.3528199017047882, |
| "raw_edge_loss_nfe_5": 0.1771838366985321, |
| "raw_diag_loss_nfe_5": 0.17563606798648834, |
| "raw_jvp_rms_nfe_5": 1.69195237159729 |
| } |
