Upload folder using huggingface_hub

.gitattributes

@@ -39,3 +39,19 @@ action_decoder/a100_bsz2_accum8/eval/iter_000000500/plots/sample000_idx0_step20_
 action_decoder/a100_bsz2_accum8/eval/iter_000000500/plots/sample001_idx88192_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
 action_decoder/stats_cache/34fb6cb2fa2ab61a2b7e42b6ba7ba69fbfc931cc14a85efa60c192807eb7d3f6 filter=lfs diff=lfs merge=lfs -text
 action_decoder/stats_cache/c3a3b8813593fa80af344f41bcf83ab0587abc15fde41a1fbf256d0ab6a41ea7 filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample000_idx0_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample001_idx12598_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample002_idx25197_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample003_idx37796_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample004_idx50395_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample005_idx62994_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample006_idx75593_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/plots/sample007_idx88192_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample000_idx0_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample001_idx12598_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample002_idx25197_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample003_idx37796_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample004_idx50395_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample005_idx62994_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample006_idx75593_step20_seed0.png filter=lfs diff=lfs merge=lfs -text
+action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/plots/sample007_idx88192_step20_seed0.png filter=lfs diff=lfs merge=lfs -text

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/metrics.json

@@ -0,0 +1,203 @@
+{
+  "checkpoint": "/home/shadeform/pdt-mimic/mimic-video/model/checkpoints/vam/so101/w2a_so101_h100_chain_bsz2_accum8_full204/checkpoints/model/iter_000000400.pt",
+  "experiment": "w2a_so101_lerobot_v2w_push_that_thing_lr1.000e-04_layer20_bsz1",
+  "job_name": "w2a_so101_h100_chain_bsz2_accum8_full204",
+  "iteration": 400,
+  "num_samples": 8,
+  "stop_steps": [
+    20
+  ],
+  "seeds": [
+    0
+  ],
+  "aggregate": {
+    "dim_0_mse/mean": 1461.276626586914,
+    "dim_0_mse/std": 1061.5213048735209,
+    "dim_1_mse/mean": 6041.07283782959,
+    "dim_1_mse/std": 4149.4244429322,
+    "dim_2_mse/mean": 5366.032791137695,
+    "dim_2_mse/std": 3634.4103096856884,
+    "dim_3_mse/mean": 510.5099515914917,
+    "dim_3_mse/std": 432.89659555098416,
+    "dim_4_mse/mean": 2623.8505783081055,
+    "dim_4_mse/std": 3084.5666077938904,
+    "dim_5_mse/mean": 0.011891428315720987,
+    "dim_5_mse/std": 0.011407825530386924,
+    "first_10_mse/mean": 2876.4423217773438,
+    "first_10_mse/std": 1433.6812578567965,
+    "first_action_mse/mean": 2098.6193923950195,
+    "first_action_mse/std": 1786.3945203566966,
+    "mae/mean": 35.434043407440186,
+    "mae/std": 9.320731239368103,
+    "mse/mean": 2667.1257247924805,
+    "mse/std": 1182.6250231736287,
+    "mse_vs_train_mean_ratio/mean": 2.5409504059414467,
+    "mse_vs_train_mean_ratio/std": 1.0154809217623397,
+    "pred_dim_std_mean/mean": 13.589832067489624,
+    "pred_dim_std_mean/std": 0.026376613730386942,
+    "sample_idx/mean": 44095.625,
+    "sample_idx/std": 28867.744650983303,
+    "seed/mean": 0.0,
+    "seed/std": 0.0,
+    "stop_step/mean": 20.0,
+    "stop_step/std": 0.0,
+    "target_dim_std_mean/mean": 13.64388757944107,
+    "target_dim_std_mean/std": 10.03583875414237,
+    "train_mean_baseline_mse/mean": 1225.073329925537,
+    "train_mean_baseline_mse/std": 725.2491605310839
+  },
+  "rows": [
+    {
+      "sample_idx": 0.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 2233.7841796875,
+      "mae": 32.45598602294922,
+      "first_action_mse": 5126.89794921875,
+      "first_10_mse": 5146.58154296875,
+      "train_mean_baseline_mse": 1425.959228515625,
+      "mse_vs_train_mean_ratio": 1.5665133581784054,
+      "pred_dim_std_mean": 13.652911186218262,
+      "target_dim_std_mean": 24.68731117248535,
+      "dim_0_mse": 1125.2979736328125,
+      "dim_1_mse": 7007.08740234375,
+      "dim_2_mse": 3756.35498046875,
+      "dim_3_mse": 561.1243286132812,
+      "dim_4_mse": 952.840087890625,
+      "dim_5_mse": 0.0012400309788063169
+    },
+    {
+      "sample_idx": 12598.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 543.3341674804688,
+      "mae": 16.21319580078125,
+      "first_action_mse": 871.6549682617188,
+      "first_10_mse": 1086.154541015625,
+      "train_mean_baseline_mse": 317.4585876464844,
+      "mse_vs_train_mean_ratio": 1.711511953444192,
+      "pred_dim_std_mean": 13.580321311950684,
+      "target_dim_std_mean": 7.664527893066406,
+      "dim_0_mse": 312.3543701171875,
+      "dim_1_mse": 897.4060668945312,
+      "dim_2_mse": 1449.2069091796875,
+      "dim_3_mse": 67.71875,
+      "dim_4_mse": 533.3180541992188,
+      "dim_5_mse": 0.0009438408887945116
+    },
+    {
+      "sample_idx": 25197.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 2375.63037109375,
+      "mae": 37.68236541748047,
+      "first_action_mse": 767.85498046875,
+      "first_10_mse": 1939.217529296875,
+      "train_mean_baseline_mse": 614.4007568359375,
+      "mse_vs_train_mean_ratio": 3.866581127484045,
+      "pred_dim_std_mean": 13.560254096984863,
+      "target_dim_std_mean": 17.704574584960938,
+      "dim_0_mse": 2626.802734375,
+      "dim_1_mse": 3700.7880859375,
+      "dim_2_mse": 3750.95947265625,
+      "dim_3_mse": 1411.08056640625,
+      "dim_4_mse": 2764.140380859375,
+      "dim_5_mse": 0.01100786030292511
+    },
+    {
+      "sample_idx": 37796.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 2589.433349609375,
+      "mae": 35.55836868286133,
+      "first_action_mse": 938.2966918945312,
+      "first_10_mse": 2693.944580078125,
+      "train_mean_baseline_mse": 2664.1845703125,
+      "mse_vs_train_mean_ratio": 0.9719421764032073,
+      "pred_dim_std_mean": 13.573594093322754,
+      "target_dim_std_mean": 3.537752151489258,
+      "dim_0_mse": 3418.8798828125,
+      "dim_1_mse": 867.8055419921875,
+      "dim_2_mse": 712.677978515625,
+      "dim_3_mse": 189.31179809570312,
+      "dim_4_mse": 10347.888671875,
+      "dim_5_mse": 0.03633066639304161
+    },
+    {
+      "sample_idx": 50395.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 3595.659423828125,
+      "mae": 44.06079864501953,
+      "first_action_mse": 1210.1441650390625,
+      "first_10_mse": 2767.2900390625,
+      "train_mean_baseline_mse": 1120.7044677734375,
+      "mse_vs_train_mean_ratio": 3.2083921562049365,
+      "pred_dim_std_mean": 13.57802677154541,
+      "target_dim_std_mean": 19.682634353637695,
+      "dim_0_mse": 1704.6829833984375,
+      "dim_1_mse": 8432.4921875,
+      "dim_2_mse": 7650.2333984375,
+      "dim_3_mse": 476.8518981933594,
+      "dim_4_mse": 3309.67529296875,
+      "dim_5_mse": 0.02147996611893177
+    },
+    {
+      "sample_idx": 62994.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 3426.2587890625,
+      "mae": 40.874267578125,
+      "first_action_mse": 631.7477416992188,
+      "first_10_mse": 2045.358642578125,
+      "train_mean_baseline_mse": 1443.3743896484375,
+      "mse_vs_train_mean_ratio": 2.373783831578883,
+      "pred_dim_std_mean": 13.57933521270752,
+      "target_dim_std_mean": 29.554487228393555,
+      "dim_0_mse": 1694.8826904296875,
+      "dim_1_mse": 8427.49609375,
+      "dim_2_mse": 7619.115234375,
+      "dim_3_mse": 913.1915893554688,
+      "dim_4_mse": 1902.8548583984375,
+      "dim_5_mse": 0.01113315112888813
+    },
+    {
+      "sample_idx": 75593.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 1838.360107421875,
+      "mae": 28.54059600830078,
+      "first_action_mse": 2179.137451171875,
+      "first_10_mse": 2050.60693359375,
+      "train_mean_baseline_mse": 469.48388671875,
+      "mse_vs_train_mean_ratio": 3.915704371177209,
+      "pred_dim_std_mean": 13.595999717712402,
+      "target_dim_std_mean": 6.319813251495361,
+      "dim_0_mse": 262.0338134765625,
+      "dim_1_mse": 4871.71435546875,
+      "dim_2_mse": 5331.11181640625,
+      "dim_3_mse": 47.469444274902344,
+      "dim_4_mse": 517.8301391601562,
+      "dim_5_mse": 0.0010640461696311831
+    },
+    {
+      "sample_idx": 88192.0,
+      "stop_step": 20.0,
+      "seed": 0.0,
+      "mse": 4734.54541015625,
+      "mae": 48.086769104003906,
+      "first_action_mse": 5063.22119140625,
+      "first_10_mse": 5282.384765625,
+      "train_mean_baseline_mse": 1745.020751953125,
+      "mse_vs_train_mean_ratio": 2.7131742730606967,
+      "pred_dim_std_mean": 13.598214149475098,
+      "target_dim_std_mean": 0.0,
+      "dim_0_mse": 545.278564453125,
+      "dim_1_mse": 14123.79296875,
+      "dim_2_mse": 12658.6025390625,
+      "dim_3_mse": 417.33123779296875,
+      "dim_4_mse": 662.2571411132812,
+      "dim_5_mse": 0.01193186454474926
+    }
+  ]
+}

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000400/predictions.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42cca6e6c2f1d68b6a7a709e37ec17890a385df35c0b7ab8931fa2d7e6975002
+size 16769

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000500/predictions.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:966018abedfb09237a183868b73eff9b947215509a0c87c69895c8b9cc125e11
+size 404540

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000900/metrics.json

