1{
 2  "seed": 3145,
 3  "device": "cuda",
 4  "math_check": {
 5    "c4_composition_max_abs": 0.0,
 6    "true_mechanism_equivariance_max_abs": 0.0,
 7    "equivariance_penalty_exact_linear": 0.0,
 8    "equivariance_penalty_non_equivariant_linear": 1.9284594058990479
 9  },
10  "params_each_model": 2690,
11  "baseline": {
12    "train_mse": 0.000663807790260762,
13    "in_distribution_mse": 0.000933704199269414,
14    "heldout_rotation_mse": 0.0010833757696673274,
15    "heldout_equivariance_penalty": 0.0005097697721794248
16  },
17  "equivariant_shared": {
18    "train_mse": 0.0007101972005330026,
19    "in_distribution_mse": 0.0006173581350594759,
20    "heldout_rotation_mse": 0.0007219823892228305,
21    "heldout_equivariance_penalty": 9.187469549942762e-05
22  }
23}