Upload generalization/full_analysis_20260324_1808/minihack_full_results.json with huggingface_hub

generalization/full_analysis_20260324_1808/minihack_full_results.json

@@ -0,0 +1,2341 @@
+{
+  "run_id": "20260324_1808",
+  "max_iter": 500,
+  "eval_eps": 10,
+  "id_envs": [
+    "MiniHack-Room-Random-5x5-v0",
+    "MiniHack-Room-Random-15x15-v0",
+    "MiniHack-Corridor-R2-v0",
+    "MiniHack-MazeWalk-9x9-v0"
+  ],
+  "ood_envs": [
+    "MiniHack-Room-Dark-15x15-v0",
+    "MiniHack-Corridor-R5-v0",
+    "MiniHack-MazeWalk-45x19-v0"
+  ],
+  "baseline_id": 0.525,
+  "baseline_ood": 0.09999999999999999,
+  "baseline_rl_id": 0.07500000000000001,
+  "baseline_rl_ood": 0.03333333333333333,
+  "kl_penalty": {
+    "id": 0.19999999999999998,
+    "ood": 0.06666666666666667
+  },
+  "frozen_backbone": {
+    "id": 0.07500000000000001,
+    "ood": 0.0
+  },
+  "bc_on_wins": {
+    "id": 0.0,
+    "ood": 0.0
+  },
+  "low_t_only": {
+    "id": 0.1,
+    "ood": 0.03333333333333333
+  },
+  "synthetic": {
+    "id": 0.05
+  },
+  "histories": {
+    "baseline_rl": {
+      "iter": [
+        10,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        100,
+        110,
+        120,
+        130,
+        140,
+        150,
+        160,
+        170,
+        180,
+        190,
+        200,
+        210,
+        220,
+        230,
+        240,
+        250,
+        260,
+        270,
+        280,
+        290,
+        300,
+        310,
+        320,
+        330,
+        340,
+        350,
+        360,
+        370,
+        380,
+        390,
+        400,
+        410,
+        420,
+        430,
+        440,
+        450,
+        460,
+        470,
+        480,
+        490,
+        500
+      ],
+      "loss": [
+        0.6148233324289322,
+        0.3809540167450905,
+        0.2913197621703148,
+        0.26417313516139984,
+        0.11857011616230011,
+        0.0840552069246769,
+        0.08531304597854614,
+        0.1835819397121668,
+        0.20300053358078002,
+        0.12927267588675023,
+        0.11205883212387562,
+        0.17100902125239373,
+        0.20254286825656892,
+        0.107823496311903,
+        0.07153078094124794,
+        0.1800502438098192,
+        0.1361511755734682,
+        0.09857502207159996,
+        0.10814670212566853,
+        0.13385814875364305,
+        0.14108883105218412,
+        0.12549171056598424,
+        0.1697872318327427,
+        0.07361965905874968,
+        0.11370712220668792,
+        0.18413102626800537,
+        0.1451867740601301,
+        0.2053861353546381,
+        0.12562402486801147,
+        0.09759063683450223,
+        0.25905307196080685,
+        0.09509452991187572,
+        0.10611449293792248,
+        0.12874542437493802,
+        0.11595193743705749,
+        0.28429544251412153,
+        0.06217188648879528,
+        0.11267267782241105,
+        0.1121125804260373,
+        0.3091689977794886,
+        0.19386135265231133,
+        0.18517458215355873,
+        0.059556832257658246,
+        0.17544590644538402,
+        0.15815371368080378,
+        0.07657972946763039,
+        0.08935816995799542,
+        0.3414760454557836,
+        0.060135122109204533,
+        0.09533649031072855
+      ],
+      "mean_return": [
+        9.893000000000004,
+        1.388000000000008,
+        -3.989999999999989,
+        1.1050000000000064,
+        4.343000000000001,
+        0.9850000000000115,
+        -0.4339999999999934,
+        3.557000000000005,
+        0.7410000000000121,
+        -1.7819999999999883,
+        -2.323999999999984,
+        1.8760000000000054,
+        -3.9299999999999864,
+        -2.0499999999999856,
+        -4.01099999999999,
+        -2.780999999999999,
+        -3.144999999999993,
+        -4.459999999999985,
+        -1.572999999999987,
+        -4.6479999999999855,
+        -2.8879999999999897,
+        1.8810000000000096,
+        -2.045999999999987,
+        -2.979999999999994,
+        2.3930000000000016,
+        1.2220000000000069,
+        -4.036999999999989,
+        -3.680999999999988,
+        -0.7959999999999952,
+        -3.362999999999991,
+        0.8120000000000054,
+        -4.1609999999999925,
+        -2.8989999999999982,
+        -0.6399999999999968,
+        -4.019999999999989,
+        0.17200000000000076,
+        -3.0829999999999975,
+        -0.9379999999999971,
+        0.14200000000000973,
+        -1.6859999999999886,
+        -3.205999999999995,
+        -3.496999999999994,
+        0.3280000000000037,
+        0.3240000000000072,
+        3.0590000000000095,
+        -3.0499999999999914,
+        -0.6909999999999927,
+        -3.3699999999999988,
+        -0.16999999999999393,
+        -3.383999999999996
+      ],
+      "win_rate_episode": [
+        0.5,
+        0.2,
+        0.0,
+        0.2,
+        0.3,
+        0.2,
+        0.1,
+        0.3,
+        0.2,
+        0.1,
+        0.1,
+        0.2,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.2,
+        0.0,
+        0.0,
+        0.2,
+        0.2,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.2,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.1,
+        0.0,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.0,
+        0.1,
+        0.1,
+        0.3,
+        0.0,
+        0.1,
+        0.0,
+        0.1,
+        0.0
+      ],
+      "id_winrate": [
+        0.07500000000000001,
+        0.025,
+        0.15000000000000002,
+        0.025,
+        0.05
+      ],
+      "id_winrate_iter": [
+        100,
+        200,
+        300,
+        400,
+        500
+      ],
+      "ood_winrate": [
+        0.0
+      ],
+      "ood_winrate_iter": [
+        500
+      ],
+      "grad_align_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "grad_align": [
+        -0.058208562433719635,
+        -0.0651715025305748,
+        0.02432907372713089,
+        -0.11967084556818008,
+        -0.06396359205245972,
+        0.05138474702835083,
+        0.06564196199178696,
+        -0.056323569267988205,
+        -0.12439217418432236,
+        0.06368804723024368,
+        0.03813093900680542,
+        -0.0025425786152482033,
+        0.07177173346281052,
+        0.07621006667613983,
+        0.08873365074396133,
+        0.06467217952013016,
+        0.10458952188491821,
+        -0.0745132565498352,
+        -0.043301280587911606,
+        -0.054070524871349335
+      ],
+      "rl_grad_norm": [
+        5.1438727378845215,
+        1.8353410959243774,
+        2.5503463745117188,
+        3.223416805267334,
+        4.39517068862915,
+        1.2743784189224243,
+        4.085864543914795,
+        5.110538959503174,
+        0.8212806582450867,
+        0.5378777980804443,
+        4.511179447174072,
+        3.9516587257385254,
+        0.2867647707462311,
+        0.440229207277298,
+        4.9942240715026855,
+        0.40748071670532227,
+        0.30211448669433594,
+        0.29640454053878784,
+        4.158074378967285,
+        0.4770386517047882
+      ],
+      "bc_grad_norm": [
+        42.94075393676758,
+        36.67488479614258,
+        23.443866729736328,
+        31.622037887573242,
+        31.358760833740234,
+        29.620054244995117,
+        33.03789520263672,
+        30.791826248168945,
+        30.711111068725586,
+        31.014888763427734,
+        28.969528198242188,
+        36.763179779052734,
+        40.4757080078125,
+        27.80027198791504,
+        35.19121551513672,
+        23.520423889160156,
+        31.927614212036133,
+        28.92644500732422,
+        25.52045249938965,
+        29.165616989135742
+      ],
+      "repr_drift_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "repr_drift": [
+        3.9174275398254395,
+        3.921295404434204,
+        3.289928913116455,
+        3.432715892791748,
+        4.955285549163818,
+        6.035404682159424,
+        3.7579100131988525,
+        6.952962875366211,
+        4.759391784667969,
+        5.946002960205078,
+        5.042165279388428,
+        5.096562385559082,
+        7.713111877441406,
+        5.057335376739502,
+        5.915785312652588,
+        6.065906524658203,
+        6.096728324890137,
+        5.633598804473877,
+        7.825881004333496,
+        5.941793441772461
+      ],
+      "t_analysis_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "norm_low_t": [
+        0.6042988896369934,
+        1.0757032632827759,
+        0.7331315279006958,
+        0.24747660756111145,
+        2.080561637878418,
+        0.9171179533004761,
+        1.712111234664917,
+        0.29089364409446716,
+        1.380139946937561,
+        0.2912912964820862,
+        9.737605094909668,
+        1.1511173248291016,
+        0.18356630206108093,
+        0.20324313640594482,
+        0.17540384829044342,
+        0.42038843035697937,
+        0.17512419819831848,
+        0.2651197910308838,
+        1.116037368774414,
+        0.371931254863739
+      ],
+      "norm_high_t": [
+        4.477694511413574,
+        1.4458259344100952,
+        1.6219706535339355,
+        2.7416775226593018,
+        8.002521514892578,
+        0.4847201108932495,
+        2.993985891342163,
+        5.996467590332031,
+        1.0836647748947144,
+        2.2951300144195557,
+        3.4846208095550537,
+        2.6846556663513184,
+        3.8576018810272217,
+        3.7125260829925537,
+        8.503217697143555,
+        0.315531849861145,
+        0.9193480014801025,
+        2.7490694522857666,
+        3.2481539249420166,
+        0.9018426537513733
+      ],
+      "lowhigh_cos": [
+        -0.03117767721414566,
+        0.3306260108947754,
+        -0.01929948851466179,
+        -0.03247307986021042,
+        0.31372562050819397,
+        0.538202702999115,
+        0.4733111262321472,
+        0.05152900516986847,
+        0.03016793541610241,
+        0.27196091413497925,
+        -0.13637898862361908,
+        0.352779746055603,
+        0.3681124448776245,
+        -0.21593783795833588,
+        0.14504966139793396,
+        0.08981875330209732,
+        -0.0734558179974556,
+        -0.16903267800807953,
+        -0.04215971753001213,
+        0.3218880891799927
+      ]
+    },
+    "kl_penalty": {
+      "iter": [
+        10,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        100,
+        110,
+        120,
+        130,
+        140,
+        150,
+        160,
+        170,
+        180,
+        190,
+        200,
+        210,
+        220,
+        230,
+        240,
+        250,
+        260,
+        270,
+        280,
+        290,
+        300,
+        310,
+        320,
+        330,
+        340,
+        350,
+        360,
+        370,
+        380,
+        390,
+        400,
+        410,
+        420,
+        430,
+        440,
+        450,
+        460,
+        470,
+        480,
+        490,
+        500
+      ],
+      "loss": [
+        0.47720463275909425,
+        0.3722280740737915,
+        0.5989133432507515,
+        0.29219049513339995,
+        0.2222285270690918,
+        0.208314248919487,
+        0.18497198522090913,
+        0.246753990650177,
+        0.167646624147892,
+        0.3369371652603149,
+        0.24976162165403365,
+        0.2363536536693573,
+        0.2619509071111679,
+        0.2550868228077888,
+        0.1868986152112484,
+        0.2387184202671051,
+        0.19962314739823342,
+        0.202347831428051,
+        0.20775072425603866,
+        0.17277054935693742,
+        0.18261205181479453,
+        0.23156674206256866,
+        0.1980122596025467,
+        0.31983775198459624,
+        0.3451811149716377,
+        0.3667415648698807,
+        0.23837146311998367,
+        0.25584650188684466,
+        0.27007018625736234,
+        0.36001030057668687,
+        0.24520354121923446,
+        0.24931957870721816,
+        0.3233412757515907,
+        0.3413666620850563,
+        0.3326790541410446,
+        0.2901922330260277,
+        0.3419711276888847,
+        0.2629153922200203,
+        0.2039576329290867,
+        0.24899442493915558,
+        0.2017705723643303,
+        0.2602558508515358,
+        0.27445500791072847,
+        0.3583936542272568,
+        0.3106272265315056,
+        0.2072318471968174,
+        0.2656280748546124,
+        0.18286765292286872,
+        0.2839070238173008,
+        0.19765938594937324
+      ],
+      "mean_return": [
+        6.715000000000005,
+        7.0890000000000075,
+        6.662000000000008,
+        6.703000000000005,
+        9.288,
+        4.207000000000005,
+        7.5290000000000035,
+        -0.5189999999999957,
+        -0.7019999999999911,
+        1.269000000000014,
+        6.418000000000011,
+        9.353000000000005,
+        14.163000000000007,
+        7.142000000000001,
+        4.8839999999999995,
+        5.996000000000007,
+        -1.004999999999994,
+        9.264000000000005,
+        9.412000000000004,
+        8.656000000000006,
+        1.6050000000000124,
+        -1.3469999999999982,
+        -0.1899999999999937,
+        6.854000000000002,
+        6.399000000000006,
+        1.1550000000000071,
+        2.0970000000000093,
+        1.3930000000000053,
+        5.5690000000000115,
+        9.957999999999998,
+        3.869000000000013,
+        1.5870000000000029,
+        5.257000000000018,
+        2.4680000000000035,
+        2.103000000000004,
+        -0.332999999999992,
+        3.5430000000000126,
+        -1.2479999999999893,
+        4.84700000000001,
+        9.819000000000003,
+        6.772999999999999,
+        7.122,
+        7.764000000000005,
+        -2.622000000000001,
+        9.799,
+        3.420000000000015,
+        2.4489999999999985,
+        1.5940000000000056,
+        11.725000000000003,
+        6.8199999999999985
+      ],
+      "win_rate_episode": [
+        0.4,
+        0.4,
+        0.4,
+        0.4,
+        0.5,
+        0.3,
+        0.4,
+        0.1,
+        0.1,
+        0.2,
+        0.4,
+        0.5,
+        0.7,
+        0.4,
+        0.3,
+        0.4,
+        0.1,
+        0.5,
+        0.5,
+        0.5,
+        0.2,
+        0.1,
+        0.1,
+        0.4,
+        0.4,
+        0.2,
+        0.2,
+        0.2,
+        0.3,
+        0.5,
+        0.3,
+        0.2,
+        0.4,
+        0.2,
+        0.2,
+        0.1,
+        0.3,
+        0.1,
+        0.3,
+        0.5,
+        0.4,
+        0.4,
+        0.4,
+        0.0,
+        0.5,
+        0.3,
+        0.2,
+        0.2,
+        0.6,
+        0.4
+      ],
+      "id_winrate": [
+        0.275,
+        0.35,
+        0.32499999999999996,
+        0.47500000000000003,
+        0.275
+      ],
+      "id_winrate_iter": [
+        100,
+        200,
+        300,
+        400,
+        500
+      ],
+      "ood_winrate": [
+        0.03333333333333333
+      ],
+      "ood_winrate_iter": [
+        500
+      ],
+      "grad_align_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "grad_align": [
+        0.12577727437019348,
+        0.475358784198761,
+        0.1447940617799759,
+        0.2517915964126587,
+        0.24051342904567719,
+        0.12343470752239227,
+        0.26251906156539917,
+        0.4390185475349426,
+        0.3158450424671173,
+        0.1859588623046875,
+        0.1288192719221115,
+        0.16457483172416687,
+        0.06687517464160919,
+        0.28203508257865906,
+        -0.0643845871090889,
+        0.43475744128227234,
+        0.0821591168642044,
+        0.15027153491973877,
+        0.2616863250732422,
+        0.06052985042333603
+      ],
+      "rl_grad_norm": [
+        1.6484646797180176,
+        0.7641434669494629,
+        0.3919910192489624,
+        0.5171300172805786,
+        0.9273585081100464,
+        5.108277797698975,
+        0.3941856324672699,
+        0.6915857195854187,
+        1.3585087060928345,
+        5.926978588104248,
+        1.4570164680480957,
+        5.051280975341797,
+        5.409900665283203,
+        4.84480619430542,
+        4.572493076324463,
+        0.41527414321899414,
+        1.148199200630188,
+        3.8179819583892822,
+        1.388190507888794,
+        1.8957138061523438
+      ],
+      "bc_grad_norm": [
+        21.5945987701416,
+        32.32539749145508,
+        16.469274520874023,
+        20.107419967651367,
+        37.188690185546875,
+        24.627458572387695,
+        22.156517028808594,
+        27.076255798339844,
+        19.586650848388672,
+        36.89309310913086,
+        30.176376342773438,
+        32.40763854980469,
+        27.814945220947266,
+        23.15130615234375,
+        19.543317794799805,
+        15.00487995147705,
+        24.16813850402832,
+        21.311819076538086,
+        27.31245994567871,
+        24.81256866455078
+      ],
+      "repr_drift_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "repr_drift": [
+        0.7076154947280884,
+        0.8270145058631897,
+        0.6357962489128113,
+        0.9219959378242493,
+        0.9769991636276245,
+        0.7882189750671387,
+        0.4730985462665558,
+        0.7068943381309509,
+        0.7026294469833374,
+        0.8599920868873596,
+        0.68658846616745,
+        1.1955540180206299,
+        0.8948519825935364,
+        0.7279329299926758,
+        0.5875495076179504,
+        0.8631494641304016,
+        0.6803992986679077,
+        0.5603808164596558,
+        1.1027553081512451,
+        0.5930135250091553
+      ],
+      "t_analysis_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "norm_low_t": [
+        0.7860373258590698,
+        0.3810654878616333,
+        0.404765784740448,
+        0.5835322141647339,
+        0.3631921410560608,
+        0.5430107712745667,
+        0.4888167381286621,
+        0.5309621095657349,
+        1.167321801185608,
+        0.5996302962303162,
+        0.8869525790214539,
+        2.478888750076294,
+        0.23427508771419525,
+        0.6303129196166992,
+        0.6949453353881836,
+        0.4048846364021301,
+        1.1991347074508667,
+        1.8658772706985474,
+        0.2984611392021179,
+        3.9312291145324707
+      ],
+      "norm_high_t": [
+        2.6291065216064453,
+        0.7011670470237732,
+        0.494642436504364,
+        0.7701104283332825,
+        2.5342204570770264,
+        5.168290615081787,
+        0.9434123635292053,
+        3.9651055335998535,
+        8.533324241638184,
+        7.611879348754883,
+        1.9443340301513672,
+        4.746319770812988,
+        2.965162754058838,
+        5.064094066619873,
+        4.081468105316162,
+        0.6930497288703918,
+        3.280083656311035,
+        3.307518243789673,
+        0.9383264183998108,
+        2.1687474250793457
+      ],
+      "lowhigh_cos": [
+        0.2833213210105896,
+        0.22898192703723907,
+        0.5208947062492371,
+        0.6059155464172363,
+        -0.015384141355752945,
+        0.12832285463809967,
+        0.15682637691497803,
+        0.2393011450767517,
+        -0.10133591294288635,
+        -0.11736318469047546,
+        0.0006828118348494172,
+        0.6370846629142761,
+        0.10531243681907654,
+        0.07136750221252441,
+        0.45528534054756165,
+        0.6974369883537292,
+        0.5258194804191589,
+        0.3584192991256714,
+        0.2281290590763092,
+        0.23442195355892181
+      ]
+    },
+    "frozen_backbone": {
+      "iter": [
+        10,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        100,
+        110,
+        120,
+        130,
+        140,
+        150,
+        160,
+        170,
+        180,
+        190,
+        200,
+        210,
+        220,
+        230,
+        240,
+        250,
+        260,
+        270,
+        280,
+        290,
+        300,
+        310,
+        320,
+        330,
+        340,
+        350,
+        360,
+        370,
+        380,
+        390,
+        400,
+        410,
+        420,
+        430,
+        440,
+        450,
+        460,
+        470,
+        480,
+        490,
+        500
+      ],
+      "loss": [
+        0.8432541787624359,
+        0.6051806628704071,
+        0.9249359101057053,
+        0.5307856887578964,
+        0.4062101155519485,
+        1.2530769169330598,
+        0.9248880535364151,
+        0.6459330469369888,
+        0.3750177398324013,
+        0.4880190521478653,
+        0.5768440961837769,
+        0.34753591269254686,
+        0.46058507561683654,
+        0.5233289957046509,
+        0.49718080908060075,
+        0.5780578717589379,
+        0.3984171137213707,
+        0.28467713445425036,
+        0.21674730628728867,
+        0.19075852558016776,
+        0.19515338838100432,
+        0.24626194760203363,
+        0.3154190480709076,
+        0.2720298781991005,
+        0.23943385034799575,
+        0.1497550018131733,
+        0.18967929482460022,
+        0.12926405742764474,
+        0.18151819109916686,
+        0.2395605966448784,
+        0.198850055038929,
+        0.30438809990882876,
+        0.21534736976027488,
+        0.22932147979736328,
+        0.19919342622160913,
+        0.18485659509897232,
+        0.2822298489511013,
+        0.1751813419163227,
+        0.1328229382634163,
+        0.1326775722205639,
+        0.11731504946947098,
+        0.14120452776551246,
+        0.12244693860411644,
+        0.16777455508708955,
+        0.1694161780178547,
+        0.1586403626948595,
+        0.10118742138147355,
+        0.14438418820500373,
+        0.12352714017033577,
+        0.15858156234025955
+      ],
+      "mean_return": [
+        14.802000000000001,
+        9.073000000000002,
+        13.052000000000003,
+        1.3720000000000057,
+        4.766000000000008,
+        10.137,
+        8.38700000000001,
+        -3.6769999999999974,
+        6.716999999999999,
+        1.3950000000000033,
+        -0.9859999999999903,
+        1.7280000000000055,
+        2.1439999999999984,
+        2.2280000000000033,
+        -2.8790000000000013,
+        -3.4519999999999955,
+        -0.8159999999999996,
+        1.6260000000000012,
+        -1.531999999999989,
+        -4.665999999999986,
+        -1.4189999999999887,
+        -0.8670000000000023,
+        -0.5189999999999984,
+        -0.9310000000000024,
+        -0.8409999999999933,
+        0.9580000000000105,
+        -3.4829999999999908,
+        0.35500000000001464,
+        5.418000000000001,
+        -3.9919999999999867,
+        -1.30199999999999,
+        -0.7849999999999947,
+        -1.5739999999999916,
+        -3.1340000000000017,
+        -1.602999999999985,
+        -3.8719999999999914,
+        -3.574999999999995,
+        -4.1189999999999936,
+        3.8230000000000004,
+        1.6980000000000022,
+        -0.9009999999999941,
+        9.245000000000001,
+        0.9290000000000103,
+        -3.066999999999987,
+        -2.8059999999999956,
+        -3.5890000000000013,
+        -0.34999999999999437,
+        -0.921000000000002,
+        -4.478999999999983,
+        -3.7299999999999947
+      ],
+      "win_rate_episode": [
+        0.7,
+        0.5,
+        0.6,
+        0.2,
+        0.3,
+        0.5,
+        0.5,
+        0.0,
+        0.4,
+        0.2,
+        0.1,
+        0.2,
+        0.2,
+        0.2,
+        0.0,
+        0.0,
+        0.1,
+        0.2,
+        0.1,
+        0.0,
+        0.1,
+        0.1,
+        0.1,
+        0.1,
+        0.1,
+        0.2,
+        0.0,
+        0.2,
+        0.3,
+        0.0,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.3,
+        0.2,
+        0.1,
+        0.5,
+        0.2,
+        0.0,
+        0.0,
+        0.0,
+        0.1,
+        0.1,
+        0.0,
+        0.0
+      ],
+      "id_winrate": [
+        0.525,
+        0.3,
+        0.07500000000000001,
+        0.05,
+        0.05
+      ],
+      "id_winrate_iter": [
+        100,
+        200,
+        300,
+        400,
+        500
+      ],
+      "ood_winrate": [
+        0.03333333333333333
+      ],
+      "ood_winrate_iter": [
+        500
+      ],
+      "grad_align_iter": [],
+      "grad_align": [],
+      "rl_grad_norm": [],
+      "bc_grad_norm": [],
+      "repr_drift_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "repr_drift": [
+        0.15479625761508942,
+        0.484932541847229,
+        0.5217902660369873,
+        0.7290679216384888,
+        0.5376307964324951,
+        0.8892215490341187,
+        1.3117743730545044,
+        1.452524185180664,
+        2.022955894470215,
+        1.654375433921814,
+        2.462423324584961,
+        2.5157599449157715,
+        2.2190136909484863,
+        2.9022059440612793,
+        2.673858880996704,
+        2.663733959197998,
+        4.088006019592285,
+        3.895555019378662,
+        4.047849655151367,
+        4.74190616607666
+      ],
+      "t_analysis_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "norm_low_t": [
+        0.5986595153808594,
+        0.6866065859794617,
+        0.42266008257865906,
+        11.257745742797852,
+        0.4751920998096466,
+        0.729705810546875,
+        0.45668742060661316,
+        0.5257764458656311,
+        0.3505500853061676,
+        0.5213647484779358,
+        0.4264509975910187,
+        0.3384961485862732,
+        0.27900227904319763,
+        0.2823605239391327,
+        0.1809273511171341,
+        0.3805403411388397,
+        0.26565390825271606,
+        3.993173122406006,
+        0.6758583188056946,
+        0.3116438686847687
+      ],
+      "norm_high_t": [
+        0.968696117401123,
+        14.156644821166992,
+        0.6386823654174805,
+        2.8380041122436523,
+        3.138822078704834,
+        1.537684440612793,
+        0.7374914884567261,
+        0.3873365819454193,
+        0.6569968461990356,
+        1.7117724418640137,
+        0.4027094542980194,
+        2.9362730979919434,
+        0.6246607899665833,
+        1.5652388334274292,
+        0.2736610174179077,
+        4.938521385192871,
+        1.3465253114700317,
+        2.811922550201416,
+        2.560199022293091,
+        0.773190975189209
+      ],
+      "lowhigh_cos": [
+        0.34187427163124084,
+        -0.013280068524181843,
+        0.1718188226222992,
+        0.4921704828739166,
+        0.4253537058830261,
+        0.10736553370952606,
+        0.10130278766155243,
+        0.36192867159843445,
+        0.06617449969053268,
+        0.31431812047958374,
+        0.14055868983268738,
+        -0.09543662518262863,
+        0.15687131881713867,
+        0.1258467733860016,
+        0.18483635783195496,
+        0.4652028977870941,
+        -0.2167549729347229,
+        0.44112205505371094,
+        0.4113374948501587,
+        0.02761063352227211
+      ]
+    },
+    "bc_on_wins": {
+      "iter": [
+        10,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        100,
+        110,
+        120,
+        130,
+        140,
+        150,
+        160,
+        170,
+        180,
+        190,
+        200,
+        210,
+        220,
+        230,
+        240,
+        250,
+        260,
+        270,
+        280,
+        290,
+        300,
+        310,
+        320,
+        330,
+        340,
+        350,
+        360,
+        370,
+        380,
+        390,
+        400,
+        410,
+        420,
+        430,
+        440,
+        450,
+        460,
+        470,
+        480,
+        490,
+        500
+      ],
+      "loss": [
+        0.0,
+        2.809250086545944,
+        1.5345448434352875,
+        1.3774227797985077,
+        0.273484418541193,
+        0.09973169527947903,
+        0.0766332889907062,
+        0.06333817462436855,
+        0.305202372930944,
+        0.13767885118722917,
+        0.13696757480502128,
+        0.12410348504781724,
+        0.2557923704385757,
+        0.223702372610569,
+        0.11811406463384629,
+        0.1622357800602913,
+        0.1446425747126341,
+        0.06279170280322433,
+        0.060602336190640926,
+        0.12183188050985336,
+        0.12065998539328575,
+        0.07745412643998861,
+        0.1753513276576996,
+        0.15074485763907433,
+        0.22228444442152978,
+        0.24410116225481032,
+        0.25474355965852735,
+        0.15105561465024947,
+        0.1774226851761341,
+        0.18747113943099974,
+        0.11809183433651924,
+        0.2158730670809746,
+        0.13357947915792465,
+        0.14638029038906097,
+        0.1506120301783085,
+        0.10705487877130508,
+        0.12562319375574588,
+        0.08112300224602223,
+        0.1059636577963829,
+        0.11252538859844208,
+        0.07254688572138548,
+        0.0704629324376583,
+        0.08125015869736671,
+        0.11510146819055081,
+        0.048482533730566504,
+        0.0742773407138884,
+        0.056103642005473374,
+        0.10363874156028033,
+        0.1402372971177101,
+        0.08852829486131668
+      ],
+      "mean_return": [
+        9.337000000000007,
+        3.8970000000000082,
+        7.945000000000002,
+        -4.483999999999985,
+        -0.8439999999999941,
+        -4.142999999999989,
+        1.8410000000000093,
+        -3.6869999999999954,
+        0.5000000000000071,
+        -3.57399999999999,
+        0.41500000000002013,
+        1.5700000000000036,
+        -1.10499999999999,
+        -3.3829999999999965,
+        -4.688999999999977,
+        -2.5409999999999995,
+        -1.394999999999992,
+        -3.8249999999999913,
+        -3.151999999999993,
+        -2.763999999999993,
+        1.6550000000000076,
+        -2.5829999999999993,
+        -0.6139999999999803,
+        0.6570000000000067,
+        -2.7069999999999856,
+        7.194,
+        0.389000000000001,
+        -3.9149999999999836,
+        -3.1179999999999954,
+        -3.7479999999999913,
+        0.7230000000000114,
+        -3.7059999999999933,
+        -3.053999999999985,
+        -3.2609999999999966,
+        -0.3719999999999978,
+        -3.333999999999996,
+        -0.9279999999999944,
+        -0.3180000000000021,
+        4.698000000000009,
+        -1.359999999999996,
+        -1.7269999999999768,
+        -0.8929999999999929,
+        -2.981999999999995,
+        -1.4949999999999903,
+        -0.6849999999999985,
+        -1.5049999999999946,
+        -3.3639999999999937,
+        -2.5750000000000006,
+        -1.2899999999999896,
+        -2.8209999999999935
+      ],
+      "win_rate_episode": [
+        0.5,
+        0.3,
+        0.4,
+        0.0,
+        0.1,
+        0.0,
+        0.2,
+        0.0,
+        0.1,
+        0.0,
+        0.2,
+        0.2,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.2,
+        0.0,
+        0.1,
+        0.2,
+        0.0,
+        0.4,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.2,
+        0.0,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.1,
+        0.1,
+        0.3,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.0,
+        0.1,
+        0.0
+      ],
+      "id_winrate": [
+        0.05,
+        0.07500000000000001,
+        0.07500000000000001,
+        0.05,
+        0.07500000000000001
+      ],
+      "id_winrate_iter": [
+        100,
+        200,
+        300,
+        400,
+        500
+      ],
+      "ood_winrate": [
+        0.0
+      ],
+      "ood_winrate_iter": [
+        500
+      ],
+      "grad_align_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "grad_align": [
+        -0.15585047006607056,
+        0.014388328418135643,
+        -0.0008289138786494732,
+        0.009685968980193138,
+        -0.14255520701408386,
+        -0.03257212042808533,
+        0.15654288232326508,
+        0.008197564631700516,
+        -0.11398962885141373,
+        0.017725320532917976,
+        -0.0008931104093790054,
+        0.03191083297133446,
+        -0.055659446865320206,
+        -0.0731707215309143,
+        0.08635709434747696,
+        -0.0048180073499679565,
+        0.05373326316475868,
+        -0.051411353051662445,
+        0.031434930860996246,
+        0.010565715841948986
+      ],
+      "rl_grad_norm": [
+        17.90909194946289,
+        9.267593383789062,
+        2.835970401763916,
+        1.5497463941574097,
+        0.648881196975708,
+        0.17218180000782013,
+        1.2748949527740479,
+        1.3406476974487305,
+        2.2597739696502686,
+        1.3221700191497803,
+        1.6656370162963867,
+        1.313543677330017,
+        1.1664280891418457,
+        1.290276288986206,
+        0.5993183255195618,
+        1.2788653373718262,
+        2.9476051330566406,
+        0.33282503485679626,
+        0.1632971167564392,
+        1.2346807718276978
+      ],
+      "bc_grad_norm": [
+        37.38460922241211,
+        57.11834716796875,
+        55.3564338684082,
+        43.19457244873047,
+        42.32614517211914,
+        49.10124588012695,
+        44.5236930847168,
+        49.081302642822266,
+        49.36444854736328,
+        46.51043701171875,
+        42.867820739746094,
+        46.32526779174805,
+        41.31015396118164,
+        45.431243896484375,
+        40.98322677612305,
+        42.76980972290039,
+        46.33472442626953,
+        45.720069885253906,
+        51.702796936035156,
+        53.660099029541016
+      ],
+      "repr_drift_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "repr_drift": [
+        6.027996063232422,
+        6.684228897094727,
+        11.888472557067871,
+        15.770934104919434,
+        19.830795288085938,
+        17.60736083984375,
+        14.056540489196777,
+        13.429742813110352,
+        13.57545280456543,
+        13.590494155883789,
+        14.123512268066406,
+        14.357796669006348,
+        15.667505264282227,
+        13.821261405944824,
+        14.207181930541992,
+        13.969894409179688,
+        14.930856704711914,
+        14.994295120239258,
+        14.834063529968262,
+        14.4537992477417
+      ],
+      "t_analysis_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "norm_low_t": [
+        20.25221824645996,
+        0.754869282245636,
+        0.17779888212680817,
+        3.0701024532318115,
+        1.7086597681045532,
+        2.237590789794922,
+        5.5478386878967285,
+        6.110817909240723,
+        0.46025317907333374,
+        0.11583580821752548,
+        3.8377528190612793,
+        0.8099531531333923,
+        1.5192604064941406,
+        5.958379745483398,
+        0.4879041016101837,
+        1.4312430620193481,
+        0.4303628206253052,
+        0.37884828448295593,
+        0.46471869945526123,
+        2.001178741455078
+      ],
+      "norm_high_t": [
+        21.141122817993164,
+        3.2488327026367188,
+        3.682577610015869,
+        2.676034450531006,
+        1.2364305257797241,
+        2.0812149047851562,
+        1.543370008468628,
+        2.039804220199585,
+        1.3472521305084229,
+        2.179379463195801,
+        5.384540557861328,
+        1.6266305446624756,
+        0.9822807908058167,
+        1.5289103984832764,
+        2.2616400718688965,
+        2.6361918449401855,
+        0.9600972533226013,
+        0.442050963640213,
+        0.3930566608905792,
+        0.7594590783119202
+      ],
+      "lowhigh_cos": [
+        0.46482086181640625,
+        -0.018056275323033333,
+        0.2649460434913635,
+        0.2176809012889862,
+        0.21259765326976776,
+        0.340675413608551,
+        -0.17475268244743347,
+        0.46381503343582153,
+        0.295942097902298,
+        0.35303354263305664,
+        0.3157437741756439,
+        0.15970274806022644,
+        0.5105319023132324,
+        0.3774409294128418,
+        0.22727873921394348,
+        0.20067614316940308,
+        0.2528092861175537,
+        0.4246519207954407,
+        -0.24365077912807465,
+        0.4615093469619751
+      ]
+    },
+    "low_t_only": {
+      "iter": [
+        10,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        100,
+        110,
+        120,
+        130,
+        140,
+        150,
+        160,
+        170,
+        180,
+        190,
+        200,
+        210,
+        220,
+        230,
+        240,
+        250,
+        260,
+        270,
+        280,
+        290,
+        300,
+        310,
+        320,
+        330,
+        340,
+        350,
+        360,
+        370,
+        380,
+        390,
+        400,
+        410,
+        420,
+        430,
+        440,
+        450,
+        460,
+        470,
+        480,
+        490,
+        500
+      ],
+      "loss": [
+        0.16954294890165328,
+        0.12819076813757418,
+        0.17230339162051678,
+        0.09248451106250286,
+        0.16420554406940938,
+        0.1693354170769453,
+        0.5027543395757675,
+        0.5454125240445137,
+        0.18514752238988877,
+        0.07633375357836485,
+        0.29205870665609834,
+        0.10658948160707951,
+        0.1487645037472248,
+        0.07421774212270975,
+        0.1682606603950262,
+        0.03632904961705208,
+        0.0382061586715281,
+        0.02044668570160866,
+        0.07106257285922765,
+        0.08296542745083571,
+        0.053635398391634226,
+        0.17159658959135413,
+        0.06913567988667638,
+        0.02520559049444273,
+        0.023581319977529346,
+        0.024397523747757076,
+        0.01780195008032024,
+        0.05103614148683846,
+        0.007415541564114392,
+        0.024072827032068745,
+        0.17588082720176318,
+        0.041048154188320043,
+        0.01608952531169052,
+        0.012731735737179405,
+        0.08324823407456279,
+        0.006074277223888203,
+        0.0398330731986789,
+        0.015381297026760877,
+        0.0053429307314218025,
+        0.006615792286174838,
+        0.00928727765185613,
+        0.006404643525092979,
+        0.0021499938324268443,
+        0.003018749266720988,
+        0.003801402702447376,
+        0.025576220805851334,
+        0.010855860910396586,
+        0.008461204392369836,
+        0.0035062126145021465,
+        0.003927168410518789
+      ],
+      "mean_return": [
+        6.852000000000008,
+        12.036000000000005,
+        4.393000000000008,
+        9.750999999999996,
+        -0.7949999999999873,
+        0.3310000000000041,
+        -0.39499999999999486,
+        -0.8029999999999967,
+        -4.120999999999986,
+        -1.315999999999994,
+        -1.9249999999999883,
+        -3.5099999999999896,
+        -4.153999999999989,
+        -3.610999999999991,
+        -1.2209999999999934,
+        -3.4219999999999944,
+        -3.9589999999999903,
+        -1.556999999999995,
+        -1.2349999999999972,
+        -0.8050000000000004,
+        -4.293999999999988,
+        0.9720000000000105,
+        -3.571999999999993,
+        -4.081999999999988,
+        -1.518999999999984,
+        -1.2259999999999966,
+        -0.10999999999999766,
+        -3.3099999999999907,
+        -3.8639999999999963,
+        -3.451999999999996,
+        0.8820000000000116,
+        2.355,
+        0.2150000000000046,
+        -4.427999999999989,
+        -4.061999999999985,
+        -3.813999999999994,
+        -2.9899999999999984,
+        -1.2689999999999966,
+        -4.1039999999999885,
+        -3.571999999999997,
+        -0.5320000000000011,
+        -3.539999999999999,
+        -0.7389999999999958,
+        -3.6379999999999995,
+        -2.521999999999996,
+        -2.839999999999999,
+        2.3600000000000034,
+        -1.5089999999999921,
+        -4.397999999999984,
+        -4.4839999999999876
+      ],
+      "win_rate_episode": [
+        0.4,
+        0.6,
+        0.3,
+        0.5,
+        0.1,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.1,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.2,
+        0.0,
+        0.0,
+        0.1,
+        0.1,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.2,
+        0.2,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.1,
+        0.0,
+        0.1,
+        0.0,
+        0.0,
+        0.0,
+        0.2,
+        0.1,
+        0.0,
+        0.0
+      ],
+      "id_winrate": [
+        0.05,
+        0.05,
+        0.0,
+        0.1,
+        0.025
+      ],
+      "id_winrate_iter": [
+        100,
+        200,
+        300,
+        400,
+        500
+      ],
+      "ood_winrate": [
+        0.0
+      ],
+      "ood_winrate_iter": [
+        500
+      ],
+      "grad_align_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "grad_align": [
+        0.2528616189956665,
+        -0.03359116613864899,
+        0.02611994370818138,
+        0.0035057030618190765,
+        0.07301062345504761,
+        -0.05015534162521362,
+        0.04849525913596153,
+        0.04365059360861778,
+        0.16888359189033508,
+        0.038295499980449677,
+        -0.020476030185818672,
+        0.11001922190189362,
+        -0.03407794237136841,
+        -0.2625855803489685,
+        0.04202434420585632,
+        0.1309511661529541,
+        -0.006483441684395075,
+        0.005877040326595306,
+        -0.003088176716119051,
+        -0.007350331172347069
+      ],
+      "rl_grad_norm": [
+        0.708732008934021,
+        4.030664920806885,
+        4.342776298522949,
+        0.7132593989372253,
+        3.507981061935425,
+        1.6703108549118042,
+        2.1692845821380615,
+        1.730566143989563,
+        0.6206149458885193,
+        1.1632376909255981,
+        4.863556385040283,
+        0.43041908740997314,
+        0.34915438294410706,
+        3.184418201446533,
+        3.9088478088378906,
+        0.632773220539093,
+        0.3219628930091858,
+        0.058308567851781845,
+        7.77617073059082,
+        0.43638089299201965
+      ],
+      "bc_grad_norm": [
+        19.74188232421875,
+        21.830142974853516,
+        22.09646224975586,
+        21.381101608276367,
+        26.10833168029785,
+        24.06924819946289,
+        23.066692352294922,
+        26.531723022460938,
+        23.48773765563965,
+        18.655624389648438,
+        22.68110466003418,
+        23.721654891967773,
+        26.712148666381836,
+        38.09318542480469,
+        13.851006507873535,
+        25.637195587158203,
+        22.518232345581055,
+        26.217267990112305,
+        20.741737365722656,
+        22.005754470825195
+      ],
+      "repr_drift_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "repr_drift": [
+        1.8782662153244019,
+        2.9116997718811035,
+        2.912844181060791,
+        3.6854569911956787,
+        3.0491065979003906,
+        4.025200366973877,
+        4.199270248413086,
+        4.503491401672363,
+        3.1839468479156494,
+        3.253112554550171,
+        3.8597023487091064,
+        5.276954650878906,
+        4.396190643310547,
+        3.5537588596343994,
+        2.945570707321167,
+        4.853464126586914,
+        4.4204020500183105,
+        4.409016132354736,
+        5.554651260375977,
+        4.029665470123291
+      ],
+      "t_analysis_iter": [
+        25,
+        50,
+        75,
+        100,
+        125,
+        150,
+        175,
+        200,
+        225,
+        250,
+        275,
+        300,
+        325,
+        350,
+        375,
+        400,
+        425,
+        450,
+        475,
+        500
+      ],
+      "norm_low_t": [
+        0.8107094168663025,
+        0.5331999659538269,
+        0.5337082147598267,
+        0.4895950257778168,
+        2.713630199432373,
+        0.49118441343307495,
+        0.08188782632350922,
+        0.39152956008911133,
+        0.2501949965953827,
+        0.1392025649547577,
+        0.33048632740974426,
+        0.1271236538887024,
+        0.29568424820899963,
+        0.11166119575500488,
+        0.28577253222465515,
+        0.20214661955833435,
+        0.00438786530867219,
+        0.0016851737163960934,
+        8.534547805786133,
+        0.08704172074794769
+      ],
+      "norm_high_t": [
+        0.9552221894264221,
+        2.923011302947998,
+        2.3088951110839844,
+        3.3186538219451904,
+        5.08151388168335,
+        2.726167678833008,
+        1.75440514087677,
+        3.37270188331604,
+        0.6986761093139648,
+        0.5467635989189148,
+        2.235100269317627,
+        0.4686378538608551,
+        0.45633357763290405,
+        2.896705389022827,
+        0.8851325511932373,
+        6.972827434539795,
+        2.036700487136841,
+        0.20503078401088715,
+        1.1935913562774658,
+        0.7886370420455933
+      ],
+      "lowhigh_cos": [
+        0.1374405026435852,
+        0.044124118983745575,
+        0.252091646194458,
+        0.2623433470726013,
+        0.2519252598285675,
+        -0.0307513065636158,
+        0.08866468071937561,
+        0.3848387598991394,
+        0.04764184355735779,
+        0.15670593082904816,
+        -0.11194318532943726,
+        -0.3121059238910675,
+        0.4413709342479706,
+        0.36320483684539795,
+        0.10349930822849274,
+        0.28844499588012695,
+        0.475638747215271,
+        -0.08917384594678879,
+        0.4083694517612457,
+        0.20888718962669373
+      ]
+    },
+    "synthetic": {
+      "iter": [
+        10,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        100,
+        110,
+        120,
+        130,
+        140,
+        150,
+        160,
+        170,
+        180,
+        190,
+        200,
+        210,
+        220,
+        230,
+        240,
+        250,
+        260,
+        270,
+        280,
+        290,
+        300,
+        310,
+        320,
+        330,
+        340,
+        350,
+        360,
+        370,
+        380,
+        390,
+        400,
+        410,
+        420,
+        430,
+        440,
+        450,
+        460,
+        470,
+        480,
+        490,
+        500
+      ],
+      "loss": [
+        0.7531764507293701,
+        0.18765906989574432,
+        0.020723845809698105,
+        0.02801017463207245,
+        0.0033709718845784664,
+        0.20654648542404175,
+        0.08307354152202606,
+        0.01641359180212021,
+        0.00023058410442899913,
+        0.006693837232887745,
+        0.0006810599006712437,
+        0.025508670136332512,
+        0.0015238457126542926,
+        0.05820140987634659,
+        0.0008688275702297688,
+        0.00020045570272486657,
+        0.00022424766211770475,
+        3.519005258567631e-05,
+        0.01258005015552044,
+        3.128516982542351e-05,
+        0.0009578695753589272,
+        0.0010310658253729343,
+        0.03165672719478607,
+        1.3986779777042102e-05,
+        0.0014118634862825274,
+        0.002570120384916663,
+        0.0007432415150105953,
+        1.3963205674372148e-05,
+        9.562265768181533e-05,
+        8.88048016349785e-05,
+        0.0001328531070612371,
+        0.00014389181160368025,
+        0.2233337163925171,
+        9.16919598239474e-05,
+        0.06965351104736328,
+        0.0024555602576583624,
+        0.00010386335634393618,
+        0.03198186308145523,
+        0.00020677264546975493,
+        0.01207452081143856,
+        0.00016542139928787947,
+        1.0498906704015099e-05,
+        2.4054688196883944e-07,
+        1.8041833754978143e-05,
+        0.0001326937781414017,
+        4.153811460128054e-05,
+        0.0006373185897246003,
+        0.059840451925992966,
+        0.03510911762714386,
+        1.009482002700679e-05
+      ],
+      "mean_return": [],
+      "win_rate_episode": [],
+      "id_winrate": [
+        0.15000000000000002,
+        0.15,
+        0.025,
+        0.05,
+        0.0
+      ],
+      "id_winrate_iter": [
+        100,
+        200,
+        300,
+        400,
+        500
+      ],
+      "ood_winrate": [],
+      "ood_winrate_iter": [],
+      "grad_align_iter": [],
+      "grad_align": [],
+      "rl_grad_norm": [],
+      "bc_grad_norm": [],
+      "repr_drift_iter": [],
+      "repr_drift": [],
+      "t_analysis_iter": [],
+      "norm_low_t": [],
+      "norm_high_t": [],
+      "lowhigh_cos": []
+    }
+  }
+}