@@ -0,0 +1,1268 @@
+{
+  "checkpoint": "/home/shadeform/pdt-mimic/mimic-video/model/checkpoints/vam/so101/w2a_so101_h100_chain_bsz2_accum8_full204/checkpoints/model/iter_000000900.pt",
+  "experiment": "w2a_so101_lerobot_v2w_push_that_thing_lr1.000e-04_layer20_bsz1",
+  "job_name": "w2a_so101_h100_chain_bsz2_accum8_full204",
+  "iteration": 900,
+  "num_samples": 32,
+  "stop_steps": [
+    25,
+    30
+  ],
+  "seeds": [
+    0
+  ],
+  "aggregate": {
+    "dim_0_mse/mean": 959.1742196083069,
+    "dim_0_mse/std": 1082.600083901787,
+    "dim_1_mse/mean": 5874.7969262599945,
+    "dim_1_mse/std": 5198.5262123905095,
+    "dim_2_mse/mean": 5140.628059387207,
+    "dim_2_mse/std": 4149.954363834257,
+    "dim_3_mse/mean": 346.1906771659851,
+    "dim_3_mse/std": 422.7644699535581,
+    "dim_4_mse/mean": 3531.630857527256,
+    "dim_4_mse/std": 5446.081901797367,
+    "dim_5_mse/mean": 0.00835959113919671,
+    "dim_5_mse/std": 0.01043779928799182,
+    "first_10_mse/mean": 1630.5958142280579,
+    "first_10_mse/std": 1435.4786659323522,
+    "first_action_mse/mean": 1118.9428907632828,
+    "first_action_mse/std": 1114.2917307000123,
+    "mae/mean": 34.4750052690506,
+    "mae/std": 13.63711438672722,
+    "mse/mean": 2642.071483373642,
+    "mse/std": 1820.3583593235128,
+    "mse_vs_train_mean_ratio/mean": 2.8850554783394613,
+    "mse_vs_train_mean_ratio/std": 2.21035832932375,
+    "pred_dim_std_mean/mean": 7.625488109886646,
+    "pred_dim_std_mean/std": 0.4080450528943499,
+    "sample_idx/mean": 44095.53125,
+    "sample_idx/std": 26267.314313591778,
+    "seed/mean": 0.0,
+    "seed/std": 0.0,
+    "stop_step/mean": 27.5,
+    "stop_step/std": 2.5,
+    "target_dim_std_mean/mean": 13.532952778041363,
+    "target_dim_std_mean/std": 8.156953974648559,
+    "train_mean_baseline_mse/mean": 1216.5151710510254,
+    "train_mean_baseline_mse/std": 839.1808489232942
+  },
+  "rows": [
+    {
+      "sample_idx": 0.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2715.065185546875,
+      "mae": 34.65090560913086,
+      "first_action_mse": 2922.53515625,
+      "first_10_mse": 3899.421875,
+      "train_mean_baseline_mse": 1425.959228515625,
+      "mse_vs_train_mean_ratio": 1.904027219889846,
+      "pred_dim_std_mean": 8.684864044189453,
+      "target_dim_std_mean": 24.68731117248535,
+      "dim_0_mse": 788.7882080078125,
+      "dim_1_mse": 9160.27734375,
+      "dim_2_mse": 4424.37548828125,
+      "dim_3_mse": 227.98402404785156,
+      "dim_4_mse": 1688.9654541015625,
+      "dim_5_mse": 0.0019032795680686831
+    },
+    {
+      "sample_idx": 0.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2016.6685791015625,
+      "mae": 31.024789810180664,
+      "first_action_mse": 2878.653076171875,
+      "first_10_mse": 3175.3740234375,
+      "train_mean_baseline_mse": 1425.959228515625,
+      "mse_vs_train_mean_ratio": 1.414254025482093,
+      "pred_dim_std_mean": 7.491584300994873,
+      "target_dim_std_mean": 24.68731117248535,
+      "dim_0_mse": 1164.8232421875,
+      "dim_1_mse": 6708.5126953125,
+      "dim_2_mse": 2780.609619140625,
+      "dim_3_mse": 275.3041687011719,
+      "dim_4_mse": 1170.760986328125,
+      "dim_5_mse": 0.0008222791948355734
+    },
+    {
+      "sample_idx": 2844.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1401.08056640625,
+      "mae": 30.06824493408203,
+      "first_action_mse": 617.1722412109375,
+      "first_10_mse": 1095.452880859375,
+      "train_mean_baseline_mse": 303.8437805175781,
+      "mse_vs_train_mean_ratio": 4.611187248985647,
+      "pred_dim_std_mean": 7.491161346435547,
+      "target_dim_std_mean": 7.692286968231201,
+      "dim_0_mse": 549.3639526367188,
+      "dim_1_mse": 3100.04931640625,
+      "dim_2_mse": 2875.781005859375,
+      "dim_3_mse": 272.2027893066406,
+      "dim_4_mse": 1609.083984375,
+      "dim_5_mse": 0.001987881027162075
+    },
+    {
+      "sample_idx": 2844.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 639.587158203125,
+      "mae": 19.60783576965332,
+      "first_action_mse": 440.3863220214844,
+      "first_10_mse": 553.7069702148438,
+      "train_mean_baseline_mse": 303.8437805175781,
+      "mse_vs_train_mean_ratio": 2.104986836043278,
+      "pred_dim_std_mean": 8.044658660888672,
+      "target_dim_std_mean": 7.692286968231201,
+      "dim_0_mse": 84.91226959228516,
+      "dim_1_mse": 1335.9854736328125,
+      "dim_2_mse": 1400.6185302734375,
+      "dim_3_mse": 297.62359619140625,
+      "dim_4_mse": 718.3820190429688,
+      "dim_5_mse": 0.0009107801597565413
+    },
+    {
+      "sample_idx": 5689.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 830.2814331054688,
+      "mae": 20.149494171142578,
+      "first_action_mse": 262.9076843261719,
+      "first_10_mse": 541.1204833984375,
+      "train_mean_baseline_mse": 1491.99658203125,
+      "mse_vs_train_mean_ratio": 0.5564901710264631,
+      "pred_dim_std_mean": 7.460916996002197,
+      "target_dim_std_mean": 3.163846015930176,
+      "dim_0_mse": 3384.656982421875,
+      "dim_1_mse": 993.3194580078125,
+      "dim_2_mse": 410.0876770019531,
+      "dim_3_mse": 80.11597442626953,
+      "dim_4_mse": 113.50711822509766,
+      "dim_5_mse": 0.0015428331680595875
+    },
+    {
+      "sample_idx": 5689.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 367.48150634765625,
+      "mae": 12.813591957092285,
+      "first_action_mse": 219.7945098876953,
+      "first_10_mse": 253.09652709960938,
+      "train_mean_baseline_mse": 1491.99658203125,
+      "mse_vs_train_mean_ratio": 0.2463018419568734,
+      "pred_dim_std_mean": 7.605101108551025,
+      "target_dim_std_mean": 3.163846015930176,
+      "dim_0_mse": 1660.97216796875,
+      "dim_1_mse": 190.0421905517578,
+      "dim_2_mse": 71.99201202392578,
+      "dim_3_mse": 129.7708282470703,
+      "dim_4_mse": 152.11119079589844,
+      "dim_5_mse": 0.000812445767223835
+    },
+    {
+      "sample_idx": 8534.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6860.7900390625,
+      "mae": 59.11018371582031,
+      "first_action_mse": 1158.6317138671875,
+      "first_10_mse": 2611.316650390625,
+      "train_mean_baseline_mse": 1506.2615966796875,
+      "mse_vs_train_mean_ratio": 4.554846285788613,
+      "pred_dim_std_mean": 7.364991664886475,
+      "target_dim_std_mean": 7.548631191253662,
+      "dim_0_mse": 774.0993041992188,
+      "dim_1_mse": 21430.248046875,
+      "dim_2_mse": 17056.11328125,
+      "dim_3_mse": 444.03265380859375,
+      "dim_4_mse": 1460.238525390625,
+      "dim_5_mse": 0.01225382462143898
+    },
+    {
+      "sample_idx": 8534.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 4884.2978515625,
+      "mae": 47.257816314697266,
+      "first_action_mse": 852.66943359375,
+      "first_10_mse": 1779.12255859375,
+      "train_mean_baseline_mse": 1506.2615966796875,
+      "mse_vs_train_mean_ratio": 3.242662404942908,
+      "pred_dim_std_mean": 7.167270660400391,
+      "target_dim_std_mean": 7.548631191253662,
+      "dim_0_mse": 109.18769836425781,
+      "dim_1_mse": 15310.1015625,
+      "dim_2_mse": 12671.412109375,
+      "dim_3_mse": 418.2569580078125,
+      "dim_4_mse": 796.814208984375,
+      "dim_5_mse": 0.010142008773982525
+    },
+    {
+      "sample_idx": 11379.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2983.939697265625,
+      "mae": 42.4905891418457,
+      "first_action_mse": 535.5656127929688,
+      "first_10_mse": 1046.447509765625,
+      "train_mean_baseline_mse": 334.44451904296875,
+      "mse_vs_train_mean_ratio": 8.922076838945758,
+      "pred_dim_std_mean": 7.591184616088867,
+      "target_dim_std_mean": 12.383785247802734,
+      "dim_0_mse": 1739.261962890625,
+      "dim_1_mse": 6317.6494140625,
+      "dim_2_mse": 6865.60009765625,
+      "dim_3_mse": 291.3958435058594,
+      "dim_4_mse": 2689.71728515625,
+      "dim_5_mse": 0.01353488489985466
+    },
+    {
+      "sample_idx": 11379.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1665.0382080078125,
+      "mae": 30.569936752319336,
+      "first_action_mse": 445.4091796875,
+      "first_10_mse": 618.3870849609375,
+      "train_mean_baseline_mse": 334.44451904296875,
+      "mse_vs_train_mean_ratio": 4.97851844835852,
+      "pred_dim_std_mean": 6.839549541473389,
+      "target_dim_std_mean": 12.383785247802734,
+      "dim_0_mse": 577.1253051757812,
+      "dim_1_mse": 3309.48681640625,
+      "dim_2_mse": 4351.96826171875,
+      "dim_3_mse": 193.8451385498047,
+      "dim_4_mse": 1557.7923583984375,
+      "dim_5_mse": 0.01081220805644989
+    },
+    {
+      "sample_idx": 14224.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 4319.02197265625,
+      "mae": 48.88478469848633,
+      "first_action_mse": 439.7796325683594,
+      "first_10_mse": 1074.5601806640625,
+      "train_mean_baseline_mse": 815.0859985351562,
+      "mse_vs_train_mean_ratio": 5.298854329994925,
+      "pred_dim_std_mean": 7.240562438964844,
+      "target_dim_std_mean": 17.710800170898438,
+      "dim_0_mse": 1506.1805419921875,
+      "dim_1_mse": 11637.806640625,
+      "dim_2_mse": 9708.51953125,
+      "dim_3_mse": 389.98870849609375,
+      "dim_4_mse": 2671.619873046875,
+      "dim_5_mse": 0.013980611227452755
+    },
+    {
+      "sample_idx": 14224.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2766.281005859375,
+      "mae": 36.770774841308594,
+      "first_action_mse": 284.8279724121094,
+      "first_10_mse": 552.061279296875,
+      "train_mean_baseline_mse": 815.0859985351562,
+      "mse_vs_train_mean_ratio": 3.393851705011296,
+      "pred_dim_std_mean": 6.898399829864502,
+      "target_dim_std_mean": 17.710800170898438,
+      "dim_0_mse": 455.7159729003906,
+      "dim_1_mse": 7441.71875,
+      "dim_2_mse": 6833.09521484375,
+      "dim_3_mse": 305.18731689453125,
+      "dim_4_mse": 1561.9576416015625,
+      "dim_5_mse": 0.011256578378379345
+    },
+    {
+      "sample_idx": 17069.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1060.6859130859375,
+      "mae": 22.310083389282227,
+      "first_action_mse": 139.50262451171875,
+      "first_10_mse": 198.25814819335938,
+      "train_mean_baseline_mse": 1071.8345947265625,
+      "mse_vs_train_mean_ratio": 0.9895985055012437,
+      "pred_dim_std_mean": 7.802321910858154,
+      "target_dim_std_mean": 17.36097526550293,
+      "dim_0_mse": 294.5861511230469,
+      "dim_1_mse": 3023.650390625,
+      "dim_2_mse": 2272.47802734375,
+      "dim_3_mse": 133.62709045410156,
+      "dim_4_mse": 639.771728515625,
+      "dim_5_mse": 0.0015158759197220206
+    },
+    {
+      "sample_idx": 17069.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 546.5440063476562,
+      "mae": 15.56402587890625,
+      "first_action_mse": 302.4043884277344,
+      "first_10_mse": 173.56094360351562,
+      "train_mean_baseline_mse": 1071.8345947265625,
+      "mse_vs_train_mean_ratio": 0.5099145045668972,
+      "pred_dim_std_mean": 8.225049018859863,
+      "target_dim_std_mean": 17.36097526550293,
+      "dim_0_mse": 238.55789184570312,
+      "dim_1_mse": 1356.126220703125,
+      "dim_2_mse": 1245.5567626953125,
+      "dim_3_mse": 176.73402404785156,
+      "dim_4_mse": 262.2875061035156,
+      "dim_5_mse": 0.0013583713443949819
+    },
+    {
+      "sample_idx": 19914.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 961.4107055664062,
+      "mae": 19.99274444580078,
+      "first_action_mse": 216.9536895751953,
+      "first_10_mse": 137.30335998535156,
+      "train_mean_baseline_mse": 1155.8238525390625,
+      "mse_vs_train_mean_ratio": 0.831796906989263,
+      "pred_dim_std_mean": 7.5299835205078125,
+      "target_dim_std_mean": 18.041248321533203,
+      "dim_0_mse": 210.5591583251953,
+      "dim_1_mse": 2606.407958984375,
+      "dim_2_mse": 2340.55859375,
+      "dim_3_mse": 97.37569427490234,
+      "dim_4_mse": 513.5570678710938,
+      "dim_5_mse": 0.0059148152358829975
+    },
+    {
+      "sample_idx": 19914.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 525.2775268554688,
+      "mae": 15.354072570800781,
+      "first_action_mse": 447.7425842285156,
+      "first_10_mse": 265.51702880859375,
+      "train_mean_baseline_mse": 1155.8238525390625,
+      "mse_vs_train_mean_ratio": 0.45446157362262635,
+      "pred_dim_std_mean": 7.907943248748779,
+      "target_dim_std_mean": 18.041248321533203,
+      "dim_0_mse": 289.09722900390625,
+      "dim_1_mse": 1158.327392578125,
+      "dim_2_mse": 1363.546875,
+      "dim_3_mse": 134.29513549804688,
+      "dim_4_mse": 206.3945770263672,
+      "dim_5_mse": 0.0039688111282885075
+    },
+    {
+      "sample_idx": 22759.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1407.662109375,
+      "mae": 26.80771827697754,
+      "first_action_mse": 208.8956298828125,
+      "first_10_mse": 250.3403778076172,
+      "train_mean_baseline_mse": 498.5845031738281,
+      "mse_vs_train_mean_ratio": 2.823317011287509,
+      "pred_dim_std_mean": 7.576528549194336,
+      "target_dim_std_mean": 11.284893035888672,
+      "dim_0_mse": 408.5557861328125,
+      "dim_1_mse": 3476.36865234375,
+      "dim_2_mse": 3665.33544921875,
+      "dim_3_mse": 27.38680648803711,
+      "dim_4_mse": 868.3234252929688,
+      "dim_5_mse": 0.0027330187149345875
+    },
+    {
+      "sample_idx": 22759.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 704.1707153320312,
+      "mae": 18.167448043823242,
+      "first_action_mse": 369.0237121582031,
+      "first_10_mse": 223.81118774414062,
+      "train_mean_baseline_mse": 498.5845031738281,
+      "mse_vs_train_mean_ratio": 1.4123397555469688,
+      "pred_dim_std_mean": 7.855295181274414,
+      "target_dim_std_mean": 11.284893035888672,
+      "dim_0_mse": 150.72879028320312,
+      "dim_1_mse": 1575.941650390625,
+      "dim_2_mse": 2078.214111328125,
+      "dim_3_mse": 45.04861068725586,
+      "dim_4_mse": 375.08966064453125,
+      "dim_5_mse": 0.0013280232669785619
+    },
+    {
+      "sample_idx": 25604.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 5329.5185546875,
+      "mae": 54.30661392211914,
+      "first_action_mse": 557.8172607421875,
+      "first_10_mse": 1223.635498046875,
+      "train_mean_baseline_mse": 1151.5936279296875,
+      "mse_vs_train_mean_ratio": 4.6279507157996385,
+      "pred_dim_std_mean": 7.213366985321045,
+      "target_dim_std_mean": 16.87040901184082,
+      "dim_0_mse": 1116.46533203125,
+      "dim_1_mse": 15264.2958984375,
+      "dim_2_mse": 11877.1904296875,
+      "dim_3_mse": 898.1690673828125,
+      "dim_4_mse": 2820.97998046875,
+      "dim_5_mse": 0.01213650219142437
+    },
+    {
+      "sample_idx": 25604.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 3612.5458984375,
+      "mae": 42.29384231567383,
+      "first_action_mse": 365.1740417480469,
+      "first_10_mse": 652.3385009765625,
+      "train_mean_baseline_mse": 1151.5936279296875,
+      "mse_vs_train_mean_ratio": 3.136997123657296,
+      "pred_dim_std_mean": 6.949882507324219,
+      "target_dim_std_mean": 16.87040901184082,
+      "dim_0_mse": 261.9459533691406,
+      "dim_1_mse": 10307.392578125,
+      "dim_2_mse": 8571.0048828125,
+      "dim_3_mse": 891.0440673828125,
+      "dim_4_mse": 1643.8778076171875,
+      "dim_5_mse": 0.009563191793859005
+    },
+    {
+      "sample_idx": 28449.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3544.0341796875,
+      "mae": 45.997127532958984,
+      "first_action_mse": 554.0211791992188,
+      "first_10_mse": 1327.8612060546875,
+      "train_mean_baseline_mse": 539.5875244140625,
+      "mse_vs_train_mean_ratio": 6.568043216966446,
+      "pred_dim_std_mean": 7.39085578918457,
+      "target_dim_std_mean": 17.05885124206543,
+      "dim_0_mse": 1713.95751953125,
+      "dim_1_mse": 8411.9609375,
+      "dim_2_mse": 7018.12646484375,
+      "dim_3_mse": 509.2244873046875,
+      "dim_4_mse": 3610.922119140625,
+      "dim_5_mse": 0.014149135909974575
+    },
+    {
+      "sample_idx": 28449.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2121.2431640625,
+      "mae": 33.79851531982422,
+      "first_action_mse": 383.9705505371094,
+      "first_10_mse": 757.2141723632812,
+      "train_mean_baseline_mse": 539.5875244140625,
+      "mse_vs_train_mean_ratio": 3.931230927486612,
+      "pred_dim_std_mean": 6.923903942108154,
+      "target_dim_std_mean": 17.05885124206543,
+      "dim_0_mse": 589.8446655273438,
+      "dim_1_mse": 4910.93994140625,
+      "dim_2_mse": 4541.3671875,
+      "dim_3_mse": 363.3932189941406,
+      "dim_4_mse": 2321.903564453125,
+      "dim_5_mse": 0.01121588796377182
+    },
+    {
+      "sample_idx": 31293.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3472.199462890625,
+      "mae": 44.87058639526367,
+      "first_action_mse": 696.8248901367188,
+      "first_10_mse": 1689.8201904296875,
+      "train_mean_baseline_mse": 370.26336669921875,
+      "mse_vs_train_mean_ratio": 9.377647845219443,
+      "pred_dim_std_mean": 8.103667259216309,
+      "target_dim_std_mean": 4.815817356109619,
+      "dim_0_mse": 448.9766540527344,
+      "dim_1_mse": 6460.17724609375,
+      "dim_2_mse": 7734.53076171875,
+      "dim_3_mse": 55.04722213745117,
+      "dim_4_mse": 6134.45556640625,
+      "dim_5_mse": 0.010775418020784855
+    },
+    {
+      "sample_idx": 31293.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2150.61328125,
+      "mae": 34.51803970336914,
+      "first_action_mse": 630.5663452148438,
+      "first_10_mse": 1087.4674072265625,
+      "train_mean_baseline_mse": 370.26336669921875,
+      "mse_vs_train_mean_ratio": 5.808333944624444,
+      "pred_dim_std_mean": 7.744465351104736,
+      "target_dim_std_mean": 4.815817356109619,
+      "dim_0_mse": 57.42213439941406,
+      "dim_1_mse": 3519.191650390625,
+      "dim_2_mse": 5048.54443359375,
+      "dim_3_mse": 109.93472290039062,
+      "dim_4_mse": 4168.57763671875,
+      "dim_5_mse": 0.007806057576090097
+    },
+    {
+      "sample_idx": 34138.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6049.15380859375,
+      "mae": 54.196434020996094,
+      "first_action_mse": 425.0455017089844,
+      "first_10_mse": 1088.71337890625,
+      "train_mean_baseline_mse": 1418.0733642578125,
+      "mse_vs_train_mean_ratio": 4.265755186622337,
+      "pred_dim_std_mean": 8.04835033416748,
+      "target_dim_std_mean": 25.433252334594727,
+      "dim_0_mse": 719.9398193359375,
+      "dim_1_mse": 17998.09765625,
+      "dim_2_mse": 15114.9599609375,
+      "dim_3_mse": 1040.057373046875,
+      "dim_4_mse": 1421.8592529296875,
+      "dim_5_mse": 0.008351304568350315
+    },
+    {
+      "sample_idx": 34138.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 4324.2314453125,
+      "mae": 43.20240020751953,
+      "first_action_mse": 190.9414520263672,
+      "first_10_mse": 608.5055541992188,
+      "train_mean_baseline_mse": 1418.0733642578125,
+      "mse_vs_train_mean_ratio": 3.0493707549296682,
+      "pred_dim_std_mean": 8.273107528686523,
+      "target_dim_std_mean": 25.433252334594727,
+      "dim_0_mse": 242.57608032226562,
+      "dim_1_mse": 12330.7705078125,
+      "dim_2_mse": 11277.2431640625,
+      "dim_3_mse": 977.1146850585938,
+      "dim_4_mse": 1117.677734375,
+      "dim_5_mse": 0.00759040005505085
+    },
+    {
+      "sample_idx": 36983.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2395.68212890625,
+      "mae": 38.63996124267578,
+      "first_action_mse": 433.5355224609375,
+      "first_10_mse": 1064.1497802734375,
+      "train_mean_baseline_mse": 267.7073974609375,
+      "mse_vs_train_mean_ratio": 8.948882816194184,
+      "pred_dim_std_mean": 7.265344142913818,
+      "target_dim_std_mean": 10.789891242980957,
+      "dim_0_mse": 2105.499755859375,
+      "dim_1_mse": 4243.37841796875,
+      "dim_2_mse": 4237.39501953125,
+      "dim_3_mse": 97.23124694824219,
+      "dim_4_mse": 3690.583251953125,
+      "dim_5_mse": 0.0048433090560138226
+    },
+    {
+      "sample_idx": 36983.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1236.5062255859375,
+      "mae": 26.694618225097656,
+      "first_action_mse": 296.2477722167969,
+      "first_10_mse": 558.1792602539062,
+      "train_mean_baseline_mse": 267.7073974609375,
+      "mse_vs_train_mean_ratio": 4.618872086888679,
+      "pred_dim_std_mean": 6.917026996612549,
+      "target_dim_std_mean": 10.789891242980957,
+      "dim_0_mse": 763.3260498046875,
+      "dim_1_mse": 1962.2998046875,
+      "dim_2_mse": 2336.3115234375,
+      "dim_3_mse": 54.35347366333008,
+      "dim_4_mse": 2302.74365234375,
+      "dim_5_mse": 0.0033966065384447575
+    },
+    {
+      "sample_idx": 39828.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 485.36273193359375,
+      "mae": 13.898509979248047,
+      "first_action_mse": 430.238525390625,
+      "first_10_mse": 230.33734130859375,
+      "train_mean_baseline_mse": 1447.6033935546875,
+      "mse_vs_train_mean_ratio": 0.33528709181991684,
+      "pred_dim_std_mean": 7.894618511199951,
+      "target_dim_std_mean": 14.908316612243652,
+      "dim_0_mse": 200.86398315429688,
+      "dim_1_mse": 1362.03369140625,
+      "dim_2_mse": 1069.41943359375,
+      "dim_3_mse": 131.3645782470703,
+      "dim_4_mse": 148.4923553466797,
+      "dim_5_mse": 0.0023813883308321238
+    },
+    {
+      "sample_idx": 39828.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 337.99322509765625,
+      "mae": 13.010416984558105,
+      "first_action_mse": 603.8759155273438,
+      "first_10_mse": 394.30499267578125,
+      "train_mean_baseline_mse": 1447.6033935546875,
+      "mse_vs_train_mean_ratio": 0.2334846868987307,
+      "pred_dim_std_mean": 7.41008996963501,
+      "target_dim_std_mean": 14.908316612243652,
+      "dim_0_mse": 639.6947631835938,
+      "dim_1_mse": 501.16265869140625,
+      "dim_2_mse": 570.2061767578125,
+      "dim_3_mse": 196.0409698486328,
+      "dim_4_mse": 120.85369110107422,
+      "dim_5_mse": 0.0010942247463390231
+    },
+    {
+      "sample_idx": 42673.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 4449.744140625,
+      "mae": 52.42909240722656,
+      "first_action_mse": 1757.181640625,
+      "first_10_mse": 2621.70751953125,
+      "train_mean_baseline_mse": 1557.8040771484375,
+      "mse_vs_train_mean_ratio": 2.856420910625849,
+      "pred_dim_std_mean": 8.12053394317627,
+      "target_dim_std_mean": 4.832499027252197,
+      "dim_0_mse": 1961.0687255859375,
+      "dim_1_mse": 6430.5654296875,
+      "dim_2_mse": 7824.33837890625,
+      "dim_3_mse": 36.615970611572266,
+      "dim_4_mse": 10445.869140625,
+      "dim_5_mse": 0.008305867202579975
+    },
+    {
+      "sample_idx": 42673.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2985.636962890625,
+      "mae": 41.77505874633789,
+      "first_action_mse": 1247.9993896484375,
+      "first_10_mse": 1834.303955078125,
+      "train_mean_baseline_mse": 1557.8040771484375,
+      "mse_vs_train_mean_ratio": 1.9165676908201688,
+      "pred_dim_std_mean": 7.806450366973877,
+      "target_dim_std_mean": 4.832499027252197,
+      "dim_0_mse": 772.3967895507812,
+      "dim_1_mse": 3318.612548828125,
+      "dim_2_mse": 4783.6513671875,
+      "dim_3_mse": 73.78472137451172,
+      "dim_4_mse": 8965.369140625,
+      "dim_5_mse": 0.00767466239631176
+    },
+    {
+      "sample_idx": 45518.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 723.506103515625,
+      "mae": 20.32315444946289,
+      "first_action_mse": 125.28692626953125,
+      "first_10_mse": 425.4751892089844,
+      "train_mean_baseline_mse": 696.74365234375,
+      "mse_vs_train_mean_ratio": 1.0384107570723462,
+      "pred_dim_std_mean": 7.70839262008667,
+      "target_dim_std_mean": 3.098416566848755,
+      "dim_0_mse": 2328.1669921875,
+      "dim_1_mse": 801.3797607421875,
+      "dim_2_mse": 782.6140747070312,
+      "dim_3_mse": 25.084028244018555,
+      "dim_4_mse": 403.78961181640625,
+      "dim_5_mse": 0.0019253201317042112
+    },
+    {
+      "sample_idx": 45518.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 241.8430938720703,
+      "mae": 11.172924041748047,
+      "first_action_mse": 98.37883758544922,
+      "first_10_mse": 144.98728942871094,
+      "train_mean_baseline_mse": 696.74365234375,
+      "mse_vs_train_mean_ratio": 0.34710483986261426,
+      "pred_dim_std_mean": 7.510776519775391,
+      "target_dim_std_mean": 3.098416566848755,
+      "dim_0_mse": 963.9204711914062,
+      "dim_1_mse": 121.45860290527344,
+      "dim_2_mse": 152.17501831054688,
+      "dim_3_mse": 64.19721984863281,
+      "dim_4_mse": 149.3065948486328,
+      "dim_5_mse": 0.0008210923988372087
+    },
+    {
+      "sample_idx": 48363.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2606.148193359375,
+      "mae": 36.855403900146484,
+      "first_action_mse": 288.2005920410156,
+      "first_10_mse": 1161.7510986328125,
+      "train_mean_baseline_mse": 493.635009765625,
+      "mse_vs_train_mean_ratio": 5.279504374288118,
+      "pred_dim_std_mean": 7.552104473114014,
+      "target_dim_std_mean": 11.497101783752441,
+      "dim_0_mse": 502.9095458984375,
+      "dim_1_mse": 7816.0341796875,
+      "dim_2_mse": 6141.27587890625,
+      "dim_3_mse": 185.62344360351562,
+      "dim_4_mse": 991.0426025390625,
+      "dim_5_mse": 0.0033503214363008738
+    },
+    {
+      "sample_idx": 48363.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1533.8013916015625,
+      "mae": 26.82679557800293,
+      "first_action_mse": 388.84912109375,
+      "first_10_mse": 720.2278442382812,
+      "train_mean_baseline_mse": 493.635009765625,
+      "mse_vs_train_mean_ratio": 3.107156828948989,
+      "pred_dim_std_mean": 7.789622783660889,
+      "target_dim_std_mean": 11.497101783752441,
+      "dim_0_mse": 95.84999084472656,
+      "dim_1_mse": 4646.46240234375,
+      "dim_2_mse": 3933.49853515625,
+      "dim_3_mse": 148.35191345214844,
+      "dim_4_mse": 378.6426086425781,
+      "dim_5_mse": 0.002613321878015995
+    },
+    {
+      "sample_idx": 51208.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 804.0179443359375,
+      "mae": 18.079256057739258,
+      "first_action_mse": 173.8828582763672,
+      "first_10_mse": 247.3124237060547,
+      "train_mean_baseline_mse": 1083.5904541015625,
+      "mse_vs_train_mean_ratio": 0.7419943035605394,
+      "pred_dim_std_mean": 7.529042720794678,
+      "target_dim_std_mean": 12.13278579711914,
+      "dim_0_mse": 69.78089904785156,
+      "dim_1_mse": 2162.550537109375,
+      "dim_2_mse": 2399.46044921875,
+      "dim_3_mse": 41.68958282470703,
+      "dim_4_mse": 150.6251678466797,
+      "dim_5_mse": 0.0010511609725654125
+    },
+    {
+      "sample_idx": 51208.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 402.6760559082031,
+      "mae": 13.382003784179688,
+      "first_action_mse": 303.268310546875,
+      "first_10_mse": 177.66864013671875,
+      "train_mean_baseline_mse": 1083.5904541015625,
+      "mse_vs_train_mean_ratio": 0.37161277527317643,
+      "pred_dim_std_mean": 7.704120635986328,
+      "target_dim_std_mean": 12.13278579711914,
+      "dim_0_mse": 365.4229736328125,
+      "dim_1_mse": 739.2015380859375,
+      "dim_2_mse": 1226.5093994140625,
+      "dim_3_mse": 29.318056106567383,
+      "dim_4_mse": 55.602298736572266,
+      "dim_5_mse": 0.0019595676567405462
+    },
+    {
+      "sample_idx": 54053.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1202.6207275390625,
+      "mae": 23.624732971191406,
+      "first_action_mse": 186.303466796875,
+      "first_10_mse": 169.11219787597656,
+      "train_mean_baseline_mse": 786.1039428710938,
+      "mse_vs_train_mean_ratio": 1.5298495045664333,
+      "pred_dim_std_mean": 7.550967693328857,
+      "target_dim_std_mean": 13.092276573181152,
+      "dim_0_mse": 182.79583740234375,
+      "dim_1_mse": 3435.501708984375,
+      "dim_2_mse": 2984.2822265625,
+      "dim_3_mse": 59.63194274902344,
+      "dim_4_mse": 553.5067749023438,
+      "dim_5_mse": 0.0054451837204396725
+    },
+    {
+      "sample_idx": 54053.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 612.5374755859375,
+      "mae": 16.881973266601562,
+      "first_action_mse": 377.225830078125,
+      "first_10_mse": 206.14695739746094,
+      "train_mean_baseline_mse": 786.1039428710938,
+      "mse_vs_train_mean_ratio": 0.7792067208679325,
+      "pred_dim_std_mean": 7.756974697113037,
+      "target_dim_std_mean": 13.092276573181152,
+      "dim_0_mse": 185.9036407470703,
+      "dim_1_mse": 1566.822509765625,
+      "dim_2_mse": 1617.1993408203125,
+      "dim_3_mse": 96.96597290039062,
+      "dim_4_mse": 208.3300323486328,
+      "dim_5_mse": 0.0034583196975290775
+    },
+    {
+      "sample_idx": 56898.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2802.240234375,
+      "mae": 42.788455963134766,
+      "first_action_mse": 1689.3046875,
+      "first_10_mse": 2807.032958984375,
+      "train_mean_baseline_mse": 685.4307250976562,
+      "mse_vs_train_mean_ratio": 4.0882909559908525,
+      "pred_dim_std_mean": 7.610069274902344,
+      "target_dim_std_mean": 5.5619215965271,
+      "dim_0_mse": 1792.7730712890625,
+      "dim_1_mse": 4171.8076171875,
+      "dim_2_mse": 5641.701171875,
+      "dim_3_mse": 69.11458587646484,
+      "dim_4_mse": 5138.0439453125,
+      "dim_5_mse": 0.0003970676043536514
+    },
+    {
+      "sample_idx": 56898.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1605.24755859375,
+      "mae": 31.64064598083496,
+      "first_action_mse": 1205.5025634765625,
+      "first_10_mse": 1940.1517333984375,
+      "train_mean_baseline_mse": 685.4307250976562,
+      "mse_vs_train_mean_ratio": 2.3419544817823033,
+      "pred_dim_std_mean": 7.245655059814453,
+      "target_dim_std_mean": 5.5619215965271,
+      "dim_0_mse": 616.651123046875,
+      "dim_1_mse": 1843.4847412109375,
+      "dim_2_mse": 3093.47021484375,
+      "dim_3_mse": 145.12986755371094,
+      "dim_4_mse": 3932.749755859375,
+      "dim_5_mse": 0.00018513997201807797
+    },
+    {
+      "sample_idx": 59742.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2726.7041015625,
+      "mae": 32.02950668334961,
+      "first_action_mse": 3632.133544921875,
+      "first_10_mse": 4839.7802734375,
+      "train_mean_baseline_mse": 1895.6773681640625,
+      "mse_vs_train_mean_ratio": 1.4383798358068047,
+      "pred_dim_std_mean": 7.945503234863281,
+      "target_dim_std_mean": 28.8035945892334,
+      "dim_0_mse": 804.0729370117188,
+      "dim_1_mse": 8572.140625,
+      "dim_2_mse": 5833.3583984375,
+      "dim_3_mse": 139.1013946533203,
+      "dim_4_mse": 1011.5481567382812,
+      "dim_5_mse": 0.004833984188735485
+    },
+    {
+      "sample_idx": 59742.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2211.062255859375,
+      "mae": 32.000701904296875,
+      "first_action_mse": 3424.454345703125,
+      "first_10_mse": 3888.603759765625,
+      "train_mean_baseline_mse": 1895.6773681640625,
+      "mse_vs_train_mean_ratio": 1.1663705507022846,
+      "pred_dim_std_mean": 7.3237762451171875,
+      "target_dim_std_mean": 28.8035945892334,
+      "dim_0_mse": 1295.27685546875,
+      "dim_1_mse": 6701.28466796875,
+      "dim_2_mse": 4211.25146484375,
+      "dim_3_mse": 223.51458740234375,
+      "dim_4_mse": 835.040283203125,
+      "dim_5_mse": 0.00547078438103199
+    },
+    {
+      "sample_idx": 62587.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6161.84130859375,
+      "mae": 58.4490966796875,
+      "first_action_mse": 3290.145751953125,
+      "first_10_mse": 4467.908203125,
+      "train_mean_baseline_mse": 2114.847412109375,
+      "mse_vs_train_mean_ratio": 2.9136103500005484,
+      "pred_dim_std_mean": 7.699039936065674,
+      "target_dim_std_mean": 28.069997787475586,
+      "dim_0_mse": 986.3435668945312,
+      "dim_1_mse": 12135.8017578125,
+      "dim_2_mse": 14045.5107421875,
+      "dim_3_mse": 2320.978271484375,
+      "dim_4_mse": 7482.40673828125,
+      "dim_5_mse": 0.00627899169921875
+    },
+    {
+      "sample_idx": 62587.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 4457.4365234375,
+      "mae": 49.39334487915039,
+      "first_action_mse": 2681.427001953125,
+      "first_10_mse": 3513.129150390625,
+      "train_mean_baseline_mse": 2114.847412109375,
+      "mse_vs_train_mean_ratio": 2.1076870595555626,
+      "pred_dim_std_mean": 7.443454265594482,
+      "target_dim_std_mean": 28.069997787475586,
+      "dim_0_mse": 549.8233642578125,
+      "dim_1_mse": 7422.24462890625,
+      "dim_2_mse": 9757.7333984375,
+      "dim_3_mse": 1941.31982421875,
+      "dim_4_mse": 7073.49462890625,
+      "dim_5_mse": 0.005352274514734745
+    },
+    {
+      "sample_idx": 65432.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 5303.484375,
+      "mae": 51.747352600097656,
+      "first_action_mse": 2868.329345703125,
+      "first_10_mse": 4017.13671875,
+      "train_mean_baseline_mse": 1502.2025146484375,
+      "mse_vs_train_mean_ratio": 3.5304723053543694,
+      "pred_dim_std_mean": 8.719014167785645,
+      "target_dim_std_mean": 10.998623847961426,
+      "dim_0_mse": 898.2756958007812,
+      "dim_1_mse": 17906.16015625,
+      "dim_2_mse": 11243.4658203125,
+      "dim_3_mse": 212.44583129882812,
+      "dim_4_mse": 1560.5531005859375,
+      "dim_5_mse": 0.007430012803524733
+    },
+    {
+      "sample_idx": 65432.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 3674.107177734375,
+      "mae": 41.829261779785156,
+      "first_action_mse": 2838.508544921875,
+      "first_10_mse": 3282.005126953125,
+      "train_mean_baseline_mse": 1502.2025146484375,
+      "mse_vs_train_mean_ratio": 2.445813491794235,
+      "pred_dim_std_mean": 7.483475208282471,
+      "target_dim_std_mean": 10.998623847961426,
+      "dim_0_mse": 465.36505126953125,
+      "dim_1_mse": 12774.7412109375,
+      "dim_2_mse": 7880.12060546875,
+      "dim_3_mse": 205.19236755371094,
+      "dim_4_mse": 719.2183837890625,
+      "dim_5_mse": 0.0053926254622638226
+    },
+    {
+      "sample_idx": 68277.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2993.975341796875,
+      "mae": 41.27082061767578,
+      "first_action_mse": 627.234375,
+      "first_10_mse": 1168.13232421875,
+      "train_mean_baseline_mse": 1187.5592041015625,
+      "mse_vs_train_mean_ratio": 2.5211167000822843,
+      "pred_dim_std_mean": 8.103791236877441,
+      "target_dim_std_mean": 24.93874168395996,
+      "dim_0_mse": 2061.515869140625,
+      "dim_1_mse": 5656.28662109375,
+      "dim_2_mse": 5771.4560546875,
+      "dim_3_mse": 871.6065673828125,
+      "dim_4_mse": 3602.9853515625,
+      "dim_5_mse": 0.00321180559694767
+    },
+    {
+      "sample_idx": 68277.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1886.2069091796875,
+      "mae": 30.722890853881836,
+      "first_action_mse": 330.8672180175781,
+      "first_10_mse": 668.3269653320312,
+      "train_mean_baseline_mse": 1187.5592041015625,
+      "mse_vs_train_mean_ratio": 1.588305578926215,
+      "pred_dim_std_mean": 7.860921382904053,
+      "target_dim_std_mean": 24.93874168395996,
+      "dim_0_mse": 1009.1412963867188,
+      "dim_1_mse": 2951.9541015625,
+      "dim_2_mse": 3750.028564453125,
+      "dim_3_mse": 648.52392578125,
+      "dim_4_mse": 2957.59033203125,
+      "dim_5_mse": 0.0028208415023982525
+    },
+    {
+      "sample_idx": 71122.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2260.4833984375,
+      "mae": 31.50975799560547,
+      "first_action_mse": 3393.128173828125,
+      "first_10_mse": 4523.8388671875,
+      "train_mean_baseline_mse": 1806.3253173828125,
+      "mse_vs_train_mean_ratio": 1.2514265158574636,
+      "pred_dim_std_mean": 8.106392860412598,
+      "target_dim_std_mean": 25.489715576171875,
+      "dim_0_mse": 496.3849792480469,
+      "dim_1_mse": 6902.28662109375,
+      "dim_2_mse": 4252.20947265625,
+      "dim_3_mse": 815.314208984375,
+      "dim_4_mse": 1096.702880859375,
+      "dim_5_mse": 0.0037050035316497087
+    },
+    {
+      "sample_idx": 71122.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1868.2843017578125,
+      "mae": 29.54847526550293,
+      "first_action_mse": 3261.265869140625,
+      "first_10_mse": 3691.69580078125,
+      "train_mean_baseline_mse": 1806.3253173828125,
+      "mse_vs_train_mean_ratio": 1.034301121607913,
+      "pred_dim_std_mean": 7.322011470794678,
+      "target_dim_std_mean": 25.489715576171875,
+      "dim_0_mse": 1429.7401123046875,
+      "dim_1_mse": 5184.8564453125,
+      "dim_2_mse": 2789.2626953125,
+      "dim_3_mse": 834.7642211914062,
+      "dim_4_mse": 971.0802001953125,
+      "dim_5_mse": 0.0024432712234556675
+    },
+    {
+      "sample_idx": 73967.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3382.081298828125,
+      "mae": 40.68802261352539,
+      "first_action_mse": 532.16162109375,
+      "first_10_mse": 1218.2757568359375,
+      "train_mean_baseline_mse": 1455.120361328125,
+      "mse_vs_train_mean_ratio": 2.324262231985548,
+      "pred_dim_std_mean": 7.70761251449585,
+      "target_dim_std_mean": 26.008773803710938,
+      "dim_0_mse": 1847.0994873046875,
+      "dim_1_mse": 3196.80517578125,
+      "dim_2_mse": 2767.542724609375,
+      "dim_3_mse": 568.5845336914062,
+      "dim_4_mse": 11912.43359375,
+      "dim_5_mse": 0.02083112858235836
+    },
+    {
+      "sample_idx": 73967.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2349.975830078125,
+      "mae": 33.127967834472656,
+      "first_action_mse": 505.7565002441406,
+      "first_10_mse": 738.3054809570312,
+      "train_mean_baseline_mse": 1455.120361328125,
+      "mse_vs_train_mean_ratio": 1.6149700688218278,
+      "pred_dim_std_mean": 7.52016019821167,
+      "target_dim_std_mean": 26.008773803710938,
+      "dim_0_mse": 714.6985473632812,
+      "dim_1_mse": 1981.63330078125,
+      "dim_2_mse": 1711.60205078125,
+      "dim_3_mse": 534.7658081054688,
+      "dim_4_mse": 9157.1396484375,
+      "dim_5_mse": 0.01676245778799057
+    },
+    {
+      "sample_idx": 76812.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2615.30322265625,
+      "mae": 36.560791015625,
+      "first_action_mse": 1144.2923583984375,
+      "first_10_mse": 2011.734130859375,
+      "train_mean_baseline_mse": 339.8664245605469,
+      "mse_vs_train_mean_ratio": 7.695091464353626,
+      "pred_dim_std_mean": 7.62019681930542,
+      "target_dim_std_mean": 4.0668721199035645,
+      "dim_0_mse": 329.3760070800781,
+      "dim_1_mse": 7824.93896484375,
+      "dim_2_mse": 6334.7333984375,
+      "dim_3_mse": 24.362499237060547,
+      "dim_4_mse": 1178.3922119140625,
+      "dim_5_mse": 0.01445804163813591
+    },
+    {
+      "sample_idx": 76812.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1563.53369140625,
+      "mae": 27.14291763305664,
+      "first_action_mse": 873.6287231445312,
+      "first_10_mse": 1243.8660888671875,
+      "train_mean_baseline_mse": 339.8664245605469,
+      "mse_vs_train_mean_ratio": 4.600435872498808,
+      "pred_dim_std_mean": 7.883703231811523,
+      "target_dim_std_mean": 4.0668721199035645,
+      "dim_0_mse": 76.1777572631836,
+      "dim_1_mse": 4660.2578125,
+      "dim_2_mse": 4102.73046875,
+      "dim_3_mse": 28.38680648803711,
+      "dim_4_mse": 513.637939453125,
+      "dim_5_mse": 0.010907999239861965
+    },
+    {
+      "sample_idx": 79657.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 7720.92919921875,
+      "mae": 69.86865234375,
+      "first_action_mse": 2146.732666015625,
+      "first_10_mse": 4350.5009765625,
+      "train_mean_baseline_mse": 3408.66943359375,
+      "mse_vs_train_mean_ratio": 2.265085937382493,
+      "pred_dim_std_mean": 7.409985065460205,
+      "target_dim_std_mean": 9.2564115524292,
+      "dim_0_mse": 6830.6953125,
+      "dim_1_mse": 5897.583984375,
+      "dim_2_mse": 6090.71533203125,
+      "dim_3_mse": 607.4854125976562,
+      "dim_4_mse": 26899.080078125,
+      "dim_5_mse": 0.01673685759305954
+    },
+    {
+      "sample_idx": 79657.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 5806.744140625,
+      "mae": 58.57997131347656,
+      "first_action_mse": 1802.4012451171875,
+      "first_10_mse": 3415.511474609375,
+      "train_mean_baseline_mse": 3408.66943359375,
+      "mse_vs_train_mean_ratio": 1.7035222258272686,
+      "pred_dim_std_mean": 7.087131023406982,
+      "target_dim_std_mean": 9.2564115524292,
+      "dim_0_mse": 4108.591796875,
+      "dim_1_mse": 3163.533935546875,
+      "dim_2_mse": 3636.00634765625,
+      "dim_3_mse": 675.347900390625,
+      "dim_4_mse": 23256.970703125,
+      "dim_5_mse": 0.014004685916006565
+    },
+    {
+      "sample_idx": 82502.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3572.919921875,
+      "mae": 37.06053924560547,
+      "first_action_mse": 2533.757568359375,
+      "first_10_mse": 3237.798583984375,
+      "train_mean_baseline_mse": 4051.765380859375,
+      "mse_vs_train_mean_ratio": 0.8818180684285297,
+      "pred_dim_std_mean": 7.945240497589111,
+      "target_dim_std_mean": 4.164361476898193,
+      "dim_0_mse": 1012.8606567382812,
+      "dim_1_mse": 381.3708190917969,
+      "dim_2_mse": 351.9069519042969,
+      "dim_3_mse": 371.06109619140625,
+      "dim_4_mse": 19320.26171875,
+      "dim_5_mse": 0.06048024445772171
+    },
+    {
+      "sample_idx": 82502.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 3060.80126953125,
+      "mae": 30.313968658447266,
+      "first_action_mse": 2430.223876953125,
+      "first_10_mse": 3036.9384765625,
+      "train_mean_baseline_mse": 4051.765380859375,
+      "mse_vs_train_mean_ratio": 0.7554241131509094,
+      "pred_dim_std_mean": 8.277018547058105,
+      "target_dim_std_mean": 4.164361476898193,
+      "dim_0_mse": 249.6647186279297,
+      "dim_1_mse": 156.29774475097656,
+      "dim_2_mse": 45.639060974121094,
+      "dim_3_mse": 538.170166015625,
+      "dim_4_mse": 17374.978515625,
+      "dim_5_mse": 0.05792795866727829
+    },
+    {
+      "sample_idx": 85347.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1922.4976806640625,
+      "mae": 34.36647033691406,
+      "first_action_mse": 340.9708557128906,
+      "first_10_mse": 869.9705810546875,
+      "train_mean_baseline_mse": 319.4601135253906,
+      "mse_vs_train_mean_ratio": 6.017958421940092,
+      "pred_dim_std_mean": 7.386026382446289,
+      "target_dim_std_mean": 11.29207992553711,
+      "dim_0_mse": 1762.592041015625,
+      "dim_1_mse": 2907.0244140625,
+      "dim_2_mse": 3132.856689453125,
+      "dim_3_mse": 40.74305725097656,
+      "dim_4_mse": 3691.7587890625,
+      "dim_5_mse": 0.01226162351667881
+    },
+    {
+      "sample_idx": 85347.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 952.5872802734375,
+      "mae": 23.999399185180664,
+      "first_action_mse": 240.9573516845703,
+      "first_10_mse": 439.2341613769531,
+      "train_mean_baseline_mse": 319.4601135253906,
+      "mse_vs_train_mean_ratio": 2.981866092017544,
+      "pred_dim_std_mean": 6.924769878387451,
+      "target_dim_std_mean": 11.29207992553711,
+      "dim_0_mse": 580.5579223632812,
+      "dim_1_mse": 1183.49560546875,
+      "dim_2_mse": 1615.8905029296875,
+      "dim_3_mse": 63.110416412353516,
+      "dim_4_mse": 2272.459716796875,
+      "dim_5_mse": 0.00963762030005455
+    },
+    {
+      "sample_idx": 88192.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6398.15869140625,
+      "mae": 57.55574417114258,
+      "first_action_mse": 3334.745849609375,
+      "first_10_mse": 4490.38232421875,
+      "train_mean_baseline_mse": 1745.020751953125,
+      "mse_vs_train_mean_ratio": 3.666522982173749,
+      "pred_dim_std_mean": 8.141156196594238,
+      "target_dim_std_mean": 0.0,
+      "dim_0_mse": 691.6264038085938,
+      "dim_1_mse": 19745.732421875,
+      "dim_2_mse": 15703.1484375,
+      "dim_3_mse": 136.36318969726562,
+      "dim_4_mse": 2112.06640625,
+      "dim_5_mse": 0.01526523195207119
+    },
+    {
+      "sample_idx": 88192.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 4519.06884765625,
+      "mae": 45.83308029174805,
+      "first_action_mse": 3226.723876953125,
+      "first_10_mse": 3657.792724609375,
+      "train_mean_baseline_mse": 1745.020751953125,
+      "mse_vs_train_mean_ratio": 2.5896934707499923,
+      "pred_dim_std_mean": 7.32410192489624,
+      "target_dim_std_mean": 0.0,
+      "dim_0_mse": 101.94429016113281,
+      "dim_1_mse": 14222.9697265625,
+      "dim_2_mse": 11580.6884765625,
+      "dim_3_mse": 116.40347290039062,
+      "dim_4_mse": 1092.3966064453125,
+      "dim_5_mse": 0.011531406082212925
+    }
+  ]
+}

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000000900/predictions.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:827585e59f73476d85300095ca4ea14ae27a60f5b4a8c8d78ec27551630cb78a
+size 129194

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000001000/metrics.json

