1{
 2  "device": "cuda",
 3  "math_checks": [
 4    {
 5      "gamma": 0.0,
 6      "pred_logdet": 0.0,
 7      "obs_logdet": 0.0,
 8      "pred_3x": 0.0,
 9      "obs_3x": 0.0,
10      "recon_max": 2.384185791015625e-07
11    },
12    {
13      "gamma": 0.25,
14      "pred_logdet": 1.0,
15      "obs_logdet": 1.0,
16      "pred_3x": 3.0,
17      "obs_3x": 3.0,
18      "recon_max": 4.76837158203125e-07
19    },
20    {
21      "gamma": 0.5,
22      "pred_logdet": 2.0,
23      "obs_logdet": 2.0,
24      "pred_3x": 6.0,
25      "obs_3x": 6.0,
26      "recon_max": 2.384185791015625e-07
27    },
28    {
29      "gamma": 1.0,
30      "pred_logdet": 4.0,
31      "obs_logdet": 4.0,
32      "pred_3x": 12.0,
33      "obs_3x": 12.0,
34      "recon_max": 2.384185791015625e-07
35    },
36    {
37      "gamma": 1.5,
38      "pred_logdet": 6.0,
39      "obs_logdet": 6.0,
40      "pred_3x": 18.0,
41      "obs_3x": 18.0,
42      "recon_max": 2.384185791015625e-07
43    },
44    {
45      "nonlinear_jacobian_logdet": 2.0182530879974365,
46      "analytic_logdet": 2.0182530879974365,
47      "jacobian_abs_error": 0.0,
48      "jacobian_sign": 1.0,
49      "nonlinear_inverse_max_error": 0.0
50    }
51  ],
52  "models": {
53    "2": {
54      "flow": [
55        1.5364100933074951,
56        1.2448464632034302
57      ],
58      "diag_mlp": [
59        4.465583324432373,
60        2.818814277648926
61      ]
62    },
63    "8": {
64      "flow": [
65        4.148821830749512,
66        1.2992744445800781
67      ],
68      "diag_mlp": [
69        17.779361724853516,
70        2.9177722930908203
71      ]
72    }
73  }
74}