@@ -0,0 +1,1268 @@
+{
+  "checkpoint": "/home/shadeform/pdt-mimic/mimic-video/model/checkpoints/vam/so101/w2a_so101_h100_chain_bsz2_accum8_full204/checkpoints/model/iter_000001000.pt",
+  "experiment": "w2a_so101_lerobot_v2w_push_that_thing_lr1.000e-04_layer20_bsz1",
+  "job_name": "w2a_so101_h100_chain_bsz2_accum8_full204",
+  "iteration": 1000,
+  "num_samples": 32,
+  "stop_steps": [
+    25,
+    30
+  ],
+  "seeds": [
+    0
+  ],
+  "aggregate": {
+    "dim_0_mse/mean": 458.927732527256,
+    "dim_0_mse/std": 573.2640494731703,
+    "dim_1_mse/mean": 4691.519800364971,
+    "dim_1_mse/std": 4666.657560034696,
+    "dim_2_mse/mean": 5677.725383520126,
+    "dim_2_mse/std": 5153.4508953357845,
+    "dim_3_mse/mean": 569.5415374040604,
+    "dim_3_mse/std": 737.5981524793643,
+    "dim_4_mse/mean": 1778.622164607048,
+    "dim_4_mse/std": 3536.9621806133596,
+    "dim_5_mse/mean": 0.0035530010895854502,
+    "dim_5_mse/std": 0.00532643283346641,
+    "first_10_mse/mean": 853.4113008975983,
+    "first_10_mse/std": 616.1107473417759,
+    "first_action_mse/mean": 852.3418889045715,
+    "first_action_mse/std": 513.1198541015891,
+    "mae/mean": 30.001748487353325,
+    "mae/std": 12.260055685564982,
+    "mse/mean": 2196.0566670894623,
+    "mse/std": 1720.4014891182633,
+    "mse_vs_train_mean_ratio/mean": 2.341390305827905,
+    "mse_vs_train_mean_ratio/std": 1.7574265235246944,
+    "pred_dim_std_mean/mean": 8.325167641043663,
+    "pred_dim_std_mean/std": 1.6667668385829646,
+    "sample_idx/mean": 44095.53125,
+    "sample_idx/std": 26267.314313591778,
+    "seed/mean": 0.0,
+    "seed/std": 0.0,
+    "stop_step/mean": 27.5,
+    "stop_step/std": 2.5,
+    "target_dim_std_mean/mean": 13.532952778041363,
+    "target_dim_std_mean/std": 8.156953974648559,
+    "train_mean_baseline_mse/mean": 1216.5151710510254,
+    "train_mean_baseline_mse/std": 839.1808489232942
+  },
+  "rows": [
+    {
+      "sample_idx": 0.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1649.673828125,
+      "mae": 29.324331283569336,
+      "first_action_mse": 1979.0693359375,
+      "first_10_mse": 1748.0826416015625,
+      "train_mean_baseline_mse": 1425.959228515625,
+      "mse_vs_train_mean_ratio": 1.1568870940596627,
+      "pred_dim_std_mean": 12.82370376586914,
+      "target_dim_std_mean": 24.68731117248535,
+      "dim_0_mse": 1002.90771484375,
+      "dim_1_mse": 5030.51904296875,
+      "dim_2_mse": 2531.66845703125,
+      "dim_3_mse": 586.5409545898438,
+      "dim_4_mse": 746.404541015625,
+      "dim_5_mse": 0.0027372571639716625
+    },
+    {
+      "sample_idx": 0.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1337.9649658203125,
+      "mae": 26.455793380737305,
+      "first_action_mse": 1889.58056640625,
+      "first_10_mse": 1529.5150146484375,
+      "train_mean_baseline_mse": 1425.959228515625,
+      "mse_vs_train_mean_ratio": 0.9382911790634354,
+      "pred_dim_std_mean": 11.087108612060547,
+      "target_dim_std_mean": 24.68731117248535,
+      "dim_0_mse": 1473.2965087890625,
+      "dim_1_mse": 3902.690673828125,
+      "dim_2_mse": 1369.4608154296875,
+      "dim_3_mse": 566.9611206054688,
+      "dim_4_mse": 715.3792724609375,
+      "dim_5_mse": 0.0007510714931413531
+    },
+    {
+      "sample_idx": 2844.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 919.7423095703125,
+      "mae": 20.774263381958008,
+      "first_action_mse": 707.0393676757812,
+      "first_10_mse": 557.0962524414062,
+      "train_mean_baseline_mse": 303.8437805175781,
+      "mse_vs_train_mean_ratio": 3.0270236501257037,
+      "pred_dim_std_mean": 7.337483882904053,
+      "target_dim_std_mean": 7.692286968231201,
+      "dim_0_mse": 168.96444702148438,
+      "dim_1_mse": 1597.1785888671875,
+      "dim_2_mse": 3418.0263671875,
+      "dim_3_mse": 19.575000762939453,
+      "dim_4_mse": 314.7076416015625,
+      "dim_5_mse": 0.0017486572032794356
+    },
+    {
+      "sample_idx": 2844.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 485.22186279296875,
+      "mae": 15.928589820861816,
+      "first_action_mse": 627.1309204101562,
+      "first_10_mse": 374.8785400390625,
+      "train_mean_baseline_mse": 303.8437805175781,
+      "mse_vs_train_mean_ratio": 1.5969451866561983,
+      "pred_dim_std_mean": 6.872797012329102,
+      "target_dim_std_mean": 7.692286968231201,
+      "dim_0_mse": 189.07144165039062,
+      "dim_1_mse": 729.0390014648438,
+      "dim_2_mse": 1468.22412109375,
+      "dim_3_mse": 20.139583587646484,
+      "dim_4_mse": 504.8573303222656,
+      "dim_5_mse": 0.0001293606183025986
+    },
+    {
+      "sample_idx": 5689.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 432.57965087890625,
+      "mae": 16.415184020996094,
+      "first_action_mse": 480.6433410644531,
+      "first_10_mse": 361.9056091308594,
+      "train_mean_baseline_mse": 1491.99658203125,
+      "mse_vs_train_mean_ratio": 0.28993340607387924,
+      "pred_dim_std_mean": 7.526397228240967,
+      "target_dim_std_mean": 3.163846015930176,
+      "dim_0_mse": 966.1071166992188,
+      "dim_1_mse": 334.1168518066406,
+      "dim_2_mse": 677.0999755859375,
+      "dim_3_mse": 93.91944122314453,
+      "dim_4_mse": 524.2332763671875,
+      "dim_5_mse": 0.0013376871356740594
+    },
+    {
+      "sample_idx": 5689.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 210.41578674316406,
+      "mae": 10.414764404296875,
+      "first_action_mse": 459.1640625,
+      "first_10_mse": 281.6194763183594,
+      "train_mean_baseline_mse": 1491.99658203125,
+      "mse_vs_train_mean_ratio": 0.14102967076284956,
+      "pred_dim_std_mean": 6.908353328704834,
+      "target_dim_std_mean": 3.163846015930176,
+      "dim_0_mse": 683.4878540039062,
+      "dim_1_mse": 70.515625,
+      "dim_2_mse": 84.06059265136719,
+      "dim_3_mse": 87.0999984741211,
+      "dim_4_mse": 337.33062744140625,
+      "dim_5_mse": 5.510118353413418e-05
+    },
+    {
+      "sample_idx": 8534.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6811.703125,
+      "mae": 54.22562026977539,
+      "first_action_mse": 1165.885009765625,
+      "first_10_mse": 2101.7236328125,
+      "train_mean_baseline_mse": 1506.2615966796875,
+      "mse_vs_train_mean_ratio": 4.522257714075237,
+      "pred_dim_std_mean": 8.046229362487793,
+      "target_dim_std_mean": 7.548631191253662,
+      "dim_0_mse": 91.54810333251953,
+      "dim_1_mse": 19043.052734375,
+      "dim_2_mse": 20380.166015625,
+      "dim_3_mse": 1167.8826904296875,
+      "dim_4_mse": 187.56849670410156,
+      "dim_5_mse": 0.0005703396163880825
+    },
+    {
+      "sample_idx": 8534.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 5185.4970703125,
+      "mae": 49.92720031738281,
+      "first_action_mse": 859.3255004882812,
+      "first_10_mse": 1533.9290771484375,
+      "train_mean_baseline_mse": 1506.2615966796875,
+      "mse_vs_train_mean_ratio": 3.4426271517132867,
+      "pred_dim_std_mean": 6.888454437255859,
+      "target_dim_std_mean": 7.548631191253662,
+      "dim_0_mse": 232.30068969726562,
+      "dim_1_mse": 15043.1640625,
+      "dim_2_mse": 14407.421875,
+      "dim_3_mse": 1064.954833984375,
+      "dim_4_mse": 365.1415710449219,
+      "dim_5_mse": 0.0037594265304505825
+    },
+    {
+      "sample_idx": 11379.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2389.17626953125,
+      "mae": 32.75593948364258,
+      "first_action_mse": 609.78662109375,
+      "first_10_mse": 440.5888366699219,
+      "train_mean_baseline_mse": 334.44451904296875,
+      "mse_vs_train_mean_ratio": 7.143714827104981,
+      "pred_dim_std_mean": 10.142245292663574,
+      "target_dim_std_mean": 12.383785247802734,
+      "dim_0_mse": 163.9875030517578,
+      "dim_1_mse": 4934.7646484375,
+      "dim_2_mse": 8212.8330078125,
+      "dim_3_mse": 630.0569458007812,
+      "dim_4_mse": 393.4166564941406,
+      "dim_5_mse": 0.0007731119985692203
+    },
+    {
+      "sample_idx": 11379.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1588.7332763671875,
+      "mae": 28.072895050048828,
+      "first_action_mse": 533.0447387695312,
+      "first_10_mse": 318.1802062988281,
+      "train_mean_baseline_mse": 334.44451904296875,
+      "mse_vs_train_mean_ratio": 4.750364218595761,
+      "pred_dim_std_mean": 8.13341236114502,
+      "target_dim_std_mean": 12.383785247802734,
+      "dim_0_mse": 172.0870361328125,
+      "dim_1_mse": 3242.00732421875,
+      "dim_2_mse": 4838.36865234375,
+      "dim_3_mse": 534.048583984375,
+      "dim_4_mse": 745.8850708007812,
+      "dim_5_mse": 0.0036519367713481188
+    },
+    {
+      "sample_idx": 14224.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3939.406005859375,
+      "mae": 40.29977798461914,
+      "first_action_mse": 558.28466796875,
+      "first_10_mse": 526.7971801757812,
+      "train_mean_baseline_mse": 815.0859985351562,
+      "mse_vs_train_mean_ratio": 4.833117012117907,
+      "pred_dim_std_mean": 8.317839622497559,
+      "target_dim_std_mean": 17.710800170898438,
+      "dim_0_mse": 82.00431060791016,
+      "dim_1_mse": 9954.2734375,
+      "dim_2_mse": 12301.505859375,
+      "dim_3_mse": 935.4491577148438,
+      "dim_4_mse": 363.202880859375,
+      "dim_5_mse": 0.0011094835354015231
+    },
+    {
+      "sample_idx": 14224.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2852.220458984375,
+      "mae": 35.58906555175781,
+      "first_action_mse": 381.6734924316406,
+      "first_10_mse": 280.66070556640625,
+      "train_mean_baseline_mse": 815.0859985351562,
+      "mse_vs_train_mean_ratio": 3.499287761181378,
+      "pred_dim_std_mean": 6.907731533050537,
+      "target_dim_std_mean": 17.710800170898438,
+      "dim_0_mse": 149.93287658691406,
+      "dim_1_mse": 7471.5498046875,
+      "dim_2_mse": 7955.14501953125,
+      "dim_3_mse": 820.438720703125,
+      "dim_4_mse": 716.2518310546875,
+      "dim_5_mse": 0.004672410897910595
+    },
+    {
+      "sample_idx": 17069.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 701.5103149414062,
+      "mae": 17.03852653503418,
+      "first_action_mse": 397.7120666503906,
+      "first_10_mse": 224.16815185546875,
+      "train_mean_baseline_mse": 1071.8345947265625,
+      "mse_vs_train_mean_ratio": 0.6544949364322111,
+      "pred_dim_std_mean": 7.473897933959961,
+      "target_dim_std_mean": 17.36097526550293,
+      "dim_0_mse": 222.5060272216797,
+      "dim_1_mse": 1518.3448486328125,
+      "dim_2_mse": 2205.2021484375,
+      "dim_3_mse": 44.90833282470703,
+      "dim_4_mse": 218.09584045410156,
+      "dim_5_mse": 0.00432552769780159
+    },
+    {
+      "sample_idx": 17069.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 467.24127197265625,
+      "mae": 14.451924324035645,
+      "first_action_mse": 550.6288452148438,
+      "first_10_mse": 305.37188720703125,
+      "train_mean_baseline_mse": 1071.8345947265625,
+      "mse_vs_train_mean_ratio": 0.435926657220702,
+      "pred_dim_std_mean": 7.560755252838135,
+      "target_dim_std_mean": 17.36097526550293,
+      "dim_0_mse": 199.9722442626953,
+      "dim_1_mse": 1042.5386962890625,
+      "dim_2_mse": 1177.4779052734375,
+      "dim_3_mse": 53.35902786254883,
+      "dim_4_mse": 330.0973205566406,
+      "dim_5_mse": 0.0021699692588299513
+    },
+    {
+      "sample_idx": 19914.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 851.6004638671875,
+      "mae": 18.11992835998535,
+      "first_action_mse": 804.0204467773438,
+      "first_10_mse": 501.6211242675781,
+      "train_mean_baseline_mse": 1155.8238525390625,
+      "mse_vs_train_mean_ratio": 0.7367908717201409,
+      "pred_dim_std_mean": 7.035090923309326,
+      "target_dim_std_mean": 18.041248321533203,
+      "dim_0_mse": 347.395751953125,
+      "dim_1_mse": 1642.23193359375,
+      "dim_2_mse": 2696.861328125,
+      "dim_3_mse": 45.99166488647461,
+      "dim_4_mse": 377.12060546875,
+      "dim_5_mse": 0.0012768215965479612
+    },
+    {
+      "sample_idx": 19914.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 600.730712890625,
+      "mae": 17.17327117919922,
+      "first_action_mse": 1049.5684814453125,
+      "first_10_mse": 725.4893188476562,
+      "train_mean_baseline_mse": 1155.8238525390625,
+      "mse_vs_train_mean_ratio": 0.5197424430815876,
+      "pred_dim_std_mean": 7.061980724334717,
+      "target_dim_std_mean": 18.041248321533203,
+      "dim_0_mse": 393.9237365722656,
+      "dim_1_mse": 1217.0916748046875,
+      "dim_2_mse": 1499.444580078125,
+      "dim_3_mse": 38.556251525878906,
+      "dim_4_mse": 455.36627197265625,
+      "dim_5_mse": 0.0018722533714026213
+    },
+    {
+      "sample_idx": 22759.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 904.9790649414062,
+      "mae": 20.591659545898438,
+      "first_action_mse": 637.5547485351562,
+      "first_10_mse": 360.6820068359375,
+      "train_mean_baseline_mse": 498.5845031738281,
+      "mse_vs_train_mean_ratio": 1.8150966569971618,
+      "pred_dim_std_mean": 7.04088020324707,
+      "target_dim_std_mean": 11.284893035888672,
+      "dim_0_mse": 197.83294677734375,
+      "dim_1_mse": 1606.94775390625,
+      "dim_2_mse": 3222.041259765625,
+      "dim_3_mse": 149.3826446533203,
+      "dim_4_mse": 253.66944885253906,
+      "dim_5_mse": 0.00028008356457576156
+    },
+    {
+      "sample_idx": 22759.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 589.8822021484375,
+      "mae": 17.707754135131836,
+      "first_action_mse": 791.3279418945312,
+      "first_10_mse": 472.0417175292969,
+      "train_mean_baseline_mse": 498.5845031738281,
+      "mse_vs_train_mean_ratio": 1.1831137919318344,
+      "pred_dim_std_mean": 7.093709468841553,
+      "target_dim_std_mean": 11.284893035888672,
+      "dim_0_mse": 191.47909545898438,
+      "dim_1_mse": 1031.1771240234375,
+      "dim_2_mse": 1830.650390625,
+      "dim_3_mse": 118.2076416015625,
+      "dim_4_mse": 367.7787780761719,
+      "dim_5_mse": 0.0001829359243856743
+    },
+    {
+      "sample_idx": 25604.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 5026.58740234375,
+      "mae": 46.728424072265625,
+      "first_action_mse": 575.4666137695312,
+      "first_10_mse": 589.7020263671875,
+      "train_mean_baseline_mse": 1151.5936279296875,
+      "mse_vs_train_mean_ratio": 4.364896852877218,
+      "pred_dim_std_mean": 8.529558181762695,
+      "target_dim_std_mean": 16.87040901184082,
+      "dim_0_mse": 94.60787963867188,
+      "dim_1_mse": 13166.470703125,
+      "dim_2_mse": 14850.5224609375,
+      "dim_3_mse": 1667.65185546875,
+      "dim_4_mse": 380.27264404296875,
+      "dim_5_mse": 0.00151045061647892
+    },
+    {
+      "sample_idx": 25604.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 3770.4931640625,
+      "mae": 42.59131622314453,
+      "first_action_mse": 388.8243103027344,
+      "first_10_mse": 326.5105895996094,
+      "train_mean_baseline_mse": 1151.5936279296875,
+      "mse_vs_train_mean_ratio": 3.274152507114006,
+      "pred_dim_std_mean": 7.04356050491333,
+      "target_dim_std_mean": 16.87040901184082,
+      "dim_0_mse": 221.85736083984375,
+      "dim_1_mse": 10154.1162109375,
+      "dim_2_mse": 9884.7958984375,
+      "dim_3_mse": 1604.0111083984375,
+      "dim_4_mse": 758.1702880859375,
+      "dim_5_mse": 0.004265849012881517
+    },
+    {
+      "sample_idx": 28449.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2976.350341796875,
+      "mae": 36.307594299316406,
+      "first_action_mse": 485.9637145996094,
+      "first_10_mse": 509.2729797363281,
+      "train_mean_baseline_mse": 539.5875244140625,
+      "mse_vs_train_mean_ratio": 5.515973233496995,
+      "pred_dim_std_mean": 9.029984474182129,
+      "target_dim_std_mean": 17.05885124206543,
+      "dim_0_mse": 103.5205307006836,
+      "dim_1_mse": 7153.68115234375,
+      "dim_2_mse": 9051.8525390625,
+      "dim_3_mse": 923.8543701171875,
+      "dim_4_mse": 625.1919555664062,
+      "dim_5_mse": 0.0009711371385492384
+    },
+    {
+      "sample_idx": 28449.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2078.15283203125,
+      "mae": 31.534873962402344,
+      "first_action_mse": 321.6697692871094,
+      "first_10_mse": 283.24224853515625,
+      "train_mean_baseline_mse": 539.5875244140625,
+      "mse_vs_train_mean_ratio": 3.8513730173578677,
+      "pred_dim_std_mean": 7.4513936042785645,
+      "target_dim_std_mean": 17.05885124206543,
+      "dim_0_mse": 139.4696044921875,
+      "dim_1_mse": 5066.16162109375,
+      "dim_2_mse": 5393.82080078125,
+      "dim_3_mse": 786.266845703125,
+      "dim_4_mse": 1083.1932373046875,
+      "dim_5_mse": 0.004375034011900425
+    },
+    {
+      "sample_idx": 31293.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2291.6826171875,
+      "mae": 33.82421875,
+      "first_action_mse": 438.7932434082031,
+      "first_10_mse": 470.4788818359375,
+      "train_mean_baseline_mse": 370.26336669921875,
+      "mse_vs_train_mean_ratio": 6.189331225546638,
+      "pred_dim_std_mean": 9.407450675964355,
+      "target_dim_std_mean": 4.815817356109619,
+      "dim_0_mse": 52.18771743774414,
+      "dim_1_mse": 3842.47021484375,
+      "dim_2_mse": 7191.15673828125,
+      "dim_3_mse": 120.87361145019531,
+      "dim_4_mse": 2543.40380859375,
+      "dim_5_mse": 0.0023061116226017475
+    },
+    {
+      "sample_idx": 31293.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1703.646484375,
+      "mae": 30.019184112548828,
+      "first_action_mse": 384.533935546875,
+      "first_10_mse": 351.2763977050781,
+      "train_mean_baseline_mse": 370.26336669921875,
+      "mse_vs_train_mean_ratio": 4.601174832829052,
+      "pred_dim_std_mean": 8.615437507629395,
+      "target_dim_std_mean": 4.815817356109619,
+      "dim_0_mse": 81.76002502441406,
+      "dim_1_mse": 2482.10791015625,
+      "dim_2_mse": 4478.94482421875,
+      "dim_3_mse": 111.27777862548828,
+      "dim_4_mse": 3067.78466796875,
+      "dim_5_mse": 0.004161580465734005
+    },
+    {
+      "sample_idx": 34138.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6709.28515625,
+      "mae": 52.60907745361328,
+      "first_action_mse": 556.34375,
+      "first_10_mse": 598.0911254882812,
+      "train_mean_baseline_mse": 1418.0733642578125,
+      "mse_vs_train_mean_ratio": 4.731268018535478,
+      "pred_dim_std_mean": 7.808330059051514,
+      "target_dim_std_mean": 25.433252334594727,
+      "dim_0_mse": 74.28401947021484,
+      "dim_1_mse": 17607.5859375,
+      "dim_2_mse": 20342.193359375,
+      "dim_3_mse": 2018.04931640625,
+      "dim_4_mse": 213.59681701660156,
+      "dim_5_mse": 0.00012749565939884633
+    },
+    {
+      "sample_idx": 34138.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 5316.76611328125,
+      "mae": 47.172908782958984,
+      "first_action_mse": 184.937744140625,
+      "first_10_mse": 212.0358123779297,
+      "train_mean_baseline_mse": 1418.0733642578125,
+      "mse_vs_train_mean_ratio": 3.749288469333831,
+      "pred_dim_std_mean": 7.477427005767822,
+      "target_dim_std_mean": 25.433252334594727,
+      "dim_0_mse": 279.0245666503906,
+      "dim_1_mse": 14122.404296875,
+      "dim_2_mse": 15450.7275390625,
+      "dim_3_mse": 1945.4013671875,
+      "dim_4_mse": 103.03580474853516,
+      "dim_5_mse": 0.0021335177589207888
+    },
+    {
+      "sample_idx": 36983.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1701.683837890625,
+      "mae": 29.17079734802246,
+      "first_action_mse": 576.8046264648438,
+      "first_10_mse": 519.8762817382812,
+      "train_mean_baseline_mse": 267.7073974609375,
+      "mse_vs_train_mean_ratio": 6.356506596493756,
+      "pred_dim_std_mean": 8.516298294067383,
+      "target_dim_std_mean": 10.789891242980957,
+      "dim_0_mse": 228.8212890625,
+      "dim_1_mse": 3278.760498046875,
+      "dim_2_mse": 5608.7724609375,
+      "dim_3_mse": 334.31805419921875,
+      "dim_4_mse": 759.4284057617188,
+      "dim_5_mse": 0.0026140001136809587
+    },
+    {
+      "sample_idx": 36983.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1078.660888671875,
+      "mae": 23.675006866455078,
+      "first_action_mse": 424.174072265625,
+      "first_10_mse": 301.39453125,
+      "train_mean_baseline_mse": 267.7073974609375,
+      "mse_vs_train_mean_ratio": 4.0292532029462045,
+      "pred_dim_std_mean": 7.0438551902771,
+      "target_dim_std_mean": 10.789891242980957,
+      "dim_0_mse": 137.7101593017578,
+      "dim_1_mse": 2007.4412841796875,
+      "dim_2_mse": 2794.57421875,
+      "dim_3_mse": 260.4513854980469,
+      "dim_4_mse": 1271.7864990234375,
+      "dim_5_mse": 0.0017391628352925181
+    },
+    {
+      "sample_idx": 39828.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 636.5552978515625,
+      "mae": 17.76345443725586,
+      "first_action_mse": 1404.5311279296875,
+      "first_10_mse": 995.2425537109375,
+      "train_mean_baseline_mse": 1447.6033935546875,
+      "mse_vs_train_mean_ratio": 0.4397304542706674,
+      "pred_dim_std_mean": 8.488480567932129,
+      "target_dim_std_mean": 14.908316612243652,
+      "dim_0_mse": 896.0125732421875,
+      "dim_1_mse": 916.0639038085938,
+      "dim_2_mse": 1348.8572998046875,
+      "dim_3_mse": 62.84722137451172,
+      "dim_4_mse": 595.550048828125,
+      "dim_5_mse": 0.0008900960092432797
+    },
+    {
+      "sample_idx": 39828.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 536.3502197265625,
+      "mae": 17.1769962310791,
+      "first_action_mse": 1433.1085205078125,
+      "first_10_mse": 1169.0045166015625,
+      "train_mean_baseline_mse": 1447.6033935546875,
+      "mse_vs_train_mean_ratio": 0.37050909255574377,
+      "pred_dim_std_mean": 7.954160690307617,
+      "target_dim_std_mean": 14.908316612243652,
+      "dim_0_mse": 995.0392456054688,
+      "dim_1_mse": 847.4630126953125,
+      "dim_2_mse": 821.8709106445312,
+      "dim_3_mse": 73.65763854980469,
+      "dim_4_mse": 480.0705871582031,
+      "dim_5_mse": 6.0356986068654805e-05
+    },
+    {
+      "sample_idx": 42673.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3447.63671875,
+      "mae": 42.547874450683594,
+      "first_action_mse": 1133.5865478515625,
+      "first_10_mse": 1551.7552490234375,
+      "train_mean_baseline_mse": 1557.8040771484375,
+      "mse_vs_train_mean_ratio": 2.2131388467418214,
+      "pred_dim_std_mean": 7.959208011627197,
+      "target_dim_std_mean": 4.832499027252197,
+      "dim_0_mse": 262.0856628417969,
+      "dim_1_mse": 6121.75341796875,
+      "dim_2_mse": 11566.837890625,
+      "dim_3_mse": 133.19444274902344,
+      "dim_4_mse": 2601.94921875,
+      "dim_5_mse": 0.00019599066581577063
+    },
+    {
+      "sample_idx": 42673.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2794.591064453125,
+      "mae": 38.43045425415039,
+      "first_action_mse": 706.1284790039062,
+      "first_10_mse": 1066.9842529296875,
+      "train_mean_baseline_mse": 1557.8040771484375,
+      "mse_vs_train_mean_ratio": 1.793929740875135,
+      "pred_dim_std_mean": 7.292335510253906,
+      "target_dim_std_mean": 4.832499027252197,
+      "dim_0_mse": 78.33853912353516,
+      "dim_1_mse": 4005.637451171875,
+      "dim_2_mse": 8031.04296875,
+      "dim_3_mse": 94.38888549804688,
+      "dim_4_mse": 4558.13525390625,
+      "dim_5_mse": 0.0015604655491188169
+    },
+    {
+      "sample_idx": 45518.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 325.8027038574219,
+      "mae": 13.792017936706543,
+      "first_action_mse": 516.8858032226562,
+      "first_10_mse": 225.05215454101562,
+      "train_mean_baseline_mse": 696.74365234375,
+      "mse_vs_train_mean_ratio": 0.4676077101836038,
+      "pred_dim_std_mean": 8.856001853942871,
+      "target_dim_std_mean": 3.098416566848755,
+      "dim_0_mse": 547.7094116210938,
+      "dim_1_mse": 193.0156707763672,
+      "dim_2_mse": 809.4246826171875,
+      "dim_3_mse": 118.88055419921875,
+      "dim_4_mse": 285.78424072265625,
+      "dim_5_mse": 0.0015045165782794356
+    },
+    {
+      "sample_idx": 45518.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 145.34994506835938,
+      "mae": 8.865190505981445,
+      "first_action_mse": 490.5796203613281,
+      "first_10_mse": 262.40301513671875,
+      "train_mean_baseline_mse": 696.74365234375,
+      "mse_vs_train_mean_ratio": 0.20861323182410363,
+      "pred_dim_std_mean": 7.910221576690674,
+      "target_dim_std_mean": 3.098416566848755,
+      "dim_0_mse": 367.0357666015625,
+      "dim_1_mse": 85.92308044433594,
+      "dim_2_mse": 126.84832763671875,
+      "dim_3_mse": 102.73124694824219,
+      "dim_4_mse": 189.56106567382812,
+      "dim_5_mse": 0.0001451280404580757
+    },
+    {
+      "sample_idx": 48363.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1696.121337890625,
+      "mae": 27.346153259277344,
+      "first_action_mse": 648.8712768554688,
+      "first_10_mse": 539.3917846679688,
+      "train_mean_baseline_mse": 493.635009765625,
+      "mse_vs_train_mean_ratio": 3.4359826680363157,
+      "pred_dim_std_mean": 7.117146015167236,
+      "target_dim_std_mean": 11.497101783752441,
+      "dim_0_mse": 117.18768310546875,
+      "dim_1_mse": 4581.8486328125,
+      "dim_2_mse": 5035.998046875,
+      "dim_3_mse": 267.5901184082031,
+      "dim_4_mse": 174.09815979003906,
+      "dim_5_mse": 0.00447862409055233
+    },
+    {
+      "sample_idx": 48363.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1177.6407470703125,
+      "mae": 23.395401000976562,
+      "first_action_mse": 747.3009643554688,
+      "first_10_mse": 465.05743408203125,
+      "train_mean_baseline_mse": 493.635009765625,
+      "mse_vs_train_mean_ratio": 2.3856507820007526,
+      "pred_dim_std_mean": 6.883031845092773,
+      "target_dim_std_mean": 11.497101783752441,
+      "dim_0_mse": 105.10868835449219,
+      "dim_1_mse": 3338.3740234375,
+      "dim_2_mse": 3139.291015625,
+      "dim_3_mse": 213.63246154785156,
+      "dim_4_mse": 269.4354248046875,
+      "dim_5_mse": 0.0028937445022165775
+    },
+    {
+      "sample_idx": 51208.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 636.5701293945312,
+      "mae": 18.134008407592773,
+      "first_action_mse": 501.477783203125,
+      "first_10_mse": 224.56492614746094,
+      "train_mean_baseline_mse": 1083.5904541015625,
+      "mse_vs_train_mean_ratio": 0.5874637663934855,
+      "pred_dim_std_mean": 7.032951354980469,
+      "target_dim_std_mean": 12.13278579711914,
+      "dim_0_mse": 271.6318664550781,
+      "dim_1_mse": 874.8794555664062,
+      "dim_2_mse": 2225.530517578125,
+      "dim_3_mse": 250.22222900390625,
+      "dim_4_mse": 197.1483917236328,
+      "dim_5_mse": 0.00843014195561409
+    },
+    {
+      "sample_idx": 51208.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 381.9914855957031,
+      "mae": 14.303567886352539,
+      "first_action_mse": 612.9700317382812,
+      "first_10_mse": 304.8699951171875,
+      "train_mean_baseline_mse": 1083.5904541015625,
+      "mse_vs_train_mean_ratio": 0.3525238563608645,
+      "pred_dim_std_mean": 7.131577968597412,
+      "target_dim_std_mean": 12.13278579711914,
+      "dim_0_mse": 332.91583251953125,
+      "dim_1_mse": 497.01531982421875,
+      "dim_2_mse": 1119.2930908203125,
+      "dim_3_mse": 196.53334045410156,
+      "dim_4_mse": 146.1868896484375,
+      "dim_5_mse": 0.004376390017569065
+    },
+    {
+      "sample_idx": 54053.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 780.2210693359375,
+      "mae": 18.155099868774414,
+      "first_action_mse": 680.7096557617188,
+      "first_10_mse": 362.6435546875,
+      "train_mean_baseline_mse": 786.1039428710938,
+      "mse_vs_train_mean_ratio": 0.9925164177224831,
+      "pred_dim_std_mean": 7.0400238037109375,
+      "target_dim_std_mean": 13.092276573181152,
+      "dim_0_mse": 190.01275634765625,
+      "dim_1_mse": 1606.889404296875,
+      "dim_2_mse": 2629.400634765625,
+      "dim_3_mse": 44.93194580078125,
+      "dim_4_mse": 210.09088134765625,
+      "dim_5_mse": 0.0008821275550872087
+    },
+    {
+      "sample_idx": 54053.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 507.80938720703125,
+      "mae": 16.003131866455078,
+      "first_action_mse": 876.4871215820312,
+      "first_10_mse": 517.836669921875,
+      "train_mean_baseline_mse": 786.1039428710938,
+      "mse_vs_train_mean_ratio": 0.6459824960963241,
+      "pred_dim_std_mean": 7.0423736572265625,
+      "target_dim_std_mean": 13.092276573181152,
+      "dim_0_mse": 216.1735076904297,
+      "dim_1_mse": 1060.0333251953125,
+      "dim_2_mse": 1432.77685546875,
+      "dim_3_mse": 42.57291793823242,
+      "dim_4_mse": 295.2980041503906,
+      "dim_5_mse": 0.0016372681129723787
+    },
+    {
+      "sample_idx": 56898.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2200.02685546875,
+      "mae": 32.0589714050293,
+      "first_action_mse": 1114.5113525390625,
+      "first_10_mse": 1772.0855712890625,
+      "train_mean_baseline_mse": 685.4307250976562,
+      "mse_vs_train_mean_ratio": 3.2096997915512215,
+      "pred_dim_std_mean": 8.439115524291992,
+      "target_dim_std_mean": 5.5619215965271,
+      "dim_0_mse": 163.56692504882812,
+      "dim_1_mse": 3874.095947265625,
+      "dim_2_mse": 8599.9912109375,
+      "dim_3_mse": 79.52222442626953,
+      "dim_4_mse": 482.9764404296875,
+      "dim_5_mse": 0.0070329452864825726
+    },
+    {
+      "sample_idx": 56898.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1467.5921630859375,
+      "mae": 27.762731552124023,
+      "first_action_mse": 710.7711791992188,
+      "first_10_mse": 1216.4415283203125,
+      "train_mean_baseline_mse": 685.4307250976562,
+      "mse_vs_train_mean_ratio": 2.1411239813868037,
+      "pred_dim_std_mean": 7.510134220123291,
+      "target_dim_std_mean": 5.5619215965271,
+      "dim_0_mse": 133.07395935058594,
+      "dim_1_mse": 2082.1962890625,
+      "dim_2_mse": 5103.63720703125,
+      "dim_3_mse": 59.025001525878906,
+      "dim_4_mse": 1427.6202392578125,
+      "dim_5_mse": 0.0016754150856286287
+    },
+    {
+      "sample_idx": 59742.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1561.2322998046875,
+      "mae": 26.127168655395508,
+      "first_action_mse": 1937.7427978515625,
+      "first_10_mse": 1869.917236328125,
+      "train_mean_baseline_mse": 1895.6773681640625,
+      "mse_vs_train_mean_ratio": 0.8235749004677518,
+      "pred_dim_std_mean": 12.375781059265137,
+      "target_dim_std_mean": 28.8035945892334,
+      "dim_0_mse": 1172.7825927734375,
+      "dim_1_mse": 4259.89990234375,
+      "dim_2_mse": 2890.492431640625,
+      "dim_3_mse": 220.77569580078125,
+      "dim_4_mse": 823.431396484375,
+      "dim_5_mse": 0.011631605215370655
+    },
+    {
+      "sample_idx": 59742.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1362.5343017578125,
+      "mae": 25.379161834716797,
+      "first_action_mse": 1866.67724609375,
+      "first_10_mse": 1657.156494140625,
+      "train_mean_baseline_mse": 1895.6773681640625,
+      "mse_vs_train_mean_ratio": 0.718758542271045,
+      "pred_dim_std_mean": 10.66865062713623,
+      "target_dim_std_mean": 28.8035945892334,
+      "dim_0_mse": 1723.5482177734375,
+      "dim_1_mse": 3558.31884765625,
+      "dim_2_mse": 2015.5706787109375,
+      "dim_3_mse": 228.25694274902344,
+      "dim_4_mse": 649.5030517578125,
+      "dim_5_mse": 0.0073789809830486774
+    },
+    {
+      "sample_idx": 62587.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 6120.060546875,
+      "mae": 57.57388687133789,
+      "first_action_mse": 1548.3551025390625,
+      "first_10_mse": 2199.7744140625,
+      "train_mean_baseline_mse": 2114.847412109375,
+      "mse_vs_train_mean_ratio": 2.893854427431611,
+      "pred_dim_std_mean": 8.130131721496582,
+      "target_dim_std_mean": 28.069997787475586,
+      "dim_0_mse": 318.08184814453125,
+      "dim_1_mse": 12267.6123046875,
+      "dim_2_mse": 18513.638671875,
+      "dim_3_mse": 3558.693115234375,
+      "dim_4_mse": 2062.33740234375,
+      "dim_5_mse": 0.0028130426071584225
+    },
+    {
+      "sample_idx": 62587.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 5009.72119140625,
+      "mae": 53.22748947143555,
+      "first_action_mse": 1175.8023681640625,
+      "first_10_mse": 1756.7476806640625,
+      "train_mean_baseline_mse": 2114.847412109375,
+      "mse_vs_train_mean_ratio": 2.368833402694283,
+      "pred_dim_std_mean": 7.1085991859436035,
+      "target_dim_std_mean": 28.069997787475586,
+      "dim_0_mse": 802.1753540039062,
+      "dim_1_mse": 9278.7060546875,
+      "dim_2_mse": 14275.0029296875,
+      "dim_3_mse": 3355.056884765625,
+      "dim_4_mse": 2347.385498046875,
+      "dim_5_mse": 0.00206027552485466
+    },
+    {
+      "sample_idx": 65432.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 3839.248779296875,
+      "mae": 41.86692428588867,
+      "first_action_mse": 1914.2890625,
+      "first_10_mse": 1812.3946533203125,
+      "train_mean_baseline_mse": 1502.2025146484375,
+      "mse_vs_train_mean_ratio": 2.5557464735008644,
+      "pred_dim_std_mean": 12.74864673614502,
+      "target_dim_std_mean": 10.998623847961426,
+      "dim_0_mse": 362.2061767578125,
+      "dim_1_mse": 12483.474609375,
+      "dim_2_mse": 9338.984375,
+      "dim_3_mse": 597.558349609375,
+      "dim_4_mse": 253.26834106445312,
+      "dim_5_mse": 0.0012073093093931675
+    },
+    {
+      "sample_idx": 65432.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2924.612060546875,
+      "mae": 37.36208724975586,
+      "first_action_mse": 1806.8536376953125,
+      "first_10_mse": 1576.85546875,
+      "train_mean_baseline_mse": 1502.2025146484375,
+      "mse_vs_train_mean_ratio": 1.9468826819473977,
+      "pred_dim_std_mean": 11.035686492919922,
+      "target_dim_std_mean": 10.998623847961426,
+      "dim_0_mse": 486.0043029785156,
+      "dim_1_mse": 9910.4638671875,
+      "dim_2_mse": 6292.60009765625,
+      "dim_3_mse": 545.9673461914062,
+      "dim_4_mse": 312.6349182128906,
+      "dim_5_mse": 0.00102979876101017
+    },
+    {
+      "sample_idx": 68277.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2766.674560546875,
+      "mae": 34.93909454345703,
+      "first_action_mse": 557.367919921875,
+      "first_10_mse": 637.3397216796875,
+      "train_mean_baseline_mse": 1187.5592041015625,
+      "mse_vs_train_mean_ratio": 2.329715058408375,
+      "pred_dim_std_mean": 8.085558891296387,
+      "target_dim_std_mean": 24.93874168395996,
+      "dim_0_mse": 281.25958251953125,
+      "dim_1_mse": 5506.39794921875,
+      "dim_2_mse": 8837.955078125,
+      "dim_3_mse": 1441.4268798828125,
+      "dim_4_mse": 533.0029296875,
+      "dim_5_mse": 0.004239230882376432
+    },
+    {
+      "sample_idx": 68277.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1991.9969482421875,
+      "mae": 30.02653694152832,
+      "first_action_mse": 266.5758361816406,
+      "first_10_mse": 288.0139465332031,
+      "train_mean_baseline_mse": 1187.5592041015625,
+      "mse_vs_train_mean_ratio": 1.677387486335231,
+      "pred_dim_std_mean": 7.629518508911133,
+      "target_dim_std_mean": 24.93874168395996,
+      "dim_0_mse": 304.68011474609375,
+      "dim_1_mse": 3662.838134765625,
+      "dim_2_mse": 5434.29931640625,
+      "dim_3_mse": 1247.0880126953125,
+      "dim_4_mse": 1303.074951171875,
+      "dim_5_mse": 0.002122327918186784
+    },
+    {
+      "sample_idx": 71122.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1537.2493896484375,
+      "mae": 30.092313766479492,
+      "first_action_mse": 1860.9783935546875,
+      "first_10_mse": 1843.0455322265625,
+      "train_mean_baseline_mse": 1806.3253173828125,
+      "mse_vs_train_mean_ratio": 0.8510368397403412,
+      "pred_dim_std_mean": 12.548169136047363,
+      "target_dim_std_mean": 25.489715576171875,
+      "dim_0_mse": 1846.227294921875,
+      "dim_1_mse": 3069.17578125,
+      "dim_2_mse": 1932.8526611328125,
+      "dim_3_mse": 1450.9364013671875,
+      "dim_4_mse": 924.3024291992188,
+      "dim_5_mse": 0.0020923190750181675
+    },
+    {
+      "sample_idx": 71122.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1347.0584716796875,
+      "mae": 27.240825653076172,
+      "first_action_mse": 1805.7142333984375,
+      "first_10_mse": 1628.12158203125,
+      "train_mean_baseline_mse": 1806.3253173828125,
+      "mse_vs_train_mean_ratio": 0.7457452202637799,
+      "pred_dim_std_mean": 10.842827796936035,
+      "target_dim_std_mean": 25.489715576171875,
+      "dim_0_mse": 2391.934326171875,
+      "dim_1_mse": 2453.29345703125,
+      "dim_2_mse": 1105.9688720703125,
+      "dim_3_mse": 1400.734375,
+      "dim_4_mse": 730.4197998046875,
+      "dim_5_mse": 0.0007468329276889563
+    },
+    {
+      "sample_idx": 73967.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 2140.21044921875,
+      "mae": 32.5174674987793,
+      "first_action_mse": 688.6858520507812,
+      "first_10_mse": 530.9181518554688,
+      "train_mean_baseline_mse": 1455.120361328125,
+      "mse_vs_train_mean_ratio": 1.4708133471964655,
+      "pred_dim_std_mean": 8.483447074890137,
+      "target_dim_std_mean": 26.008773803710938,
+      "dim_0_mse": 684.64892578125,
+      "dim_1_mse": 1581.1629638671875,
+      "dim_2_mse": 2027.3817138671875,
+      "dim_3_mse": 1034.788330078125,
+      "dim_4_mse": 7513.27490234375,
+      "dim_5_mse": 0.005234782118350267
+    },
+    {
+      "sample_idx": 73967.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 2029.025634765625,
+      "mae": 30.195598602294922,
+      "first_action_mse": 630.6936645507812,
+      "first_10_mse": 404.59747314453125,
+      "train_mean_baseline_mse": 1455.120361328125,
+      "mse_vs_train_mean_ratio": 1.394403987937934,
+      "pred_dim_std_mean": 7.616910457611084,
+      "target_dim_std_mean": 26.008773803710938,
+      "dim_0_mse": 477.16015625,
+      "dim_1_mse": 1081.707763671875,
+      "dim_2_mse": 1144.0625,
+      "dim_3_mse": 955.2425537109375,
+      "dim_4_mse": 8515.970703125,
+      "dim_5_mse": 0.009893289767205715
+    },
+    {
+      "sample_idx": 76812.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1849.2000732421875,
+      "mae": 28.467744827270508,
+      "first_action_mse": 759.69873046875,
+      "first_10_mse": 1029.820556640625,
+      "train_mean_baseline_mse": 339.8664245605469,
+      "mse_vs_train_mean_ratio": 5.440961329537729,
+      "pred_dim_std_mean": 6.959598541259766,
+      "target_dim_std_mean": 4.0668721199035645,
+      "dim_0_mse": 63.68870544433594,
+      "dim_1_mse": 4816.33837890625,
+      "dim_2_mse": 5959.2373046875,
+      "dim_3_mse": 56.42986297607422,
+      "dim_4_mse": 199.5022735595703,
+      "dim_5_mse": 0.0031653509940952063
+    },
+    {
+      "sample_idx": 76812.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1255.6241455078125,
+      "mae": 24.220102310180664,
+      "first_action_mse": 498.2957763671875,
+      "first_10_mse": 670.2645263671875,
+      "train_mean_baseline_mse": 339.8664245605469,
+      "mse_vs_train_mean_ratio": 3.69446363267968,
+      "pred_dim_std_mean": 6.905824184417725,
+      "target_dim_std_mean": 4.0668721199035645,
+      "dim_0_mse": 88.67900085449219,
+      "dim_1_mse": 3367.25,
+      "dim_2_mse": 3745.265380859375,
+      "dim_3_mse": 36.42916488647461,
+      "dim_4_mse": 296.114990234375,
+      "dim_5_mse": 0.0065711126662790775
+    },
+    {
+      "sample_idx": 79657.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 5298.72216796875,
+      "mae": 56.2122688293457,
+      "first_action_mse": 773.9064331054688,
+      "first_10_mse": 1964.5987548828125,
+      "train_mean_baseline_mse": 3408.66943359375,
+      "mse_vs_train_mean_ratio": 1.5544840211690234,
+      "pred_dim_std_mean": 9.068045616149902,
+      "target_dim_std_mean": 9.2564115524292,
+      "dim_0_mse": 2718.4169921875,
+      "dim_1_mse": 4360.404296875,
+      "dim_2_mse": 7876.18212890625,
+      "dim_3_mse": 131.35833740234375,
+      "dim_4_mse": 16705.97265625,
+      "dim_5_mse": 0.0018534342525526881
+    },
+    {
+      "sample_idx": 79657.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 4717.57861328125,
+      "mae": 50.53145980834961,
+      "first_action_mse": 721.4744262695312,
+      "first_10_mse": 1734.751953125,
+      "train_mean_baseline_mse": 3408.66943359375,
+      "mse_vs_train_mean_ratio": 1.3839941669871816,
+      "pred_dim_std_mean": 7.646022796630859,
+      "target_dim_std_mean": 9.2564115524292,
+      "dim_0_mse": 1847.2957763671875,
+      "dim_1_mse": 2628.024169921875,
+      "dim_2_mse": 4433.95263671875,
+      "dim_3_mse": 149.85000610351562,
+      "dim_4_mse": 19246.34375,
+      "dim_5_mse": 0.0057388306595385075
+    },
+    {
+      "sample_idx": 82502.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1498.3662109375,
+      "mae": 25.266407012939453,
+      "first_action_mse": 475.4281005859375,
+      "first_10_mse": 761.9863891601562,
+      "train_mean_baseline_mse": 4051.765380859375,
+      "mse_vs_train_mean_ratio": 0.3698057686202201,
+      "pred_dim_std_mean": 7.575662136077881,
+      "target_dim_std_mean": 4.164361476898193,
+      "dim_0_mse": 109.35230255126953,
+      "dim_1_mse": 344.2492980957031,
+      "dim_2_mse": 1329.949951171875,
+      "dim_3_mse": 68.42500305175781,
+      "dim_4_mse": 7138.19921875,
+      "dim_5_mse": 0.02159762941300869
+    },
+    {
+      "sample_idx": 82502.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 1768.421875,
+      "mae": 23.22665023803711,
+      "first_action_mse": 593.9186401367188,
+      "first_10_mse": 949.8501586914062,
+      "train_mean_baseline_mse": 4051.765380859375,
+      "mse_vs_train_mean_ratio": 0.43645712640570505,
+      "pred_dim_std_mean": 6.957513332366943,
+      "target_dim_std_mean": 4.164361476898193,
+      "dim_0_mse": 64.18871307373047,
+      "dim_1_mse": 55.82847213745117,
+      "dim_2_mse": 464.16180419921875,
+      "dim_3_mse": 77.2138900756836,
+      "dim_4_mse": 9949.1025390625,
+      "dim_5_mse": 0.03643798828125
+    },
+    {
+      "sample_idx": 85347.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 1119.26708984375,
+      "mae": 24.227705001831055,
+      "first_action_mse": 358.5816955566406,
+      "first_10_mse": 278.45843505859375,
+      "train_mean_baseline_mse": 319.4601135253906,
+      "mse_vs_train_mean_ratio": 3.5036207728474085,
+      "pred_dim_std_mean": 8.248956680297852,
+      "target_dim_std_mean": 11.29207992553711,
+      "dim_0_mse": 235.62191772460938,
+      "dim_1_mse": 1614.7979736328125,
+      "dim_2_mse": 3315.44970703125,
+      "dim_3_mse": 280.1076354980469,
+      "dim_4_mse": 1269.6248779296875,
+      "dim_5_mse": 0.0013326009502634406
+    },
+    {
+      "sample_idx": 85347.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 733.7714233398438,
+      "mae": 19.846405029296875,
+      "first_action_mse": 314.9831237792969,
+      "first_10_mse": 185.83827209472656,
+      "train_mean_baseline_mse": 319.4601135253906,
+      "mse_vs_train_mean_ratio": 2.2969109202470865,
+      "pred_dim_std_mean": 6.867852687835693,
+      "target_dim_std_mean": 11.29207992553711,
+      "dim_0_mse": 143.34445190429688,
+      "dim_1_mse": 790.8534545898438,
+      "dim_2_mse": 1475.164794921875,
+      "dim_3_mse": 247.63055419921875,
+      "dim_4_mse": 1745.631591796875,
+      "dim_5_mse": 0.003836229210719466
+    },
+    {
+      "sample_idx": 88192.0,
+      "stop_step": 25.0,
+      "seed": 0.0,
+      "mse": 4732.5068359375,
+      "mae": 45.62256622314453,
+      "first_action_mse": 1824.4547119140625,
+      "first_10_mse": 1786.7923583984375,
+      "train_mean_baseline_mse": 1745.020751953125,
+      "mse_vs_train_mean_ratio": 2.7120060495788447,
+      "pred_dim_std_mean": 12.575286865234375,
+      "target_dim_std_mean": 0.0,
+      "dim_0_mse": 111.61729431152344,
+      "dim_1_mse": 13740.1162109375,
+      "dim_2_mse": 13798.70703125,
+      "dim_3_mse": 467.7861022949219,
+      "dim_4_mse": 276.81365966796875,
+      "dim_5_mse": 0.0012925890041515231
+    },
+    {
+      "sample_idx": 88192.0,
+      "stop_step": 30.0,
+      "seed": 0.0,
+      "mse": 3638.697021484375,
+      "mae": 41.30709457397461,
+      "first_action_mse": 1772.53173828125,
+      "first_10_mse": 1571.5140380859375,
+      "train_mean_baseline_mse": 1745.020751953125,
+      "mse_vs_train_mean_ratio": 2.0851883952736614,
+      "pred_dim_std_mean": 10.893909454345703,
+      "target_dim_std_mean": 0.0,
+      "dim_0_mse": 120.51985931396484,
+      "dim_1_mse": 11048.7607421875,
+      "dim_2_mse": 9853.724609375,
+      "dim_3_mse": 409.54443359375,
+      "dim_4_mse": 399.6302185058594,
+      "dim_5_mse": 0.0037455239798873663
+    }
+  ]
+}

action_decoder/chain/w2a_so101_h100_chain_bsz2_accum8_full204/eval/iter_000001000/predictions.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ebaa916d5463c1c6a9350d825dcd3a63113e3431ca532edab43e89223c506f5
+size 127760