{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 28.06361085126286,
  "eval_steps": 500,
  "global_step": 30000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.009354536950420954,
      "flow_matching_loss": 1.06994286775589,
      "grad_norm": 11.036585807800293,
      "learning_rate": 6.000000000000001e-07,
      "loss": 1.1576,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.08761446848511696,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 1.912585484981537,
      "state_query_norm": 122.23323211669921,
      "state_relation_loss": 0.08761446848511696,
      "step": 10,
      "student_distance_mean": 107.70182037353516,
      "teacher_distance_mean": 1.137965452671051,
      "weighted_auxiliary_loss": 0.08761446848511696,
      "weighted_relation_loss": 0.08761446848511696
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.018709073900841908,
      "flow_matching_loss": 1.0576623678207397,
      "grad_norm": 12.550166130065918,
      "learning_rate": 1.2666666666666667e-06,
      "loss": 1.1416,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.08391660377383232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 1.8951416492462159,
      "state_query_norm": 121.24369277954102,
      "state_relation_loss": 0.08391660377383232,
      "step": 20,
      "student_distance_mean": 107.01396560668945,
      "teacher_distance_mean": 1.1540406823158265,
      "weighted_auxiliary_loss": 0.08391660377383232,
      "weighted_relation_loss": 0.08391660377383232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.02806361085126286,
      "flow_matching_loss": 0.8500910043716431,
      "grad_norm": 6.968210697174072,
      "learning_rate": 1.9333333333333336e-06,
      "loss": 0.9386,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.08848222866654396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 1.9533559203147888,
      "state_query_norm": 119.92243041992188,
      "state_relation_loss": 0.08848222866654396,
      "step": 30,
      "student_distance_mean": 109.72795715332032,
      "teacher_distance_mean": 1.144095528125763,
      "weighted_auxiliary_loss": 0.08848222866654396,
      "weighted_relation_loss": 0.08848222866654396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.037418147801683815,
      "flow_matching_loss": 0.6073942959308625,
      "grad_norm": 5.914402961730957,
      "learning_rate": 2.6e-06,
      "loss": 0.6955,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.08814469128847122,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.114087736606598,
      "state_query_norm": 124.4527359008789,
      "state_relation_loss": 0.08814469128847122,
      "step": 40,
      "student_distance_mean": 120.74618301391601,
      "teacher_distance_mean": 1.1482280254364015,
      "weighted_auxiliary_loss": 0.08814469128847122,
      "weighted_relation_loss": 0.08814469128847122
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.04677268475210477,
      "flow_matching_loss": 0.46203915774822235,
      "grad_norm": 1.7596853971481323,
      "learning_rate": 3.2666666666666666e-06,
      "loss": 0.5418,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.07972318902611733,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.186553120613098,
      "state_query_norm": 130.74512481689453,
      "state_relation_loss": 0.07972318902611733,
      "step": 50,
      "student_distance_mean": 124.04701385498046,
      "teacher_distance_mean": 1.1488664031028748,
      "weighted_auxiliary_loss": 0.07972318902611733,
      "weighted_relation_loss": 0.07972318902611733
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.05612722170252572,
      "flow_matching_loss": 0.37373038530349734,
      "grad_norm": 1.0894638299942017,
      "learning_rate": 3.9333333333333335e-06,
      "loss": 0.4452,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.07148488312959671,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.294408369064331,
      "state_query_norm": 140.68495330810546,
      "state_relation_loss": 0.07148488312959671,
      "step": 60,
      "student_distance_mean": 129.13553161621093,
      "teacher_distance_mean": 1.1556007981300354,
      "weighted_auxiliary_loss": 0.07148488312959671,
      "weighted_relation_loss": 0.07148488312959671
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.06548175865294668,
      "flow_matching_loss": 0.31855241656303407,
      "grad_norm": 0.9638234376907349,
      "learning_rate": 4.6e-06,
      "loss": 0.3771,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.05854181759059429,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.388656663894653,
      "state_query_norm": 143.36868438720703,
      "state_relation_loss": 0.05854181759059429,
      "step": 70,
      "student_distance_mean": 133.66366424560547,
      "teacher_distance_mean": 1.143768858909607,
      "weighted_auxiliary_loss": 0.05854181759059429,
      "weighted_relation_loss": 0.05854181759059429
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.07483629560336763,
      "flow_matching_loss": 0.2771622985601425,
      "grad_norm": 0.605579674243927,
      "learning_rate": 5.266666666666667e-06,
      "loss": 0.3268,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.04962201826274395,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.539447808265686,
      "state_query_norm": 144.86495819091797,
      "state_relation_loss": 0.04962201826274395,
      "step": 80,
      "student_distance_mean": 140.41712493896483,
      "teacher_distance_mean": 1.1609897375106812,
      "weighted_auxiliary_loss": 0.04962201826274395,
      "weighted_relation_loss": 0.04962201826274395
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.08419083255378859,
      "flow_matching_loss": 0.25133520513772967,
      "grad_norm": 0.5952714085578918,
      "learning_rate": 5.933333333333334e-06,
      "loss": 0.2944,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.04302356205880642,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.5729240655899046,
      "state_query_norm": 144.43463134765625,
      "state_relation_loss": 0.04302356205880642,
      "step": 90,
      "student_distance_mean": 142.00726165771485,
      "teacher_distance_mean": 1.1499913096427918,
      "weighted_auxiliary_loss": 0.04302356205880642,
      "weighted_relation_loss": 0.04302356205880642
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.09354536950420954,
      "flow_matching_loss": 0.2197645053267479,
      "grad_norm": 0.5051930546760559,
      "learning_rate": 6.6e-06,
      "loss": 0.2579,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.03810863196849823,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.5795870304107664,
      "state_query_norm": 141.77390899658204,
      "state_relation_loss": 0.03810863196849823,
      "step": 100,
      "student_distance_mean": 141.70136413574218,
      "teacher_distance_mean": 1.1468847751617433,
      "weighted_auxiliary_loss": 0.03810863196849823,
      "weighted_relation_loss": 0.03810863196849823
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.1028999064546305,
      "flow_matching_loss": 0.2065592274069786,
      "grad_norm": 0.6919609308242798,
      "learning_rate": 7.266666666666668e-06,
      "loss": 0.2401,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.033564095944166185,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.6853400230407716,
      "state_query_norm": 144.98010711669923,
      "state_relation_loss": 0.033564095944166185,
      "step": 110,
      "student_distance_mean": 147.53838958740235,
      "teacher_distance_mean": 1.1590776085853576,
      "weighted_auxiliary_loss": 0.033564095944166185,
      "weighted_relation_loss": 0.033564095944166185
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.11225444340505145,
      "flow_matching_loss": 0.19171424806118012,
      "grad_norm": 0.5677187442779541,
      "learning_rate": 7.933333333333334e-06,
      "loss": 0.2205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.02878142911940813,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 2.9120297193527223,
      "state_query_norm": 155.4535888671875,
      "state_relation_loss": 0.02878142911940813,
      "step": 120,
      "student_distance_mean": 160.3805694580078,
      "teacher_distance_mean": 1.152875304222107,
      "weighted_auxiliary_loss": 0.02878142911940813,
      "weighted_relation_loss": 0.02878142911940813
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.1216089803554724,
      "flow_matching_loss": 0.17957597821950913,
      "grad_norm": 0.6089605689048767,
      "learning_rate": 8.599999999999999e-06,
      "loss": 0.2052,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.025632178597152232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.1894851207733153,
      "state_query_norm": 167.62225952148438,
      "state_relation_loss": 0.025632178597152232,
      "step": 130,
      "student_distance_mean": 176.32713775634767,
      "teacher_distance_mean": 1.1681151270866394,
      "weighted_auxiliary_loss": 0.025632178597152232,
      "weighted_relation_loss": 0.025632178597152232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.13096351730589337,
      "flow_matching_loss": 0.1724351689219475,
      "grad_norm": 0.6334270238876343,
      "learning_rate": 9.266666666666667e-06,
      "loss": 0.1942,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.02173627931624651,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.3083380460739136,
      "state_query_norm": 168.3145309448242,
      "state_relation_loss": 0.02173627931624651,
      "step": 140,
      "student_distance_mean": 182.4239486694336,
      "teacher_distance_mean": 1.157258892059326,
      "weighted_auxiliary_loss": 0.02173627931624651,
      "weighted_relation_loss": 0.02173627931624651
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.1403180542563143,
      "flow_matching_loss": 0.16478968113660813,
      "grad_norm": 0.52126145362854,
      "learning_rate": 9.933333333333334e-06,
      "loss": 0.1852,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.020365650206804274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.576279044151306,
      "state_query_norm": 182.20634918212892,
      "state_relation_loss": 0.020365650206804274,
      "step": 150,
      "student_distance_mean": 198.57768859863282,
      "teacher_distance_mean": 1.1668389916419983,
      "weighted_auxiliary_loss": 0.020365650206804274,
      "weighted_relation_loss": 0.020365650206804274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.14967259120673526,
      "flow_matching_loss": 0.15129084885120392,
      "grad_norm": 0.5395965576171875,
      "learning_rate": 1.06e-05,
      "loss": 0.1699,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.018603138998150824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.6677271127700806,
      "state_query_norm": 187.50961456298828,
      "state_relation_loss": 0.018603138998150824,
      "step": 160,
      "student_distance_mean": 204.29331817626954,
      "teacher_distance_mean": 1.1370608806610107,
      "weighted_auxiliary_loss": 0.018603138998150824,
      "weighted_relation_loss": 0.018603138998150824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.15902712815715622,
      "flow_matching_loss": 0.14535031467676163,
      "grad_norm": 0.5902556777000427,
      "learning_rate": 1.1266666666666667e-05,
      "loss": 0.1613,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.01599673256278038,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.7667226791381836,
      "state_query_norm": 192.45580291748047,
      "state_relation_loss": 0.01599673256278038,
      "step": 170,
      "student_distance_mean": 209.86038665771486,
      "teacher_distance_mean": 1.1411669731140137,
      "weighted_auxiliary_loss": 0.01599673256278038,
      "weighted_relation_loss": 0.01599673256278038
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.16838166510757718,
      "flow_matching_loss": 0.14158526808023453,
      "grad_norm": 0.6628186702728271,
      "learning_rate": 1.1933333333333333e-05,
      "loss": 0.157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.015462531615048647,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.9215491771698,
      "state_query_norm": 196.8627685546875,
      "state_relation_loss": 0.015462531615048647,
      "step": 180,
      "student_distance_mean": 218.67012786865234,
      "teacher_distance_mean": 1.1530103802680969,
      "weighted_auxiliary_loss": 0.015462531615048647,
      "weighted_relation_loss": 0.015462531615048647
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.17773620205799812,
      "flow_matching_loss": 0.13818033635616303,
      "grad_norm": 0.548592746257782,
      "learning_rate": 1.2600000000000001e-05,
      "loss": 0.1518,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.013577943108975888,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 3.9869842529296875,
      "state_query_norm": 204.01189422607422,
      "state_relation_loss": 0.013577943108975888,
      "step": 190,
      "student_distance_mean": 223.1490280151367,
      "teacher_distance_mean": 1.1439606904983521,
      "weighted_auxiliary_loss": 0.013577943108975888,
      "weighted_relation_loss": 0.013577943108975888
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.18709073900841908,
      "flow_matching_loss": 0.139235956966877,
      "grad_norm": 0.49099427461624146,
      "learning_rate": 1.3266666666666666e-05,
      "loss": 0.1526,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.01337873712182045,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.180413627624512,
      "state_query_norm": 213.6262222290039,
      "state_relation_loss": 0.01337873712182045,
      "step": 200,
      "student_distance_mean": 234.83850860595703,
      "teacher_distance_mean": 1.1578856825828552,
      "weighted_auxiliary_loss": 0.01337873712182045,
      "weighted_relation_loss": 0.01337873712182045
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.19644527595884004,
      "flow_matching_loss": 0.1284155033528805,
      "grad_norm": 0.6125730276107788,
      "learning_rate": 1.3933333333333334e-05,
      "loss": 0.1404,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.011971226893365383,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.323363971710205,
      "state_query_norm": 218.7675308227539,
      "state_relation_loss": 0.011971226893365383,
      "step": 210,
      "student_distance_mean": 242.58125762939454,
      "teacher_distance_mean": 1.1613598227500916,
      "weighted_auxiliary_loss": 0.011971226893365383,
      "weighted_relation_loss": 0.011971226893365383
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.205799812909261,
      "flow_matching_loss": 0.13189892172813417,
      "grad_norm": 0.5268071293830872,
      "learning_rate": 1.4599999999999999e-05,
      "loss": 0.1433,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.011438625678420068,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.424276733398438,
      "state_query_norm": 224.7982925415039,
      "state_relation_loss": 0.011438625678420068,
      "step": 220,
      "student_distance_mean": 248.9791473388672,
      "teacher_distance_mean": 1.1646481275558471,
      "weighted_auxiliary_loss": 0.011438625678420068,
      "weighted_relation_loss": 0.011438625678420068
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.21515434985968196,
      "flow_matching_loss": 0.12362681776285171,
      "grad_norm": 0.37326371669769287,
      "learning_rate": 1.5266666666666667e-05,
      "loss": 0.1343,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.010704933479428292,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.494705057144165,
      "state_query_norm": 230.37777862548828,
      "state_relation_loss": 0.010704933479428292,
      "step": 230,
      "student_distance_mean": 253.93264617919922,
      "teacher_distance_mean": 1.146237075328827,
      "weighted_auxiliary_loss": 0.010704933479428292,
      "weighted_relation_loss": 0.010704933479428292
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.2245088868101029,
      "flow_matching_loss": 0.12420597970485688,
      "grad_norm": 0.500545084476471,
      "learning_rate": 1.5933333333333332e-05,
      "loss": 0.1347,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.010487499460577965,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.633782243728637,
      "state_query_norm": 231.69873504638673,
      "state_relation_loss": 0.010487499460577965,
      "step": 240,
      "student_distance_mean": 261.6099060058594,
      "teacher_distance_mean": 1.1522658824920655,
      "weighted_auxiliary_loss": 0.010487499460577965,
      "weighted_relation_loss": 0.010487499460577965
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.23386342376052385,
      "flow_matching_loss": 0.12334694266319275,
      "grad_norm": 0.434024840593338,
      "learning_rate": 1.66e-05,
      "loss": 0.1335,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.010174830444157124,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.679224348068237,
      "state_query_norm": 235.44737548828124,
      "state_relation_loss": 0.010174830444157124,
      "step": 250,
      "student_distance_mean": 265.1591430664063,
      "teacher_distance_mean": 1.1429988384246825,
      "weighted_auxiliary_loss": 0.010174830444157124,
      "weighted_relation_loss": 0.010174830444157124
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.2432179607109448,
      "flow_matching_loss": 0.12008417174220085,
      "grad_norm": 0.5392580032348633,
      "learning_rate": 1.726666666666667e-05,
      "loss": 0.13,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.009921714756637811,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.84806079864502,
      "state_query_norm": 243.50420989990235,
      "state_relation_loss": 0.009921714756637811,
      "step": 260,
      "student_distance_mean": 274.5570129394531,
      "teacher_distance_mean": 1.1598688602447509,
      "weighted_auxiliary_loss": 0.009921714756637811,
      "weighted_relation_loss": 0.009921714756637811
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.25257249766136575,
      "flow_matching_loss": 0.1122573345899582,
      "grad_norm": 0.43989697098731995,
      "learning_rate": 1.7933333333333337e-05,
      "loss": 0.1216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.009373194258660077,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.860552597045898,
      "state_query_norm": 245.321826171875,
      "state_relation_loss": 0.009373194258660077,
      "step": 270,
      "student_distance_mean": 276.5357971191406,
      "teacher_distance_mean": 1.1415817856788635,
      "weighted_auxiliary_loss": 0.009373194258660077,
      "weighted_relation_loss": 0.009373194258660077
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.26192703461178674,
      "flow_matching_loss": 0.11017514243721963,
      "grad_norm": 0.5808083415031433,
      "learning_rate": 1.86e-05,
      "loss": 0.119,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.008810459729284048,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.937539672851562,
      "state_query_norm": 248.3888427734375,
      "state_relation_loss": 0.008810459729284048,
      "step": 280,
      "student_distance_mean": 280.6804138183594,
      "teacher_distance_mean": 1.1477226853370666,
      "weighted_auxiliary_loss": 0.008810459729284048,
      "weighted_relation_loss": 0.008810459729284048
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.27128157156220767,
      "flow_matching_loss": 0.11428771242499351,
      "grad_norm": 0.4130600690841675,
      "learning_rate": 1.926666666666667e-05,
      "loss": 0.1228,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.008523033559322357,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 4.988988399505615,
      "state_query_norm": 251.62523193359374,
      "state_relation_loss": 0.008523033559322357,
      "step": 290,
      "student_distance_mean": 284.3015563964844,
      "teacher_distance_mean": 1.1395103096961976,
      "weighted_auxiliary_loss": 0.008523033559322357,
      "weighted_relation_loss": 0.008523033559322357
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.2806361085126286,
      "flow_matching_loss": 0.11149175688624383,
      "grad_norm": 0.6376433968544006,
      "learning_rate": 1.9933333333333334e-05,
      "loss": 0.12,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.008468803577125072,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.175398874282837,
      "state_query_norm": 259.06895141601564,
      "state_relation_loss": 0.008468803577125072,
      "step": 300,
      "student_distance_mean": 294.39154663085935,
      "teacher_distance_mean": 1.1515833497047425,
      "weighted_auxiliary_loss": 0.008468803577125072,
      "weighted_relation_loss": 0.008468803577125072
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.2899906454630496,
      "flow_matching_loss": 0.11139505729079247,
      "grad_norm": 0.49724438786506653,
      "learning_rate": 2.06e-05,
      "loss": 0.1194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.007979746675118804,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.245955085754394,
      "state_query_norm": 261.9921401977539,
      "state_relation_loss": 0.007979746675118804,
      "step": 310,
      "student_distance_mean": 297.7452880859375,
      "teacher_distance_mean": 1.1489434480667113,
      "weighted_auxiliary_loss": 0.007979746675118804,
      "weighted_relation_loss": 0.007979746675118804
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.2993451824134705,
      "flow_matching_loss": 0.10828804224729538,
      "grad_norm": 0.4152394235134125,
      "learning_rate": 2.1266666666666667e-05,
      "loss": 0.1165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.008170874416828155,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.353177165985107,
      "state_query_norm": 265.990771484375,
      "state_relation_loss": 0.008170874416828155,
      "step": 320,
      "student_distance_mean": 305.14752197265625,
      "teacher_distance_mean": 1.1579443216323853,
      "weighted_auxiliary_loss": 0.008170874416828155,
      "weighted_relation_loss": 0.008170874416828155
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.3086997193638915,
      "flow_matching_loss": 0.1071076400578022,
      "grad_norm": 0.49486997723579407,
      "learning_rate": 2.1933333333333332e-05,
      "loss": 0.1144,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.007261159271001816,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.4189537525177,
      "state_query_norm": 271.2239929199219,
      "state_relation_loss": 0.007261159271001816,
      "step": 330,
      "student_distance_mean": 309.0622924804687,
      "teacher_distance_mean": 1.1571632623672485,
      "weighted_auxiliary_loss": 0.007261159271001816,
      "weighted_relation_loss": 0.007261159271001816
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.31805425631431244,
      "flow_matching_loss": 0.10828360617160797,
      "grad_norm": 0.49699074029922485,
      "learning_rate": 2.26e-05,
      "loss": 0.1158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0074671595357358456,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.478713274002075,
      "state_query_norm": 275.1707275390625,
      "state_relation_loss": 0.0074671595357358456,
      "step": 340,
      "student_distance_mean": 312.77305908203124,
      "teacher_distance_mean": 1.1601409435272216,
      "weighted_auxiliary_loss": 0.0074671595357358456,
      "weighted_relation_loss": 0.0074671595357358456
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.3274087932647334,
      "flow_matching_loss": 0.10375735834240914,
      "grad_norm": 0.520050585269928,
      "learning_rate": 2.326666666666667e-05,
      "loss": 0.1113,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.007506386423483491,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.333050203323364,
      "state_query_norm": 269.05199279785154,
      "state_relation_loss": 0.007506386423483491,
      "step": 350,
      "student_distance_mean": 304.4993530273438,
      "teacher_distance_mean": 1.139188325405121,
      "weighted_auxiliary_loss": 0.007506386423483491,
      "weighted_relation_loss": 0.007506386423483491
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.33676333021515437,
      "flow_matching_loss": 0.10412324666976928,
      "grad_norm": 0.48651543259620667,
      "learning_rate": 2.3933333333333337e-05,
      "loss": 0.1113,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0071356741711497305,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.531252431869507,
      "state_query_norm": 280.1615509033203,
      "state_relation_loss": 0.0071356741711497305,
      "step": 360,
      "student_distance_mean": 317.13751831054685,
      "teacher_distance_mean": 1.1634278655052186,
      "weighted_auxiliary_loss": 0.0071356741711497305,
      "weighted_relation_loss": 0.0071356741711497305
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.3461178671655753,
      "flow_matching_loss": 0.10350734740495682,
      "grad_norm": 0.442334920167923,
      "learning_rate": 2.46e-05,
      "loss": 0.1104,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.006863347161561251,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.554446649551392,
      "state_query_norm": 274.81434936523436,
      "state_relation_loss": 0.006863347161561251,
      "step": 370,
      "student_distance_mean": 318.16126403808596,
      "teacher_distance_mean": 1.1494271159172058,
      "weighted_auxiliary_loss": 0.006863347161561251,
      "weighted_relation_loss": 0.006863347161561251
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.35547240411599623,
      "flow_matching_loss": 0.10137954577803612,
      "grad_norm": 0.5168814659118652,
      "learning_rate": 2.5266666666666666e-05,
      "loss": 0.1081,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.006707955105230212,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.613587093353272,
      "state_query_norm": 282.34519958496094,
      "state_relation_loss": 0.006707955105230212,
      "step": 380,
      "student_distance_mean": 322.481982421875,
      "teacher_distance_mean": 1.1458631873130798,
      "weighted_auxiliary_loss": 0.006707955105230212,
      "weighted_relation_loss": 0.006707955105230212
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.3648269410664172,
      "flow_matching_loss": 0.1063816525042057,
      "grad_norm": 0.4626791179180145,
      "learning_rate": 2.5933333333333338e-05,
      "loss": 0.1129,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0064720848575234415,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.708681344985962,
      "state_query_norm": 284.74302673339844,
      "state_relation_loss": 0.0064720848575234415,
      "step": 390,
      "student_distance_mean": 327.03224182128906,
      "teacher_distance_mean": 1.1625988006591796,
      "weighted_auxiliary_loss": 0.0064720848575234415,
      "weighted_relation_loss": 0.0064720848575234415
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.37418147801683815,
      "flow_matching_loss": 0.10370319187641144,
      "grad_norm": 0.3331562578678131,
      "learning_rate": 2.6600000000000003e-05,
      "loss": 0.1102,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.006509578879922628,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.801465177536011,
      "state_query_norm": 291.6082489013672,
      "state_relation_loss": 0.006509578879922628,
      "step": 400,
      "student_distance_mean": 333.0099304199219,
      "teacher_distance_mean": 1.1496968269348145,
      "weighted_auxiliary_loss": 0.006509578879922628,
      "weighted_relation_loss": 0.006509578879922628
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.38353601496725914,
      "flow_matching_loss": 0.09941594749689102,
      "grad_norm": 0.3693245053291321,
      "learning_rate": 2.7266666666666668e-05,
      "loss": 0.1058,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.006391395488753915,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.862775707244873,
      "state_query_norm": 295.34031677246094,
      "state_relation_loss": 0.006391395488753915,
      "step": 410,
      "student_distance_mean": 336.72654724121094,
      "teacher_distance_mean": 1.1487054705619812,
      "weighted_auxiliary_loss": 0.006391395488753915,
      "weighted_relation_loss": 0.006391395488753915
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.3928905519176801,
      "flow_matching_loss": 0.09932756051421165,
      "grad_norm": 0.5055780410766602,
      "learning_rate": 2.7933333333333332e-05,
      "loss": 0.1057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.006323883822187781,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 5.776380252838135,
      "state_query_norm": 285.68701171875,
      "state_relation_loss": 0.006323883822187781,
      "step": 420,
      "student_distance_mean": 332.4546875,
      "teacher_distance_mean": 1.1462031126022338,
      "weighted_auxiliary_loss": 0.006323883822187781,
      "weighted_relation_loss": 0.006323883822187781
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.402245088868101,
      "flow_matching_loss": 0.10103094577789307,
      "grad_norm": 0.4062526524066925,
      "learning_rate": 2.86e-05,
      "loss": 0.1069,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005903813801705837,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.025854778289795,
      "state_query_norm": 298.7015655517578,
      "state_relation_loss": 0.005903813801705837,
      "step": 430,
      "student_distance_mean": 346.64598388671874,
      "teacher_distance_mean": 1.1556927561759949,
      "weighted_auxiliary_loss": 0.005903813801705837,
      "weighted_relation_loss": 0.005903813801705837
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.411599625818522,
      "flow_matching_loss": 0.10010676309466363,
      "grad_norm": 0.3803205192089081,
      "learning_rate": 2.926666666666667e-05,
      "loss": 0.1062,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.006085875257849693,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.006131362915039,
      "state_query_norm": 298.0793518066406,
      "state_relation_loss": 0.006085875257849693,
      "step": 440,
      "student_distance_mean": 344.91428527832034,
      "teacher_distance_mean": 1.1371147990226746,
      "weighted_auxiliary_loss": 0.006085875257849693,
      "weighted_relation_loss": 0.006085875257849693
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.42095416276894293,
      "flow_matching_loss": 0.10092598795890809,
      "grad_norm": 0.3550468981266022,
      "learning_rate": 2.9933333333333337e-05,
      "loss": 0.1069,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005939097935333848,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.137590217590332,
      "state_query_norm": 304.79259338378904,
      "state_relation_loss": 0.005939097935333848,
      "step": 450,
      "student_distance_mean": 353.50104370117185,
      "teacher_distance_mean": 1.1514466166496278,
      "weighted_auxiliary_loss": 0.005939097935333848,
      "weighted_relation_loss": 0.005939097935333848
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.4303086997193639,
      "flow_matching_loss": 0.09822389483451843,
      "grad_norm": 0.4678654074668884,
      "learning_rate": 3.06e-05,
      "loss": 0.1039,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005718402517959476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.211316823959351,
      "state_query_norm": 305.46324157714844,
      "state_relation_loss": 0.005718402517959476,
      "step": 460,
      "student_distance_mean": 356.25604248046875,
      "teacher_distance_mean": 1.154858934879303,
      "weighted_auxiliary_loss": 0.005718402517959476,
      "weighted_relation_loss": 0.005718402517959476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.43966323666978485,
      "flow_matching_loss": 0.0942557193338871,
      "grad_norm": 0.3847038447856903,
      "learning_rate": 3.126666666666666e-05,
      "loss": 0.1,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00572289451956749,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.2870447635650635,
      "state_query_norm": 311.3675109863281,
      "state_relation_loss": 0.00572289451956749,
      "step": 470,
      "student_distance_mean": 361.7296478271484,
      "teacher_distance_mean": 1.1578816413879394,
      "weighted_auxiliary_loss": 0.00572289451956749,
      "weighted_relation_loss": 0.00572289451956749
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.4490177736202058,
      "flow_matching_loss": 0.09775345027446747,
      "grad_norm": 0.3290293216705322,
      "learning_rate": 3.1933333333333335e-05,
      "loss": 0.1031,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005392692098394036,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.281968021392823,
      "state_query_norm": 312.50294494628906,
      "state_relation_loss": 0.005392692098394036,
      "step": 480,
      "student_distance_mean": 361.49163208007815,
      "teacher_distance_mean": 1.1533711791038512,
      "weighted_auxiliary_loss": 0.005392692098394036,
      "weighted_relation_loss": 0.005392692098394036
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.4583723105706268,
      "flow_matching_loss": 0.09613204970955849,
      "grad_norm": 0.2751481235027313,
      "learning_rate": 3.26e-05,
      "loss": 0.1015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005380218615755439,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.3268531322479244,
      "state_query_norm": 317.6311401367187,
      "state_relation_loss": 0.005380218615755439,
      "step": 490,
      "student_distance_mean": 363.7427124023437,
      "teacher_distance_mean": 1.1540603160858154,
      "weighted_auxiliary_loss": 0.005380218615755439,
      "weighted_relation_loss": 0.005380218615755439
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.4677268475210477,
      "flow_matching_loss": 0.09820147529244423,
      "grad_norm": 0.3896028995513916,
      "learning_rate": 3.326666666666667e-05,
      "loss": 0.1035,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005269604502245784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.472756338119507,
      "state_query_norm": 317.0560638427734,
      "state_relation_loss": 0.005269604502245784,
      "step": 500,
      "student_distance_mean": 372.45896911621094,
      "teacher_distance_mean": 1.1731183767318725,
      "weighted_auxiliary_loss": 0.005269604502245784,
      "weighted_relation_loss": 0.005269604502245784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.47708138447146864,
      "flow_matching_loss": 0.09566998481750488,
      "grad_norm": 0.46462276577949524,
      "learning_rate": 3.3933333333333336e-05,
      "loss": 0.1007,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00506074489094317,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.507755136489868,
      "state_query_norm": 318.82781372070315,
      "state_relation_loss": 0.00506074489094317,
      "step": 510,
      "student_distance_mean": 373.3120178222656,
      "teacher_distance_mean": 1.1629413604736327,
      "weighted_auxiliary_loss": 0.00506074489094317,
      "weighted_relation_loss": 0.00506074489094317
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.4864359214218896,
      "flow_matching_loss": 0.0967143453657627,
      "grad_norm": 0.45278650522232056,
      "learning_rate": 3.46e-05,
      "loss": 0.1021,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005382002936676145,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.470574855804443,
      "state_query_norm": 312.9590179443359,
      "state_relation_loss": 0.005382002936676145,
      "step": 520,
      "student_distance_mean": 371.79808349609374,
      "teacher_distance_mean": 1.1395187258720398,
      "weighted_auxiliary_loss": 0.005382002936676145,
      "weighted_relation_loss": 0.005382002936676145
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.49579045837231056,
      "flow_matching_loss": 0.09268488958477974,
      "grad_norm": 0.33856070041656494,
      "learning_rate": 3.526666666666667e-05,
      "loss": 0.098,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.005317030567675829,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.568705463409424,
      "state_query_norm": 324.5697326660156,
      "state_relation_loss": 0.005317030567675829,
      "step": 530,
      "student_distance_mean": 378.4930816650391,
      "teacher_distance_mean": 1.151533567905426,
      "weighted_auxiliary_loss": 0.005317030567675829,
      "weighted_relation_loss": 0.005317030567675829
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5051449953227315,
      "flow_matching_loss": 0.09781154841184617,
      "grad_norm": 0.4559934437274933,
      "learning_rate": 3.593333333333334e-05,
      "loss": 0.1028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0049972254782915115,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.511228418350219,
      "state_query_norm": 321.9759918212891,
      "state_relation_loss": 0.0049972254782915115,
      "step": 540,
      "student_distance_mean": 375.64873046875,
      "teacher_distance_mean": 1.149820578098297,
      "weighted_auxiliary_loss": 0.0049972254782915115,
      "weighted_relation_loss": 0.0049972254782915115
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5144995322731525,
      "flow_matching_loss": 0.09617534056305885,
      "grad_norm": 0.3240475356578827,
      "learning_rate": 3.66e-05,
      "loss": 0.101,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004812247818335891,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.6093329906463625,
      "state_query_norm": 324.1118469238281,
      "state_relation_loss": 0.004812247818335891,
      "step": 550,
      "student_distance_mean": 379.71815490722656,
      "teacher_distance_mean": 1.1573933482170105,
      "weighted_auxiliary_loss": 0.004812247818335891,
      "weighted_relation_loss": 0.004812247818335891
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5238540692235735,
      "flow_matching_loss": 0.09385082647204399,
      "grad_norm": 0.3761182427406311,
      "learning_rate": 3.726666666666667e-05,
      "loss": 0.0987,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004874040139839053,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.664743232727051,
      "state_query_norm": 329.7113464355469,
      "state_relation_loss": 0.004874040139839053,
      "step": 560,
      "student_distance_mean": 383.3904998779297,
      "teacher_distance_mean": 1.162396204471588,
      "weighted_auxiliary_loss": 0.004874040139839053,
      "weighted_relation_loss": 0.004874040139839053
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5332086061739943,
      "flow_matching_loss": 0.09453307092189789,
      "grad_norm": 0.405642032623291,
      "learning_rate": 3.793333333333334e-05,
      "loss": 0.0994,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004865025589242578,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.631131649017334,
      "state_query_norm": 327.0867584228516,
      "state_relation_loss": 0.004865025589242578,
      "step": 570,
      "student_distance_mean": 381.5397003173828,
      "teacher_distance_mean": 1.1442010641098022,
      "weighted_auxiliary_loss": 0.004865025589242578,
      "weighted_relation_loss": 0.004865025589242578
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5425631431244153,
      "flow_matching_loss": 0.09548690691590309,
      "grad_norm": 0.36122825741767883,
      "learning_rate": 3.86e-05,
      "loss": 0.1,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004508915217593312,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.813356256484985,
      "state_query_norm": 334.63221435546876,
      "state_relation_loss": 0.004508915217593312,
      "step": 580,
      "student_distance_mean": 391.5412841796875,
      "teacher_distance_mean": 1.1540161490440368,
      "weighted_auxiliary_loss": 0.004508915217593312,
      "weighted_relation_loss": 0.004508915217593312
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5519176800748363,
      "flow_matching_loss": 0.0964415803551674,
      "grad_norm": 0.32572224736213684,
      "learning_rate": 3.926666666666667e-05,
      "loss": 0.1011,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004658235795795917,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.767948770523072,
      "state_query_norm": 333.1564880371094,
      "state_relation_loss": 0.004658235795795917,
      "step": 590,
      "student_distance_mean": 389.0386199951172,
      "teacher_distance_mean": 1.147804057598114,
      "weighted_auxiliary_loss": 0.004658235795795917,
      "weighted_relation_loss": 0.004658235795795917
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5612722170252572,
      "flow_matching_loss": 0.09363847598433495,
      "grad_norm": 0.3763466775417328,
      "learning_rate": 3.993333333333333e-05,
      "loss": 0.0979,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004289700696244836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.0475935459136965,
      "state_query_norm": 348.83734741210935,
      "state_relation_loss": 0.004289700696244836,
      "step": 600,
      "student_distance_mean": 406.7082794189453,
      "teacher_distance_mean": 1.170185375213623,
      "weighted_auxiliary_loss": 0.004289700696244836,
      "weighted_relation_loss": 0.004289700696244836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5706267539756782,
      "flow_matching_loss": 0.09428174495697021,
      "grad_norm": 0.31770971417427063,
      "learning_rate": 4.0600000000000004e-05,
      "loss": 0.0989,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004631484020501375,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 6.86405611038208,
      "state_query_norm": 335.6419616699219,
      "state_relation_loss": 0.004631484020501375,
      "step": 610,
      "student_distance_mean": 394.3836395263672,
      "teacher_distance_mean": 1.1376185417175293,
      "weighted_auxiliary_loss": 0.004631484020501375,
      "weighted_relation_loss": 0.004631484020501375
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5799812909260992,
      "flow_matching_loss": 0.09041579216718673,
      "grad_norm": 0.32017648220062256,
      "learning_rate": 4.126666666666667e-05,
      "loss": 0.0949,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004450333397835493,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.116427755355835,
      "state_query_norm": 349.6931640625,
      "state_relation_loss": 0.004450333397835493,
      "step": 620,
      "student_distance_mean": 409.2827545166016,
      "teacher_distance_mean": 1.154456377029419,
      "weighted_auxiliary_loss": 0.004450333397835493,
      "weighted_relation_loss": 0.004450333397835493
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.5893358278765201,
      "flow_matching_loss": 0.09276990294456482,
      "grad_norm": 0.4294535517692566,
      "learning_rate": 4.1933333333333334e-05,
      "loss": 0.0971,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004303237609565258,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.18673186302185,
      "state_query_norm": 354.1222686767578,
      "state_relation_loss": 0.004303237609565258,
      "step": 630,
      "student_distance_mean": 413.7368560791016,
      "teacher_distance_mean": 1.1543882131576537,
      "weighted_auxiliary_loss": 0.004303237609565258,
      "weighted_relation_loss": 0.004303237609565258
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.598690364826941,
      "flow_matching_loss": 0.09419972822070122,
      "grad_norm": 0.3376505970954895,
      "learning_rate": 4.26e-05,
      "loss": 0.0984,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004242202453315258,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.17914719581604,
      "state_query_norm": 348.01707763671874,
      "state_relation_loss": 0.004242202453315258,
      "step": 640,
      "student_distance_mean": 412.22003784179685,
      "teacher_distance_mean": 1.1528164267539978,
      "weighted_auxiliary_loss": 0.004242202453315258,
      "weighted_relation_loss": 0.004242202453315258
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.608044901777362,
      "flow_matching_loss": 0.08879634141921997,
      "grad_norm": 0.3268992006778717,
      "learning_rate": 4.3266666666666664e-05,
      "loss": 0.093,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004191657318733633,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.2037430763244625,
      "state_query_norm": 356.2533782958984,
      "state_relation_loss": 0.004191657318733633,
      "step": 650,
      "student_distance_mean": 415.61394958496095,
      "teacher_distance_mean": 1.1529419660568236,
      "weighted_auxiliary_loss": 0.004191657318733633,
      "weighted_relation_loss": 0.004191657318733633
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.617399438727783,
      "flow_matching_loss": 0.09184950888156891,
      "grad_norm": 0.35693493485450745,
      "learning_rate": 4.3933333333333335e-05,
      "loss": 0.0959,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004015258699655533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.322900199890137,
      "state_query_norm": 356.21246948242185,
      "state_relation_loss": 0.004015258699655533,
      "step": 660,
      "student_distance_mean": 421.0307586669922,
      "teacher_distance_mean": 1.1539456605911256,
      "weighted_auxiliary_loss": 0.004015258699655533,
      "weighted_relation_loss": 0.004015258699655533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6267539756782039,
      "flow_matching_loss": 0.09220405668020248,
      "grad_norm": 0.34807148575782776,
      "learning_rate": 4.46e-05,
      "loss": 0.0964,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004222638974897564,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.3958573818206785,
      "state_query_norm": 354.93853149414065,
      "state_relation_loss": 0.004222638974897564,
      "step": 670,
      "student_distance_mean": 425.37119750976564,
      "teacher_distance_mean": 1.159468674659729,
      "weighted_auxiliary_loss": 0.004222638974897564,
      "weighted_relation_loss": 0.004222638974897564
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6361085126286249,
      "flow_matching_loss": 0.09393062368035317,
      "grad_norm": 0.5769522786140442,
      "learning_rate": 4.526666666666667e-05,
      "loss": 0.0979,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0040111837442964315,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.285066747665406,
      "state_query_norm": 353.0783325195313,
      "state_relation_loss": 0.0040111837442964315,
      "step": 680,
      "student_distance_mean": 419.4762359619141,
      "teacher_distance_mean": 1.1521036505699158,
      "weighted_auxiliary_loss": 0.0040111837442964315,
      "weighted_relation_loss": 0.0040111837442964315
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6454630495790459,
      "flow_matching_loss": 0.09049093127250671,
      "grad_norm": 0.428073912858963,
      "learning_rate": 4.5933333333333336e-05,
      "loss": 0.0944,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003876669728197157,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.317030191421509,
      "state_query_norm": 357.1012664794922,
      "state_relation_loss": 0.003876669728197157,
      "step": 690,
      "student_distance_mean": 419.21552734375,
      "teacher_distance_mean": 1.1337384223937987,
      "weighted_auxiliary_loss": 0.003876669728197157,
      "weighted_relation_loss": 0.003876669728197157
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6548175865294668,
      "flow_matching_loss": 0.08973315954208375,
      "grad_norm": 0.4248066842556,
      "learning_rate": 4.660000000000001e-05,
      "loss": 0.0935,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0038035745499655606,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.4815083026885985,
      "state_query_norm": 359.1538330078125,
      "state_relation_loss": 0.0038035745499655606,
      "step": 700,
      "student_distance_mean": 429.1734161376953,
      "teacher_distance_mean": 1.1549351692199707,
      "weighted_auxiliary_loss": 0.0038035745499655606,
      "weighted_relation_loss": 0.0038035745499655606
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6641721234798877,
      "flow_matching_loss": 0.09200526997447014,
      "grad_norm": 0.3436557352542877,
      "learning_rate": 4.726666666666667e-05,
      "loss": 0.0958,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0038157393923029304,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.712765502929687,
      "state_query_norm": 371.5950561523438,
      "state_relation_loss": 0.0038157393923029304,
      "step": 710,
      "student_distance_mean": 442.8421600341797,
      "teacher_distance_mean": 1.160798680782318,
      "weighted_auxiliary_loss": 0.0038157393923029304,
      "weighted_relation_loss": 0.0038157393923029304
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6735266604303087,
      "flow_matching_loss": 0.0921943537890911,
      "grad_norm": 0.4417186677455902,
      "learning_rate": 4.793333333333334e-05,
      "loss": 0.0962,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.004007930099032819,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.651881551742553,
      "state_query_norm": 373.4845825195313,
      "state_relation_loss": 0.004007930099032819,
      "step": 720,
      "student_distance_mean": 439.6769134521484,
      "teacher_distance_mean": 1.1501890897750855,
      "weighted_auxiliary_loss": 0.004007930099032819,
      "weighted_relation_loss": 0.004007930099032819
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6828811973807296,
      "flow_matching_loss": 0.09078895673155785,
      "grad_norm": 0.383171021938324,
      "learning_rate": 4.86e-05,
      "loss": 0.0947,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00392225831747055,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.674455785751343,
      "state_query_norm": 376.0504669189453,
      "state_relation_loss": 0.00392225831747055,
      "step": 730,
      "student_distance_mean": 441.71002807617185,
      "teacher_distance_mean": 1.1537112593650818,
      "weighted_auxiliary_loss": 0.00392225831747055,
      "weighted_relation_loss": 0.00392225831747055
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.6922357343311506,
      "flow_matching_loss": 0.09232239499688148,
      "grad_norm": 0.3248549699783325,
      "learning_rate": 4.926666666666667e-05,
      "loss": 0.0962,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0038347460329532625,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.73379955291748,
      "state_query_norm": 374.4445495605469,
      "state_relation_loss": 0.0038347460329532625,
      "step": 740,
      "student_distance_mean": 444.88270263671876,
      "teacher_distance_mean": 1.1549984455108642,
      "weighted_auxiliary_loss": 0.0038347460329532625,
      "weighted_relation_loss": 0.0038347460329532625
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7015902712815716,
      "flow_matching_loss": 0.08721776455640792,
      "grad_norm": 0.3720647990703583,
      "learning_rate": 4.993333333333334e-05,
      "loss": 0.091,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003732963162474334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.645658540725708,
      "state_query_norm": 374.0692199707031,
      "state_relation_loss": 0.003732963162474334,
      "step": 750,
      "student_distance_mean": 440.43149719238284,
      "teacher_distance_mean": 1.1386087656021118,
      "weighted_auxiliary_loss": 0.003732963162474334,
      "weighted_relation_loss": 0.003732963162474334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7109448082319925,
      "flow_matching_loss": 0.0870901696383953,
      "grad_norm": 0.34803953766822815,
      "learning_rate": 5.0600000000000003e-05,
      "loss": 0.091,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003937054821290076,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 7.962924003601074,
      "state_query_norm": 388.7931793212891,
      "state_relation_loss": 0.003937054821290076,
      "step": 760,
      "student_distance_mean": 458.13915100097654,
      "teacher_distance_mean": 1.150578761100769,
      "weighted_auxiliary_loss": 0.003937054821290076,
      "weighted_relation_loss": 0.003937054821290076
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7202993451824135,
      "flow_matching_loss": 0.08559707254171371,
      "grad_norm": 0.3584212064743042,
      "learning_rate": 5.1266666666666675e-05,
      "loss": 0.0894,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003838554583489895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.094408750534058,
      "state_query_norm": 385.0616851806641,
      "state_relation_loss": 0.003838554583489895,
      "step": 770,
      "student_distance_mean": 463.6553100585937,
      "teacher_distance_mean": 1.160267949104309,
      "weighted_auxiliary_loss": 0.003838554583489895,
      "weighted_relation_loss": 0.003838554583489895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7296538821328344,
      "flow_matching_loss": 0.08696652501821518,
      "grad_norm": 0.2725257873535156,
      "learning_rate": 5.193333333333333e-05,
      "loss": 0.0906,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0035908238729462027,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.125619554519654,
      "state_query_norm": 393.3303558349609,
      "state_relation_loss": 0.0035908238729462027,
      "step": 780,
      "student_distance_mean": 466.3972534179687,
      "teacher_distance_mean": 1.1607200980186463,
      "weighted_auxiliary_loss": 0.0035908238729462027,
      "weighted_relation_loss": 0.0035908238729462027
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7390084190832554,
      "flow_matching_loss": 0.08626338467001915,
      "grad_norm": 0.22874845564365387,
      "learning_rate": 5.2600000000000005e-05,
      "loss": 0.0898,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00354378349147737,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.078540897369384,
      "state_query_norm": 393.6493347167969,
      "state_relation_loss": 0.00354378349147737,
      "step": 790,
      "student_distance_mean": 466.78350219726565,
      "teacher_distance_mean": 1.1466712474822998,
      "weighted_auxiliary_loss": 0.00354378349147737,
      "weighted_relation_loss": 0.00354378349147737
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7483629560336763,
      "flow_matching_loss": 0.08768970146775246,
      "grad_norm": 0.31346237659454346,
      "learning_rate": 5.326666666666666e-05,
      "loss": 0.0914,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003740439983084798,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.20908203125,
      "state_query_norm": 395.92112426757814,
      "state_relation_loss": 0.003740439983084798,
      "step": 800,
      "student_distance_mean": 472.46824035644534,
      "teacher_distance_mean": 1.151691210269928,
      "weighted_auxiliary_loss": 0.003740439983084798,
      "weighted_relation_loss": 0.003740439983084798
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7577174929840973,
      "flow_matching_loss": 0.08400671854615212,
      "grad_norm": 0.2390723079442978,
      "learning_rate": 5.3933333333333334e-05,
      "loss": 0.0875,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003521554032340646,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.270304822921753,
      "state_query_norm": 404.4766021728516,
      "state_relation_loss": 0.003521554032340646,
      "step": 810,
      "student_distance_mean": 478.1989410400391,
      "teacher_distance_mean": 1.1463620066642761,
      "weighted_auxiliary_loss": 0.003521554032340646,
      "weighted_relation_loss": 0.003521554032340646
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7670720299345183,
      "flow_matching_loss": 0.08778889626264572,
      "grad_norm": 0.24652402102947235,
      "learning_rate": 5.4600000000000006e-05,
      "loss": 0.0913,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003528173197992146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.31774549484253,
      "state_query_norm": 396.8185272216797,
      "state_relation_loss": 0.003528173197992146,
      "step": 820,
      "student_distance_mean": 476.75389709472654,
      "teacher_distance_mean": 1.1586206674575805,
      "weighted_auxiliary_loss": 0.003528173197992146,
      "weighted_relation_loss": 0.003528173197992146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7764265668849392,
      "flow_matching_loss": 0.0890668086707592,
      "grad_norm": 0.2716830372810364,
      "learning_rate": 5.5266666666666664e-05,
      "loss": 0.0926,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0035085358656942846,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.244839429855347,
      "state_query_norm": 396.81768798828125,
      "state_relation_loss": 0.0035085358656942846,
      "step": 830,
      "student_distance_mean": 475.7045104980469,
      "teacher_distance_mean": 1.1521310091018677,
      "weighted_auxiliary_loss": 0.0035085358656942846,
      "weighted_relation_loss": 0.0035085358656942846
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7857811038353602,
      "flow_matching_loss": 0.08361260369420051,
      "grad_norm": 0.27004027366638184,
      "learning_rate": 5.5933333333333335e-05,
      "loss": 0.087,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0034002944361418487,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.062776231765747,
      "state_query_norm": 394.4241912841797,
      "state_relation_loss": 0.0034002944361418487,
      "step": 840,
      "student_distance_mean": 465.8226806640625,
      "teacher_distance_mean": 1.1458372116088866,
      "weighted_auxiliary_loss": 0.0034002944361418487,
      "weighted_relation_loss": 0.0034002944361418487
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.7951356407857811,
      "flow_matching_loss": 0.08670368790626526,
      "grad_norm": 0.21587876975536346,
      "learning_rate": 5.66e-05,
      "loss": 0.0901,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00335076421033591,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.399567604064941,
      "state_query_norm": 400.4198028564453,
      "state_relation_loss": 0.00335076421033591,
      "step": 850,
      "student_distance_mean": 482.11144104003904,
      "teacher_distance_mean": 1.1605466246604919,
      "weighted_auxiliary_loss": 0.00335076421033591,
      "weighted_relation_loss": 0.00335076421033591
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.804490177736202,
      "flow_matching_loss": 0.08493279293179512,
      "grad_norm": 0.3561612069606781,
      "learning_rate": 5.726666666666667e-05,
      "loss": 0.0882,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003290463099256158,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.46015510559082,
      "state_query_norm": 407.6709686279297,
      "state_relation_loss": 0.003290463099256158,
      "step": 860,
      "student_distance_mean": 485.27170715332034,
      "teacher_distance_mean": 1.1442485332489014,
      "weighted_auxiliary_loss": 0.003290463099256158,
      "weighted_relation_loss": 0.003290463099256158
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.813844714686623,
      "flow_matching_loss": 0.09056695774197579,
      "grad_norm": 0.3674977719783783,
      "learning_rate": 5.7933333333333337e-05,
      "loss": 0.0938,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0032732578460127116,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.480609607696532,
      "state_query_norm": 406.60328063964846,
      "state_relation_loss": 0.0032732578460127116,
      "step": 870,
      "student_distance_mean": 487.40794372558594,
      "teacher_distance_mean": 1.1449895262718202,
      "weighted_auxiliary_loss": 0.0032732578460127116,
      "weighted_relation_loss": 0.0032732578460127116
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.823199251637044,
      "flow_matching_loss": 0.08422692939639091,
      "grad_norm": 0.26504504680633545,
      "learning_rate": 5.86e-05,
      "loss": 0.0876,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0033354156417772173,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.581858158111572,
      "state_query_norm": 411.6161285400391,
      "state_relation_loss": 0.0033354156417772173,
      "step": 880,
      "student_distance_mean": 493.6275177001953,
      "teacher_distance_mean": 1.1550048828125,
      "weighted_auxiliary_loss": 0.0033354156417772173,
      "weighted_relation_loss": 0.0033354156417772173
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8325537885874649,
      "flow_matching_loss": 0.08609007596969605,
      "grad_norm": 0.2770549952983856,
      "learning_rate": 5.926666666666667e-05,
      "loss": 0.0894,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0032759944442659616,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.627408885955811,
      "state_query_norm": 410.46693725585936,
      "state_relation_loss": 0.0032759944442659616,
      "step": 890,
      "student_distance_mean": 495.1732543945312,
      "teacher_distance_mean": 1.1483906030654907,
      "weighted_auxiliary_loss": 0.0032759944442659616,
      "weighted_relation_loss": 0.0032759944442659616
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8419083255378859,
      "flow_matching_loss": 0.0862847961485386,
      "grad_norm": 0.29617220163345337,
      "learning_rate": 5.9933333333333345e-05,
      "loss": 0.0896,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0032957901246845724,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.654057121276855,
      "state_query_norm": 413.8448455810547,
      "state_relation_loss": 0.0032957901246845724,
      "step": 900,
      "student_distance_mean": 497.16480712890626,
      "teacher_distance_mean": 1.1561429142951964,
      "weighted_auxiliary_loss": 0.0032957901246845724,
      "weighted_relation_loss": 0.0032957901246845724
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8512628624883068,
      "flow_matching_loss": 0.08478893786668777,
      "grad_norm": 0.3209947943687439,
      "learning_rate": 6.06e-05,
      "loss": 0.088,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003253478114493191,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.778784084320069,
      "state_query_norm": 422.0820770263672,
      "state_relation_loss": 0.003253478114493191,
      "step": 910,
      "student_distance_mean": 505.56592712402346,
      "teacher_distance_mean": 1.1564711809158326,
      "weighted_auxiliary_loss": 0.003253478114493191,
      "weighted_relation_loss": 0.003253478114493191
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8606173994387278,
      "flow_matching_loss": 0.08393731266260147,
      "grad_norm": 0.2740541994571686,
      "learning_rate": 6.126666666666667e-05,
      "loss": 0.0872,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003249171352945268,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.63735237121582,
      "state_query_norm": 413.14272155761716,
      "state_relation_loss": 0.003249171352945268,
      "step": 920,
      "student_distance_mean": 495.90497436523435,
      "teacher_distance_mean": 1.1389930248260498,
      "weighted_auxiliary_loss": 0.003249171352945268,
      "weighted_relation_loss": 0.003249171352945268
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8699719363891487,
      "flow_matching_loss": 0.08641351014375687,
      "grad_norm": 0.3016621470451355,
      "learning_rate": 6.193333333333333e-05,
      "loss": 0.0897,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003286628099158406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.63754415512085,
      "state_query_norm": 419.6200439453125,
      "state_relation_loss": 0.003286628099158406,
      "step": 930,
      "student_distance_mean": 497.53240356445315,
      "teacher_distance_mean": 1.145218849182129,
      "weighted_auxiliary_loss": 0.003286628099158406,
      "weighted_relation_loss": 0.003286628099158406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8793264733395697,
      "flow_matching_loss": 0.08558017313480377,
      "grad_norm": 0.3370536267757416,
      "learning_rate": 6.26e-05,
      "loss": 0.0887,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0030761512462049723,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.909135913848877,
      "state_query_norm": 424.41429138183594,
      "state_relation_loss": 0.0030761512462049723,
      "step": 940,
      "student_distance_mean": 511.87499084472654,
      "teacher_distance_mean": 1.148470687866211,
      "weighted_auxiliary_loss": 0.0030761512462049723,
      "weighted_relation_loss": 0.0030761512462049723
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8886810102899907,
      "flow_matching_loss": 0.08586962521076202,
      "grad_norm": 0.2639080286026001,
      "learning_rate": 6.326666666666667e-05,
      "loss": 0.089,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0031405562767758965,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 8.951270866394044,
      "state_query_norm": 427.23412170410154,
      "state_relation_loss": 0.0031405562767758965,
      "step": 950,
      "student_distance_mean": 514.4122741699218,
      "teacher_distance_mean": 1.1434989094734191,
      "weighted_auxiliary_loss": 0.0031405562767758965,
      "weighted_relation_loss": 0.0031405562767758965
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.8980355472404116,
      "flow_matching_loss": 0.08360518589615822,
      "grad_norm": 0.28401297330856323,
      "learning_rate": 6.393333333333333e-05,
      "loss": 0.0867,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0030739766778424384,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.01827564239502,
      "state_query_norm": 428.5092315673828,
      "state_relation_loss": 0.0030739766778424384,
      "step": 960,
      "student_distance_mean": 517.4803344726563,
      "teacher_distance_mean": 1.1556237101554871,
      "weighted_auxiliary_loss": 0.0030739766778424384,
      "weighted_relation_loss": 0.0030739766778424384
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9073900841908326,
      "flow_matching_loss": 0.08427013158798217,
      "grad_norm": 0.31301942467689514,
      "learning_rate": 6.460000000000001e-05,
      "loss": 0.0874,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0030835901852697134,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.07306137084961,
      "state_query_norm": 436.5423156738281,
      "state_relation_loss": 0.0030835901852697134,
      "step": 970,
      "student_distance_mean": 522.361849975586,
      "teacher_distance_mean": 1.1729799389839173,
      "weighted_auxiliary_loss": 0.0030835901852697134,
      "weighted_relation_loss": 0.0030835901852697134
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9167446211412535,
      "flow_matching_loss": 0.08458714187145233,
      "grad_norm": 0.4083026647567749,
      "learning_rate": 6.526666666666666e-05,
      "loss": 0.0877,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0031143719563260673,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.125057411193847,
      "state_query_norm": 437.40008850097655,
      "state_relation_loss": 0.0031143719563260673,
      "step": 980,
      "student_distance_mean": 523.41650390625,
      "teacher_distance_mean": 1.1647521734237671,
      "weighted_auxiliary_loss": 0.0031143719563260673,
      "weighted_relation_loss": 0.0031143719563260673
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9260991580916744,
      "flow_matching_loss": 0.08605562746524811,
      "grad_norm": 0.2353024035692215,
      "learning_rate": 6.593333333333334e-05,
      "loss": 0.089,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0029744884930551054,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.10974760055542,
      "state_query_norm": 441.8790985107422,
      "state_relation_loss": 0.0029744884930551054,
      "step": 990,
      "student_distance_mean": 522.6790008544922,
      "teacher_distance_mean": 1.1457409262657166,
      "weighted_auxiliary_loss": 0.0029744884930551054,
      "weighted_relation_loss": 0.0029744884930551054
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9354536950420954,
      "flow_matching_loss": 0.08759394064545631,
      "grad_norm": 0.31530800461769104,
      "learning_rate": 6.66e-05,
      "loss": 0.0907,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003110022679902613,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.2250470161438,
      "state_query_norm": 439.5970489501953,
      "state_relation_loss": 0.003110022679902613,
      "step": 1000,
      "student_distance_mean": 529.6276916503906,
      "teacher_distance_mean": 1.155034077167511,
      "weighted_auxiliary_loss": 0.003110022679902613,
      "weighted_relation_loss": 0.003110022679902613
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9448082319925164,
      "flow_matching_loss": 0.08522051200270653,
      "grad_norm": 0.3785005509853363,
      "learning_rate": 6.726666666666667e-05,
      "loss": 0.0883,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003040390065871179,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.139633274078369,
      "state_query_norm": 434.1042938232422,
      "state_relation_loss": 0.003040390065871179,
      "step": 1010,
      "student_distance_mean": 523.6177337646484,
      "teacher_distance_mean": 1.142064106464386,
      "weighted_auxiliary_loss": 0.003040390065871179,
      "weighted_relation_loss": 0.003040390065871179
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9541627689429373,
      "flow_matching_loss": 0.08356651738286018,
      "grad_norm": 0.32783210277557373,
      "learning_rate": 6.793333333333334e-05,
      "loss": 0.0868,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0032531765289604666,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.371899509429932,
      "state_query_norm": 443.9478240966797,
      "state_relation_loss": 0.0032531765289604666,
      "step": 1020,
      "student_distance_mean": 536.9468017578125,
      "teacher_distance_mean": 1.1711750745773315,
      "weighted_auxiliary_loss": 0.0032531765289604666,
      "weighted_relation_loss": 0.0032531765289604666
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9635173058933583,
      "flow_matching_loss": 0.08587965220212937,
      "grad_norm": 0.3441413342952728,
      "learning_rate": 6.860000000000001e-05,
      "loss": 0.0891,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0032672979170456527,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.273048210144044,
      "state_query_norm": 439.2505218505859,
      "state_relation_loss": 0.0032672979170456527,
      "step": 1030,
      "student_distance_mean": 534.314697265625,
      "teacher_distance_mean": 1.159760308265686,
      "weighted_auxiliary_loss": 0.0032672979170456527,
      "weighted_relation_loss": 0.0032672979170456527
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9728718428437793,
      "flow_matching_loss": 0.08481471762061119,
      "grad_norm": 0.31570225954055786,
      "learning_rate": 6.926666666666667e-05,
      "loss": 0.088,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.003204711712896824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.544803047180176,
      "state_query_norm": 454.800927734375,
      "state_relation_loss": 0.003204711712896824,
      "step": 1040,
      "student_distance_mean": 547.999365234375,
      "teacher_distance_mean": 1.1588602304458617,
      "weighted_auxiliary_loss": 0.003204711712896824,
      "weighted_relation_loss": 0.003204711712896824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9822263797942001,
      "flow_matching_loss": 0.08341924399137497,
      "grad_norm": 0.24193088710308075,
      "learning_rate": 6.993333333333334e-05,
      "loss": 0.0863,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0028868028661236168,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.60545244216919,
      "state_query_norm": 457.32896423339844,
      "state_relation_loss": 0.0028868028661236168,
      "step": 1050,
      "student_distance_mean": 550.8728942871094,
      "teacher_distance_mean": 1.1754607558250427,
      "weighted_auxiliary_loss": 0.0028868028661236168,
      "weighted_relation_loss": 0.0028868028661236168
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 0.9915809167446211,
      "flow_matching_loss": 0.0835106298327446,
      "grad_norm": 0.35493025183677673,
      "learning_rate": 7.06e-05,
      "loss": 0.0865,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0029499415075406434,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.453960609436034,
      "state_query_norm": 446.55406494140624,
      "state_relation_loss": 0.0029499415075406434,
      "step": 1060,
      "student_distance_mean": 541.6322662353516,
      "teacher_distance_mean": 1.146171462535858,
      "weighted_auxiliary_loss": 0.0029499415075406434,
      "weighted_relation_loss": 0.0029499415075406434
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.000935453695042,
      "flow_matching_loss": 0.08097771853208542,
      "grad_norm": 0.2616291046142578,
      "learning_rate": 7.126666666666667e-05,
      "loss": 0.0837,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.002709265286102891,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 9.445436573028564,
      "state_query_norm": 450.8688171386719,
      "state_relation_loss": 0.002709265286102891,
      "step": 1070,
      "student_distance_mean": 540.7023132324218,
      "teacher_distance_mean": 1.1471204519271851,
      "weighted_auxiliary_loss": 0.002709265286102891,
      "weighted_relation_loss": 0.002709265286102891
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.010289990645463,
      "flow_matching_loss": 0.0839642271399498,
      "grad_norm": 0.3087005615234375,
      "learning_rate": 7.193333333333334e-05,
      "loss": 0.0868,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002811162150464952,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.709772682189941,
      "state_query_norm": 458.60753479003904,
      "state_relation_loss": 0.002811162150464952,
      "step": 1080,
      "student_distance_mean": 556.8316650390625,
      "teacher_distance_mean": 1.1668587446212768,
      "weighted_auxiliary_loss": 0.002811162150464952,
      "weighted_relation_loss": 0.002811162150464952
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.019644527595884,
      "flow_matching_loss": 0.08224052786827088,
      "grad_norm": 0.2546474039554596,
      "learning_rate": 7.26e-05,
      "loss": 0.0851,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002875513443723321,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.60481767654419,
      "state_query_norm": 453.9708526611328,
      "state_relation_loss": 0.002875513443723321,
      "step": 1090,
      "student_distance_mean": 551.0307739257812,
      "teacher_distance_mean": 1.155555784702301,
      "weighted_auxiliary_loss": 0.002875513443723321,
      "weighted_relation_loss": 0.002875513443723321
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.028999064546305,
      "flow_matching_loss": 0.08645965829491616,
      "grad_norm": 0.32125937938690186,
      "learning_rate": 7.326666666666667e-05,
      "loss": 0.0895,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0030200196197256446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.601666736602784,
      "state_query_norm": 447.60899353027344,
      "state_relation_loss": 0.0030200196197256446,
      "step": 1100,
      "student_distance_mean": 549.0656005859375,
      "teacher_distance_mean": 1.1447080969810486,
      "weighted_auxiliary_loss": 0.0030200196197256446,
      "weighted_relation_loss": 0.0030200196197256446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.0383536014967258,
      "flow_matching_loss": 0.08321336358785629,
      "grad_norm": 0.3063764274120331,
      "learning_rate": 7.393333333333333e-05,
      "loss": 0.086,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0027913110796362163,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.626701354980469,
      "state_query_norm": 457.57240600585936,
      "state_relation_loss": 0.0027913110796362163,
      "step": 1110,
      "student_distance_mean": 553.8706909179688,
      "teacher_distance_mean": 1.145009982585907,
      "weighted_auxiliary_loss": 0.0027913110796362163,
      "weighted_relation_loss": 0.0027913110796362163
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.047708138447147,
      "flow_matching_loss": 0.08224016726016999,
      "grad_norm": 0.19827330112457275,
      "learning_rate": 7.46e-05,
      "loss": 0.0851,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002843470126390457,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.721747493743896,
      "state_query_norm": 460.98236389160155,
      "state_relation_loss": 0.002843470126390457,
      "step": 1120,
      "student_distance_mean": 556.3695678710938,
      "teacher_distance_mean": 1.145697784423828,
      "weighted_auxiliary_loss": 0.002843470126390457,
      "weighted_relation_loss": 0.002843470126390457
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.0570626753975678,
      "flow_matching_loss": 0.0810593992471695,
      "grad_norm": 0.309683233499527,
      "learning_rate": 7.526666666666668e-05,
      "loss": 0.0838,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0027633332647383213,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.613244533538818,
      "state_query_norm": 454.62738342285155,
      "state_relation_loss": 0.0027633332647383213,
      "step": 1130,
      "student_distance_mean": 550.8255187988282,
      "teacher_distance_mean": 1.1384383916854859,
      "weighted_auxiliary_loss": 0.0027633332647383213,
      "weighted_relation_loss": 0.0027633332647383213
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.0664172123479887,
      "flow_matching_loss": 0.08281072527170182,
      "grad_norm": 0.2100631147623062,
      "learning_rate": 7.593333333333334e-05,
      "loss": 0.0855,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0027145362459123135,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.782097434997558,
      "state_query_norm": 464.82215881347656,
      "state_relation_loss": 0.0027145362459123135,
      "step": 1140,
      "student_distance_mean": 560.4823608398438,
      "teacher_distance_mean": 1.1476758003234864,
      "weighted_auxiliary_loss": 0.0027145362459123135,
      "weighted_relation_loss": 0.0027145362459123135
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.0757717492984098,
      "flow_matching_loss": 0.08200741112232209,
      "grad_norm": 0.2946639657020569,
      "learning_rate": 7.66e-05,
      "loss": 0.0848,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0027850523591041567,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.911156558990479,
      "state_query_norm": 466.21059265136716,
      "state_relation_loss": 0.0027850523591041567,
      "step": 1150,
      "student_distance_mean": 568.0750122070312,
      "teacher_distance_mean": 1.1552786350250244,
      "weighted_auxiliary_loss": 0.0027850523591041567,
      "weighted_relation_loss": 0.0027850523591041567
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.0851262862488307,
      "flow_matching_loss": 0.08329515233635902,
      "grad_norm": 0.3351810574531555,
      "learning_rate": 7.726666666666667e-05,
      "loss": 0.086,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0027271662373095753,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.09463529586792,
      "state_query_norm": 472.3872894287109,
      "state_relation_loss": 0.0027271662373095753,
      "step": 1160,
      "student_distance_mean": 577.7437622070313,
      "teacher_distance_mean": 1.1427403807640075,
      "weighted_auxiliary_loss": 0.0027271662373095753,
      "weighted_relation_loss": 0.0027271662373095753
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.0944808231992516,
      "flow_matching_loss": 0.08328018561005593,
      "grad_norm": 0.3400173783302307,
      "learning_rate": 7.793333333333333e-05,
      "loss": 0.0859,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0025935641489923,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.269233989715577,
      "state_query_norm": 487.3688659667969,
      "state_relation_loss": 0.0025935641489923,
      "step": 1170,
      "student_distance_mean": 588.0437072753906,
      "teacher_distance_mean": 1.1540670990943909,
      "weighted_auxiliary_loss": 0.0025935641489923,
      "weighted_relation_loss": 0.0025935641489923
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1038353601496727,
      "flow_matching_loss": 0.08174395933747292,
      "grad_norm": 0.2901643216609955,
      "learning_rate": 7.860000000000001e-05,
      "loss": 0.0844,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0026170045835897326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.223327159881592,
      "state_query_norm": 477.7791809082031,
      "state_relation_loss": 0.0026170045835897326,
      "step": 1180,
      "student_distance_mean": 584.5100830078125,
      "teacher_distance_mean": 1.1654307126998902,
      "weighted_auxiliary_loss": 0.0026170045835897326,
      "weighted_relation_loss": 0.0026170045835897326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1131898971000935,
      "flow_matching_loss": 0.08503565043210984,
      "grad_norm": 0.27708473801612854,
      "learning_rate": 7.926666666666666e-05,
      "loss": 0.0877,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0026999316178262234,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.21167974472046,
      "state_query_norm": 479.7337982177734,
      "state_relation_loss": 0.0026999316178262234,
      "step": 1190,
      "student_distance_mean": 584.9914672851562,
      "teacher_distance_mean": 1.1447321057319642,
      "weighted_auxiliary_loss": 0.0026999316178262234,
      "weighted_relation_loss": 0.0026999316178262234
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1225444340505144,
      "flow_matching_loss": 0.08308178335428237,
      "grad_norm": 0.1752166599035263,
      "learning_rate": 7.993333333333334e-05,
      "loss": 0.0857,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0026543170446529985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.514822101593017,
      "state_query_norm": 494.4262176513672,
      "state_relation_loss": 0.0026543170446529985,
      "step": 1200,
      "student_distance_mean": 604.0842346191406,
      "teacher_distance_mean": 1.1642672538757324,
      "weighted_auxiliary_loss": 0.0026543170446529985,
      "weighted_relation_loss": 0.0026543170446529985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1318989710009355,
      "flow_matching_loss": 0.08471024185419082,
      "grad_norm": 0.3720015287399292,
      "learning_rate": 8.060000000000001e-05,
      "loss": 0.0872,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0025046603521332146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 9.96740083694458,
      "state_query_norm": 468.4440643310547,
      "state_relation_loss": 0.0025046603521332146,
      "step": 1210,
      "student_distance_mean": 570.8582885742187,
      "teacher_distance_mean": 1.1424437642097474,
      "weighted_auxiliary_loss": 0.0025046603521332146,
      "weighted_relation_loss": 0.0025046603521332146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1412535079513564,
      "flow_matching_loss": 0.0802742637693882,
      "grad_norm": 0.22013375163078308,
      "learning_rate": 8.126666666666667e-05,
      "loss": 0.0829,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00260711710434407,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.087808227539062,
      "state_query_norm": 477.4094207763672,
      "state_relation_loss": 0.00260711710434407,
      "step": 1220,
      "student_distance_mean": 579.0872375488282,
      "teacher_distance_mean": 1.1408269166946412,
      "weighted_auxiliary_loss": 0.00260711710434407,
      "weighted_relation_loss": 0.00260711710434407
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1506080449017775,
      "flow_matching_loss": 0.0807521864771843,
      "grad_norm": 0.2318732738494873,
      "learning_rate": 8.193333333333334e-05,
      "loss": 0.0833,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0025771592743694784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.09104299545288,
      "state_query_norm": 478.3634368896484,
      "state_relation_loss": 0.0025771592743694784,
      "step": 1230,
      "student_distance_mean": 578.1020568847656,
      "teacher_distance_mean": 1.149958348274231,
      "weighted_auxiliary_loss": 0.0025771592743694784,
      "weighted_relation_loss": 0.0025771592743694784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1599625818521984,
      "flow_matching_loss": 0.08276034370064736,
      "grad_norm": 0.37311264872550964,
      "learning_rate": 8.26e-05,
      "loss": 0.0853,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002517397119663656,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.216317844390868,
      "state_query_norm": 480.6615447998047,
      "state_relation_loss": 0.002517397119663656,
      "step": 1240,
      "student_distance_mean": 585.587451171875,
      "teacher_distance_mean": 1.1556093573570252,
      "weighted_auxiliary_loss": 0.002517397119663656,
      "weighted_relation_loss": 0.002517397119663656
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1693171188026192,
      "flow_matching_loss": 0.08443114459514618,
      "grad_norm": 0.2775542140007019,
      "learning_rate": 8.326666666666667e-05,
      "loss": 0.087,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002570974477566779,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.259323596954346,
      "state_query_norm": 483.12990417480466,
      "state_relation_loss": 0.002570974477566779,
      "step": 1250,
      "student_distance_mean": 588.1174255371094,
      "teacher_distance_mean": 1.1402223944664,
      "weighted_auxiliary_loss": 0.002570974477566779,
      "weighted_relation_loss": 0.002570974477566779
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1786716557530403,
      "flow_matching_loss": 0.08478128388524056,
      "grad_norm": 0.28829729557037354,
      "learning_rate": 8.393333333333335e-05,
      "loss": 0.0874,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0026233874028548597,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.372071743011475,
      "state_query_norm": 486.64471130371095,
      "state_relation_loss": 0.0026233874028548597,
      "step": 1260,
      "student_distance_mean": 594.7375,
      "teacher_distance_mean": 1.1481234669685363,
      "weighted_auxiliary_loss": 0.0026233874028548597,
      "weighted_relation_loss": 0.0026233874028548597
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.1880261927034612,
      "flow_matching_loss": 0.08075293824076653,
      "grad_norm": 0.29464319348335266,
      "learning_rate": 8.46e-05,
      "loss": 0.0834,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002630456956103444,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.646674156188965,
      "state_query_norm": 504.7584197998047,
      "state_relation_loss": 0.002630456956103444,
      "step": 1270,
      "student_distance_mean": 610.5739196777344,
      "teacher_distance_mean": 1.1571772933006286,
      "weighted_auxiliary_loss": 0.002630456956103444,
      "weighted_relation_loss": 0.002630456956103444
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.197380729653882,
      "flow_matching_loss": 0.08423047885298729,
      "grad_norm": 0.4273032546043396,
      "learning_rate": 8.526666666666667e-05,
      "loss": 0.0868,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0025227148784324528,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.683392524719238,
      "state_query_norm": 501.435009765625,
      "state_relation_loss": 0.0025227148784324528,
      "step": 1280,
      "student_distance_mean": 611.8894470214843,
      "teacher_distance_mean": 1.1525228142738342,
      "weighted_auxiliary_loss": 0.0025227148784324528,
      "weighted_relation_loss": 0.0025227148784324528
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.2067352666043032,
      "flow_matching_loss": 0.08403920754790306,
      "grad_norm": 0.2496773898601532,
      "learning_rate": 8.593333333333333e-05,
      "loss": 0.0866,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002602634625509381,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.757257175445556,
      "state_query_norm": 498.7322113037109,
      "state_relation_loss": 0.002602634625509381,
      "step": 1290,
      "student_distance_mean": 614.40078125,
      "teacher_distance_mean": 1.1657190680503846,
      "weighted_auxiliary_loss": 0.002602634625509381,
      "weighted_relation_loss": 0.002602634625509381
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.216089803554724,
      "flow_matching_loss": 0.08087452203035354,
      "grad_norm": 0.22538411617279053,
      "learning_rate": 8.66e-05,
      "loss": 0.0833,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002446908038109541,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.926614570617676,
      "state_query_norm": 506.6776885986328,
      "state_relation_loss": 0.002446908038109541,
      "step": 1300,
      "student_distance_mean": 624.1839111328125,
      "teacher_distance_mean": 1.1680084586143493,
      "weighted_auxiliary_loss": 0.002446908038109541,
      "weighted_relation_loss": 0.002446908038109541
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.225444340505145,
      "flow_matching_loss": 0.08323376774787902,
      "grad_norm": 0.1752716600894928,
      "learning_rate": 8.726666666666667e-05,
      "loss": 0.0858,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00252909529954195,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.561883640289306,
      "state_query_norm": 498.37037658691406,
      "state_relation_loss": 0.00252909529954195,
      "step": 1310,
      "student_distance_mean": 604.7279541015625,
      "teacher_distance_mean": 1.1451438307762145,
      "weighted_auxiliary_loss": 0.00252909529954195,
      "weighted_relation_loss": 0.00252909529954195
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.234798877455566,
      "flow_matching_loss": 0.08115599453449249,
      "grad_norm": 0.2700255811214447,
      "learning_rate": 8.793333333333333e-05,
      "loss": 0.0835,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002386771887540817,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.690519523620605,
      "state_query_norm": 499.6940887451172,
      "state_relation_loss": 0.002386771887540817,
      "step": 1320,
      "student_distance_mean": 612.6345520019531,
      "teacher_distance_mean": 1.1619114756584168,
      "weighted_auxiliary_loss": 0.002386771887540817,
      "weighted_relation_loss": 0.002386771887540817
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.244153414405987,
      "flow_matching_loss": 0.0807044856250286,
      "grad_norm": 0.28498637676239014,
      "learning_rate": 8.86e-05,
      "loss": 0.0832,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002453887485899031,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.657210636138917,
      "state_query_norm": 499.9916595458984,
      "state_relation_loss": 0.002453887485899031,
      "step": 1330,
      "student_distance_mean": 611.0804382324219,
      "teacher_distance_mean": 1.1544017672538758,
      "weighted_auxiliary_loss": 0.002453887485899031,
      "weighted_relation_loss": 0.002453887485899031
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.2535079513564078,
      "flow_matching_loss": 0.08120447099208832,
      "grad_norm": 0.22871047258377075,
      "learning_rate": 8.926666666666668e-05,
      "loss": 0.0836,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002418512408621609,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.640367984771729,
      "state_query_norm": 496.97218627929686,
      "state_relation_loss": 0.002418512408621609,
      "step": 1340,
      "student_distance_mean": 608.4369750976563,
      "teacher_distance_mean": 1.144587504863739,
      "weighted_auxiliary_loss": 0.002418512408621609,
      "weighted_relation_loss": 0.002418512408621609
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.262862488306829,
      "flow_matching_loss": 0.07985543683171273,
      "grad_norm": 0.3572023808956146,
      "learning_rate": 8.993333333333334e-05,
      "loss": 0.0822,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002338220877572894,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.823288059234619,
      "state_query_norm": 505.5861541748047,
      "state_relation_loss": 0.002338220877572894,
      "step": 1350,
      "student_distance_mean": 620.9577087402344,
      "teacher_distance_mean": 1.150049352645874,
      "weighted_auxiliary_loss": 0.002338220877572894,
      "weighted_relation_loss": 0.002338220877572894
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.2722170252572498,
      "flow_matching_loss": 0.07871490120887756,
      "grad_norm": 0.28486040234565735,
      "learning_rate": 9.06e-05,
      "loss": 0.0811,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0023792160907760264,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 10.954993534088135,
      "state_query_norm": 514.4484985351562,
      "state_relation_loss": 0.0023792160907760264,
      "step": 1360,
      "student_distance_mean": 625.8045959472656,
      "teacher_distance_mean": 1.1559850573539734,
      "weighted_auxiliary_loss": 0.0023792160907760264,
      "weighted_relation_loss": 0.0023792160907760264
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.2815715622076707,
      "flow_matching_loss": 0.07660514116287231,
      "grad_norm": 0.2622716724872589,
      "learning_rate": 9.126666666666667e-05,
      "loss": 0.079,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0023537838831543924,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.046446704864502,
      "state_query_norm": 511.6098114013672,
      "state_relation_loss": 0.0023537838831543924,
      "step": 1370,
      "student_distance_mean": 629.0564025878906,
      "teacher_distance_mean": 1.1473851084709168,
      "weighted_auxiliary_loss": 0.0023537838831543924,
      "weighted_relation_loss": 0.0023537838831543924
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.2909260991580918,
      "flow_matching_loss": 0.08223071470856666,
      "grad_norm": 0.2832670509815216,
      "learning_rate": 9.193333333333334e-05,
      "loss": 0.0846,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0023785881232470272,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.19566478729248,
      "state_query_norm": 526.1260162353516,
      "state_relation_loss": 0.0023785881232470272,
      "step": 1380,
      "student_distance_mean": 640.3471618652344,
      "teacher_distance_mean": 1.1615562438964844,
      "weighted_auxiliary_loss": 0.0023785881232470272,
      "weighted_relation_loss": 0.0023785881232470272
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3002806361085126,
      "flow_matching_loss": 0.08253677561879158,
      "grad_norm": 0.2521263062953949,
      "learning_rate": 9.260000000000001e-05,
      "loss": 0.085,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002423637663014233,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.226747226715087,
      "state_query_norm": 525.0012145996094,
      "state_relation_loss": 0.002423637663014233,
      "step": 1390,
      "student_distance_mean": 644.5232543945312,
      "teacher_distance_mean": 1.1701722383499145,
      "weighted_auxiliary_loss": 0.002423637663014233,
      "weighted_relation_loss": 0.002423637663014233
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3096351730589335,
      "flow_matching_loss": 0.08455777913331985,
      "grad_norm": 0.33885326981544495,
      "learning_rate": 9.326666666666667e-05,
      "loss": 0.087,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00242352036293596,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.273293018341064,
      "state_query_norm": 521.5875885009766,
      "state_relation_loss": 0.00242352036293596,
      "step": 1400,
      "student_distance_mean": 643.4583190917969,
      "teacher_distance_mean": 1.143809962272644,
      "weighted_auxiliary_loss": 0.00242352036293596,
      "weighted_relation_loss": 0.00242352036293596
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3189897100093546,
      "flow_matching_loss": 0.07935408875346184,
      "grad_norm": 0.26643210649490356,
      "learning_rate": 9.393333333333334e-05,
      "loss": 0.0816,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022436139639467,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.56492919921875,
      "state_query_norm": 542.7016723632812,
      "state_relation_loss": 0.0022436139639467,
      "step": 1410,
      "student_distance_mean": 663.6206237792969,
      "teacher_distance_mean": 1.1668159723281861,
      "weighted_auxiliary_loss": 0.0022436139639467,
      "weighted_relation_loss": 0.0022436139639467
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3283442469597755,
      "flow_matching_loss": 0.08104558140039445,
      "grad_norm": 0.36364924907684326,
      "learning_rate": 9.46e-05,
      "loss": 0.0833,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022364139556884767,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.63763656616211,
      "state_query_norm": 541.2416015625,
      "state_relation_loss": 0.0022364139556884767,
      "step": 1420,
      "student_distance_mean": 666.2486328125,
      "teacher_distance_mean": 1.1707327127456666,
      "weighted_auxiliary_loss": 0.0022364139556884767,
      "weighted_relation_loss": 0.0022364139556884767
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3376987839101964,
      "flow_matching_loss": 0.07860133349895478,
      "grad_norm": 0.3850550353527069,
      "learning_rate": 9.526666666666667e-05,
      "loss": 0.0808,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022160806925967335,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.34944019317627,
      "state_query_norm": 524.468295288086,
      "state_relation_loss": 0.0022160806925967335,
      "step": 1430,
      "student_distance_mean": 646.7030822753907,
      "teacher_distance_mean": 1.144745111465454,
      "weighted_auxiliary_loss": 0.0022160806925967335,
      "weighted_relation_loss": 0.0022160806925967335
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3470533208606175,
      "flow_matching_loss": 0.08176040649414062,
      "grad_norm": 0.2931636869907379,
      "learning_rate": 9.593333333333334e-05,
      "loss": 0.084,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002249034680426121,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.401490688323975,
      "state_query_norm": 530.8308288574219,
      "state_relation_loss": 0.002249034680426121,
      "step": 1440,
      "student_distance_mean": 651.86728515625,
      "teacher_distance_mean": 1.1505134463310243,
      "weighted_auxiliary_loss": 0.002249034680426121,
      "weighted_relation_loss": 0.002249034680426121
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3564078578110383,
      "flow_matching_loss": 0.08061363324522972,
      "grad_norm": 0.27290648221969604,
      "learning_rate": 9.66e-05,
      "loss": 0.0829,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0023095279932022093,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.568145751953125,
      "state_query_norm": 538.1887939453125,
      "state_relation_loss": 0.0023095279932022093,
      "step": 1450,
      "student_distance_mean": 662.2705871582032,
      "teacher_distance_mean": 1.1591590642929077,
      "weighted_auxiliary_loss": 0.0023095279932022093,
      "weighted_relation_loss": 0.0023095279932022093
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3657623947614592,
      "flow_matching_loss": 0.080695141851902,
      "grad_norm": 0.2159356325864792,
      "learning_rate": 9.726666666666667e-05,
      "loss": 0.0829,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022188361966982485,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.56308193206787,
      "state_query_norm": 532.9557067871094,
      "state_relation_loss": 0.0022188361966982485,
      "step": 1460,
      "student_distance_mean": 660.4668823242188,
      "teacher_distance_mean": 1.1599047780036926,
      "weighted_auxiliary_loss": 0.0022188361966982485,
      "weighted_relation_loss": 0.0022188361966982485
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3751169317118803,
      "flow_matching_loss": 0.07677315473556519,
      "grad_norm": 0.23781423270702362,
      "learning_rate": 9.793333333333333e-05,
      "loss": 0.079,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00226524070603773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.649554538726807,
      "state_query_norm": 539.9195404052734,
      "state_relation_loss": 0.00226524070603773,
      "step": 1470,
      "student_distance_mean": 668.1720031738281,
      "teacher_distance_mean": 1.1656119227409363,
      "weighted_auxiliary_loss": 0.00226524070603773,
      "weighted_relation_loss": 0.00226524070603773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.3844714686623012,
      "flow_matching_loss": 0.08022217229008674,
      "grad_norm": 0.30400368571281433,
      "learning_rate": 9.86e-05,
      "loss": 0.0825,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002257693000137806,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.5325758934021,
      "state_query_norm": 533.4408874511719,
      "state_relation_loss": 0.002257693000137806,
      "step": 1480,
      "student_distance_mean": 659.1077270507812,
      "teacher_distance_mean": 1.155841886997223,
      "weighted_auxiliary_loss": 0.002257693000137806,
      "weighted_relation_loss": 0.002257693000137806
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.393826005612722,
      "flow_matching_loss": 0.08176343366503716,
      "grad_norm": 0.32895636558532715,
      "learning_rate": 9.926666666666668e-05,
      "loss": 0.0841,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002289640926755965,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.58389720916748,
      "state_query_norm": 537.6516906738282,
      "state_relation_loss": 0.002289640926755965,
      "step": 1490,
      "student_distance_mean": 662.944775390625,
      "teacher_distance_mean": 1.1431068658828736,
      "weighted_auxiliary_loss": 0.002289640926755965,
      "weighted_relation_loss": 0.002289640926755965
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4031805425631432,
      "flow_matching_loss": 0.07899172827601433,
      "grad_norm": 0.23546093702316284,
      "learning_rate": 9.993333333333334e-05,
      "loss": 0.0814,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002364831930026412,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.732314205169677,
      "state_query_norm": 547.3171691894531,
      "state_relation_loss": 0.002364831930026412,
      "step": 1500,
      "student_distance_mean": 670.5461059570313,
      "teacher_distance_mean": 1.1554579854011535,
      "weighted_auxiliary_loss": 0.002364831930026412,
      "weighted_relation_loss": 0.002364831930026412
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.412535079513564,
      "flow_matching_loss": 0.08127077519893647,
      "grad_norm": 0.24550005793571472,
      "learning_rate": 9.999997539434007e-05,
      "loss": 0.0835,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002248318656347692,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.699183368682862,
      "state_query_norm": 541.5641235351562,
      "state_relation_loss": 0.002248318656347692,
      "step": 1510,
      "student_distance_mean": 668.3925231933594,
      "teacher_distance_mean": 1.1523568153381347,
      "weighted_auxiliary_loss": 0.002248318656347692,
      "weighted_relation_loss": 0.002248318656347692
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.421889616463985,
      "flow_matching_loss": 0.08252245038747788,
      "grad_norm": 0.27995797991752625,
      "learning_rate": 9.999989033776898e-05,
      "loss": 0.0848,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022337281727232037,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.62009687423706,
      "state_query_norm": 534.5384887695312,
      "state_relation_loss": 0.0022337281727232037,
      "step": 1520,
      "student_distance_mean": 663.8138122558594,
      "teacher_distance_mean": 1.1593607425689698,
      "weighted_auxiliary_loss": 0.0022337281727232037,
      "weighted_relation_loss": 0.0022337281727232037
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.431244153414406,
      "flow_matching_loss": 0.0795646220445633,
      "grad_norm": 0.3145272731781006,
      "learning_rate": 9.999974452661641e-05,
      "loss": 0.0818,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022027330240234734,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.892511463165283,
      "state_query_norm": 549.9366821289062,
      "state_relation_loss": 0.0022027330240234734,
      "step": 1530,
      "student_distance_mean": 680.0905639648438,
      "teacher_distance_mean": 1.1654606342315674,
      "weighted_auxiliary_loss": 0.0022027330240234734,
      "weighted_relation_loss": 0.0022027330240234734
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.440598690364827,
      "flow_matching_loss": 0.07828773185610771,
      "grad_norm": 0.29971402883529663,
      "learning_rate": 9.999953796105959e-05,
      "loss": 0.0805,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0022457811748608945,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.837709999084472,
      "state_query_norm": 540.7881103515625,
      "state_relation_loss": 0.0022457811748608945,
      "step": 1540,
      "student_distance_mean": 672.6812072753906,
      "teacher_distance_mean": 1.1563432455062865,
      "weighted_auxiliary_loss": 0.0022457811748608945,
      "weighted_relation_loss": 0.0022457811748608945
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4499532273152478,
      "flow_matching_loss": 0.07724201828241348,
      "grad_norm": 0.20807760953903198,
      "learning_rate": 9.999927064134949e-05,
      "loss": 0.0795,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002229971089400351,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.78499174118042,
      "state_query_norm": 539.218310546875,
      "state_relation_loss": 0.002229971089400351,
      "step": 1550,
      "student_distance_mean": 670.7113891601563,
      "teacher_distance_mean": 1.1394943356513978,
      "weighted_auxiliary_loss": 0.002229971089400351,
      "weighted_relation_loss": 0.002229971089400351
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4593077642656689,
      "flow_matching_loss": 0.0805919848382473,
      "grad_norm": 0.28402775526046753,
      "learning_rate": 9.999894256781095e-05,
      "loss": 0.0827,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00212213397026062,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.954427528381348,
      "state_query_norm": 549.1665100097656,
      "state_relation_loss": 0.00212213397026062,
      "step": 1560,
      "student_distance_mean": 682.1801025390625,
      "teacher_distance_mean": 1.161312198638916,
      "weighted_auxiliary_loss": 0.00212213397026062,
      "weighted_relation_loss": 0.00212213397026062
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4686623012160898,
      "flow_matching_loss": 0.07781435549259186,
      "grad_norm": 0.2727651000022888,
      "learning_rate": 9.99985537408426e-05,
      "loss": 0.0799,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0021287115290760995,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.013211345672607,
      "state_query_norm": 555.5052917480468,
      "state_relation_loss": 0.0021287115290760995,
      "step": 1570,
      "student_distance_mean": 684.7549194335937,
      "teacher_distance_mean": 1.1601760506629943,
      "weighted_auxiliary_loss": 0.0021287115290760995,
      "weighted_relation_loss": 0.0021287115290760995
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4780168381665106,
      "flow_matching_loss": 0.07745357751846313,
      "grad_norm": 0.2695757746696472,
      "learning_rate": 9.999810416091688e-05,
      "loss": 0.0795,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0020633397274650634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.112625312805175,
      "state_query_norm": 560.9433044433594,
      "state_relation_loss": 0.0020633397274650634,
      "step": 1580,
      "student_distance_mean": 692.9669311523437,
      "teacher_distance_mean": 1.1488985896110535,
      "weighted_auxiliary_loss": 0.0020633397274650634,
      "weighted_relation_loss": 0.0020633397274650634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4873713751169317,
      "flow_matching_loss": 0.07837220355868339,
      "grad_norm": 0.23582147061824799,
      "learning_rate": 9.99975938285801e-05,
      "loss": 0.0804,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002065959561150521,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.994457817077636,
      "state_query_norm": 550.9630981445313,
      "state_relation_loss": 0.002065959561150521,
      "step": 1590,
      "student_distance_mean": 684.8271545410156,
      "teacher_distance_mean": 1.1569951534271241,
      "weighted_auxiliary_loss": 0.002065959561150521,
      "weighted_relation_loss": 0.002065959561150521
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.4967259120673526,
      "flow_matching_loss": 0.08010712936520577,
      "grad_norm": 0.20454561710357666,
      "learning_rate": 9.999702274445236e-05,
      "loss": 0.0822,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0021322476444765926,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 11.95124111175537,
      "state_query_norm": 547.1314270019532,
      "state_relation_loss": 0.0021322476444765926,
      "step": 1600,
      "student_distance_mean": 678.3248046875,
      "teacher_distance_mean": 1.137067186832428,
      "weighted_auxiliary_loss": 0.0021322476444765926,
      "weighted_relation_loss": 0.0021322476444765926
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5060804490177735,
      "flow_matching_loss": 0.07705634012818337,
      "grad_norm": 0.21527151763439178,
      "learning_rate": 9.999639090922756e-05,
      "loss": 0.0791,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002052609226666391,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.258218002319335,
      "state_query_norm": 562.0056945800782,
      "state_relation_loss": 0.002052609226666391,
      "step": 1610,
      "student_distance_mean": 699.5926940917968,
      "teacher_distance_mean": 1.147902822494507,
      "weighted_auxiliary_loss": 0.002052609226666391,
      "weighted_relation_loss": 0.002052609226666391
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5154349859681946,
      "flow_matching_loss": 0.07392210513353348,
      "grad_norm": 0.30437734723091125,
      "learning_rate": 9.999569832367346e-05,
      "loss": 0.0759,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0020125136245042084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.470887851715087,
      "state_query_norm": 579.859375,
      "state_relation_loss": 0.0020125136245042084,
      "step": 1620,
      "student_distance_mean": 712.9078125,
      "teacher_distance_mean": 1.1555162072181702,
      "weighted_auxiliary_loss": 0.0020125136245042084,
      "weighted_relation_loss": 0.0020125136245042084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5247895229186157,
      "flow_matching_loss": 0.07991473004221916,
      "grad_norm": 0.20899251103401184,
      "learning_rate": 9.999494498863162e-05,
      "loss": 0.082,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002052618295419961,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.127881050109863,
      "state_query_norm": 559.6109436035156,
      "state_relation_loss": 0.002052618295419961,
      "step": 1630,
      "student_distance_mean": 692.653955078125,
      "teacher_distance_mean": 1.130436325073242,
      "weighted_auxiliary_loss": 0.002052618295419961,
      "weighted_relation_loss": 0.002052618295419961
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5341440598690363,
      "flow_matching_loss": 0.07706919834017753,
      "grad_norm": 0.32717210054397583,
      "learning_rate": 9.99941309050174e-05,
      "loss": 0.0791,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002031221357174218,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.118582439422607,
      "state_query_norm": 560.7157043457031,
      "state_relation_loss": 0.002031221357174218,
      "step": 1640,
      "student_distance_mean": 692.643212890625,
      "teacher_distance_mean": 1.1345916152000428,
      "weighted_auxiliary_loss": 0.002031221357174218,
      "weighted_relation_loss": 0.002031221357174218
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5434985968194574,
      "flow_matching_loss": 0.0797169640660286,
      "grad_norm": 0.27529555559158325,
      "learning_rate": 9.999325607381999e-05,
      "loss": 0.0818,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002107193041592836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.507458972930909,
      "state_query_norm": 577.3951171875,
      "state_relation_loss": 0.002107193041592836,
      "step": 1650,
      "student_distance_mean": 713.5609375,
      "teacher_distance_mean": 1.1556610703468322,
      "weighted_auxiliary_loss": 0.002107193041592836,
      "weighted_relation_loss": 0.002107193041592836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5528531337698785,
      "flow_matching_loss": 0.08060325980186463,
      "grad_norm": 0.23515701293945312,
      "learning_rate": 9.999232049610238e-05,
      "loss": 0.0828,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002170978381764144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.520276927947998,
      "state_query_norm": 571.3786499023438,
      "state_relation_loss": 0.002170978381764144,
      "step": 1660,
      "student_distance_mean": 714.4024536132813,
      "teacher_distance_mean": 1.1557549715042115,
      "weighted_auxiliary_loss": 0.002170978381764144,
      "weighted_relation_loss": 0.002170978381764144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5622076707202992,
      "flow_matching_loss": 0.07672021836042404,
      "grad_norm": 0.2737828493118286,
      "learning_rate": 9.999132417300142e-05,
      "loss": 0.0788,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002116403146646917,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.414067649841309,
      "state_query_norm": 572.8826782226563,
      "state_relation_loss": 0.002116403146646917,
      "step": 1670,
      "student_distance_mean": 708.2482788085938,
      "teacher_distance_mean": 1.148399245738983,
      "weighted_auxiliary_loss": 0.002116403146646917,
      "weighted_relation_loss": 0.002116403146646917
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5715622076707203,
      "flow_matching_loss": 0.0774283267557621,
      "grad_norm": 0.2490275800228119,
      "learning_rate": 9.99902671057277e-05,
      "loss": 0.0795,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002051581535488367,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.522239780426025,
      "state_query_norm": 572.551953125,
      "state_relation_loss": 0.002051581535488367,
      "step": 1680,
      "student_distance_mean": 715.0232666015625,
      "teacher_distance_mean": 1.1602219820022583,
      "weighted_auxiliary_loss": 0.002051581535488367,
      "weighted_relation_loss": 0.002051581535488367
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5809167446211414,
      "flow_matching_loss": 0.07935866191983224,
      "grad_norm": 0.33887192606925964,
      "learning_rate": 9.998914929556569e-05,
      "loss": 0.0814,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0020342711941339076,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.634984397888184,
      "state_query_norm": 579.7035705566407,
      "state_relation_loss": 0.0020342711941339076,
      "step": 1690,
      "student_distance_mean": 720.669384765625,
      "teacher_distance_mean": 1.1521984934806824,
      "weighted_auxiliary_loss": 0.0020342711941339076,
      "weighted_relation_loss": 0.0020342711941339076
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.590271281571562,
      "flow_matching_loss": 0.07963919118046761,
      "grad_norm": 0.24728156626224518,
      "learning_rate": 9.998797074387361e-05,
      "loss": 0.0817,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0020866064820438623,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.690489292144775,
      "state_query_norm": 583.3160278320313,
      "state_relation_loss": 0.0020866064820438623,
      "step": 1700,
      "student_distance_mean": 724.151123046875,
      "teacher_distance_mean": 1.1495561718940734,
      "weighted_auxiliary_loss": 0.0020866064820438623,
      "weighted_relation_loss": 0.0020866064820438623
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.5996258185219832,
      "flow_matching_loss": 0.0790396086871624,
      "grad_norm": 0.2767429053783417,
      "learning_rate": 9.99867314520835e-05,
      "loss": 0.0812,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002130486175883561,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.648562812805176,
      "state_query_norm": 580.1040893554688,
      "state_relation_loss": 0.002130486175883561,
      "step": 1710,
      "student_distance_mean": 720.8067321777344,
      "teacher_distance_mean": 1.1650105714797974,
      "weighted_auxiliary_loss": 0.002130486175883561,
      "weighted_relation_loss": 0.002130486175883561
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.6089803554724043,
      "flow_matching_loss": 0.07567426860332489,
      "grad_norm": 0.2604134976863861,
      "learning_rate": 9.998543142170126e-05,
      "loss": 0.0777,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001993534434586763,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.559395599365235,
      "state_query_norm": 581.8912048339844,
      "state_relation_loss": 0.001993534434586763,
      "step": 1720,
      "student_distance_mean": 719.1592163085937,
      "teacher_distance_mean": 1.1412727952003479,
      "weighted_auxiliary_loss": 0.001993534434586763,
      "weighted_relation_loss": 0.001993534434586763
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.618334892422825,
      "flow_matching_loss": 0.07999316677451133,
      "grad_norm": 0.2964591681957245,
      "learning_rate": 9.99840706543065e-05,
      "loss": 0.0819,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019537028623744846,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.73389778137207,
      "state_query_norm": 587.2112976074219,
      "state_relation_loss": 0.0019537028623744846,
      "step": 1730,
      "student_distance_mean": 726.5922607421875,
      "teacher_distance_mean": 1.1447595477104187,
      "weighted_auxiliary_loss": 0.0019537028623744846,
      "weighted_relation_loss": 0.0019537028623744846
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.627689429373246,
      "flow_matching_loss": 0.07504526302218437,
      "grad_norm": 0.3154752254486084,
      "learning_rate": 9.998264915155274e-05,
      "loss": 0.077,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019271098892204464,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.829965114593506,
      "state_query_norm": 598.405078125,
      "state_relation_loss": 0.0019271098892204464,
      "step": 1740,
      "student_distance_mean": 730.9726867675781,
      "teacher_distance_mean": 1.1371015191078186,
      "weighted_auxiliary_loss": 0.0019271098892204464,
      "weighted_relation_loss": 0.0019271098892204464
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.637043966323667,
      "flow_matching_loss": 0.07841207906603813,
      "grad_norm": 0.2759057581424713,
      "learning_rate": 9.998116691516718e-05,
      "loss": 0.0804,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.002009621646720916,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.242793941497803,
      "state_query_norm": 611.8898681640625,
      "state_relation_loss": 0.002009621646720916,
      "step": 1750,
      "student_distance_mean": 757.4458190917969,
      "teacher_distance_mean": 1.157972764968872,
      "weighted_auxiliary_loss": 0.002009621646720916,
      "weighted_relation_loss": 0.002009621646720916
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.646398503274088,
      "flow_matching_loss": 0.07919657379388809,
      "grad_norm": 0.23307956755161285,
      "learning_rate": 9.997962394695091e-05,
      "loss": 0.0811,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019522063434123994,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.255432605743408,
      "state_query_norm": 603.2278686523438,
      "state_relation_loss": 0.0019522063434123994,
      "step": 1760,
      "student_distance_mean": 754.6495971679688,
      "teacher_distance_mean": 1.1577337503433227,
      "weighted_auxiliary_loss": 0.0019522063434123994,
      "weighted_relation_loss": 0.0019522063434123994
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.6557530402245089,
      "flow_matching_loss": 0.07290752008557319,
      "grad_norm": 0.2658368647098541,
      "learning_rate": 9.997802024877875e-05,
      "loss": 0.0748,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019055966869927943,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.121117401123048,
      "state_query_norm": 597.1864562988281,
      "state_relation_loss": 0.0019055966869927943,
      "step": 1770,
      "student_distance_mean": 747.8144836425781,
      "teacher_distance_mean": 1.166729474067688,
      "weighted_auxiliary_loss": 0.0019055966869927943,
      "weighted_relation_loss": 0.0019055966869927943
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.66510757717493,
      "flow_matching_loss": 0.07536658495664597,
      "grad_norm": 0.1892297863960266,
      "learning_rate": 9.99763558225994e-05,
      "loss": 0.0773,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019543166039511562,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.102494239807129,
      "state_query_norm": 597.2389221191406,
      "state_relation_loss": 0.0019543166039511562,
      "step": 1780,
      "student_distance_mean": 747.5089050292969,
      "teacher_distance_mean": 1.1620692968368531,
      "weighted_auxiliary_loss": 0.0019543166039511562,
      "weighted_relation_loss": 0.0019543166039511562
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.6744621141253508,
      "flow_matching_loss": 0.07562625780701637,
      "grad_norm": 0.28864675760269165,
      "learning_rate": 9.997463067043526e-05,
      "loss": 0.0775,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019086108193732798,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.85399169921875,
      "state_query_norm": 583.2835021972656,
      "state_relation_loss": 0.0019086108193732798,
      "step": 1790,
      "student_distance_mean": 732.8851013183594,
      "teacher_distance_mean": 1.1560049653053284,
      "weighted_auxiliary_loss": 0.0019086108193732798,
      "weighted_relation_loss": 0.0019086108193732798
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.6838166510757717,
      "flow_matching_loss": 0.07715069651603698,
      "grad_norm": 0.23410208523273468,
      "learning_rate": 9.997284479438253e-05,
      "loss": 0.079,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018639955436810852,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.907737731933594,
      "state_query_norm": 587.0585571289063,
      "state_relation_loss": 0.0018639955436810852,
      "step": 1800,
      "student_distance_mean": 734.7563171386719,
      "teacher_distance_mean": 1.1410681843757629,
      "weighted_auxiliary_loss": 0.0018639955436810852,
      "weighted_relation_loss": 0.0018639955436810852
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.6931711880261928,
      "flow_matching_loss": 0.07502134963870048,
      "grad_norm": 0.1824643909931183,
      "learning_rate": 9.997099819661127e-05,
      "loss": 0.0768,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018196841818280518,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.324853134155273,
      "state_query_norm": 600.421240234375,
      "state_relation_loss": 0.0018196841818280518,
      "step": 1810,
      "student_distance_mean": 757.2390075683594,
      "teacher_distance_mean": 1.1663371324539185,
      "weighted_auxiliary_loss": 0.0018196841818280518,
      "weighted_relation_loss": 0.0018196841818280518
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7025257249766137,
      "flow_matching_loss": 0.07763098627328872,
      "grad_norm": 0.32398685812950134,
      "learning_rate": 9.996909087936524e-05,
      "loss": 0.0795,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018350357539020478,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.219750308990479,
      "state_query_norm": 601.0189514160156,
      "state_relation_loss": 0.0018350357539020478,
      "step": 1820,
      "student_distance_mean": 750.9692687988281,
      "teacher_distance_mean": 1.147036349773407,
      "weighted_auxiliary_loss": 0.0018350357539020478,
      "weighted_relation_loss": 0.0018350357539020478
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7118802619270346,
      "flow_matching_loss": 0.07909542396664619,
      "grad_norm": 0.21097253262996674,
      "learning_rate": 9.996712284496201e-05,
      "loss": 0.0809,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018276833812706172,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.39379425048828,
      "state_query_norm": 611.2896118164062,
      "state_relation_loss": 0.0018276833812706172,
      "step": 1830,
      "student_distance_mean": 763.1278137207031,
      "teacher_distance_mean": 1.1551605343818665,
      "weighted_auxiliary_loss": 0.0018276833812706172,
      "weighted_relation_loss": 0.0018276833812706172
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7212347988774557,
      "flow_matching_loss": 0.07601451277732849,
      "grad_norm": 0.19119423627853394,
      "learning_rate": 9.996509409579293e-05,
      "loss": 0.0779,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019078150624409318,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.56267032623291,
      "state_query_norm": 616.0325805664063,
      "state_relation_loss": 0.0019078150624409318,
      "step": 1840,
      "student_distance_mean": 770.8647094726563,
      "teacher_distance_mean": 1.156461977958679,
      "weighted_auxiliary_loss": 0.0019078150624409318,
      "weighted_relation_loss": 0.0019078150624409318
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7305893358278766,
      "flow_matching_loss": 0.07711068615317344,
      "grad_norm": 0.21183381974697113,
      "learning_rate": 9.996300463432312e-05,
      "loss": 0.079,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001843694702256471,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.06559944152832,
      "state_query_norm": 602.4593933105468,
      "state_relation_loss": 0.001843694702256471,
      "step": 1850,
      "student_distance_mean": 747.7722900390625,
      "teacher_distance_mean": 1.1509055018424987,
      "weighted_auxiliary_loss": 0.001843694702256471,
      "weighted_relation_loss": 0.001843694702256471
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7399438727782974,
      "flow_matching_loss": 0.07205189019441605,
      "grad_norm": 0.1751106232404709,
      "learning_rate": 9.996085446309148e-05,
      "loss": 0.0739,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001889584120362997,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 12.981561660766602,
      "state_query_norm": 591.9867858886719,
      "state_relation_loss": 0.001889584120362997,
      "step": 1860,
      "student_distance_mean": 739.2992126464844,
      "teacher_distance_mean": 1.147829830646515,
      "weighted_auxiliary_loss": 0.001889584120362997,
      "weighted_relation_loss": 0.001889584120362997
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7492984097287185,
      "flow_matching_loss": 0.07644594237208366,
      "grad_norm": 0.1716964691877365,
      "learning_rate": 9.995864358471066e-05,
      "loss": 0.0784,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0019383563310839235,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.151847839355469,
      "state_query_norm": 597.2633056640625,
      "state_relation_loss": 0.0019383563310839235,
      "step": 1870,
      "student_distance_mean": 749.3913208007813,
      "teacher_distance_mean": 1.1367177724838258,
      "weighted_auxiliary_loss": 0.0019383563310839235,
      "weighted_relation_loss": 0.0019383563310839235
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7586529466791394,
      "flow_matching_loss": 0.0760963313281536,
      "grad_norm": 0.21011169254779816,
      "learning_rate": 9.99563720018671e-05,
      "loss": 0.078,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018780162557959557,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.604470825195312,
      "state_query_norm": 620.8941955566406,
      "state_relation_loss": 0.0018780162557959557,
      "step": 1880,
      "student_distance_mean": 774.8585632324218,
      "teacher_distance_mean": 1.1623044610023499,
      "weighted_auxiliary_loss": 0.0018780162557959557,
      "weighted_relation_loss": 0.0018780162557959557
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7680074836295603,
      "flow_matching_loss": 0.0782677747309208,
      "grad_norm": 0.22496891021728516,
      "learning_rate": 9.995403971732098e-05,
      "loss": 0.0801,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001828056713566184,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.6186674118042,
      "state_query_norm": 619.4435668945313,
      "state_relation_loss": 0.001828056713566184,
      "step": 1890,
      "student_distance_mean": 777.2219543457031,
      "teacher_distance_mean": 1.1534602999687196,
      "weighted_auxiliary_loss": 0.001828056713566184,
      "weighted_relation_loss": 0.001828056713566184
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7773620205799814,
      "flow_matching_loss": 0.07313693836331367,
      "grad_norm": 0.30813199281692505,
      "learning_rate": 9.995164673390625e-05,
      "loss": 0.075,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018286168342456222,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.681418895721436,
      "state_query_norm": 624.3576904296875,
      "state_relation_loss": 0.0018286168342456222,
      "step": 1900,
      "student_distance_mean": 777.2791442871094,
      "teacher_distance_mean": 1.154014766216278,
      "weighted_auxiliary_loss": 0.0018286168342456222,
      "weighted_relation_loss": 0.0018286168342456222
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7867165575304023,
      "flow_matching_loss": 0.07439156845211983,
      "grad_norm": 0.2740948498249054,
      "learning_rate": 9.994919305453059e-05,
      "loss": 0.0762,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018105880706571042,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.724625301361083,
      "state_query_norm": 620.11591796875,
      "state_relation_loss": 0.0018105880706571042,
      "step": 1910,
      "student_distance_mean": 779.8014404296875,
      "teacher_distance_mean": 1.1448947191238403,
      "weighted_auxiliary_loss": 0.0018105880706571042,
      "weighted_relation_loss": 0.0018105880706571042
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.7960710944808231,
      "flow_matching_loss": 0.0759547583758831,
      "grad_norm": 0.17486189305782318,
      "learning_rate": 9.994667868217548e-05,
      "loss": 0.0778,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018369866418652236,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.655205631256104,
      "state_query_norm": 623.8283569335938,
      "state_relation_loss": 0.0018369866418652236,
      "step": 1920,
      "student_distance_mean": 778.5023803710938,
      "teacher_distance_mean": 1.1507623672485352,
      "weighted_auxiliary_loss": 0.0018369866418652236,
      "weighted_relation_loss": 0.0018369866418652236
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8054256314312442,
      "flow_matching_loss": 0.0765747532248497,
      "grad_norm": 0.13994553685188293,
      "learning_rate": 9.99441036198961e-05,
      "loss": 0.0784,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00182295796694234,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.545649528503418,
      "state_query_norm": 613.529345703125,
      "state_relation_loss": 0.00182295796694234,
      "step": 1930,
      "student_distance_mean": 769.5511291503906,
      "teacher_distance_mean": 1.1464751362800598,
      "weighted_auxiliary_loss": 0.00182295796694234,
      "weighted_relation_loss": 0.00182295796694234
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8147801683816651,
      "flow_matching_loss": 0.07328036800026894,
      "grad_norm": 0.16248945891857147,
      "learning_rate": 9.99414678708214e-05,
      "loss": 0.0751,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018631587852723896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.787501239776612,
      "state_query_norm": 625.8742736816406,
      "state_relation_loss": 0.0018631587852723896,
      "step": 1940,
      "student_distance_mean": 786.6523559570312,
      "teacher_distance_mean": 1.1631656646728517,
      "weighted_auxiliary_loss": 0.0018631587852723896,
      "weighted_relation_loss": 0.0018631587852723896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.824134705332086,
      "flow_matching_loss": 0.0739493265748024,
      "grad_norm": 0.3006563186645508,
      "learning_rate": 9.993877143815407e-05,
      "loss": 0.0758,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001838396571110934,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.909656143188476,
      "state_query_norm": 626.4418395996094,
      "state_relation_loss": 0.001838396571110934,
      "step": 1950,
      "student_distance_mean": 788.6162963867188,
      "teacher_distance_mean": 1.1555147886276245,
      "weighted_auxiliary_loss": 0.001838396571110934,
      "weighted_relation_loss": 0.001838396571110934
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.833489242282507,
      "flow_matching_loss": 0.07395718544721604,
      "grad_norm": 0.34227341413497925,
      "learning_rate": 9.993601432517053e-05,
      "loss": 0.0758,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017933124909177423,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.011356735229493,
      "state_query_norm": 634.6170654296875,
      "state_relation_loss": 0.0017933124909177423,
      "step": 1960,
      "student_distance_mean": 796.4142211914062,
      "teacher_distance_mean": 1.1641674757003784,
      "weighted_auxiliary_loss": 0.0017933124909177423,
      "weighted_relation_loss": 0.0017933124909177423
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.842843779232928,
      "flow_matching_loss": 0.07428508549928665,
      "grad_norm": 0.2865685224533081,
      "learning_rate": 9.993319653522091e-05,
      "loss": 0.0761,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017974664689972997,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.743811416625977,
      "state_query_norm": 619.7040283203125,
      "state_relation_loss": 0.0017974664689972997,
      "step": 1970,
      "student_distance_mean": 781.6456970214844,
      "teacher_distance_mean": 1.139522922039032,
      "weighted_auxiliary_loss": 0.0017974664689972997,
      "weighted_relation_loss": 0.0017974664689972997
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8521983161833488,
      "flow_matching_loss": 0.07503869235515595,
      "grad_norm": 0.287648469209671,
      "learning_rate": 9.993031807172911e-05,
      "loss": 0.0769,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018154399003833532,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.881764316558838,
      "state_query_norm": 628.2779602050781,
      "state_relation_loss": 0.0018154399003833532,
      "step": 1980,
      "student_distance_mean": 789.8670593261719,
      "teacher_distance_mean": 1.167342746257782,
      "weighted_auxiliary_loss": 0.0018154399003833532,
      "weighted_relation_loss": 0.0018154399003833532
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.86155285313377,
      "flow_matching_loss": 0.07463488653302193,
      "grad_norm": 0.2418784201145172,
      "learning_rate": 9.992737893819273e-05,
      "loss": 0.0764,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017817501327954232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.285348415374756,
      "state_query_norm": 644.8524169921875,
      "state_relation_loss": 0.0017817501327954232,
      "step": 1990,
      "student_distance_mean": 813.4955871582031,
      "teacher_distance_mean": 1.158066749572754,
      "weighted_auxiliary_loss": 0.0017817501327954232,
      "weighted_relation_loss": 0.0017817501327954232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8709073900841908,
      "flow_matching_loss": 0.07544285655021668,
      "grad_norm": 0.20557093620300293,
      "learning_rate": 9.992437913818312e-05,
      "loss": 0.0773,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0018762001651339232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.057274436950683,
      "state_query_norm": 635.9193054199219,
      "state_relation_loss": 0.0018762001651339232,
      "step": 2000,
      "student_distance_mean": 799.1137939453125,
      "teacher_distance_mean": 1.142703914642334,
      "weighted_auxiliary_loss": 0.0018762001651339232,
      "weighted_relation_loss": 0.0018762001651339232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8802619270346117,
      "flow_matching_loss": 0.07253347858786582,
      "grad_norm": 0.17635680735111237,
      "learning_rate": 9.992131867534526e-05,
      "loss": 0.0743,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017662487807683647,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.173650550842286,
      "state_query_norm": 638.6480651855469,
      "state_relation_loss": 0.0017662487807683647,
      "step": 2010,
      "student_distance_mean": 805.9291320800781,
      "teacher_distance_mean": 1.150196921825409,
      "weighted_auxiliary_loss": 0.0017662487807683647,
      "weighted_relation_loss": 0.0017662487807683647
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8896164639850328,
      "flow_matching_loss": 0.0748682476580143,
      "grad_norm": 0.2115204781293869,
      "learning_rate": 9.991819755339796e-05,
      "loss": 0.0765,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016672416124492884,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 13.956680297851562,
      "state_query_norm": 631.4730285644531,
      "state_relation_loss": 0.0016672416124492884,
      "step": 2020,
      "student_distance_mean": 794.6367431640625,
      "teacher_distance_mean": 1.1443756341934204,
      "weighted_auxiliary_loss": 0.0016672416124492884,
      "weighted_relation_loss": 0.0016672416124492884
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.8989710009354537,
      "flow_matching_loss": 0.07436864823102951,
      "grad_norm": 0.1850007027387619,
      "learning_rate": 9.991501577613365e-05,
      "loss": 0.0761,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00169675083598122,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.218309307098389,
      "state_query_norm": 634.9357604980469,
      "state_relation_loss": 0.00169675083598122,
      "step": 2030,
      "student_distance_mean": 807.7757446289063,
      "teacher_distance_mean": 1.1569784164428711,
      "weighted_auxiliary_loss": 0.00169675083598122,
      "weighted_relation_loss": 0.00169675083598122
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9083255378858746,
      "flow_matching_loss": 0.0732714667916298,
      "grad_norm": 0.2155812531709671,
      "learning_rate": 9.99117733474185e-05,
      "loss": 0.075,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016813104506582023,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.304940795898437,
      "state_query_norm": 642.1004577636719,
      "state_relation_loss": 0.0016813104506582023,
      "step": 2040,
      "student_distance_mean": 811.3817932128907,
      "teacher_distance_mean": 1.149510097503662,
      "weighted_auxiliary_loss": 0.0016813104506582023,
      "weighted_relation_loss": 0.0016813104506582023
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9176800748362957,
      "flow_matching_loss": 0.07665898948907852,
      "grad_norm": 0.20650628209114075,
      "learning_rate": 9.990847027119234e-05,
      "loss": 0.0784,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00169181392993778,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.24866189956665,
      "state_query_norm": 642.8491394042969,
      "state_relation_loss": 0.00169181392993778,
      "step": 2050,
      "student_distance_mean": 811.5463012695312,
      "teacher_distance_mean": 1.152880322933197,
      "weighted_auxiliary_loss": 0.00169181392993778,
      "weighted_relation_loss": 0.00169181392993778
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9270346117867165,
      "flow_matching_loss": 0.0721564069390297,
      "grad_norm": 0.24281634390354156,
      "learning_rate": 9.990510655146877e-05,
      "loss": 0.0738,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001667894481215626,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.46126413345337,
      "state_query_norm": 652.6361267089844,
      "state_relation_loss": 0.001667894481215626,
      "step": 2060,
      "student_distance_mean": 823.2527587890625,
      "teacher_distance_mean": 1.1665914535522461,
      "weighted_auxiliary_loss": 0.001667894481215626,
      "weighted_relation_loss": 0.001667894481215626
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9363891487371374,
      "flow_matching_loss": 0.0736817479133606,
      "grad_norm": 0.2624732553958893,
      "learning_rate": 9.990168219233496e-05,
      "loss": 0.0754,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016800191253423691,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.072542858123779,
      "state_query_norm": 635.151708984375,
      "state_relation_loss": 0.0016800191253423691,
      "step": 2070,
      "student_distance_mean": 799.206494140625,
      "teacher_distance_mean": 1.128300964832306,
      "weighted_auxiliary_loss": 0.0016800191253423691,
      "weighted_relation_loss": 0.0016800191253423691
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9457436856875585,
      "flow_matching_loss": 0.07370823100209237,
      "grad_norm": 0.18516846001148224,
      "learning_rate": 9.989819719795188e-05,
      "loss": 0.0754,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016855562687851488,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.04415683746338,
      "state_query_norm": 640.1000793457031,
      "state_relation_loss": 0.0016855562687851488,
      "step": 2080,
      "student_distance_mean": 799.0228759765625,
      "teacher_distance_mean": 1.1358654618263244,
      "weighted_auxiliary_loss": 0.0016855562687851488,
      "weighted_relation_loss": 0.0016855562687851488
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9550982226379794,
      "flow_matching_loss": 0.07176201194524764,
      "grad_norm": 0.314350962638855,
      "learning_rate": 9.989465157255412e-05,
      "loss": 0.0734,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00162404595175758,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.057073020935059,
      "state_query_norm": 642.6400756835938,
      "state_relation_loss": 0.00162404595175758,
      "step": 2090,
      "student_distance_mean": 799.5229370117188,
      "teacher_distance_mean": 1.1527240991592407,
      "weighted_auxiliary_loss": 0.00162404595175758,
      "weighted_relation_loss": 0.00162404595175758
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9644527595884003,
      "flow_matching_loss": 0.07437191605567932,
      "grad_norm": 0.21992647647857666,
      "learning_rate": 9.989104532044994e-05,
      "loss": 0.076,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016108899260871112,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.15410737991333,
      "state_query_norm": 635.2526306152344,
      "state_relation_loss": 0.0016108899260871112,
      "step": 2100,
      "student_distance_mean": 806.0351989746093,
      "teacher_distance_mean": 1.1638254880905152,
      "weighted_auxiliary_loss": 0.0016108899260871112,
      "weighted_relation_loss": 0.0016108899260871112
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9738072965388214,
      "flow_matching_loss": 0.07265281900763512,
      "grad_norm": 0.33479559421539307,
      "learning_rate": 9.988737844602128e-05,
      "loss": 0.0744,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017328095622360707,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.060911941528321,
      "state_query_norm": 637.7822326660156,
      "state_relation_loss": 0.0017328095622360707,
      "step": 2110,
      "student_distance_mean": 800.2331970214843,
      "teacher_distance_mean": 1.1435122609138488,
      "weighted_auxiliary_loss": 0.0017328095622360707,
      "weighted_relation_loss": 0.0017328095622360707
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9831618334892422,
      "flow_matching_loss": 0.0730179376900196,
      "grad_norm": 0.2647485136985779,
      "learning_rate": 9.988365095372372e-05,
      "loss": 0.0747,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001646563084796071,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.315241432189941,
      "state_query_norm": 637.3870239257812,
      "state_relation_loss": 0.001646563084796071,
      "step": 2120,
      "student_distance_mean": 811.4862854003907,
      "teacher_distance_mean": 1.1482300281524658,
      "weighted_auxiliary_loss": 0.001646563084796071,
      "weighted_relation_loss": 0.001646563084796071
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 1.9925163704396631,
      "flow_matching_loss": 0.07434166297316551,
      "grad_norm": 0.3129539489746094,
      "learning_rate": 9.987986284808654e-05,
      "loss": 0.076,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016945034149102867,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.761244964599609,
      "state_query_norm": 670.0125305175782,
      "state_relation_loss": 0.0016945034149102867,
      "step": 2130,
      "student_distance_mean": 841.7317810058594,
      "teacher_distance_mean": 1.1625724196434022,
      "weighted_auxiliary_loss": 0.0016945034149102867,
      "weighted_relation_loss": 0.0016945034149102867
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.001870907390084,
      "flow_matching_loss": 0.07524389326572418,
      "grad_norm": 0.3267081379890442,
      "learning_rate": 9.987601413371264e-05,
      "loss": 0.0769,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0016719648730941117,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 14.541229724884033,
      "state_query_norm": 657.0002014160157,
      "state_relation_loss": 0.0016719648730941117,
      "step": 2140,
      "student_distance_mean": 827.7344177246093,
      "teacher_distance_mean": 1.1537338495254517,
      "weighted_auxiliary_loss": 0.0016719648730941117,
      "weighted_relation_loss": 0.0016719648730941117
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.0112254443405053,
      "flow_matching_loss": 0.07850624695420265,
      "grad_norm": 0.2736548185348511,
      "learning_rate": 9.987210481527855e-05,
      "loss": 0.0802,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017007286078296602,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.63303747177124,
      "state_query_norm": 658.6677917480469,
      "state_relation_loss": 0.0017007286078296602,
      "step": 2150,
      "student_distance_mean": 830.9818542480468,
      "teacher_distance_mean": 1.1571331739425659,
      "weighted_auxiliary_loss": 0.0017007286078296602,
      "weighted_relation_loss": 0.0017007286078296602
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.020579981290926,
      "flow_matching_loss": 0.07541133463382721,
      "grad_norm": 0.22195252776145935,
      "learning_rate": 9.98681348975345e-05,
      "loss": 0.077,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00161962965503335,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.633486270904541,
      "state_query_norm": 661.7378356933593,
      "state_relation_loss": 0.00161962965503335,
      "step": 2160,
      "student_distance_mean": 832.7548156738281,
      "teacher_distance_mean": 1.155735433101654,
      "weighted_auxiliary_loss": 0.00161962965503335,
      "weighted_relation_loss": 0.00161962965503335
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.029934518241347,
      "flow_matching_loss": 0.07511784210801124,
      "grad_norm": 0.17414402961730957,
      "learning_rate": 9.986410438530427e-05,
      "loss": 0.0769,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0017402138211764396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.44195327758789,
      "state_query_norm": 646.8072387695313,
      "state_relation_loss": 0.0017402138211764396,
      "step": 2170,
      "student_distance_mean": 820.1239624023438,
      "teacher_distance_mean": 1.1471267461776733,
      "weighted_auxiliary_loss": 0.0017402138211764396,
      "weighted_relation_loss": 0.0017402138211764396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.039289055191768,
      "flow_matching_loss": 0.07344180159270763,
      "grad_norm": 0.24165847897529602,
      "learning_rate": 9.986001328348534e-05,
      "loss": 0.0751,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016788125154562294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.768287944793702,
      "state_query_norm": 663.7076721191406,
      "state_relation_loss": 0.0016788125154562294,
      "step": 2180,
      "student_distance_mean": 838.7929626464844,
      "teacher_distance_mean": 1.1599724173545838,
      "weighted_auxiliary_loss": 0.0016788125154562294,
      "weighted_relation_loss": 0.0016788125154562294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.048643592142189,
      "flow_matching_loss": 0.07185257002711296,
      "grad_norm": 0.21191003918647766,
      "learning_rate": 9.985586159704878e-05,
      "loss": 0.0734,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015971318585798143,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.684493160247802,
      "state_query_norm": 654.9853942871093,
      "state_relation_loss": 0.0015971318585798143,
      "step": 2190,
      "student_distance_mean": 834.3022277832031,
      "teacher_distance_mean": 1.1650349855422975,
      "weighted_auxiliary_loss": 0.0015971318585798143,
      "weighted_relation_loss": 0.0015971318585798143
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.05799812909261,
      "flow_matching_loss": 0.07411423921585084,
      "grad_norm": 0.22347021102905273,
      "learning_rate": 9.985164933103929e-05,
      "loss": 0.0757,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016348467441275715,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.43292760848999,
      "state_query_norm": 651.8951904296875,
      "state_relation_loss": 0.0016348467441275715,
      "step": 2200,
      "student_distance_mean": 821.8958984375,
      "teacher_distance_mean": 1.1519717931747437,
      "weighted_auxiliary_loss": 0.0016348467441275715,
      "weighted_relation_loss": 0.0016348467441275715
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.067352666043031,
      "flow_matching_loss": 0.07303120717406272,
      "grad_norm": 0.1607319563627243,
      "learning_rate": 9.984737649057513e-05,
      "loss": 0.0746,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016025081276893616,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.481811141967773,
      "state_query_norm": 653.6377136230469,
      "state_relation_loss": 0.0016025081276893616,
      "step": 2210,
      "student_distance_mean": 823.9777465820313,
      "teacher_distance_mean": 1.1447540283203126,
      "weighted_auxiliary_loss": 0.0016025081276893616,
      "weighted_relation_loss": 0.0016025081276893616
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.0767072029934517,
      "flow_matching_loss": 0.07521739900112152,
      "grad_norm": 0.2842220366001129,
      "learning_rate": 9.984304308084827e-05,
      "loss": 0.0769,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016470106202177703,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.7533465385437,
      "state_query_norm": 661.0559631347656,
      "state_relation_loss": 0.0016470106202177703,
      "step": 2220,
      "student_distance_mean": 837.7868713378906,
      "teacher_distance_mean": 1.1594070553779603,
      "weighted_auxiliary_loss": 0.0016470106202177703,
      "weighted_relation_loss": 0.0016470106202177703
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.086061739943873,
      "flow_matching_loss": 0.0747406616806984,
      "grad_norm": 0.2686719000339508,
      "learning_rate": 9.983864910712416e-05,
      "loss": 0.0764,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016213595867156983,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.86608657836914,
      "state_query_norm": 663.4883850097656,
      "state_relation_loss": 0.0016213595867156983,
      "step": 2230,
      "student_distance_mean": 843.4019836425781,
      "teacher_distance_mean": 1.1616759896278381,
      "weighted_auxiliary_loss": 0.0016213595867156983,
      "weighted_relation_loss": 0.0016213595867156983
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.095416276894294,
      "flow_matching_loss": 0.07294152304530144,
      "grad_norm": 0.25205758213996887,
      "learning_rate": 9.98341945747419e-05,
      "loss": 0.0746,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016262311371974647,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.6483642578125,
      "state_query_norm": 652.1051391601562,
      "state_relation_loss": 0.0016262311371974647,
      "step": 2240,
      "student_distance_mean": 831.94873046875,
      "teacher_distance_mean": 1.1476080894470215,
      "weighted_auxiliary_loss": 0.0016262311371974647,
      "weighted_relation_loss": 0.0016262311371974647
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1047708138447145,
      "flow_matching_loss": 0.0742723897099495,
      "grad_norm": 0.21919094026088715,
      "learning_rate": 9.98296794891142e-05,
      "loss": 0.0759,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015787965152412654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.746866035461426,
      "state_query_norm": 663.9723327636718,
      "state_relation_loss": 0.0015787965152412654,
      "step": 2250,
      "student_distance_mean": 838.1876403808594,
      "teacher_distance_mean": 1.1482947587966919,
      "weighted_auxiliary_loss": 0.0015787965152412654,
      "weighted_relation_loss": 0.0015787965152412654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1141253507951356,
      "flow_matching_loss": 0.07535260394215584,
      "grad_norm": 0.18256624042987823,
      "learning_rate": 9.982510385572725e-05,
      "loss": 0.0769,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015905752312391996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.92207374572754,
      "state_query_norm": 667.6341796875,
      "state_relation_loss": 0.0015905752312391996,
      "step": 2260,
      "student_distance_mean": 847.9487060546875,
      "teacher_distance_mean": 1.1564378499984742,
      "weighted_auxiliary_loss": 0.0015905752312391996,
      "weighted_relation_loss": 0.0015905752312391996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1234798877455567,
      "flow_matching_loss": 0.07497590854763984,
      "grad_norm": 0.22810275852680206,
      "learning_rate": 9.982046768014094e-05,
      "loss": 0.0766,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015767781180329621,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.851778507232666,
      "state_query_norm": 663.6166381835938,
      "state_relation_loss": 0.0015767781180329621,
      "step": 2270,
      "student_distance_mean": 841.8980163574219,
      "teacher_distance_mean": 1.1484689831733703,
      "weighted_auxiliary_loss": 0.0015767781180329621,
      "weighted_relation_loss": 0.0015767781180329621
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1328344246959774,
      "flow_matching_loss": 0.0708786565810442,
      "grad_norm": 0.23528899252414703,
      "learning_rate": 9.981577096798863e-05,
      "loss": 0.0725,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001585434831213206,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.828202533721925,
      "state_query_norm": 667.1094177246093,
      "state_relation_loss": 0.001585434831213206,
      "step": 2280,
      "student_distance_mean": 842.8076904296875,
      "teacher_distance_mean": 1.1479369044303893,
      "weighted_auxiliary_loss": 0.001585434831213206,
      "weighted_relation_loss": 0.001585434831213206
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1421889616463985,
      "flow_matching_loss": 0.07293438650667668,
      "grad_norm": 0.2653783857822418,
      "learning_rate": 9.981101372497727e-05,
      "loss": 0.0745,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015858312137424946,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.069859504699707,
      "state_query_norm": 666.3919250488282,
      "state_relation_loss": 0.0015858312137424946,
      "step": 2290,
      "student_distance_mean": 856.2255737304688,
      "teacher_distance_mean": 1.166939389705658,
      "weighted_auxiliary_loss": 0.0015858312137424946,
      "weighted_relation_loss": 0.0015858312137424946
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1515434985968196,
      "flow_matching_loss": 0.07072190567851067,
      "grad_norm": 0.28406545519828796,
      "learning_rate": 9.980619595688737e-05,
      "loss": 0.0723,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016168489353731274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.79095516204834,
      "state_query_norm": 664.1426635742188,
      "state_relation_loss": 0.0016168489353731274,
      "step": 2300,
      "student_distance_mean": 837.8688232421875,
      "teacher_distance_mean": 1.1548645615577697,
      "weighted_auxiliary_loss": 0.0016168489353731274,
      "weighted_relation_loss": 0.0016168489353731274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1608980355472402,
      "flow_matching_loss": 0.07335647493600846,
      "grad_norm": 0.27683424949645996,
      "learning_rate": 9.980131766957295e-05,
      "loss": 0.075,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0016156450612470508,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.104425716400147,
      "state_query_norm": 670.6073364257812,
      "state_relation_loss": 0.0016156450612470508,
      "step": 2310,
      "student_distance_mean": 857.9377197265625,
      "teacher_distance_mean": 1.1589620471000672,
      "weighted_auxiliary_loss": 0.0016156450612470508,
      "weighted_relation_loss": 0.0016156450612470508
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1702525724976613,
      "flow_matching_loss": 0.07298567071557045,
      "grad_norm": 0.2620185911655426,
      "learning_rate": 9.979637886896163e-05,
      "loss": 0.0745,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015619151992723346,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.04588441848755,
      "state_query_norm": 682.1636779785156,
      "state_relation_loss": 0.0015619151992723346,
      "step": 2320,
      "student_distance_mean": 855.7879577636719,
      "teacher_distance_mean": 1.1632193684577943,
      "weighted_auxiliary_loss": 0.0015619151992723346,
      "weighted_relation_loss": 0.0015619151992723346
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.1796071094480824,
      "flow_matching_loss": 0.07431837990880012,
      "grad_norm": 0.23274633288383484,
      "learning_rate": 9.979137956105447e-05,
      "loss": 0.0759,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015370976412668823,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.868980312347412,
      "state_query_norm": 661.5570617675781,
      "state_relation_loss": 0.0015370976412668823,
      "step": 2330,
      "student_distance_mean": 842.6657958984375,
      "teacher_distance_mean": 1.1460992813110351,
      "weighted_auxiliary_loss": 0.0015370976412668823,
      "weighted_relation_loss": 0.0015370976412668823
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.188961646398503,
      "flow_matching_loss": 0.07293166592717171,
      "grad_norm": 0.21090193092823029,
      "learning_rate": 9.978631975192613e-05,
      "loss": 0.0745,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015927263419143855,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 14.980064868927002,
      "state_query_norm": 682.3815368652344,
      "state_relation_loss": 0.0015927263419143855,
      "step": 2340,
      "student_distance_mean": 852.8685913085938,
      "teacher_distance_mean": 1.147693943977356,
      "weighted_auxiliary_loss": 0.0015927263419143855,
      "weighted_relation_loss": 0.0015927263419143855
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.198316183348924,
      "flow_matching_loss": 0.0699254609644413,
      "grad_norm": 0.1537030041217804,
      "learning_rate": 9.978119944772475e-05,
      "loss": 0.0714,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015223652357235552,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.067367553710938,
      "state_query_norm": 673.7156677246094,
      "state_relation_loss": 0.0015223652357235552,
      "step": 2350,
      "student_distance_mean": 856.7509338378907,
      "teacher_distance_mean": 1.1465827465057372,
      "weighted_auxiliary_loss": 0.0015223652357235552,
      "weighted_relation_loss": 0.0015223652357235552
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.2076707202993453,
      "flow_matching_loss": 0.07413184866309167,
      "grad_norm": 0.16336601972579956,
      "learning_rate": 9.977601865467197e-05,
      "loss": 0.0756,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014925727387890219,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.114593696594238,
      "state_query_norm": 676.5016845703125,
      "state_relation_loss": 0.0014925727387890219,
      "step": 2360,
      "student_distance_mean": 858.9870178222657,
      "teacher_distance_mean": 1.1610548138618468,
      "weighted_auxiliary_loss": 0.0014925727387890219,
      "weighted_relation_loss": 0.0014925727387890219
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.217025257249766,
      "flow_matching_loss": 0.07164428234100342,
      "grad_norm": 0.24021361768245697,
      "learning_rate": 9.977077737906297e-05,
      "loss": 0.0731,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00150006792973727,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.23119068145752,
      "state_query_norm": 680.7447021484375,
      "state_relation_loss": 0.00150006792973727,
      "step": 2370,
      "student_distance_mean": 865.5287414550781,
      "teacher_distance_mean": 1.1551414370536803,
      "weighted_auxiliary_loss": 0.00150006792973727,
      "weighted_relation_loss": 0.00150006792973727
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.226379794200187,
      "flow_matching_loss": 0.07343701720237732,
      "grad_norm": 0.20783251523971558,
      "learning_rate": 9.976547562726636e-05,
      "loss": 0.075,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015341468271799386,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.359695053100586,
      "state_query_norm": 679.9660339355469,
      "state_relation_loss": 0.0015341468271799386,
      "step": 2380,
      "student_distance_mean": 869.5773498535157,
      "teacher_distance_mean": 1.1518880367279052,
      "weighted_auxiliary_loss": 0.0015341468271799386,
      "weighted_relation_loss": 0.0015341468271799386
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.235734331150608,
      "flow_matching_loss": 0.06886420175433158,
      "grad_norm": 0.29645252227783203,
      "learning_rate": 9.976011340572429e-05,
      "loss": 0.0704,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015620503341779112,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.321971702575684,
      "state_query_norm": 677.8196838378906,
      "state_relation_loss": 0.0015620503341779112,
      "step": 2390,
      "student_distance_mean": 868.5393432617187,
      "teacher_distance_mean": 1.1567612528800963,
      "weighted_auxiliary_loss": 0.0015620503341779112,
      "weighted_relation_loss": 0.0015620503341779112
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.245088868101029,
      "flow_matching_loss": 0.07214475274085999,
      "grad_norm": 0.2567208409309387,
      "learning_rate": 9.975469072095237e-05,
      "loss": 0.0736,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014959274441935122,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.349958133697509,
      "state_query_norm": 683.3470458984375,
      "state_relation_loss": 0.0014959274441935122,
      "step": 2400,
      "student_distance_mean": 868.7168701171875,
      "teacher_distance_mean": 1.1556586503982544,
      "weighted_auxiliary_loss": 0.0014959274441935122,
      "weighted_relation_loss": 0.0014959274441935122
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.25444340505145,
      "flow_matching_loss": 0.07163106203079224,
      "grad_norm": 0.24809525907039642,
      "learning_rate": 9.974920757953965e-05,
      "loss": 0.0731,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014340708032250405,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.189360427856446,
      "state_query_norm": 675.3263122558594,
      "state_relation_loss": 0.0014340708032250405,
      "step": 2410,
      "student_distance_mean": 862.3469421386719,
      "teacher_distance_mean": 1.1506392359733582,
      "weighted_auxiliary_loss": 0.0014340708032250405,
      "weighted_relation_loss": 0.0014340708032250405
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.263797942001871,
      "flow_matching_loss": 0.07139974422752857,
      "grad_norm": 0.21281422674655914,
      "learning_rate": 9.97436639881487e-05,
      "loss": 0.0728,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014409854775294662,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.38465929031372,
      "state_query_norm": 686.7196838378907,
      "state_relation_loss": 0.0014409854775294662,
      "step": 2420,
      "student_distance_mean": 873.0026062011718,
      "teacher_distance_mean": 1.1407102346420288,
      "weighted_auxiliary_loss": 0.0014409854775294662,
      "weighted_relation_loss": 0.0014409854775294662
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.2731524789522917,
      "flow_matching_loss": 0.0693709172308445,
      "grad_norm": 0.20583291351795197,
      "learning_rate": 9.973805995351545e-05,
      "loss": 0.0709,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015009113238193094,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.371563911437988,
      "state_query_norm": 686.8006591796875,
      "state_relation_loss": 0.0015009113238193094,
      "step": 2430,
      "student_distance_mean": 872.4851379394531,
      "teacher_distance_mean": 1.1488651037216187,
      "weighted_auxiliary_loss": 0.0015009113238193094,
      "weighted_relation_loss": 0.0015009113238193094
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.2825070159027128,
      "flow_matching_loss": 0.0734326496720314,
      "grad_norm": 0.2213982194662094,
      "learning_rate": 9.973239548244939e-05,
      "loss": 0.075,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015575955156236887,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.28019094467163,
      "state_query_norm": 679.1342346191407,
      "state_relation_loss": 0.0015575955156236887,
      "step": 2440,
      "student_distance_mean": 864.1163452148437,
      "teacher_distance_mean": 1.1440345883369445,
      "weighted_auxiliary_loss": 0.0015575955156236887,
      "weighted_relation_loss": 0.0015575955156236887
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.291861552853134,
      "flow_matching_loss": 0.07256495729088783,
      "grad_norm": 0.2028656005859375,
      "learning_rate": 9.972667058183333e-05,
      "loss": 0.074,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014133295975625516,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.47707691192627,
      "state_query_norm": 697.8563598632812,
      "state_relation_loss": 0.0014133295975625516,
      "step": 2450,
      "student_distance_mean": 880.8043151855469,
      "teacher_distance_mean": 1.1619381070137025,
      "weighted_auxiliary_loss": 0.0014133295975625516,
      "weighted_relation_loss": 0.0014133295975625516
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.301216089803555,
      "flow_matching_loss": 0.06842116974294185,
      "grad_norm": 0.18232570588588715,
      "learning_rate": 9.972088525862362e-05,
      "loss": 0.0699,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015008613700047135,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.4300931930542,
      "state_query_norm": 693.3426940917968,
      "state_relation_loss": 0.0015008613700047135,
      "step": 2460,
      "student_distance_mean": 874.9178588867187,
      "teacher_distance_mean": 1.1590028285980225,
      "weighted_auxiliary_loss": 0.0015008613700047135,
      "weighted_relation_loss": 0.0015008613700047135
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3105706267539756,
      "flow_matching_loss": 0.07038565538823605,
      "grad_norm": 0.18303126096725464,
      "learning_rate": 9.971503951984995e-05,
      "loss": 0.0718,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001441091811284423,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.879146480560303,
      "state_query_norm": 699.2589538574218,
      "state_relation_loss": 0.001441091811284423,
      "step": 2470,
      "student_distance_mean": 897.9959350585938,
      "teacher_distance_mean": 1.1635619044303893,
      "weighted_auxiliary_loss": 0.001441091811284423,
      "weighted_relation_loss": 0.001441091811284423
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3199251637043967,
      "flow_matching_loss": 0.07254122570157051,
      "grad_norm": 0.16771933436393738,
      "learning_rate": 9.970913337261543e-05,
      "loss": 0.074,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001418481464497745,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.918201541900634,
      "state_query_norm": 714.8692138671875,
      "state_relation_loss": 0.001418481464497745,
      "step": 2480,
      "student_distance_mean": 905.9972351074218,
      "teacher_distance_mean": 1.1695952892303467,
      "weighted_auxiliary_loss": 0.001418481464497745,
      "weighted_relation_loss": 0.001418481464497745
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3292797006548174,
      "flow_matching_loss": 0.071743543446064,
      "grad_norm": 0.18032614886760712,
      "learning_rate": 9.97031668240966e-05,
      "loss": 0.0732,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001433874003123492,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.639496612548829,
      "state_query_norm": 695.7482360839844,
      "state_relation_loss": 0.001433874003123492,
      "step": 2490,
      "student_distance_mean": 887.7346374511719,
      "teacher_distance_mean": 1.1463501691818236,
      "weighted_auxiliary_loss": 0.001433874003123492,
      "weighted_relation_loss": 0.001433874003123492
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3386342376052385,
      "flow_matching_loss": 0.07014854624867439,
      "grad_norm": 0.18798239529132843,
      "learning_rate": 9.969713988154339e-05,
      "loss": 0.0716,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014828551560640336,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.909021663665772,
      "state_query_norm": 711.8352355957031,
      "state_relation_loss": 0.0014828551560640336,
      "step": 2500,
      "student_distance_mean": 902.22490234375,
      "teacher_distance_mean": 1.1599424362182618,
      "weighted_auxiliary_loss": 0.0014828551560640336,
      "weighted_relation_loss": 0.0014828551560640336
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3479887745556596,
      "flow_matching_loss": 0.067243180423975,
      "grad_norm": 0.178035706281662,
      "learning_rate": 9.969105255227906e-05,
      "loss": 0.0687,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014826479251496493,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.775590419769287,
      "state_query_norm": 712.5939208984375,
      "state_relation_loss": 0.0014826479251496493,
      "step": 2510,
      "student_distance_mean": 898.5018127441406,
      "teacher_distance_mean": 1.1644224166870116,
      "weighted_auxiliary_loss": 0.0014826479251496493,
      "weighted_relation_loss": 0.0014826479251496493
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3573433115060807,
      "flow_matching_loss": 0.06860526762902737,
      "grad_norm": 0.20106005668640137,
      "learning_rate": 9.968490484370035e-05,
      "loss": 0.0701,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014730179100297391,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.729355716705323,
      "state_query_norm": 694.3784606933593,
      "state_relation_loss": 0.0014730179100297391,
      "step": 2520,
      "student_distance_mean": 890.1100280761718,
      "teacher_distance_mean": 1.1613133430480957,
      "weighted_auxiliary_loss": 0.0014730179100297391,
      "weighted_relation_loss": 0.0014730179100297391
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3666978484565013,
      "flow_matching_loss": 0.07146027348935605,
      "grad_norm": 0.14267459511756897,
      "learning_rate": 9.967869676327726e-05,
      "loss": 0.0729,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014192119357176124,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.501744079589844,
      "state_query_norm": 695.3307678222657,
      "state_relation_loss": 0.0014192119357176124,
      "step": 2530,
      "student_distance_mean": 882.0869995117188,
      "teacher_distance_mean": 1.1463262796401978,
      "weighted_auxiliary_loss": 0.0014192119357176124,
      "weighted_relation_loss": 0.0014192119357176124
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.3760523854069224,
      "flow_matching_loss": 0.06580617614090442,
      "grad_norm": 0.12527702748775482,
      "learning_rate": 9.96724283185532e-05,
      "loss": 0.0672,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014365148032084107,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.682070446014404,
      "state_query_norm": 694.370556640625,
      "state_relation_loss": 0.0014365148032084107,
      "step": 2540,
      "student_distance_mean": 889.4680786132812,
      "teacher_distance_mean": 1.1516673803329467,
      "weighted_auxiliary_loss": 0.0014365148032084107,
      "weighted_relation_loss": 0.0014365148032084107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.385406922357343,
      "flow_matching_loss": 0.06875670328736305,
      "grad_norm": 0.21440960466861725,
      "learning_rate": 9.966609951714494e-05,
      "loss": 0.0702,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014145763707347214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.767085742950439,
      "state_query_norm": 697.8744323730468,
      "state_relation_loss": 0.0014145763707347214,
      "step": 2550,
      "student_distance_mean": 892.4533142089844,
      "teacher_distance_mean": 1.155202567577362,
      "weighted_auxiliary_loss": 0.0014145763707347214,
      "weighted_relation_loss": 0.0014145763707347214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.394761459307764,
      "flow_matching_loss": 0.06737622879445553,
      "grad_norm": 0.12610967457294464,
      "learning_rate": 9.965971036674255e-05,
      "loss": 0.0688,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00144444250036031,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.70587329864502,
      "state_query_norm": 697.0919616699218,
      "state_relation_loss": 0.00144444250036031,
      "step": 2560,
      "student_distance_mean": 889.4948669433594,
      "teacher_distance_mean": 1.1411049723625184,
      "weighted_auxiliary_loss": 0.00144444250036031,
      "weighted_relation_loss": 0.00144444250036031
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.4041159962581853,
      "flow_matching_loss": 0.06871109753847122,
      "grad_norm": 0.15060743689537048,
      "learning_rate": 9.965326087510947e-05,
      "loss": 0.0701,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014104280737228692,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.90842809677124,
      "state_query_norm": 706.9473999023437,
      "state_relation_loss": 0.0014104280737228692,
      "step": 2570,
      "student_distance_mean": 903.0892150878906,
      "teacher_distance_mean": 1.1673221945762635,
      "weighted_auxiliary_loss": 0.0014104280737228692,
      "weighted_relation_loss": 0.0014104280737228692
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.4134705332086064,
      "flow_matching_loss": 0.07134409323334694,
      "grad_norm": 0.20914588868618011,
      "learning_rate": 9.964675105008244e-05,
      "loss": 0.0728,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001437283877748996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.846607685089111,
      "state_query_norm": 704.1692626953125,
      "state_relation_loss": 0.001437283877748996,
      "step": 2580,
      "student_distance_mean": 898.4569763183594,
      "teacher_distance_mean": 1.1523389220237732,
      "weighted_auxiliary_loss": 0.001437283877748996,
      "weighted_relation_loss": 0.001437283877748996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.422825070159027,
      "flow_matching_loss": 0.07112221643328667,
      "grad_norm": 0.22808276116847992,
      "learning_rate": 9.964018089957147e-05,
      "loss": 0.0726,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014848702936433255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.772211170196533,
      "state_query_norm": 698.91953125,
      "state_relation_loss": 0.0014848702936433255,
      "step": 2590,
      "student_distance_mean": 891.8319763183594,
      "teacher_distance_mean": 1.148923933506012,
      "weighted_auxiliary_loss": 0.0014848702936433255,
      "weighted_relation_loss": 0.0014848702936433255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.432179607109448,
      "flow_matching_loss": 0.07051147818565369,
      "grad_norm": 0.20792938768863678,
      "learning_rate": 9.963355043155996e-05,
      "loss": 0.072,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015204290742985905,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.22307300567627,
      "state_query_norm": 713.0792236328125,
      "state_relation_loss": 0.0015204290742985905,
      "step": 2600,
      "student_distance_mean": 918.4112487792969,
      "teacher_distance_mean": 1.1743061780929565,
      "weighted_auxiliary_loss": 0.0015204290742985905,
      "weighted_relation_loss": 0.0015204290742985905
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.441534144059869,
      "flow_matching_loss": 0.06829880475997925,
      "grad_norm": 0.23014071583747864,
      "learning_rate": 9.962685965410455e-05,
      "loss": 0.0697,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014173304662108421,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.905363178253173,
      "state_query_norm": 712.0779846191406,
      "state_relation_loss": 0.0014173304662108421,
      "step": 2610,
      "student_distance_mean": 903.3007202148438,
      "teacher_distance_mean": 1.1422237277030944,
      "weighted_auxiliary_loss": 0.0014173304662108421,
      "weighted_relation_loss": 0.0014173304662108421
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.45088868101029,
      "flow_matching_loss": 0.07034763507544994,
      "grad_norm": 0.17355388402938843,
      "learning_rate": 9.962010857533514e-05,
      "loss": 0.0718,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014803583966568113,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.95347843170166,
      "state_query_norm": 711.3197204589844,
      "state_relation_loss": 0.0014803583966568113,
      "step": 2620,
      "student_distance_mean": 904.5494201660156,
      "teacher_distance_mean": 1.1422609329223632,
      "weighted_auxiliary_loss": 0.0014803583966568113,
      "weighted_relation_loss": 0.0014803583966568113
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.460243217960711,
      "flow_matching_loss": 0.07224676758050919,
      "grad_norm": 0.1535940170288086,
      "learning_rate": 9.961329720345493e-05,
      "loss": 0.0738,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015226659830659627,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.018435382843016,
      "state_query_norm": 706.3573852539063,
      "state_relation_loss": 0.0015226659830659627,
      "step": 2630,
      "student_distance_mean": 905.6172180175781,
      "teacher_distance_mean": 1.155264949798584,
      "weighted_auxiliary_loss": 0.0015226659830659627,
      "weighted_relation_loss": 0.0015226659830659627
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.469597754911132,
      "flow_matching_loss": 0.07213574945926667,
      "grad_norm": 0.22586938738822937,
      "learning_rate": 9.96064255467404e-05,
      "loss": 0.0736,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014511424582451583,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.899784755706786,
      "state_query_norm": 702.8613342285156,
      "state_relation_loss": 0.0014511424582451583,
      "step": 2640,
      "student_distance_mean": 901.5603332519531,
      "teacher_distance_mean": 1.1491597533226012,
      "weighted_auxiliary_loss": 0.0014511424582451583,
      "weighted_relation_loss": 0.0014511424582451583
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.4789522918615527,
      "flow_matching_loss": 0.06889676973223686,
      "grad_norm": 0.20343492925167084,
      "learning_rate": 9.959949361354126e-05,
      "loss": 0.0704,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014621768612414598,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.886749267578125,
      "state_query_norm": 705.1831604003906,
      "state_relation_loss": 0.0014621768612414598,
      "step": 2650,
      "student_distance_mean": 901.9769897460938,
      "teacher_distance_mean": 1.1421310663223267,
      "weighted_auxiliary_loss": 0.0014621768612414598,
      "weighted_relation_loss": 0.0014621768612414598
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.488306828811974,
      "flow_matching_loss": 0.0694273967295885,
      "grad_norm": 0.15379540622234344,
      "learning_rate": 9.959250141228045e-05,
      "loss": 0.0709,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014443458523601294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.120902347564698,
      "state_query_norm": 712.8470825195312,
      "state_relation_loss": 0.0014443458523601294,
      "step": 2660,
      "student_distance_mean": 912.5272521972656,
      "teacher_distance_mean": 1.1515992283821106,
      "weighted_auxiliary_loss": 0.0014443458523601294,
      "weighted_relation_loss": 0.0014443458523601294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.497661365762395,
      "flow_matching_loss": 0.06959854923188687,
      "grad_norm": 0.16592969000339508,
      "learning_rate": 9.958544895145414e-05,
      "loss": 0.0711,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014572352869436146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.801433753967284,
      "state_query_norm": 696.4358764648438,
      "state_relation_loss": 0.0014572352869436146,
      "step": 2670,
      "student_distance_mean": 893.130908203125,
      "teacher_distance_mean": 1.1414517521858216,
      "weighted_auxiliary_loss": 0.0014572352869436146,
      "weighted_relation_loss": 0.0014572352869436146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5070159027128156,
      "flow_matching_loss": 0.06966987475752831,
      "grad_norm": 0.25336092710494995,
      "learning_rate": 9.957833623963177e-05,
      "loss": 0.0711,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014530545915476978,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.18629398345947,
      "state_query_norm": 716.1901123046875,
      "state_relation_loss": 0.0014530545915476978,
      "step": 2680,
      "student_distance_mean": 916.8274108886719,
      "teacher_distance_mean": 1.1578286170959473,
      "weighted_auxiliary_loss": 0.0014530545915476978,
      "weighted_relation_loss": 0.0014530545915476978
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5163704396632367,
      "flow_matching_loss": 0.07049149572849274,
      "grad_norm": 0.21556958556175232,
      "learning_rate": 9.957116328545593e-05,
      "loss": 0.0719,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013945602695457637,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.24327383041382,
      "state_query_norm": 718.0304443359375,
      "state_relation_loss": 0.0013945602695457637,
      "step": 2690,
      "student_distance_mean": 922.4367126464844,
      "teacher_distance_mean": 1.153695321083069,
      "weighted_auxiliary_loss": 0.0013945602695457637,
      "weighted_relation_loss": 0.0013945602695457637
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.525724976613658,
      "flow_matching_loss": 0.0704803004860878,
      "grad_norm": 0.2686383128166199,
      "learning_rate": 9.956393009764244e-05,
      "loss": 0.072,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001535562553908676,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.181145858764648,
      "state_query_norm": 716.6304992675781,
      "state_relation_loss": 0.001535562553908676,
      "step": 2700,
      "student_distance_mean": 916.3850891113282,
      "teacher_distance_mean": 1.1500471591949464,
      "weighted_auxiliary_loss": 0.001535562553908676,
      "weighted_relation_loss": 0.001535562553908676
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5350795135640785,
      "flow_matching_loss": 0.07253988981246948,
      "grad_norm": 0.15364424884319305,
      "learning_rate": 9.955663668498032e-05,
      "loss": 0.074,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014560835086740553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.4527756690979,
      "state_query_norm": 732.0369140625,
      "state_relation_loss": 0.0014560835086740553,
      "step": 2710,
      "student_distance_mean": 936.6380310058594,
      "teacher_distance_mean": 1.1595804691314697,
      "weighted_auxiliary_loss": 0.0014560835086740553,
      "weighted_relation_loss": 0.0014560835086740553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5444340505144996,
      "flow_matching_loss": 0.07137565687298775,
      "grad_norm": 0.16671401262283325,
      "learning_rate": 9.954928305633173e-05,
      "loss": 0.0729,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0015123833320103586,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.414645957946778,
      "state_query_norm": 726.9693420410156,
      "state_relation_loss": 0.0015123833320103586,
      "step": 2720,
      "student_distance_mean": 928.3467590332032,
      "teacher_distance_mean": 1.164177918434143,
      "weighted_auxiliary_loss": 0.0015123833320103586,
      "weighted_relation_loss": 0.0015123833320103586
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.55378858746492,
      "flow_matching_loss": 0.06771515645086765,
      "grad_norm": 0.16767658293247223,
      "learning_rate": 9.954186922063204e-05,
      "loss": 0.0692,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014740120619535445,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 15.956373500823975,
      "state_query_norm": 707.6818969726562,
      "state_relation_loss": 0.0014740120619535445,
      "step": 2730,
      "student_distance_mean": 902.2422302246093,
      "teacher_distance_mean": 1.1368435621261597,
      "weighted_auxiliary_loss": 0.0014740120619535445,
      "weighted_relation_loss": 0.0014740120619535445
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5631431244153413,
      "flow_matching_loss": 0.0681543193757534,
      "grad_norm": 0.23829154670238495,
      "learning_rate": 9.953439518688974e-05,
      "loss": 0.0696,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001405826723203063,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.456497764587404,
      "state_query_norm": 732.9711486816407,
      "state_relation_loss": 0.001405826723203063,
      "step": 2740,
      "student_distance_mean": 934.0225402832032,
      "teacher_distance_mean": 1.1441065430641175,
      "weighted_auxiliary_loss": 0.001405826723203063,
      "weighted_relation_loss": 0.001405826723203063
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5724976613657624,
      "flow_matching_loss": 0.07135378159582614,
      "grad_norm": 0.16945049166679382,
      "learning_rate": 9.952686096418652e-05,
      "loss": 0.0728,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001438826124649495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.90984191894531,
      "state_query_norm": 744.9072998046875,
      "state_relation_loss": 0.001438826124649495,
      "step": 2750,
      "student_distance_mean": 958.2289489746094,
      "teacher_distance_mean": 1.1644675970077514,
      "weighted_auxiliary_loss": 0.001438826124649495,
      "weighted_relation_loss": 0.001438826124649495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.5818521983161835,
      "flow_matching_loss": 0.06928888782858848,
      "grad_norm": 0.24265894293785095,
      "learning_rate": 9.951926656167714e-05,
      "loss": 0.0707,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014495628769509495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.802453231811523,
      "state_query_norm": 741.8277648925781,
      "state_relation_loss": 0.0014495628769509495,
      "step": 2760,
      "student_distance_mean": 948.6819702148438,
      "teacher_distance_mean": 1.1556137919425964,
      "weighted_auxiliary_loss": 0.0014495628769509495,
      "weighted_relation_loss": 0.0014495628769509495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.591206735266604,
      "flow_matching_loss": 0.06838748380541801,
      "grad_norm": 0.1420225352048874,
      "learning_rate": 9.951161198858953e-05,
      "loss": 0.0698,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014143984066322447,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.71926612854004,
      "state_query_norm": 743.5330322265625,
      "state_relation_loss": 0.0014143984066322447,
      "step": 2770,
      "student_distance_mean": 946.7732604980469,
      "teacher_distance_mean": 1.1592402458190918,
      "weighted_auxiliary_loss": 0.0014143984066322447,
      "weighted_relation_loss": 0.0014143984066322447
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.6005612722170253,
      "flow_matching_loss": 0.06920456886291504,
      "grad_norm": 0.15883633494377136,
      "learning_rate": 9.950389725422471e-05,
      "loss": 0.0705,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013369563850574196,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.579966259002685,
      "state_query_norm": 730.2845336914063,
      "state_relation_loss": 0.0013369563850574196,
      "step": 2780,
      "student_distance_mean": 938.7540588378906,
      "teacher_distance_mean": 1.1451243042945862,
      "weighted_auxiliary_loss": 0.0013369563850574196,
      "weighted_relation_loss": 0.0013369563850574196
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.6099158091674464,
      "flow_matching_loss": 0.06906823888421058,
      "grad_norm": 0.16023865342140198,
      "learning_rate": 9.949612236795682e-05,
      "loss": 0.0705,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013952738139778376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.55421648025513,
      "state_query_norm": 733.3504333496094,
      "state_relation_loss": 0.0013952738139778376,
      "step": 2790,
      "student_distance_mean": 936.6558532714844,
      "teacher_distance_mean": 1.1464686632156371,
      "weighted_auxiliary_loss": 0.0013952738139778376,
      "weighted_relation_loss": 0.0013952738139778376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.619270346117867,
      "flow_matching_loss": 0.07165778130292892,
      "grad_norm": 0.2666313052177429,
      "learning_rate": 9.948828733923305e-05,
      "loss": 0.073,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013444487587548793,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.822303009033202,
      "state_query_norm": 741.656982421875,
      "state_relation_loss": 0.0013444487587548793,
      "step": 2800,
      "student_distance_mean": 951.8124633789063,
      "teacher_distance_mean": 1.163680100440979,
      "weighted_auxiliary_loss": 0.0013444487587548793,
      "weighted_relation_loss": 0.0013444487587548793
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.628624883068288,
      "flow_matching_loss": 0.07142695263028145,
      "grad_norm": 0.25620901584625244,
      "learning_rate": 9.948039217757374e-05,
      "loss": 0.0728,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013665847247466445,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.578653717041014,
      "state_query_norm": 732.7944946289062,
      "state_relation_loss": 0.0013665847247466445,
      "step": 2810,
      "student_distance_mean": 936.9557556152344,
      "teacher_distance_mean": 1.143812358379364,
      "weighted_auxiliary_loss": 0.0013665847247466445,
      "weighted_relation_loss": 0.0013665847247466445
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.637979420018709,
      "flow_matching_loss": 0.0688992589712143,
      "grad_norm": 0.22904300689697266,
      "learning_rate": 9.947243689257225e-05,
      "loss": 0.0702,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013398499810136854,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.870139408111573,
      "state_query_norm": 742.74365234375,
      "state_relation_loss": 0.0013398499810136854,
      "step": 2820,
      "student_distance_mean": 952.3557983398438,
      "teacher_distance_mean": 1.1570000648498535,
      "weighted_auxiliary_loss": 0.0013398499810136854,
      "weighted_relation_loss": 0.0013398499810136854
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.64733395696913,
      "flow_matching_loss": 0.06810216903686524,
      "grad_norm": 0.17579643428325653,
      "learning_rate": 9.946442149389497e-05,
      "loss": 0.0695,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013736078515648841,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.7501127243042,
      "state_query_norm": 741.8648254394532,
      "state_relation_loss": 0.0013736078515648841,
      "step": 2830,
      "student_distance_mean": 946.1168395996094,
      "teacher_distance_mean": 1.1488034009933472,
      "weighted_auxiliary_loss": 0.0013736078515648841,
      "weighted_relation_loss": 0.0013736078515648841
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.656688493919551,
      "flow_matching_loss": 0.06849814355373382,
      "grad_norm": 0.16792455315589905,
      "learning_rate": 9.945634599128139e-05,
      "loss": 0.0699,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014213057584129274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.804044914245605,
      "state_query_norm": 742.1039428710938,
      "state_relation_loss": 0.0014213057584129274,
      "step": 2840,
      "student_distance_mean": 948.9595153808593,
      "teacher_distance_mean": 1.1641866207122802,
      "weighted_auxiliary_loss": 0.0014213057584129274,
      "weighted_relation_loss": 0.0014213057584129274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.666043030869972,
      "flow_matching_loss": 0.07153939753770829,
      "grad_norm": 0.2058432251214981,
      "learning_rate": 9.944821039454402e-05,
      "loss": 0.0729,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001320945320185274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.529775142669678,
      "state_query_norm": 733.4887878417969,
      "state_relation_loss": 0.001320945320185274,
      "step": 2850,
      "student_distance_mean": 935.737060546875,
      "teacher_distance_mean": 1.1459463596343995,
      "weighted_auxiliary_loss": 0.001320945320185274,
      "weighted_relation_loss": 0.001320945320185274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.6753975678203927,
      "flow_matching_loss": 0.06693740971386433,
      "grad_norm": 0.1311621069908142,
      "learning_rate": 9.944001471356835e-05,
      "loss": 0.0683,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013475467450916768,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.013300895690918,
      "state_query_norm": 746.6695495605469,
      "state_relation_loss": 0.0013475467450916768,
      "step": 2860,
      "student_distance_mean": 961.7202941894532,
      "teacher_distance_mean": 1.1701145887374877,
      "weighted_auxiliary_loss": 0.0013475467450916768,
      "weighted_relation_loss": 0.0013475467450916768
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.684752104770814,
      "flow_matching_loss": 0.06962588243186474,
      "grad_norm": 0.24597589671611786,
      "learning_rate": 9.94317589583129e-05,
      "loss": 0.071,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014136143145151436,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.578171730041504,
      "state_query_norm": 729.5015441894532,
      "state_relation_loss": 0.0014136143145151436,
      "step": 2870,
      "student_distance_mean": 941.8791931152343,
      "teacher_distance_mean": 1.1457627296447754,
      "weighted_auxiliary_loss": 0.0014136143145151436,
      "weighted_relation_loss": 0.0014136143145151436
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.694106641721235,
      "flow_matching_loss": 0.06800616309046745,
      "grad_norm": 0.2514389753341675,
      "learning_rate": 9.942344313880922e-05,
      "loss": 0.0693,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013160495669580995,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.515931701660158,
      "state_query_norm": 726.6583251953125,
      "state_relation_loss": 0.0013160495669580995,
      "step": 2880,
      "student_distance_mean": 933.8889343261719,
      "teacher_distance_mean": 1.1391291260719298,
      "weighted_auxiliary_loss": 0.0013160495669580995,
      "weighted_relation_loss": 0.0013160495669580995
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.703461178671656,
      "flow_matching_loss": 0.07087229043245316,
      "grad_norm": 0.2626822590827942,
      "learning_rate": 9.941506726516179e-05,
      "loss": 0.0722,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013374968664720654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.690913963317872,
      "state_query_norm": 747.3611267089843,
      "state_relation_loss": 0.0013374968664720654,
      "step": 2890,
      "student_distance_mean": 943.8179565429688,
      "teacher_distance_mean": 1.139746403694153,
      "weighted_auxiliary_loss": 0.0013374968664720654,
      "weighted_relation_loss": 0.0013374968664720654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7128157156220767,
      "flow_matching_loss": 0.06890916228294372,
      "grad_norm": 0.21382732689380646,
      "learning_rate": 9.94066313475481e-05,
      "loss": 0.0702,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013329016510397196,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.143061637878418,
      "state_query_norm": 749.5222900390625,
      "state_relation_loss": 0.0013329016510397196,
      "step": 2900,
      "student_distance_mean": 965.5879699707032,
      "teacher_distance_mean": 1.160078799724579,
      "weighted_auxiliary_loss": 0.0013329016510397196,
      "weighted_relation_loss": 0.0013329016510397196
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.722170252572498,
      "flow_matching_loss": 0.07100431770086288,
      "grad_norm": 0.1838572770357132,
      "learning_rate": 9.939813539621857e-05,
      "loss": 0.0723,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013296561548486352,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.61986598968506,
      "state_query_norm": 736.8249328613281,
      "state_relation_loss": 0.0013296561548486352,
      "step": 2910,
      "student_distance_mean": 941.1138671875,
      "teacher_distance_mean": 1.1404417991638183,
      "weighted_auxiliary_loss": 0.0013296561548486352,
      "weighted_relation_loss": 0.0013296561548486352
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7315247895229184,
      "flow_matching_loss": 0.06963260844349861,
      "grad_norm": 0.20860116183757782,
      "learning_rate": 9.93895794214966e-05,
      "loss": 0.0709,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012815836467780172,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.62027931213379,
      "state_query_norm": 728.323876953125,
      "state_relation_loss": 0.0012815836467780172,
      "step": 2920,
      "student_distance_mean": 938.8585021972656,
      "teacher_distance_mean": 1.1355662226676941,
      "weighted_auxiliary_loss": 0.0012815836467780172,
      "weighted_relation_loss": 0.0012815836467780172
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7408793264733395,
      "flow_matching_loss": 0.06827115863561631,
      "grad_norm": 0.3129938542842865,
      "learning_rate": 9.938096343377852e-05,
      "loss": 0.0696,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001330548233818263,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.693357086181642,
      "state_query_norm": 740.6237243652344,
      "state_relation_loss": 0.001330548233818263,
      "step": 2930,
      "student_distance_mean": 945.5949340820313,
      "teacher_distance_mean": 1.1343170166015626,
      "weighted_auxiliary_loss": 0.001330548233818263,
      "weighted_relation_loss": 0.001330548233818263
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7502338634237606,
      "flow_matching_loss": 0.06887351684272289,
      "grad_norm": 0.2434074878692627,
      "learning_rate": 9.937228744353353e-05,
      "loss": 0.0702,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013336451025679707,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.00752124786377,
      "state_query_norm": 749.4504211425781,
      "state_relation_loss": 0.0013336451025679707,
      "step": 2940,
      "student_distance_mean": 960.3546875,
      "teacher_distance_mean": 1.1449601292610168,
      "weighted_auxiliary_loss": 0.0013336451025679707,
      "weighted_relation_loss": 0.0013336451025679707
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7595884003741817,
      "flow_matching_loss": 0.06832868009805679,
      "grad_norm": 0.18562304973602295,
      "learning_rate": 9.936355146130379e-05,
      "loss": 0.0697,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013615385163575412,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.16761474609375,
      "state_query_norm": 759.7364501953125,
      "state_relation_loss": 0.0013615385163575412,
      "step": 2950,
      "student_distance_mean": 970.7710205078125,
      "teacher_distance_mean": 1.1503335118293763,
      "weighted_auxiliary_loss": 0.0013615385163575412,
      "weighted_relation_loss": 0.0013615385163575412
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7689429373246024,
      "flow_matching_loss": 0.07120953872799873,
      "grad_norm": 0.22258637845516205,
      "learning_rate": 9.935475549770435e-05,
      "loss": 0.0725,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013376794988289476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.099216842651366,
      "state_query_norm": 758.4655639648438,
      "state_relation_loss": 0.0013376794988289476,
      "step": 2960,
      "student_distance_mean": 968.0466003417969,
      "teacher_distance_mean": 1.1453680634498595,
      "weighted_auxiliary_loss": 0.0013376794988289476,
      "weighted_relation_loss": 0.0013376794988289476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7782974742750235,
      "flow_matching_loss": 0.06838905066251755,
      "grad_norm": 0.24474599957466125,
      "learning_rate": 9.934589956342315e-05,
      "loss": 0.0697,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012995674973353744,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.43121452331543,
      "state_query_norm": 769.9009460449219,
      "state_relation_loss": 0.0012995674973353744,
      "step": 2970,
      "student_distance_mean": 986.5227905273438,
      "teacher_distance_mean": 1.1628435611724854,
      "weighted_auxiliary_loss": 0.0012995674973353744,
      "weighted_relation_loss": 0.0012995674973353744
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.787652011225444,
      "flow_matching_loss": 0.07089051455259324,
      "grad_norm": 0.22899994254112244,
      "learning_rate": 9.933698366922093e-05,
      "loss": 0.0722,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012861897703260183,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.99616470336914,
      "state_query_norm": 742.9482482910156,
      "state_relation_loss": 0.0012861897703260183,
      "step": 2980,
      "student_distance_mean": 960.8915466308594,
      "teacher_distance_mean": 1.1375767111778259,
      "weighted_auxiliary_loss": 0.0012861897703260183,
      "weighted_relation_loss": 0.0012861897703260183
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.7970065481758652,
      "flow_matching_loss": 0.07134996056556701,
      "grad_norm": 0.2347521036863327,
      "learning_rate": 9.93280078259314e-05,
      "loss": 0.0728,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0014201165875419974,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.22537269592285,
      "state_query_norm": 748.7498046875,
      "state_relation_loss": 0.0014201165875419974,
      "step": 2990,
      "student_distance_mean": 969.5734375,
      "teacher_distance_mean": 1.1527654767036437,
      "weighted_auxiliary_loss": 0.0014201165875419974,
      "weighted_relation_loss": 0.0014201165875419974
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.8063610851262863,
      "flow_matching_loss": 0.07102693468332291,
      "grad_norm": 0.18199124932289124,
      "learning_rate": 9.931897204446104e-05,
      "loss": 0.0724,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001360873063094914,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.011366653442384,
      "state_query_norm": 741.7423828125,
      "state_relation_loss": 0.001360873063094914,
      "step": 3000,
      "student_distance_mean": 961.86689453125,
      "teacher_distance_mean": 1.1462706804275513,
      "weighted_auxiliary_loss": 0.001360873063094914,
      "weighted_relation_loss": 0.001360873063094914
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.8157156220767074,
      "flow_matching_loss": 0.06823561154305935,
      "grad_norm": 0.18902255594730377,
      "learning_rate": 9.930987633578915e-05,
      "loss": 0.0696,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001357497915159911,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.258910179138184,
      "state_query_norm": 758.6583618164062,
      "state_relation_loss": 0.001357497915159911,
      "step": 3010,
      "student_distance_mean": 976.1985595703125,
      "teacher_distance_mean": 1.1547426104545593,
      "weighted_auxiliary_loss": 0.001357497915159911,
      "weighted_relation_loss": 0.001357497915159911
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.825070159027128,
      "flow_matching_loss": 0.06816608123481274,
      "grad_norm": 0.15312491357326508,
      "learning_rate": 9.93007207109679e-05,
      "loss": 0.0696,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001446070394013077,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 16.964902687072755,
      "state_query_norm": 739.7635009765625,
      "state_relation_loss": 0.001446070394013077,
      "step": 3020,
      "student_distance_mean": 958.6951293945312,
      "teacher_distance_mean": 1.1437357664108276,
      "weighted_auxiliary_loss": 0.001446070394013077,
      "weighted_relation_loss": 0.001446070394013077
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.834424695977549,
      "flow_matching_loss": 0.07015755847096443,
      "grad_norm": 0.17363744974136353,
      "learning_rate": 9.929150518112224e-05,
      "loss": 0.0715,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013734745560213923,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.09639263153076,
      "state_query_norm": 753.3534851074219,
      "state_relation_loss": 0.0013734745560213923,
      "step": 3030,
      "student_distance_mean": 967.2445129394531,
      "teacher_distance_mean": 1.1426693201065063,
      "weighted_auxiliary_loss": 0.0013734745560213923,
      "weighted_relation_loss": 0.0013734745560213923
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.84377923292797,
      "flow_matching_loss": 0.0676470085978508,
      "grad_norm": 0.13971447944641113,
      "learning_rate": 9.928222975744991e-05,
      "loss": 0.069,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013251700322143734,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.563316345214844,
      "state_query_norm": 765.8498291015625,
      "state_relation_loss": 0.0013251700322143734,
      "step": 3040,
      "student_distance_mean": 992.5040466308594,
      "teacher_distance_mean": 1.167258095741272,
      "weighted_auxiliary_loss": 0.0013251700322143734,
      "weighted_relation_loss": 0.0013251700322143734
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.853133769878391,
      "flow_matching_loss": 0.06816319338977336,
      "grad_norm": 0.17844268679618835,
      "learning_rate": 9.92728944512214e-05,
      "loss": 0.0695,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013181876158341766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.45041160583496,
      "state_query_norm": 764.7248779296875,
      "state_relation_loss": 0.0013181876158341766,
      "step": 3050,
      "student_distance_mean": 986.3061401367188,
      "teacher_distance_mean": 1.149627447128296,
      "weighted_auxiliary_loss": 0.0013181876158341766,
      "weighted_relation_loss": 0.0013181876158341766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.862488306828812,
      "flow_matching_loss": 0.07002448737621307,
      "grad_norm": 0.12627464532852173,
      "learning_rate": 9.926349927378001e-05,
      "loss": 0.0714,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013263702508993446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.36690502166748,
      "state_query_norm": 756.4324157714843,
      "state_relation_loss": 0.0013263702508993446,
      "step": 3060,
      "student_distance_mean": 981.3261291503907,
      "teacher_distance_mean": 1.1401768565177917,
      "weighted_auxiliary_loss": 0.0013263702508993446,
      "weighted_relation_loss": 0.0013263702508993446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.871842843779233,
      "flow_matching_loss": 0.06658383123576642,
      "grad_norm": 0.1452697366476059,
      "learning_rate": 9.925404423654174e-05,
      "loss": 0.0679,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013586997403763235,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.60118999481201,
      "state_query_norm": 771.1363403320313,
      "state_relation_loss": 0.0013586997403763235,
      "step": 3070,
      "student_distance_mean": 992.7151184082031,
      "teacher_distance_mean": 1.1606330513954162,
      "weighted_auxiliary_loss": 0.0013586997403763235,
      "weighted_relation_loss": 0.0013586997403763235
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.881197380729654,
      "flow_matching_loss": 0.06954890489578247,
      "grad_norm": 0.21808265149593353,
      "learning_rate": 9.924452935099537e-05,
      "loss": 0.0709,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013154686777852476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.35814151763916,
      "state_query_norm": 764.3766967773438,
      "state_relation_loss": 0.0013154686777852476,
      "step": 3080,
      "student_distance_mean": 982.2410339355469,
      "teacher_distance_mean": 1.1532782793045044,
      "weighted_auxiliary_loss": 0.0013154686777852476,
      "weighted_relation_loss": 0.0013154686777852476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.890551917680075,
      "flow_matching_loss": 0.06521319188177585,
      "grad_norm": 0.203239306807518,
      "learning_rate": 9.92349546287024e-05,
      "loss": 0.0665,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0013275943812914193,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.48331241607666,
      "state_query_norm": 768.6451354980469,
      "state_relation_loss": 0.0013275943812914193,
      "step": 3090,
      "student_distance_mean": 985.3960205078125,
      "teacher_distance_mean": 1.1511108040809632,
      "weighted_auxiliary_loss": 0.0013275943812914193,
      "weighted_relation_loss": 0.0013275943812914193
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.8999064546304956,
      "flow_matching_loss": 0.06627746298909187,
      "grad_norm": 0.1669435203075409,
      "learning_rate": 9.9225320081297e-05,
      "loss": 0.0675,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012442453065887094,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.554809188842775,
      "state_query_norm": 766.2998474121093,
      "state_relation_loss": 0.0012442453065887094,
      "step": 3100,
      "student_distance_mean": 992.0640014648437,
      "teacher_distance_mean": 1.147732424736023,
      "weighted_auxiliary_loss": 0.0012442453065887094,
      "weighted_relation_loss": 0.0012442453065887094
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9092609915809167,
      "flow_matching_loss": 0.06659607328474522,
      "grad_norm": 0.19494228065013885,
      "learning_rate": 9.921562572048606e-05,
      "loss": 0.0678,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001211655093356967,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.64340114593506,
      "state_query_norm": 764.4751892089844,
      "state_relation_loss": 0.001211655093356967,
      "step": 3110,
      "student_distance_mean": 994.7200500488282,
      "teacher_distance_mean": 1.1512393951416016,
      "weighted_auxiliary_loss": 0.001211655093356967,
      "weighted_relation_loss": 0.001211655093356967
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9186155285313378,
      "flow_matching_loss": 0.0652515284717083,
      "grad_norm": 0.22887766361236572,
      "learning_rate": 9.920587155804913e-05,
      "loss": 0.0665,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012446220964193345,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.59754161834717,
      "state_query_norm": 767.8043212890625,
      "state_relation_loss": 0.0012446220964193345,
      "step": 3120,
      "student_distance_mean": 995.6300598144531,
      "teacher_distance_mean": 1.1573487162590026,
      "weighted_auxiliary_loss": 0.0012446220964193345,
      "weighted_relation_loss": 0.0012446220964193345
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.927970065481759,
      "flow_matching_loss": 0.06827176697552204,
      "grad_norm": 0.24378149211406708,
      "learning_rate": 9.919605760583845e-05,
      "loss": 0.0695,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012441562954336404,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.487155723571778,
      "state_query_norm": 766.5017211914062,
      "state_relation_loss": 0.0012441562954336404,
      "step": 3130,
      "student_distance_mean": 988.3407104492187,
      "teacher_distance_mean": 1.1497314691543579,
      "weighted_auxiliary_loss": 0.0012441562954336404,
      "weighted_relation_loss": 0.0012441562954336404
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9373246024321795,
      "flow_matching_loss": 0.06499265395104885,
      "grad_norm": 0.2363097220659256,
      "learning_rate": 9.91861838757789e-05,
      "loss": 0.0662,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001222050259821117,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.83629093170166,
      "state_query_norm": 784.0715637207031,
      "state_relation_loss": 0.001222050259821117,
      "step": 3140,
      "student_distance_mean": 1009.2415161132812,
      "teacher_distance_mean": 1.1634830117225647,
      "weighted_auxiliary_loss": 0.001222050259821117,
      "weighted_relation_loss": 0.001222050259821117
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9466791393826006,
      "flow_matching_loss": 0.06832056194543838,
      "grad_norm": 0.29218071699142456,
      "learning_rate": 9.917625037986798e-05,
      "loss": 0.0695,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011972142499871552,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.598493003845213,
      "state_query_norm": 772.1787719726562,
      "state_relation_loss": 0.0011972142499871552,
      "step": 3150,
      "student_distance_mean": 994.1703430175781,
      "teacher_distance_mean": 1.1384762883186341,
      "weighted_auxiliary_loss": 0.0011972142499871552,
      "weighted_relation_loss": 0.0011972142499871552
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9560336763330213,
      "flow_matching_loss": 0.06634801253676414,
      "grad_norm": 0.23822364211082458,
      "learning_rate": 9.916625713017583e-05,
      "loss": 0.0676,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012426610104739667,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.50737056732178,
      "state_query_norm": 761.5946655273438,
      "state_relation_loss": 0.0012426610104739667,
      "step": 3160,
      "student_distance_mean": 987.8262939453125,
      "teacher_distance_mean": 1.1388943910598754,
      "weighted_auxiliary_loss": 0.0012426610104739667,
      "weighted_relation_loss": 0.0012426610104739667
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9653882132834424,
      "flow_matching_loss": 0.06687211096286774,
      "grad_norm": 0.20045948028564453,
      "learning_rate": 9.915620413884519e-05,
      "loss": 0.068,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011755022802390157,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.76091537475586,
      "state_query_norm": 776.0143005371094,
      "state_relation_loss": 0.0011755022802390157,
      "step": 3170,
      "student_distance_mean": 1003.8632446289063,
      "teacher_distance_mean": 1.1465513586997986,
      "weighted_auxiliary_loss": 0.0011755022802390157,
      "weighted_relation_loss": 0.0011755022802390157
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9747427502338635,
      "flow_matching_loss": 0.06779996939003467,
      "grad_norm": 0.19696946442127228,
      "learning_rate": 9.914609141809139e-05,
      "loss": 0.069,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012479233439080418,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.01971569061279,
      "state_query_norm": 784.7968383789063,
      "state_relation_loss": 0.0012479233439080418,
      "step": 3180,
      "student_distance_mean": 1018.2109375,
      "teacher_distance_mean": 1.15795738697052,
      "weighted_auxiliary_loss": 0.0012479233439080418,
      "weighted_relation_loss": 0.0012479233439080418
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9840972871842846,
      "flow_matching_loss": 0.07243924289941787,
      "grad_norm": 0.20755939185619354,
      "learning_rate": 9.913591898020235e-05,
      "loss": 0.0737,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001250332815106958,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.098137855529785,
      "state_query_norm": 792.6031005859375,
      "state_relation_loss": 0.001250332815106958,
      "step": 3190,
      "student_distance_mean": 1022.4997436523438,
      "teacher_distance_mean": 1.1641355752944946,
      "weighted_auxiliary_loss": 0.001250332815106958,
      "weighted_relation_loss": 0.001250332815106958
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 2.9934518241347052,
      "flow_matching_loss": 0.071672198548913,
      "grad_norm": 0.2249118685722351,
      "learning_rate": 9.912568683753853e-05,
      "loss": 0.0729,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012441449682228268,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.702602767944335,
      "state_query_norm": 779.1783996582031,
      "state_relation_loss": 0.0012441449682228268,
      "step": 3200,
      "student_distance_mean": 998.1238403320312,
      "teacher_distance_mean": 1.1475433826446533,
      "weighted_auxiliary_loss": 0.0012441449682228268,
      "weighted_relation_loss": 0.0012441449682228268
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.0028063610851263,
      "flow_matching_loss": 0.06772123910486698,
      "grad_norm": 0.24653439223766327,
      "learning_rate": 9.911539500253295e-05,
      "loss": 0.069,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0012485761311836542,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 18.11190586090088,
      "state_query_norm": 791.5013732910156,
      "state_relation_loss": 0.0012485761311836542,
      "step": 3210,
      "student_distance_mean": 1023.0475646972657,
      "teacher_distance_mean": 1.1717887043952941,
      "weighted_auxiliary_loss": 0.0012485761311836542,
      "weighted_relation_loss": 0.0012485761311836542
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.0121608980355474,
      "flow_matching_loss": 0.06677384860813618,
      "grad_norm": 0.1925474852323532,
      "learning_rate": 9.910504348769118e-05,
      "loss": 0.068,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012265856377780438,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.812471008300783,
      "state_query_norm": 777.0395751953125,
      "state_relation_loss": 0.0012265856377780438,
      "step": 3220,
      "student_distance_mean": 1005.1295654296875,
      "teacher_distance_mean": 1.140077543258667,
      "weighted_auxiliary_loss": 0.0012265856377780438,
      "weighted_relation_loss": 0.0012265856377780438
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.021515434985968,
      "flow_matching_loss": 0.06606617160141467,
      "grad_norm": 0.19625861942768097,
      "learning_rate": 9.909463230559127e-05,
      "loss": 0.0673,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012625584146007895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.97645511627197,
      "state_query_norm": 786.8817932128907,
      "state_relation_loss": 0.0012625584146007895,
      "step": 3230,
      "student_distance_mean": 1014.9067687988281,
      "teacher_distance_mean": 1.1556745171546936,
      "weighted_auxiliary_loss": 0.0012625584146007895,
      "weighted_relation_loss": 0.0012625584146007895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.030869971936389,
      "flow_matching_loss": 0.06891879998147488,
      "grad_norm": 0.1853356957435608,
      "learning_rate": 9.908416146888376e-05,
      "loss": 0.0702,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001278481283225119,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.058235740661623,
      "state_query_norm": 786.1998596191406,
      "state_relation_loss": 0.001278481283225119,
      "step": 3240,
      "student_distance_mean": 1017.3999755859375,
      "teacher_distance_mean": 1.1561914205551147,
      "weighted_auxiliary_loss": 0.001278481283225119,
      "weighted_relation_loss": 0.001278481283225119
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.0402245088868103,
      "flow_matching_loss": 0.06775466986000538,
      "grad_norm": 0.16392414271831512,
      "learning_rate": 9.907363099029175e-05,
      "loss": 0.069,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00129102022619918,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.692810821533204,
      "state_query_norm": 771.8754455566407,
      "state_relation_loss": 0.00129102022619918,
      "step": 3250,
      "student_distance_mean": 997.642041015625,
      "teacher_distance_mean": 1.149860441684723,
      "weighted_auxiliary_loss": 0.00129102022619918,
      "weighted_relation_loss": 0.00129102022619918
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.049579045837231,
      "flow_matching_loss": 0.06745827198028564,
      "grad_norm": 0.12411076575517654,
      "learning_rate": 9.906304088261073e-05,
      "loss": 0.0688,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001291971025057137,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.920029640197754,
      "state_query_norm": 785.8454040527344,
      "state_relation_loss": 0.001291971025057137,
      "step": 3260,
      "student_distance_mean": 1011.9639892578125,
      "teacher_distance_mean": 1.152755868434906,
      "weighted_auxiliary_loss": 0.001291971025057137,
      "weighted_relation_loss": 0.001291971025057137
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.058933582787652,
      "flow_matching_loss": 0.06795569062232971,
      "grad_norm": 0.14712034165859222,
      "learning_rate": 9.905239115870872e-05,
      "loss": 0.0692,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012527968152426182,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.201232719421387,
      "state_query_norm": 797.7730834960937,
      "state_relation_loss": 0.0012527968152426182,
      "step": 3270,
      "student_distance_mean": 1029.0849365234376,
      "teacher_distance_mean": 1.1637607574462892,
      "weighted_auxiliary_loss": 0.0012527968152426182,
      "weighted_relation_loss": 0.0012527968152426182
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.068288119738073,
      "flow_matching_loss": 0.06937971636652947,
      "grad_norm": 0.22718903422355652,
      "learning_rate": 9.90416818315261e-05,
      "loss": 0.0706,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012173073482699692,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.56394805908203,
      "state_query_norm": 779.6317565917968,
      "state_relation_loss": 0.0012173073482699692,
      "step": 3280,
      "student_distance_mean": 995.4809326171875,
      "teacher_distance_mean": 1.1279528975486754,
      "weighted_auxiliary_loss": 0.0012173073482699692,
      "weighted_relation_loss": 0.0012173073482699692
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.077642656688494,
      "flow_matching_loss": 0.06407920196652413,
      "grad_norm": 0.14668308198451996,
      "learning_rate": 9.903091291407573e-05,
      "loss": 0.0653,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012487764586694538,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.278777885437012,
      "state_query_norm": 794.1041564941406,
      "state_relation_loss": 0.0012487764586694538,
      "step": 3290,
      "student_distance_mean": 1028.8852172851562,
      "teacher_distance_mean": 1.1765576720237731,
      "weighted_auxiliary_loss": 0.0012487764586694538,
      "weighted_relation_loss": 0.0012487764586694538
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.086997193638915,
      "flow_matching_loss": 0.06886253878474236,
      "grad_norm": 0.15745584666728973,
      "learning_rate": 9.902008441944286e-05,
      "loss": 0.0701,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001217026577796787,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.864340400695802,
      "state_query_norm": 778.1311462402343,
      "state_relation_loss": 0.001217026577796787,
      "step": 3300,
      "student_distance_mean": 1007.9912902832032,
      "teacher_distance_mean": 1.1521009087562561,
      "weighted_auxiliary_loss": 0.001217026577796787,
      "weighted_relation_loss": 0.001217026577796787
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.096351730589336,
      "flow_matching_loss": 0.06298879384994507,
      "grad_norm": 0.21580487489700317,
      "learning_rate": 9.900919636078512e-05,
      "loss": 0.0643,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001266385184135288,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 17.939774131774904,
      "state_query_norm": 785.9353149414062,
      "state_relation_loss": 0.001266385184135288,
      "step": 3310,
      "student_distance_mean": 1013.5968383789062,
      "teacher_distance_mean": 1.1430381298065186,
      "weighted_auxiliary_loss": 0.001266385184135288,
      "weighted_relation_loss": 0.001266385184135288
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1057062675397566,
      "flow_matching_loss": 0.0651952538639307,
      "grad_norm": 0.1600891649723053,
      "learning_rate": 9.899824875133255e-05,
      "loss": 0.0664,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001235263084527105,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.51973514556885,
      "state_query_norm": 810.4757019042969,
      "state_relation_loss": 0.001235263084527105,
      "step": 3320,
      "student_distance_mean": 1045.277520751953,
      "teacher_distance_mean": 1.1609339118003845,
      "weighted_auxiliary_loss": 0.001235263084527105,
      "weighted_relation_loss": 0.001235263084527105
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1150608044901777,
      "flow_matching_loss": 0.06542024128139019,
      "grad_norm": 0.1785973608493805,
      "learning_rate": 9.898724160438749e-05,
      "loss": 0.0666,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012270602164790034,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.11857452392578,
      "state_query_norm": 788.8426635742187,
      "state_relation_loss": 0.0012270602164790034,
      "step": 3330,
      "student_distance_mean": 1023.3915954589844,
      "teacher_distance_mean": 1.144622266292572,
      "weighted_auxiliary_loss": 0.0012270602164790034,
      "weighted_relation_loss": 0.0012270602164790034
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.124415341440599,
      "flow_matching_loss": 0.06636526063084602,
      "grad_norm": 0.14499658346176147,
      "learning_rate": 9.89761749333247e-05,
      "loss": 0.0675,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001173705910332501,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.129253959655763,
      "state_query_norm": 786.5108215332032,
      "state_relation_loss": 0.001173705910332501,
      "step": 3340,
      "student_distance_mean": 1022.1250305175781,
      "teacher_distance_mean": 1.145331120491028,
      "weighted_auxiliary_loss": 0.001173705910332501,
      "weighted_relation_loss": 0.001173705910332501
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1337698783910195,
      "flow_matching_loss": 0.06501750163733959,
      "grad_norm": 0.14336048066616058,
      "learning_rate": 9.896504875159122e-05,
      "loss": 0.0662,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001176670112181455,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.25575523376465,
      "state_query_norm": 795.5326843261719,
      "state_relation_loss": 0.001176670112181455,
      "step": 3350,
      "student_distance_mean": 1029.0236877441407,
      "teacher_distance_mean": 1.1545001149177552,
      "weighted_auxiliary_loss": 0.001176670112181455,
      "weighted_relation_loss": 0.001176670112181455
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1431244153414406,
      "flow_matching_loss": 0.07073229067027569,
      "grad_norm": 0.16259858012199402,
      "learning_rate": 9.89538630727064e-05,
      "loss": 0.0719,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001193944166880101,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.10523338317871,
      "state_query_norm": 790.165185546875,
      "state_relation_loss": 0.001193944166880101,
      "step": 3360,
      "student_distance_mean": 1021.0081298828125,
      "teacher_distance_mean": 1.1495692491531373,
      "weighted_auxiliary_loss": 0.001193944166880101,
      "weighted_relation_loss": 0.001193944166880101
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1524789522918617,
      "flow_matching_loss": 0.06739816702902317,
      "grad_norm": 0.21466104686260223,
      "learning_rate": 9.894261791026189e-05,
      "loss": 0.0686,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012157735996879636,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.290324401855468,
      "state_query_norm": 796.6047607421875,
      "state_relation_loss": 0.0012157735996879636,
      "step": 3370,
      "student_distance_mean": 1029.605535888672,
      "teacher_distance_mean": 1.1600698590278626,
      "weighted_auxiliary_loss": 0.0012157735996879636,
      "weighted_relation_loss": 0.0012157735996879636
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1618334892422824,
      "flow_matching_loss": 0.06560203731060028,
      "grad_norm": 0.24754318594932556,
      "learning_rate": 9.893131327792165e-05,
      "loss": 0.0668,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00116726397536695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.247476196289064,
      "state_query_norm": 795.506689453125,
      "state_relation_loss": 0.00116726397536695,
      "step": 3380,
      "student_distance_mean": 1029.0518127441405,
      "teacher_distance_mean": 1.1440191507339477,
      "weighted_auxiliary_loss": 0.00116726397536695,
      "weighted_relation_loss": 0.00116726397536695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1711880261927035,
      "flow_matching_loss": 0.06652941703796386,
      "grad_norm": 0.35295552015304565,
      "learning_rate": 9.891994918942182e-05,
      "loss": 0.0678,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012337747495621443,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.388595199584962,
      "state_query_norm": 807.2938293457031,
      "state_relation_loss": 0.0012337747495621443,
      "step": 3390,
      "student_distance_mean": 1038.2673828125,
      "teacher_distance_mean": 1.1544067859649658,
      "weighted_auxiliary_loss": 0.0012337747495621443,
      "weighted_relation_loss": 0.0012337747495621443
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1805425631431246,
      "flow_matching_loss": 0.06803090125322342,
      "grad_norm": 0.18290089070796967,
      "learning_rate": 9.890852565857092e-05,
      "loss": 0.0692,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012148475274443627,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.61448230743408,
      "state_query_norm": 824.1489990234375,
      "state_relation_loss": 0.0012148475274443627,
      "step": 3400,
      "student_distance_mean": 1054.316339111328,
      "teacher_distance_mean": 1.1676112771034242,
      "weighted_auxiliary_loss": 0.0012148475274443627,
      "weighted_relation_loss": 0.0012148475274443627
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.189897100093545,
      "flow_matching_loss": 0.06377182900905609,
      "grad_norm": 0.17585782706737518,
      "learning_rate": 9.889704269924954e-05,
      "loss": 0.0649,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011693202890455724,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.673690605163575,
      "state_query_norm": 818.5699584960937,
      "state_relation_loss": 0.0011693202890455724,
      "step": 3410,
      "student_distance_mean": 1053.7814453125,
      "teacher_distance_mean": 1.1685713410377503,
      "weighted_auxiliary_loss": 0.0011693202890455724,
      "weighted_relation_loss": 0.0011693202890455724
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.1992516370439663,
      "flow_matching_loss": 0.06557110771536827,
      "grad_norm": 0.18703477084636688,
      "learning_rate": 9.888550032541059e-05,
      "loss": 0.0668,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012530556414276361,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.435154342651366,
      "state_query_norm": 800.2729309082031,
      "state_relation_loss": 0.0012530556414276361,
      "step": 3420,
      "student_distance_mean": 1040.2571716308594,
      "teacher_distance_mean": 1.1597554206848144,
      "weighted_auxiliary_loss": 0.0012530556414276361,
      "weighted_relation_loss": 0.0012530556414276361
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.2086061739943874,
      "flow_matching_loss": 0.06706833019852639,
      "grad_norm": 0.2032107263803482,
      "learning_rate": 9.887389855107916e-05,
      "loss": 0.0683,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001245393615681678,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.33184642791748,
      "state_query_norm": 795.8142150878906,
      "state_relation_loss": 0.001245393615681678,
      "step": 3430,
      "student_distance_mean": 1028.68642578125,
      "teacher_distance_mean": 1.1545194387435913,
      "weighted_auxiliary_loss": 0.001245393615681678,
      "weighted_relation_loss": 0.001245393615681678
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.217960710944808,
      "flow_matching_loss": 0.0664651270955801,
      "grad_norm": 0.21406608819961548,
      "learning_rate": 9.886223739035248e-05,
      "loss": 0.0676,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011588625959120691,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.537525177001953,
      "state_query_norm": 815.0171447753906,
      "state_relation_loss": 0.0011588625959120691,
      "step": 3440,
      "student_distance_mean": 1048.4635131835937,
      "teacher_distance_mean": 1.1531961679458618,
      "weighted_auxiliary_loss": 0.0011588625959120691,
      "weighted_relation_loss": 0.0011588625959120691
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.227315247895229,
      "flow_matching_loss": 0.06401881873607636,
      "grad_norm": 0.20220090448856354,
      "learning_rate": 9.885051685739997e-05,
      "loss": 0.0652,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011849697912111878,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.68363857269287,
      "state_query_norm": 814.8586853027343,
      "state_relation_loss": 0.0011849697912111878,
      "step": 3450,
      "student_distance_mean": 1052.8892333984375,
      "teacher_distance_mean": 1.1503219127655029,
      "weighted_auxiliary_loss": 0.0011849697912111878,
      "weighted_relation_loss": 0.0011849697912111878
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.2366697848456503,
      "flow_matching_loss": 0.06534346230328084,
      "grad_norm": 0.2544252574443817,
      "learning_rate": 9.883873696646316e-05,
      "loss": 0.0666,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001289498677942902,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.819431877136232,
      "state_query_norm": 810.6829406738282,
      "state_relation_loss": 0.001289498677942902,
      "step": 3460,
      "student_distance_mean": 1058.3991271972657,
      "teacher_distance_mean": 1.1617146492004395,
      "weighted_auxiliary_loss": 0.001289498677942902,
      "weighted_relation_loss": 0.001289498677942902
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.246024321796071,
      "flow_matching_loss": 0.065619732812047,
      "grad_norm": 0.232905313372612,
      "learning_rate": 9.882689773185575e-05,
      "loss": 0.0669,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012666449882090091,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.726467895507813,
      "state_query_norm": 810.3847900390625,
      "state_relation_loss": 0.0012666449882090091,
      "step": 3470,
      "student_distance_mean": 1057.8032104492188,
      "teacher_distance_mean": 1.1601975798606872,
      "weighted_auxiliary_loss": 0.0012666449882090091,
      "weighted_relation_loss": 0.0012666449882090091
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.255378858746492,
      "flow_matching_loss": 0.06765155903995038,
      "grad_norm": 0.19734831154346466,
      "learning_rate": 9.881499916796353e-05,
      "loss": 0.0689,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012323539471253754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.91249351501465,
      "state_query_norm": 814.4706420898438,
      "state_relation_loss": 0.0012323539471253754,
      "step": 3480,
      "student_distance_mean": 1062.9320983886719,
      "teacher_distance_mean": 1.1578136205673217,
      "weighted_auxiliary_loss": 0.0012323539471253754,
      "weighted_relation_loss": 0.0012323539471253754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.264733395696913,
      "flow_matching_loss": 0.06408189125359058,
      "grad_norm": 0.21427327394485474,
      "learning_rate": 9.880304128924434e-05,
      "loss": 0.0653,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001225099165458232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.456141662597656,
      "state_query_norm": 802.8167541503906,
      "state_relation_loss": 0.001225099165458232,
      "step": 3490,
      "student_distance_mean": 1040.8529235839844,
      "teacher_distance_mean": 1.1487868309020997,
      "weighted_auxiliary_loss": 0.001225099165458232,
      "weighted_relation_loss": 0.001225099165458232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.2740879326473338,
      "flow_matching_loss": 0.06555304527282715,
      "grad_norm": 0.19275851547718048,
      "learning_rate": 9.879102411022817e-05,
      "loss": 0.0667,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011636224575340749,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.519970703125,
      "state_query_norm": 811.1768859863281,
      "state_relation_loss": 0.0011636224575340749,
      "step": 3500,
      "student_distance_mean": 1047.565496826172,
      "teacher_distance_mean": 1.149129903316498,
      "weighted_auxiliary_loss": 0.0011636224575340749,
      "weighted_relation_loss": 0.0011636224575340749
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.283442469597755,
      "flow_matching_loss": 0.06278864294290543,
      "grad_norm": 0.18517856299877167,
      "learning_rate": 9.877894764551703e-05,
      "loss": 0.0639,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011472117737866938,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.416942977905272,
      "state_query_norm": 792.5243957519531,
      "state_relation_loss": 0.0011472117737866938,
      "step": 3510,
      "student_distance_mean": 1036.2966186523438,
      "teacher_distance_mean": 1.1454955816268921,
      "weighted_auxiliary_loss": 0.0011472117737866938,
      "weighted_relation_loss": 0.0011472117737866938
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.292797006548176,
      "flow_matching_loss": 0.0647862870246172,
      "grad_norm": 0.16901737451553345,
      "learning_rate": 9.876681190978494e-05,
      "loss": 0.0659,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00112664116313681,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.743206405639647,
      "state_query_norm": 804.1699584960937,
      "state_relation_loss": 0.00112664116313681,
      "step": 3520,
      "student_distance_mean": 1055.827392578125,
      "teacher_distance_mean": 1.1488818287849427,
      "weighted_auxiliary_loss": 0.00112664116313681,
      "weighted_relation_loss": 0.00112664116313681
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3021515434985966,
      "flow_matching_loss": 0.0644435778260231,
      "grad_norm": 0.1676226407289505,
      "learning_rate": 9.875461691777797e-05,
      "loss": 0.0657,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012084480375051498,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.947247314453126,
      "state_query_norm": 825.1987365722656,
      "state_relation_loss": 0.0012084480375051498,
      "step": 3530,
      "student_distance_mean": 1069.3481384277343,
      "teacher_distance_mean": 1.1522369742393495,
      "weighted_auxiliary_loss": 0.0012084480375051498,
      "weighted_relation_loss": 0.0012084480375051498
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3115060804490177,
      "flow_matching_loss": 0.0667919110506773,
      "grad_norm": 0.12176376581192017,
      "learning_rate": 9.874236268431417e-05,
      "loss": 0.0679,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011556146782822907,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.541206550598144,
      "state_query_norm": 810.4483093261719,
      "state_relation_loss": 0.0011556146782822907,
      "step": 3540,
      "student_distance_mean": 1043.821795654297,
      "teacher_distance_mean": 1.1438045382499695,
      "weighted_auxiliary_loss": 0.0011556146782822907,
      "weighted_relation_loss": 0.0011556146782822907
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.320860617399439,
      "flow_matching_loss": 0.06607943400740623,
      "grad_norm": 0.1474902480840683,
      "learning_rate": 9.873004922428361e-05,
      "loss": 0.0673,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011867829016409814,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.34115924835205,
      "state_query_norm": 806.7633361816406,
      "state_relation_loss": 0.0011867829016409814,
      "step": 3550,
      "student_distance_mean": 1036.9686401367187,
      "teacher_distance_mean": 1.1367150902748109,
      "weighted_auxiliary_loss": 0.0011867829016409814,
      "weighted_relation_loss": 0.0011867829016409814
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.33021515434986,
      "flow_matching_loss": 0.0651279129087925,
      "grad_norm": 0.25279104709625244,
      "learning_rate": 9.871767655264829e-05,
      "loss": 0.0663,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012064375448971986,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.84895763397217,
      "state_query_norm": 814.2171142578125,
      "state_relation_loss": 0.0012064375448971986,
      "step": 3560,
      "student_distance_mean": 1061.91591796875,
      "teacher_distance_mean": 1.1509929060935975,
      "weighted_auxiliary_loss": 0.0012064375448971986,
      "weighted_relation_loss": 0.0012064375448971986
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3395696913002806,
      "flow_matching_loss": 0.06506252810359,
      "grad_norm": 0.1947307288646698,
      "learning_rate": 9.87052446844422e-05,
      "loss": 0.0662,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011435053078457714,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.56947193145752,
      "state_query_norm": 803.5981323242188,
      "state_relation_loss": 0.0011435053078457714,
      "step": 3570,
      "student_distance_mean": 1050.4497314453124,
      "teacher_distance_mean": 1.1443710803985596,
      "weighted_auxiliary_loss": 0.0011435053078457714,
      "weighted_relation_loss": 0.0011435053078457714
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3489242282507017,
      "flow_matching_loss": 0.06735938191413879,
      "grad_norm": 0.24682794511318207,
      "learning_rate": 9.869275363477122e-05,
      "loss": 0.0685,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011808783980086445,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.744368934631346,
      "state_query_norm": 816.0007141113281,
      "state_relation_loss": 0.0011808783980086445,
      "step": 3580,
      "student_distance_mean": 1055.0221252441406,
      "teacher_distance_mean": 1.1525105714797974,
      "weighted_auxiliary_loss": 0.0011808783980086445,
      "weighted_relation_loss": 0.0011808783980086445
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3582787652011223,
      "flow_matching_loss": 0.06520397551357746,
      "grad_norm": 0.20693610608577728,
      "learning_rate": 9.868020341881312e-05,
      "loss": 0.0664,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011641841381788253,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.803137397766115,
      "state_query_norm": 813.1369262695313,
      "state_relation_loss": 0.0011641841381788253,
      "step": 3590,
      "student_distance_mean": 1061.7736511230469,
      "teacher_distance_mean": 1.1631354212760925,
      "weighted_auxiliary_loss": 0.0011641841381788253,
      "weighted_relation_loss": 0.0011641841381788253
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3676333021515434,
      "flow_matching_loss": 0.06859066635370255,
      "grad_norm": 0.19800998270511627,
      "learning_rate": 9.866759405181765e-05,
      "loss": 0.0698,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012554575223475695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.870628356933594,
      "state_query_norm": 816.1632263183594,
      "state_relation_loss": 0.0012554575223475695,
      "step": 3600,
      "student_distance_mean": 1063.0932495117188,
      "teacher_distance_mean": 1.15861177444458,
      "weighted_auxiliary_loss": 0.0012554575223475695,
      "weighted_relation_loss": 0.0012554575223475695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3769878391019645,
      "flow_matching_loss": 0.06584264449775219,
      "grad_norm": 0.21940511465072632,
      "learning_rate": 9.865492554910633e-05,
      "loss": 0.0671,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012457569362595677,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.92666015625,
      "state_query_norm": 817.0292541503907,
      "state_relation_loss": 0.0012457569362595677,
      "step": 3610,
      "student_distance_mean": 1065.293115234375,
      "teacher_distance_mean": 1.1479739665985107,
      "weighted_auxiliary_loss": 0.0012457569362595677,
      "weighted_relation_loss": 0.0012457569362595677
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3863423760523856,
      "flow_matching_loss": 0.0662817344069481,
      "grad_norm": 0.21361415088176727,
      "learning_rate": 9.864219792607262e-05,
      "loss": 0.0676,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012772840680554509,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.143818283081053,
      "state_query_norm": 820.921533203125,
      "state_relation_loss": 0.0012772840680554509,
      "step": 3620,
      "student_distance_mean": 1076.5071533203125,
      "teacher_distance_mean": 1.1452356100082397,
      "weighted_auxiliary_loss": 0.0012772840680554509,
      "weighted_relation_loss": 0.0012772840680554509
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.3956969130028063,
      "flow_matching_loss": 0.06356234885752202,
      "grad_norm": 0.21495972573757172,
      "learning_rate": 9.862941119818177e-05,
      "loss": 0.0648,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012134063988924026,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.816412925720215,
      "state_query_norm": 810.943896484375,
      "state_relation_loss": 0.0012134063988924026,
      "step": 3630,
      "student_distance_mean": 1059.6951293945312,
      "teacher_distance_mean": 1.1352076053619384,
      "weighted_auxiliary_loss": 0.0012134063988924026,
      "weighted_relation_loss": 0.0012134063988924026
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.4050514499532274,
      "flow_matching_loss": 0.06578862443566322,
      "grad_norm": 0.1939457803964615,
      "learning_rate": 9.861656538097086e-05,
      "loss": 0.067,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011846694862470031,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.796840095520018,
      "state_query_norm": 813.5523315429688,
      "state_relation_loss": 0.0011846694862470031,
      "step": 3640,
      "student_distance_mean": 1061.0750122070312,
      "teacher_distance_mean": 1.1515138864517211,
      "weighted_auxiliary_loss": 0.0011846694862470031,
      "weighted_relation_loss": 0.0011846694862470031
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.414405986903648,
      "flow_matching_loss": 0.06379021555185319,
      "grad_norm": 0.19785863161087036,
      "learning_rate": 9.860366049004877e-05,
      "loss": 0.0649,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001130991883110255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.000416374206544,
      "state_query_norm": 820.9862121582031,
      "state_relation_loss": 0.001130991883110255,
      "step": 3650,
      "student_distance_mean": 1069.8442077636719,
      "teacher_distance_mean": 1.1519143223762511,
      "weighted_auxiliary_loss": 0.001130991883110255,
      "weighted_relation_loss": 0.001130991883110255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.423760523854069,
      "flow_matching_loss": 0.06445912122726441,
      "grad_norm": 0.19551976025104523,
      "learning_rate": 9.859069654109615e-05,
      "loss": 0.0656,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011454757885076105,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.120783615112305,
      "state_query_norm": 830.112939453125,
      "state_relation_loss": 0.0011454757885076105,
      "step": 3660,
      "student_distance_mean": 1080.1843383789062,
      "teacher_distance_mean": 1.1579799771308898,
      "weighted_auxiliary_loss": 0.0011454757885076105,
      "weighted_relation_loss": 0.0011454757885076105
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.4331150608044902,
      "flow_matching_loss": 0.062251859530806544,
      "grad_norm": 0.13259389996528625,
      "learning_rate": 9.857767354986545e-05,
      "loss": 0.0634,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001172226876951754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.054544830322264,
      "state_query_norm": 818.6254943847656,
      "state_relation_loss": 0.001172226876951754,
      "step": 3670,
      "student_distance_mean": 1072.501025390625,
      "teacher_distance_mean": 1.1493220210075379,
      "weighted_auxiliary_loss": 0.001172226876951754,
      "weighted_relation_loss": 0.001172226876951754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.4424695977549113,
      "flow_matching_loss": 0.061817314848303793,
      "grad_norm": 0.2050231248140335,
      "learning_rate": 9.856459153218078e-05,
      "loss": 0.063,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011658877017907798,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.921368408203126,
      "state_query_norm": 816.7981994628906,
      "state_relation_loss": 0.0011658877017907798,
      "step": 3680,
      "student_distance_mean": 1063.4130737304688,
      "teacher_distance_mean": 1.144253957271576,
      "weighted_auxiliary_loss": 0.0011658877017907798,
      "weighted_relation_loss": 0.0011658877017907798
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.451824134705332,
      "flow_matching_loss": 0.06460364423692226,
      "grad_norm": 0.22126713395118713,
      "learning_rate": 9.855145050393808e-05,
      "loss": 0.0658,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001215158950071782,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.97944030761719,
      "state_query_norm": 828.0836303710937,
      "state_relation_loss": 0.001215158950071782,
      "step": 3690,
      "student_distance_mean": 1070.732421875,
      "teacher_distance_mean": 1.1388055086135864,
      "weighted_auxiliary_loss": 0.001215158950071782,
      "weighted_relation_loss": 0.001215158950071782
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.461178671655753,
      "flow_matching_loss": 0.0650600802153349,
      "grad_norm": 0.19469167292118073,
      "learning_rate": 9.85382504811049e-05,
      "loss": 0.0662,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011815589386969805,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.29099349975586,
      "state_query_norm": 826.9976379394532,
      "state_relation_loss": 0.0011815589386969805,
      "step": 3700,
      "student_distance_mean": 1086.125927734375,
      "teacher_distance_mean": 1.1604084730148316,
      "weighted_auxiliary_loss": 0.0011815589386969805,
      "weighted_relation_loss": 0.0011815589386969805
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.4705332086061738,
      "flow_matching_loss": 0.06622116379439831,
      "grad_norm": 0.17048858106136322,
      "learning_rate": 9.852499147972054e-05,
      "loss": 0.0674,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011997393798083068,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.891605186462403,
      "state_query_norm": 823.4629211425781,
      "state_relation_loss": 0.0011997393798083068,
      "step": 3710,
      "student_distance_mean": 1067.0478576660157,
      "teacher_distance_mean": 1.144926655292511,
      "weighted_auxiliary_loss": 0.0011997393798083068,
      "weighted_relation_loss": 0.0011997393798083068
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.479887745556595,
      "flow_matching_loss": 0.06344525180757046,
      "grad_norm": 0.1964299976825714,
      "learning_rate": 9.851167351589592e-05,
      "loss": 0.0646,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011778082814998924,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 18.806630897521973,
      "state_query_norm": 812.0213745117187,
      "state_relation_loss": 0.0011778082814998924,
      "step": 3720,
      "student_distance_mean": 1058.83564453125,
      "teacher_distance_mean": 1.1449398159980775,
      "weighted_auxiliary_loss": 0.0011778082814998924,
      "weighted_relation_loss": 0.0011778082814998924
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.489242282507016,
      "flow_matching_loss": 0.06572880260646344,
      "grad_norm": 0.20873525738716125,
      "learning_rate": 9.849829660581363e-05,
      "loss": 0.0669,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011325525818392634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.219413375854494,
      "state_query_norm": 826.0972717285156,
      "state_relation_loss": 0.0011325525818392634,
      "step": 3730,
      "student_distance_mean": 1081.1735595703126,
      "teacher_distance_mean": 1.1522449374198913,
      "weighted_auxiliary_loss": 0.0011325525818392634,
      "weighted_relation_loss": 0.0011325525818392634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.498596819457437,
      "flow_matching_loss": 0.06666396334767341,
      "grad_norm": 0.171140655875206,
      "learning_rate": 9.848486076572787e-05,
      "loss": 0.0678,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011209307471290232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.415994071960448,
      "state_query_norm": 853.3111389160156,
      "state_relation_loss": 0.0011209307471290232,
      "step": 3740,
      "student_distance_mean": 1098.14140625,
      "teacher_distance_mean": 1.1580164074897765,
      "weighted_auxiliary_loss": 0.0011209307471290232,
      "weighted_relation_loss": 0.0011209307471290232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5079513564078577,
      "flow_matching_loss": 0.06475951932370663,
      "grad_norm": 0.21072086691856384,
      "learning_rate": 9.847136601196446e-05,
      "loss": 0.0659,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011227496899664402,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.027905654907226,
      "state_query_norm": 825.2449401855469,
      "state_relation_loss": 0.0011227496899664402,
      "step": 3750,
      "student_distance_mean": 1070.463916015625,
      "teacher_distance_mean": 1.142364001274109,
      "weighted_auxiliary_loss": 0.0011227496899664402,
      "weighted_relation_loss": 0.0011227496899664402
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.517305893358279,
      "flow_matching_loss": 0.06585575491189957,
      "grad_norm": 0.1747412532567978,
      "learning_rate": 9.845781236092078e-05,
      "loss": 0.0669,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001066799706313759,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.35721492767334,
      "state_query_norm": 836.1636291503906,
      "state_relation_loss": 0.001066799706313759,
      "step": 3760,
      "student_distance_mean": 1089.6784790039062,
      "teacher_distance_mean": 1.1555549502372742,
      "weighted_auxiliary_loss": 0.001066799706313759,
      "weighted_relation_loss": 0.001066799706313759
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5266604303086995,
      "flow_matching_loss": 0.0665702048689127,
      "grad_norm": 0.13287752866744995,
      "learning_rate": 9.844419982906583e-05,
      "loss": 0.0677,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011664067511446773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.307546615600586,
      "state_query_norm": 839.1032653808594,
      "state_relation_loss": 0.0011664067511446773,
      "step": 3770,
      "student_distance_mean": 1085.4138793945312,
      "teacher_distance_mean": 1.148223841190338,
      "weighted_auxiliary_loss": 0.0011664067511446773,
      "weighted_relation_loss": 0.0011664067511446773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5360149672591206,
      "flow_matching_loss": 0.06628324389457703,
      "grad_norm": 0.11859215050935745,
      "learning_rate": 9.843052843294008e-05,
      "loss": 0.0674,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011243403889238835,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.001902770996093,
      "state_query_norm": 820.6361999511719,
      "state_relation_loss": 0.0011243403889238835,
      "step": 3780,
      "student_distance_mean": 1068.6291870117188,
      "teacher_distance_mean": 1.1309312343597413,
      "weighted_auxiliary_loss": 0.0011243403889238835,
      "weighted_relation_loss": 0.0011243403889238835
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5453695042095417,
      "flow_matching_loss": 0.06276890896260738,
      "grad_norm": 0.19503425061702728,
      "learning_rate": 9.841679818915559e-05,
      "loss": 0.0639,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001120909082237631,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.102560806274415,
      "state_query_norm": 832.3030517578125,
      "state_relation_loss": 0.001120909082237631,
      "step": 3790,
      "student_distance_mean": 1075.5966674804688,
      "teacher_distance_mean": 1.1523022532463074,
      "weighted_auxiliary_loss": 0.001120909082237631,
      "weighted_relation_loss": 0.001120909082237631
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5547240411599628,
      "flow_matching_loss": 0.06657339744269848,
      "grad_norm": 0.17421327531337738,
      "learning_rate": 9.840300911439591e-05,
      "loss": 0.0677,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011666528473142535,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.389758682250978,
      "state_query_norm": 835.1653137207031,
      "state_relation_loss": 0.0011666528473142535,
      "step": 3800,
      "student_distance_mean": 1088.8525634765624,
      "teacher_distance_mean": 1.163909125328064,
      "weighted_auxiliary_loss": 0.0011666528473142535,
      "weighted_relation_loss": 0.0011666528473142535
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5640785781103834,
      "flow_matching_loss": 0.06744766347110272,
      "grad_norm": 0.1951306015253067,
      "learning_rate": 9.838916122541603e-05,
      "loss": 0.0686,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011452534003183247,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.50404758453369,
      "state_query_norm": 843.693212890625,
      "state_relation_loss": 0.0011452534003183247,
      "step": 3810,
      "student_distance_mean": 1098.9120361328125,
      "teacher_distance_mean": 1.144069218635559,
      "weighted_auxiliary_loss": 0.0011452534003183247,
      "weighted_relation_loss": 0.0011452534003183247
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5734331150608045,
      "flow_matching_loss": 0.06531758196651935,
      "grad_norm": 0.20998989045619965,
      "learning_rate": 9.837525453904246e-05,
      "loss": 0.0665,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011613438022322954,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.298747062683105,
      "state_query_norm": 834.4424865722656,
      "state_relation_loss": 0.0011613438022322954,
      "step": 3820,
      "student_distance_mean": 1087.4294555664062,
      "teacher_distance_mean": 1.1357304811477662,
      "weighted_auxiliary_loss": 0.0011613438022322954,
      "weighted_relation_loss": 0.0011613438022322954
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.582787652011225,
      "flow_matching_loss": 0.0653331682085991,
      "grad_norm": 0.16577592492103577,
      "learning_rate": 9.836128907217314e-05,
      "loss": 0.0665,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011341675883159042,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.742482566833495,
      "state_query_norm": 849.2970092773437,
      "state_relation_loss": 0.0011341675883159042,
      "step": 3830,
      "student_distance_mean": 1110.20322265625,
      "teacher_distance_mean": 1.1539686918258667,
      "weighted_auxiliary_loss": 0.0011341675883159042,
      "weighted_relation_loss": 0.0011341675883159042
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.5921421889616463,
      "flow_matching_loss": 0.06394464299082755,
      "grad_norm": 0.1828376203775406,
      "learning_rate": 9.834726484177743e-05,
      "loss": 0.0651,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001140665216371417,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.40423164367676,
      "state_query_norm": 846.568701171875,
      "state_relation_loss": 0.001140665216371417,
      "step": 3840,
      "student_distance_mean": 1093.0839599609376,
      "teacher_distance_mean": 1.1423158049583435,
      "weighted_auxiliary_loss": 0.001140665216371417,
      "weighted_relation_loss": 0.001140665216371417
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.6014967259120674,
      "flow_matching_loss": 0.06617428250610828,
      "grad_norm": 0.20300191640853882,
      "learning_rate": 9.833318186489609e-05,
      "loss": 0.0673,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011251914082095027,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.488720512390138,
      "state_query_norm": 844.0125671386719,
      "state_relation_loss": 0.0011251914082095027,
      "step": 3850,
      "student_distance_mean": 1098.1050048828124,
      "teacher_distance_mean": 1.1534923315048218,
      "weighted_auxiliary_loss": 0.0011251914082095027,
      "weighted_relation_loss": 0.0011251914082095027
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.6108512628624885,
      "flow_matching_loss": 0.0628185722976923,
      "grad_norm": 0.21348516643047333,
      "learning_rate": 9.831904015864126e-05,
      "loss": 0.064,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011377123300917447,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.812080955505373,
      "state_query_norm": 849.6161437988281,
      "state_relation_loss": 0.0011377123300917447,
      "step": 3860,
      "student_distance_mean": 1114.411865234375,
      "teacher_distance_mean": 1.1602153420448302,
      "weighted_auxiliary_loss": 0.0011377123300917447,
      "weighted_relation_loss": 0.0011377123300917447
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.620205799812909,
      "flow_matching_loss": 0.06298139467835426,
      "grad_norm": 0.21814903616905212,
      "learning_rate": 9.830483974019645e-05,
      "loss": 0.0641,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011223490117117762,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.250298690795898,
      "state_query_norm": 877.3742736816406,
      "state_relation_loss": 0.0011223490117117762,
      "step": 3870,
      "student_distance_mean": 1136.9483642578125,
      "teacher_distance_mean": 1.174721598625183,
      "weighted_auxiliary_loss": 0.0011223490117117762,
      "weighted_relation_loss": 0.0011223490117117762
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.6295603367633302,
      "flow_matching_loss": 0.06691219583153725,
      "grad_norm": 0.22245319187641144,
      "learning_rate": 9.82905806268165e-05,
      "loss": 0.068,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001106680423254147,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.686574363708495,
      "state_query_norm": 847.9074523925781,
      "state_relation_loss": 0.001106680423254147,
      "step": 3880,
      "student_distance_mean": 1106.5656616210938,
      "teacher_distance_mean": 1.1509364247322083,
      "weighted_auxiliary_loss": 0.001106680423254147,
      "weighted_relation_loss": 0.001106680423254147
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.6389148737137513,
      "flow_matching_loss": 0.06597784869372844,
      "grad_norm": 0.19294001162052155,
      "learning_rate": 9.82762628358276e-05,
      "loss": 0.0671,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011250935029238462,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.03305892944336,
      "state_query_norm": 862.5111145019531,
      "state_relation_loss": 0.0011250935029238462,
      "step": 3890,
      "student_distance_mean": 1126.86513671875,
      "teacher_distance_mean": 1.164388406276703,
      "weighted_auxiliary_loss": 0.0011250935029238462,
      "weighted_relation_loss": 0.0011250935029238462
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.648269410664172,
      "flow_matching_loss": 0.06676178053021431,
      "grad_norm": 0.1646394282579422,
      "learning_rate": 9.826188638462718e-05,
      "loss": 0.0679,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011499042389914393,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.599499320983888,
      "state_query_norm": 847.3091186523437,
      "state_relation_loss": 0.0011499042389914393,
      "step": 3900,
      "student_distance_mean": 1104.3850952148437,
      "teacher_distance_mean": 1.1456454396247864,
      "weighted_auxiliary_loss": 0.0011499042389914393,
      "weighted_relation_loss": 0.0011499042389914393
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.657623947614593,
      "flow_matching_loss": 0.06598784178495407,
      "grad_norm": 0.2528783679008484,
      "learning_rate": 9.824745129068402e-05,
      "loss": 0.0671,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011139589245431126,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.675207138061523,
      "state_query_norm": 847.7506103515625,
      "state_relation_loss": 0.0011139589245431126,
      "step": 3910,
      "student_distance_mean": 1105.6224975585938,
      "teacher_distance_mean": 1.1434816002845765,
      "weighted_auxiliary_loss": 0.0011139589245431126,
      "weighted_relation_loss": 0.0011139589245431126
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.666978484565014,
      "flow_matching_loss": 0.0646111886948347,
      "grad_norm": 0.22058513760566711,
      "learning_rate": 9.82329575715381e-05,
      "loss": 0.0658,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001142613054253161,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.14977912902832,
      "state_query_norm": 861.9942810058594,
      "state_relation_loss": 0.001142613054253161,
      "step": 3920,
      "student_distance_mean": 1132.2471313476562,
      "teacher_distance_mean": 1.165307056903839,
      "weighted_auxiliary_loss": 0.001142613054253161,
      "weighted_relation_loss": 0.001142613054253161
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.676333021515435,
      "flow_matching_loss": 0.06465921141207218,
      "grad_norm": 0.2219148427248001,
      "learning_rate": 9.821840524480066e-05,
      "loss": 0.0658,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011383779463358223,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.89946632385254,
      "state_query_norm": 852.4838806152344,
      "state_relation_loss": 0.0011383779463358223,
      "step": 3930,
      "student_distance_mean": 1119.8407836914062,
      "teacher_distance_mean": 1.1520728707313537,
      "weighted_auxiliary_loss": 0.0011383779463358223,
      "weighted_relation_loss": 0.0011383779463358223
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.685687558465856,
      "flow_matching_loss": 0.06383973956108094,
      "grad_norm": 0.21280063688755035,
      "learning_rate": 9.820379432815414e-05,
      "loss": 0.065,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001139411434996873,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.17243537902832,
      "state_query_norm": 869.6273254394531,
      "state_relation_loss": 0.001139411434996873,
      "step": 3940,
      "student_distance_mean": 1134.6212768554688,
      "teacher_distance_mean": 1.1640572786331176,
      "weighted_auxiliary_loss": 0.001139411434996873,
      "weighted_relation_loss": 0.001139411434996873
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.695042095416277,
      "flow_matching_loss": 0.06663345508277416,
      "grad_norm": 0.27230799198150635,
      "learning_rate": 9.81891248393522e-05,
      "loss": 0.0677,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011084812576882541,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.367545700073244,
      "state_query_norm": 872.1958801269532,
      "state_relation_loss": 0.0011084812576882541,
      "step": 3950,
      "student_distance_mean": 1145.5585693359376,
      "teacher_distance_mean": 1.1762829542160034,
      "weighted_auxiliary_loss": 0.0011084812576882541,
      "weighted_relation_loss": 0.0011084812576882541
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7043966323666977,
      "flow_matching_loss": 0.06378688327968121,
      "grad_norm": 0.17659255862236023,
      "learning_rate": 9.817439679621963e-05,
      "loss": 0.0649,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010948896408081055,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.764728164672853,
      "state_query_norm": 852.2411743164063,
      "state_relation_loss": 0.0010948896408081055,
      "step": 3960,
      "student_distance_mean": 1110.7558471679688,
      "teacher_distance_mean": 1.1501981377601624,
      "weighted_auxiliary_loss": 0.0010948896408081055,
      "weighted_relation_loss": 0.0010948896408081055
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.713751169317119,
      "flow_matching_loss": 0.06596810184419155,
      "grad_norm": 0.1994720846414566,
      "learning_rate": 9.815961021665243e-05,
      "loss": 0.0671,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011461558286100626,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.811380767822264,
      "state_query_norm": 851.5127807617188,
      "state_relation_loss": 0.0011461558286100626,
      "step": 3970,
      "student_distance_mean": 1114.2067504882812,
      "teacher_distance_mean": 1.156564712524414,
      "weighted_auxiliary_loss": 0.0011461558286100626,
      "weighted_relation_loss": 0.0011461558286100626
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.72310570626754,
      "flow_matching_loss": 0.06311617121100425,
      "grad_norm": 0.13809819519519806,
      "learning_rate": 9.814476511861763e-05,
      "loss": 0.0642,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010997130302712321,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.591697120666502,
      "state_query_norm": 853.7866394042969,
      "state_relation_loss": 0.0010997130302712321,
      "step": 3980,
      "student_distance_mean": 1107.1901611328126,
      "teacher_distance_mean": 1.1480016112327576,
      "weighted_auxiliary_loss": 0.0010997130302712321,
      "weighted_relation_loss": 0.0010997130302712321
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7324602432179605,
      "flow_matching_loss": 0.06372886039316654,
      "grad_norm": 0.2247590869665146,
      "learning_rate": 9.812986152015348e-05,
      "loss": 0.0648,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010516814480070024,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.061648750305174,
      "state_query_norm": 864.9793151855469,
      "state_relation_loss": 0.0010516814480070024,
      "step": 3990,
      "student_distance_mean": 1129.90859375,
      "teacher_distance_mean": 1.1610413789749146,
      "weighted_auxiliary_loss": 0.0010516814480070024,
      "weighted_relation_loss": 0.0010516814480070024
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7418147801683816,
      "flow_matching_loss": 0.06301248073577881,
      "grad_norm": 0.20620764791965485,
      "learning_rate": 9.811489943936922e-05,
      "loss": 0.0641,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011270788614638149,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.077602767944335,
      "state_query_norm": 866.2919738769531,
      "state_relation_loss": 0.0011270788614638149,
      "step": 4000,
      "student_distance_mean": 1132.6494995117187,
      "teacher_distance_mean": 1.1551192998886108,
      "weighted_auxiliary_loss": 0.0011270788614638149,
      "weighted_relation_loss": 0.0011270788614638149
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7511693171188027,
      "flow_matching_loss": 0.06149840094149113,
      "grad_norm": 0.14520740509033203,
      "learning_rate": 9.809987889444522e-05,
      "loss": 0.0626,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011169016594067215,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.889905548095705,
      "state_query_norm": 850.1192504882813,
      "state_relation_loss": 0.0011169016594067215,
      "step": 4010,
      "student_distance_mean": 1119.5145202636718,
      "teacher_distance_mean": 1.1559602260589599,
      "weighted_auxiliary_loss": 0.0011169016594067215,
      "weighted_relation_loss": 0.0011169016594067215
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7605238540692234,
      "flow_matching_loss": 0.06326527446508408,
      "grad_norm": 0.19780059158802032,
      "learning_rate": 9.808479990363282e-05,
      "loss": 0.0644,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011116775800473987,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.452994918823244,
      "state_query_norm": 884.2417053222656,
      "state_relation_loss": 0.0011116775800473987,
      "step": 4020,
      "student_distance_mean": 1150.7923828125,
      "teacher_distance_mean": 1.1665475130081178,
      "weighted_auxiliary_loss": 0.0011116775800473987,
      "weighted_relation_loss": 0.0011116775800473987
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7698783910196445,
      "flow_matching_loss": 0.06531900465488434,
      "grad_norm": 0.25538164377212524,
      "learning_rate": 9.806966248525445e-05,
      "loss": 0.0664,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001099144760519266,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.292953491210938,
      "state_query_norm": 871.037451171875,
      "state_relation_loss": 0.001099144760519266,
      "step": 4030,
      "student_distance_mean": 1142.50234375,
      "teacher_distance_mean": 1.1589829921722412,
      "weighted_auxiliary_loss": 0.001099144760519266,
      "weighted_relation_loss": 0.001099144760519266
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7792329279700656,
      "flow_matching_loss": 0.0630754791200161,
      "grad_norm": 0.20450003445148468,
      "learning_rate": 9.805446665770348e-05,
      "loss": 0.0642,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010857031447812915,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.60446891784668,
      "state_query_norm": 847.0193359375,
      "state_relation_loss": 0.0010857031447812915,
      "step": 4040,
      "student_distance_mean": 1105.3504760742187,
      "teacher_distance_mean": 1.1363637685775756,
      "weighted_auxiliary_loss": 0.0010857031447812915,
      "weighted_relation_loss": 0.0010857031447812915
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7885874649204867,
      "flow_matching_loss": 0.06575223952531814,
      "grad_norm": 0.17515000700950623,
      "learning_rate": 9.803921243944429e-05,
      "loss": 0.0668,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010715884913224728,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.336012077331542,
      "state_query_norm": 873.0581359863281,
      "state_relation_loss": 0.0010715884913224728,
      "step": 4050,
      "student_distance_mean": 1145.855322265625,
      "teacher_distance_mean": 1.1634166359901428,
      "weighted_auxiliary_loss": 0.0010715884913224728,
      "weighted_relation_loss": 0.0010715884913224728
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.7979420018709074,
      "flow_matching_loss": 0.06852025128901004,
      "grad_norm": 0.19678223133087158,
      "learning_rate": 9.802389984901218e-05,
      "loss": 0.0696,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001089409104315564,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.39882926940918,
      "state_query_norm": 874.800048828125,
      "state_relation_loss": 0.001089409104315564,
      "step": 4060,
      "student_distance_mean": 1146.59755859375,
      "teacher_distance_mean": 1.161970067024231,
      "weighted_auxiliary_loss": 0.001089409104315564,
      "weighted_relation_loss": 0.001089409104315564
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.8072965388213285,
      "flow_matching_loss": 0.06295167021453381,
      "grad_norm": 0.1569962203502655,
      "learning_rate": 9.80085289050134e-05,
      "loss": 0.0641,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011599910794757307,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.175167083740234,
      "state_query_norm": 860.1434814453125,
      "state_relation_loss": 0.0011599910794757307,
      "step": 4070,
      "student_distance_mean": 1132.5763549804688,
      "teacher_distance_mean": 1.1553489565849304,
      "weighted_auxiliary_loss": 0.0011599910794757307,
      "weighted_relation_loss": 0.0011599910794757307
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.816651075771749,
      "flow_matching_loss": 0.0667047955095768,
      "grad_norm": 0.14839735627174377,
      "learning_rate": 9.799309962612508e-05,
      "loss": 0.0678,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011153411818668247,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 19.73217144012451,
      "state_query_norm": 849.1932250976563,
      "state_relation_loss": 0.0011153411818668247,
      "step": 4080,
      "student_distance_mean": 1110.4033447265624,
      "teacher_distance_mean": 1.1413289070129395,
      "weighted_auxiliary_loss": 0.0011153411818668247,
      "weighted_relation_loss": 0.0011153411818668247
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.82600561272217,
      "flow_matching_loss": 0.0630173709243536,
      "grad_norm": 0.21345415711402893,
      "learning_rate": 9.797761203109527e-05,
      "loss": 0.0641,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010632076824549585,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.254308128356932,
      "state_query_norm": 867.1464477539063,
      "state_relation_loss": 0.0010632076824549585,
      "step": 4090,
      "student_distance_mean": 1135.1778076171875,
      "teacher_distance_mean": 1.1535690069198608,
      "weighted_auxiliary_loss": 0.0010632076824549585,
      "weighted_relation_loss": 0.0010632076824549585
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.8353601496725913,
      "flow_matching_loss": 0.06404738537967206,
      "grad_norm": 0.2150295525789261,
      "learning_rate": 9.796206613874283e-05,
      "loss": 0.0651,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010771532834041864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.001571083068846,
      "state_query_norm": 856.6465576171875,
      "state_relation_loss": 0.0010771532834041864,
      "step": 4100,
      "student_distance_mean": 1125.6486206054688,
      "teacher_distance_mean": 1.1422832489013672,
      "weighted_auxiliary_loss": 0.0010771532834041864,
      "weighted_relation_loss": 0.0010771532834041864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.8447146866230124,
      "flow_matching_loss": 0.06348854042589665,
      "grad_norm": 0.21712322533130646,
      "learning_rate": 9.794646196795754e-05,
      "loss": 0.0646,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011202564113773406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.027596473693848,
      "state_query_norm": 861.6693298339844,
      "state_relation_loss": 0.0011202564113773406,
      "step": 4110,
      "student_distance_mean": 1128.1250244140624,
      "teacher_distance_mean": 1.143946611881256,
      "weighted_auxiliary_loss": 0.0011202564113773406,
      "weighted_relation_loss": 0.0011202564113773406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.854069223573433,
      "flow_matching_loss": 0.06407177783548831,
      "grad_norm": 0.1288752257823944,
      "learning_rate": 9.793079953769987e-05,
      "loss": 0.0652,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010967139271087945,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.215862464904784,
      "state_query_norm": 865.8019958496094,
      "state_relation_loss": 0.0010967139271087945,
      "step": 4120,
      "student_distance_mean": 1136.1588745117188,
      "teacher_distance_mean": 1.1354409098625182,
      "weighted_auxiliary_loss": 0.0010967139271087945,
      "weighted_relation_loss": 0.0010967139271087945
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.863423760523854,
      "flow_matching_loss": 0.06427050195634365,
      "grad_norm": 0.1048569604754448,
      "learning_rate": 9.79150788670012e-05,
      "loss": 0.0654,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001091219182126224,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.68347454071045,
      "state_query_norm": 889.1442932128906,
      "state_relation_loss": 0.001091219182126224,
      "step": 4130,
      "student_distance_mean": 1160.4697998046875,
      "teacher_distance_mean": 1.167117190361023,
      "weighted_auxiliary_loss": 0.001091219182126224,
      "weighted_relation_loss": 0.001091219182126224
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.872778297474275,
      "flow_matching_loss": 0.06212002076208591,
      "grad_norm": 0.14544036984443665,
      "learning_rate": 9.78992999749636e-05,
      "loss": 0.0632,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011005614360328764,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.550920677185058,
      "state_query_norm": 879.5044738769532,
      "state_relation_loss": 0.0011005614360328764,
      "step": 4140,
      "student_distance_mean": 1156.0776000976562,
      "teacher_distance_mean": 1.1643259048461914,
      "weighted_auxiliary_loss": 0.0011005614360328764,
      "weighted_relation_loss": 0.0011005614360328764
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.882132834424696,
      "flow_matching_loss": 0.06272890046238899,
      "grad_norm": 0.19270822405815125,
      "learning_rate": 9.788346288075994e-05,
      "loss": 0.0638,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010899179149419069,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.560153579711915,
      "state_query_norm": 878.2455200195312,
      "state_relation_loss": 0.0010899179149419069,
      "step": 4150,
      "student_distance_mean": 1153.7798583984375,
      "teacher_distance_mean": 1.1623017072677613,
      "weighted_auxiliary_loss": 0.0010899179149419069,
      "weighted_relation_loss": 0.0010899179149419069
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.891487371375117,
      "flow_matching_loss": 0.06526200696825982,
      "grad_norm": 0.2091587781906128,
      "learning_rate": 9.786756760363373e-05,
      "loss": 0.0663,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001055745081976056,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.23820571899414,
      "state_query_norm": 876.2778198242188,
      "state_relation_loss": 0.001055745081976056,
      "step": 4160,
      "student_distance_mean": 1140.1797241210938,
      "teacher_distance_mean": 1.143203341960907,
      "weighted_auxiliary_loss": 0.001055745081976056,
      "weighted_relation_loss": 0.001055745081976056
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.900841908325538,
      "flow_matching_loss": 0.06314588487148284,
      "grad_norm": 0.13645806908607483,
      "learning_rate": 9.78516141628993e-05,
      "loss": 0.0642,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010865888209082187,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.67452735900879,
      "state_query_norm": 893.33876953125,
      "state_relation_loss": 0.0010865888209082187,
      "step": 4170,
      "student_distance_mean": 1162.587451171875,
      "teacher_distance_mean": 1.1594462513923645,
      "weighted_auxiliary_loss": 0.0010865888209082187,
      "weighted_relation_loss": 0.0010865888209082187
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9101964452759588,
      "flow_matching_loss": 0.06281136646866799,
      "grad_norm": 0.12441578507423401,
      "learning_rate": 9.783560257794154e-05,
      "loss": 0.0639,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001062568003544584,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.28863410949707,
      "state_query_norm": 876.1578247070313,
      "state_relation_loss": 0.001062568003544584,
      "step": 4180,
      "student_distance_mean": 1143.3174560546875,
      "teacher_distance_mean": 1.150883173942566,
      "weighted_auxiliary_loss": 0.001062568003544584,
      "weighted_relation_loss": 0.001062568003544584
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.91955098222638,
      "flow_matching_loss": 0.06332811266183853,
      "grad_norm": 0.1403486579656601,
      "learning_rate": 9.781953286821603e-05,
      "loss": 0.0644,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001114270999096334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.29814510345459,
      "state_query_norm": 872.9014038085937,
      "state_relation_loss": 0.001114270999096334,
      "step": 4190,
      "student_distance_mean": 1143.4692260742188,
      "teacher_distance_mean": 1.1449089050292969,
      "weighted_auxiliary_loss": 0.001114270999096334,
      "weighted_relation_loss": 0.001114270999096334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9289055191768005,
      "flow_matching_loss": 0.06258428394794464,
      "grad_norm": 0.12909969687461853,
      "learning_rate": 9.780340505324901e-05,
      "loss": 0.0636,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010161421610973775,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.395573806762695,
      "state_query_norm": 875.0433654785156,
      "state_relation_loss": 0.0010161421610973775,
      "step": 4200,
      "student_distance_mean": 1144.7137329101563,
      "teacher_distance_mean": 1.1523443937301636,
      "weighted_auxiliary_loss": 0.0010161421610973775,
      "weighted_relation_loss": 0.0010161421610973775
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9382600561272216,
      "flow_matching_loss": 0.06432996168732644,
      "grad_norm": 0.16630645096302032,
      "learning_rate": 9.778721915263727e-05,
      "loss": 0.0654,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010483376507181674,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.23790035247803,
      "state_query_norm": 863.9381042480469,
      "state_relation_loss": 0.0010483376507181674,
      "step": 4210,
      "student_distance_mean": 1135.5918212890624,
      "teacher_distance_mean": 1.1435330748558044,
      "weighted_auxiliary_loss": 0.0010483376507181674,
      "weighted_relation_loss": 0.0010483376507181674
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9476145930776427,
      "flow_matching_loss": 0.06319602727890014,
      "grad_norm": 0.18253645300865173,
      "learning_rate": 9.777097518604824e-05,
      "loss": 0.0643,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011216580867767334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.735617256164552,
      "state_query_norm": 881.8639587402344,
      "state_relation_loss": 0.0011216580867767334,
      "step": 4220,
      "student_distance_mean": 1162.73193359375,
      "teacher_distance_mean": 1.1598063707351685,
      "weighted_auxiliary_loss": 0.0011216580867767334,
      "weighted_relation_loss": 0.0011216580867767334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.956969130028064,
      "flow_matching_loss": 0.061267454922199246,
      "grad_norm": 0.1563195437192917,
      "learning_rate": 9.775467317321984e-05,
      "loss": 0.0624,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010861793765798212,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.716399002075196,
      "state_query_norm": 886.478271484375,
      "state_relation_loss": 0.0010861793765798212,
      "step": 4230,
      "student_distance_mean": 1162.3116577148437,
      "teacher_distance_mean": 1.164093255996704,
      "weighted_auxiliary_loss": 0.0010861793765798212,
      "weighted_relation_loss": 0.0010861793765798212
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9663236669784845,
      "flow_matching_loss": 0.06121411211788654,
      "grad_norm": 0.13742344081401825,
      "learning_rate": 9.773831313396055e-05,
      "loss": 0.0622,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010252790583763272,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.40555057525635,
      "state_query_norm": 887.2636352539063,
      "state_relation_loss": 0.0010252790583763272,
      "step": 4240,
      "student_distance_mean": 1150.7853515625,
      "teacher_distance_mean": 1.1392409920692443,
      "weighted_auxiliary_loss": 0.0010252790583763272,
      "weighted_relation_loss": 0.0010252790583763272
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9756782039289056,
      "flow_matching_loss": 0.06020795442163944,
      "grad_norm": 0.13725504279136658,
      "learning_rate": 9.77218950881494e-05,
      "loss": 0.0613,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010948940936941654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.157149505615234,
      "state_query_norm": 908.2756958007812,
      "state_relation_loss": 0.0010948940936941654,
      "step": 4250,
      "student_distance_mean": 1189.2836181640625,
      "teacher_distance_mean": 1.167861270904541,
      "weighted_auxiliary_loss": 0.0010948940936941654,
      "weighted_relation_loss": 0.0010948940936941654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9850327408793262,
      "flow_matching_loss": 0.06227084621787071,
      "grad_norm": 0.1844046264886856,
      "learning_rate": 9.770541905573583e-05,
      "loss": 0.0633,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001074435334885493,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.622572708129884,
      "state_query_norm": 890.8994567871093,
      "state_relation_loss": 0.001074435334885493,
      "step": 4260,
      "student_distance_mean": 1158.5599243164063,
      "teacher_distance_mean": 1.1407257318496704,
      "weighted_auxiliary_loss": 0.001074435334885493,
      "weighted_relation_loss": 0.001074435334885493
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 3.9943872778297473,
      "flow_matching_loss": 0.062486904487013814,
      "grad_norm": 0.13281311094760895,
      "learning_rate": 9.768888505673976e-05,
      "loss": 0.0635,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010419605358038097,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.778749465942383,
      "state_query_norm": 893.3359008789063,
      "state_relation_loss": 0.0010419605358038097,
      "step": 4270,
      "student_distance_mean": 1169.6280029296875,
      "teacher_distance_mean": 1.157858657836914,
      "weighted_auxiliary_loss": 0.0010419605358038097,
      "weighted_relation_loss": 0.0010419605358038097
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.003741814780168,
      "flow_matching_loss": 0.06099091283977032,
      "grad_norm": 0.13937720656394958,
      "learning_rate": 9.767229311125162e-05,
      "loss": 0.0621,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0011024746694602073,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 20.54302215576172,
      "state_query_norm": 877.4690795898438,
      "state_relation_loss": 0.0011024746694602073,
      "step": 4280,
      "student_distance_mean": 1158.0109008789063,
      "teacher_distance_mean": 1.1621626973152162,
      "weighted_auxiliary_loss": 0.0011024746694602073,
      "weighted_relation_loss": 0.0011024746694602073
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.0130963517305895,
      "flow_matching_loss": 0.060305919125676156,
      "grad_norm": 0.13975633680820465,
      "learning_rate": 9.765564323943211e-05,
      "loss": 0.0614,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011361646000295877,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.566643714904785,
      "state_query_norm": 886.7838684082031,
      "state_relation_loss": 0.0011361646000295877,
      "step": 4290,
      "student_distance_mean": 1157.40166015625,
      "teacher_distance_mean": 1.145495593547821,
      "weighted_auxiliary_loss": 0.0011361646000295877,
      "weighted_relation_loss": 0.0011361646000295877
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.022450888681011,
      "flow_matching_loss": 0.0600750494748354,
      "grad_norm": 0.18723569810390472,
      "learning_rate": 9.763893546151244e-05,
      "loss": 0.0612,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010981909814290701,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.841653633117676,
      "state_query_norm": 885.9658020019531,
      "state_relation_loss": 0.0010981909814290701,
      "step": 4300,
      "student_distance_mean": 1168.1848999023437,
      "teacher_distance_mean": 1.1542216062545776,
      "weighted_auxiliary_loss": 0.0010981909814290701,
      "weighted_relation_loss": 0.0010981909814290701
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.031805425631431,
      "flow_matching_loss": 0.0614531334489584,
      "grad_norm": 0.17388348281383514,
      "learning_rate": 9.762216979779412e-05,
      "loss": 0.0626,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011376157752238214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.094824028015136,
      "state_query_norm": 909.6409362792969,
      "state_relation_loss": 0.0011376157752238214,
      "step": 4310,
      "student_distance_mean": 1185.9901733398438,
      "teacher_distance_mean": 1.1612117290496826,
      "weighted_auxiliary_loss": 0.0011376157752238214,
      "weighted_relation_loss": 0.0011376157752238214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.041159962581852,
      "flow_matching_loss": 0.06002825126051903,
      "grad_norm": 0.19137385487556458,
      "learning_rate": 9.760534626864902e-05,
      "loss": 0.0612,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011342328623868525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.716765022277833,
      "state_query_norm": 883.7436279296875,
      "state_relation_loss": 0.0011342328623868525,
      "step": 4320,
      "student_distance_mean": 1161.576171875,
      "teacher_distance_mean": 1.1586446404457091,
      "weighted_auxiliary_loss": 0.0011342328623868525,
      "weighted_relation_loss": 0.0011342328623868525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.050514499532273,
      "flow_matching_loss": 0.06321249343454838,
      "grad_norm": 0.22141359746456146,
      "learning_rate": 9.758846489451931e-05,
      "loss": 0.0643,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011071604676544666,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.551207542419434,
      "state_query_norm": 881.0888244628907,
      "state_relation_loss": 0.0011071604676544666,
      "step": 4330,
      "student_distance_mean": 1153.8283569335938,
      "teacher_distance_mean": 1.1483384847640992,
      "weighted_auxiliary_loss": 0.0011071604676544666,
      "weighted_relation_loss": 0.0011071604676544666
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.059869036482694,
      "flow_matching_loss": 0.060582848638296126,
      "grad_norm": 0.18667452037334442,
      "learning_rate": 9.757152569591748e-05,
      "loss": 0.0617,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011571259470656513,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.89083652496338,
      "state_query_norm": 891.1888732910156,
      "state_relation_loss": 0.0011571259470656513,
      "step": 4340,
      "student_distance_mean": 1171.8626220703125,
      "teacher_distance_mean": 1.165063488483429,
      "weighted_auxiliary_loss": 0.0011571259470656513,
      "weighted_relation_loss": 0.0011571259470656513
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.069223573433115,
      "flow_matching_loss": 0.06135772876441479,
      "grad_norm": 0.17626824975013733,
      "learning_rate": 9.75545286934262e-05,
      "loss": 0.0626,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0012075229664333166,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.142477416992186,
      "state_query_norm": 899.1848754882812,
      "state_relation_loss": 0.0012075229664333166,
      "step": 4350,
      "student_distance_mean": 1186.2167358398438,
      "teacher_distance_mean": 1.1704258918762207,
      "weighted_auxiliary_loss": 0.0012075229664333166,
      "weighted_relation_loss": 0.0012075229664333166
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.078578110383536,
      "flow_matching_loss": 0.06093643978238106,
      "grad_norm": 0.14116552472114563,
      "learning_rate": 9.753747390769847e-05,
      "loss": 0.062,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011017664801329374,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.00676383972168,
      "state_query_norm": 898.8165893554688,
      "state_relation_loss": 0.0011017664801329374,
      "step": 4360,
      "student_distance_mean": 1180.6588012695313,
      "teacher_distance_mean": 1.1515238285064697,
      "weighted_auxiliary_loss": 0.0011017664801329374,
      "weighted_relation_loss": 0.0011017664801329374
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.087932647333957,
      "flow_matching_loss": 0.06036153696477413,
      "grad_norm": 0.18250635266304016,
      "learning_rate": 9.752036135945744e-05,
      "loss": 0.0614,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001083977794041857,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.92748031616211,
      "state_query_norm": 895.8787048339843,
      "state_relation_loss": 0.001083977794041857,
      "step": 4370,
      "student_distance_mean": 1174.3621215820312,
      "teacher_distance_mean": 1.1555330514907838,
      "weighted_auxiliary_loss": 0.001083977794041857,
      "weighted_relation_loss": 0.001083977794041857
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.097287184284378,
      "flow_matching_loss": 0.06083359569311142,
      "grad_norm": 0.1843128651380539,
      "learning_rate": 9.750319106949649e-05,
      "loss": 0.0619,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001046496455091983,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.034004974365235,
      "state_query_norm": 897.0452453613282,
      "state_relation_loss": 0.001046496455091983,
      "step": 4380,
      "student_distance_mean": 1180.81669921875,
      "teacher_distance_mean": 1.161818504333496,
      "weighted_auxiliary_loss": 0.001046496455091983,
      "weighted_relation_loss": 0.001046496455091983
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.106641721234799,
      "flow_matching_loss": 0.06458953395485878,
      "grad_norm": 0.19868701696395874,
      "learning_rate": 9.748596305867913e-05,
      "loss": 0.0657,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0011159998422954232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.071802139282227,
      "state_query_norm": 899.5074096679688,
      "state_relation_loss": 0.0011159998422954232,
      "step": 4390,
      "student_distance_mean": 1183.2124755859375,
      "teacher_distance_mean": 1.156242561340332,
      "weighted_auxiliary_loss": 0.0011159998422954232,
      "weighted_relation_loss": 0.0011159998422954232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.11599625818522,
      "flow_matching_loss": 0.06277989000082015,
      "grad_norm": 0.1326817274093628,
      "learning_rate": 9.746867734793903e-05,
      "loss": 0.0638,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010681081912480295,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.963792037963866,
      "state_query_norm": 892.9460083007813,
      "state_relation_loss": 0.0010681081912480295,
      "step": 4400,
      "student_distance_mean": 1173.7004028320312,
      "teacher_distance_mean": 1.1443888664245605,
      "weighted_auxiliary_loss": 0.0010681081912480295,
      "weighted_relation_loss": 0.0010681081912480295
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.125350795135641,
      "flow_matching_loss": 0.061240780353546145,
      "grad_norm": 0.14399059116840363,
      "learning_rate": 9.745133395827993e-05,
      "loss": 0.0622,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010071339085698128,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.172788047790526,
      "state_query_norm": 903.4506896972656,
      "state_relation_loss": 0.0010071339085698128,
      "step": 4410,
      "student_distance_mean": 1186.6146606445313,
      "teacher_distance_mean": 1.1531034231185913,
      "weighted_auxiliary_loss": 0.0010071339085698128,
      "weighted_relation_loss": 0.0010071339085698128
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.134705332086062,
      "flow_matching_loss": 0.062281325832009314,
      "grad_norm": 0.21536517143249512,
      "learning_rate": 9.743393291077572e-05,
      "loss": 0.0633,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009962473937775939,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.168668937683105,
      "state_query_norm": 905.5382751464844,
      "state_relation_loss": 0.0009962473937775939,
      "step": 4420,
      "student_distance_mean": 1189.7023681640626,
      "teacher_distance_mean": 1.1487982511520385,
      "weighted_auxiliary_loss": 0.0009962473937775939,
      "weighted_relation_loss": 0.0009962473937775939
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.144059869036482,
      "flow_matching_loss": 0.06107497178018093,
      "grad_norm": 0.1736196130514145,
      "learning_rate": 9.741647422657028e-05,
      "loss": 0.0621,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010464766819495708,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.967351531982423,
      "state_query_norm": 893.0588745117187,
      "state_relation_loss": 0.0010464766819495708,
      "step": 4430,
      "student_distance_mean": 1175.6685546875,
      "teacher_distance_mean": 1.1478741765022278,
      "weighted_auxiliary_loss": 0.0010464766819495708,
      "weighted_relation_loss": 0.0010464766819495708
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.153414405986903,
      "flow_matching_loss": 0.05805813521146774,
      "grad_norm": 0.17157290875911713,
      "learning_rate": 9.739895792687758e-05,
      "loss": 0.0591,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010334159596823156,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.024270820617676,
      "state_query_norm": 891.0668151855468,
      "state_relation_loss": 0.0010334159596823156,
      "step": 4440,
      "student_distance_mean": 1176.0423706054687,
      "teacher_distance_mean": 1.1557563662528991,
      "weighted_auxiliary_loss": 0.0010334159596823156,
      "weighted_relation_loss": 0.0010334159596823156
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.1627689429373245,
      "flow_matching_loss": 0.060365797206759456,
      "grad_norm": 0.16781458258628845,
      "learning_rate": 9.738138403298157e-05,
      "loss": 0.0614,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001024201576365158,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.235922622680665,
      "state_query_norm": 904.9447326660156,
      "state_relation_loss": 0.001024201576365158,
      "step": 4450,
      "student_distance_mean": 1191.9028076171876,
      "teacher_distance_mean": 1.1517783403396606,
      "weighted_auxiliary_loss": 0.001024201576365158,
      "weighted_relation_loss": 0.001024201576365158
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.172123479887746,
      "flow_matching_loss": 0.061835238710045815,
      "grad_norm": 0.1718689203262329,
      "learning_rate": 9.736375256623619e-05,
      "loss": 0.0629,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010663600405678153,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.324294090270996,
      "state_query_norm": 905.9666687011719,
      "state_relation_loss": 0.0010663600405678153,
      "step": 4460,
      "student_distance_mean": 1192.751318359375,
      "teacher_distance_mean": 1.1569496273994446,
      "weighted_auxiliary_loss": 0.0010663600405678153,
      "weighted_relation_loss": 0.0010663600405678153
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.181478016838167,
      "flow_matching_loss": 0.06338896304368973,
      "grad_norm": 0.1416356861591339,
      "learning_rate": 9.734606354806533e-05,
      "loss": 0.0644,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010209081461653112,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.459431457519532,
      "state_query_norm": 913.9358703613282,
      "state_relation_loss": 0.0010209081461653112,
      "step": 4470,
      "student_distance_mean": 1202.4557739257812,
      "teacher_distance_mean": 1.1582276582717896,
      "weighted_auxiliary_loss": 0.0010209081461653112,
      "weighted_relation_loss": 0.0010209081461653112
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.190832553788588,
      "flow_matching_loss": 0.059955406561493875,
      "grad_norm": 0.1836944967508316,
      "learning_rate": 9.73283169999628e-05,
      "loss": 0.061,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010311436955817044,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.07018928527832,
      "state_query_norm": 896.152880859375,
      "state_relation_loss": 0.0010311436955817044,
      "step": 4480,
      "student_distance_mean": 1178.781884765625,
      "teacher_distance_mean": 1.1450532793998718,
      "weighted_auxiliary_loss": 0.0010311436955817044,
      "weighted_relation_loss": 0.0010311436955817044
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.200187090739009,
      "flow_matching_loss": 0.05937126688659191,
      "grad_norm": 0.1522144228219986,
      "learning_rate": 9.731051294349238e-05,
      "loss": 0.0604,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001053798897191882,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.5232234954834,
      "state_query_norm": 904.2024230957031,
      "state_relation_loss": 0.001053798897191882,
      "step": 4490,
      "student_distance_mean": 1203.5108154296875,
      "teacher_distance_mean": 1.1675137281417847,
      "weighted_auxiliary_loss": 0.001053798897191882,
      "weighted_relation_loss": 0.001053798897191882
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.209541627689429,
      "flow_matching_loss": 0.05973884947597981,
      "grad_norm": 0.13751870393753052,
      "learning_rate": 9.729265140028762e-05,
      "loss": 0.0608,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001041619013994932,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 20.940510177612303,
      "state_query_norm": 892.238818359375,
      "state_relation_loss": 0.001041619013994932,
      "step": 4500,
      "student_distance_mean": 1175.34326171875,
      "teacher_distance_mean": 1.1423842430114746,
      "weighted_auxiliary_loss": 0.001041619013994932,
      "weighted_relation_loss": 0.001041619013994932
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.21889616463985,
      "flow_matching_loss": 0.05932054482400417,
      "grad_norm": 0.12488225102424622,
      "learning_rate": 9.727473239205201e-05,
      "loss": 0.0603,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010192581394221634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.508905792236327,
      "state_query_norm": 906.6364501953125,
      "state_relation_loss": 0.0010192581394221634,
      "step": 4510,
      "student_distance_mean": 1202.307666015625,
      "teacher_distance_mean": 1.1616639256477357,
      "weighted_auxiliary_loss": 0.0010192581394221634,
      "weighted_relation_loss": 0.0010192581394221634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.228250701590271,
      "flow_matching_loss": 0.06350913047790527,
      "grad_norm": 0.15210574865341187,
      "learning_rate": 9.725675594055883e-05,
      "loss": 0.0645,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010083548841066658,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.156729698181152,
      "state_query_norm": 900.3487487792969,
      "state_relation_loss": 0.0010083548841066658,
      "step": 4520,
      "student_distance_mean": 1186.1385620117187,
      "teacher_distance_mean": 1.1426487922668458,
      "weighted_auxiliary_loss": 0.0010083548841066658,
      "weighted_relation_loss": 0.0010083548841066658
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.237605238540692,
      "flow_matching_loss": 0.06081272028386593,
      "grad_norm": 0.1545821875333786,
      "learning_rate": 9.723872206765116e-05,
      "loss": 0.0618,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010264367330819369,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.03904781341553,
      "state_query_norm": 896.1480651855469,
      "state_relation_loss": 0.0010264367330819369,
      "step": 4530,
      "student_distance_mean": 1177.7121215820312,
      "teacher_distance_mean": 1.1433652758598327,
      "weighted_auxiliary_loss": 0.0010264367330819369,
      "weighted_relation_loss": 0.0010264367330819369
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.2469597754911135,
      "flow_matching_loss": 0.06296824775636196,
      "grad_norm": 0.18090003728866577,
      "learning_rate": 9.722063079524185e-05,
      "loss": 0.064,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010241901967674495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.214754104614258,
      "state_query_norm": 904.7933715820312,
      "state_relation_loss": 0.0010241901967674495,
      "step": 4540,
      "student_distance_mean": 1192.0671020507812,
      "teacher_distance_mean": 1.156449556350708,
      "weighted_auxiliary_loss": 0.0010241901967674495,
      "weighted_relation_loss": 0.0010241901967674495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.256314312441534,
      "flow_matching_loss": 0.0630191445350647,
      "grad_norm": 0.14424985647201538,
      "learning_rate": 9.720248214531351e-05,
      "loss": 0.0641,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010496518574655057,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.36712074279785,
      "state_query_norm": 910.0011291503906,
      "state_relation_loss": 0.0010496518574655057,
      "step": 4550,
      "student_distance_mean": 1199.9269897460938,
      "teacher_distance_mean": 1.1604981899261475,
      "weighted_auxiliary_loss": 0.0010496518574655057,
      "weighted_relation_loss": 0.0010496518574655057
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.265668849391955,
      "flow_matching_loss": 0.059853121265769006,
      "grad_norm": 0.155291348695755,
      "learning_rate": 9.718427613991848e-05,
      "loss": 0.0608,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000993924430804327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.335840034484864,
      "state_query_norm": 911.0713012695312,
      "state_relation_loss": 0.000993924430804327,
      "step": 4560,
      "student_distance_mean": 1199.4367797851562,
      "teacher_distance_mean": 1.1510987043380738,
      "weighted_auxiliary_loss": 0.000993924430804327,
      "weighted_relation_loss": 0.000993924430804327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.275023386342376,
      "flow_matching_loss": 0.06055922284722328,
      "grad_norm": 0.15530362725257874,
      "learning_rate": 9.716601280117873e-05,
      "loss": 0.0616,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010236845817416906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.39531021118164,
      "state_query_norm": 913.3276184082031,
      "state_relation_loss": 0.0010236845817416906,
      "step": 4570,
      "student_distance_mean": 1200.3316650390625,
      "teacher_distance_mean": 1.155099368095398,
      "weighted_auxiliary_loss": 0.0010236845817416906,
      "weighted_relation_loss": 0.0010236845817416906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.284377923292797,
      "flow_matching_loss": 0.06097394973039627,
      "grad_norm": 0.20950835943222046,
      "learning_rate": 9.714769215128596e-05,
      "loss": 0.062,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010026949224993587,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.304269790649414,
      "state_query_norm": 911.2627563476562,
      "state_relation_loss": 0.0010026949224993587,
      "step": 4580,
      "student_distance_mean": 1195.5877197265625,
      "teacher_distance_mean": 1.142809510231018,
      "weighted_auxiliary_loss": 0.0010026949224993587,
      "weighted_relation_loss": 0.0010026949224993587
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.293732460243218,
      "flow_matching_loss": 0.06148798391222954,
      "grad_norm": 0.20565882325172424,
      "learning_rate": 9.712931421250152e-05,
      "loss": 0.0625,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010115842102095484,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.58515110015869,
      "state_query_norm": 918.5130615234375,
      "state_relation_loss": 0.0010115842102095484,
      "step": 4590,
      "student_distance_mean": 1209.0933959960937,
      "teacher_distance_mean": 1.1562533378601074,
      "weighted_auxiliary_loss": 0.0010115842102095484,
      "weighted_relation_loss": 0.0010115842102095484
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.303086997193639,
      "flow_matching_loss": 0.06252912692725658,
      "grad_norm": 0.1515369415283203,
      "learning_rate": 9.711087900715627e-05,
      "loss": 0.0636,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010264841141179204,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.38353900909424,
      "state_query_norm": 917.3600402832031,
      "state_relation_loss": 0.0010264841141179204,
      "step": 4600,
      "student_distance_mean": 1201.35185546875,
      "teacher_distance_mean": 1.1473510146141053,
      "weighted_auxiliary_loss": 0.0010264841141179204,
      "weighted_relation_loss": 0.0010264841141179204
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.31244153414406,
      "flow_matching_loss": 0.06157912611961365,
      "grad_norm": 0.14916731417179108,
      "learning_rate": 9.709238655765078e-05,
      "loss": 0.0626,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001054473390104249,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.367045402526855,
      "state_query_norm": 913.6577209472656,
      "state_relation_loss": 0.001054473390104249,
      "step": 4610,
      "student_distance_mean": 1198.6176513671876,
      "teacher_distance_mean": 1.1596540212631226,
      "weighted_auxiliary_loss": 0.001054473390104249,
      "weighted_relation_loss": 0.001054473390104249
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.3217960710944805,
      "flow_matching_loss": 0.060056252777576445,
      "grad_norm": 0.16912637650966644,
      "learning_rate": 9.707383688645511e-05,
      "loss": 0.0611,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010577210632618517,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.08152542114258,
      "state_query_norm": 943.683203125,
      "state_relation_loss": 0.0010577210632618517,
      "step": 4620,
      "student_distance_mean": 1239.0566040039062,
      "teacher_distance_mean": 1.1730785727500916,
      "weighted_auxiliary_loss": 0.0010577210632618517,
      "weighted_relation_loss": 0.0010577210632618517
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.331150608044902,
      "flow_matching_loss": 0.06045241989195347,
      "grad_norm": 0.19954366981983185,
      "learning_rate": 9.705523001610883e-05,
      "loss": 0.0615,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001057524880161509,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.932540321350096,
      "state_query_norm": 934.381396484375,
      "state_relation_loss": 0.001057524880161509,
      "step": 4630,
      "student_distance_mean": 1227.406591796875,
      "teacher_distance_mean": 1.1633740782737731,
      "weighted_auxiliary_loss": 0.001057524880161509,
      "weighted_relation_loss": 0.001057524880161509
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.340505144995323,
      "flow_matching_loss": 0.061776905134320256,
      "grad_norm": 0.18386664986610413,
      "learning_rate": 9.703656596922107e-05,
      "loss": 0.0628,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010521979245822876,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.779433250427246,
      "state_query_norm": 928.3874206542969,
      "state_relation_loss": 0.0010521979245822876,
      "step": 4640,
      "student_distance_mean": 1219.477880859375,
      "teacher_distance_mean": 1.1514361262321473,
      "weighted_auxiliary_loss": 0.0010521979245822876,
      "weighted_relation_loss": 0.0010521979245822876
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.349859681945744,
      "flow_matching_loss": 0.06195047944784164,
      "grad_norm": 0.22209881246089935,
      "learning_rate": 9.70178447684704e-05,
      "loss": 0.063,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010684320121072232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.62687244415283,
      "state_query_norm": 920.5624267578125,
      "state_relation_loss": 0.0010684320121072232,
      "step": 4650,
      "student_distance_mean": 1209.6454223632813,
      "teacher_distance_mean": 1.1599594950675964,
      "weighted_auxiliary_loss": 0.0010684320121072232,
      "weighted_relation_loss": 0.0010684320121072232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.359214218896165,
      "flow_matching_loss": 0.05908207669854164,
      "grad_norm": 0.15276803076267242,
      "learning_rate": 9.699906643660483e-05,
      "loss": 0.0601,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010065419133752584,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.371796989440917,
      "state_query_norm": 909.7534790039062,
      "state_relation_loss": 0.0010065419133752584,
      "step": 4660,
      "student_distance_mean": 1198.8306030273438,
      "teacher_distance_mean": 1.147034239768982,
      "weighted_auxiliary_loss": 0.0010065419133752584,
      "weighted_relation_loss": 0.0010065419133752584
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.368568755846586,
      "flow_matching_loss": 0.057823384180665016,
      "grad_norm": 0.17401988804340363,
      "learning_rate": 9.698023099644185e-05,
      "loss": 0.0588,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009668952377978712,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.48394889831543,
      "state_query_norm": 911.2006530761719,
      "state_relation_loss": 0.0009668952377978712,
      "step": 4670,
      "student_distance_mean": 1204.4806884765626,
      "teacher_distance_mean": 1.151465380191803,
      "weighted_auxiliary_loss": 0.0009668952377978712,
      "weighted_relation_loss": 0.0009668952377978712
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.377923292797006,
      "flow_matching_loss": 0.06204337999224663,
      "grad_norm": 0.20451588928699493,
      "learning_rate": 9.696133847086823e-05,
      "loss": 0.063,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009661897318437695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.484501457214357,
      "state_query_norm": 909.1344360351562,
      "state_relation_loss": 0.0009661897318437695,
      "step": 4680,
      "student_distance_mean": 1203.9071533203125,
      "teacher_distance_mean": 1.1482334971427917,
      "weighted_auxiliary_loss": 0.0009661897318437695,
      "weighted_relation_loss": 0.0009661897318437695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.387277829747427,
      "flow_matching_loss": 0.06118642576038837,
      "grad_norm": 0.204630047082901,
      "learning_rate": 9.694238888284022e-05,
      "loss": 0.0622,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00100019559613429,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.845648765563965,
      "state_query_norm": 931.4640014648437,
      "state_relation_loss": 0.00100019559613429,
      "step": 4690,
      "student_distance_mean": 1225.7524658203124,
      "teacher_distance_mean": 1.167278778553009,
      "weighted_auxiliary_loss": 0.00100019559613429,
      "weighted_relation_loss": 0.00100019559613429
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.396632366697848,
      "flow_matching_loss": 0.05794200152158737,
      "grad_norm": 0.13930359482765198,
      "learning_rate": 9.692338225538333e-05,
      "loss": 0.059,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001015056186588481,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.585604858398437,
      "state_query_norm": 918.94013671875,
      "state_relation_loss": 0.001015056186588481,
      "step": 4700,
      "student_distance_mean": 1209.201416015625,
      "teacher_distance_mean": 1.1497017383575439,
      "weighted_auxiliary_loss": 0.001015056186588481,
      "weighted_relation_loss": 0.001015056186588481
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.4059869036482695,
      "flow_matching_loss": 0.059277668222785,
      "grad_norm": 0.1442646086215973,
      "learning_rate": 9.690431861159241e-05,
      "loss": 0.0603,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001034323702333495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.608759307861327,
      "state_query_norm": 925.9995483398437,
      "state_relation_loss": 0.001034323702333495,
      "step": 4710,
      "student_distance_mean": 1215.482275390625,
      "teacher_distance_mean": 1.14324893951416,
      "weighted_auxiliary_loss": 0.001034323702333495,
      "weighted_relation_loss": 0.001034323702333495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.415341440598691,
      "flow_matching_loss": 0.06202391311526299,
      "grad_norm": 0.16048212349414825,
      "learning_rate": 9.688519797463161e-05,
      "loss": 0.0631,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010401061386801302,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.851583862304686,
      "state_query_norm": 934.1140869140625,
      "state_relation_loss": 0.0010401061386801302,
      "step": 4720,
      "student_distance_mean": 1225.666552734375,
      "teacher_distance_mean": 1.1544199943542481,
      "weighted_auxiliary_loss": 0.0010401061386801302,
      "weighted_relation_loss": 0.0010401061386801302
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.424695977549112,
      "flow_matching_loss": 0.05614550970494747,
      "grad_norm": 0.15562643110752106,
      "learning_rate": 9.686602036773426e-05,
      "loss": 0.0572,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010215920046903193,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.10341854095459,
      "state_query_norm": 941.43818359375,
      "state_relation_loss": 0.0010215920046903193,
      "step": 4730,
      "student_distance_mean": 1239.1068969726562,
      "teacher_distance_mean": 1.16261146068573,
      "weighted_auxiliary_loss": 0.0010215920046903193,
      "weighted_relation_loss": 0.0010215920046903193
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.434050514499532,
      "flow_matching_loss": 0.05903582461178303,
      "grad_norm": 0.19037891924381256,
      "learning_rate": 9.684678581420302e-05,
      "loss": 0.06,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009732325444929302,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.98509407043457,
      "state_query_norm": 940.4308288574218,
      "state_relation_loss": 0.0009732325444929302,
      "step": 4740,
      "student_distance_mean": 1233.2066040039062,
      "teacher_distance_mean": 1.1545543432235719,
      "weighted_auxiliary_loss": 0.0009732325444929302,
      "weighted_relation_loss": 0.0009732325444929302
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.443405051449953,
      "flow_matching_loss": 0.05934835486114025,
      "grad_norm": 0.16512620449066162,
      "learning_rate": 9.682749433740962e-05,
      "loss": 0.0603,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009833747113589197,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.761736488342287,
      "state_query_norm": 931.411865234375,
      "state_relation_loss": 0.0009833747113589197,
      "step": 4750,
      "student_distance_mean": 1221.7443603515626,
      "teacher_distance_mean": 1.1393617153167725,
      "weighted_auxiliary_loss": 0.0009833747113589197,
      "weighted_relation_loss": 0.0009833747113589197
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.452759588400374,
      "flow_matching_loss": 0.06003738008439541,
      "grad_norm": 0.11225531250238419,
      "learning_rate": 9.680814596079507e-05,
      "loss": 0.061,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000971581757767126,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.143515205383302,
      "state_query_norm": 946.6491943359375,
      "state_relation_loss": 0.000971581757767126,
      "step": 4760,
      "student_distance_mean": 1241.0630615234375,
      "teacher_distance_mean": 1.1693717241287231,
      "weighted_auxiliary_loss": 0.000971581757767126,
      "weighted_relation_loss": 0.000971581757767126
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.462114125350795,
      "flow_matching_loss": 0.06352456584572792,
      "grad_norm": 0.20902135968208313,
      "learning_rate": 9.678874070786945e-05,
      "loss": 0.0645,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010167768807150424,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.274003982543945,
      "state_query_norm": 951.4236389160156,
      "state_relation_loss": 0.0010167768807150424,
      "step": 4770,
      "student_distance_mean": 1251.2204223632812,
      "teacher_distance_mean": 1.1614799857139588,
      "weighted_auxiliary_loss": 0.0010167768807150424,
      "weighted_relation_loss": 0.0010167768807150424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.471468662301216,
      "flow_matching_loss": 0.06305555365979672,
      "grad_norm": 0.17513422667980194,
      "learning_rate": 9.676927860221199e-05,
      "loss": 0.0641,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001011597568867728,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.08226490020752,
      "state_query_norm": 935.8916381835937,
      "state_relation_loss": 0.001011597568867728,
      "step": 4780,
      "student_distance_mean": 1238.8247436523438,
      "teacher_distance_mean": 1.1653536915779115,
      "weighted_auxiliary_loss": 0.001011597568867728,
      "weighted_relation_loss": 0.001011597568867728
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.480823199251637,
      "flow_matching_loss": 0.06209450177848339,
      "grad_norm": 0.14450876414775848,
      "learning_rate": 9.674975966747097e-05,
      "loss": 0.0631,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010119273734744637,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.850765991210938,
      "state_query_norm": 919.5924865722657,
      "state_relation_loss": 0.0010119273734744637,
      "step": 4790,
      "student_distance_mean": 1219.5238525390625,
      "teacher_distance_mean": 1.1479776620864868,
      "weighted_auxiliary_loss": 0.0010119273734744637,
      "weighted_relation_loss": 0.0010119273734744637
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.490177736202058,
      "flow_matching_loss": 0.06349525973200798,
      "grad_norm": 0.18112830817699432,
      "learning_rate": 9.673018392736374e-05,
      "loss": 0.0645,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009645625192206353,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.939248466491698,
      "state_query_norm": 933.6039428710938,
      "state_relation_loss": 0.0009645625192206353,
      "step": 4800,
      "student_distance_mean": 1230.3451049804687,
      "teacher_distance_mean": 1.143717896938324,
      "weighted_auxiliary_loss": 0.0009645625192206353,
      "weighted_relation_loss": 0.0009645625192206353
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.499532273152479,
      "flow_matching_loss": 0.062079979851841924,
      "grad_norm": 0.1925283670425415,
      "learning_rate": 9.671055140567667e-05,
      "loss": 0.0631,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001031505543505773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.069429779052733,
      "state_query_norm": 937.9296813964844,
      "state_relation_loss": 0.001031505543505773,
      "step": 4810,
      "student_distance_mean": 1233.7389770507812,
      "teacher_distance_mean": 1.1467426896095276,
      "weighted_auxiliary_loss": 0.001031505543505773,
      "weighted_relation_loss": 0.001031505543505773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.5088868101029,
      "flow_matching_loss": 0.05950799249112606,
      "grad_norm": 0.23085421323776245,
      "learning_rate": 9.669086212626511e-05,
      "loss": 0.0605,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009506979491561651,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.864538955688477,
      "state_query_norm": 929.4881958007812,
      "state_relation_loss": 0.0009506979491561651,
      "step": 4820,
      "student_distance_mean": 1226.8374633789062,
      "teacher_distance_mean": 1.137363064289093,
      "weighted_auxiliary_loss": 0.0009506979491561651,
      "weighted_relation_loss": 0.0009506979491561651
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.518241347053321,
      "flow_matching_loss": 0.06155535764992237,
      "grad_norm": 0.17681831121444702,
      "learning_rate": 9.667111611305341e-05,
      "loss": 0.0625,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000936388096306473,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.255270195007324,
      "state_query_norm": 948.4407592773438,
      "state_relation_loss": 0.000936388096306473,
      "step": 4830,
      "student_distance_mean": 1246.315478515625,
      "teacher_distance_mean": 1.162479317188263,
      "weighted_auxiliary_loss": 0.000936388096306473,
      "weighted_relation_loss": 0.000936388096306473
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.527595884003742,
      "flow_matching_loss": 0.06174096949398518,
      "grad_norm": 0.1655592918395996,
      "learning_rate": 9.665131339003486e-05,
      "loss": 0.0628,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010232329717837275,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.265739250183106,
      "state_query_norm": 950.1158874511718,
      "state_relation_loss": 0.0010232329717837275,
      "step": 4840,
      "student_distance_mean": 1246.35615234375,
      "teacher_distance_mean": 1.1656245231628417,
      "weighted_auxiliary_loss": 0.0010232329717837275,
      "weighted_relation_loss": 0.0010232329717837275
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.536950420954163,
      "flow_matching_loss": 0.059608450159430504,
      "grad_norm": 0.19809432327747345,
      "learning_rate": 9.663145398127158e-05,
      "loss": 0.0606,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000987456733128056,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.819224548339843,
      "state_query_norm": 924.4114196777343,
      "state_relation_loss": 0.000987456733128056,
      "step": 4850,
      "student_distance_mean": 1222.4077026367188,
      "teacher_distance_mean": 1.131963813304901,
      "weighted_auxiliary_loss": 0.000987456733128056,
      "weighted_relation_loss": 0.000987456733128056
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.546304957904583,
      "flow_matching_loss": 0.05906212516129017,
      "grad_norm": 0.1336357146501541,
      "learning_rate": 9.661153791089467e-05,
      "loss": 0.06,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009763335576280952,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.71925582885742,
      "state_query_norm": 967.6433776855469,
      "state_relation_loss": 0.0009763335576280952,
      "step": 4860,
      "student_distance_mean": 1273.3395874023438,
      "teacher_distance_mean": 1.1594160199165344,
      "weighted_auxiliary_loss": 0.0009763335576280952,
      "weighted_relation_loss": 0.0009763335576280952
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.555659494855004,
      "flow_matching_loss": 0.058166321739554404,
      "grad_norm": 0.14032527804374695,
      "learning_rate": 9.659156520310402e-05,
      "loss": 0.0592,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.001004792482126504,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.47955188751221,
      "state_query_norm": 951.4026977539063,
      "state_relation_loss": 0.001004792482126504,
      "step": 4870,
      "student_distance_mean": 1258.5713134765624,
      "teacher_distance_mean": 1.1626646518707275,
      "weighted_auxiliary_loss": 0.001004792482126504,
      "weighted_relation_loss": 0.001004792482126504
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.5650140318054255,
      "flow_matching_loss": 0.0606164027005434,
      "grad_norm": 0.14863456785678864,
      "learning_rate": 9.657153588216835e-05,
      "loss": 0.0616,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009360017138533294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.033281517028808,
      "state_query_norm": 938.3299987792968,
      "state_relation_loss": 0.0009360017138533294,
      "step": 4880,
      "student_distance_mean": 1236.4344116210937,
      "teacher_distance_mean": 1.148675799369812,
      "weighted_auxiliary_loss": 0.0009360017138533294,
      "weighted_relation_loss": 0.0009360017138533294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.574368568755847,
      "flow_matching_loss": 0.061402928456664084,
      "grad_norm": 0.15999466180801392,
      "learning_rate": 9.655144997242516e-05,
      "loss": 0.0623,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009180062857922167,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.00348014831543,
      "state_query_norm": 928.0048217773438,
      "state_relation_loss": 0.0009180062857922167,
      "step": 4890,
      "student_distance_mean": 1231.6714599609375,
      "teacher_distance_mean": 1.1505772948265076,
      "weighted_auxiliary_loss": 0.0009180062857922167,
      "weighted_relation_loss": 0.0009180062857922167
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.583723105706268,
      "flow_matching_loss": 0.060607961192727086,
      "grad_norm": 0.1579321324825287,
      "learning_rate": 9.653130749828075e-05,
      "loss": 0.0616,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009660880139563232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.02334518432617,
      "state_query_norm": 927.7783630371093,
      "state_relation_loss": 0.0009660880139563232,
      "step": 4900,
      "student_distance_mean": 1232.0077758789062,
      "teacher_distance_mean": 1.145547831058502,
      "weighted_auxiliary_loss": 0.0009660880139563232,
      "weighted_relation_loss": 0.0009660880139563232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.593077642656689,
      "flow_matching_loss": 0.05914253406226635,
      "grad_norm": 0.12805137038230896,
      "learning_rate": 9.65111084842101e-05,
      "loss": 0.0601,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009964481578208506,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 21.870430374145506,
      "state_query_norm": 934.3760559082032,
      "state_relation_loss": 0.0009964481578208506,
      "step": 4910,
      "student_distance_mean": 1225.9715087890625,
      "teacher_distance_mean": 1.1464900374412537,
      "weighted_auxiliary_loss": 0.0009964481578208506,
      "weighted_relation_loss": 0.0009964481578208506
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.60243217960711,
      "flow_matching_loss": 0.06008644960820675,
      "grad_norm": 0.14358992874622345,
      "learning_rate": 9.649085295475695e-05,
      "loss": 0.0611,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010434807685669512,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.2419792175293,
      "state_query_norm": 946.6627258300781,
      "state_relation_loss": 0.0010434807685669512,
      "step": 4920,
      "student_distance_mean": 1245.478515625,
      "teacher_distance_mean": 1.1614628911018372,
      "weighted_auxiliary_loss": 0.0010434807685669512,
      "weighted_relation_loss": 0.0010434807685669512
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.61178671655753,
      "flow_matching_loss": 0.05611129775643349,
      "grad_norm": 0.13881775736808777,
      "learning_rate": 9.647054093453365e-05,
      "loss": 0.0571,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009757551946677267,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.148769760131835,
      "state_query_norm": 952.4428833007812,
      "state_relation_loss": 0.0009757551946677267,
      "step": 4930,
      "student_distance_mean": 1244.0886474609374,
      "teacher_distance_mean": 1.1502949953079225,
      "weighted_auxiliary_loss": 0.0009757551946677267,
      "weighted_relation_loss": 0.0009757551946677267
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.621141253507951,
      "flow_matching_loss": 0.05888822115957737,
      "grad_norm": 0.16272900998592377,
      "learning_rate": 9.645017244822123e-05,
      "loss": 0.0599,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010277948633302004,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.432292175292968,
      "state_query_norm": 946.0485961914062,
      "state_relation_loss": 0.0010277948633302004,
      "step": 4940,
      "student_distance_mean": 1252.5858276367187,
      "teacher_distance_mean": 1.1631193041801453,
      "weighted_auxiliary_loss": 0.0010277948633302004,
      "weighted_relation_loss": 0.0010277948633302004
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.630495790458372,
      "flow_matching_loss": 0.05870286487042904,
      "grad_norm": 0.12226808071136475,
      "learning_rate": 9.642974752056931e-05,
      "loss": 0.0597,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009541264153085649,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.45389747619629,
      "state_query_norm": 954.2419677734375,
      "state_relation_loss": 0.0009541264153085649,
      "step": 4950,
      "student_distance_mean": 1259.3507202148437,
      "teacher_distance_mean": 1.1650814890861512,
      "weighted_auxiliary_loss": 0.0009541264153085649,
      "weighted_relation_loss": 0.0009541264153085649
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.639850327408793,
      "flow_matching_loss": 0.06038271449506283,
      "grad_norm": 0.12406609952449799,
      "learning_rate": 9.640926617639613e-05,
      "loss": 0.0614,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009765610739123076,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.453820419311523,
      "state_query_norm": 955.1663879394531,
      "state_relation_loss": 0.0009765610739123076,
      "step": 4960,
      "student_distance_mean": 1261.1055053710938,
      "teacher_distance_mean": 1.1584358215332031,
      "weighted_auxiliary_loss": 0.0009765610739123076,
      "weighted_relation_loss": 0.0009765610739123076
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.6492048643592145,
      "flow_matching_loss": 0.05902659818530083,
      "grad_norm": 0.12183074653148651,
      "learning_rate": 9.638872844058843e-05,
      "loss": 0.06,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009596493560820818,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.72193851470947,
      "state_query_norm": 961.7709228515625,
      "state_relation_loss": 0.0009596493560820818,
      "step": 4970,
      "student_distance_mean": 1271.4364501953125,
      "teacher_distance_mean": 1.16543447971344,
      "weighted_auxiliary_loss": 0.0009596493560820818,
      "weighted_relation_loss": 0.0009596493560820818
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.658559401309635,
      "flow_matching_loss": 0.0595804650336504,
      "grad_norm": 0.17292307317256927,
      "learning_rate": 9.63681343381015e-05,
      "loss": 0.0606,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010049735370557754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.281280517578125,
      "state_query_norm": 948.3658447265625,
      "state_relation_loss": 0.0010049735370557754,
      "step": 4980,
      "student_distance_mean": 1247.7301635742188,
      "teacher_distance_mean": 1.1426178216934204,
      "weighted_auxiliary_loss": 0.0010049735370557754,
      "weighted_relation_loss": 0.0010049735370557754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.667913938260056,
      "flow_matching_loss": 0.06140161827206612,
      "grad_norm": 0.17430830001831055,
      "learning_rate": 9.634748389395914e-05,
      "loss": 0.0624,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009673227847088129,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.052608489990234,
      "state_query_norm": 938.5183166503906,
      "state_relation_loss": 0.0009673227847088129,
      "step": 4990,
      "student_distance_mean": 1234.5803466796874,
      "teacher_distance_mean": 1.137339973449707,
      "weighted_auxiliary_loss": 0.0009673227847088129,
      "weighted_relation_loss": 0.0009673227847088129
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.677268475210477,
      "flow_matching_loss": 0.06009722463786602,
      "grad_norm": 0.17153723537921906,
      "learning_rate": 9.632677713325353e-05,
      "loss": 0.061,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009507472510449589,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.193123626708985,
      "state_query_norm": 941.9546508789062,
      "state_relation_loss": 0.0009507472510449589,
      "step": 5000,
      "student_distance_mean": 1243.0999633789063,
      "teacher_distance_mean": 1.1537708640098572,
      "weighted_auxiliary_loss": 0.0009507472510449589,
      "weighted_relation_loss": 0.0009507472510449589
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.686623012160898,
      "flow_matching_loss": 0.057904092594981194,
      "grad_norm": 0.19199195504188538,
      "learning_rate": 9.63060140811454e-05,
      "loss": 0.0589,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010057318722829223,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.51002655029297,
      "state_query_norm": 946.0360656738281,
      "state_relation_loss": 0.0010057318722829223,
      "step": 5010,
      "student_distance_mean": 1256.1741333007812,
      "teacher_distance_mean": 1.1595860362052917,
      "weighted_auxiliary_loss": 0.0010057318722829223,
      "weighted_relation_loss": 0.0010057318722829223
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.695977549111319,
      "flow_matching_loss": 0.05720580704510212,
      "grad_norm": 0.17032937705516815,
      "learning_rate": 9.628519476286379e-05,
      "loss": 0.0582,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010008592333178967,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.17185878753662,
      "state_query_norm": 941.2516235351562,
      "state_relation_loss": 0.0010008592333178967,
      "step": 5020,
      "student_distance_mean": 1242.6795532226563,
      "teacher_distance_mean": 1.142333209514618,
      "weighted_auxiliary_loss": 0.0010008592333178967,
      "weighted_relation_loss": 0.0010008592333178967
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.70533208606174,
      "flow_matching_loss": 0.058770664408802985,
      "grad_norm": 0.2553747296333313,
      "learning_rate": 9.626431920370612e-05,
      "loss": 0.0597,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009476788865868003,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.326325035095216,
      "state_query_norm": 950.4345092773438,
      "state_relation_loss": 0.0009476788865868003,
      "step": 5030,
      "student_distance_mean": 1252.5176147460938,
      "teacher_distance_mean": 1.13669251203537,
      "weighted_auxiliary_loss": 0.0009476788865868003,
      "weighted_relation_loss": 0.0009476788865868003
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.714686623012161,
      "flow_matching_loss": 0.059935740754008295,
      "grad_norm": 0.18004073202610016,
      "learning_rate": 9.624338742903819e-05,
      "loss": 0.0609,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009697107423562556,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.39676914215088,
      "state_query_norm": 949.0685852050781,
      "state_relation_loss": 0.0009697107423562556,
      "step": 5040,
      "student_distance_mean": 1254.9331298828124,
      "teacher_distance_mean": 1.1473371148109437,
      "weighted_auxiliary_loss": 0.0009697107423562556,
      "weighted_relation_loss": 0.0009697107423562556
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.724041159962582,
      "flow_matching_loss": 0.060578296333551405,
      "grad_norm": 0.2013590782880783,
      "learning_rate": 9.622239946429406e-05,
      "loss": 0.0615,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009293617156799882,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.87989559173584,
      "state_query_norm": 968.8873107910156,
      "state_relation_loss": 0.0009293617156799882,
      "step": 5050,
      "student_distance_mean": 1280.83720703125,
      "teacher_distance_mean": 1.1547225952148437,
      "weighted_auxiliary_loss": 0.0009293617156799882,
      "weighted_relation_loss": 0.0009293617156799882
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.733395696913003,
      "flow_matching_loss": 0.061778100579977034,
      "grad_norm": 0.19648487865924835,
      "learning_rate": 9.620135533497609e-05,
      "loss": 0.0627,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009689110505860299,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.5940523147583,
      "state_query_norm": 961.8879821777343,
      "state_relation_loss": 0.0009689110505860299,
      "step": 5060,
      "student_distance_mean": 1266.3015747070312,
      "teacher_distance_mean": 1.1391722559928894,
      "weighted_auxiliary_loss": 0.0009689110505860299,
      "weighted_relation_loss": 0.0009689110505860299
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.742750233863424,
      "flow_matching_loss": 0.05881409235298633,
      "grad_norm": 0.15495626628398895,
      "learning_rate": 9.61802550666549e-05,
      "loss": 0.0598,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000966739299474284,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.38996276855469,
      "state_query_norm": 955.2772583007812,
      "state_relation_loss": 0.000966739299474284,
      "step": 5070,
      "student_distance_mean": 1252.0634399414062,
      "teacher_distance_mean": 1.1434747099876403,
      "weighted_auxiliary_loss": 0.000966739299474284,
      "weighted_relation_loss": 0.000966739299474284
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.752104770813845,
      "flow_matching_loss": 0.059654251486063,
      "grad_norm": 0.20085681974887848,
      "learning_rate": 9.615909868496928e-05,
      "loss": 0.0606,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009643943340051919,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.936357498168945,
      "state_query_norm": 974.5832458496094,
      "state_relation_loss": 0.0009643943340051919,
      "step": 5080,
      "student_distance_mean": 1284.7193359375,
      "teacher_distance_mean": 1.1577672600746154,
      "weighted_auxiliary_loss": 0.0009643943340051919,
      "weighted_relation_loss": 0.0009643943340051919
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.761459307764266,
      "flow_matching_loss": 0.061951649561524394,
      "grad_norm": 0.19780860841274261,
      "learning_rate": 9.613788621562622e-05,
      "loss": 0.0629,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009588920511305332,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.75831527709961,
      "state_query_norm": 964.9334533691406,
      "state_relation_loss": 0.0009588920511305332,
      "step": 5090,
      "student_distance_mean": 1272.8251098632813,
      "teacher_distance_mean": 1.1468449115753174,
      "weighted_auxiliary_loss": 0.0009588920511305332,
      "weighted_relation_loss": 0.0009588920511305332
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.770813844714686,
      "flow_matching_loss": 0.06147348023951053,
      "grad_norm": 0.20162642002105713,
      "learning_rate": 9.611661768440093e-05,
      "loss": 0.0624,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009488932206295431,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.011916160583496,
      "state_query_norm": 966.0841857910157,
      "state_relation_loss": 0.0009488932206295431,
      "step": 5100,
      "student_distance_mean": 1287.1159545898438,
      "teacher_distance_mean": 1.1637944102287292,
      "weighted_auxiliary_loss": 0.0009488932206295431,
      "weighted_relation_loss": 0.0009488932206295431
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.780168381665107,
      "flow_matching_loss": 0.05704403556883335,
      "grad_norm": 0.22752273082733154,
      "learning_rate": 9.609529311713661e-05,
      "loss": 0.058,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009831567527726292,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.760149574279787,
      "state_query_norm": 960.8161987304687,
      "state_relation_loss": 0.0009831567527726292,
      "step": 5110,
      "student_distance_mean": 1271.907666015625,
      "teacher_distance_mean": 1.154425287246704,
      "weighted_auxiliary_loss": 0.0009831567527726292,
      "weighted_relation_loss": 0.0009831567527726292
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.789522918615528,
      "flow_matching_loss": 0.057885149121284486,
      "grad_norm": 0.2490161806344986,
      "learning_rate": 9.607391253974466e-05,
      "loss": 0.0589,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010140148166101425,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.263697242736818,
      "state_query_norm": 945.407421875,
      "state_relation_loss": 0.0010140148166101425,
      "step": 5120,
      "student_distance_mean": 1247.638427734375,
      "teacher_distance_mean": 1.130066978931427,
      "weighted_auxiliary_loss": 0.0010140148166101425,
      "weighted_relation_loss": 0.0010140148166101425
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.7988774555659495,
      "flow_matching_loss": 0.05909014008939266,
      "grad_norm": 0.20991559326648712,
      "learning_rate": 9.605247597820448e-05,
      "loss": 0.0601,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009834331169258804,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.377702140808104,
      "state_query_norm": 949.9300415039063,
      "state_relation_loss": 0.0009834331169258804,
      "step": 5130,
      "student_distance_mean": 1253.950390625,
      "teacher_distance_mean": 1.1372756600379943,
      "weighted_auxiliary_loss": 0.0009834331169258804,
      "weighted_relation_loss": 0.0009834331169258804
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.808231992516371,
      "flow_matching_loss": 0.06055972464382649,
      "grad_norm": 0.2216065227985382,
      "learning_rate": 9.603098345856354e-05,
      "loss": 0.0615,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009805318724829704,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.382287979125977,
      "state_query_norm": 949.7417175292969,
      "state_relation_loss": 0.0009805318724829704,
      "step": 5140,
      "student_distance_mean": 1253.6763427734375,
      "teacher_distance_mean": 1.1467603206634522,
      "weighted_auxiliary_loss": 0.0009805318724829704,
      "weighted_relation_loss": 0.0009805318724829704
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.817586529466792,
      "flow_matching_loss": 0.06010716371238232,
      "grad_norm": 0.24414609372615814,
      "learning_rate": 9.600943500693725e-05,
      "loss": 0.061,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009328082669526339,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.53838348388672,
      "state_query_norm": 956.3105529785156,
      "state_relation_loss": 0.0009328082669526339,
      "step": 5150,
      "student_distance_mean": 1263.586279296875,
      "teacher_distance_mean": 1.1448371171951295,
      "weighted_auxiliary_loss": 0.0009328082669526339,
      "weighted_relation_loss": 0.0009328082669526339
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.826941066417213,
      "flow_matching_loss": 0.059084751084446906,
      "grad_norm": 0.16949769854545593,
      "learning_rate": 9.598783064950902e-05,
      "loss": 0.06,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009466750663705171,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.031368827819826,
      "state_query_norm": 972.3361450195313,
      "state_relation_loss": 0.0009466750663705171,
      "step": 5160,
      "student_distance_mean": 1288.5492431640625,
      "teacher_distance_mean": 1.159613883495331,
      "weighted_auxiliary_loss": 0.0009466750663705171,
      "weighted_relation_loss": 0.0009466750663705171
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.836295603367633,
      "flow_matching_loss": 0.0575584352016449,
      "grad_norm": 0.17703039944171906,
      "learning_rate": 9.596617041253018e-05,
      "loss": 0.0585,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009399178612511605,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.82541217803955,
      "state_query_norm": 969.5875427246094,
      "state_relation_loss": 0.0009399178612511605,
      "step": 5170,
      "student_distance_mean": 1278.3114013671875,
      "teacher_distance_mean": 1.1430239796638488,
      "weighted_auxiliary_loss": 0.0009399178612511605,
      "weighted_relation_loss": 0.0009399178612511605
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.845650140318054,
      "flow_matching_loss": 0.058947717398405076,
      "grad_norm": 0.19700153172016144,
      "learning_rate": 9.594445432231996e-05,
      "loss": 0.0599,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009898255812004209,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.565270614624023,
      "state_query_norm": 954.3418579101562,
      "state_relation_loss": 0.0009898255812004209,
      "step": 5180,
      "student_distance_mean": 1262.7852416992187,
      "teacher_distance_mean": 1.1361716628074645,
      "weighted_auxiliary_loss": 0.0009898255812004209,
      "weighted_relation_loss": 0.0009898255812004209
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.855004677268475,
      "flow_matching_loss": 0.05924712233245373,
      "grad_norm": 0.20257756114006042,
      "learning_rate": 9.592268240526547e-05,
      "loss": 0.0602,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009433899656869471,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.409083747863768,
      "state_query_norm": 954.6501281738281,
      "state_relation_loss": 0.0009433899656869471,
      "step": 5190,
      "student_distance_mean": 1255.27333984375,
      "teacher_distance_mean": 1.1294666290283204,
      "weighted_auxiliary_loss": 0.0009433899656869471,
      "weighted_relation_loss": 0.0009433899656869471
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.864359214218896,
      "flow_matching_loss": 0.05709194727241993,
      "grad_norm": 0.1654173582792282,
      "learning_rate": 9.590085468782162e-05,
      "loss": 0.0581,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009713212668430061,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.231773948669435,
      "state_query_norm": 985.5262634277344,
      "state_relation_loss": 0.0009713212668430061,
      "step": 5200,
      "student_distance_mean": 1300.3269653320312,
      "teacher_distance_mean": 1.1706985473632812,
      "weighted_auxiliary_loss": 0.0009713212668430061,
      "weighted_relation_loss": 0.0009713212668430061
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.873713751169317,
      "flow_matching_loss": 0.06128414869308472,
      "grad_norm": 0.1348421722650528,
      "learning_rate": 9.587897119651116e-05,
      "loss": 0.0623,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010083395696710794,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.286575317382812,
      "state_query_norm": 985.5228210449219,
      "state_relation_loss": 0.0010083395696710794,
      "step": 5210,
      "student_distance_mean": 1304.080517578125,
      "teacher_distance_mean": 1.1640434622764588,
      "weighted_auxiliary_loss": 0.0010083395696710794,
      "weighted_relation_loss": 0.0010083395696710794
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.883068288119738,
      "flow_matching_loss": 0.05919506438076496,
      "grad_norm": 0.1345810443162918,
      "learning_rate": 9.585703195792459e-05,
      "loss": 0.0601,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009241015999577939,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.50185604095459,
      "state_query_norm": 996.1209899902344,
      "state_relation_loss": 0.0009241015999577939,
      "step": 5220,
      "student_distance_mean": 1315.8386352539062,
      "teacher_distance_mean": 1.1630983233451844,
      "weighted_auxiliary_loss": 0.0009241015999577939,
      "weighted_relation_loss": 0.0009241015999577939
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.892422825070159,
      "flow_matching_loss": 0.05762579552829265,
      "grad_norm": 0.11418034136295319,
      "learning_rate": 9.583503699872016e-05,
      "loss": 0.0586,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010084554087370634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.99597759246826,
      "state_query_norm": 976.3986389160157,
      "state_relation_loss": 0.0010084554087370634,
      "step": 5230,
      "student_distance_mean": 1289.3386840820312,
      "teacher_distance_mean": 1.1431981563568114,
      "weighted_auxiliary_loss": 0.0010084554087370634,
      "weighted_relation_loss": 0.0010084554087370634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.90177736202058,
      "flow_matching_loss": 0.05850395783782005,
      "grad_norm": 0.1274721622467041,
      "learning_rate": 9.581298634562381e-05,
      "loss": 0.0595,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009835069300606846,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.058894920349122,
      "state_query_norm": 974.4572082519531,
      "state_relation_loss": 0.0009835069300606846,
      "step": 5240,
      "student_distance_mean": 1291.2283203125,
      "teacher_distance_mean": 1.1431464672088623,
      "weighted_auxiliary_loss": 0.0009835069300606846,
      "weighted_relation_loss": 0.0009835069300606846
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.911131898971001,
      "flow_matching_loss": 0.06023257896304131,
      "grad_norm": 0.17580269277095795,
      "learning_rate": 9.579088002542917e-05,
      "loss": 0.0613,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010251756524667145,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.939052200317384,
      "state_query_norm": 968.4556762695313,
      "state_relation_loss": 0.0010251756524667145,
      "step": 5250,
      "student_distance_mean": 1280.7944213867188,
      "teacher_distance_mean": 1.1382758736610412,
      "weighted_auxiliary_loss": 0.0010251756524667145,
      "weighted_relation_loss": 0.0010251756524667145
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.920486435921422,
      "flow_matching_loss": 0.0596564032137394,
      "grad_norm": 0.10488204658031464,
      "learning_rate": 9.57687180649975e-05,
      "loss": 0.0606,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009552438335958868,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.412686157226563,
      "state_query_norm": 991.950830078125,
      "state_relation_loss": 0.0009552438335958868,
      "step": 5260,
      "student_distance_mean": 1310.8240600585937,
      "teacher_distance_mean": 1.1624203324317932,
      "weighted_auxiliary_loss": 0.0009552438335958868,
      "weighted_relation_loss": 0.0009552438335958868
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.929840972871843,
      "flow_matching_loss": 0.05702955760061741,
      "grad_norm": 0.16159872710704803,
      "learning_rate": 9.574650049125768e-05,
      "loss": 0.058,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000928573968121782,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.61591682434082,
      "state_query_norm": 1002.9892822265625,
      "state_relation_loss": 0.000928573968121782,
      "step": 5270,
      "student_distance_mean": 1321.6224243164063,
      "teacher_distance_mean": 1.1742310643196106,
      "weighted_auxiliary_loss": 0.000928573968121782,
      "weighted_relation_loss": 0.000928573968121782
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.939195509822264,
      "flow_matching_loss": 0.0583682831376791,
      "grad_norm": 0.2488957941532135,
      "learning_rate": 9.572422733120614e-05,
      "loss": 0.0593,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009401082061231136,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.83637237548828,
      "state_query_norm": 972.00322265625,
      "state_relation_loss": 0.0009401082061231136,
      "step": 5280,
      "student_distance_mean": 1279.4180053710938,
      "teacher_distance_mean": 1.1353227853775025,
      "weighted_auxiliary_loss": 0.0009401082061231136,
      "weighted_relation_loss": 0.0009401082061231136
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.948550046772684,
      "flow_matching_loss": 0.06008892431855202,
      "grad_norm": 0.14223235845565796,
      "learning_rate": 9.570189861190689e-05,
      "loss": 0.061,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000944878940936178,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.491666984558105,
      "state_query_norm": 1000.0428405761719,
      "state_relation_loss": 0.000944878940936178,
      "step": 5290,
      "student_distance_mean": 1315.6930297851563,
      "teacher_distance_mean": 1.1663668394088744,
      "weighted_auxiliary_loss": 0.000944878940936178,
      "weighted_relation_loss": 0.000944878940936178
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.9579045837231055,
      "flow_matching_loss": 0.05672279484570027,
      "grad_norm": 0.22439730167388916,
      "learning_rate": 9.56795143604914e-05,
      "loss": 0.0577,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009754897502716631,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.827354621887206,
      "state_query_norm": 968.6460571289062,
      "state_relation_loss": 0.0009754897502716631,
      "step": 5300,
      "student_distance_mean": 1277.6951293945312,
      "teacher_distance_mean": 1.1354077458381653,
      "weighted_auxiliary_loss": 0.0009754897502716631,
      "weighted_relation_loss": 0.0009754897502716631
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.967259120673527,
      "flow_matching_loss": 0.05768042281270027,
      "grad_norm": 0.1351708173751831,
      "learning_rate": 9.565707460415869e-05,
      "loss": 0.0587,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009788055322133005,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.498548698425292,
      "state_query_norm": 986.6535400390625,
      "state_relation_loss": 0.0009788055322133005,
      "step": 5310,
      "student_distance_mean": 1309.5163330078126,
      "teacher_distance_mean": 1.162442922592163,
      "weighted_auxiliary_loss": 0.0009788055322133005,
      "weighted_relation_loss": 0.0009788055322133005
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.976613657623948,
      "flow_matching_loss": 0.056654227897524834,
      "grad_norm": 0.15519441664218903,
      "learning_rate": 9.563457937017515e-05,
      "loss": 0.0576,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009568508365191519,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.030282020568848,
      "state_query_norm": 974.2480102539063,
      "state_relation_loss": 0.0009568508365191519,
      "step": 5320,
      "student_distance_mean": 1289.9420043945313,
      "teacher_distance_mean": 1.1352164626121521,
      "weighted_auxiliary_loss": 0.0009568508365191519,
      "weighted_relation_loss": 0.0009568508365191519
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.985968194574369,
      "flow_matching_loss": 0.05831940323114395,
      "grad_norm": 0.17055220901966095,
      "learning_rate": 9.56120286858746e-05,
      "loss": 0.0593,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000968499795999378,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.618523788452148,
      "state_query_norm": 989.8745422363281,
      "state_relation_loss": 0.000968499795999378,
      "step": 5330,
      "student_distance_mean": 1319.1547119140625,
      "teacher_distance_mean": 1.1631481170654296,
      "weighted_auxiliary_loss": 0.000968499795999378,
      "weighted_relation_loss": 0.000968499795999378
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 4.99532273152479,
      "flow_matching_loss": 0.061989843100309375,
      "grad_norm": 0.191450297832489,
      "learning_rate": 9.558942257865829e-05,
      "loss": 0.0629,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009431204292923212,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.308593559265137,
      "state_query_norm": 979.3922729492188,
      "state_relation_loss": 0.0009431204292923212,
      "step": 5340,
      "student_distance_mean": 1299.7523193359375,
      "teacher_distance_mean": 1.1426249861717224,
      "weighted_auxiliary_loss": 0.0009431204292923212,
      "weighted_relation_loss": 0.0009431204292923212
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.00467726847521,
      "flow_matching_loss": 0.05680481605231762,
      "grad_norm": 0.16768477857112885,
      "learning_rate": 9.556676107599472e-05,
      "loss": 0.0578,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0009622408833820373,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 23.29658432006836,
      "state_query_norm": 985.7153991699219,
      "state_relation_loss": 0.0009622408833820373,
      "step": 5350,
      "student_distance_mean": 1302.3911865234375,
      "teacher_distance_mean": 1.1495639562606812,
      "weighted_auxiliary_loss": 0.0009622408833820373,
      "weighted_relation_loss": 0.0009622408833820373
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.014031805425631,
      "flow_matching_loss": 0.06009898856282234,
      "grad_norm": 0.22961877286434174,
      "learning_rate": 9.554404420541978e-05,
      "loss": 0.0611,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009740928479004651,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.402357864379884,
      "state_query_norm": 982.2803405761719,
      "state_relation_loss": 0.0009740928479004651,
      "step": 5360,
      "student_distance_mean": 1309.099462890625,
      "teacher_distance_mean": 1.1533791661262511,
      "weighted_auxiliary_loss": 0.0009740928479004651,
      "weighted_relation_loss": 0.0009740928479004651
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.023386342376052,
      "flow_matching_loss": 0.056257544457912444,
      "grad_norm": 0.16647624969482422,
      "learning_rate": 9.55212719945366e-05,
      "loss": 0.0572,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009579761419445277,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.502399826049803,
      "state_query_norm": 992.1555358886719,
      "state_relation_loss": 0.0009579761419445277,
      "step": 5370,
      "student_distance_mean": 1313.7210693359375,
      "teacher_distance_mean": 1.1568437337875366,
      "weighted_auxiliary_loss": 0.0009579761419445277,
      "weighted_relation_loss": 0.0009579761419445277
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.032740879326473,
      "flow_matching_loss": 0.05515318512916565,
      "grad_norm": 0.22015458345413208,
      "learning_rate": 9.549844447101559e-05,
      "loss": 0.0561,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009479169151745736,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.470462226867674,
      "state_query_norm": 994.0835327148437,
      "state_relation_loss": 0.0009479169151745736,
      "step": 5380,
      "student_distance_mean": 1314.0315307617188,
      "teacher_distance_mean": 1.150702452659607,
      "weighted_auxiliary_loss": 0.0009479169151745736,
      "weighted_relation_loss": 0.0009479169151745736
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.0420954162768945,
      "flow_matching_loss": 0.05753067396581173,
      "grad_norm": 0.17036010324954987,
      "learning_rate": 9.547556166259433e-05,
      "loss": 0.0585,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009614097827579827,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.222580909729004,
      "state_query_norm": 981.8799865722656,
      "state_relation_loss": 0.0009614097827579827,
      "step": 5390,
      "student_distance_mean": 1297.2593994140625,
      "teacher_distance_mean": 1.1381472945213318,
      "weighted_auxiliary_loss": 0.0009614097827579827,
      "weighted_relation_loss": 0.0009614097827579827
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.051449953227316,
      "flow_matching_loss": 0.05434639938175678,
      "grad_norm": 0.157206729054451,
      "learning_rate": 9.545262359707756e-05,
      "loss": 0.0553,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009510072763077914,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.672765922546386,
      "state_query_norm": 1002.6566772460938,
      "state_relation_loss": 0.0009510072763077914,
      "step": 5400,
      "student_distance_mean": 1327.8721435546875,
      "teacher_distance_mean": 1.1605736136436462,
      "weighted_auxiliary_loss": 0.0009510072763077914,
      "weighted_relation_loss": 0.0009510072763077914
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.060804490177736,
      "flow_matching_loss": 0.056975755468010904,
      "grad_norm": 0.15422679483890533,
      "learning_rate": 9.542963030233724e-05,
      "loss": 0.0579,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009000597288832069,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.47278251647949,
      "state_query_norm": 982.7701599121094,
      "state_relation_loss": 0.0009000597288832069,
      "step": 5410,
      "student_distance_mean": 1309.0455810546875,
      "teacher_distance_mean": 1.1528385400772094,
      "weighted_auxiliary_loss": 0.0009000597288832069,
      "weighted_relation_loss": 0.0009000597288832069
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.070159027128157,
      "flow_matching_loss": 0.05607793778181076,
      "grad_norm": 0.1715058982372284,
      "learning_rate": 9.540658180631237e-05,
      "loss": 0.057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008968022943008691,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 22.856502342224122,
      "state_query_norm": 968.81904296875,
      "state_relation_loss": 0.0008968022943008691,
      "step": 5420,
      "student_distance_mean": 1280.5290771484374,
      "teacher_distance_mean": 1.1331905841827392,
      "weighted_auxiliary_loss": 0.0008968022943008691,
      "weighted_relation_loss": 0.0008968022943008691
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.079513564078578,
      "flow_matching_loss": 0.05607590973377228,
      "grad_norm": 0.20273391902446747,
      "learning_rate": 9.538347813700904e-05,
      "loss": 0.057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009717295353766531,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.094791412353516,
      "state_query_norm": 983.569140625,
      "state_relation_loss": 0.0009717295353766531,
      "step": 5430,
      "student_distance_mean": 1293.16962890625,
      "teacher_distance_mean": 1.140835702419281,
      "weighted_auxiliary_loss": 0.0009717295353766531,
      "weighted_relation_loss": 0.0009717295353766531
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.088868101028999,
      "flow_matching_loss": 0.05908995643258095,
      "grad_norm": 0.18113744258880615,
      "learning_rate": 9.536031932250036e-05,
      "loss": 0.06,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009155611332971603,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.62693977355957,
      "state_query_norm": 996.331298828125,
      "state_relation_loss": 0.0009155611332971603,
      "step": 5440,
      "student_distance_mean": 1318.8318115234374,
      "teacher_distance_mean": 1.1493406295776367,
      "weighted_auxiliary_loss": 0.0009155611332971603,
      "weighted_relation_loss": 0.0009155611332971603
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.09822263797942,
      "flow_matching_loss": 0.056305373832583425,
      "grad_norm": 0.2204769253730774,
      "learning_rate": 9.533710539092653e-05,
      "loss": 0.0573,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009553211915772408,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.081988906860353,
      "state_query_norm": 1014.5739318847657,
      "state_relation_loss": 0.0009553211915772408,
      "step": 5450,
      "student_distance_mean": 1345.3166381835938,
      "teacher_distance_mean": 1.1664481520652772,
      "weighted_auxiliary_loss": 0.0009553211915772408,
      "weighted_relation_loss": 0.0009553211915772408
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.107577174929841,
      "flow_matching_loss": 0.0571349062025547,
      "grad_norm": 0.17715507745742798,
      "learning_rate": 9.531383637049464e-05,
      "loss": 0.0581,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009379583003465086,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.547042655944825,
      "state_query_norm": 995.5287353515625,
      "state_relation_loss": 0.0009379583003465086,
      "step": 5460,
      "student_distance_mean": 1316.9524291992188,
      "teacher_distance_mean": 1.1554317116737365,
      "weighted_auxiliary_loss": 0.0009379583003465086,
      "weighted_relation_loss": 0.0009379583003465086
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.1169317118802615,
      "flow_matching_loss": 0.058018684014678,
      "grad_norm": 0.23899060487747192,
      "learning_rate": 9.529051228947875e-05,
      "loss": 0.059,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009877409378532321,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.58627986907959,
      "state_query_norm": 987.9650573730469,
      "state_relation_loss": 0.0009877409378532321,
      "step": 5470,
      "student_distance_mean": 1318.4378540039063,
      "teacher_distance_mean": 1.1512339949607848,
      "weighted_auxiliary_loss": 0.0009877409378532321,
      "weighted_relation_loss": 0.0009877409378532321
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.126286248830683,
      "flow_matching_loss": 0.05636413544416428,
      "grad_norm": 0.1452212929725647,
      "learning_rate": 9.52671331762198e-05,
      "loss": 0.0572,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008789078739937395,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.80296173095703,
      "state_query_norm": 1011.2097839355469,
      "state_relation_loss": 0.0008789078739937395,
      "step": 5480,
      "student_distance_mean": 1334.8532470703126,
      "teacher_distance_mean": 1.1571123242378234,
      "weighted_auxiliary_loss": 0.0008789078739937395,
      "weighted_relation_loss": 0.0008789078739937395
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.135640785781104,
      "flow_matching_loss": 0.05822174921631813,
      "grad_norm": 0.1388719528913498,
      "learning_rate": 9.524369905912565e-05,
      "loss": 0.0591,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000905337737640366,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.1054349899292,
      "state_query_norm": 1013.4976013183593,
      "state_relation_loss": 0.000905337737640366,
      "step": 5490,
      "student_distance_mean": 1345.6623901367188,
      "teacher_distance_mean": 1.1692411780357361,
      "weighted_auxiliary_loss": 0.000905337737640366,
      "weighted_relation_loss": 0.000905337737640366
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.144995322731525,
      "flow_matching_loss": 0.05605010837316513,
      "grad_norm": 0.16219383478164673,
      "learning_rate": 9.522020996667092e-05,
      "loss": 0.057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009242089872714132,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.96171417236328,
      "state_query_norm": 1018.4369995117188,
      "state_relation_loss": 0.0009242089872714132,
      "step": 5500,
      "student_distance_mean": 1342.3741333007813,
      "teacher_distance_mean": 1.1623955845832825,
      "weighted_auxiliary_loss": 0.0009242089872714132,
      "weighted_relation_loss": 0.0009242089872714132
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.154349859681946,
      "flow_matching_loss": 0.05784024931490421,
      "grad_norm": 0.1670115888118744,
      "learning_rate": 9.519666592739709e-05,
      "loss": 0.0588,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009465110255405307,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.51611042022705,
      "state_query_norm": 991.3208129882812,
      "state_relation_loss": 0.0009465110255405307,
      "step": 5510,
      "student_distance_mean": 1312.5510498046874,
      "teacher_distance_mean": 1.1368989109992982,
      "weighted_auxiliary_loss": 0.0009465110255405307,
      "weighted_relation_loss": 0.0009465110255405307
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.163704396632367,
      "flow_matching_loss": 0.05487678423523903,
      "grad_norm": 0.0982099324464798,
      "learning_rate": 9.517306696991241e-05,
      "loss": 0.0558,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009446488169487566,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.722127532958986,
      "state_query_norm": 999.3442077636719,
      "state_relation_loss": 0.0009446488169487566,
      "step": 5520,
      "student_distance_mean": 1327.1730102539063,
      "teacher_distance_mean": 1.147361147403717,
      "weighted_auxiliary_loss": 0.0009446488169487566,
      "weighted_relation_loss": 0.0009446488169487566
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.173058933582787,
      "flow_matching_loss": 0.05562197156250477,
      "grad_norm": 0.14928288757801056,
      "learning_rate": 9.51494131228918e-05,
      "loss": 0.0566,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009643825411330909,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.03196792602539,
      "state_query_norm": 1016.8061950683593,
      "state_relation_loss": 0.0009643825411330909,
      "step": 5530,
      "student_distance_mean": 1344.1677978515625,
      "teacher_distance_mean": 1.1600943207740784,
      "weighted_auxiliary_loss": 0.0009643825411330909,
      "weighted_relation_loss": 0.0009643825411330909
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.182413470533208,
      "flow_matching_loss": 0.05762588120996952,
      "grad_norm": 0.16927184164524078,
      "learning_rate": 9.512570441507695e-05,
      "loss": 0.0586,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010030006757006048,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.98661289215088,
      "state_query_norm": 1010.4011901855469,
      "state_relation_loss": 0.0010030006757006048,
      "step": 5540,
      "student_distance_mean": 1341.3672119140624,
      "teacher_distance_mean": 1.1614895820617677,
      "weighted_auxiliary_loss": 0.0010030006757006048,
      "weighted_relation_loss": 0.0010030006757006048
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.191768007483629,
      "flow_matching_loss": 0.057702282816171645,
      "grad_norm": 0.1527412384748459,
      "learning_rate": 9.510194087527615e-05,
      "loss": 0.0587,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009643344208598136,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.225298500061037,
      "state_query_norm": 981.8610229492188,
      "state_relation_loss": 0.0009643344208598136,
      "step": 5550,
      "student_distance_mean": 1297.2787353515625,
      "teacher_distance_mean": 1.1390271306037902,
      "weighted_auxiliary_loss": 0.0009643344208598136,
      "weighted_relation_loss": 0.0009643344208598136
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.2011225444340505,
      "flow_matching_loss": 0.056653580814599994,
      "grad_norm": 0.2527863383293152,
      "learning_rate": 9.507812253236435e-05,
      "loss": 0.0576,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000968475389527157,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.98434829711914,
      "state_query_norm": 1016.8227416992188,
      "state_relation_loss": 0.000968475389527157,
      "step": 5560,
      "student_distance_mean": 1341.4711547851562,
      "teacher_distance_mean": 1.155326735973358,
      "weighted_auxiliary_loss": 0.000968475389527157,
      "weighted_relation_loss": 0.000968475389527157
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.210477081384472,
      "flow_matching_loss": 0.05633353404700756,
      "grad_norm": 0.19643601775169373,
      "learning_rate": 9.505424941528309e-05,
      "loss": 0.0573,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009887084946967662,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.850411224365235,
      "state_query_norm": 1001.0579711914063,
      "state_relation_loss": 0.0009887084946967662,
      "step": 5570,
      "student_distance_mean": 1333.9932861328125,
      "teacher_distance_mean": 1.1377329349517822,
      "weighted_auxiliary_loss": 0.0009887084946967662,
      "weighted_relation_loss": 0.0009887084946967662
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.219831618334893,
      "flow_matching_loss": 0.05838188081979752,
      "grad_norm": 0.18496161699295044,
      "learning_rate": 9.503032155304046e-05,
      "loss": 0.0594,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009957184898667037,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.06293659210205,
      "state_query_norm": 1012.0622802734375,
      "state_relation_loss": 0.0009957184898667037,
      "step": 5580,
      "student_distance_mean": 1342.837109375,
      "teacher_distance_mean": 1.1495798110961915,
      "weighted_auxiliary_loss": 0.0009957184898667037,
      "weighted_relation_loss": 0.0009957184898667037
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.229186155285314,
      "flow_matching_loss": 0.05871273763477802,
      "grad_norm": 0.11882977932691574,
      "learning_rate": 9.500633897471106e-05,
      "loss": 0.0597,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0010046226787380874,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.664142036437987,
      "state_query_norm": 1031.496923828125,
      "state_relation_loss": 0.0010046226787380874,
      "step": 5590,
      "student_distance_mean": 1373.0992065429687,
      "teacher_distance_mean": 1.157350480556488,
      "weighted_auxiliary_loss": 0.0010046226787380874,
      "weighted_relation_loss": 0.0010046226787380874
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.238540692235734,
      "flow_matching_loss": 0.05599871389567852,
      "grad_norm": 0.18865983188152313,
      "learning_rate": 9.498230170943596e-05,
      "loss": 0.057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009676769550424069,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.187715911865233,
      "state_query_norm": 1015.7341918945312,
      "state_relation_loss": 0.0009676769550424069,
      "step": 5600,
      "student_distance_mean": 1350.9669555664063,
      "teacher_distance_mean": 1.1468792676925659,
      "weighted_auxiliary_loss": 0.0009676769550424069,
      "weighted_relation_loss": 0.0009676769550424069
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.247895229186155,
      "flow_matching_loss": 0.057614358887076375,
      "grad_norm": 0.15021438896656036,
      "learning_rate": 9.495820978642275e-05,
      "loss": 0.0586,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009627455903682858,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.835597801208497,
      "state_query_norm": 1002.0857788085938,
      "state_relation_loss": 0.0009627455903682858,
      "step": 5610,
      "student_distance_mean": 1330.3,
      "teacher_distance_mean": 1.1373219966888428,
      "weighted_auxiliary_loss": 0.0009627455903682858,
      "weighted_relation_loss": 0.0009627455903682858
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.257249766136576,
      "flow_matching_loss": 0.05496595613658428,
      "grad_norm": 0.15882331132888794,
      "learning_rate": 9.493406323494535e-05,
      "loss": 0.0559,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009753967635333538,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.09910068511963,
      "state_query_norm": 1007.6837097167969,
      "state_relation_loss": 0.0009753967635333538,
      "step": 5620,
      "student_distance_mean": 1345.78701171875,
      "teacher_distance_mean": 1.1370011210441588,
      "weighted_auxiliary_loss": 0.0009753967635333538,
      "weighted_relation_loss": 0.0009753967635333538
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.266604303086997,
      "flow_matching_loss": 0.05544258914887905,
      "grad_norm": 0.2195267230272293,
      "learning_rate": 9.490986208434413e-05,
      "loss": 0.0564,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000998881895793602,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.33470230102539,
      "state_query_norm": 1016.1888854980468,
      "state_relation_loss": 0.000998881895793602,
      "step": 5630,
      "student_distance_mean": 1353.8438232421875,
      "teacher_distance_mean": 1.144615149497986,
      "weighted_auxiliary_loss": 0.000998881895793602,
      "weighted_relation_loss": 0.000998881895793602
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.275958840037418,
      "flow_matching_loss": 0.05761600434780121,
      "grad_norm": 0.15049703419208527,
      "learning_rate": 9.488560636402577e-05,
      "loss": 0.0586,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009572198381647467,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.52350425720215,
      "state_query_norm": 1032.0977416992187,
      "state_relation_loss": 0.0009572198381647467,
      "step": 5640,
      "student_distance_mean": 1369.3447265625,
      "teacher_distance_mean": 1.158760631084442,
      "weighted_auxiliary_loss": 0.0009572198381647467,
      "weighted_relation_loss": 0.0009572198381647467
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.285313376987839,
      "flow_matching_loss": 0.05563274286687374,
      "grad_norm": 0.172663614153862,
      "learning_rate": 9.486129610346321e-05,
      "loss": 0.0566,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009631633758544922,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.0714900970459,
      "state_query_norm": 1014.3427124023438,
      "state_relation_loss": 0.0009631633758544922,
      "step": 5650,
      "student_distance_mean": 1341.9513916015626,
      "teacher_distance_mean": 1.1455072522163392,
      "weighted_auxiliary_loss": 0.0009631633758544922,
      "weighted_relation_loss": 0.0009631633758544922
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.29466791393826,
      "flow_matching_loss": 0.055123671516776086,
      "grad_norm": 0.15005972981452942,
      "learning_rate": 9.483693133219576e-05,
      "loss": 0.0561,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009358135343063622,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.850681114196778,
      "state_query_norm": 1045.1900024414062,
      "state_relation_loss": 0.0009358135343063622,
      "step": 5660,
      "student_distance_mean": 1387.057763671875,
      "teacher_distance_mean": 1.1788893461227417,
      "weighted_auxiliary_loss": 0.0009358135343063622,
      "weighted_relation_loss": 0.0009358135343063622
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.304022450888681,
      "flow_matching_loss": 0.057818103954195975,
      "grad_norm": 0.14100204408168793,
      "learning_rate": 9.481251207982888e-05,
      "loss": 0.0587,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009300841600634157,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.236711502075195,
      "state_query_norm": 1026.1345581054688,
      "state_relation_loss": 0.0009300841600634157,
      "step": 5670,
      "student_distance_mean": 1356.769091796875,
      "teacher_distance_mean": 1.15795156955719,
      "weighted_auxiliary_loss": 0.0009300841600634157,
      "weighted_relation_loss": 0.0009300841600634157
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.313376987839102,
      "flow_matching_loss": 0.056061134487390515,
      "grad_norm": 0.15678009390830994,
      "learning_rate": 9.47880383760343e-05,
      "loss": 0.057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009237576858140528,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.06719856262207,
      "state_query_norm": 1015.6839599609375,
      "state_relation_loss": 0.0009237576858140528,
      "step": 5680,
      "student_distance_mean": 1345.3032104492188,
      "teacher_distance_mean": 1.1475780844688415,
      "weighted_auxiliary_loss": 0.0009237576858140528,
      "weighted_relation_loss": 0.0009237576858140528
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.322731524789523,
      "flow_matching_loss": 0.05672193244099617,
      "grad_norm": 0.1539320945739746,
      "learning_rate": 9.476351025054983e-05,
      "loss": 0.0576,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009171729790978134,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 23.908107376098634,
      "state_query_norm": 1001.5156005859375,
      "state_relation_loss": 0.0009171729790978134,
      "step": 5690,
      "student_distance_mean": 1335.8162719726563,
      "teacher_distance_mean": 1.137979006767273,
      "weighted_auxiliary_loss": 0.0009171729790978134,
      "weighted_relation_loss": 0.0009171729790978134
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.332086061739944,
      "flow_matching_loss": 0.05735205449163914,
      "grad_norm": 0.194584921002388,
      "learning_rate": 9.473892773317952e-05,
      "loss": 0.0582,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008467351086437702,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.417834854125978,
      "state_query_norm": 1027.9483154296875,
      "state_relation_loss": 0.0008467351086437702,
      "step": 5700,
      "student_distance_mean": 1364.1886352539063,
      "teacher_distance_mean": 1.1606817960739135,
      "weighted_auxiliary_loss": 0.0008467351086437702,
      "weighted_relation_loss": 0.0008467351086437702
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.341440598690365,
      "flow_matching_loss": 0.054096310213208196,
      "grad_norm": 0.21720965206623077,
      "learning_rate": 9.471429085379338e-05,
      "loss": 0.055,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009029300126712769,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.6887508392334,
      "state_query_norm": 1037.1441772460937,
      "state_relation_loss": 0.0009029300126712769,
      "step": 5710,
      "student_distance_mean": 1376.9096313476562,
      "teacher_distance_mean": 1.1668609976768494,
      "weighted_auxiliary_loss": 0.0009029300126712769,
      "weighted_relation_loss": 0.0009029300126712769
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.3507951356407855,
      "flow_matching_loss": 0.054237513616681096,
      "grad_norm": 0.14597047865390778,
      "learning_rate": 9.468959964232757e-05,
      "loss": 0.0551,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008697077922988683,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.38226490020752,
      "state_query_norm": 1028.6933654785157,
      "state_relation_loss": 0.0008697077922988683,
      "step": 5720,
      "student_distance_mean": 1364.4807983398437,
      "teacher_distance_mean": 1.1531468391418458,
      "weighted_auxiliary_loss": 0.0008697077922988683,
      "weighted_relation_loss": 0.0008697077922988683
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.360149672591207,
      "flow_matching_loss": 0.058267517015337944,
      "grad_norm": 0.13512761890888214,
      "learning_rate": 9.466485412878425e-05,
      "loss": 0.0591,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008559520472772419,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.228095245361327,
      "state_query_norm": 1021.5940490722656,
      "state_relation_loss": 0.0008559520472772419,
      "step": 5730,
      "student_distance_mean": 1356.3052612304687,
      "teacher_distance_mean": 1.1472254037857055,
      "weighted_auxiliary_loss": 0.0008559520472772419,
      "weighted_relation_loss": 0.0008559520472772419
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.369504209541628,
      "flow_matching_loss": 0.055230018123984335,
      "grad_norm": 0.1560611128807068,
      "learning_rate": 9.464005434323154e-05,
      "loss": 0.0561,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009192054742015899,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.169540023803712,
      "state_query_norm": 1015.7660583496094,
      "state_relation_loss": 0.0009192054742015899,
      "step": 5740,
      "student_distance_mean": 1348.5036987304688,
      "teacher_distance_mean": 1.149523711204529,
      "weighted_auxiliary_loss": 0.0009192054742015899,
      "weighted_relation_loss": 0.0009192054742015899
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.378858746492049,
      "flow_matching_loss": 0.05567675158381462,
      "grad_norm": 0.14355209469795227,
      "learning_rate": 9.461520031580352e-05,
      "loss": 0.0565,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008296882850117981,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.570145988464354,
      "state_query_norm": 1027.7836669921876,
      "state_relation_loss": 0.0008296882850117981,
      "step": 5750,
      "student_distance_mean": 1371.8669189453126,
      "teacher_distance_mean": 1.1663121700286865,
      "weighted_auxiliary_loss": 0.0008296882850117981,
      "weighted_relation_loss": 0.0008296882850117981
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.38821328344247,
      "flow_matching_loss": 0.0554638959467411,
      "grad_norm": 0.180445596575737,
      "learning_rate": 9.459029207670019e-05,
      "loss": 0.0563,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008605821698438376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.086973190307617,
      "state_query_norm": 1019.3866638183594,
      "state_relation_loss": 0.0008605821698438376,
      "step": 5760,
      "student_distance_mean": 1349.6328857421875,
      "teacher_distance_mean": 1.1515079855918884,
      "weighted_auxiliary_loss": 0.0008605821698438376,
      "weighted_relation_loss": 0.0008605821698438376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.397567820392891,
      "flow_matching_loss": 0.056532207131385806,
      "grad_norm": 0.14613080024719238,
      "learning_rate": 9.456532965618737e-05,
      "loss": 0.0574,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008698184974491596,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.157265853881835,
      "state_query_norm": 1019.9687255859375,
      "state_relation_loss": 0.0008698184974491596,
      "step": 5770,
      "student_distance_mean": 1349.2511474609375,
      "teacher_distance_mean": 1.157909345626831,
      "weighted_auxiliary_loss": 0.0008698184974491596,
      "weighted_relation_loss": 0.0008698184974491596
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.406922357343311,
      "flow_matching_loss": 0.05837411768734455,
      "grad_norm": 0.17017494142055511,
      "learning_rate": 9.454031308459681e-05,
      "loss": 0.0593,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008948403643444181,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.21909980773926,
      "state_query_norm": 1014.7032531738281,
      "state_relation_loss": 0.0008948403643444181,
      "step": 5780,
      "student_distance_mean": 1350.302490234375,
      "teacher_distance_mean": 1.1482689023017882,
      "weighted_auxiliary_loss": 0.0008948403643444181,
      "weighted_relation_loss": 0.0008948403643444181
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.416276894293732,
      "flow_matching_loss": 0.05443265996873379,
      "grad_norm": 0.1680678427219391,
      "learning_rate": 9.451524239232595e-05,
      "loss": 0.0553,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008352081349585205,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.311636924743652,
      "state_query_norm": 1031.6737854003907,
      "state_relation_loss": 0.0008352081349585205,
      "step": 5790,
      "student_distance_mean": 1360.0624267578125,
      "teacher_distance_mean": 1.1499229431152345,
      "weighted_auxiliary_loss": 0.0008352081349585205,
      "weighted_relation_loss": 0.0008352081349585205
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.425631431244153,
      "flow_matching_loss": 0.05589016489684582,
      "grad_norm": 0.1505247801542282,
      "learning_rate": 9.449011760983809e-05,
      "loss": 0.0568,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008617502462584525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.423492622375488,
      "state_query_norm": 1024.4345764160157,
      "state_relation_loss": 0.0008617502462584525,
      "step": 5800,
      "student_distance_mean": 1364.832373046875,
      "teacher_distance_mean": 1.146150326728821,
      "weighted_auxiliary_loss": 0.0008617502462584525,
      "weighted_relation_loss": 0.0008617502462584525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.4349859681945745,
      "flow_matching_loss": 0.05451410934329033,
      "grad_norm": 0.19114020466804504,
      "learning_rate": 9.446493876766218e-05,
      "loss": 0.0554,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008491171291097998,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.447880363464357,
      "state_query_norm": 1031.803936767578,
      "state_relation_loss": 0.0008491171291097998,
      "step": 5810,
      "student_distance_mean": 1366.3747314453126,
      "teacher_distance_mean": 1.1495680928230285,
      "weighted_auxiliary_loss": 0.0008491171291097998,
      "weighted_relation_loss": 0.0008491171291097998
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.444340505144996,
      "flow_matching_loss": 0.056779808923602106,
      "grad_norm": 0.16084441542625427,
      "learning_rate": 9.44397058963929e-05,
      "loss": 0.0576,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000867805618327111,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.756318855285645,
      "state_query_norm": 1034.778338623047,
      "state_relation_loss": 0.000867805618327111,
      "step": 5820,
      "student_distance_mean": 1380.5687622070313,
      "teacher_distance_mean": 1.1707778692245483,
      "weighted_auxiliary_loss": 0.000867805618327111,
      "weighted_relation_loss": 0.000867805618327111
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.453695042095417,
      "flow_matching_loss": 0.05596901960670948,
      "grad_norm": 0.12485325336456299,
      "learning_rate": 9.441441902669056e-05,
      "loss": 0.0568,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008748581632971764,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.308226585388184,
      "state_query_norm": 1019.9424438476562,
      "state_relation_loss": 0.0008748581632971764,
      "step": 5830,
      "student_distance_mean": 1357.77333984375,
      "teacher_distance_mean": 1.1524789094924928,
      "weighted_auxiliary_loss": 0.0008748581632971764,
      "weighted_relation_loss": 0.0008748581632971764
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.463049579045837,
      "flow_matching_loss": 0.05606691762804985,
      "grad_norm": 0.13778355717658997,
      "learning_rate": 9.43890781892811e-05,
      "loss": 0.057,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009192822384648025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.38010025024414,
      "state_query_norm": 1020.8671264648438,
      "state_relation_loss": 0.0009192822384648025,
      "step": 5840,
      "student_distance_mean": 1361.6535522460938,
      "teacher_distance_mean": 1.1556440830230712,
      "weighted_auxiliary_loss": 0.0009192822384648025,
      "weighted_relation_loss": 0.0009192822384648025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.472404115996258,
      "flow_matching_loss": 0.055260326340794565,
      "grad_norm": 0.17786163091659546,
      "learning_rate": 9.436368341495603e-05,
      "loss": 0.0562,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008985989377833902,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.663329124450684,
      "state_query_norm": 1030.2477294921875,
      "state_relation_loss": 0.0008985989377833902,
      "step": 5850,
      "student_distance_mean": 1374.0994995117187,
      "teacher_distance_mean": 1.152322494983673,
      "weighted_auxiliary_loss": 0.0008985989377833902,
      "weighted_relation_loss": 0.0008985989377833902
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.481758652946679,
      "flow_matching_loss": 0.0563531544059515,
      "grad_norm": 0.19257405400276184,
      "learning_rate": 9.43382347345724e-05,
      "loss": 0.0572,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008579885063227267,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.366442680358887,
      "state_query_norm": 1029.0408081054688,
      "state_relation_loss": 0.0008579885063227267,
      "step": 5860,
      "student_distance_mean": 1364.0684326171875,
      "teacher_distance_mean": 1.1400808930397033,
      "weighted_auxiliary_loss": 0.0008579885063227267,
      "weighted_relation_loss": 0.0008579885063227267
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.4911131898971,
      "flow_matching_loss": 0.05582466647028923,
      "grad_norm": 0.11425263434648514,
      "learning_rate": 9.431273217905273e-05,
      "loss": 0.0567,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008809252118226141,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.657215881347657,
      "state_query_norm": 1036.9784606933595,
      "state_relation_loss": 0.0008809252118226141,
      "step": 5870,
      "student_distance_mean": 1376.071826171875,
      "teacher_distance_mean": 1.153286373615265,
      "weighted_auxiliary_loss": 0.0008809252118226141,
      "weighted_relation_loss": 0.0008809252118226141
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.500467726847521,
      "flow_matching_loss": 0.054895517975091936,
      "grad_norm": 0.14844799041748047,
      "learning_rate": 9.428717577938504e-05,
      "loss": 0.0558,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009072262502741069,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.542073249816895,
      "state_query_norm": 1031.6692504882812,
      "state_relation_loss": 0.0009072262502741069,
      "step": 5880,
      "student_distance_mean": 1368.3260131835937,
      "teacher_distance_mean": 1.149471938610077,
      "weighted_auxiliary_loss": 0.0009072262502741069,
      "weighted_relation_loss": 0.0009072262502741069
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.509822263797942,
      "flow_matching_loss": 0.057869695499539374,
      "grad_norm": 0.1392412781715393,
      "learning_rate": 9.426156556662276e-05,
      "loss": 0.0588,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009126884047873318,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.941844940185547,
      "state_query_norm": 1047.2609313964845,
      "state_relation_loss": 0.0009126884047873318,
      "step": 5890,
      "student_distance_mean": 1394.798828125,
      "teacher_distance_mean": 1.166048288345337,
      "weighted_auxiliary_loss": 0.0009126884047873318,
      "weighted_relation_loss": 0.0009126884047873318
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.519176800748363,
      "flow_matching_loss": 0.05550290755927563,
      "grad_norm": 0.21486429870128632,
      "learning_rate": 9.423590157188474e-05,
      "loss": 0.0564,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008934262907132507,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.576721572875975,
      "state_query_norm": 1038.7936584472657,
      "state_relation_loss": 0.0008934262907132507,
      "step": 5900,
      "student_distance_mean": 1374.0550170898437,
      "teacher_distance_mean": 1.139813983440399,
      "weighted_auxiliary_loss": 0.0008934262907132507,
      "weighted_relation_loss": 0.0008934262907132507
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.528531337698784,
      "flow_matching_loss": 0.05824618376791477,
      "grad_norm": 0.1959284394979477,
      "learning_rate": 9.421018382635513e-05,
      "loss": 0.0592,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009260841587092727,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.75323371887207,
      "state_query_norm": 1037.3191162109374,
      "state_relation_loss": 0.0009260841587092727,
      "step": 5910,
      "student_distance_mean": 1380.1071166992188,
      "teacher_distance_mean": 1.1480309009552,
      "weighted_auxiliary_loss": 0.0009260841587092727,
      "weighted_relation_loss": 0.0009260841587092727
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.537885874649205,
      "flow_matching_loss": 0.056514698266983035,
      "grad_norm": 0.1339508295059204,
      "learning_rate": 9.418441236128343e-05,
      "loss": 0.0574,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009011664136778563,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.129327392578126,
      "state_query_norm": 1052.31005859375,
      "state_relation_loss": 0.0009011664136778563,
      "step": 5920,
      "student_distance_mean": 1402.6604858398437,
      "teacher_distance_mean": 1.165479338169098,
      "weighted_auxiliary_loss": 0.0009011664136778563,
      "weighted_relation_loss": 0.0009011664136778563
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.547240411599626,
      "flow_matching_loss": 0.056500920280814174,
      "grad_norm": 0.134428009390831,
      "learning_rate": 9.41585872079844e-05,
      "loss": 0.0574,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008959305472671985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.863261795043947,
      "state_query_norm": 1035.4991333007813,
      "state_relation_loss": 0.0008959305472671985,
      "step": 5930,
      "student_distance_mean": 1389.0773071289063,
      "teacher_distance_mean": 1.1544844150543212,
      "weighted_auxiliary_loss": 0.0008959305472671985,
      "weighted_relation_loss": 0.0008959305472671985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.556594948550047,
      "flow_matching_loss": 0.05571128278970718,
      "grad_norm": 0.14275436103343964,
      "learning_rate": 9.413270839783802e-05,
      "loss": 0.0566,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009051762172020972,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.70593547821045,
      "state_query_norm": 1031.7835876464844,
      "state_relation_loss": 0.0009051762172020972,
      "step": 5940,
      "student_distance_mean": 1374.9020263671875,
      "teacher_distance_mean": 1.1463708519935607,
      "weighted_auxiliary_loss": 0.0009051762172020972,
      "weighted_relation_loss": 0.0009051762172020972
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.565949485500468,
      "flow_matching_loss": 0.056573083996772765,
      "grad_norm": 0.14146015048027039,
      "learning_rate": 9.41067759622895e-05,
      "loss": 0.0575,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000885425164597109,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.954277801513673,
      "state_query_norm": 1042.6751586914063,
      "state_relation_loss": 0.000885425164597109,
      "step": 5950,
      "student_distance_mean": 1388.123876953125,
      "teacher_distance_mean": 1.15753333568573,
      "weighted_auxiliary_loss": 0.000885425164597109,
      "weighted_relation_loss": 0.000885425164597109
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.575304022450888,
      "flow_matching_loss": 0.056259974092245105,
      "grad_norm": 0.17515027523040771,
      "learning_rate": 9.408078993284917e-05,
      "loss": 0.0572,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009018414421007037,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.983591270446777,
      "state_query_norm": 1051.433856201172,
      "state_relation_loss": 0.0009018414421007037,
      "step": 5960,
      "student_distance_mean": 1395.33017578125,
      "teacher_distance_mean": 1.147224235534668,
      "weighted_auxiliary_loss": 0.0009018414421007037,
      "weighted_relation_loss": 0.0009018414421007037
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.584658559401309,
      "flow_matching_loss": 0.056915925815701485,
      "grad_norm": 0.16032619774341583,
      "learning_rate": 9.405475034109254e-05,
      "loss": 0.0578,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008942277228925378,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.390536308288574,
      "state_query_norm": 1057.9114013671874,
      "state_relation_loss": 0.0008942277228925378,
      "step": 5970,
      "student_distance_mean": 1412.1525390625,
      "teacher_distance_mean": 1.1679882168769837,
      "weighted_auxiliary_loss": 0.0008942277228925378,
      "weighted_relation_loss": 0.0008942277228925378
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.5940130963517305,
      "flow_matching_loss": 0.054587554559111594,
      "grad_norm": 0.19517390429973602,
      "learning_rate": 9.402865721866015e-05,
      "loss": 0.0555,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008959495928138495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.627683258056642,
      "state_query_norm": 1034.1107360839844,
      "state_relation_loss": 0.0008959495928138495,
      "step": 5980,
      "student_distance_mean": 1376.4171997070312,
      "teacher_distance_mean": 1.1474486708641052,
      "weighted_auxiliary_loss": 0.0008959495928138495,
      "weighted_relation_loss": 0.0008959495928138495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.603367633302152,
      "flow_matching_loss": 0.05499638020992279,
      "grad_norm": 0.14761076867580414,
      "learning_rate": 9.400251059725762e-05,
      "loss": 0.0559,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008839828835334629,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.041243934631346,
      "state_query_norm": 1043.5840759277344,
      "state_relation_loss": 0.0008839828835334629,
      "step": 5990,
      "student_distance_mean": 1397.3862915039062,
      "teacher_distance_mean": 1.1531166315078736,
      "weighted_auxiliary_loss": 0.0008839828835334629,
      "weighted_relation_loss": 0.0008839828835334629
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.612722170252573,
      "flow_matching_loss": 0.055307618156075475,
      "grad_norm": 0.13809484243392944,
      "learning_rate": 9.397631050865554e-05,
      "loss": 0.0562,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009043097787071019,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.182266235351562,
      "state_query_norm": 1057.8501708984375,
      "state_relation_loss": 0.0009043097787071019,
      "step": 6000,
      "student_distance_mean": 1404.4429077148438,
      "teacher_distance_mean": 1.1648571968078614,
      "weighted_auxiliary_loss": 0.0009043097787071019,
      "weighted_relation_loss": 0.0009043097787071019
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.622076707202994,
      "flow_matching_loss": 0.053784435242414476,
      "grad_norm": 0.1449497640132904,
      "learning_rate": 9.395005698468949e-05,
      "loss": 0.0546,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000863071100320667,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.66465358734131,
      "state_query_norm": 1039.4723327636718,
      "state_relation_loss": 0.000863071100320667,
      "step": 6010,
      "student_distance_mean": 1376.315966796875,
      "teacher_distance_mean": 1.136637771129608,
      "weighted_auxiliary_loss": 0.000863071100320667,
      "weighted_relation_loss": 0.000863071100320667
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.631431244153415,
      "flow_matching_loss": 0.05482274852693081,
      "grad_norm": 0.16903908550739288,
      "learning_rate": 9.392375005726e-05,
      "loss": 0.0557,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008318132488057018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.263718605041504,
      "state_query_norm": 1065.9537231445313,
      "state_relation_loss": 0.0008318132488057018,
      "step": 6020,
      "student_distance_mean": 1413.8500366210938,
      "teacher_distance_mean": 1.1686526894569398,
      "weighted_auxiliary_loss": 0.0008318132488057018,
      "weighted_relation_loss": 0.0008318132488057018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.640785781103835,
      "flow_matching_loss": 0.05364150330424309,
      "grad_norm": 0.13051171600818634,
      "learning_rate": 9.389738975833242e-05,
      "loss": 0.0545,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008656551479361951,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.028732299804688,
      "state_query_norm": 1050.5167602539063,
      "state_relation_loss": 0.0008656551479361951,
      "step": 6030,
      "student_distance_mean": 1396.9521606445312,
      "teacher_distance_mean": 1.1578692674636841,
      "weighted_auxiliary_loss": 0.0008656551479361951,
      "weighted_relation_loss": 0.0008656551479361951
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.650140318054256,
      "flow_matching_loss": 0.05334464535117149,
      "grad_norm": 0.1320236772298813,
      "learning_rate": 9.387097611993707e-05,
      "loss": 0.0542,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008782927470747381,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.299194526672363,
      "state_query_norm": 1061.33798828125,
      "state_relation_loss": 0.0008782927470747381,
      "step": 6040,
      "student_distance_mean": 1409.521337890625,
      "teacher_distance_mean": 1.1551748752593993,
      "weighted_auxiliary_loss": 0.0008782927470747381,
      "weighted_relation_loss": 0.0008782927470747381
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.659494855004677,
      "flow_matching_loss": 0.05568895824253559,
      "grad_norm": 0.11614929884672165,
      "learning_rate": 9.384450917416894e-05,
      "loss": 0.0565,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008300393412355334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.483870697021484,
      "state_query_norm": 1070.6051330566406,
      "state_relation_loss": 0.0008300393412355334,
      "step": 6050,
      "student_distance_mean": 1424.8217041015625,
      "teacher_distance_mean": 1.1666643857955932,
      "weighted_auxiliary_loss": 0.0008300393412355334,
      "weighted_relation_loss": 0.0008300393412355334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.668849391955098,
      "flow_matching_loss": 0.05530536733567715,
      "grad_norm": 0.2033710479736328,
      "learning_rate": 9.381798895318792e-05,
      "loss": 0.0561,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008370100753381848,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.974700736999512,
      "state_query_norm": 1053.8403442382812,
      "state_relation_loss": 0.0008370100753381848,
      "step": 6060,
      "student_distance_mean": 1394.9871704101563,
      "teacher_distance_mean": 1.1515660643577577,
      "weighted_auxiliary_loss": 0.0008370100753381848,
      "weighted_relation_loss": 0.0008370100753381848
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.6782039289055195,
      "flow_matching_loss": 0.05366047732532024,
      "grad_norm": 0.20752818882465363,
      "learning_rate": 9.379141548921853e-05,
      "loss": 0.0545,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000865703314775601,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.01685733795166,
      "state_query_norm": 1040.6366333007813,
      "state_relation_loss": 0.000865703314775601,
      "step": 6070,
      "student_distance_mean": 1393.392724609375,
      "teacher_distance_mean": 1.1490006566047668,
      "weighted_auxiliary_loss": 0.000865703314775601,
      "weighted_relation_loss": 0.000865703314775601
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.68755846585594,
      "flow_matching_loss": 0.05737585909664631,
      "grad_norm": 0.12423579394817352,
      "learning_rate": 9.376478881455009e-05,
      "loss": 0.0583,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008867157797794789,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.990546798706056,
      "state_query_norm": 1045.7280151367188,
      "state_relation_loss": 0.0008867157797794789,
      "step": 6080,
      "student_distance_mean": 1393.8709594726563,
      "teacher_distance_mean": 1.1446243405342102,
      "weighted_auxiliary_loss": 0.0008867157797794789,
      "weighted_relation_loss": 0.0008867157797794789
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.696913002806361,
      "flow_matching_loss": 0.055649083107709885,
      "grad_norm": 0.17617563903331757,
      "learning_rate": 9.373810896153647e-05,
      "loss": 0.0565,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008574992069043219,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.31802577972412,
      "state_query_norm": 1052.4367614746093,
      "state_relation_loss": 0.0008574992069043219,
      "step": 6090,
      "student_distance_mean": 1412.461865234375,
      "teacher_distance_mean": 1.1555991530418397,
      "weighted_auxiliary_loss": 0.0008574992069043219,
      "weighted_relation_loss": 0.0008574992069043219
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.706267539756782,
      "flow_matching_loss": 0.05588410943746567,
      "grad_norm": 0.15895065665245056,
      "learning_rate": 9.371137596259623e-05,
      "loss": 0.0568,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008690449059940875,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.151115798950194,
      "state_query_norm": 1055.9965698242188,
      "state_relation_loss": 0.0008690449059940875,
      "step": 6100,
      "student_distance_mean": 1403.7101318359375,
      "teacher_distance_mean": 1.1511848211288451,
      "weighted_auxiliary_loss": 0.0008690449059940875,
      "weighted_relation_loss": 0.0008690449059940875
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.715622076707203,
      "flow_matching_loss": 0.057149849459528924,
      "grad_norm": 0.15084300935268402,
      "learning_rate": 9.368458985021248e-05,
      "loss": 0.058,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008198432740755379,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.14193801879883,
      "state_query_norm": 1064.00947265625,
      "state_relation_loss": 0.0008198432740755379,
      "step": 6110,
      "student_distance_mean": 1408.2316040039063,
      "teacher_distance_mean": 1.1552374243736268,
      "weighted_auxiliary_loss": 0.0008198432740755379,
      "weighted_relation_loss": 0.0008198432740755379
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.724976613657624,
      "flow_matching_loss": 0.057809867337346076,
      "grad_norm": 0.10081761330366135,
      "learning_rate": 9.365775065693287e-05,
      "loss": 0.0587,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008787830593064427,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.112697029113768,
      "state_query_norm": 1053.6849487304687,
      "state_relation_loss": 0.0008787830593064427,
      "step": 6120,
      "student_distance_mean": 1402.1968505859375,
      "teacher_distance_mean": 1.1494207620620727,
      "weighted_auxiliary_loss": 0.0008787830593064427,
      "weighted_relation_loss": 0.0008787830593064427
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.734331150608045,
      "flow_matching_loss": 0.053877703845500946,
      "grad_norm": 0.16179358959197998,
      "learning_rate": 9.363085841536957e-05,
      "loss": 0.0547,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008657482743728906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.170842552185057,
      "state_query_norm": 1054.991436767578,
      "state_relation_loss": 0.0008657482743728906,
      "step": 6130,
      "student_distance_mean": 1404.1560668945312,
      "teacher_distance_mean": 1.1469181895256042,
      "weighted_auxiliary_loss": 0.0008657482743728906,
      "weighted_relation_loss": 0.0008657482743728906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.743685687558466,
      "flow_matching_loss": 0.05410079620778561,
      "grad_norm": 0.13808362185955048,
      "learning_rate": 9.360391315819917e-05,
      "loss": 0.055,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008512799511663615,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.994293022155762,
      "state_query_norm": 1052.1306396484374,
      "state_relation_loss": 0.0008512799511663615,
      "step": 6140,
      "student_distance_mean": 1396.8599243164062,
      "teacher_distance_mean": 1.1423433542251586,
      "weighted_auxiliary_loss": 0.0008512799511663615,
      "weighted_relation_loss": 0.0008512799511663615
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.7530402245088865,
      "flow_matching_loss": 0.05361007452011109,
      "grad_norm": 0.12967047095298767,
      "learning_rate": 9.357691491816269e-05,
      "loss": 0.0545,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008925458183512092,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.50744514465332,
      "state_query_norm": 1065.313916015625,
      "state_relation_loss": 0.0008925458183512092,
      "step": 6150,
      "student_distance_mean": 1421.67607421875,
      "teacher_distance_mean": 1.1664572358131409,
      "weighted_auxiliary_loss": 0.0008925458183512092,
      "weighted_relation_loss": 0.0008925458183512092
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.762394761459308,
      "flow_matching_loss": 0.055154294893145564,
      "grad_norm": 0.16237570345401764,
      "learning_rate": 9.354986372806557e-05,
      "loss": 0.056,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008649255498312414,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.355334854125978,
      "state_query_norm": 1069.7149841308594,
      "state_relation_loss": 0.0008649255498312414,
      "step": 6160,
      "student_distance_mean": 1415.45419921875,
      "teacher_distance_mean": 1.157591128349304,
      "weighted_auxiliary_loss": 0.0008649255498312414,
      "weighted_relation_loss": 0.0008649255498312414
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.771749298409729,
      "flow_matching_loss": 0.05285686701536178,
      "grad_norm": 0.15665115416049957,
      "learning_rate": 9.352275962077752e-05,
      "loss": 0.0537,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008730229397770018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.885629653930664,
      "state_query_norm": 1050.0872863769532,
      "state_relation_loss": 0.0008730229397770018,
      "step": 6170,
      "student_distance_mean": 1388.6062377929688,
      "teacher_distance_mean": 1.140917956829071,
      "weighted_auxiliary_loss": 0.0008730229397770018,
      "weighted_relation_loss": 0.0008730229397770018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.78110383536015,
      "flow_matching_loss": 0.05458650141954422,
      "grad_norm": 0.14972060918807983,
      "learning_rate": 9.349560262923262e-05,
      "loss": 0.0554,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008598885498940945,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.948772621154784,
      "state_query_norm": 1046.3471374511719,
      "state_relation_loss": 0.0008598885498940945,
      "step": 6180,
      "student_distance_mean": 1390.2234008789062,
      "teacher_distance_mean": 1.148751938343048,
      "weighted_auxiliary_loss": 0.0008598885498940945,
      "weighted_relation_loss": 0.0008598885498940945
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.790458372310571,
      "flow_matching_loss": 0.05315508991479874,
      "grad_norm": 0.1158437579870224,
      "learning_rate": 9.346839278642913e-05,
      "loss": 0.054,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008531579340342432,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.07814998626709,
      "state_query_norm": 1048.1191955566405,
      "state_relation_loss": 0.0008531579340342432,
      "step": 6190,
      "student_distance_mean": 1397.9298583984375,
      "teacher_distance_mean": 1.1507298469543457,
      "weighted_auxiliary_loss": 0.0008531579340342432,
      "weighted_relation_loss": 0.0008531579340342432
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.799812909260991,
      "flow_matching_loss": 0.054360829293727875,
      "grad_norm": 0.11720101535320282,
      "learning_rate": 9.344113012542963e-05,
      "loss": 0.0552,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008355589816346765,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.689587020874022,
      "state_query_norm": 1077.64833984375,
      "state_relation_loss": 0.0008355589816346765,
      "step": 6200,
      "student_distance_mean": 1430.9681762695313,
      "teacher_distance_mean": 1.1594958662986756,
      "weighted_auxiliary_loss": 0.0008355589816346765,
      "weighted_relation_loss": 0.0008355589816346765
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.809167446211412,
      "flow_matching_loss": 0.055102888494729996,
      "grad_norm": 0.13708163797855377,
      "learning_rate": 9.341381467936079e-05,
      "loss": 0.056,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008953167009167374,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.19840564727783,
      "state_query_norm": 1054.9315246582032,
      "state_relation_loss": 0.0008953167009167374,
      "step": 6210,
      "student_distance_mean": 1404.1579345703126,
      "teacher_distance_mean": 1.1440280079841614,
      "weighted_auxiliary_loss": 0.0008953167009167374,
      "weighted_relation_loss": 0.0008953167009167374
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.818521983161833,
      "flow_matching_loss": 0.05421672835946083,
      "grad_norm": 0.12172046303749084,
      "learning_rate": 9.338644648141346e-05,
      "loss": 0.0551,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008581461850553751,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.556084442138673,
      "state_query_norm": 1076.9492065429688,
      "state_relation_loss": 0.0008581461850553751,
      "step": 6220,
      "student_distance_mean": 1427.437060546875,
      "teacher_distance_mean": 1.1670879244804382,
      "weighted_auxiliary_loss": 0.0008581461850553751,
      "weighted_relation_loss": 0.0008581461850553751
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.827876520112254,
      "flow_matching_loss": 0.05290352925658226,
      "grad_norm": 0.10524383932352066,
      "learning_rate": 9.335902556484257e-05,
      "loss": 0.0537,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008460600976832211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.97976303100586,
      "state_query_norm": 1050.9630432128906,
      "state_relation_loss": 0.0008460600976832211,
      "step": 6230,
      "student_distance_mean": 1394.0353393554688,
      "teacher_distance_mean": 1.1418160200119019,
      "weighted_auxiliary_loss": 0.0008460600976832211,
      "weighted_relation_loss": 0.0008460600976832211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.8372310570626755,
      "flow_matching_loss": 0.05698956698179245,
      "grad_norm": 0.12857021391391754,
      "learning_rate": 9.333155196296712e-05,
      "loss": 0.0578,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000849236745852977,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.1747688293457,
      "state_query_norm": 1058.5179931640625,
      "state_relation_loss": 0.000849236745852977,
      "step": 6240,
      "student_distance_mean": 1405.174853515625,
      "teacher_distance_mean": 1.1567137479782104,
      "weighted_auxiliary_loss": 0.000849236745852977,
      "weighted_relation_loss": 0.000849236745852977
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.846585594013097,
      "flow_matching_loss": 0.05460745356976986,
      "grad_norm": 0.16889828443527222,
      "learning_rate": 9.330402570917016e-05,
      "loss": 0.0554,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000819819892058149,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 24.958666038513183,
      "state_query_norm": 1050.4134155273437,
      "state_relation_loss": 0.000819819892058149,
      "step": 6250,
      "student_distance_mean": 1395.198046875,
      "teacher_distance_mean": 1.1441627502441407,
      "weighted_auxiliary_loss": 0.000819819892058149,
      "weighted_relation_loss": 0.000819819892058149
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.855940130963518,
      "flow_matching_loss": 0.05496603846549988,
      "grad_norm": 0.16079023480415344,
      "learning_rate": 9.327644683689865e-05,
      "loss": 0.0558,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008502177894115448,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.204310417175293,
      "state_query_norm": 1053.1987670898438,
      "state_relation_loss": 0.0008502177894115448,
      "step": 6260,
      "student_distance_mean": 1406.7592407226562,
      "teacher_distance_mean": 1.1524961948394776,
      "weighted_auxiliary_loss": 0.0008502177894115448,
      "weighted_relation_loss": 0.0008502177894115448
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.865294667913938,
      "flow_matching_loss": 0.05338669642806053,
      "grad_norm": 0.14444337785243988,
      "learning_rate": 9.324881537966354e-05,
      "loss": 0.0542,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008338697429280728,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.031983375549316,
      "state_query_norm": 1047.0078918457032,
      "state_relation_loss": 0.0008338697429280728,
      "step": 6270,
      "student_distance_mean": 1399.1063232421875,
      "teacher_distance_mean": 1.1465437769889832,
      "weighted_auxiliary_loss": 0.0008338697429280728,
      "weighted_relation_loss": 0.0008338697429280728
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.874649204864359,
      "flow_matching_loss": 0.054916157945990564,
      "grad_norm": 0.1149958074092865,
      "learning_rate": 9.322113137103964e-05,
      "loss": 0.0558,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008628584386315197,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.900370788574218,
      "state_query_norm": 1073.8496215820312,
      "state_relation_loss": 0.0008628584386315197,
      "step": 6280,
      "student_distance_mean": 1439.0774536132812,
      "teacher_distance_mean": 1.1695262312889099,
      "weighted_auxiliary_loss": 0.0008628584386315197,
      "weighted_relation_loss": 0.0008628584386315197
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.88400374181478,
      "flow_matching_loss": 0.0563779316842556,
      "grad_norm": 0.1483526974916458,
      "learning_rate": 9.319339484466564e-05,
      "loss": 0.0573,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008752369438298046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.008460426330565,
      "state_query_norm": 1042.1031188964844,
      "state_relation_loss": 0.0008752369438298046,
      "step": 6290,
      "student_distance_mean": 1396.1904052734376,
      "teacher_distance_mean": 1.1336139678955077,
      "weighted_auxiliary_loss": 0.0008752369438298046,
      "weighted_relation_loss": 0.0008752369438298046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.893358278765201,
      "flow_matching_loss": 0.05390057228505611,
      "grad_norm": 0.14831146597862244,
      "learning_rate": 9.316560583424403e-05,
      "loss": 0.0548,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008587040880229324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.111012077331544,
      "state_query_norm": 1045.59013671875,
      "state_relation_loss": 0.0008587040880229324,
      "step": 6300,
      "student_distance_mean": 1397.8421752929687,
      "teacher_distance_mean": 1.139680600166321,
      "weighted_auxiliary_loss": 0.0008587040880229324,
      "weighted_relation_loss": 0.0008587040880229324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.902712815715622,
      "flow_matching_loss": 0.05645358636975288,
      "grad_norm": 0.18004436790943146,
      "learning_rate": 9.313776437354108e-05,
      "loss": 0.0574,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009269262955058366,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.548662185668945,
      "state_query_norm": 1064.4880310058593,
      "state_relation_loss": 0.0009269262955058366,
      "step": 6310,
      "student_distance_mean": 1422.6177978515625,
      "teacher_distance_mean": 1.1550017952919007,
      "weighted_auxiliary_loss": 0.0009269262955058366,
      "weighted_relation_loss": 0.0009269262955058366
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.9120673526660426,
      "flow_matching_loss": 0.055098751559853554,
      "grad_norm": 0.15565867722034454,
      "learning_rate": 9.31098704963868e-05,
      "loss": 0.056,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009031480934936553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.593285369873048,
      "state_query_norm": 1065.6549194335937,
      "state_relation_loss": 0.0009031480934936553,
      "step": 6320,
      "student_distance_mean": 1426.9117431640625,
      "teacher_distance_mean": 1.1556081891059875,
      "weighted_auxiliary_loss": 0.0009031480934936553,
      "weighted_relation_loss": 0.0009031480934936553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.921421889616464,
      "flow_matching_loss": 0.053823181241750714,
      "grad_norm": 0.11705271154642105,
      "learning_rate": 9.308192423667485e-05,
      "loss": 0.0547,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008447455707937479,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.228721618652344,
      "state_query_norm": 1047.2173095703124,
      "state_relation_loss": 0.0008447455707937479,
      "step": 6330,
      "student_distance_mean": 1405.9308837890626,
      "teacher_distance_mean": 1.147297751903534,
      "weighted_auxiliary_loss": 0.0008447455707937479,
      "weighted_relation_loss": 0.0008447455707937479
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.930776426566885,
      "flow_matching_loss": 0.05588550642132759,
      "grad_norm": 0.12452146410942078,
      "learning_rate": 9.305392562836262e-05,
      "loss": 0.0567,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008617395593319088,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.424684143066408,
      "state_query_norm": 1064.757012939453,
      "state_relation_loss": 0.0008617395593319088,
      "step": 6340,
      "student_distance_mean": 1417.1156616210938,
      "teacher_distance_mean": 1.1547052979469299,
      "weighted_auxiliary_loss": 0.0008617395593319088,
      "weighted_relation_loss": 0.0008617395593319088
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.940130963517306,
      "flow_matching_loss": 0.05388821586966515,
      "grad_norm": 0.14747892320156097,
      "learning_rate": 9.3025874705471e-05,
      "loss": 0.0548,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008619753178209066,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.330206680297852,
      "state_query_norm": 1053.9664245605468,
      "state_relation_loss": 0.0008619753178209066,
      "step": 6350,
      "student_distance_mean": 1415.2732299804688,
      "teacher_distance_mean": 1.1521711707115174,
      "weighted_auxiliary_loss": 0.0008619753178209066,
      "weighted_relation_loss": 0.0008619753178209066
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.949485500467727,
      "flow_matching_loss": 0.0531207587569952,
      "grad_norm": 0.16607074439525604,
      "learning_rate": 9.299777150208457e-05,
      "loss": 0.054,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008557953988201916,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.445859527587892,
      "state_query_norm": 1056.9733642578126,
      "state_relation_loss": 0.0008557953988201916,
      "step": 6360,
      "student_distance_mean": 1418.4678833007813,
      "teacher_distance_mean": 1.15538409948349,
      "weighted_auxiliary_loss": 0.0008557953988201916,
      "weighted_relation_loss": 0.0008557953988201916
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.958840037418148,
      "flow_matching_loss": 0.05449603982269764,
      "grad_norm": 0.1404436081647873,
      "learning_rate": 9.296961605235133e-05,
      "loss": 0.0554,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008685312641318887,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.694962310791016,
      "state_query_norm": 1066.6299560546875,
      "state_relation_loss": 0.0008685312641318887,
      "step": 6370,
      "student_distance_mean": 1434.3348266601563,
      "teacher_distance_mean": 1.1607060194015504,
      "weighted_auxiliary_loss": 0.0008685312641318887,
      "weighted_relation_loss": 0.0008685312641318887
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.968194574368569,
      "flow_matching_loss": 0.055932608991861345,
      "grad_norm": 0.15826669335365295,
      "learning_rate": 9.29414083904828e-05,
      "loss": 0.0568,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008815527311526239,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.819362258911134,
      "state_query_norm": 1078.421826171875,
      "state_relation_loss": 0.0008815527311526239,
      "step": 6380,
      "student_distance_mean": 1440.4668334960938,
      "teacher_distance_mean": 1.1620265126228333,
      "weighted_auxiliary_loss": 0.0008815527311526239,
      "weighted_relation_loss": 0.0008815527311526239
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.977549111318989,
      "flow_matching_loss": 0.05712302215397358,
      "grad_norm": 0.18484124541282654,
      "learning_rate": 9.291314855075397e-05,
      "loss": 0.058,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008867624157574028,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.591328048706053,
      "state_query_norm": 1064.34228515625,
      "state_relation_loss": 0.0008867624157574028,
      "step": 6390,
      "student_distance_mean": 1424.674609375,
      "teacher_distance_mean": 1.1449324131011962,
      "weighted_auxiliary_loss": 0.0008867624157574028,
      "weighted_relation_loss": 0.0008867624157574028
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.9869036482694105,
      "flow_matching_loss": 0.05626343749463558,
      "grad_norm": 0.11492419987916946,
      "learning_rate": 9.288483656750322e-05,
      "loss": 0.0571,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008854017592966557,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.848783683776855,
      "state_query_norm": 1078.4949951171875,
      "state_relation_loss": 0.0008854017592966557,
      "step": 6400,
      "student_distance_mean": 1440.0058837890624,
      "teacher_distance_mean": 1.1552909016609192,
      "weighted_auxiliary_loss": 0.0008854017592966557,
      "weighted_relation_loss": 0.0008854017592966557
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 5.996258185219832,
      "flow_matching_loss": 0.05481532327830792,
      "grad_norm": 0.11550776660442352,
      "learning_rate": 9.285647247513225e-05,
      "loss": 0.0557,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009279645280912518,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.65220947265625,
      "state_query_norm": 1065.2459594726563,
      "state_relation_loss": 0.0009279645280912518,
      "step": 6410,
      "student_distance_mean": 1429.134033203125,
      "teacher_distance_mean": 1.1517932057380675,
      "weighted_auxiliary_loss": 0.0009279645280912518,
      "weighted_relation_loss": 0.0009279645280912518
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.005612722170253,
      "flow_matching_loss": 0.05307253450155258,
      "grad_norm": 0.11455747485160828,
      "learning_rate": 9.282805630810614e-05,
      "loss": 0.054,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0009462649410124868,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 26.15867748260498,
      "state_query_norm": 1086.1181274414062,
      "state_relation_loss": 0.0009462649410124868,
      "step": 6420,
      "student_distance_mean": 1455.4979125976563,
      "teacher_distance_mean": 1.1610220551490784,
      "weighted_auxiliary_loss": 0.0009462649410124868,
      "weighted_relation_loss": 0.0009462649410124868
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.014967259120674,
      "flow_matching_loss": 0.0540423858910799,
      "grad_norm": 0.14807593822479248,
      "learning_rate": 9.279958810095317e-05,
      "loss": 0.0549,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008998233010061085,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.12746524810791,
      "state_query_norm": 1088.7029541015625,
      "state_relation_loss": 0.0008998233010061085,
      "step": 6430,
      "student_distance_mean": 1454.3530517578124,
      "teacher_distance_mean": 1.1558572053909302,
      "weighted_auxiliary_loss": 0.0008998233010061085,
      "weighted_relation_loss": 0.0008998233010061085
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.024321796071095,
      "flow_matching_loss": 0.054117562249302864,
      "grad_norm": 0.14951546490192413,
      "learning_rate": 9.277106788826494e-05,
      "loss": 0.055,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0009073495748452842,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.372257804870607,
      "state_query_norm": 1058.5934692382812,
      "state_relation_loss": 0.0009073495748452842,
      "step": 6440,
      "student_distance_mean": 1414.2987670898438,
      "teacher_distance_mean": 1.1330116748809815,
      "weighted_auxiliary_loss": 0.0009073495748452842,
      "weighted_relation_loss": 0.0009073495748452842
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.033676333021515,
      "flow_matching_loss": 0.05291886776685715,
      "grad_norm": 0.1605883091688156,
      "learning_rate": 9.274249570469618e-05,
      "loss": 0.0538,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008373852702789009,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.86741352081299,
      "state_query_norm": 1088.9481018066406,
      "state_relation_loss": 0.0008373852702789009,
      "step": 6450,
      "student_distance_mean": 1443.1951171875,
      "teacher_distance_mean": 1.1616435289382934,
      "weighted_auxiliary_loss": 0.0008373852702789009,
      "weighted_relation_loss": 0.0008373852702789009
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.043030869971936,
      "flow_matching_loss": 0.05535526983439922,
      "grad_norm": 0.14414529502391815,
      "learning_rate": 9.271387158496476e-05,
      "loss": 0.0563,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008966762688942254,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.23195285797119,
      "state_query_norm": 1059.7404174804688,
      "state_relation_loss": 0.0008966762688942254,
      "step": 6460,
      "student_distance_mean": 1406.1820068359375,
      "teacher_distance_mean": 1.1408214330673219,
      "weighted_auxiliary_loss": 0.0008966762688942254,
      "weighted_relation_loss": 0.0008966762688942254
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.052385406922357,
      "flow_matching_loss": 0.0551991418004036,
      "grad_norm": 0.18340209126472473,
      "learning_rate": 9.268519556385173e-05,
      "loss": 0.0561,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000882442865986377,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.72846965789795,
      "state_query_norm": 1073.5345092773437,
      "state_relation_loss": 0.000882442865986377,
      "step": 6470,
      "student_distance_mean": 1432.1575561523437,
      "teacher_distance_mean": 1.157139241695404,
      "weighted_auxiliary_loss": 0.000882442865986377,
      "weighted_relation_loss": 0.000882442865986377
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.061739943872778,
      "flow_matching_loss": 0.05558663122355938,
      "grad_norm": 0.15601742267608643,
      "learning_rate": 9.265646767620113e-05,
      "loss": 0.0565,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000873440969735384,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.35418701171875,
      "state_query_norm": 1058.84619140625,
      "state_relation_loss": 0.000873440969735384,
      "step": 6480,
      "student_distance_mean": 1413.1673217773437,
      "teacher_distance_mean": 1.1428634643554687,
      "weighted_auxiliary_loss": 0.000873440969735384,
      "weighted_relation_loss": 0.000873440969735384
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.0710944808231995,
      "flow_matching_loss": 0.05328549407422543,
      "grad_norm": 0.1848648637533188,
      "learning_rate": 9.262768795692006e-05,
      "loss": 0.0541,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008475486014503985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.968098640441895,
      "state_query_norm": 1082.0667114257812,
      "state_relation_loss": 0.0008475486014503985,
      "step": 6490,
      "student_distance_mean": 1446.848583984375,
      "teacher_distance_mean": 1.162591326236725,
      "weighted_auxiliary_loss": 0.0008475486014503985,
      "weighted_relation_loss": 0.0008475486014503985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.080449017773621,
      "flow_matching_loss": 0.05492933690547943,
      "grad_norm": 0.2280506044626236,
      "learning_rate": 9.259885644097861e-05,
      "loss": 0.0558,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008789031126070768,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.731753158569337,
      "state_query_norm": 1076.7087829589843,
      "state_relation_loss": 0.0008789031126070768,
      "step": 6500,
      "student_distance_mean": 1435.5853637695313,
      "teacher_distance_mean": 1.1428379774093629,
      "weighted_auxiliary_loss": 0.0008789031126070768,
      "weighted_relation_loss": 0.0008789031126070768
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.089803554724041,
      "flow_matching_loss": 0.05397066511213779,
      "grad_norm": 0.2684487998485565,
      "learning_rate": 9.256997316340976e-05,
      "loss": 0.0548,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008619795553386211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.16898365020752,
      "state_query_norm": 1095.6921997070312,
      "state_relation_loss": 0.0008619795553386211,
      "step": 6510,
      "student_distance_mean": 1460.7631469726562,
      "teacher_distance_mean": 1.1540901899337768,
      "weighted_auxiliary_loss": 0.0008619795553386211,
      "weighted_relation_loss": 0.0008619795553386211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.099158091674462,
      "flow_matching_loss": 0.051345253363251686,
      "grad_norm": 0.18078909814357758,
      "learning_rate": 9.254103815930943e-05,
      "loss": 0.0522,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008605172391980886,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.883103561401366,
      "state_query_norm": 1069.2557373046875,
      "state_relation_loss": 0.0008605172391980886,
      "step": 6520,
      "student_distance_mean": 1441.8007690429688,
      "teacher_distance_mean": 1.150248658657074,
      "weighted_auxiliary_loss": 0.0008605172391980886,
      "weighted_relation_loss": 0.0008605172391980886
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.108512628624883,
      "flow_matching_loss": 0.053357267007231715,
      "grad_norm": 0.16708189249038696,
      "learning_rate": 9.251205146383636e-05,
      "loss": 0.0542,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008642373082693666,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.24172706604004,
      "state_query_norm": 1103.5281005859374,
      "state_relation_loss": 0.0008642373082693666,
      "step": 6530,
      "student_distance_mean": 1464.6190795898438,
      "teacher_distance_mean": 1.1627000570297241,
      "weighted_auxiliary_loss": 0.0008642373082693666,
      "weighted_relation_loss": 0.0008642373082693666
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.117867165575304,
      "flow_matching_loss": 0.055328698828816414,
      "grad_norm": 0.11891195923089981,
      "learning_rate": 9.248301311221216e-05,
      "loss": 0.0562,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008543218427803367,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.98254528045654,
      "state_query_norm": 1080.1243896484375,
      "state_relation_loss": 0.0008543218427803367,
      "step": 6540,
      "student_distance_mean": 1444.4815795898437,
      "teacher_distance_mean": 1.1476960897445678,
      "weighted_auxiliary_loss": 0.0008543218427803367,
      "weighted_relation_loss": 0.0008543218427803367
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.127221702525725,
      "flow_matching_loss": 0.053312602639198306,
      "grad_norm": 0.1402360200881958,
      "learning_rate": 9.245392313972115e-05,
      "loss": 0.0541,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008311531564686447,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.672488975524903,
      "state_query_norm": 1074.0455322265625,
      "state_relation_loss": 0.0008311531564686447,
      "step": 6550,
      "student_distance_mean": 1426.7578979492187,
      "teacher_distance_mean": 1.1429704666137694,
      "weighted_auxiliary_loss": 0.0008311531564686447,
      "weighted_relation_loss": 0.0008311531564686447
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.136576239476146,
      "flow_matching_loss": 0.05381741523742676,
      "grad_norm": 0.1106438860297203,
      "learning_rate": 9.242478158171037e-05,
      "loss": 0.0547,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008608836738858372,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.675711822509765,
      "state_query_norm": 1066.4909912109374,
      "state_relation_loss": 0.0008608836738858372,
      "step": 6560,
      "student_distance_mean": 1427.6592529296875,
      "teacher_distance_mean": 1.1484552025794983,
      "weighted_auxiliary_loss": 0.0008608836738858372,
      "weighted_relation_loss": 0.0008608836738858372
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.1459307764265665,
      "flow_matching_loss": 0.0543169267475605,
      "grad_norm": 0.13304945826530457,
      "learning_rate": 9.239558847358957e-05,
      "loss": 0.0552,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008352979028131812,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.617472648620605,
      "state_query_norm": 1058.4569091796875,
      "state_relation_loss": 0.0008352979028131812,
      "step": 6570,
      "student_distance_mean": 1424.2479858398438,
      "teacher_distance_mean": 1.14616060256958,
      "weighted_auxiliary_loss": 0.0008352979028131812,
      "weighted_relation_loss": 0.0008352979028131812
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.155285313376988,
      "flow_matching_loss": 0.052563071995973584,
      "grad_norm": 0.12862583994865417,
      "learning_rate": 9.236634385083114e-05,
      "loss": 0.0534,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008329321804922074,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.0387845993042,
      "state_query_norm": 1089.1737548828125,
      "state_relation_loss": 0.0008329321804922074,
      "step": 6580,
      "student_distance_mean": 1452.6777709960938,
      "teacher_distance_mean": 1.1550159454345703,
      "weighted_auxiliary_loss": 0.0008329321804922074,
      "weighted_relation_loss": 0.0008329321804922074
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.164639850327409,
      "flow_matching_loss": 0.05400613807141781,
      "grad_norm": 0.1343463957309723,
      "learning_rate": 9.233704774897005e-05,
      "loss": 0.0549,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008454088179860264,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.02432975769043,
      "state_query_norm": 1084.4842529296875,
      "state_relation_loss": 0.0008454088179860264,
      "step": 6590,
      "student_distance_mean": 1451.859521484375,
      "teacher_distance_mean": 1.1563290715217591,
      "weighted_auxiliary_loss": 0.0008454088179860264,
      "weighted_relation_loss": 0.0008454088179860264
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.17399438727783,
      "flow_matching_loss": 0.05259507410228252,
      "grad_norm": 0.14711014926433563,
      "learning_rate": 9.230770020360382e-05,
      "loss": 0.0534,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008447994769085199,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.28918876647949,
      "state_query_norm": 1089.579736328125,
      "state_relation_loss": 0.0008447994769085199,
      "step": 6600,
      "student_distance_mean": 1464.2271850585937,
      "teacher_distance_mean": 1.1660949468612671,
      "weighted_auxiliary_loss": 0.0008447994769085199,
      "weighted_relation_loss": 0.0008447994769085199
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.183348924228251,
      "flow_matching_loss": 0.05292349644005299,
      "grad_norm": 0.12556122243404388,
      "learning_rate": 9.227830125039247e-05,
      "loss": 0.0538,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008382220461498946,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.932187461853026,
      "state_query_norm": 1081.1680541992187,
      "state_relation_loss": 0.0008382220461498946,
      "step": 6610,
      "student_distance_mean": 1444.1958129882812,
      "teacher_distance_mean": 1.1503484487533568,
      "weighted_auxiliary_loss": 0.0008382220461498946,
      "weighted_relation_loss": 0.0008382220461498946
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.192703461178672,
      "flow_matching_loss": 0.052148013934493066,
      "grad_norm": 0.13176700472831726,
      "learning_rate": 9.224885092505853e-05,
      "loss": 0.053,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008298297470901161,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.29610233306885,
      "state_query_norm": 1103.1356567382813,
      "state_relation_loss": 0.0008298297470901161,
      "step": 6620,
      "student_distance_mean": 1468.9651977539063,
      "teacher_distance_mean": 1.1622495174407959,
      "weighted_auxiliary_loss": 0.0008298297470901161,
      "weighted_relation_loss": 0.0008298297470901161
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.202057998129092,
      "flow_matching_loss": 0.053973545879125596,
      "grad_norm": 0.11811382323503494,
      "learning_rate": 9.221934926338691e-05,
      "loss": 0.0548,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008231075771618634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.952346420288087,
      "state_query_norm": 1083.4628173828125,
      "state_relation_loss": 0.0008231075771618634,
      "step": 6630,
      "student_distance_mean": 1448.0054809570313,
      "teacher_distance_mean": 1.1479345083236694,
      "weighted_auxiliary_loss": 0.0008231075771618634,
      "weighted_relation_loss": 0.0008231075771618634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.211412535079513,
      "flow_matching_loss": 0.050901488214731214,
      "grad_norm": 0.13918165862560272,
      "learning_rate": 9.21897963012249e-05,
      "loss": 0.0517,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008438927703537047,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.166483688354493,
      "state_query_norm": 1096.7684814453125,
      "state_relation_loss": 0.0008438927703537047,
      "step": 6640,
      "student_distance_mean": 1458.6014526367187,
      "teacher_distance_mean": 1.1526550769805908,
      "weighted_auxiliary_loss": 0.0008438927703537047,
      "weighted_relation_loss": 0.0008438927703537047
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.220767072029934,
      "flow_matching_loss": 0.05562729164958,
      "grad_norm": 0.1351659595966339,
      "learning_rate": 9.216019207448217e-05,
      "loss": 0.0565,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008984138257801533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.00370807647705,
      "state_query_norm": 1088.1914916992187,
      "state_relation_loss": 0.0008984138257801533,
      "step": 6650,
      "student_distance_mean": 1451.8232055664062,
      "teacher_distance_mean": 1.1474331259727477,
      "weighted_auxiliary_loss": 0.0008984138257801533,
      "weighted_relation_loss": 0.0008984138257801533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.2301216089803555,
      "flow_matching_loss": 0.05350041911005974,
      "grad_norm": 0.165306955575943,
      "learning_rate": 9.21305366191306e-05,
      "loss": 0.0543,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008241747680585831,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.168746948242188,
      "state_query_norm": 1097.1041259765625,
      "state_relation_loss": 0.0008241747680585831,
      "step": 6660,
      "student_distance_mean": 1459.7986572265625,
      "teacher_distance_mean": 1.1573107838630676,
      "weighted_auxiliary_loss": 0.0008241747680585831,
      "weighted_relation_loss": 0.0008241747680585831
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.239476145930777,
      "flow_matching_loss": 0.055141483619809153,
      "grad_norm": 0.1537160873413086,
      "learning_rate": 9.21008299712044e-05,
      "loss": 0.056,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008290050143841654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.135285949707033,
      "state_query_norm": 1088.1917358398437,
      "state_relation_loss": 0.0008290050143841654,
      "step": 6670,
      "student_distance_mean": 1454.6146240234375,
      "teacher_distance_mean": 1.160685157775879,
      "weighted_auxiliary_loss": 0.0008290050143841654,
      "weighted_relation_loss": 0.0008290050143841654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.248830682881198,
      "flow_matching_loss": 0.05293594971299172,
      "grad_norm": 0.1249435618519783,
      "learning_rate": 9.207107216679993e-05,
      "loss": 0.0538,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008573844213970005,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.888319396972655,
      "state_query_norm": 1081.2540405273437,
      "state_relation_loss": 0.0008573844213970005,
      "step": 6680,
      "student_distance_mean": 1444.833544921875,
      "teacher_distance_mean": 1.1515561580657958,
      "weighted_auxiliary_loss": 0.0008573844213970005,
      "weighted_relation_loss": 0.0008573844213970005
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.258185219831619,
      "flow_matching_loss": 0.05251542590558529,
      "grad_norm": 0.12570171058177948,
      "learning_rate": 9.204126324207574e-05,
      "loss": 0.0533,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008013643266167492,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.743854522705078,
      "state_query_norm": 1079.5659545898438,
      "state_relation_loss": 0.0008013643266167492,
      "step": 6690,
      "student_distance_mean": 1436.9754516601563,
      "teacher_distance_mean": 1.1450424075126648,
      "weighted_auxiliary_loss": 0.0008013643266167492,
      "weighted_relation_loss": 0.0008013643266167492
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.267539756782039,
      "flow_matching_loss": 0.05326295532286167,
      "grad_norm": 0.13321126997470856,
      "learning_rate": 9.201140323325248e-05,
      "loss": 0.0541,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008079148072283715,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.609782028198243,
      "state_query_norm": 1071.3910278320313,
      "state_relation_loss": 0.0008079148072283715,
      "step": 6700,
      "student_distance_mean": 1431.3590209960937,
      "teacher_distance_mean": 1.1439270734786988,
      "weighted_auxiliary_loss": 0.0008079148072283715,
      "weighted_relation_loss": 0.0008079148072283715
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.27689429373246,
      "flow_matching_loss": 0.053780867531895635,
      "grad_norm": 0.14290004968643188,
      "learning_rate": 9.198149217661287e-05,
      "loss": 0.0546,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008385066408663988,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.786290550231932,
      "state_query_norm": 1080.0414916992188,
      "state_relation_loss": 0.0008385066408663988,
      "step": 6710,
      "student_distance_mean": 1439.86669921875,
      "teacher_distance_mean": 1.1442141652107238,
      "weighted_auxiliary_loss": 0.0008385066408663988,
      "weighted_relation_loss": 0.0008385066408663988
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.286248830682881,
      "flow_matching_loss": 0.05268605127930641,
      "grad_norm": 0.1629393994808197,
      "learning_rate": 9.195153010850167e-05,
      "loss": 0.0535,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008412499388214201,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.221005821228026,
      "state_query_norm": 1090.9391479492188,
      "state_relation_loss": 0.0008412499388214201,
      "step": 6720,
      "student_distance_mean": 1463.2732543945312,
      "teacher_distance_mean": 1.164817476272583,
      "weighted_auxiliary_loss": 0.0008412499388214201,
      "weighted_relation_loss": 0.0008412499388214201
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.295603367633302,
      "flow_matching_loss": 0.052707308530807497,
      "grad_norm": 0.14744199812412262,
      "learning_rate": 9.192151706532562e-05,
      "loss": 0.0535,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008087058900855482,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.184882354736327,
      "state_query_norm": 1082.35322265625,
      "state_relation_loss": 0.0008087058900855482,
      "step": 6730,
      "student_distance_mean": 1457.9301513671876,
      "teacher_distance_mean": 1.1543872833251954,
      "weighted_auxiliary_loss": 0.0008087058900855482,
      "weighted_relation_loss": 0.0008087058900855482
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.304957904583723,
      "flow_matching_loss": 0.050709442049264905,
      "grad_norm": 0.1296660304069519,
      "learning_rate": 9.189145308355337e-05,
      "loss": 0.0515,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008403381507378072,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.388525009155273,
      "state_query_norm": 1098.82724609375,
      "state_relation_loss": 0.0008403381507378072,
      "step": 6740,
      "student_distance_mean": 1468.6477661132812,
      "teacher_distance_mean": 1.1600340247154235,
      "weighted_auxiliary_loss": 0.0008403381507378072,
      "weighted_relation_loss": 0.0008403381507378072
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.314312441534144,
      "flow_matching_loss": 0.051888054609298705,
      "grad_norm": 0.1494949758052826,
      "learning_rate": 9.186133819971555e-05,
      "loss": 0.0527,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000819621077971533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.977644157409667,
      "state_query_norm": 1093.076220703125,
      "state_relation_loss": 0.000819621077971533,
      "step": 6750,
      "student_distance_mean": 1451.60380859375,
      "teacher_distance_mean": 1.143548274040222,
      "weighted_auxiliary_loss": 0.000819621077971533,
      "weighted_relation_loss": 0.000819621077971533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.323666978484565,
      "flow_matching_loss": 0.05216542854905128,
      "grad_norm": 0.10540354251861572,
      "learning_rate": 9.183117245040455e-05,
      "loss": 0.053,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008054434845689684,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.48818874359131,
      "state_query_norm": 1100.9606323242188,
      "state_relation_loss": 0.0008054434845689684,
      "step": 6760,
      "student_distance_mean": 1474.6708862304688,
      "teacher_distance_mean": 1.165096116065979,
      "weighted_auxiliary_loss": 0.0008054434845689684,
      "weighted_relation_loss": 0.0008054434845689684
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.333021515434986,
      "flow_matching_loss": 0.05220129266381264,
      "grad_norm": 0.16554298996925354,
      "learning_rate": 9.180095587227458e-05,
      "loss": 0.053,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000811746419640258,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.693875312805176,
      "state_query_norm": 1068.055517578125,
      "state_relation_loss": 0.000811746419640258,
      "step": 6770,
      "student_distance_mean": 1431.213720703125,
      "teacher_distance_mean": 1.1344622492790222,
      "weighted_auxiliary_loss": 0.000811746419640258,
      "weighted_relation_loss": 0.000811746419640258
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.342376052385407,
      "flow_matching_loss": 0.053311976790428164,
      "grad_norm": 0.11983513087034225,
      "learning_rate": 9.177068850204167e-05,
      "loss": 0.0541,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007879537355620414,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 25.84125518798828,
      "state_query_norm": 1080.7481811523437,
      "state_relation_loss": 0.0007879537355620414,
      "step": 6780,
      "student_distance_mean": 1440.098974609375,
      "teacher_distance_mean": 1.140079176425934,
      "weighted_auxiliary_loss": 0.0007879537355620414,
      "weighted_relation_loss": 0.0007879537355620414
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.351730589335828,
      "flow_matching_loss": 0.051589440926909445,
      "grad_norm": 0.17926588654518127,
      "learning_rate": 9.174037037648351e-05,
      "loss": 0.0524,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008513665816280991,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.338827323913574,
      "state_query_norm": 1097.5080078125,
      "state_relation_loss": 0.0008513665816280991,
      "step": 6790,
      "student_distance_mean": 1465.0146728515624,
      "teacher_distance_mean": 1.1495097279548645,
      "weighted_auxiliary_loss": 0.0008513665816280991,
      "weighted_relation_loss": 0.0008513665816280991
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.361085126286249,
      "flow_matching_loss": 0.049983826279640195,
      "grad_norm": 0.19223928451538086,
      "learning_rate": 9.171000153243947e-05,
      "loss": 0.0508,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008261111797764897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.20730495452881,
      "state_query_norm": 1089.301708984375,
      "state_relation_loss": 0.0008261111797764897,
      "step": 6800,
      "student_distance_mean": 1459.9931396484376,
      "teacher_distance_mean": 1.1458650827407837,
      "weighted_auxiliary_loss": 0.0008261111797764897,
      "weighted_relation_loss": 0.0008261111797764897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.37043966323667,
      "flow_matching_loss": 0.05326565280556679,
      "grad_norm": 0.12041887640953064,
      "learning_rate": 9.167958200681057e-05,
      "loss": 0.0541,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008111677598208189,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.311501884460448,
      "state_query_norm": 1092.4974365234375,
      "state_relation_loss": 0.0008111677598208189,
      "step": 6810,
      "student_distance_mean": 1465.8932739257812,
      "teacher_distance_mean": 1.1491985082626344,
      "weighted_auxiliary_loss": 0.0008111677598208189,
      "weighted_relation_loss": 0.0008111677598208189
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.37979420018709,
      "flow_matching_loss": 0.051247164607048035,
      "grad_norm": 0.14169812202453613,
      "learning_rate": 9.164911183655943e-05,
      "loss": 0.0521,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000826694507850334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.382359886169432,
      "state_query_norm": 1090.58876953125,
      "state_relation_loss": 0.000826694507850334,
      "step": 6820,
      "student_distance_mean": 1469.505810546875,
      "teacher_distance_mean": 1.1548988223075867,
      "weighted_auxiliary_loss": 0.000826694507850334,
      "weighted_relation_loss": 0.000826694507850334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.3891487371375115,
      "flow_matching_loss": 0.05110749043524265,
      "grad_norm": 0.1121506616473198,
      "learning_rate": 9.161859105871013e-05,
      "loss": 0.0519,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008273475978057832,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.421445274353026,
      "state_query_norm": 1090.2079345703125,
      "state_relation_loss": 0.0008273475978057832,
      "step": 6830,
      "student_distance_mean": 1470.5293090820312,
      "teacher_distance_mean": 1.151909601688385,
      "weighted_auxiliary_loss": 0.0008273475978057832,
      "weighted_relation_loss": 0.0008273475978057832
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.398503274087933,
      "flow_matching_loss": 0.05205156952142716,
      "grad_norm": 0.11892655491828918,
      "learning_rate": 9.158801971034832e-05,
      "loss": 0.0529,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008143524406477809,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.219715881347657,
      "state_query_norm": 1083.5780029296875,
      "state_relation_loss": 0.0008143524406477809,
      "step": 6840,
      "student_distance_mean": 1457.2377075195313,
      "teacher_distance_mean": 1.142713725566864,
      "weighted_auxiliary_loss": 0.0008143524406477809,
      "weighted_relation_loss": 0.0008143524406477809
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.407857811038354,
      "flow_matching_loss": 0.05125122331082821,
      "grad_norm": 0.15150152146816254,
      "learning_rate": 9.155739782862107e-05,
      "loss": 0.052,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007928341859951615,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.258788299560546,
      "state_query_norm": 1094.9441284179688,
      "state_relation_loss": 0.0007928341859951615,
      "step": 6850,
      "student_distance_mean": 1464.5373291015626,
      "teacher_distance_mean": 1.1486747145652771,
      "weighted_auxiliary_loss": 0.0007928341859951615,
      "weighted_relation_loss": 0.0007928341859951615
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.417212347988775,
      "flow_matching_loss": 0.05129800476133824,
      "grad_norm": 0.16377291083335876,
      "learning_rate": 9.152672545073687e-05,
      "loss": 0.0521,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008453478978481144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.25857467651367,
      "state_query_norm": 1088.22890625,
      "state_relation_loss": 0.0008453478978481144,
      "step": 6860,
      "student_distance_mean": 1459.5937622070312,
      "teacher_distance_mean": 1.146351659297943,
      "weighted_auxiliary_loss": 0.0008453478978481144,
      "weighted_relation_loss": 0.0008453478978481144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.426566884939196,
      "flow_matching_loss": 0.05357868857681751,
      "grad_norm": 0.13787841796875,
      "learning_rate": 9.149600261396551e-05,
      "loss": 0.0544,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008302063099108637,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.505550384521484,
      "state_query_norm": 1100.9586059570313,
      "state_relation_loss": 0.0008302063099108637,
      "step": 6870,
      "student_distance_mean": 1475.2672729492188,
      "teacher_distance_mean": 1.1552507877349854,
      "weighted_auxiliary_loss": 0.0008302063099108637,
      "weighted_relation_loss": 0.0008302063099108637
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.435921421889616,
      "flow_matching_loss": 0.04899468533694744,
      "grad_norm": 0.13227519392967224,
      "learning_rate": 9.146522935563817e-05,
      "loss": 0.0498,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000848836312070489,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.45686569213867,
      "state_query_norm": 1107.8357299804688,
      "state_relation_loss": 0.000848836312070489,
      "step": 6880,
      "student_distance_mean": 1477.856103515625,
      "teacher_distance_mean": 1.1520871996879578,
      "weighted_auxiliary_loss": 0.000848836312070489,
      "weighted_relation_loss": 0.000848836312070489
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.445275958840037,
      "flow_matching_loss": 0.05133989416062832,
      "grad_norm": 0.1756073534488678,
      "learning_rate": 9.143440571314723e-05,
      "loss": 0.0521,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007787814945913851,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.55211944580078,
      "state_query_norm": 1107.15751953125,
      "state_relation_loss": 0.0007787814945913851,
      "step": 6890,
      "student_distance_mean": 1479.1510498046875,
      "teacher_distance_mean": 1.16342191696167,
      "weighted_auxiliary_loss": 0.0007787814945913851,
      "weighted_relation_loss": 0.0007787814945913851
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.454630495790458,
      "flow_matching_loss": 0.052909844741225245,
      "grad_norm": 0.16688495874404907,
      "learning_rate": 9.140353172394636e-05,
      "loss": 0.0537,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008209490973968059,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.28299160003662,
      "state_query_norm": 1096.710498046875,
      "state_relation_loss": 0.0008209490973968059,
      "step": 6900,
      "student_distance_mean": 1467.1595825195313,
      "teacher_distance_mean": 1.1498406648635864,
      "weighted_auxiliary_loss": 0.0008209490973968059,
      "weighted_relation_loss": 0.0008209490973968059
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.463985032740879,
      "flow_matching_loss": 0.05319203287363052,
      "grad_norm": 0.14570674300193787,
      "learning_rate": 9.137260742555032e-05,
      "loss": 0.054,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008210320607759059,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.81219348907471,
      "state_query_norm": 1116.0814819335938,
      "state_relation_loss": 0.0008210320607759059,
      "step": 6910,
      "student_distance_mean": 1493.5334350585938,
      "teacher_distance_mean": 1.1616217851638795,
      "weighted_auxiliary_loss": 0.0008210320607759059,
      "weighted_relation_loss": 0.0008210320607759059
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.4733395696913005,
      "flow_matching_loss": 0.05381365604698658,
      "grad_norm": 0.13565775752067566,
      "learning_rate": 9.13416328555351e-05,
      "loss": 0.0546,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007763827743474394,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.81450309753418,
      "state_query_norm": 1113.7809814453126,
      "state_relation_loss": 0.0007763827743474394,
      "step": 6920,
      "student_distance_mean": 1494.5125854492187,
      "teacher_distance_mean": 1.1599271059036256,
      "weighted_auxiliary_loss": 0.0007763827743474394,
      "weighted_relation_loss": 0.0007763827743474394
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.482694106641722,
      "flow_matching_loss": 0.05173491202294826,
      "grad_norm": 0.12567070126533508,
      "learning_rate": 9.131060805153769e-05,
      "loss": 0.0525,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008092425880022347,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.664759826660156,
      "state_query_norm": 1099.8254760742188,
      "state_relation_loss": 0.0008092425880022347,
      "step": 6930,
      "student_distance_mean": 1480.5244140625,
      "teacher_distance_mean": 1.1509940862655639,
      "weighted_auxiliary_loss": 0.0008092425880022347,
      "weighted_relation_loss": 0.0008092425880022347
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.492048643592142,
      "flow_matching_loss": 0.0531077429652214,
      "grad_norm": 0.13073717057704926,
      "learning_rate": 9.127953305125618e-05,
      "loss": 0.0539,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008105483080726117,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.687911605834962,
      "state_query_norm": 1104.530322265625,
      "state_relation_loss": 0.0008105483080726117,
      "step": 6940,
      "student_distance_mean": 1483.7767944335938,
      "teacher_distance_mean": 1.1456781029701233,
      "weighted_auxiliary_loss": 0.0008105483080726117,
      "weighted_relation_loss": 0.0008105483080726117
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.501403180542563,
      "flow_matching_loss": 0.05276038348674774,
      "grad_norm": 0.11367229372262955,
      "learning_rate": 9.124840789244958e-05,
      "loss": 0.0536,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000851612922269851,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.963486099243163,
      "state_query_norm": 1112.8530029296876,
      "state_relation_loss": 0.000851612922269851,
      "step": 6950,
      "student_distance_mean": 1497.9542236328125,
      "teacher_distance_mean": 1.1565261006355285,
      "weighted_auxiliary_loss": 0.000851612922269851,
      "weighted_relation_loss": 0.000851612922269851
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.510757717492984,
      "flow_matching_loss": 0.0538957305252552,
      "grad_norm": 0.17372582852840424,
      "learning_rate": 9.121723261293794e-05,
      "loss": 0.0547,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007926520309410989,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.845581436157225,
      "state_query_norm": 1111.5967529296875,
      "state_relation_loss": 0.0007926520309410989,
      "step": 6960,
      "student_distance_mean": 1494.8857177734376,
      "teacher_distance_mean": 1.1577036261558533,
      "weighted_auxiliary_loss": 0.0007926520309410989,
      "weighted_relation_loss": 0.0007926520309410989
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.520112254443405,
      "flow_matching_loss": 0.049447083845734596,
      "grad_norm": 0.12115083634853363,
      "learning_rate": 9.118600725060214e-05,
      "loss": 0.0502,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007913839304819703,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.16537570953369,
      "state_query_norm": 1127.1950317382812,
      "state_relation_loss": 0.0007913839304819703,
      "step": 6970,
      "student_distance_mean": 1511.556884765625,
      "teacher_distance_mean": 1.1604792952537537,
      "weighted_auxiliary_loss": 0.0007913839304819703,
      "weighted_relation_loss": 0.0007913839304819703
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.529466791393826,
      "flow_matching_loss": 0.05197048224508762,
      "grad_norm": 0.14813607931137085,
      "learning_rate": 9.115473184338392e-05,
      "loss": 0.0528,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007861710269935429,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.16769618988037,
      "state_query_norm": 1118.4878051757812,
      "state_relation_loss": 0.0007861710269935429,
      "step": 6980,
      "student_distance_mean": 1509.417578125,
      "teacher_distance_mean": 1.1672661185264588,
      "weighted_auxiliary_loss": 0.0007861710269935429,
      "weighted_relation_loss": 0.0007861710269935429
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.538821328344247,
      "flow_matching_loss": 0.05317265577614307,
      "grad_norm": 0.19765284657478333,
      "learning_rate": 9.11234064292859e-05,
      "loss": 0.054,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007854317896999419,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.027472305297852,
      "state_query_norm": 1120.0164916992187,
      "state_relation_loss": 0.0007854317896999419,
      "step": 6990,
      "student_distance_mean": 1505.765283203125,
      "teacher_distance_mean": 1.1623998761177063,
      "weighted_auxiliary_loss": 0.0007854317896999419,
      "weighted_relation_loss": 0.0007854317896999419
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.5481758652946676,
      "flow_matching_loss": 0.05303711742162705,
      "grad_norm": 0.18560680747032166,
      "learning_rate": 9.109203104637137e-05,
      "loss": 0.0538,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007751599652692676,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.711203002929686,
      "state_query_norm": 1113.3006225585937,
      "state_relation_loss": 0.0007751599652692676,
      "step": 7000,
      "student_distance_mean": 1489.3268676757812,
      "teacher_distance_mean": 1.1507430672645569,
      "weighted_auxiliary_loss": 0.0007751599652692676,
      "weighted_relation_loss": 0.0007751599652692676
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.557530402245089,
      "flow_matching_loss": 0.054736285284161565,
      "grad_norm": 0.15622426569461823,
      "learning_rate": 9.10606057327644e-05,
      "loss": 0.0555,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007807463814970105,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.593360137939452,
      "state_query_norm": 1103.2366821289063,
      "state_relation_loss": 0.0007807463814970105,
      "step": 7010,
      "student_distance_mean": 1481.21953125,
      "teacher_distance_mean": 1.1466729640960693,
      "weighted_auxiliary_loss": 0.0007807463814970105,
      "weighted_relation_loss": 0.0007807463814970105
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.56688493919551,
      "flow_matching_loss": 0.051376907899975774,
      "grad_norm": 0.1506839245557785,
      "learning_rate": 9.10291305266497e-05,
      "loss": 0.0522,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007985943462699652,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.94077262878418,
      "state_query_norm": 1116.5166015625,
      "state_relation_loss": 0.0007985943462699652,
      "step": 7020,
      "student_distance_mean": 1498.57490234375,
      "teacher_distance_mean": 1.1500297784805298,
      "weighted_auxiliary_loss": 0.0007985943462699652,
      "weighted_relation_loss": 0.0007985943462699652
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.576239476145931,
      "flow_matching_loss": 0.05461766980588436,
      "grad_norm": 0.14533568918704987,
      "learning_rate": 9.099760546627261e-05,
      "loss": 0.0554,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007987197197508067,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.759513092041015,
      "state_query_norm": 1113.6426147460938,
      "state_relation_loss": 0.0007987197197508067,
      "step": 7030,
      "student_distance_mean": 1491.744189453125,
      "teacher_distance_mean": 1.1405842900276184,
      "weighted_auxiliary_loss": 0.0007987197197508067,
      "weighted_relation_loss": 0.0007987197197508067
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.585594013096352,
      "flow_matching_loss": 0.05150098651647568,
      "grad_norm": 0.15986357629299164,
      "learning_rate": 9.096603058993907e-05,
      "loss": 0.0523,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008372721786145121,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.743911933898925,
      "state_query_norm": 1113.6238403320312,
      "state_relation_loss": 0.0008372721786145121,
      "step": 7040,
      "student_distance_mean": 1488.8484252929688,
      "teacher_distance_mean": 1.1549704670906067,
      "weighted_auxiliary_loss": 0.0008372721786145121,
      "weighted_relation_loss": 0.0008372721786145121
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.594948550046773,
      "flow_matching_loss": 0.052480964362621306,
      "grad_norm": 0.14559432864189148,
      "learning_rate": 9.093440593601552e-05,
      "loss": 0.0533,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008069760631769895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.18503017425537,
      "state_query_norm": 1119.9052001953125,
      "state_relation_loss": 0.0008069760631769895,
      "step": 7050,
      "student_distance_mean": 1509.0843383789063,
      "teacher_distance_mean": 1.1681195855140687,
      "weighted_auxiliary_loss": 0.0008069760631769895,
      "weighted_relation_loss": 0.0008069760631769895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.604303086997193,
      "flow_matching_loss": 0.05030200369656086,
      "grad_norm": 0.16681647300720215,
      "learning_rate": 9.090273154292889e-05,
      "loss": 0.0511,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008071450341958552,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.487063598632812,
      "state_query_norm": 1102.3310791015624,
      "state_relation_loss": 0.0008071450341958552,
      "step": 7060,
      "student_distance_mean": 1475.1478271484375,
      "teacher_distance_mean": 1.1370999574661256,
      "weighted_auxiliary_loss": 0.0008071450341958552,
      "weighted_relation_loss": 0.0008071450341958552
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.613657623947614,
      "flow_matching_loss": 0.05311484821140766,
      "grad_norm": 0.1694062203168869,
      "learning_rate": 9.087100744916656e-05,
      "loss": 0.0539,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008203546633012593,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.758731842041016,
      "state_query_norm": 1104.1432861328126,
      "state_relation_loss": 0.0008203546633012593,
      "step": 7070,
      "student_distance_mean": 1487.4151123046875,
      "teacher_distance_mean": 1.1602235555648803,
      "weighted_auxiliary_loss": 0.0008203546633012593,
      "weighted_relation_loss": 0.0008203546633012593
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.6230121608980355,
      "flow_matching_loss": 0.051842575147747996,
      "grad_norm": 0.12244708836078644,
      "learning_rate": 9.08392336932763e-05,
      "loss": 0.0527,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008385569031815976,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.82108917236328,
      "state_query_norm": 1111.3651977539062,
      "state_relation_loss": 0.0008385569031815976,
      "step": 7080,
      "student_distance_mean": 1493.2157836914062,
      "teacher_distance_mean": 1.1619582891464233,
      "weighted_auxiliary_loss": 0.0008385569031815976,
      "weighted_relation_loss": 0.0008385569031815976
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.632366697848457,
      "flow_matching_loss": 0.051978171616792676,
      "grad_norm": 0.15284475684165955,
      "learning_rate": 9.080741031386619e-05,
      "loss": 0.0528,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008239686663728208,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.404158782958984,
      "state_query_norm": 1134.1251708984375,
      "state_relation_loss": 0.0008239686663728208,
      "step": 7090,
      "student_distance_mean": 1523.8092163085937,
      "teacher_distance_mean": 1.1691144466400147,
      "weighted_auxiliary_loss": 0.0008239686663728208,
      "weighted_relation_loss": 0.0008239686663728208
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.641721234798878,
      "flow_matching_loss": 0.05205826833844185,
      "grad_norm": 0.1488076001405716,
      "learning_rate": 9.077553734960468e-05,
      "loss": 0.0529,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008096672710962594,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.870612335205077,
      "state_query_norm": 1117.1535522460938,
      "state_relation_loss": 0.0008096672710962594,
      "step": 7100,
      "student_distance_mean": 1495.6119995117188,
      "teacher_distance_mean": 1.1518354177474976,
      "weighted_auxiliary_loss": 0.0008096672710962594,
      "weighted_relation_loss": 0.0008096672710962594
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.651075771749299,
      "flow_matching_loss": 0.050068586692214014,
      "grad_norm": 0.14862285554409027,
      "learning_rate": 9.074361483922041e-05,
      "loss": 0.0509,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008138683682773262,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.930113983154296,
      "state_query_norm": 1111.18955078125,
      "state_relation_loss": 0.0008138683682773262,
      "step": 7110,
      "student_distance_mean": 1493.4775634765624,
      "teacher_distance_mean": 1.1490979313850402,
      "weighted_auxiliary_loss": 0.0008138683682773262,
      "weighted_relation_loss": 0.0008138683682773262
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.66043030869972,
      "flow_matching_loss": 0.0514182548969984,
      "grad_norm": 0.12293051928281784,
      "learning_rate": 9.071164282150224e-05,
      "loss": 0.0522,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008061498578172177,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.04173812866211,
      "state_query_norm": 1125.1993896484375,
      "state_relation_loss": 0.0008061498578172177,
      "step": 7120,
      "student_distance_mean": 1503.7447143554687,
      "teacher_distance_mean": 1.153918945789337,
      "weighted_auxiliary_loss": 0.0008061498578172177,
      "weighted_relation_loss": 0.0008061498578172177
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.66978484565014,
      "flow_matching_loss": 0.05154693238437176,
      "grad_norm": 0.15261688828468323,
      "learning_rate": 9.067962133529919e-05,
      "loss": 0.0524,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008146876934915781,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.27861633300781,
      "state_query_norm": 1136.4538818359374,
      "state_relation_loss": 0.0008146876934915781,
      "step": 7130,
      "student_distance_mean": 1520.3600341796875,
      "teacher_distance_mean": 1.1584664702415466,
      "weighted_auxiliary_loss": 0.0008146876934915781,
      "weighted_relation_loss": 0.0008146876934915781
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.679139382600561,
      "flow_matching_loss": 0.05037154071033001,
      "grad_norm": 0.12631431221961975,
      "learning_rate": 9.064755041952036e-05,
      "loss": 0.0512,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008042719855438918,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.377744102478026,
      "state_query_norm": 1139.2041625976562,
      "state_relation_loss": 0.0008042719855438918,
      "step": 7140,
      "student_distance_mean": 1523.469580078125,
      "teacher_distance_mean": 1.1510871767997741,
      "weighted_auxiliary_loss": 0.0008042719855438918,
      "weighted_relation_loss": 0.0008042719855438918
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.688493919550982,
      "flow_matching_loss": 0.052776817977428434,
      "grad_norm": 0.13270701467990875,
      "learning_rate": 9.061543011313497e-05,
      "loss": 0.0536,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000804930180311203,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.035833740234374,
      "state_query_norm": 1120.8883666992188,
      "state_relation_loss": 0.000804930180311203,
      "step": 7150,
      "student_distance_mean": 1503.7063354492188,
      "teacher_distance_mean": 1.143418025970459,
      "weighted_auxiliary_loss": 0.000804930180311203,
      "weighted_relation_loss": 0.000804930180311203
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.697848456501403,
      "flow_matching_loss": 0.052303094789385794,
      "grad_norm": 0.1265735775232315,
      "learning_rate": 9.05832604551722e-05,
      "loss": 0.0531,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008294686558656394,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.197256660461427,
      "state_query_norm": 1135.918994140625,
      "state_relation_loss": 0.0008294686558656394,
      "step": 7160,
      "student_distance_mean": 1514.0926879882813,
      "teacher_distance_mean": 1.154975950717926,
      "weighted_auxiliary_loss": 0.0008294686558656394,
      "weighted_relation_loss": 0.0008294686558656394
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.7072029934518245,
      "flow_matching_loss": 0.049378127232193945,
      "grad_norm": 0.12423447519540787,
      "learning_rate": 9.055104148472124e-05,
      "loss": 0.0502,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008224859018810093,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.241493797302248,
      "state_query_norm": 1124.7141845703125,
      "state_relation_loss": 0.0008224859018810093,
      "step": 7170,
      "student_distance_mean": 1515.7161865234375,
      "teacher_distance_mean": 1.1539496541023255,
      "weighted_auxiliary_loss": 0.0008224859018810093,
      "weighted_relation_loss": 0.0008224859018810093
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.716557530402245,
      "flow_matching_loss": 0.05335555970668793,
      "grad_norm": 0.18562841415405273,
      "learning_rate": 9.051877324093114e-05,
      "loss": 0.0542,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000857312313746661,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.263312339782715,
      "state_query_norm": 1133.4074829101562,
      "state_relation_loss": 0.000857312313746661,
      "step": 7180,
      "student_distance_mean": 1516.880029296875,
      "teacher_distance_mean": 1.1628609418869018,
      "weighted_auxiliary_loss": 0.000857312313746661,
      "weighted_relation_loss": 0.000857312313746661
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.725912067352666,
      "flow_matching_loss": 0.04943303316831589,
      "grad_norm": 0.21210519969463348,
      "learning_rate": 9.04864557630109e-05,
      "loss": 0.0502,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008136116550303996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.173739433288574,
      "state_query_norm": 1123.0962280273438,
      "state_relation_loss": 0.0008136116550303996,
      "step": 7190,
      "student_distance_mean": 1510.605908203125,
      "teacher_distance_mean": 1.158105731010437,
      "weighted_auxiliary_loss": 0.0008136116550303996,
      "weighted_relation_loss": 0.0008136116550303996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.735266604303087,
      "flow_matching_loss": 0.05424777790904045,
      "grad_norm": 0.12029723078012466,
      "learning_rate": 9.045408909022928e-05,
      "loss": 0.055,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007622925098985433,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.990926361083986,
      "state_query_norm": 1121.5781616210938,
      "state_relation_loss": 0.0007622925098985433,
      "step": 7200,
      "student_distance_mean": 1504.3927001953125,
      "teacher_distance_mean": 1.1541468858718873,
      "weighted_auxiliary_loss": 0.0007622925098985433,
      "weighted_relation_loss": 0.0007622925098985433
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.744621141253508,
      "flow_matching_loss": 0.05393092185258865,
      "grad_norm": 0.2116503268480301,
      "learning_rate": 9.042167326191484e-05,
      "loss": 0.0547,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008062004926614463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.66452865600586,
      "state_query_norm": 1143.844091796875,
      "state_relation_loss": 0.0008062004926614463,
      "step": 7210,
      "student_distance_mean": 1539.4643310546876,
      "teacher_distance_mean": 1.1733983993530273,
      "weighted_auxiliary_loss": 0.0008062004926614463,
      "weighted_relation_loss": 0.0008062004926614463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.753975678203929,
      "flow_matching_loss": 0.052571256458759305,
      "grad_norm": 0.17706532776355743,
      "learning_rate": 9.038920831745587e-05,
      "loss": 0.0534,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007912096451036632,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.173241806030273,
      "state_query_norm": 1128.9558715820312,
      "state_relation_loss": 0.0007912096451036632,
      "step": 7220,
      "student_distance_mean": 1512.7201416015625,
      "teacher_distance_mean": 1.1426467418670654,
      "weighted_auxiliary_loss": 0.0007912096451036632,
      "weighted_relation_loss": 0.0007912096451036632
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.76333021515435,
      "flow_matching_loss": 0.05354878716170788,
      "grad_norm": 0.11147145926952362,
      "learning_rate": 9.035669429630036e-05,
      "loss": 0.0543,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007776529353577643,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.25881862640381,
      "state_query_norm": 1141.1360961914063,
      "state_relation_loss": 0.0007776529353577643,
      "step": 7230,
      "student_distance_mean": 1521.01328125,
      "teacher_distance_mean": 1.1480881094932556,
      "weighted_auxiliary_loss": 0.0007776529353577643,
      "weighted_relation_loss": 0.0007776529353577643
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.772684752104771,
      "flow_matching_loss": 0.04991618134081364,
      "grad_norm": 0.1408117562532425,
      "learning_rate": 9.032413123795588e-05,
      "loss": 0.0506,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007316353148780763,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.29192237854004,
      "state_query_norm": 1134.1728881835938,
      "state_relation_loss": 0.0007316353148780763,
      "step": 7240,
      "student_distance_mean": 1518.9466796875,
      "teacher_distance_mean": 1.145373237133026,
      "weighted_auxiliary_loss": 0.0007316353148780763,
      "weighted_relation_loss": 0.0007316353148780763
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.7820392890551915,
      "flow_matching_loss": 0.050749833881855014,
      "grad_norm": 0.18876172602176666,
      "learning_rate": 9.029151918198962e-05,
      "loss": 0.0516,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008015273662749678,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.122357177734376,
      "state_query_norm": 1128.9006225585938,
      "state_relation_loss": 0.0008015273662749678,
      "step": 7250,
      "student_distance_mean": 1511.8814208984375,
      "teacher_distance_mean": 1.1393626928329468,
      "weighted_auxiliary_loss": 0.0008015273662749678,
      "weighted_relation_loss": 0.0008015273662749678
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.791393826005613,
      "flow_matching_loss": 0.05035816058516503,
      "grad_norm": 0.14533773064613342,
      "learning_rate": 9.025885816802833e-05,
      "loss": 0.0512,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007963240146636963,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.6502592086792,
      "state_query_norm": 1141.255517578125,
      "state_relation_loss": 0.0007963240146636963,
      "step": 7260,
      "student_distance_mean": 1539.6926391601562,
      "teacher_distance_mean": 1.1583477854728699,
      "weighted_auxiliary_loss": 0.0007963240146636963,
      "weighted_relation_loss": 0.0007963240146636963
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.800748362956034,
      "flow_matching_loss": 0.05142630785703659,
      "grad_norm": 0.13325762748718262,
      "learning_rate": 9.02261482357582e-05,
      "loss": 0.0522,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007965078810229897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.689296340942384,
      "state_query_norm": 1147.8856689453125,
      "state_relation_loss": 0.0007965078810229897,
      "step": 7270,
      "student_distance_mean": 1540.2786865234375,
      "teacher_distance_mean": 1.1631543517112732,
      "weighted_auxiliary_loss": 0.0007965078810229897,
      "weighted_relation_loss": 0.0007965078810229897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.810102899906455,
      "flow_matching_loss": 0.05046956874430179,
      "grad_norm": 0.1395898163318634,
      "learning_rate": 9.019338942492485e-05,
      "loss": 0.0513,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008101387065835297,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.192023086547852,
      "state_query_norm": 1124.7653686523438,
      "state_relation_loss": 0.0008101387065835297,
      "step": 7280,
      "student_distance_mean": 1510.0705078125,
      "teacher_distance_mean": 1.1430625796318055,
      "weighted_auxiliary_loss": 0.0008101387065835297,
      "weighted_relation_loss": 0.0008101387065835297
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.819457436856876,
      "flow_matching_loss": 0.05317705497145653,
      "grad_norm": 0.12014435231685638,
      "learning_rate": 9.01605817753334e-05,
      "loss": 0.054,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007983733259607107,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.860311698913574,
      "state_query_norm": 1151.85771484375,
      "state_relation_loss": 0.0007983733259607107,
      "step": 7290,
      "student_distance_mean": 1549.3204711914063,
      "teacher_distance_mean": 1.1680981874465943,
      "weighted_auxiliary_loss": 0.0007983733259607107,
      "weighted_relation_loss": 0.0007983733259607107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.828811973807296,
      "flow_matching_loss": 0.05181320048868656,
      "grad_norm": 0.11365775018930435,
      "learning_rate": 9.012772532684818e-05,
      "loss": 0.0526,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007730497804004699,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.465356636047364,
      "state_query_norm": 1147.7003540039063,
      "state_relation_loss": 0.0007730497804004699,
      "step": 7300,
      "student_distance_mean": 1531.1548950195313,
      "teacher_distance_mean": 1.1536117553710938,
      "weighted_auxiliary_loss": 0.0007730497804004699,
      "weighted_relation_loss": 0.0007730497804004699
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.838166510757717,
      "flow_matching_loss": 0.05039530247449875,
      "grad_norm": 0.1619308441877365,
      "learning_rate": 9.00948201193929e-05,
      "loss": 0.0512,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007955512613989413,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.14375591278076,
      "state_query_norm": 1125.7929565429688,
      "state_relation_loss": 0.0007955512613989413,
      "step": 7310,
      "student_distance_mean": 1511.396826171875,
      "teacher_distance_mean": 1.152135169506073,
      "weighted_auxiliary_loss": 0.0007955512613989413,
      "weighted_relation_loss": 0.0007955512613989413
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.847521047708138,
      "flow_matching_loss": 0.05121591500937939,
      "grad_norm": 0.11305305361747742,
      "learning_rate": 9.006186619295047e-05,
      "loss": 0.052,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007532191171776504,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.2058464050293,
      "state_query_norm": 1128.1773803710937,
      "state_relation_loss": 0.0007532191171776504,
      "step": 7320,
      "student_distance_mean": 1512.8025268554688,
      "teacher_distance_mean": 1.1440853238105775,
      "weighted_auxiliary_loss": 0.0007532191171776504,
      "weighted_relation_loss": 0.0007532191171776504
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.856875584658559,
      "flow_matching_loss": 0.051529668271541595,
      "grad_norm": 0.1185077577829361,
      "learning_rate": 9.002886358756305e-05,
      "loss": 0.0523,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007705391384661197,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.5930757522583,
      "state_query_norm": 1138.249072265625,
      "state_relation_loss": 0.0007705391384661197,
      "step": 7330,
      "student_distance_mean": 1534.3607299804687,
      "teacher_distance_mean": 1.152604615688324,
      "weighted_auxiliary_loss": 0.0007705391384661197,
      "weighted_relation_loss": 0.0007705391384661197
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.8662301216089805,
      "flow_matching_loss": 0.05187285207211971,
      "grad_norm": 0.10848328471183777,
      "learning_rate": 8.999581234333188e-05,
      "loss": 0.0527,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007832440896891057,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.285427856445313,
      "state_query_norm": 1132.2359619140625,
      "state_relation_loss": 0.0007832440896891057,
      "step": 7340,
      "student_distance_mean": 1520.1670043945312,
      "teacher_distance_mean": 1.152870202064514,
      "weighted_auxiliary_loss": 0.0007832440896891057,
      "weighted_relation_loss": 0.0007832440896891057
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.875584658559402,
      "flow_matching_loss": 0.04966067187488079,
      "grad_norm": 0.14168542623519897,
      "learning_rate": 8.996271250041734e-05,
      "loss": 0.0504,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007600149023346603,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.770338821411134,
      "state_query_norm": 1104.5586059570312,
      "state_relation_loss": 0.0007600149023346603,
      "step": 7350,
      "student_distance_mean": 1491.1626708984375,
      "teacher_distance_mean": 1.140852415561676,
      "weighted_auxiliary_loss": 0.0007600149023346603,
      "weighted_relation_loss": 0.0007600149023346603
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.884939195509823,
      "flow_matching_loss": 0.05152480565011501,
      "grad_norm": 0.11494863033294678,
      "learning_rate": 8.99295640990389e-05,
      "loss": 0.0523,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007892089197412133,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.314362716674804,
      "state_query_norm": 1133.742041015625,
      "state_relation_loss": 0.0007892089197412133,
      "step": 7360,
      "student_distance_mean": 1522.5475219726563,
      "teacher_distance_mean": 1.1574462532997132,
      "weighted_auxiliary_loss": 0.0007892089197412133,
      "weighted_relation_loss": 0.0007892089197412133
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.894293732460243,
      "flow_matching_loss": 0.05220405496656895,
      "grad_norm": 0.17654871940612793,
      "learning_rate": 8.989636717947494e-05,
      "loss": 0.053,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007515823817811906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 26.940713500976564,
      "state_query_norm": 1126.4122192382813,
      "state_relation_loss": 0.0007515823817811906,
      "step": 7370,
      "student_distance_mean": 1504.0833251953125,
      "teacher_distance_mean": 1.147915518283844,
      "weighted_auxiliary_loss": 0.0007515823817811906,
      "weighted_relation_loss": 0.0007515823817811906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.903648269410664,
      "flow_matching_loss": 0.04982322603464127,
      "grad_norm": 0.14072731137275696,
      "learning_rate": 8.986312178206291e-05,
      "loss": 0.0506,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008103873173240572,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.34815845489502,
      "state_query_norm": 1128.8091430664062,
      "state_relation_loss": 0.0008103873173240572,
      "step": 7380,
      "student_distance_mean": 1520.3099975585938,
      "teacher_distance_mean": 1.1552500963211059,
      "weighted_auxiliary_loss": 0.0008103873173240572,
      "weighted_relation_loss": 0.0008103873173240572
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.913002806361085,
      "flow_matching_loss": 0.05068164430558682,
      "grad_norm": 0.13891784846782684,
      "learning_rate": 8.982982794719904e-05,
      "loss": 0.0514,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007656061497982591,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.16382713317871,
      "state_query_norm": 1137.9172119140626,
      "state_relation_loss": 0.0007656061497982591,
      "step": 7390,
      "student_distance_mean": 1514.8644653320312,
      "teacher_distance_mean": 1.1419543266296386,
      "weighted_auxiliary_loss": 0.0007656061497982591,
      "weighted_relation_loss": 0.0007656061497982591
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.922357343311506,
      "flow_matching_loss": 0.05102501697838306,
      "grad_norm": 0.18192288279533386,
      "learning_rate": 8.979648571533852e-05,
      "loss": 0.0519,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008299614593852312,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.629748916625978,
      "state_query_norm": 1151.8838500976562,
      "state_relation_loss": 0.0008299614593852312,
      "step": 7400,
      "student_distance_mean": 1537.7716552734375,
      "teacher_distance_mean": 1.1665118217468262,
      "weighted_auxiliary_loss": 0.0008299614593852312,
      "weighted_relation_loss": 0.0008299614593852312
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.931711880261927,
      "flow_matching_loss": 0.050193680077791215,
      "grad_norm": 0.20030547678470612,
      "learning_rate": 8.976309512699528e-05,
      "loss": 0.051,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007988739525899291,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.788753318786622,
      "state_query_norm": 1149.1217041015625,
      "state_relation_loss": 0.0007988739525899291,
      "step": 7410,
      "student_distance_mean": 1544.3786987304688,
      "teacher_distance_mean": 1.1634599447250367,
      "weighted_auxiliary_loss": 0.0007988739525899291,
      "weighted_relation_loss": 0.0007988739525899291
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.9410664172123475,
      "flow_matching_loss": 0.05061730891466141,
      "grad_norm": 0.18240448832511902,
      "learning_rate": 8.972965622274205e-05,
      "loss": 0.0514,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007450170756783337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.197398948669434,
      "state_query_norm": 1135.63359375,
      "state_relation_loss": 0.0007450170756783337,
      "step": 7420,
      "student_distance_mean": 1513.60947265625,
      "teacher_distance_mean": 1.136962080001831,
      "weighted_auxiliary_loss": 0.0007450170756783337,
      "weighted_relation_loss": 0.0007450170756783337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.950420954162769,
      "flow_matching_loss": 0.05020713768899441,
      "grad_norm": 0.11280254274606705,
      "learning_rate": 8.969616904321026e-05,
      "loss": 0.051,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007796379213687032,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.655071258544922,
      "state_query_norm": 1146.0945922851563,
      "state_relation_loss": 0.0007796379213687032,
      "step": 7430,
      "student_distance_mean": 1537.967138671875,
      "teacher_distance_mean": 1.1622326493263244,
      "weighted_auxiliary_loss": 0.0007796379213687032,
      "weighted_relation_loss": 0.0007796379213687032
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.95977549111319,
      "flow_matching_loss": 0.04993008859455585,
      "grad_norm": 0.1407899260520935,
      "learning_rate": 8.966263362908998e-05,
      "loss": 0.0507,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007573712675366551,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.403904914855957,
      "state_query_norm": 1140.99892578125,
      "state_relation_loss": 0.0007573712675366551,
      "step": 7440,
      "student_distance_mean": 1525.087841796875,
      "teacher_distance_mean": 1.1508483052253724,
      "weighted_auxiliary_loss": 0.0007573712675366551,
      "weighted_relation_loss": 0.0007573712675366551
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.969130028063611,
      "flow_matching_loss": 0.05253262743353844,
      "grad_norm": 0.1289072334766388,
      "learning_rate": 8.962905002112989e-05,
      "loss": 0.0533,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007926720310933888,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.623559761047364,
      "state_query_norm": 1155.3873046875,
      "state_relation_loss": 0.0007926720310933888,
      "step": 7450,
      "student_distance_mean": 1538.4037963867188,
      "teacher_distance_mean": 1.1518338799476624,
      "weighted_auxiliary_loss": 0.0007926720310933888,
      "weighted_relation_loss": 0.0007926720310933888
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.978484565014032,
      "flow_matching_loss": 0.04962190426886082,
      "grad_norm": 0.10556233674287796,
      "learning_rate": 8.959541826013724e-05,
      "loss": 0.0504,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007863137871026993,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.523157119750977,
      "state_query_norm": 1148.12236328125,
      "state_relation_loss": 0.0007863137871026993,
      "step": 7460,
      "student_distance_mean": 1532.5935913085937,
      "teacher_distance_mean": 1.1405160546302795,
      "weighted_auxiliary_loss": 0.0007863137871026993,
      "weighted_relation_loss": 0.0007863137871026993
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.987839101964453,
      "flow_matching_loss": 0.05010876879096031,
      "grad_norm": 0.1576019674539566,
      "learning_rate": 8.956173838697781e-05,
      "loss": 0.0509,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007796032237820327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.650955200195312,
      "state_query_norm": 1143.7647705078125,
      "state_relation_loss": 0.0007796032237820327,
      "step": 7470,
      "student_distance_mean": 1534.9620727539063,
      "teacher_distance_mean": 1.1442965626716615,
      "weighted_auxiliary_loss": 0.0007796032237820327,
      "weighted_relation_loss": 0.0007796032237820327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 6.997193638914874,
      "flow_matching_loss": 0.05158641189336777,
      "grad_norm": 0.2038889229297638,
      "learning_rate": 8.952801044257581e-05,
      "loss": 0.0524,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008046987699344754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.825044441223145,
      "state_query_norm": 1156.2558471679688,
      "state_relation_loss": 0.0008046987699344754,
      "step": 7480,
      "student_distance_mean": 1547.2404541015626,
      "teacher_distance_mean": 1.1547141313552856,
      "weighted_auxiliary_loss": 0.0008046987699344754,
      "weighted_relation_loss": 0.0008046987699344754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.006548175865294,
      "flow_matching_loss": 0.0521313015371561,
      "grad_norm": 0.14516372978687286,
      "learning_rate": 8.949423446791387e-05,
      "loss": 0.053,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0008333578472957015,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 27.416086769104005,
      "state_query_norm": 1134.3488525390626,
      "state_relation_loss": 0.0008333578472957015,
      "step": 7490,
      "student_distance_mean": 1526.6992309570312,
      "teacher_distance_mean": 1.1453593730926515,
      "weighted_auxiliary_loss": 0.0008333578472957015,
      "weighted_relation_loss": 0.0008333578472957015
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.015902712815715,
      "flow_matching_loss": 0.04692152813076973,
      "grad_norm": 0.18692536652088165,
      "learning_rate": 8.9460410504033e-05,
      "loss": 0.0477,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007601128425449132,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.766261863708497,
      "state_query_norm": 1149.93681640625,
      "state_relation_loss": 0.0007601128425449132,
      "step": 7500,
      "student_distance_mean": 1542.4004028320312,
      "teacher_distance_mean": 1.156622552871704,
      "weighted_auxiliary_loss": 0.0007601128425449132,
      "weighted_relation_loss": 0.0007601128425449132
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.0252572497661365,
      "flow_matching_loss": 0.049546368047595025,
      "grad_norm": 0.13414685428142548,
      "learning_rate": 8.942653859203248e-05,
      "loss": 0.0503,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007844153675250709,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.614675521850586,
      "state_query_norm": 1145.3830078125,
      "state_relation_loss": 0.0007844153675250709,
      "step": 7510,
      "student_distance_mean": 1538.156103515625,
      "teacher_distance_mean": 1.1412006139755249,
      "weighted_auxiliary_loss": 0.0007844153675250709,
      "weighted_relation_loss": 0.0007844153675250709
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.034611786716558,
      "flow_matching_loss": 0.04845043197274208,
      "grad_norm": 0.17539191246032715,
      "learning_rate": 8.93926187730699e-05,
      "loss": 0.0493,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0008039083681069314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.49437313079834,
      "state_query_norm": 1135.0001220703125,
      "state_relation_loss": 0.0008039083681069314,
      "step": 7520,
      "student_distance_mean": 1525.538671875,
      "teacher_distance_mean": 1.1380829453468322,
      "weighted_auxiliary_loss": 0.0008039083681069314,
      "weighted_relation_loss": 0.0008039083681069314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.043966323666979,
      "flow_matching_loss": 0.050808051228523256,
      "grad_norm": 0.1326930820941925,
      "learning_rate": 8.935865108836102e-05,
      "loss": 0.0516,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007565188396256417,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.19144401550293,
      "state_query_norm": 1127.4996948242188,
      "state_relation_loss": 0.0007565188396256417,
      "step": 7530,
      "student_distance_mean": 1512.72119140625,
      "teacher_distance_mean": 1.1337053179740906,
      "weighted_auxiliary_loss": 0.0007565188396256417,
      "weighted_relation_loss": 0.0007565188396256417
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.0533208606174,
      "flow_matching_loss": 0.0498900443315506,
      "grad_norm": 0.16345994174480438,
      "learning_rate": 8.932463557917981e-05,
      "loss": 0.0506,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007521665829699486,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.863855361938477,
      "state_query_norm": 1152.3214721679688,
      "state_relation_loss": 0.0007521665829699486,
      "step": 7540,
      "student_distance_mean": 1547.8966064453125,
      "teacher_distance_mean": 1.1587033987045288,
      "weighted_auxiliary_loss": 0.0007521665829699486,
      "weighted_relation_loss": 0.0007521665829699486
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.06267539756782,
      "flow_matching_loss": 0.04901385083794594,
      "grad_norm": 0.14108294248580933,
      "learning_rate": 8.929057228685829e-05,
      "loss": 0.0498,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007709231751505286,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.935148429870605,
      "state_query_norm": 1159.2183837890625,
      "state_relation_loss": 0.0007709231751505286,
      "step": 7550,
      "student_distance_mean": 1553.7142822265625,
      "teacher_distance_mean": 1.1552353739738463,
      "weighted_auxiliary_loss": 0.0007709231751505286,
      "weighted_relation_loss": 0.0007709231751505286
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.072029934518241,
      "flow_matching_loss": 0.048254984244704244,
      "grad_norm": 0.14481166005134583,
      "learning_rate": 8.925646125278658e-05,
      "loss": 0.049,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007638846582267433,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.85441608428955,
      "state_query_norm": 1155.4212646484375,
      "state_relation_loss": 0.0007638846582267433,
      "step": 7560,
      "student_distance_mean": 1547.8510375976562,
      "teacher_distance_mean": 1.1538455486297607,
      "weighted_auxiliary_loss": 0.0007638846582267433,
      "weighted_relation_loss": 0.0007638846582267433
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.081384471468662,
      "flow_matching_loss": 0.05066346079111099,
      "grad_norm": 0.1545996069908142,
      "learning_rate": 8.92223025184128e-05,
      "loss": 0.0514,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007420470006763935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.76728801727295,
      "state_query_norm": 1155.1858520507812,
      "state_relation_loss": 0.0007420470006763935,
      "step": 7570,
      "student_distance_mean": 1545.4718627929688,
      "teacher_distance_mean": 1.1463476538658142,
      "weighted_auxiliary_loss": 0.0007420470006763935,
      "weighted_relation_loss": 0.0007420470006763935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.090739008419083,
      "flow_matching_loss": 0.049073098599910735,
      "grad_norm": 0.11874005198478699,
      "learning_rate": 8.918809612524303e-05,
      "loss": 0.0499,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007898372597992421,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.86024055480957,
      "state_query_norm": 1146.4521362304688,
      "state_relation_loss": 0.0007898372597992421,
      "step": 7580,
      "student_distance_mean": 1546.3418334960938,
      "teacher_distance_mean": 1.1506149053573609,
      "weighted_auxiliary_loss": 0.0007898372597992421,
      "weighted_relation_loss": 0.0007898372597992421
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.100093545369504,
      "flow_matching_loss": 0.04969823770225048,
      "grad_norm": 0.18561366200447083,
      "learning_rate": 8.91538421148413e-05,
      "loss": 0.0505,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007974698964972049,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.585749053955077,
      "state_query_norm": 1146.3548706054687,
      "state_relation_loss": 0.0007974698964972049,
      "step": 7590,
      "student_distance_mean": 1534.0303588867187,
      "teacher_distance_mean": 1.1412864327430725,
      "weighted_auxiliary_loss": 0.0007974698964972049,
      "weighted_relation_loss": 0.0007974698964972049
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.1094480823199255,
      "flow_matching_loss": 0.04973345771431923,
      "grad_norm": 0.13771294057369232,
      "learning_rate": 8.91195405288294e-05,
      "loss": 0.0505,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007574463612399996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.86193714141846,
      "state_query_norm": 1155.780908203125,
      "state_relation_loss": 0.0007574463612399996,
      "step": 7600,
      "student_distance_mean": 1550.89775390625,
      "teacher_distance_mean": 1.1597257494926452,
      "weighted_auxiliary_loss": 0.0007574463612399996,
      "weighted_relation_loss": 0.0007574463612399996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.118802619270346,
      "flow_matching_loss": 0.04975742399692536,
      "grad_norm": 0.12314369529485703,
      "learning_rate": 8.908519140888705e-05,
      "loss": 0.0505,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007504246197640896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.94643955230713,
      "state_query_norm": 1158.3306762695313,
      "state_relation_loss": 0.0007504246197640896,
      "step": 7610,
      "student_distance_mean": 1555.436279296875,
      "teacher_distance_mean": 1.1582237124443053,
      "weighted_auxiliary_loss": 0.0007504246197640896,
      "weighted_relation_loss": 0.0007504246197640896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.128157156220767,
      "flow_matching_loss": 0.04770250245928764,
      "grad_norm": 0.1689693182706833,
      "learning_rate": 8.905079479675164e-05,
      "loss": 0.0485,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007901963486801833,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.936686706542968,
      "state_query_norm": 1153.3175537109375,
      "state_relation_loss": 0.0007901963486801833,
      "step": 7620,
      "student_distance_mean": 1553.699365234375,
      "teacher_distance_mean": 1.1556433200836183,
      "weighted_auxiliary_loss": 0.0007901963486801833,
      "weighted_relation_loss": 0.0007901963486801833
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.137511693171188,
      "flow_matching_loss": 0.04792879037559032,
      "grad_norm": 0.145249143242836,
      "learning_rate": 8.901635073421831e-05,
      "loss": 0.0487,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007886424951720983,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.223401260375976,
      "state_query_norm": 1157.7010620117187,
      "state_relation_loss": 0.0007886424951720983,
      "step": 7630,
      "student_distance_mean": 1566.8249267578126,
      "teacher_distance_mean": 1.1644098162651062,
      "weighted_auxiliary_loss": 0.0007886424951720983,
      "weighted_relation_loss": 0.0007886424951720983
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.146866230121609,
      "flow_matching_loss": 0.05008870847523213,
      "grad_norm": 0.13912837207317352,
      "learning_rate": 8.898185926313983e-05,
      "loss": 0.0509,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000770534633193165,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.848622512817382,
      "state_query_norm": 1153.0811889648437,
      "state_relation_loss": 0.000770534633193165,
      "step": 7640,
      "student_distance_mean": 1549.3487915039063,
      "teacher_distance_mean": 1.1425512313842774,
      "weighted_auxiliary_loss": 0.000770534633193165,
      "weighted_relation_loss": 0.000770534633193165
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.15622076707203,
      "flow_matching_loss": 0.047335998713970186,
      "grad_norm": 0.11145764589309692,
      "learning_rate": 8.894732042542659e-05,
      "loss": 0.0481,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007637434173375368,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.867947387695313,
      "state_query_norm": 1151.9797485351562,
      "state_relation_loss": 0.0007637434173375368,
      "step": 7650,
      "student_distance_mean": 1548.8202880859376,
      "teacher_distance_mean": 1.1450827598571778,
      "weighted_auxiliary_loss": 0.0007637434173375368,
      "weighted_relation_loss": 0.0007637434173375368
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.165575304022451,
      "flow_matching_loss": 0.049068994075059894,
      "grad_norm": 0.11499954760074615,
      "learning_rate": 8.891273426304654e-05,
      "loss": 0.0498,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007697652676142752,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.131257820129395,
      "state_query_norm": 1168.6351196289063,
      "state_relation_loss": 0.0007697652676142752,
      "step": 7660,
      "student_distance_mean": 1565.1001831054687,
      "teacher_distance_mean": 1.1532617807388306,
      "weighted_auxiliary_loss": 0.0007697652676142752,
      "weighted_relation_loss": 0.0007697652676142752
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.1749298409728715,
      "flow_matching_loss": 0.04599080681800842,
      "grad_norm": 0.1432107836008072,
      "learning_rate": 8.887810081802515e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007827353081665933,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.85662136077881,
      "state_query_norm": 1157.2162353515625,
      "state_relation_loss": 0.0007827353081665933,
      "step": 7670,
      "student_distance_mean": 1546.904296875,
      "teacher_distance_mean": 1.1394083976745606,
      "weighted_auxiliary_loss": 0.0007827353081665933,
      "weighted_relation_loss": 0.0007827353081665933
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.1842843779232926,
      "flow_matching_loss": 0.048825708031654355,
      "grad_norm": 0.13055793941020966,
      "learning_rate": 8.884342013244529e-05,
      "loss": 0.0496,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007473091944120824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.01428508758545,
      "state_query_norm": 1158.3463256835937,
      "state_relation_loss": 0.0007473091944120824,
      "step": 7680,
      "student_distance_mean": 1555.1633544921874,
      "teacher_distance_mean": 1.1497917413711547,
      "weighted_auxiliary_loss": 0.0007473091944120824,
      "weighted_relation_loss": 0.0007473091944120824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.193638914873714,
      "flow_matching_loss": 0.05082498379051685,
      "grad_norm": 0.17859503626823425,
      "learning_rate": 8.880869224844725e-05,
      "loss": 0.0516,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007274823961779475,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.021845626831055,
      "state_query_norm": 1153.9455810546874,
      "state_relation_loss": 0.0007274823961779475,
      "step": 7690,
      "student_distance_mean": 1553.1921020507812,
      "teacher_distance_mean": 1.1539485216140748,
      "weighted_auxiliary_loss": 0.0007274823961779475,
      "weighted_relation_loss": 0.0007274823961779475
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.202993451824135,
      "flow_matching_loss": 0.0502738781273365,
      "grad_norm": 0.15348976850509644,
      "learning_rate": 8.877391720822875e-05,
      "loss": 0.051,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007607055129483342,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.93851947784424,
      "state_query_norm": 1163.4998168945312,
      "state_relation_loss": 0.0007607055129483342,
      "step": 7700,
      "student_distance_mean": 1555.7356079101562,
      "teacher_distance_mean": 1.152375543117523,
      "weighted_auxiliary_loss": 0.0007607055129483342,
      "weighted_relation_loss": 0.0007607055129483342
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.212347988774556,
      "flow_matching_loss": 0.048909717798233034,
      "grad_norm": 0.11986485868692398,
      "learning_rate": 8.873909505404468e-05,
      "loss": 0.0497,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007451478042639792,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.123132514953614,
      "state_query_norm": 1163.3934692382813,
      "state_relation_loss": 0.0007451478042639792,
      "step": 7710,
      "student_distance_mean": 1564.413427734375,
      "teacher_distance_mean": 1.1498121261596679,
      "weighted_auxiliary_loss": 0.0007451478042639792,
      "weighted_relation_loss": 0.0007451478042639792
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.221702525724977,
      "flow_matching_loss": 0.04797378517687321,
      "grad_norm": 0.1237185150384903,
      "learning_rate": 8.870422582820726e-05,
      "loss": 0.0488,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007811465882696211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.304250144958495,
      "state_query_norm": 1171.1558471679687,
      "state_relation_loss": 0.0007811465882696211,
      "step": 7720,
      "student_distance_mean": 1569.3705688476562,
      "teacher_distance_mean": 1.1540920853614807,
      "weighted_auxiliary_loss": 0.0007811465882696211,
      "weighted_relation_loss": 0.0007811465882696211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.231057062675397,
      "flow_matching_loss": 0.04927553720772267,
      "grad_norm": 0.13057702779769897,
      "learning_rate": 8.866930957308588e-05,
      "loss": 0.05,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000761904229875654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.126053047180175,
      "state_query_norm": 1161.745703125,
      "state_relation_loss": 0.000761904229875654,
      "step": 7730,
      "student_distance_mean": 1563.7599731445312,
      "teacher_distance_mean": 1.1554493188858033,
      "weighted_auxiliary_loss": 0.000761904229875654,
      "weighted_relation_loss": 0.000761904229875654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.240411599625818,
      "flow_matching_loss": 0.04807343259453774,
      "grad_norm": 0.14416353404521942,
      "learning_rate": 8.86343463311071e-05,
      "loss": 0.0488,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00076455173548311,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.680313873291016,
      "state_query_norm": 1138.8487426757813,
      "state_relation_loss": 0.00076455173548311,
      "step": 7740,
      "student_distance_mean": 1533.7740356445313,
      "teacher_distance_mean": 1.138894760608673,
      "weighted_auxiliary_loss": 0.00076455173548311,
      "weighted_relation_loss": 0.00076455173548311
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.249766136576239,
      "flow_matching_loss": 0.0477653443813324,
      "grad_norm": 0.112228162586689,
      "learning_rate": 8.859933614475452e-05,
      "loss": 0.0485,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007617100374773145,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.93837547302246,
      "state_query_norm": 1164.9254638671875,
      "state_relation_loss": 0.0007617100374773145,
      "step": 7750,
      "student_distance_mean": 1555.8750610351562,
      "teacher_distance_mean": 1.1560649037361146,
      "weighted_auxiliary_loss": 0.0007617100374773145,
      "weighted_relation_loss": 0.0007617100374773145
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.2591206735266605,
      "flow_matching_loss": 0.04800588041543961,
      "grad_norm": 0.13473612070083618,
      "learning_rate": 8.85642790565689e-05,
      "loss": 0.0488,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007496949110645801,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.14143486022949,
      "state_query_norm": 1160.3108154296874,
      "state_relation_loss": 0.0007496949110645801,
      "step": 7760,
      "student_distance_mean": 1561.22763671875,
      "teacher_distance_mean": 1.1488704204559326,
      "weighted_auxiliary_loss": 0.0007496949110645801,
      "weighted_relation_loss": 0.0007496949110645801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.268475210477082,
      "flow_matching_loss": 0.04796276725828648,
      "grad_norm": 0.13891617953777313,
      "learning_rate": 8.852917510914782e-05,
      "loss": 0.0487,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000738112727412954,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.942287063598634,
      "state_query_norm": 1158.5205200195312,
      "state_relation_loss": 0.000738112727412954,
      "step": 7770,
      "student_distance_mean": 1551.0873046875,
      "teacher_distance_mean": 1.1442638635635376,
      "weighted_auxiliary_loss": 0.000738112727412954,
      "weighted_relation_loss": 0.000738112727412954
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.277829747427503,
      "flow_matching_loss": 0.05032388232648373,
      "grad_norm": 0.1289941966533661,
      "learning_rate": 8.84940243451459e-05,
      "loss": 0.0511,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007482176646590233,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.008922386169434,
      "state_query_norm": 1166.5550903320313,
      "state_relation_loss": 0.0007482176646590233,
      "step": 7780,
      "student_distance_mean": 1559.8906372070312,
      "teacher_distance_mean": 1.155056357383728,
      "weighted_auxiliary_loss": 0.0007482176646590233,
      "weighted_relation_loss": 0.0007482176646590233
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.287184284377924,
      "flow_matching_loss": 0.05183723866939545,
      "grad_norm": 0.14326873421669006,
      "learning_rate": 8.845882680727469e-05,
      "loss": 0.0526,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007504391309339553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.870098495483397,
      "state_query_norm": 1154.2218383789063,
      "state_relation_loss": 0.0007504391309339553,
      "step": 7790,
      "student_distance_mean": 1549.4648193359376,
      "teacher_distance_mean": 1.153936505317688,
      "weighted_auxiliary_loss": 0.0007504391309339553,
      "weighted_relation_loss": 0.0007504391309339553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.296538821328344,
      "flow_matching_loss": 0.04947849847376347,
      "grad_norm": 0.17892394959926605,
      "learning_rate": 8.842358253830246e-05,
      "loss": 0.0502,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007420152192935348,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.129657173156737,
      "state_query_norm": 1163.3712280273437,
      "state_relation_loss": 0.0007420152192935348,
      "step": 7800,
      "student_distance_mean": 1563.8091064453124,
      "teacher_distance_mean": 1.156695258617401,
      "weighted_auxiliary_loss": 0.0007420152192935348,
      "weighted_relation_loss": 0.0007420152192935348
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.305893358278765,
      "flow_matching_loss": 0.05165784880518913,
      "grad_norm": 0.13741743564605713,
      "learning_rate": 8.838829158105434e-05,
      "loss": 0.0524,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007222918327897787,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.006228828430174,
      "state_query_norm": 1160.3194946289063,
      "state_relation_loss": 0.0007222918327897787,
      "step": 7810,
      "student_distance_mean": 1557.635107421875,
      "teacher_distance_mean": 1.1452922701835633,
      "weighted_auxiliary_loss": 0.0007222918327897787,
      "weighted_relation_loss": 0.0007222918327897787
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.315247895229186,
      "flow_matching_loss": 0.04847164712846279,
      "grad_norm": 0.1370110660791397,
      "learning_rate": 8.835295397841217e-05,
      "loss": 0.0492,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007338572002481669,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.457568359375,
      "state_query_norm": 1171.71884765625,
      "state_relation_loss": 0.0007338572002481669,
      "step": 7820,
      "student_distance_mean": 1581.0157470703125,
      "teacher_distance_mean": 1.1614913940429688,
      "weighted_auxiliary_loss": 0.0007338572002481669,
      "weighted_relation_loss": 0.0007338572002481669
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.324602432179607,
      "flow_matching_loss": 0.04716640301048756,
      "grad_norm": 0.12350431084632874,
      "learning_rate": 8.831756977331447e-05,
      "loss": 0.0479,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000715754204429686,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.326322555541992,
      "state_query_norm": 1180.6494995117187,
      "state_relation_loss": 0.000715754204429686,
      "step": 7830,
      "student_distance_mean": 1577.2805786132812,
      "teacher_distance_mean": 1.1611127138137818,
      "weighted_auxiliary_loss": 0.000715754204429686,
      "weighted_relation_loss": 0.000715754204429686
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.333956969130028,
      "flow_matching_loss": 0.049824358522892,
      "grad_norm": 0.11678425222635269,
      "learning_rate": 8.828213900875638e-05,
      "loss": 0.0506,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007602072786539793,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.09724712371826,
      "state_query_norm": 1161.0596557617187,
      "state_relation_loss": 0.0007602072786539793,
      "step": 7840,
      "student_distance_mean": 1559.543994140625,
      "teacher_distance_mean": 1.1430541276931763,
      "weighted_auxiliary_loss": 0.0007602072786539793,
      "weighted_relation_loss": 0.0007602072786539793
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.343311506080449,
      "flow_matching_loss": 0.04711429700255394,
      "grad_norm": 0.13127751648426056,
      "learning_rate": 8.824666172778964e-05,
      "loss": 0.0478,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006829094782005996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.904328536987304,
      "state_query_norm": 1152.4687866210938,
      "state_relation_loss": 0.0006829094782005996,
      "step": 7850,
      "student_distance_mean": 1551.6392456054687,
      "teacher_distance_mean": 1.1345389246940614,
      "weighted_auxiliary_loss": 0.0006829094782005996,
      "weighted_relation_loss": 0.0006829094782005996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.35266604303087,
      "flow_matching_loss": 0.049357809126377106,
      "grad_norm": 0.1490107774734497,
      "learning_rate": 8.821113797352245e-05,
      "loss": 0.0501,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00077009322703816,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.23026123046875,
      "state_query_norm": 1170.2903564453125,
      "state_relation_loss": 0.00077009322703816,
      "step": 7860,
      "student_distance_mean": 1568.4629516601562,
      "teacher_distance_mean": 1.1544017314910888,
      "weighted_auxiliary_loss": 0.00077009322703816,
      "weighted_relation_loss": 0.00077009322703816
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.362020579981291,
      "flow_matching_loss": 0.04726620018482208,
      "grad_norm": 0.16940200328826904,
      "learning_rate": 8.817556778911956e-05,
      "loss": 0.048,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007018181611783802,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.621755599975586,
      "state_query_norm": 1184.9838500976562,
      "state_relation_loss": 0.0007018181611783802,
      "step": 7870,
      "student_distance_mean": 1592.0874633789062,
      "teacher_distance_mean": 1.1659305810928344,
      "weighted_auxiliary_loss": 0.0007018181611783802,
      "weighted_relation_loss": 0.0007018181611783802
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.371375116931712,
      "flow_matching_loss": 0.04560569263994694,
      "grad_norm": 0.12085536867380142,
      "learning_rate": 8.81399512178021e-05,
      "loss": 0.0464,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007752272940706462,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.274363899230956,
      "state_query_norm": 1168.67568359375,
      "state_relation_loss": 0.0007752272940706462,
      "step": 7880,
      "student_distance_mean": 1567.2984985351563,
      "teacher_distance_mean": 1.143255603313446,
      "weighted_auxiliary_loss": 0.0007752272940706462,
      "weighted_relation_loss": 0.0007752272940706462
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.380729653882133,
      "flow_matching_loss": 0.04701472967863083,
      "grad_norm": 0.12235661596059799,
      "learning_rate": 8.810428830284752e-05,
      "loss": 0.0478,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007375820947345346,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.506939506530763,
      "state_query_norm": 1180.4111206054688,
      "state_relation_loss": 0.0007375820947345346,
      "step": 7890,
      "student_distance_mean": 1580.3777709960937,
      "teacher_distance_mean": 1.1512137055397034,
      "weighted_auxiliary_loss": 0.0007375820947345346,
      "weighted_relation_loss": 0.0007375820947345346
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.390084190832554,
      "flow_matching_loss": 0.04842742495238781,
      "grad_norm": 0.12397646903991699,
      "learning_rate": 8.806857908758967e-05,
      "loss": 0.0492,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007502281805500388,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.01457099914551,
      "state_query_norm": 1167.8736938476563,
      "state_relation_loss": 0.0007502281805500388,
      "step": 7900,
      "student_distance_mean": 1561.0190551757812,
      "teacher_distance_mean": 1.1417987108230592,
      "weighted_auxiliary_loss": 0.0007502281805500388,
      "weighted_relation_loss": 0.0007502281805500388
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.399438727782975,
      "flow_matching_loss": 0.04728815518319607,
      "grad_norm": 0.1647876501083374,
      "learning_rate": 8.80328236154186e-05,
      "loss": 0.048,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007367155922111124,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.35532112121582,
      "state_query_norm": 1173.6582153320312,
      "state_relation_loss": 0.0007367155922111124,
      "step": 7910,
      "student_distance_mean": 1574.1425537109376,
      "teacher_distance_mean": 1.1544857263565063,
      "weighted_auxiliary_loss": 0.0007367155922111124,
      "weighted_relation_loss": 0.0007367155922111124
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.408793264733395,
      "flow_matching_loss": 0.0468885637819767,
      "grad_norm": 0.13399238884449005,
      "learning_rate": 8.799702192978056e-05,
      "loss": 0.0476,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000761122687254101,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.58064498901367,
      "state_query_norm": 1185.3832153320313,
      "state_relation_loss": 0.000761122687254101,
      "step": 7920,
      "student_distance_mean": 1589.031591796875,
      "teacher_distance_mean": 1.1556477427482605,
      "weighted_auxiliary_loss": 0.000761122687254101,
      "weighted_relation_loss": 0.000761122687254101
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.4181478016838165,
      "flow_matching_loss": 0.04832918047904968,
      "grad_norm": 0.16075319051742554,
      "learning_rate": 8.7961174074178e-05,
      "loss": 0.049,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000712432194268331,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.597032737731933,
      "state_query_norm": 1172.9951171875,
      "state_relation_loss": 0.000712432194268331,
      "step": 7930,
      "student_distance_mean": 1584.375244140625,
      "teacher_distance_mean": 1.1609779953956605,
      "weighted_auxiliary_loss": 0.000712432194268331,
      "weighted_relation_loss": 0.000712432194268331
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.427502338634238,
      "flow_matching_loss": 0.04723826423287392,
      "grad_norm": 0.13767452538013458,
      "learning_rate": 8.792528009216942e-05,
      "loss": 0.048,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007307680323719978,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.848383140563964,
      "state_query_norm": 1194.9185302734375,
      "state_relation_loss": 0.0007307680323719978,
      "step": 7940,
      "student_distance_mean": 1601.4799438476562,
      "teacher_distance_mean": 1.173900830745697,
      "weighted_auxiliary_loss": 0.0007307680323719978,
      "weighted_relation_loss": 0.0007307680323719978
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.436856875584659,
      "flow_matching_loss": 0.05035501904785633,
      "grad_norm": 0.11873073130846024,
      "learning_rate": 8.788934002736943e-05,
      "loss": 0.0511,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007538710488006473,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 27.872670936584473,
      "state_query_norm": 1147.4078735351563,
      "state_relation_loss": 0.0007538710488006473,
      "step": 7950,
      "student_distance_mean": 1545.7779418945313,
      "teacher_distance_mean": 1.1332378387451172,
      "weighted_auxiliary_loss": 0.0007538710488006473,
      "weighted_relation_loss": 0.0007538710488006473
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.44621141253508,
      "flow_matching_loss": 0.048720003291964534,
      "grad_norm": 0.12908442318439484,
      "learning_rate": 8.785335392344857e-05,
      "loss": 0.0495,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007424200535751879,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.51763172149658,
      "state_query_norm": 1179.2838256835937,
      "state_relation_loss": 0.0007424200535751879,
      "step": 7960,
      "student_distance_mean": 1585.5762451171875,
      "teacher_distance_mean": 1.1618157744407653,
      "weighted_auxiliary_loss": 0.0007424200535751879,
      "weighted_relation_loss": 0.0007424200535751879
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.4555659494855,
      "flow_matching_loss": 0.05085556358098984,
      "grad_norm": 0.15011024475097656,
      "learning_rate": 8.781732182413336e-05,
      "loss": 0.0516,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007618999050464481,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.506147384643555,
      "state_query_norm": 1171.9888061523438,
      "state_relation_loss": 0.0007618999050464481,
      "step": 7970,
      "student_distance_mean": 1581.0873046875,
      "teacher_distance_mean": 1.14929541349411,
      "weighted_auxiliary_loss": 0.0007618999050464481,
      "weighted_relation_loss": 0.0007618999050464481
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.464920486435921,
      "flow_matching_loss": 0.04974446706473827,
      "grad_norm": 0.1292283833026886,
      "learning_rate": 8.778124377320618e-05,
      "loss": 0.0505,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007627116399817168,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.509806442260743,
      "state_query_norm": 1184.21650390625,
      "state_relation_loss": 0.0007627116399817168,
      "step": 7980,
      "student_distance_mean": 1584.9879028320313,
      "teacher_distance_mean": 1.1534072041511536,
      "weighted_auxiliary_loss": 0.0007627116399817168,
      "weighted_relation_loss": 0.0007627116399817168
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.474275023386342,
      "flow_matching_loss": 0.0485960952937603,
      "grad_norm": 0.14455577731132507,
      "learning_rate": 8.774511981450529e-05,
      "loss": 0.0493,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007444027869496495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.963877296447755,
      "state_query_norm": 1195.3528930664063,
      "state_relation_loss": 0.0007444027869496495,
      "step": 7990,
      "student_distance_mean": 1608.2637451171875,
      "teacher_distance_mean": 1.1729727745056153,
      "weighted_auxiliary_loss": 0.0007444027869496495,
      "weighted_relation_loss": 0.0007444027869496495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.483629560336763,
      "flow_matching_loss": 0.051304514333605764,
      "grad_norm": 0.18832442164421082,
      "learning_rate": 8.770894999192467e-05,
      "loss": 0.052,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007338560419157148,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.84093952178955,
      "state_query_norm": 1194.7860717773438,
      "state_relation_loss": 0.0007338560419157148,
      "step": 8000,
      "student_distance_mean": 1603.4834594726562,
      "teacher_distance_mean": 1.1570884227752685,
      "weighted_auxiliary_loss": 0.0007338560419157148,
      "weighted_relation_loss": 0.0007338560419157148
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.492984097287184,
      "flow_matching_loss": 0.04893839992582798,
      "grad_norm": 0.15630920231342316,
      "learning_rate": 8.767273434941412e-05,
      "loss": 0.0497,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000757391972001642,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.70643901824951,
      "state_query_norm": 1194.3105346679688,
      "state_relation_loss": 0.000757391972001642,
      "step": 8010,
      "student_distance_mean": 1593.807470703125,
      "teacher_distance_mean": 1.1508715867996215,
      "weighted_auxiliary_loss": 0.000757391972001642,
      "weighted_relation_loss": 0.000757391972001642
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.5023386342376055,
      "flow_matching_loss": 0.0504963468760252,
      "grad_norm": 0.13329152762889862,
      "learning_rate": 8.7636472930979e-05,
      "loss": 0.0513,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007879379030782729,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.93271369934082,
      "state_query_norm": 1200.5692138671875,
      "state_relation_loss": 0.0007879379030782729,
      "step": 8020,
      "student_distance_mean": 1606.8753540039063,
      "teacher_distance_mean": 1.1677302241325378,
      "weighted_auxiliary_loss": 0.0007879379030782729,
      "weighted_relation_loss": 0.0007879379030782729
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.511693171188027,
      "flow_matching_loss": 0.04824888408184051,
      "grad_norm": 0.12880438566207886,
      "learning_rate": 8.76001657806804e-05,
      "loss": 0.049,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007608968764543534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.395308685302734,
      "state_query_norm": 1165.8171875,
      "state_relation_loss": 0.0007608968764543534,
      "step": 8030,
      "student_distance_mean": 1574.9645874023438,
      "teacher_distance_mean": 1.140112578868866,
      "weighted_auxiliary_loss": 0.0007608968764543534,
      "weighted_relation_loss": 0.0007608968764543534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.521047708138447,
      "flow_matching_loss": 0.049384049326181415,
      "grad_norm": 0.17048227787017822,
      "learning_rate": 8.756381294263491e-05,
      "loss": 0.0501,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007497772574424743,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.68793315887451,
      "state_query_norm": 1184.504833984375,
      "state_relation_loss": 0.0007497772574424743,
      "step": 8040,
      "student_distance_mean": 1592.4205200195313,
      "teacher_distance_mean": 1.1580359101295472,
      "weighted_auxiliary_loss": 0.0007497772574424743,
      "weighted_relation_loss": 0.0007497772574424743
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.530402245088868,
      "flow_matching_loss": 0.04839806109666824,
      "grad_norm": 0.17258578538894653,
      "learning_rate": 8.752741446101464e-05,
      "loss": 0.0492,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007896720198914409,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.67513542175293,
      "state_query_norm": 1181.3316162109375,
      "state_relation_loss": 0.0007896720198914409,
      "step": 8050,
      "student_distance_mean": 1591.0661010742188,
      "teacher_distance_mean": 1.1596700549125671,
      "weighted_auxiliary_loss": 0.0007896720198914409,
      "weighted_relation_loss": 0.0007896720198914409
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.539756782039289,
      "flow_matching_loss": 0.04743288345634937,
      "grad_norm": 0.20961718261241913,
      "learning_rate": 8.749097038004722e-05,
      "loss": 0.0482,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007213975652121008,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.734172058105468,
      "state_query_norm": 1181.8855834960937,
      "state_relation_loss": 0.0007213975652121008,
      "step": 8060,
      "student_distance_mean": 1596.4623901367188,
      "teacher_distance_mean": 1.156033742427826,
      "weighted_auxiliary_loss": 0.0007213975652121008,
      "weighted_relation_loss": 0.0007213975652121008
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.54911131898971,
      "flow_matching_loss": 0.04809533953666687,
      "grad_norm": 0.18321356177330017,
      "learning_rate": 8.745448074401561e-05,
      "loss": 0.0489,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007554428186267615,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.803008842468262,
      "state_query_norm": 1183.9197998046875,
      "state_relation_loss": 0.0007554428186267615,
      "step": 8070,
      "student_distance_mean": 1596.2710205078124,
      "teacher_distance_mean": 1.1596787810325622,
      "weighted_auxiliary_loss": 0.0007554428186267615,
      "weighted_relation_loss": 0.0007554428186267615
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.558465855940131,
      "flow_matching_loss": 0.05004945024847984,
      "grad_norm": 0.22547249495983124,
      "learning_rate": 8.741794559725818e-05,
      "loss": 0.0508,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007420914655085654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.78066463470459,
      "state_query_norm": 1186.0933959960937,
      "state_relation_loss": 0.0007420914655085654,
      "step": 8080,
      "student_distance_mean": 1597.2482788085938,
      "teacher_distance_mean": 1.1534748315811156,
      "weighted_auxiliary_loss": 0.0007420914655085654,
      "weighted_relation_loss": 0.0007420914655085654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.567820392890552,
      "flow_matching_loss": 0.050094505026936534,
      "grad_norm": 0.14113330841064453,
      "learning_rate": 8.738136498416856e-05,
      "loss": 0.0508,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007308740052394569,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.54603729248047,
      "state_query_norm": 1179.311962890625,
      "state_relation_loss": 0.0007308740052394569,
      "step": 8090,
      "student_distance_mean": 1585.9335083007813,
      "teacher_distance_mean": 1.137184476852417,
      "weighted_auxiliary_loss": 0.0007308740052394569,
      "weighted_relation_loss": 0.0007308740052394569
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.5771749298409725,
      "flow_matching_loss": 0.04901162832975388,
      "grad_norm": 0.11213424056768417,
      "learning_rate": 8.734473894919564e-05,
      "loss": 0.0498,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007581947138532996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.79541072845459,
      "state_query_norm": 1191.31220703125,
      "state_relation_loss": 0.0007581947138532996,
      "step": 8100,
      "student_distance_mean": 1595.7722900390625,
      "teacher_distance_mean": 1.1519330978393554,
      "weighted_auxiliary_loss": 0.0007581947138532996,
      "weighted_relation_loss": 0.0007581947138532996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.586529466791394,
      "flow_matching_loss": 0.048732516914606096,
      "grad_norm": 0.12974707782268524,
      "learning_rate": 8.730806753684353e-05,
      "loss": 0.0495,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007391940744128078,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.75209140777588,
      "state_query_norm": 1188.4638305664062,
      "state_relation_loss": 0.0007391940744128078,
      "step": 8110,
      "student_distance_mean": 1595.7804077148437,
      "teacher_distance_mean": 1.1491058945655823,
      "weighted_auxiliary_loss": 0.0007391940744128078,
      "weighted_relation_loss": 0.0007391940744128078
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.595884003741815,
      "flow_matching_loss": 0.04872421510517597,
      "grad_norm": 0.1987489014863968,
      "learning_rate": 8.727135079167144e-05,
      "loss": 0.0495,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007496413891203701,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.034572410583497,
      "state_query_norm": 1199.0878295898438,
      "state_relation_loss": 0.0007496413891203701,
      "step": 8120,
      "student_distance_mean": 1610.8467041015624,
      "teacher_distance_mean": 1.1593021631240845,
      "weighted_auxiliary_loss": 0.0007496413891203701,
      "weighted_relation_loss": 0.0007496413891203701
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.605238540692236,
      "flow_matching_loss": 0.04857559204101562,
      "grad_norm": 0.14106757938861847,
      "learning_rate": 8.723458875829368e-05,
      "loss": 0.0493,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007623721729032695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.48089599609375,
      "state_query_norm": 1177.9173095703125,
      "state_relation_loss": 0.0007623721729032695,
      "step": 8130,
      "student_distance_mean": 1580.3606811523437,
      "teacher_distance_mean": 1.139595639705658,
      "weighted_auxiliary_loss": 0.0007623721729032695,
      "weighted_relation_loss": 0.0007623721729032695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.614593077642657,
      "flow_matching_loss": 0.049950672313570976,
      "grad_norm": 0.1342075765132904,
      "learning_rate": 8.719778148137958e-05,
      "loss": 0.0507,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007202196342404932,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.087995719909667,
      "state_query_norm": 1197.4977661132812,
      "state_relation_loss": 0.0007202196342404932,
      "step": 8140,
      "student_distance_mean": 1614.3708862304688,
      "teacher_distance_mean": 1.1663396120071412,
      "weighted_auxiliary_loss": 0.0007202196342404932,
      "weighted_relation_loss": 0.0007202196342404932
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.623947614593078,
      "flow_matching_loss": 0.05036841332912445,
      "grad_norm": 0.1814807653427124,
      "learning_rate": 8.716092900565346e-05,
      "loss": 0.0511,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007129361678380519,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.527856826782227,
      "state_query_norm": 1212.0825561523438,
      "state_relation_loss": 0.0007129361678380519,
      "step": 8150,
      "student_distance_mean": 1637.5280883789062,
      "teacher_distance_mean": 1.176620399951935,
      "weighted_auxiliary_loss": 0.0007129361678380519,
      "weighted_relation_loss": 0.0007129361678380519
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.633302151543498,
      "flow_matching_loss": 0.04956393167376518,
      "grad_norm": 0.1373046636581421,
      "learning_rate": 8.712403137589456e-05,
      "loss": 0.0503,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007542446721345187,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.00810604095459,
      "state_query_norm": 1195.6701538085938,
      "state_relation_loss": 0.0007542446721345187,
      "step": 8160,
      "student_distance_mean": 1607.8619018554687,
      "teacher_distance_mean": 1.1552972316741943,
      "weighted_auxiliary_loss": 0.0007542446721345187,
      "weighted_relation_loss": 0.0007542446721345187
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.642656688493919,
      "flow_matching_loss": 0.044989919662475585,
      "grad_norm": 0.12587426602840424,
      "learning_rate": 8.708708863693697e-05,
      "loss": 0.0457,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007434672152157873,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.83796920776367,
      "state_query_norm": 1185.4739990234375,
      "state_relation_loss": 0.0007434672152157873,
      "step": 8170,
      "student_distance_mean": 1601.7666870117187,
      "teacher_distance_mean": 1.1436989903450012,
      "weighted_auxiliary_loss": 0.0007434672152157873,
      "weighted_relation_loss": 0.0007434672152157873
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.65201122544434,
      "flow_matching_loss": 0.04659094102680683,
      "grad_norm": 0.1289907544851303,
      "learning_rate": 8.70501008336696e-05,
      "loss": 0.0473,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007404350966680795,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.262326049804688,
      "state_query_norm": 1201.1842651367188,
      "state_relation_loss": 0.0007404350966680795,
      "step": 8180,
      "student_distance_mean": 1619.70107421875,
      "teacher_distance_mean": 1.1612870931625365,
      "weighted_auxiliary_loss": 0.0007404350966680795,
      "weighted_relation_loss": 0.0007404350966680795
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.6613657623947615,
      "flow_matching_loss": 0.04731418937444687,
      "grad_norm": 0.13599230349063873,
      "learning_rate": 8.701306801103611e-05,
      "loss": 0.048,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007076576934196055,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.267988204956055,
      "state_query_norm": 1215.5564331054688,
      "state_relation_loss": 0.0007076576934196055,
      "step": 8190,
      "student_distance_mean": 1625.6657592773438,
      "teacher_distance_mean": 1.1616599798202514,
      "weighted_auxiliary_loss": 0.0007076576934196055,
      "weighted_relation_loss": 0.0007076576934196055
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.670720299345183,
      "flow_matching_loss": 0.05124166049063206,
      "grad_norm": 0.12990400195121765,
      "learning_rate": 8.697599021403489e-05,
      "loss": 0.052,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007233622833155096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.057356071472167,
      "state_query_norm": 1190.271435546875,
      "state_relation_loss": 0.0007233622833155096,
      "step": 8200,
      "student_distance_mean": 1609.885888671875,
      "teacher_distance_mean": 1.1496419787406922,
      "weighted_auxiliary_loss": 0.0007233622833155096,
      "weighted_relation_loss": 0.0007233622833155096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.680074836295604,
      "flow_matching_loss": 0.04850235730409622,
      "grad_norm": 0.17668037116527557,
      "learning_rate": 8.693886748771896e-05,
      "loss": 0.0492,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007446867180988193,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.002720069885253,
      "state_query_norm": 1189.7129516601562,
      "state_relation_loss": 0.0007446867180988193,
      "step": 8210,
      "student_distance_mean": 1607.4335571289062,
      "teacher_distance_mean": 1.1483680367469788,
      "weighted_auxiliary_loss": 0.0007446867180988193,
      "weighted_relation_loss": 0.0007446867180988193
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.689429373246024,
      "flow_matching_loss": 0.04621488824486732,
      "grad_norm": 0.12510302662849426,
      "learning_rate": 8.690169987719593e-05,
      "loss": 0.047,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007364607765339315,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.04489688873291,
      "state_query_norm": 1189.7752563476563,
      "state_relation_loss": 0.0007364607765339315,
      "step": 8220,
      "student_distance_mean": 1608.7810791015625,
      "teacher_distance_mean": 1.1521748661994935,
      "weighted_auxiliary_loss": 0.0007364607765339315,
      "weighted_relation_loss": 0.0007364607765339315
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.698783910196445,
      "flow_matching_loss": 0.05115893185138702,
      "grad_norm": 0.15540769696235657,
      "learning_rate": 8.686448742762791e-05,
      "loss": 0.0519,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007867510779760778,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.021224975585938,
      "state_query_norm": 1193.3797607421875,
      "state_relation_loss": 0.0007867510779760778,
      "step": 8230,
      "student_distance_mean": 1610.120068359375,
      "teacher_distance_mean": 1.1606377601623534,
      "weighted_auxiliary_loss": 0.0007867510779760778,
      "weighted_relation_loss": 0.0007867510779760778
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.708138447146866,
      "flow_matching_loss": 0.04696136675775051,
      "grad_norm": 0.12132484465837479,
      "learning_rate": 8.682723018423161e-05,
      "loss": 0.0477,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007546171778813005,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.858472061157226,
      "state_query_norm": 1191.2328491210938,
      "state_relation_loss": 0.0007546171778813005,
      "step": 8240,
      "student_distance_mean": 1601.4047973632812,
      "teacher_distance_mean": 1.153060793876648,
      "weighted_auxiliary_loss": 0.0007546171778813005,
      "weighted_relation_loss": 0.0007546171778813005
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.717492984097287,
      "flow_matching_loss": 0.04763316847383976,
      "grad_norm": 0.15993988513946533,
      "learning_rate": 8.678992819227804e-05,
      "loss": 0.0484,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007410961494315416,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.160471534729005,
      "state_query_norm": 1202.9308837890626,
      "state_relation_loss": 0.0007410961494315416,
      "step": 8250,
      "student_distance_mean": 1619.5310668945312,
      "teacher_distance_mean": 1.167177712917328,
      "weighted_auxiliary_loss": 0.0007410961494315416,
      "weighted_relation_loss": 0.0007410961494315416
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.726847521047708,
      "flow_matching_loss": 0.046640570089221,
      "grad_norm": 0.13946828246116638,
      "learning_rate": 8.675258149709265e-05,
      "loss": 0.0474,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00074758545961231,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.426636505126954,
      "state_query_norm": 1169.649853515625,
      "state_relation_loss": 0.00074758545961231,
      "step": 8260,
      "student_distance_mean": 1576.6949951171875,
      "teacher_distance_mean": 1.1319271802902222,
      "weighted_auxiliary_loss": 0.00074758545961231,
      "weighted_relation_loss": 0.00074758545961231
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.736202057998129,
      "flow_matching_loss": 0.0461408369243145,
      "grad_norm": 0.17371118068695068,
      "learning_rate": 8.67151901440552e-05,
      "loss": 0.0469,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007264220796059818,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.82546272277832,
      "state_query_norm": 1185.5869262695312,
      "state_relation_loss": 0.0007264220796059818,
      "step": 8270,
      "student_distance_mean": 1597.1094604492187,
      "teacher_distance_mean": 1.1450185418128966,
      "weighted_auxiliary_loss": 0.0007264220796059818,
      "weighted_relation_loss": 0.0007264220796059818
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.74555659494855,
      "flow_matching_loss": 0.046501710638403894,
      "grad_norm": 0.13839657604694366,
      "learning_rate": 8.667775417859971e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007393954729195684,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.130057716369627,
      "state_query_norm": 1204.2145751953126,
      "state_relation_loss": 0.0007393954729195684,
      "step": 8280,
      "student_distance_mean": 1616.8533935546875,
      "teacher_distance_mean": 1.1509634494781493,
      "weighted_auxiliary_loss": 0.0007393954729195684,
      "weighted_relation_loss": 0.0007393954729195684
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.754911131898971,
      "flow_matching_loss": 0.046504365280270576,
      "grad_norm": 0.12406903505325317,
      "learning_rate": 8.664027364621441e-05,
      "loss": 0.0473,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007469684584066272,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.160436820983886,
      "state_query_norm": 1198.4765747070312,
      "state_relation_loss": 0.0007469684584066272,
      "step": 8290,
      "student_distance_mean": 1616.6705078125,
      "teacher_distance_mean": 1.1483561515808105,
      "weighted_auxiliary_loss": 0.0007469684584066272,
      "weighted_relation_loss": 0.0007469684584066272
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.764265668849392,
      "flow_matching_loss": 0.04656244553625584,
      "grad_norm": 0.1256696581840515,
      "learning_rate": 8.660274859244168e-05,
      "loss": 0.0473,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007114873849786818,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.30813980102539,
      "state_query_norm": 1201.1533813476562,
      "state_relation_loss": 0.0007114873849786818,
      "step": 8300,
      "student_distance_mean": 1624.5554809570312,
      "teacher_distance_mean": 1.149780285358429,
      "weighted_auxiliary_loss": 0.0007114873849786818,
      "weighted_relation_loss": 0.0007114873849786818
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.773620205799813,
      "flow_matching_loss": 0.048295757919549945,
      "grad_norm": 0.12416248023509979,
      "learning_rate": 8.656517906287799e-05,
      "loss": 0.0491,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007689695165026933,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.445046043395998,
      "state_query_norm": 1217.5040405273437,
      "state_relation_loss": 0.0007689695165026933,
      "step": 8310,
      "student_distance_mean": 1632.0801635742187,
      "teacher_distance_mean": 1.1518667817115784,
      "weighted_auxiliary_loss": 0.0007689695165026933,
      "weighted_relation_loss": 0.0007689695165026933
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.782974742750234,
      "flow_matching_loss": 0.0494898971170187,
      "grad_norm": 0.16435672342777252,
      "learning_rate": 8.652756510317387e-05,
      "loss": 0.0502,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007018768577836454,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.480620765686034,
      "state_query_norm": 1224.2997802734376,
      "state_relation_loss": 0.0007018768577836454,
      "step": 8320,
      "student_distance_mean": 1638.2761108398438,
      "teacher_distance_mean": 1.1591464757919312,
      "weighted_auxiliary_loss": 0.0007018768577836454,
      "weighted_relation_loss": 0.0007018768577836454
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.792329279700655,
      "flow_matching_loss": 0.0473087303340435,
      "grad_norm": 0.1285381317138672,
      "learning_rate": 8.648990675903382e-05,
      "loss": 0.0481,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007483379857148975,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.426570320129393,
      "state_query_norm": 1211.0658325195313,
      "state_relation_loss": 0.0007483379857148975,
      "step": 8330,
      "student_distance_mean": 1631.0468139648438,
      "teacher_distance_mean": 1.1610714793205261,
      "weighted_auxiliary_loss": 0.0007483379857148975,
      "weighted_relation_loss": 0.0007483379857148975
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.801683816651076,
      "flow_matching_loss": 0.04778914712369442,
      "grad_norm": 0.13135157525539398,
      "learning_rate": 8.64522040762163e-05,
      "loss": 0.0485,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007507081085350364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.157710456848143,
      "state_query_norm": 1202.830322265625,
      "state_relation_loss": 0.0007507081085350364,
      "step": 8340,
      "student_distance_mean": 1617.4497192382812,
      "teacher_distance_mean": 1.1514349937438966,
      "weighted_auxiliary_loss": 0.0007507081085350364,
      "weighted_relation_loss": 0.0007507081085350364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.8110383536014965,
      "flow_matching_loss": 0.04659496285021305,
      "grad_norm": 0.1296379715204239,
      "learning_rate": 8.64144571005336e-05,
      "loss": 0.0474,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000758265337208286,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.93864860534668,
      "state_query_norm": 1229.69375,
      "state_relation_loss": 0.000758265337208286,
      "step": 8350,
      "student_distance_mean": 1656.8660400390625,
      "teacher_distance_mean": 1.1709938287734984,
      "weighted_auxiliary_loss": 0.000758265337208286,
      "weighted_relation_loss": 0.000758265337208286
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.8203928905519176,
      "flow_matching_loss": 0.0464199136942625,
      "grad_norm": 0.12009727954864502,
      "learning_rate": 8.637666587785184e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007633155735675245,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.671912956237794,
      "state_query_norm": 1224.1689086914062,
      "state_relation_loss": 0.0007633155735675245,
      "step": 8360,
      "student_distance_mean": 1642.448974609375,
      "teacher_distance_mean": 1.1545225262641907,
      "weighted_auxiliary_loss": 0.0007633155735675245,
      "weighted_relation_loss": 0.0007633155735675245
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.829747427502339,
      "flow_matching_loss": 0.04862712137401104,
      "grad_norm": 0.12374918907880783,
      "learning_rate": 8.633883045409096e-05,
      "loss": 0.0494,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007535503944382071,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.162109184265137,
      "state_query_norm": 1206.0494018554687,
      "state_relation_loss": 0.0007535503944382071,
      "step": 8370,
      "student_distance_mean": 1617.3815795898438,
      "teacher_distance_mean": 1.14229416847229,
      "weighted_auxiliary_loss": 0.0007535503944382071,
      "weighted_relation_loss": 0.0007535503944382071
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.83910196445276,
      "flow_matching_loss": 0.04760638922452927,
      "grad_norm": 0.17623203992843628,
      "learning_rate": 8.630095087522457e-05,
      "loss": 0.0483,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007338814728427678,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.507408332824706,
      "state_query_norm": 1209.2107666015625,
      "state_relation_loss": 0.0007338814728427678,
      "step": 8380,
      "student_distance_mean": 1633.7612426757812,
      "teacher_distance_mean": 1.1566290855407715,
      "weighted_auxiliary_loss": 0.0007338814728427678,
      "weighted_relation_loss": 0.0007338814728427678
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.848456501403181,
      "flow_matching_loss": 0.05014626011252403,
      "grad_norm": 0.16805708408355713,
      "learning_rate": 8.62630271872799e-05,
      "loss": 0.0509,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007328037638217211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.263854026794434,
      "state_query_norm": 1208.2423583984375,
      "state_relation_loss": 0.0007328037638217211,
      "step": 8390,
      "student_distance_mean": 1625.1229858398438,
      "teacher_distance_mean": 1.1532922506332397,
      "weighted_auxiliary_loss": 0.0007328037638217211,
      "weighted_relation_loss": 0.0007328037638217211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.857811038353601,
      "flow_matching_loss": 0.045177629590034483,
      "grad_norm": 0.1362423449754715,
      "learning_rate": 8.62250594363378e-05,
      "loss": 0.0459,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007599485223181545,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.04572811126709,
      "state_query_norm": 1196.6623413085938,
      "state_relation_loss": 0.0007599485223181545,
      "step": 8400,
      "student_distance_mean": 1612.5690307617188,
      "teacher_distance_mean": 1.1475250482559205,
      "weighted_auxiliary_loss": 0.0007599485223181545,
      "weighted_relation_loss": 0.0007599485223181545
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.867165575304022,
      "flow_matching_loss": 0.046162960678339006,
      "grad_norm": 0.13288713991641998,
      "learning_rate": 8.61870476685327e-05,
      "loss": 0.0469,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007215273391921074,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.02126407623291,
      "state_query_norm": 1227.5092407226562,
      "state_relation_loss": 0.0007215273391921074,
      "step": 8410,
      "student_distance_mean": 1660.298291015625,
      "teacher_distance_mean": 1.1681220054626464,
      "weighted_auxiliary_loss": 0.0007215273391921074,
      "weighted_relation_loss": 0.0007215273391921074
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.876520112254443,
      "flow_matching_loss": 0.046883418783545495,
      "grad_norm": 0.14580412209033966,
      "learning_rate": 8.614899193005248e-05,
      "loss": 0.0476,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007447915326338261,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.50554676055908,
      "state_query_norm": 1214.694482421875,
      "state_relation_loss": 0.0007447915326338261,
      "step": 8420,
      "student_distance_mean": 1637.31552734375,
      "teacher_distance_mean": 1.157834804058075,
      "weighted_auxiliary_loss": 0.0007447915326338261,
      "weighted_relation_loss": 0.0007447915326338261
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.885874649204864,
      "flow_matching_loss": 0.046433756873011586,
      "grad_norm": 0.12293489277362823,
      "learning_rate": 8.611089226713842e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007254281314089895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.710219192504884,
      "state_query_norm": 1220.2320678710937,
      "state_relation_loss": 0.0007254281314089895,
      "step": 8430,
      "student_distance_mean": 1646.098291015625,
      "teacher_distance_mean": 1.166000521183014,
      "weighted_auxiliary_loss": 0.0007254281314089895,
      "weighted_relation_loss": 0.0007254281314089895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.8952291861552855,
      "flow_matching_loss": 0.046226216852664946,
      "grad_norm": 0.16364306211471558,
      "learning_rate": 8.607274872608521e-05,
      "loss": 0.0469,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007113507715985179,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.700990867614745,
      "state_query_norm": 1182.0754150390626,
      "state_relation_loss": 0.0007113507715985179,
      "step": 8440,
      "student_distance_mean": 1593.732861328125,
      "teacher_distance_mean": 1.125552487373352,
      "weighted_auxiliary_loss": 0.0007113507715985179,
      "weighted_relation_loss": 0.0007113507715985179
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.904583723105707,
      "flow_matching_loss": 0.046398724615573886,
      "grad_norm": 0.12770028412342072,
      "learning_rate": 8.603456135324088e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007667694706469774,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.269750785827636,
      "state_query_norm": 1204.2650756835938,
      "state_relation_loss": 0.0007667694706469774,
      "step": 8450,
      "student_distance_mean": 1622.6454833984376,
      "teacher_distance_mean": 1.1510425686836243,
      "weighted_auxiliary_loss": 0.0007667694706469774,
      "weighted_relation_loss": 0.0007667694706469774
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.913938260056128,
      "flow_matching_loss": 0.04545519538223743,
      "grad_norm": 0.11795072257518768,
      "learning_rate": 8.599633019500665e-05,
      "loss": 0.0462,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007316072122193872,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.404209518432616,
      "state_query_norm": 1221.1795532226563,
      "state_relation_loss": 0.0007316072122193872,
      "step": 8460,
      "student_distance_mean": 1634.6785400390625,
      "teacher_distance_mean": 1.1564115166664124,
      "weighted_auxiliary_loss": 0.0007316072122193872,
      "weighted_relation_loss": 0.0007316072122193872
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.923292797006548,
      "flow_matching_loss": 0.04827189520001411,
      "grad_norm": 0.13168682157993317,
      "learning_rate": 8.595805529783702e-05,
      "loss": 0.049,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007145338866394013,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.373897361755372,
      "state_query_norm": 1211.36123046875,
      "state_relation_loss": 0.0007145338866394013,
      "step": 8470,
      "student_distance_mean": 1627.0124755859374,
      "teacher_distance_mean": 1.1575315117835998,
      "weighted_auxiliary_loss": 0.0007145338866394013,
      "weighted_relation_loss": 0.0007145338866394013
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.932647333956969,
      "flow_matching_loss": 0.048682589828968045,
      "grad_norm": 0.13643312454223633,
      "learning_rate": 8.591973670823958e-05,
      "loss": 0.0494,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007150117715355008,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.775081825256347,
      "state_query_norm": 1192.2287719726562,
      "state_relation_loss": 0.0007150117715355008,
      "step": 8480,
      "student_distance_mean": 1598.091259765625,
      "teacher_distance_mean": 1.1396965026855468,
      "weighted_auxiliary_loss": 0.0007150117715355008,
      "weighted_relation_loss": 0.0007150117715355008
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.94200187090739,
      "flow_matching_loss": 0.04666789621114731,
      "grad_norm": 0.13899099826812744,
      "learning_rate": 8.588137447277503e-05,
      "loss": 0.0474,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006888279574923217,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 28.80754985809326,
      "state_query_norm": 1186.0307739257812,
      "state_relation_loss": 0.0006888279574923217,
      "step": 8490,
      "student_distance_mean": 1598.816650390625,
      "teacher_distance_mean": 1.1391934156417847,
      "weighted_auxiliary_loss": 0.0006888279574923217,
      "weighted_relation_loss": 0.0006888279574923217
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.951356407857811,
      "flow_matching_loss": 0.04836431108415127,
      "grad_norm": 0.1575760841369629,
      "learning_rate": 8.584296863805709e-05,
      "loss": 0.0491,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007538862759247422,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.21902713775635,
      "state_query_norm": 1199.3841918945313,
      "state_relation_loss": 0.0007538862759247422,
      "step": 8500,
      "student_distance_mean": 1620.04755859375,
      "teacher_distance_mean": 1.1510718226432801,
      "weighted_auxiliary_loss": 0.0007538862759247422,
      "weighted_relation_loss": 0.0007538862759247422
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.960710944808232,
      "flow_matching_loss": 0.04643992744386196,
      "grad_norm": 0.1522206813097,
      "learning_rate": 8.580451925075248e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007427826756611466,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.243922805786134,
      "state_query_norm": 1202.2934692382812,
      "state_relation_loss": 0.0007427826756611466,
      "step": 8510,
      "student_distance_mean": 1619.9452392578125,
      "teacher_distance_mean": 1.151408588886261,
      "weighted_auxiliary_loss": 0.0007427826756611466,
      "weighted_relation_loss": 0.0007427826756611466
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.9700654817586525,
      "flow_matching_loss": 0.04922537095844746,
      "grad_norm": 0.15048913657665253,
      "learning_rate": 8.576602635758085e-05,
      "loss": 0.05,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007411122322082519,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.867474174499513,
      "state_query_norm": 1227.9500854492187,
      "state_relation_loss": 0.0007411122322082519,
      "step": 8520,
      "student_distance_mean": 1657.1314575195313,
      "teacher_distance_mean": 1.168204951286316,
      "weighted_auxiliary_loss": 0.0007411122322082519,
      "weighted_relation_loss": 0.0007411122322082519
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.979420018709074,
      "flow_matching_loss": 0.047418785840272905,
      "grad_norm": 0.11292584985494614,
      "learning_rate": 8.572749000531469e-05,
      "loss": 0.0481,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006824746960774064,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.8134973526001,
      "state_query_norm": 1237.8109375,
      "state_relation_loss": 0.0006824746960774064,
      "step": 8530,
      "student_distance_mean": 1652.7178955078125,
      "teacher_distance_mean": 1.1585428714752197,
      "weighted_auxiliary_loss": 0.0006824746960774064,
      "weighted_relation_loss": 0.0006824746960774064
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.988774555659495,
      "flow_matching_loss": 0.0458662174642086,
      "grad_norm": 0.11256349086761475,
      "learning_rate": 8.568891024077926e-05,
      "loss": 0.0466,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006918931612744927,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.3484769821167,
      "state_query_norm": 1219.665234375,
      "state_relation_loss": 0.0006918931612744927,
      "step": 8540,
      "student_distance_mean": 1630.3181640625,
      "teacher_distance_mean": 1.1483950018882751,
      "weighted_auxiliary_loss": 0.0006918931612744927,
      "weighted_relation_loss": 0.0006918931612744927
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 7.998129092609916,
      "flow_matching_loss": 0.046981073915958405,
      "grad_norm": 0.11513198912143707,
      "learning_rate": 8.565028711085265e-05,
      "loss": 0.0477,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007473201025277376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.299674797058106,
      "state_query_norm": 1207.5508666992187,
      "state_relation_loss": 0.0007473201025277376,
      "step": 8550,
      "student_distance_mean": 1626.88134765625,
      "teacher_distance_mean": 1.1426936388015747,
      "weighted_auxiliary_loss": 0.0007473201025277376,
      "weighted_relation_loss": 0.0007473201025277376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.007483629560337,
      "flow_matching_loss": 0.04781044907867908,
      "grad_norm": 0.13072094321250916,
      "learning_rate": 8.561162066246561e-05,
      "loss": 0.0485,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0007094629399944097,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 29.436054039001466,
      "state_query_norm": 1212.1496215820312,
      "state_relation_loss": 0.0007094629399944097,
      "step": 8560,
      "student_distance_mean": 1635.5921142578125,
      "teacher_distance_mean": 1.1421643853187562,
      "weighted_auxiliary_loss": 0.0007094629399944097,
      "weighted_relation_loss": 0.0007094629399944097
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.016838166510757,
      "flow_matching_loss": 0.04845139905810356,
      "grad_norm": 0.12739118933677673,
      "learning_rate": 8.557291094260152e-05,
      "loss": 0.0492,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007656002067960799,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.463352012634278,
      "state_query_norm": 1243.7330444335937,
      "state_relation_loss": 0.0007656002067960799,
      "step": 8570,
      "student_distance_mean": 1686.0810791015624,
      "teacher_distance_mean": 1.1793560147285462,
      "weighted_auxiliary_loss": 0.0007656002067960799,
      "weighted_relation_loss": 0.0007656002067960799
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.026192703461179,
      "flow_matching_loss": 0.045764248073101046,
      "grad_norm": 0.14885061979293823,
      "learning_rate": 8.553415799829629e-05,
      "loss": 0.0465,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007348965213168412,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.718510818481445,
      "state_query_norm": 1217.5445922851563,
      "state_relation_loss": 0.0007348965213168412,
      "step": 8580,
      "student_distance_mean": 1645.7425170898437,
      "teacher_distance_mean": 1.1531318783760072,
      "weighted_auxiliary_loss": 0.0007348965213168412,
      "weighted_relation_loss": 0.0007348965213168412
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.0355472404116,
      "flow_matching_loss": 0.04939300268888473,
      "grad_norm": 0.16002309322357178,
      "learning_rate": 8.549536187663847e-05,
      "loss": 0.0501,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000732630823040381,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.702034950256348,
      "state_query_norm": 1216.179296875,
      "state_relation_loss": 0.000732630823040381,
      "step": 8590,
      "student_distance_mean": 1643.3815551757812,
      "teacher_distance_mean": 1.153227198123932,
      "weighted_auxiliary_loss": 0.000732630823040381,
      "weighted_relation_loss": 0.000732630823040381
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.044901777362021,
      "flow_matching_loss": 0.047426749020814896,
      "grad_norm": 0.11329764872789383,
      "learning_rate": 8.545652262476897e-05,
      "loss": 0.0482,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007368555699940771,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.71462059020996,
      "state_query_norm": 1223.3296508789062,
      "state_relation_loss": 0.0007368555699940771,
      "step": 8600,
      "student_distance_mean": 1648.2651245117188,
      "teacher_distance_mean": 1.1588240146636963,
      "weighted_auxiliary_loss": 0.0007368555699940771,
      "weighted_relation_loss": 0.0007368555699940771
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.054256314312441,
      "flow_matching_loss": 0.0480233408510685,
      "grad_norm": 0.10492725670337677,
      "learning_rate": 8.541764028988115e-05,
      "loss": 0.0487,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006902271474245935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.61661605834961,
      "state_query_norm": 1220.7323852539062,
      "state_relation_loss": 0.0006902271474245935,
      "step": 8610,
      "student_distance_mean": 1645.2155151367188,
      "teacher_distance_mean": 1.1488860845565796,
      "weighted_auxiliary_loss": 0.0006902271474245935,
      "weighted_relation_loss": 0.0006902271474245935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.063610851262862,
      "flow_matching_loss": 0.045709742978215215,
      "grad_norm": 0.12392457574605942,
      "learning_rate": 8.537871491922071e-05,
      "loss": 0.0464,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007142982387449593,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.655992126464845,
      "state_query_norm": 1216.7952270507812,
      "state_relation_loss": 0.0007142982387449593,
      "step": 8620,
      "student_distance_mean": 1642.765625,
      "teacher_distance_mean": 1.1461669087409974,
      "weighted_auxiliary_loss": 0.0007142982387449593,
      "weighted_relation_loss": 0.0007142982387449593
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.072965388213284,
      "flow_matching_loss": 0.048790520057082176,
      "grad_norm": 0.12754906713962555,
      "learning_rate": 8.533974656008566e-05,
      "loss": 0.0495,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007037885719910264,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.435237312316893,
      "state_query_norm": 1205.1001098632812,
      "state_relation_loss": 0.0007037885719910264,
      "step": 8630,
      "student_distance_mean": 1629.60517578125,
      "teacher_distance_mean": 1.1378964543342591,
      "weighted_auxiliary_loss": 0.0007037885719910264,
      "weighted_relation_loss": 0.0007037885719910264
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.082319925163704,
      "flow_matching_loss": 0.049805917963385585,
      "grad_norm": 0.11808574199676514,
      "learning_rate": 8.530073525982621e-05,
      "loss": 0.0505,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007131826423574239,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.961980628967286,
      "state_query_norm": 1232.9691162109375,
      "state_relation_loss": 0.0007131826423574239,
      "step": 8640,
      "student_distance_mean": 1663.4621948242188,
      "teacher_distance_mean": 1.1608106255531312,
      "weighted_auxiliary_loss": 0.0007131826423574239,
      "weighted_relation_loss": 0.0007131826423574239
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.091674462114126,
      "flow_matching_loss": 0.04651513323187828,
      "grad_norm": 0.12089912593364716,
      "learning_rate": 8.526168106584476e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007065419747959822,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.738997650146484,
      "state_query_norm": 1221.2878051757812,
      "state_relation_loss": 0.0007065419747959822,
      "step": 8650,
      "student_distance_mean": 1645.6823120117188,
      "teacher_distance_mean": 1.1545154690742492,
      "weighted_auxiliary_loss": 0.0007065419747959822,
      "weighted_relation_loss": 0.0007065419747959822
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.101028999064546,
      "flow_matching_loss": 0.046882163360714914,
      "grad_norm": 0.1522672325372696,
      "learning_rate": 8.522258402559586e-05,
      "loss": 0.0476,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007335858652368187,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.215277862548827,
      "state_query_norm": 1240.8037963867187,
      "state_relation_loss": 0.0007335858652368187,
      "step": 8660,
      "student_distance_mean": 1672.153173828125,
      "teacher_distance_mean": 1.1695695161819457,
      "weighted_auxiliary_loss": 0.0007335858652368187,
      "weighted_relation_loss": 0.0007335858652368187
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.110383536014968,
      "flow_matching_loss": 0.045449203625321385,
      "grad_norm": 0.1233115866780281,
      "learning_rate": 8.518344418658613e-05,
      "loss": 0.0462,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007177695515565574,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.056159019470215,
      "state_query_norm": 1236.1456176757813,
      "state_relation_loss": 0.0007177695515565574,
      "step": 8670,
      "student_distance_mean": 1669.0123168945313,
      "teacher_distance_mean": 1.165192222595215,
      "weighted_auxiliary_loss": 0.0007177695515565574,
      "weighted_relation_loss": 0.0007177695515565574
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.119738072965388,
      "flow_matching_loss": 0.04493583664298058,
      "grad_norm": 0.1327926069498062,
      "learning_rate": 8.51442615963741e-05,
      "loss": 0.0456,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007119536225218326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.88677558898926,
      "state_query_norm": 1228.4793212890625,
      "state_relation_loss": 0.0007119536225218326,
      "step": 8680,
      "student_distance_mean": 1656.1620361328125,
      "teacher_distance_mean": 1.1530622482299804,
      "weighted_auxiliary_loss": 0.0007119536225218326,
      "weighted_relation_loss": 0.0007119536225218326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.129092609915809,
      "flow_matching_loss": 0.04457283988595009,
      "grad_norm": 0.10176560282707214,
      "learning_rate": 8.510503630257034e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007264784537255764,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.789075088500976,
      "state_query_norm": 1223.9714599609374,
      "state_relation_loss": 0.0007264784537255764,
      "step": 8690,
      "student_distance_mean": 1654.336181640625,
      "teacher_distance_mean": 1.1562951683998108,
      "weighted_auxiliary_loss": 0.0007264784537255764,
      "weighted_relation_loss": 0.0007264784537255764
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.13844714686623,
      "flow_matching_loss": 0.04606135152280331,
      "grad_norm": 0.12754249572753906,
      "learning_rate": 8.50657683528373e-05,
      "loss": 0.0467,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006846133968792856,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.748287773132326,
      "state_query_norm": 1223.62392578125,
      "state_relation_loss": 0.0006846133968792856,
      "step": 8700,
      "student_distance_mean": 1648.3775634765625,
      "teacher_distance_mean": 1.1476344108581542,
      "weighted_auxiliary_loss": 0.0006846133968792856,
      "weighted_relation_loss": 0.0006846133968792856
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.14780168381665,
      "flow_matching_loss": 0.0460821270942688,
      "grad_norm": 0.11760639399290085,
      "learning_rate": 8.502645779488922e-05,
      "loss": 0.0467,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006624005152843892,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.88634147644043,
      "state_query_norm": 1231.160595703125,
      "state_relation_loss": 0.0006624005152843892,
      "step": 8710,
      "student_distance_mean": 1659.7512817382812,
      "teacher_distance_mean": 1.1535073876380921,
      "weighted_auxiliary_loss": 0.0006624005152843892,
      "weighted_relation_loss": 0.0006624005152843892
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.157156220767073,
      "flow_matching_loss": 0.04506725892424583,
      "grad_norm": 0.14766651391983032,
      "learning_rate": 8.498710467649213e-05,
      "loss": 0.0458,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000724473933223635,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.664343643188477,
      "state_query_norm": 1226.1322631835938,
      "state_relation_loss": 0.000724473933223635,
      "step": 8720,
      "student_distance_mean": 1645.5338989257812,
      "teacher_distance_mean": 1.1527413845062255,
      "weighted_auxiliary_loss": 0.000724473933223635,
      "weighted_relation_loss": 0.000724473933223635
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.166510757717493,
      "flow_matching_loss": 0.04557272121310234,
      "grad_norm": 0.14685386419296265,
      "learning_rate": 8.494770904546381e-05,
      "loss": 0.0463,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007132260652724653,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.553066062927247,
      "state_query_norm": 1220.272265625,
      "state_relation_loss": 0.0007132260652724653,
      "step": 8730,
      "student_distance_mean": 1639.3363037109375,
      "teacher_distance_mean": 1.149713158607483,
      "weighted_auxiliary_loss": 0.0007132260652724653,
      "weighted_relation_loss": 0.0007132260652724653
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.175865294667913,
      "flow_matching_loss": 0.04645274430513382,
      "grad_norm": 0.13251672685146332,
      "learning_rate": 8.490827094967363e-05,
      "loss": 0.0471,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000692098360741511,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.188324165344238,
      "state_query_norm": 1200.9419921875,
      "state_relation_loss": 0.000692098360741511,
      "step": 8740,
      "student_distance_mean": 1618.551708984375,
      "teacher_distance_mean": 1.1346566438674928,
      "weighted_auxiliary_loss": 0.000692098360741511,
      "weighted_relation_loss": 0.000692098360741511
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.185219831618335,
      "flow_matching_loss": 0.04454544186592102,
      "grad_norm": 0.14446775615215302,
      "learning_rate": 8.486879043704263e-05,
      "loss": 0.0452,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006702438928186894,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.490914726257323,
      "state_query_norm": 1212.8177124023437,
      "state_relation_loss": 0.0006702438928186894,
      "step": 8750,
      "student_distance_mean": 1638.5841064453125,
      "teacher_distance_mean": 1.1521270632743836,
      "weighted_auxiliary_loss": 0.0006702438928186894,
      "weighted_relation_loss": 0.0006702438928186894
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.194574368568755,
      "flow_matching_loss": 0.04569541364908218,
      "grad_norm": 0.1364871859550476,
      "learning_rate": 8.482926755554332e-05,
      "loss": 0.0464,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006709624489303678,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.91262912750244,
      "state_query_norm": 1220.657958984375,
      "state_relation_loss": 0.0006709624489303678,
      "step": 8760,
      "student_distance_mean": 1654.7264892578125,
      "teacher_distance_mean": 1.1653096437454225,
      "weighted_auxiliary_loss": 0.0006709624489303678,
      "weighted_relation_loss": 0.0006709624489303678
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.203928905519177,
      "flow_matching_loss": 0.046665092930197716,
      "grad_norm": 0.16240786015987396,
      "learning_rate": 8.478970235319975e-05,
      "loss": 0.0474,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007021075289230794,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.324656295776368,
      "state_query_norm": 1212.08837890625,
      "state_relation_loss": 0.0007021075289230794,
      "step": 8770,
      "student_distance_mean": 1629.4680541992188,
      "teacher_distance_mean": 1.1452075123786927,
      "weighted_auxiliary_loss": 0.0007021075289230794,
      "weighted_relation_loss": 0.0007021075289230794
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.213283442469598,
      "flow_matching_loss": 0.04343659915030003,
      "grad_norm": 0.11935156583786011,
      "learning_rate": 8.475009487808737e-05,
      "loss": 0.0441,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006900595559272915,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.101931953430174,
      "state_query_norm": 1239.9209228515624,
      "state_relation_loss": 0.0006900595559272915,
      "step": 8780,
      "student_distance_mean": 1669.1459594726562,
      "teacher_distance_mean": 1.1628754615783692,
      "weighted_auxiliary_loss": 0.0006900595559272915,
      "weighted_relation_loss": 0.0006900595559272915
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.22263797942002,
      "flow_matching_loss": 0.04603551551699638,
      "grad_norm": 0.11226548999547958,
      "learning_rate": 8.471044517833298e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007273404218722135,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.243135833740233,
      "state_query_norm": 1240.67197265625,
      "state_relation_loss": 0.0007273404218722135,
      "step": 8790,
      "student_distance_mean": 1672.5499267578125,
      "teacher_distance_mean": 1.1670518398284913,
      "weighted_auxiliary_loss": 0.0007273404218722135,
      "weighted_relation_loss": 0.0007273404218722135
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.23199251637044,
      "flow_matching_loss": 0.04517314061522484,
      "grad_norm": 0.16959884762763977,
      "learning_rate": 8.467075330211474e-05,
      "loss": 0.0459,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000734483520500362,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.464461898803712,
      "state_query_norm": 1205.7843505859375,
      "state_relation_loss": 0.000734483520500362,
      "step": 8800,
      "student_distance_mean": 1635.8552856445312,
      "teacher_distance_mean": 1.1468252062797546,
      "weighted_auxiliary_loss": 0.000734483520500362,
      "weighted_relation_loss": 0.000734483520500362
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.24134705332086,
      "flow_matching_loss": 0.04634832590818405,
      "grad_norm": 0.19941341876983643,
      "learning_rate": 8.463101929766198e-05,
      "loss": 0.0471,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007229871116578579,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.739528274536134,
      "state_query_norm": 1226.239208984375,
      "state_relation_loss": 0.0007229871116578579,
      "step": 8810,
      "student_distance_mean": 1650.6771728515625,
      "teacher_distance_mean": 1.1561475157737733,
      "weighted_auxiliary_loss": 0.0007229871116578579,
      "weighted_relation_loss": 0.0007229871116578579
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.250701590271282,
      "flow_matching_loss": 0.044903894513845445,
      "grad_norm": 0.16016338765621185,
      "learning_rate": 8.459124321325528e-05,
      "loss": 0.0456,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007427857315633446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.59688320159912,
      "state_query_norm": 1215.9206787109374,
      "state_relation_loss": 0.0007427857315633446,
      "step": 8820,
      "student_distance_mean": 1640.246533203125,
      "teacher_distance_mean": 1.144491934776306,
      "weighted_auxiliary_loss": 0.0007427857315633446,
      "weighted_relation_loss": 0.0007427857315633446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.260056127221702,
      "flow_matching_loss": 0.04610229693353176,
      "grad_norm": 0.12467060983181,
      "learning_rate": 8.455142509722634e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006936846417374909,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.836146354675293,
      "state_query_norm": 1218.8272094726562,
      "state_relation_loss": 0.0006936846417374909,
      "step": 8830,
      "student_distance_mean": 1652.2815307617188,
      "teacher_distance_mean": 1.151377272605896,
      "weighted_auxiliary_loss": 0.0006936846417374909,
      "weighted_relation_loss": 0.0006936846417374909
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.269410664172124,
      "flow_matching_loss": 0.04554670974612236,
      "grad_norm": 0.15128226578235626,
      "learning_rate": 8.451156499795792e-05,
      "loss": 0.0463,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007328996434807777,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.662582397460938,
      "state_query_norm": 1224.2974609375,
      "state_relation_loss": 0.0007328996434807777,
      "step": 8840,
      "student_distance_mean": 1643.8960815429687,
      "teacher_distance_mean": 1.1484894633293152,
      "weighted_auxiliary_loss": 0.0007328996434807777,
      "weighted_relation_loss": 0.0007328996434807777
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.278765201122544,
      "flow_matching_loss": 0.0445561345666647,
      "grad_norm": 0.15609051287174225,
      "learning_rate": 8.447166296388381e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007475078862626105,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.268678283691408,
      "state_query_norm": 1242.7786499023437,
      "state_relation_loss": 0.0007475078862626105,
      "step": 8850,
      "student_distance_mean": 1676.2086059570313,
      "teacher_distance_mean": 1.1663904309272766,
      "weighted_auxiliary_loss": 0.0007475078862626105,
      "weighted_relation_loss": 0.0007475078862626105
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.288119738072965,
      "flow_matching_loss": 0.04811318665742874,
      "grad_norm": 0.14924803376197815,
      "learning_rate": 8.443171904348872e-05,
      "loss": 0.0488,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007129146775696427,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.712349128723144,
      "state_query_norm": 1222.5796752929687,
      "state_relation_loss": 0.0007129146775696427,
      "step": 8860,
      "student_distance_mean": 1646.0626831054688,
      "teacher_distance_mean": 1.145149290561676,
      "weighted_auxiliary_loss": 0.0007129146775696427,
      "weighted_relation_loss": 0.0007129146775696427
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.297474275023387,
      "flow_matching_loss": 0.046611401811242105,
      "grad_norm": 0.13871853053569794,
      "learning_rate": 8.43917332853083e-05,
      "loss": 0.0473,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007120152644347399,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.060349655151366,
      "state_query_norm": 1232.4333129882812,
      "state_relation_loss": 0.0007120152644347399,
      "step": 8870,
      "student_distance_mean": 1665.1899291992188,
      "teacher_distance_mean": 1.1516376852989196,
      "weighted_auxiliary_loss": 0.0007120152644347399,
      "weighted_relation_loss": 0.0007120152644347399
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.306828811973807,
      "flow_matching_loss": 0.045358624681830405,
      "grad_norm": 0.13993139564990997,
      "learning_rate": 8.435170573792901e-05,
      "loss": 0.046,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006801258015912027,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.774624633789063,
      "state_query_norm": 1226.396044921875,
      "state_relation_loss": 0.0006801258015912027,
      "step": 8880,
      "student_distance_mean": 1646.327783203125,
      "teacher_distance_mean": 1.1451245546340942,
      "weighted_auxiliary_loss": 0.0006801258015912027,
      "weighted_relation_loss": 0.0006801258015912027
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.316183348924229,
      "flow_matching_loss": 0.04530370123684406,
      "grad_norm": 0.13637545704841614,
      "learning_rate": 8.431163644998808e-05,
      "loss": 0.046,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007229422917589545,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.071583557128907,
      "state_query_norm": 1232.2750244140625,
      "state_relation_loss": 0.0007229422917589545,
      "step": 8890,
      "student_distance_mean": 1662.8682983398437,
      "teacher_distance_mean": 1.1567638516426086,
      "weighted_auxiliary_loss": 0.0007229422917589545,
      "weighted_relation_loss": 0.0007229422917589545
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.325537885874649,
      "flow_matching_loss": 0.044697601720690726,
      "grad_norm": 0.1437443494796753,
      "learning_rate": 8.42715254701735e-05,
      "loss": 0.0454,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006919394654687494,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.342304611206053,
      "state_query_norm": 1217.364404296875,
      "state_relation_loss": 0.0006919394654687494,
      "step": 8900,
      "student_distance_mean": 1629.3282592773437,
      "teacher_distance_mean": 1.1326342225074768,
      "weighted_auxiliary_loss": 0.0006919394654687494,
      "weighted_relation_loss": 0.0006919394654687494
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.334892422825071,
      "flow_matching_loss": 0.04678560085594654,
      "grad_norm": 0.13205358386039734,
      "learning_rate": 8.42313728472239e-05,
      "loss": 0.0475,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007101976429112255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.38589382171631,
      "state_query_norm": 1208.592919921875,
      "state_relation_loss": 0.0007101976429112255,
      "step": 8910,
      "student_distance_mean": 1631.5960693359375,
      "teacher_distance_mean": 1.1410062432289123,
      "weighted_auxiliary_loss": 0.0007101976429112255,
      "weighted_relation_loss": 0.0007101976429112255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.344246959775491,
      "flow_matching_loss": 0.04610676094889641,
      "grad_norm": 0.15790040791034698,
      "learning_rate": 8.419117862992844e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006992590264417231,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.069332122802734,
      "state_query_norm": 1233.2333618164062,
      "state_relation_loss": 0.0006992590264417231,
      "step": 8920,
      "student_distance_mean": 1665.1265747070313,
      "teacher_distance_mean": 1.158819854259491,
      "weighted_auxiliary_loss": 0.0006992590264417231,
      "weighted_relation_loss": 0.0006992590264417231
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.353601496725911,
      "flow_matching_loss": 0.045475820824503896,
      "grad_norm": 0.11707915365695953,
      "learning_rate": 8.415094286712694e-05,
      "loss": 0.0462,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006954409996978939,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.295104789733887,
      "state_query_norm": 1239.5097412109376,
      "state_relation_loss": 0.0006954409996978939,
      "step": 8930,
      "student_distance_mean": 1680.235009765625,
      "teacher_distance_mean": 1.167378270626068,
      "weighted_auxiliary_loss": 0.0006954409996978939,
      "weighted_relation_loss": 0.0006954409996978939
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.362956033676333,
      "flow_matching_loss": 0.044621435925364494,
      "grad_norm": 0.12276992946863174,
      "learning_rate": 8.411066560770965e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007041644828859716,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.795483779907226,
      "state_query_norm": 1225.9417114257812,
      "state_relation_loss": 0.0007041644828859716,
      "step": 8940,
      "student_distance_mean": 1652.7115844726563,
      "teacher_distance_mean": 1.1477660298347474,
      "weighted_auxiliary_loss": 0.0007041644828859716,
      "weighted_relation_loss": 0.0007041644828859716
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.372310570626754,
      "flow_matching_loss": 0.04423740841448307,
      "grad_norm": 0.14587250351905823,
      "learning_rate": 8.407034690061721e-05,
      "loss": 0.045,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007304087514057756,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.908213996887206,
      "state_query_norm": 1229.0702026367187,
      "state_relation_loss": 0.0007304087514057756,
      "step": 8950,
      "student_distance_mean": 1658.9898681640625,
      "teacher_distance_mean": 1.1523898124694825,
      "weighted_auxiliary_loss": 0.0007304087514057756,
      "weighted_relation_loss": 0.0007304087514057756
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.381665107577176,
      "flow_matching_loss": 0.04359632208943367,
      "grad_norm": 0.09002635627985,
      "learning_rate": 8.402998679484067e-05,
      "loss": 0.0443,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000712400523480028,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.763445091247558,
      "state_query_norm": 1222.0856689453126,
      "state_relation_loss": 0.000712400523480028,
      "step": 8960,
      "student_distance_mean": 1648.210205078125,
      "teacher_distance_mean": 1.143178606033325,
      "weighted_auxiliary_loss": 0.000712400523480028,
      "weighted_relation_loss": 0.000712400523480028
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.391019644527596,
      "flow_matching_loss": 0.04397016130387783,
      "grad_norm": 0.14678114652633667,
      "learning_rate": 8.398958533942135e-05,
      "loss": 0.0447,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007483221008442343,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.26480712890625,
      "state_query_norm": 1242.5158203125,
      "state_relation_loss": 0.0007483221008442343,
      "step": 8970,
      "student_distance_mean": 1674.7518432617187,
      "teacher_distance_mean": 1.161120069026947,
      "weighted_auxiliary_loss": 0.0007483221008442343,
      "weighted_relation_loss": 0.0007483221008442343
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.400374181478018,
      "flow_matching_loss": 0.04374243542551994,
      "grad_norm": 0.11952260136604309,
      "learning_rate": 8.394914258345083e-05,
      "loss": 0.0445,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007120559050235898,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.55906734466553,
      "state_query_norm": 1255.4415649414063,
      "state_relation_loss": 0.0007120559050235898,
      "step": 8980,
      "student_distance_mean": 1694.1866088867187,
      "teacher_distance_mean": 1.168886387348175,
      "weighted_auxiliary_loss": 0.0007120559050235898,
      "weighted_relation_loss": 0.0007120559050235898
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.409728718428438,
      "flow_matching_loss": 0.04363497644662857,
      "grad_norm": 0.11668038368225098,
      "learning_rate": 8.390865857607089e-05,
      "loss": 0.0443,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006803067168220878,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.478341102600098,
      "state_query_norm": 1248.9474487304688,
      "state_relation_loss": 0.0006803067168220878,
      "step": 8990,
      "student_distance_mean": 1690.2158569335938,
      "teacher_distance_mean": 1.1671538829803467,
      "weighted_auxiliary_loss": 0.0006803067168220878,
      "weighted_relation_loss": 0.0006803067168220878
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.419083255378858,
      "flow_matching_loss": 0.046144478395581244,
      "grad_norm": 0.16700822114944458,
      "learning_rate": 8.386813336647339e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006859313114546239,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.806050491333007,
      "state_query_norm": 1231.5552368164062,
      "state_relation_loss": 0.0006859313114546239,
      "step": 9000,
      "student_distance_mean": 1653.2432373046875,
      "teacher_distance_mean": 1.141900360584259,
      "weighted_auxiliary_loss": 0.0006859313114546239,
      "weighted_relation_loss": 0.0006859313114546239
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.42843779232928,
      "flow_matching_loss": 0.045661048591136934,
      "grad_norm": 0.1167205423116684,
      "learning_rate": 8.38275670039003e-05,
      "loss": 0.0464,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006936197285540402,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.332564544677734,
      "state_query_norm": 1248.2303100585937,
      "state_relation_loss": 0.0006936197285540402,
      "step": 9010,
      "student_distance_mean": 1682.3928466796874,
      "teacher_distance_mean": 1.1588507652282716,
      "weighted_auxiliary_loss": 0.0006936197285540402,
      "weighted_relation_loss": 0.0006936197285540402
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.4377923292797,
      "flow_matching_loss": 0.04353260025382042,
      "grad_norm": 0.10252255201339722,
      "learning_rate": 8.378695953764356e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006995211762841791,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.36208839416504,
      "state_query_norm": 1237.4176635742188,
      "state_relation_loss": 0.0006995211762841791,
      "step": 9020,
      "student_distance_mean": 1680.1681884765626,
      "teacher_distance_mean": 1.1604443907737731,
      "weighted_auxiliary_loss": 0.0006995211762841791,
      "weighted_relation_loss": 0.0006995211762841791
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.447146866230122,
      "flow_matching_loss": 0.043252181634306906,
      "grad_norm": 0.16557152569293976,
      "learning_rate": 8.374631101704509e-05,
      "loss": 0.0439,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006503494747448713,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.848117065429687,
      "state_query_norm": 1231.5771850585938,
      "state_relation_loss": 0.0006503494747448713,
      "step": 9030,
      "student_distance_mean": 1658.5067993164062,
      "teacher_distance_mean": 1.1489060282707215,
      "weighted_auxiliary_loss": 0.0006503494747448713,
      "weighted_relation_loss": 0.0006503494747448713
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.456501403180543,
      "flow_matching_loss": 0.04434061422944069,
      "grad_norm": 0.13934852182865143,
      "learning_rate": 8.370562149149665e-05,
      "loss": 0.045,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00069877709611319,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.689387321472168,
      "state_query_norm": 1223.3417846679688,
      "state_relation_loss": 0.00069877709611319,
      "step": 9040,
      "student_distance_mean": 1647.4610717773437,
      "teacher_distance_mean": 1.1427777528762817,
      "weighted_auxiliary_loss": 0.00069877709611319,
      "weighted_relation_loss": 0.00069877709611319
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.465855940130963,
      "flow_matching_loss": 0.044606448337435725,
      "grad_norm": 0.13046249747276306,
      "learning_rate": 8.366489101043989e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007158930879086256,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.445627975463868,
      "state_query_norm": 1249.6479370117188,
      "state_relation_loss": 0.0007158930879086256,
      "step": 9050,
      "student_distance_mean": 1687.1848022460938,
      "teacher_distance_mean": 1.162188673019409,
      "weighted_auxiliary_loss": 0.0007158930879086256,
      "weighted_relation_loss": 0.0007158930879086256
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.475210477081385,
      "flow_matching_loss": 0.0431698739528656,
      "grad_norm": 0.10137064754962921,
      "learning_rate": 8.362411962336614e-05,
      "loss": 0.0439,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007089614111464471,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.85168399810791,
      "state_query_norm": 1219.69609375,
      "state_relation_loss": 0.0007089614111464471,
      "step": 9060,
      "student_distance_mean": 1654.1288818359376,
      "teacher_distance_mean": 1.1486578702926635,
      "weighted_auxiliary_loss": 0.0007089614111464471,
      "weighted_relation_loss": 0.0007089614111464471
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.484565014031805,
      "flow_matching_loss": 0.04718571230769157,
      "grad_norm": 0.13675062358379364,
      "learning_rate": 8.358330737981651e-05,
      "loss": 0.0479,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006827022123616188,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.14477405548096,
      "state_query_norm": 1238.8401000976562,
      "state_relation_loss": 0.0006827022123616188,
      "step": 9070,
      "student_distance_mean": 1668.2357543945313,
      "teacher_distance_mean": 1.1523144960403442,
      "weighted_auxiliary_loss": 0.0006827022123616188,
      "weighted_relation_loss": 0.0006827022123616188
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.493919550982227,
      "flow_matching_loss": 0.046292032673954965,
      "grad_norm": 0.13084730505943298,
      "learning_rate": 8.35424543293817e-05,
      "loss": 0.047,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007122288981918246,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.478404426574706,
      "state_query_norm": 1250.9433227539062,
      "state_relation_loss": 0.0007122288981918246,
      "step": 9080,
      "student_distance_mean": 1686.2260986328124,
      "teacher_distance_mean": 1.1549646139144898,
      "weighted_auxiliary_loss": 0.0007122288981918246,
      "weighted_relation_loss": 0.0007122288981918246
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.503274087932647,
      "flow_matching_loss": 0.045025890693068504,
      "grad_norm": 0.17953085899353027,
      "learning_rate": 8.350156052170206e-05,
      "loss": 0.0457,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007066189544275403,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.216673851013184,
      "state_query_norm": 1249.8221801757813,
      "state_relation_loss": 0.0007066189544275403,
      "step": 9090,
      "student_distance_mean": 1679.1845092773438,
      "teacher_distance_mean": 1.1517578840255738,
      "weighted_auxiliary_loss": 0.0007066189544275403,
      "weighted_relation_loss": 0.0007066189544275403
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.512628624883067,
      "flow_matching_loss": 0.04511316679418087,
      "grad_norm": 0.0982985645532608,
      "learning_rate": 8.346062600646738e-05,
      "loss": 0.0458,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007228795671835542,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.393585777282716,
      "state_query_norm": 1243.5139526367188,
      "state_relation_loss": 0.0007228795671835542,
      "step": 9100,
      "student_distance_mean": 1681.0855712890625,
      "teacher_distance_mean": 1.1546444177627564,
      "weighted_auxiliary_loss": 0.0007228795671835542,
      "weighted_relation_loss": 0.0007228795671835542
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.52198316183349,
      "flow_matching_loss": 0.0492880504578352,
      "grad_norm": 0.15754158794879913,
      "learning_rate": 8.341965083341697e-05,
      "loss": 0.05,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007069197192322463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.39359474182129,
      "state_query_norm": 1251.8106079101562,
      "state_relation_loss": 0.0007069197192322463,
      "step": 9110,
      "student_distance_mean": 1683.9615966796875,
      "teacher_distance_mean": 1.1535227656364442,
      "weighted_auxiliary_loss": 0.0007069197192322463,
      "weighted_relation_loss": 0.0007069197192322463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.53133769878391,
      "flow_matching_loss": 0.04558880142867565,
      "grad_norm": 0.11972485482692719,
      "learning_rate": 8.337863505233953e-05,
      "loss": 0.0463,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006618430372327567,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.148429298400877,
      "state_query_norm": 1240.89013671875,
      "state_relation_loss": 0.0006618430372327567,
      "step": 9120,
      "student_distance_mean": 1669.8823486328124,
      "teacher_distance_mean": 1.1487578630447388,
      "weighted_auxiliary_loss": 0.0006618430372327567,
      "weighted_relation_loss": 0.0006618430372327567
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.540692235734332,
      "flow_matching_loss": 0.04638255499303341,
      "grad_norm": 0.15184268355369568,
      "learning_rate": 8.33375787130731e-05,
      "loss": 0.0471,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006974261661525816,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.81006565093994,
      "state_query_norm": 1253.5108276367187,
      "state_relation_loss": 0.0006974261661525816,
      "step": 9130,
      "student_distance_mean": 1703.7082275390626,
      "teacher_distance_mean": 1.166794729232788,
      "weighted_auxiliary_loss": 0.0006974261661525816,
      "weighted_relation_loss": 0.0006974261661525816
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.550046772684752,
      "flow_matching_loss": 0.04500385038554668,
      "grad_norm": 0.16149012744426727,
      "learning_rate": 8.3296481865505e-05,
      "loss": 0.0457,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007226287387311458,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.306458473205566,
      "state_query_norm": 1242.9006591796874,
      "state_relation_loss": 0.0007226287387311458,
      "step": 9140,
      "student_distance_mean": 1677.0064697265625,
      "teacher_distance_mean": 1.1478225231170653,
      "weighted_auxiliary_loss": 0.0007226287387311458,
      "weighted_relation_loss": 0.0007226287387311458
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.559401309635174,
      "flow_matching_loss": 0.04469682946801186,
      "grad_norm": 0.13259357213974,
      "learning_rate": 8.325534455957181e-05,
      "loss": 0.0454,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006901349464897066,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.50434055328369,
      "state_query_norm": 1256.543798828125,
      "state_relation_loss": 0.0006901349464897066,
      "step": 9150,
      "student_distance_mean": 1690.9140014648438,
      "teacher_distance_mean": 1.1529470801353454,
      "weighted_auxiliary_loss": 0.0006901349464897066,
      "weighted_relation_loss": 0.0006901349464897066
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.568755846585594,
      "flow_matching_loss": 0.04460294544696808,
      "grad_norm": 0.169998437166214,
      "learning_rate": 8.321416684525917e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006922222499269992,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.302963638305663,
      "state_query_norm": 1249.3396240234374,
      "state_relation_loss": 0.0006922222499269992,
      "step": 9160,
      "student_distance_mean": 1677.9263061523438,
      "teacher_distance_mean": 1.1455645561218262,
      "weighted_auxiliary_loss": 0.0006922222499269992,
      "weighted_relation_loss": 0.0006922222499269992
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.578110383536014,
      "flow_matching_loss": 0.04611380584537983,
      "grad_norm": 0.17252174019813538,
      "learning_rate": 8.317294877260193e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007042325276415795,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.700997734069823,
      "state_query_norm": 1251.9987915039062,
      "state_relation_loss": 0.0007042325276415795,
      "step": 9170,
      "student_distance_mean": 1696.0326538085938,
      "teacher_distance_mean": 1.1598599553108215,
      "weighted_auxiliary_loss": 0.0007042325276415795,
      "weighted_relation_loss": 0.0007042325276415795
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.587464920486436,
      "flow_matching_loss": 0.04379087910056114,
      "grad_norm": 0.11502914130687714,
      "learning_rate": 8.313169039168394e-05,
      "loss": 0.0445,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006605107628274709,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.49268207550049,
      "state_query_norm": 1253.4849609375,
      "state_relation_loss": 0.0006605107628274709,
      "step": 9180,
      "student_distance_mean": 1691.2051879882813,
      "teacher_distance_mean": 1.1555001497268678,
      "weighted_auxiliary_loss": 0.0006605107628274709,
      "weighted_relation_loss": 0.0006605107628274709
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.596819457436856,
      "flow_matching_loss": 0.04591975323855877,
      "grad_norm": 0.12483374774456024,
      "learning_rate": 8.3090391752638e-05,
      "loss": 0.0466,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006740771990735084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.14549674987793,
      "state_query_norm": 1242.6259765625,
      "state_relation_loss": 0.0006740771990735084,
      "step": 9190,
      "student_distance_mean": 1672.36298828125,
      "teacher_distance_mean": 1.1437472581863404,
      "weighted_auxiliary_loss": 0.0006740771990735084,
      "weighted_relation_loss": 0.0006740771990735084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.606173994387278,
      "flow_matching_loss": 0.043821733072400094,
      "grad_norm": 0.16362646222114563,
      "learning_rate": 8.304905290564586e-05,
      "loss": 0.0445,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006615884543862194,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.269782066345215,
      "state_query_norm": 1250.6205078125,
      "state_relation_loss": 0.0006615884543862194,
      "step": 9200,
      "student_distance_mean": 1676.4725708007813,
      "teacher_distance_mean": 1.1428682208061218,
      "weighted_auxiliary_loss": 0.0006615884543862194,
      "weighted_relation_loss": 0.0006615884543862194
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.615528531337699,
      "flow_matching_loss": 0.04274322390556336,
      "grad_norm": 0.1899222582578659,
      "learning_rate": 8.300767390093813e-05,
      "loss": 0.0434,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000692516251001507,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.053717422485352,
      "state_query_norm": 1269.2151733398437,
      "state_relation_loss": 0.000692516251001507,
      "step": 9210,
      "student_distance_mean": 1718.8399658203125,
      "teacher_distance_mean": 1.1607048630714416,
      "weighted_auxiliary_loss": 0.000692516251001507,
      "weighted_relation_loss": 0.000692516251001507
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.62488306828812,
      "flow_matching_loss": 0.04458607621490955,
      "grad_norm": 0.13239659368991852,
      "learning_rate": 8.296625478879417e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006656400219071656,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.316280364990234,
      "state_query_norm": 1243.4053344726562,
      "state_relation_loss": 0.0006656400219071656,
      "step": 9220,
      "student_distance_mean": 1678.4694702148438,
      "teacher_distance_mean": 1.1399323344230652,
      "weighted_auxiliary_loss": 0.0006656400219071656,
      "weighted_relation_loss": 0.0006656400219071656
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.63423760523854,
      "flow_matching_loss": 0.04459172636270523,
      "grad_norm": 0.10657663643360138,
      "learning_rate": 8.292479561954214e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007076329726260156,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.35756645202637,
      "state_query_norm": 1243.0371948242187,
      "state_relation_loss": 0.0007076329726260156,
      "step": 9230,
      "student_distance_mean": 1679.1274047851562,
      "teacher_distance_mean": 1.1546016454696655,
      "weighted_auxiliary_loss": 0.0007076329726260156,
      "weighted_relation_loss": 0.0007076329726260156
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.643592142188961,
      "flow_matching_loss": 0.04470423199236393,
      "grad_norm": 0.11779104173183441,
      "learning_rate": 8.288329644355883e-05,
      "loss": 0.0454,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006541577691677958,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.15251178741455,
      "state_query_norm": 1238.5758911132812,
      "state_relation_loss": 0.0006541577691677958,
      "step": 9240,
      "student_distance_mean": 1671.6395385742187,
      "teacher_distance_mean": 1.151599609851837,
      "weighted_auxiliary_loss": 0.0006541577691677958,
      "weighted_relation_loss": 0.0006541577691677958
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.652946679139383,
      "flow_matching_loss": 0.04529704563319683,
      "grad_norm": 0.13223248720169067,
      "learning_rate": 8.284175731126965e-05,
      "loss": 0.046,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006827078235801309,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 29.99404296875,
      "state_query_norm": 1239.9388305664063,
      "state_relation_loss": 0.0006827078235801309,
      "step": 9250,
      "student_distance_mean": 1665.532421875,
      "teacher_distance_mean": 1.1370399355888368,
      "weighted_auxiliary_loss": 0.0006827078235801309,
      "weighted_relation_loss": 0.0006827078235801309
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.662301216089803,
      "flow_matching_loss": 0.04332416988909245,
      "grad_norm": 0.1088465228676796,
      "learning_rate": 8.280017827314854e-05,
      "loss": 0.044,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006757777417078614,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.346651458740233,
      "state_query_norm": 1251.1588134765625,
      "state_relation_loss": 0.0006757777417078614,
      "step": 9260,
      "student_distance_mean": 1682.1444580078125,
      "teacher_distance_mean": 1.1538326025009156,
      "weighted_auxiliary_loss": 0.0006757777417078614,
      "weighted_relation_loss": 0.0006757777417078614
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.671655753040225,
      "flow_matching_loss": 0.046555579826235774,
      "grad_norm": 0.12758447229862213,
      "learning_rate": 8.2758559379718e-05,
      "loss": 0.0472,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006567743956111371,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.491870307922362,
      "state_query_norm": 1259.5972534179687,
      "state_relation_loss": 0.0006567743956111371,
      "step": 9270,
      "student_distance_mean": 1692.3914916992187,
      "teacher_distance_mean": 1.1501836895942688,
      "weighted_auxiliary_loss": 0.0006567743956111371,
      "weighted_relation_loss": 0.0006567743956111371
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.681010289990645,
      "flow_matching_loss": 0.04588857032358647,
      "grad_norm": 0.12032122164964676,
      "learning_rate": 8.271690068154887e-05,
      "loss": 0.0466,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000681693886872381,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.680109977722168,
      "state_query_norm": 1251.9789672851562,
      "state_relation_loss": 0.000681693886872381,
      "step": 9280,
      "student_distance_mean": 1695.9334228515625,
      "teacher_distance_mean": 1.155985414981842,
      "weighted_auxiliary_loss": 0.000681693886872381,
      "weighted_relation_loss": 0.000681693886872381
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.690364826941066,
      "flow_matching_loss": 0.04446878582239151,
      "grad_norm": 0.16520124673843384,
      "learning_rate": 8.26752022292604e-05,
      "loss": 0.0451,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006723255093675107,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.40771141052246,
      "state_query_norm": 1247.1486938476562,
      "state_relation_loss": 0.0006723255093675107,
      "step": 9290,
      "student_distance_mean": 1685.787353515625,
      "teacher_distance_mean": 1.1480619311332703,
      "weighted_auxiliary_loss": 0.0006723255093675107,
      "weighted_relation_loss": 0.0006723255093675107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.699719363891488,
      "flow_matching_loss": 0.04325496889650822,
      "grad_norm": 0.13704951107501984,
      "learning_rate": 8.263346407352018e-05,
      "loss": 0.0439,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006741501740179956,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.338766860961915,
      "state_query_norm": 1247.3259033203126,
      "state_relation_loss": 0.0006741501740179956,
      "step": 9300,
      "student_distance_mean": 1680.9450561523438,
      "teacher_distance_mean": 1.1383753299713135,
      "weighted_auxiliary_loss": 0.0006741501740179956,
      "weighted_relation_loss": 0.0006741501740179956
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.709073900841908,
      "flow_matching_loss": 0.04552045129239559,
      "grad_norm": 0.15557539463043213,
      "learning_rate": 8.259168626504395e-05,
      "loss": 0.0462,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007052890199702233,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.73412628173828,
      "state_query_norm": 1254.096923828125,
      "state_relation_loss": 0.0007052890199702233,
      "step": 9310,
      "student_distance_mean": 1700.0581909179687,
      "teacher_distance_mean": 1.1655097723007202,
      "weighted_auxiliary_loss": 0.0007052890199702233,
      "weighted_relation_loss": 0.0007052890199702233
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.71842843779233,
      "flow_matching_loss": 0.04554929099977016,
      "grad_norm": 0.13101881742477417,
      "learning_rate": 8.254986885459572e-05,
      "loss": 0.0462,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006945288681890815,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.761048126220704,
      "state_query_norm": 1265.1921752929688,
      "state_relation_loss": 0.0006945288681890815,
      "step": 9320,
      "student_distance_mean": 1705.2057006835937,
      "teacher_distance_mean": 1.159329903125763,
      "weighted_auxiliary_loss": 0.0006945288681890815,
      "weighted_relation_loss": 0.0006945288681890815
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.72778297474275,
      "flow_matching_loss": 0.04578020349144936,
      "grad_norm": 0.12621721625328064,
      "learning_rate": 8.250801189298757e-05,
      "loss": 0.0465,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007001327932812274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.176651573181154,
      "state_query_norm": 1237.2245483398438,
      "state_relation_loss": 0.0007001327932812274,
      "step": 9330,
      "student_distance_mean": 1667.5197631835938,
      "teacher_distance_mean": 1.1406297922134399,
      "weighted_auxiliary_loss": 0.0007001327932812274,
      "weighted_relation_loss": 0.0007001327932812274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.737137511693172,
      "flow_matching_loss": 0.04349540285766125,
      "grad_norm": 0.14987149834632874,
      "learning_rate": 8.246611543107967e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006727960251737386,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.37312412261963,
      "state_query_norm": 1249.8369873046875,
      "state_relation_loss": 0.0006727960251737386,
      "step": 9340,
      "student_distance_mean": 1684.9757568359375,
      "teacher_distance_mean": 1.1515897154808044,
      "weighted_auxiliary_loss": 0.0006727960251737386,
      "weighted_relation_loss": 0.0006727960251737386
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.746492048643592,
      "flow_matching_loss": 0.044591257348656654,
      "grad_norm": 0.12237004190683365,
      "learning_rate": 8.242417951978018e-05,
      "loss": 0.0453,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006999209173955024,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.936956787109374,
      "state_query_norm": 1264.8190063476563,
      "state_relation_loss": 0.0006999209173955024,
      "step": 9350,
      "student_distance_mean": 1709.462060546875,
      "teacher_distance_mean": 1.1603633403778075,
      "weighted_auxiliary_loss": 0.0006999209173955024,
      "weighted_relation_loss": 0.0006999209173955024
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.755846585594012,
      "flow_matching_loss": 0.04481491446495056,
      "grad_norm": 0.13779331743717194,
      "learning_rate": 8.238220421004517e-05,
      "loss": 0.0455,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007029718486592174,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.231536293029784,
      "state_query_norm": 1243.0627807617188,
      "state_relation_loss": 0.0007029718486592174,
      "step": 9360,
      "student_distance_mean": 1675.8867431640624,
      "teacher_distance_mean": 1.137600588798523,
      "weighted_auxiliary_loss": 0.0007029718486592174,
      "weighted_relation_loss": 0.0007029718486592174
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.765201122544434,
      "flow_matching_loss": 0.04616307392716408,
      "grad_norm": 0.144220232963562,
      "learning_rate": 8.234018955287859e-05,
      "loss": 0.0468,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000668482220498845,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.52797679901123,
      "state_query_norm": 1258.9203857421876,
      "state_relation_loss": 0.000668482220498845,
      "step": 9370,
      "student_distance_mean": 1690.969482421875,
      "teacher_distance_mean": 1.1471747636795044,
      "weighted_auxiliary_loss": 0.000668482220498845,
      "weighted_relation_loss": 0.000668482220498845
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.774555659494855,
      "flow_matching_loss": 0.04369700849056244,
      "grad_norm": 0.125376895070076,
      "learning_rate": 8.229813559933224e-05,
      "loss": 0.0444,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006703424907755107,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.29770698547363,
      "state_query_norm": 1232.4867797851562,
      "state_relation_loss": 0.0006703424907755107,
      "step": 9380,
      "student_distance_mean": 1672.7048583984374,
      "teacher_distance_mean": 1.136982011795044,
      "weighted_auxiliary_loss": 0.0006703424907755107,
      "weighted_relation_loss": 0.0006703424907755107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.783910196445277,
      "flow_matching_loss": 0.04735737778246403,
      "grad_norm": 0.12021992355585098,
      "learning_rate": 8.22560424005056e-05,
      "loss": 0.0481,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000701923226006329,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.535723876953124,
      "state_query_norm": 1264.0624267578125,
      "state_relation_loss": 0.000701923226006329,
      "step": 9390,
      "student_distance_mean": 1694.9343383789062,
      "teacher_distance_mean": 1.1544134497642518,
      "weighted_auxiliary_loss": 0.000701923226006329,
      "weighted_relation_loss": 0.000701923226006329
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.793264733395697,
      "flow_matching_loss": 0.043064073845744136,
      "grad_norm": 0.12301108241081238,
      "learning_rate": 8.22139100075459e-05,
      "loss": 0.0437,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006742972356732935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.79334373474121,
      "state_query_norm": 1258.8289428710937,
      "state_relation_loss": 0.0006742972356732935,
      "step": 9400,
      "student_distance_mean": 1704.0615478515624,
      "teacher_distance_mean": 1.1578336000442504,
      "weighted_auxiliary_loss": 0.0006742972356732935,
      "weighted_relation_loss": 0.0006742972356732935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.802619270346117,
      "flow_matching_loss": 0.04455126002430916,
      "grad_norm": 0.12569309771060944,
      "learning_rate": 8.217173847164799e-05,
      "loss": 0.0452,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006647712783887982,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.02418632507324,
      "state_query_norm": 1279.5185424804688,
      "state_relation_loss": 0.0006647712783887982,
      "step": 9410,
      "student_distance_mean": 1722.0360717773438,
      "teacher_distance_mean": 1.1575594902038575,
      "weighted_auxiliary_loss": 0.0006647712783887982,
      "weighted_relation_loss": 0.0006647712783887982
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.811973807296539,
      "flow_matching_loss": 0.04358151257038116,
      "grad_norm": 0.11178319901227951,
      "learning_rate": 8.212952784405422e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006535143184009939,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.03263053894043,
      "state_query_norm": 1266.11494140625,
      "state_relation_loss": 0.0006535143184009939,
      "step": 9420,
      "student_distance_mean": 1716.7060668945312,
      "teacher_distance_mean": 1.1554224133491515,
      "weighted_auxiliary_loss": 0.0006535143184009939,
      "weighted_relation_loss": 0.0006535143184009939
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.82132834424696,
      "flow_matching_loss": 0.04302969872951508,
      "grad_norm": 0.10751693695783615,
      "learning_rate": 8.208727817605451e-05,
      "loss": 0.0437,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006510165927466005,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.372972297668458,
      "state_query_norm": 1285.1647583007812,
      "state_relation_loss": 0.0006510165927466005,
      "step": 9430,
      "student_distance_mean": 1736.5140502929687,
      "teacher_distance_mean": 1.1750047206878662,
      "weighted_auxiliary_loss": 0.0006510165927466005,
      "weighted_relation_loss": 0.0006510165927466005
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.830682881197381,
      "flow_matching_loss": 0.044520004093647,
      "grad_norm": 0.1107029914855957,
      "learning_rate": 8.204498951898618e-05,
      "loss": 0.0452,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006562496942933649,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.67365951538086,
      "state_query_norm": 1259.2129028320312,
      "state_relation_loss": 0.0006562496942933649,
      "step": 9440,
      "student_distance_mean": 1698.3043090820313,
      "teacher_distance_mean": 1.1503455996513368,
      "weighted_auxiliary_loss": 0.0006562496942933649,
      "weighted_relation_loss": 0.0006562496942933649
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.840037418147801,
      "flow_matching_loss": 0.04410296492278576,
      "grad_norm": 0.12267929315567017,
      "learning_rate": 8.200266192423395e-05,
      "loss": 0.0447,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006275797379203141,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.73872318267822,
      "state_query_norm": 1266.6413452148438,
      "state_relation_loss": 0.0006275797379203141,
      "step": 9450,
      "student_distance_mean": 1704.2194213867188,
      "teacher_distance_mean": 1.1472239017486572,
      "weighted_auxiliary_loss": 0.0006275797379203141,
      "weighted_relation_loss": 0.0006275797379203141
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.849391955098223,
      "flow_matching_loss": 0.04319850206375122,
      "grad_norm": 0.14437811076641083,
      "learning_rate": 8.196029544322981e-05,
      "loss": 0.0438,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006508171558380127,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.684038352966308,
      "state_query_norm": 1260.4023681640624,
      "state_relation_loss": 0.0006508171558380127,
      "step": 9460,
      "student_distance_mean": 1700.1608520507812,
      "teacher_distance_mean": 1.1490601658821107,
      "weighted_auxiliary_loss": 0.0006508171558380127,
      "weighted_relation_loss": 0.0006508171558380127
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.858746492048644,
      "flow_matching_loss": 0.04522765204310417,
      "grad_norm": 0.14147445559501648,
      "learning_rate": 8.191789012745305e-05,
      "loss": 0.0459,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000630235968856141,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.42685546875,
      "state_query_norm": 1250.0712524414062,
      "state_relation_loss": 0.000630235968856141,
      "step": 9470,
      "student_distance_mean": 1685.4239868164063,
      "teacher_distance_mean": 1.1430324554443358,
      "weighted_auxiliary_loss": 0.000630235968856141,
      "weighted_relation_loss": 0.000630235968856141
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.868101028999064,
      "flow_matching_loss": 0.04410983845591545,
      "grad_norm": 0.12467405200004578,
      "learning_rate": 8.187544602843014e-05,
      "loss": 0.0448,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006465188111178577,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.29993782043457,
      "state_query_norm": 1252.3714965820313,
      "state_relation_loss": 0.0006465188111178577,
      "step": 9480,
      "student_distance_mean": 1680.0638305664063,
      "teacher_distance_mean": 1.1366968750953674,
      "weighted_auxiliary_loss": 0.0006465188111178577,
      "weighted_relation_loss": 0.0006465188111178577
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.877455565949486,
      "flow_matching_loss": 0.04511747136712074,
      "grad_norm": 0.11856278032064438,
      "learning_rate": 8.183296319773466e-05,
      "loss": 0.0457,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006265515345148742,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.12976245880127,
      "state_query_norm": 1238.1589233398438,
      "state_relation_loss": 0.0006265515345148742,
      "step": 9490,
      "student_distance_mean": 1669.830029296875,
      "teacher_distance_mean": 1.1305497288703918,
      "weighted_auxiliary_loss": 0.0006265515345148742,
      "weighted_relation_loss": 0.0006265515345148742
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.886810102899906,
      "flow_matching_loss": 0.042937219887971875,
      "grad_norm": 0.13436263799667358,
      "learning_rate": 8.179044168698721e-05,
      "loss": 0.0436,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006542315299157053,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.091695022583007,
      "state_query_norm": 1236.4875366210938,
      "state_relation_loss": 0.0006542315299157053,
      "step": 9500,
      "student_distance_mean": 1668.4586303710937,
      "teacher_distance_mean": 1.1302085638046264,
      "weighted_auxiliary_loss": 0.0006542315299157053,
      "weighted_relation_loss": 0.0006542315299157053
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.896164639850328,
      "flow_matching_loss": 0.041662536934018134,
      "grad_norm": 0.11506451666355133,
      "learning_rate": 8.174788154785548e-05,
      "loss": 0.0423,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006803661643061786,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.616520500183107,
      "state_query_norm": 1266.478515625,
      "state_relation_loss": 0.0006803661643061786,
      "step": 9510,
      "student_distance_mean": 1698.5646850585938,
      "teacher_distance_mean": 1.150040292739868,
      "weighted_auxiliary_loss": 0.0006803661643061786,
      "weighted_relation_loss": 0.0006803661643061786
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.905519176800748,
      "flow_matching_loss": 0.042617413029074666,
      "grad_norm": 0.16260512173175812,
      "learning_rate": 8.170528283205403e-05,
      "loss": 0.0433,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006833866937085986,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.443201637268068,
      "state_query_norm": 1289.8753173828125,
      "state_relation_loss": 0.0006833866937085986,
      "step": 9520,
      "student_distance_mean": 1740.427392578125,
      "teacher_distance_mean": 1.1640980958938598,
      "weighted_auxiliary_loss": 0.0006833866937085986,
      "weighted_relation_loss": 0.0006833866937085986
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.91487371375117,
      "flow_matching_loss": 0.045046129077672956,
      "grad_norm": 0.1314931958913803,
      "learning_rate": 8.166264559134435e-05,
      "loss": 0.0457,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006830926693510264,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.158213424682618,
      "state_query_norm": 1270.9246459960937,
      "state_relation_loss": 0.0006830926693510264,
      "step": 9530,
      "student_distance_mean": 1722.4886474609375,
      "teacher_distance_mean": 1.1568221926689148,
      "weighted_auxiliary_loss": 0.0006830926693510264,
      "weighted_relation_loss": 0.0006830926693510264
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.92422825070159,
      "flow_matching_loss": 0.043263446912169455,
      "grad_norm": 0.1383940577507019,
      "learning_rate": 8.161996987753467e-05,
      "loss": 0.044,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007077262329403311,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.92098274230957,
      "state_query_norm": 1268.2640014648437,
      "state_relation_loss": 0.0007077262329403311,
      "step": 9540,
      "student_distance_mean": 1712.7908447265625,
      "teacher_distance_mean": 1.156819260120392,
      "weighted_auxiliary_loss": 0.0007077262329403311,
      "weighted_relation_loss": 0.0007077262329403311
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.93358278765201,
      "flow_matching_loss": 0.044919416308403015,
      "grad_norm": 0.13318809866905212,
      "learning_rate": 8.157725574248e-05,
      "loss": 0.0456,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006313674617558718,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.431596755981445,
      "state_query_norm": 1246.3203247070312,
      "state_relation_loss": 0.0006313674617558718,
      "step": 9550,
      "student_distance_mean": 1683.9147705078126,
      "teacher_distance_mean": 1.148082196712494,
      "weighted_auxiliary_loss": 0.0006313674617558718,
      "weighted_relation_loss": 0.0006313674617558718
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.942937324602433,
      "flow_matching_loss": 0.046249279752373695,
      "grad_norm": 0.12246598303318024,
      "learning_rate": 8.153450323808209e-05,
      "loss": 0.0469,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006645103101618588,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.773986434936525,
      "state_query_norm": 1256.4687866210938,
      "state_relation_loss": 0.0006645103101618588,
      "step": 9560,
      "student_distance_mean": 1703.4568481445312,
      "teacher_distance_mean": 1.1541512250900268,
      "weighted_auxiliary_loss": 0.0006645103101618588,
      "weighted_relation_loss": 0.0006645103101618588
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.952291861552853,
      "flow_matching_loss": 0.042257055267691614,
      "grad_norm": 0.16277098655700684,
      "learning_rate": 8.149171241628923e-05,
      "loss": 0.043,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000710747775156051,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.787689018249512,
      "state_query_norm": 1261.0209350585938,
      "state_relation_loss": 0.000710747775156051,
      "step": 9570,
      "student_distance_mean": 1703.02763671875,
      "teacher_distance_mean": 1.1523284912109375,
      "weighted_auxiliary_loss": 0.000710747775156051,
      "weighted_relation_loss": 0.000710747775156051
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.961646398503275,
      "flow_matching_loss": 0.044416191428899764,
      "grad_norm": 0.20015689730644226,
      "learning_rate": 8.144888332909631e-05,
      "loss": 0.0451,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0007302799262106418,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.14699821472168,
      "state_query_norm": 1269.153466796875,
      "state_relation_loss": 0.0007302799262106418,
      "step": 9580,
      "student_distance_mean": 1722.36455078125,
      "teacher_distance_mean": 1.1656362652778625,
      "weighted_auxiliary_loss": 0.0007302799262106418,
      "weighted_relation_loss": 0.0007302799262106418
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.971000935453695,
      "flow_matching_loss": 0.04480639770627022,
      "grad_norm": 0.1874643862247467,
      "learning_rate": 8.14060160285447e-05,
      "loss": 0.0455,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006652112351730466,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.907257652282716,
      "state_query_norm": 1258.1495361328125,
      "state_relation_loss": 0.0006652112351730466,
      "step": 9590,
      "student_distance_mean": 1707.0703247070312,
      "teacher_distance_mean": 1.155396056175232,
      "weighted_auxiliary_loss": 0.0006652112351730466,
      "weighted_relation_loss": 0.0006652112351730466
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.980355472404115,
      "flow_matching_loss": 0.04327602572739124,
      "grad_norm": 0.15458886325359344,
      "learning_rate": 8.136311056672223e-05,
      "loss": 0.0439,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006709240493364632,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.875215911865233,
      "state_query_norm": 1266.2217651367187,
      "state_relation_loss": 0.0006709240493364632,
      "step": 9600,
      "student_distance_mean": 1712.3776123046875,
      "teacher_distance_mean": 1.1544523119926453,
      "weighted_auxiliary_loss": 0.0006709240493364632,
      "weighted_relation_loss": 0.0006709240493364632
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.989710009354537,
      "flow_matching_loss": 0.04278177134692669,
      "grad_norm": 0.14550568163394928,
      "learning_rate": 8.132016699576308e-05,
      "loss": 0.0434,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006530496990308166,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.892094993591307,
      "state_query_norm": 1272.47177734375,
      "state_relation_loss": 0.0006530496990308166,
      "step": 9610,
      "student_distance_mean": 1712.897509765625,
      "teacher_distance_mean": 1.159641683101654,
      "weighted_auxiliary_loss": 0.0006530496990308166,
      "weighted_relation_loss": 0.0006530496990308166
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 8.999064546304957,
      "flow_matching_loss": 0.04450306221842766,
      "grad_norm": 0.09546048939228058,
      "learning_rate": 8.127718536784771e-05,
      "loss": 0.0452,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006740988872479647,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.88813304901123,
      "state_query_norm": 1266.669921875,
      "state_relation_loss": 0.0006740988872479647,
      "step": 9620,
      "student_distance_mean": 1711.624755859375,
      "teacher_distance_mean": 1.1569618225097655,
      "weighted_auxiliary_loss": 0.0006740988872479647,
      "weighted_relation_loss": 0.0006740988872479647
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.00841908325538,
      "flow_matching_loss": 0.04315745569765568,
      "grad_norm": 0.12373901903629303,
      "learning_rate": 8.12341657352029e-05,
      "loss": 0.0438,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0006336303427815438,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 30.199437141418457,
      "state_query_norm": 1238.7754272460938,
      "state_relation_loss": 0.0006336303427815438,
      "step": 9630,
      "student_distance_mean": 1674.15234375,
      "teacher_distance_mean": 1.1327766299247741,
      "weighted_auxiliary_loss": 0.0006336303427815438,
      "weighted_relation_loss": 0.0006336303427815438
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.0177736202058,
      "flow_matching_loss": 0.042226272821426394,
      "grad_norm": 0.1281135231256485,
      "learning_rate": 8.119110815010151e-05,
      "loss": 0.0429,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006487023551017046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.00098400115967,
      "state_query_norm": 1272.6696166992188,
      "state_relation_loss": 0.0006487023551017046,
      "step": 9640,
      "student_distance_mean": 1719.7667236328125,
      "teacher_distance_mean": 1.1503705978393555,
      "weighted_auxiliary_loss": 0.0006487023551017046,
      "weighted_relation_loss": 0.0006487023551017046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.027128157156222,
      "flow_matching_loss": 0.04341490566730499,
      "grad_norm": 0.11115766316652298,
      "learning_rate": 8.114801266486261e-05,
      "loss": 0.0441,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006367563153617084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.493419075012206,
      "state_query_norm": 1244.7033447265626,
      "state_relation_loss": 0.0006367563153617084,
      "step": 9650,
      "student_distance_mean": 1687.1348999023437,
      "teacher_distance_mean": 1.138063871860504,
      "weighted_auxiliary_loss": 0.0006367563153617084,
      "weighted_relation_loss": 0.0006367563153617084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.036482694106642,
      "flow_matching_loss": 0.043617401644587515,
      "grad_norm": 0.15002799034118652,
      "learning_rate": 8.110487933185122e-05,
      "loss": 0.0443,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006570875993929803,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.67475299835205,
      "state_query_norm": 1262.9828125,
      "state_relation_loss": 0.0006570875993929803,
      "step": 9660,
      "student_distance_mean": 1701.4732055664062,
      "teacher_distance_mean": 1.1549769639968872,
      "weighted_auxiliary_loss": 0.0006570875993929803,
      "weighted_relation_loss": 0.0006570875993929803
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.045837231057062,
      "flow_matching_loss": 0.04222849942743778,
      "grad_norm": 0.158666729927063,
      "learning_rate": 8.106170820347847e-05,
      "loss": 0.0429,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006845917727332562,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.872846603393555,
      "state_query_norm": 1270.1118530273438,
      "state_relation_loss": 0.0006845917727332562,
      "step": 9670,
      "student_distance_mean": 1710.5386474609375,
      "teacher_distance_mean": 1.1475272297859191,
      "weighted_auxiliary_loss": 0.0006845917727332562,
      "weighted_relation_loss": 0.0006845917727332562
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.055191768007484,
      "flow_matching_loss": 0.04293194562196732,
      "grad_norm": 0.1469157487154007,
      "learning_rate": 8.101849933220132e-05,
      "loss": 0.0436,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006981622369494289,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.130269050598145,
      "state_query_norm": 1269.1040161132812,
      "state_relation_loss": 0.0006981622369494289,
      "step": 9680,
      "student_distance_mean": 1719.901708984375,
      "teacher_distance_mean": 1.1563839793205262,
      "weighted_auxiliary_loss": 0.0006981622369494289,
      "weighted_relation_loss": 0.0006981622369494289
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.064546304957904,
      "flow_matching_loss": 0.04263115376234054,
      "grad_norm": 0.1362462043762207,
      "learning_rate": 8.097525277052264e-05,
      "loss": 0.0433,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006697849545162171,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.46279697418213,
      "state_query_norm": 1283.8246826171876,
      "state_relation_loss": 0.0006697849545162171,
      "step": 9690,
      "student_distance_mean": 1741.9174194335938,
      "teacher_distance_mean": 1.1634058117866517,
      "weighted_auxiliary_loss": 0.0006697849545162171,
      "weighted_relation_loss": 0.0006697849545162171
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.073900841908326,
      "flow_matching_loss": 0.042507351562380793,
      "grad_norm": 0.1187974289059639,
      "learning_rate": 8.093196857099106e-05,
      "loss": 0.0432,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006766703154426068,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.01728935241699,
      "state_query_norm": 1263.4220825195312,
      "state_relation_loss": 0.0006766703154426068,
      "step": 9700,
      "student_distance_mean": 1715.0142578125,
      "teacher_distance_mean": 1.1440882802009582,
      "weighted_auxiliary_loss": 0.0006766703154426068,
      "weighted_relation_loss": 0.0006766703154426068
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.083255378858746,
      "flow_matching_loss": 0.04217822812497616,
      "grad_norm": 0.14329217374324799,
      "learning_rate": 8.088864678620096e-05,
      "loss": 0.0429,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006938945094589144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.81668186187744,
      "state_query_norm": 1259.8606201171874,
      "state_relation_loss": 0.0006938945094589144,
      "step": 9710,
      "student_distance_mean": 1703.4413452148438,
      "teacher_distance_mean": 1.1368178844451904,
      "weighted_auxiliary_loss": 0.0006938945094589144,
      "weighted_relation_loss": 0.0006938945094589144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.092609915809167,
      "flow_matching_loss": 0.04151851125061512,
      "grad_norm": 0.14017373323440552,
      "learning_rate": 8.084528746879243e-05,
      "loss": 0.0422,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006872840574942529,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.709693145751952,
      "state_query_norm": 1289.7929321289062,
      "state_relation_loss": 0.0006872840574942529,
      "step": 9720,
      "student_distance_mean": 1753.4612670898437,
      "teacher_distance_mean": 1.16261328458786,
      "weighted_auxiliary_loss": 0.0006872840574942529,
      "weighted_relation_loss": 0.0006872840574942529
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.101964452759589,
      "flow_matching_loss": 0.04068919010460377,
      "grad_norm": 0.10916679352521896,
      "learning_rate": 8.080189067145108e-05,
      "loss": 0.0414,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006731141300406307,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.14230880737305,
      "state_query_norm": 1267.2659790039063,
      "state_relation_loss": 0.0006731141300406307,
      "step": 9730,
      "student_distance_mean": 1720.2693359375,
      "teacher_distance_mean": 1.1491454124450684,
      "weighted_auxiliary_loss": 0.0006731141300406307,
      "weighted_relation_loss": 0.0006731141300406307
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.111318989710009,
      "flow_matching_loss": 0.04264632500708103,
      "grad_norm": 0.16430771350860596,
      "learning_rate": 8.075845644690813e-05,
      "loss": 0.0433,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006764221005141735,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.605856323242186,
      "state_query_norm": 1257.0862915039063,
      "state_relation_loss": 0.0006764221005141735,
      "step": 9740,
      "student_distance_mean": 1695.3241455078125,
      "teacher_distance_mean": 1.138247787952423,
      "weighted_auxiliary_loss": 0.0006764221005141735,
      "weighted_relation_loss": 0.0006764221005141735
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.12067352666043,
      "flow_matching_loss": 0.04088568650186062,
      "grad_norm": 0.11409866064786911,
      "learning_rate": 8.07149848479403e-05,
      "loss": 0.0415,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006381989049259573,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.128714179992677,
      "state_query_norm": 1265.3627075195313,
      "state_relation_loss": 0.0006381989049259573,
      "step": 9750,
      "student_distance_mean": 1716.7595581054688,
      "teacher_distance_mean": 1.1516259431838989,
      "weighted_auxiliary_loss": 0.0006381989049259573,
      "weighted_relation_loss": 0.0006381989049259573
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.130028063610851,
      "flow_matching_loss": 0.04369465857744217,
      "grad_norm": 0.14284445345401764,
      "learning_rate": 8.067147592736962e-05,
      "loss": 0.0444,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006946639041416347,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.364645957946777,
      "state_query_norm": 1268.9493530273437,
      "state_relation_loss": 0.0006946639041416347,
      "step": 9760,
      "student_distance_mean": 1727.9720703125,
      "teacher_distance_mean": 1.1525431513786315,
      "weighted_auxiliary_loss": 0.0006946639041416347,
      "weighted_relation_loss": 0.0006946639041416347
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.139382600561273,
      "flow_matching_loss": 0.041270026192069056,
      "grad_norm": 0.15870164334774017,
      "learning_rate": 8.062792973806357e-05,
      "loss": 0.0419,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006562753929756582,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.687466621398926,
      "state_query_norm": 1297.8641235351563,
      "state_relation_loss": 0.0006562753929756582,
      "step": 9770,
      "student_distance_mean": 1754.9553466796874,
      "teacher_distance_mean": 1.167230224609375,
      "weighted_auxiliary_loss": 0.0006562753929756582,
      "weighted_relation_loss": 0.0006562753929756582
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.148737137511693,
      "flow_matching_loss": 0.041693564131855966,
      "grad_norm": 0.18604987859725952,
      "learning_rate": 8.058434633293485e-05,
      "loss": 0.0424,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000663147569866851,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.747622299194337,
      "state_query_norm": 1249.2367797851562,
      "state_relation_loss": 0.000663147569866851,
      "step": 9780,
      "student_distance_mean": 1698.0700439453126,
      "teacher_distance_mean": 1.13895583152771,
      "weighted_auxiliary_loss": 0.000663147569866851,
      "weighted_relation_loss": 0.000663147569866851
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.158091674462113,
      "flow_matching_loss": 0.04107689820230007,
      "grad_norm": 0.14419694244861603,
      "learning_rate": 8.054072576494143e-05,
      "loss": 0.0417,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006357929029036314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.206426239013673,
      "state_query_norm": 1271.1227661132812,
      "state_relation_loss": 0.0006357929029036314,
      "step": 9790,
      "student_distance_mean": 1720.5864624023438,
      "teacher_distance_mean": 1.1479517340660095,
      "weighted_auxiliary_loss": 0.0006357929029036314,
      "weighted_relation_loss": 0.0006357929029036314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.167446211412535,
      "flow_matching_loss": 0.04336763881146908,
      "grad_norm": 0.1451789289712906,
      "learning_rate": 8.04970680870864e-05,
      "loss": 0.044,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006435157672967761,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.832576560974122,
      "state_query_norm": 1305.4216064453126,
      "state_relation_loss": 0.0006435157672967761,
      "step": 9800,
      "student_distance_mean": 1762.132421875,
      "teacher_distance_mean": 1.1744984626770019,
      "weighted_auxiliary_loss": 0.0006435157672967761,
      "weighted_relation_loss": 0.0006435157672967761
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.176800748362956,
      "flow_matching_loss": 0.04349592737853527,
      "grad_norm": 0.13548052310943604,
      "learning_rate": 8.045337335241793e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006641324609518052,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.85632495880127,
      "state_query_norm": 1271.66328125,
      "state_relation_loss": 0.0006641324609518052,
      "step": 9810,
      "student_distance_mean": 1705.1018188476562,
      "teacher_distance_mean": 1.1371872782707215,
      "weighted_auxiliary_loss": 0.0006641324609518052,
      "weighted_relation_loss": 0.0006641324609518052
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.186155285313378,
      "flow_matching_loss": 0.042365067452192304,
      "grad_norm": 0.16155371069908142,
      "learning_rate": 8.040964161402932e-05,
      "loss": 0.043,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006481553544290363,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.21950168609619,
      "state_query_norm": 1277.0816650390625,
      "state_relation_loss": 0.0006481553544290363,
      "step": 9820,
      "student_distance_mean": 1726.0714111328125,
      "teacher_distance_mean": 1.1554757475852966,
      "weighted_auxiliary_loss": 0.0006481553544290363,
      "weighted_relation_loss": 0.0006481553544290363
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.195509822263798,
      "flow_matching_loss": 0.04312896691262722,
      "grad_norm": 0.1509213000535965,
      "learning_rate": 8.036587292505869e-05,
      "loss": 0.0438,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006488408718723804,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.24388904571533,
      "state_query_norm": 1279.1415161132813,
      "state_relation_loss": 0.0006488408718723804,
      "step": 9830,
      "student_distance_mean": 1728.138232421875,
      "teacher_distance_mean": 1.1575557351112367,
      "weighted_auxiliary_loss": 0.0006488408718723804,
      "weighted_relation_loss": 0.0006488408718723804
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.204864359214218,
      "flow_matching_loss": 0.04321721829473972,
      "grad_norm": 0.11570801585912704,
      "learning_rate": 8.032206733868912e-05,
      "loss": 0.0439,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006646930647548288,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.516779136657714,
      "state_query_norm": 1298.4740600585938,
      "state_relation_loss": 0.0006646930647548288,
      "step": 9840,
      "student_distance_mean": 1744.8457885742187,
      "teacher_distance_mean": 1.1651877522468568,
      "weighted_auxiliary_loss": 0.0006646930647548288,
      "weighted_relation_loss": 0.0006646930647548288
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.21421889616464,
      "flow_matching_loss": 0.043351497128605844,
      "grad_norm": 0.12079368531703949,
      "learning_rate": 8.027822490814858e-05,
      "loss": 0.044,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006462867720983922,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.345148658752443,
      "state_query_norm": 1290.903173828125,
      "state_relation_loss": 0.0006462867720983922,
      "step": 9850,
      "student_distance_mean": 1734.901220703125,
      "teacher_distance_mean": 1.1597049355506897,
      "weighted_auxiliary_loss": 0.0006462867720983922,
      "weighted_relation_loss": 0.0006462867720983922
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.22357343311506,
      "flow_matching_loss": 0.04398294948041439,
      "grad_norm": 0.13945960998535156,
      "learning_rate": 8.023434568670972e-05,
      "loss": 0.0446,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006401668360922486,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.125455856323242,
      "state_query_norm": 1271.983056640625,
      "state_relation_loss": 0.0006401668360922486,
      "step": 9860,
      "student_distance_mean": 1721.8006225585937,
      "teacher_distance_mean": 1.1523326873779296,
      "weighted_auxiliary_loss": 0.0006401668360922486,
      "weighted_relation_loss": 0.0006401668360922486
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.232927970065482,
      "flow_matching_loss": 0.0431855134665966,
      "grad_norm": 0.118395134806633,
      "learning_rate": 8.019042972768992e-05,
      "loss": 0.0438,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006438149779569357,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.771267700195313,
      "state_query_norm": 1252.5525146484374,
      "state_relation_loss": 0.0006438149779569357,
      "step": 9870,
      "student_distance_mean": 1701.923828125,
      "teacher_distance_mean": 1.13990079164505,
      "weighted_auxiliary_loss": 0.0006438149779569357,
      "weighted_relation_loss": 0.0006438149779569357
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.242282507015902,
      "flow_matching_loss": 0.041560623049736026,
      "grad_norm": 0.1258152425289154,
      "learning_rate": 8.014647708445124e-05,
      "loss": 0.0422,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006668473477475345,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.889685821533202,
      "state_query_norm": 1259.8413696289062,
      "state_relation_loss": 0.0006668473477475345,
      "step": 9880,
      "student_distance_mean": 1707.0703247070312,
      "teacher_distance_mean": 1.140949547290802,
      "weighted_auxiliary_loss": 0.0006668473477475345,
      "weighted_relation_loss": 0.0006668473477475345
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.251637043966324,
      "flow_matching_loss": 0.04132894240319729,
      "grad_norm": 0.12208378314971924,
      "learning_rate": 8.010248781040027e-05,
      "loss": 0.042,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000666171609191224,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.611869239807127,
      "state_query_norm": 1291.3858642578125,
      "state_relation_loss": 0.000666171609191224,
      "step": 9890,
      "student_distance_mean": 1747.3298950195312,
      "teacher_distance_mean": 1.1697723984718322,
      "weighted_auxiliary_loss": 0.000666171609191224,
      "weighted_relation_loss": 0.000666171609191224
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.260991580916745,
      "flow_matching_loss": 0.0445284191519022,
      "grad_norm": 0.17654408514499664,
      "learning_rate": 8.005846195898814e-05,
      "loss": 0.0452,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006616875296458602,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.712086486816407,
      "state_query_norm": 1259.7589599609375,
      "state_relation_loss": 0.0006616875296458602,
      "step": 9900,
      "student_distance_mean": 1698.0638671875,
      "teacher_distance_mean": 1.131806993484497,
      "weighted_auxiliary_loss": 0.0006616875296458602,
      "weighted_relation_loss": 0.0006616875296458602
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.270346117867165,
      "flow_matching_loss": 0.041373286023736,
      "grad_norm": 0.20139889419078827,
      "learning_rate": 8.00143995837104e-05,
      "loss": 0.042,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006310939905233681,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.05142250061035,
      "state_query_norm": 1284.146435546875,
      "state_relation_loss": 0.0006310939905233681,
      "step": 9910,
      "student_distance_mean": 1719.5724487304688,
      "teacher_distance_mean": 1.143282151222229,
      "weighted_auxiliary_loss": 0.0006310939905233681,
      "weighted_relation_loss": 0.0006310939905233681
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.279700654817587,
      "flow_matching_loss": 0.04282294362783432,
      "grad_norm": 0.1967705339193344,
      "learning_rate": 7.997030073810698e-05,
      "loss": 0.0435,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006445126025937497,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.62106876373291,
      "state_query_norm": 1261.226318359375,
      "state_relation_loss": 0.0006445126025937497,
      "step": 9920,
      "student_distance_mean": 1697.3462768554687,
      "teacher_distance_mean": 1.1272514939308167,
      "weighted_auxiliary_loss": 0.0006445126025937497,
      "weighted_relation_loss": 0.0006445126025937497
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.289055191768007,
      "flow_matching_loss": 0.0435240600258112,
      "grad_norm": 0.16049236059188843,
      "learning_rate": 7.992616547576217e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006670775299426168,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.807789611816407,
      "state_query_norm": 1297.96357421875,
      "state_relation_loss": 0.0006670775299426168,
      "step": 9930,
      "student_distance_mean": 1756.2162231445313,
      "teacher_distance_mean": 1.1712688326835632,
      "weighted_auxiliary_loss": 0.0006670775299426168,
      "weighted_relation_loss": 0.0006670775299426168
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.298409728718429,
      "flow_matching_loss": 0.04238249734044075,
      "grad_norm": 0.15000395476818085,
      "learning_rate": 7.988199385030445e-05,
      "loss": 0.043,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006467239232733846,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.449240684509277,
      "state_query_norm": 1291.8298706054688,
      "state_relation_loss": 0.0006467239232733846,
      "step": 9940,
      "student_distance_mean": 1743.7959106445312,
      "teacher_distance_mean": 1.157607626914978,
      "weighted_auxiliary_loss": 0.0006467239232733846,
      "weighted_relation_loss": 0.0006467239232733846
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.30776426566885,
      "flow_matching_loss": 0.041751202940940854,
      "grad_norm": 0.11050473898649216,
      "learning_rate": 7.98377859154065e-05,
      "loss": 0.0424,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006409651658032089,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.00356044769287,
      "state_query_norm": 1258.6512939453125,
      "state_relation_loss": 0.0006409651658032089,
      "step": 9950,
      "student_distance_mean": 1713.4538208007812,
      "teacher_distance_mean": 1.1376975297927856,
      "weighted_auxiliary_loss": 0.0006409651658032089,
      "weighted_relation_loss": 0.0006409651658032089
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.31711880261927,
      "flow_matching_loss": 0.04429178610444069,
      "grad_norm": 0.1690734624862671,
      "learning_rate": 7.979354172478516e-05,
      "loss": 0.0449,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006487795268185437,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.44848575592041,
      "state_query_norm": 1283.744482421875,
      "state_relation_loss": 0.0006487795268185437,
      "step": 9960,
      "student_distance_mean": 1739.6074951171875,
      "teacher_distance_mean": 1.1506540298461914,
      "weighted_auxiliary_loss": 0.0006487795268185437,
      "weighted_relation_loss": 0.0006487795268185437
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.326473339569691,
      "flow_matching_loss": 0.04433740675449371,
      "grad_norm": 0.1396246999502182,
      "learning_rate": 7.974926133220127e-05,
      "loss": 0.045,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006698153680190444,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.715876007080077,
      "state_query_norm": 1296.5284912109375,
      "state_relation_loss": 0.0006698153680190444,
      "step": 9970,
      "student_distance_mean": 1753.0944213867188,
      "teacher_distance_mean": 1.1585864901542664,
      "weighted_auxiliary_loss": 0.0006698153680190444,
      "weighted_relation_loss": 0.0006698153680190444
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.335827876520112,
      "flow_matching_loss": 0.04578882269561291,
      "grad_norm": 0.14260002970695496,
      "learning_rate": 7.970494479145967e-05,
      "loss": 0.0464,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006342910812236369,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.2001615524292,
      "state_query_norm": 1284.9541748046875,
      "state_relation_loss": 0.0006342910812236369,
      "step": 9980,
      "student_distance_mean": 1727.69091796875,
      "teacher_distance_mean": 1.147135329246521,
      "weighted_auxiliary_loss": 0.0006342910812236369,
      "weighted_relation_loss": 0.0006342910812236369
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.345182413470534,
      "flow_matching_loss": 0.042101599276065826,
      "grad_norm": 0.14272546768188477,
      "learning_rate": 7.966059215640919e-05,
      "loss": 0.0427,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006463309866376222,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.234094619750977,
      "state_query_norm": 1277.636669921875,
      "state_relation_loss": 0.0006463309866376222,
      "step": 9990,
      "student_distance_mean": 1729.5172973632812,
      "teacher_distance_mean": 1.148346769809723,
      "weighted_auxiliary_loss": 0.0006463309866376222,
      "weighted_relation_loss": 0.0006463309866376222
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.354536950420954,
      "flow_matching_loss": 0.040786541625857355,
      "grad_norm": 0.1364801824092865,
      "learning_rate": 7.961620348094241e-05,
      "loss": 0.0414,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006606221955735237,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.224611663818358,
      "state_query_norm": 1283.112060546875,
      "state_relation_loss": 0.0006606221955735237,
      "step": 10000,
      "student_distance_mean": 1727.8334716796876,
      "teacher_distance_mean": 1.1494966387748717,
      "weighted_auxiliary_loss": 0.0006606221955735237,
      "weighted_relation_loss": 0.0006606221955735237
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.363891487371376,
      "flow_matching_loss": 0.04489863701164722,
      "grad_norm": 0.1279076784849167,
      "learning_rate": 7.957177881899577e-05,
      "loss": 0.0456,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006591998739168048,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.320408439636232,
      "state_query_norm": 1279.7652587890625,
      "state_relation_loss": 0.0006591998739168048,
      "step": 10010,
      "student_distance_mean": 1729.7418701171875,
      "teacher_distance_mean": 1.1481656432151794,
      "weighted_auxiliary_loss": 0.0006591998739168048,
      "weighted_relation_loss": 0.0006591998739168048
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.373246024321796,
      "flow_matching_loss": 0.042040206491947174,
      "grad_norm": 0.12403611093759537,
      "learning_rate": 7.952731822454944e-05,
      "loss": 0.0427,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006542129965964704,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.71023178100586,
      "state_query_norm": 1259.5717895507812,
      "state_relation_loss": 0.0006542129965964704,
      "step": 10020,
      "student_distance_mean": 1701.3747680664062,
      "teacher_distance_mean": 1.1322522401809691,
      "weighted_auxiliary_loss": 0.0006542129965964704,
      "weighted_relation_loss": 0.0006542129965964704
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.382600561272216,
      "flow_matching_loss": 0.04304079078137875,
      "grad_norm": 0.10292224586009979,
      "learning_rate": 7.948282175162722e-05,
      "loss": 0.0437,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006539354857522994,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.880969047546387,
      "state_query_norm": 1297.2803100585938,
      "state_relation_loss": 0.0006539354857522994,
      "step": 10030,
      "student_distance_mean": 1761.579150390625,
      "teacher_distance_mean": 1.1706356406211853,
      "weighted_auxiliary_loss": 0.0006539354857522994,
      "weighted_relation_loss": 0.0006539354857522994
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.391955098222638,
      "flow_matching_loss": 0.04232733100652695,
      "grad_norm": 0.11574596911668777,
      "learning_rate": 7.943828945429652e-05,
      "loss": 0.0429,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006150923145469278,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.234082794189455,
      "state_query_norm": 1281.264501953125,
      "state_relation_loss": 0.0006150923145469278,
      "step": 10040,
      "student_distance_mean": 1731.6218505859374,
      "teacher_distance_mean": 1.1499807715415955,
      "weighted_auxiliary_loss": 0.0006150923145469278,
      "weighted_relation_loss": 0.0006150923145469278
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.401309635173059,
      "flow_matching_loss": 0.04344829022884369,
      "grad_norm": 0.16879624128341675,
      "learning_rate": 7.939372138666827e-05,
      "loss": 0.0441,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006712823815178126,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.15550422668457,
      "state_query_norm": 1267.9963012695312,
      "state_relation_loss": 0.0006712823815178126,
      "step": 10050,
      "student_distance_mean": 1725.5145751953125,
      "teacher_distance_mean": 1.1519816637039184,
      "weighted_auxiliary_loss": 0.0006712823815178126,
      "weighted_relation_loss": 0.0006712823815178126
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.41066417212348,
      "flow_matching_loss": 0.03972742967307567,
      "grad_norm": 0.10778705030679703,
      "learning_rate": 7.934911760289692e-05,
      "loss": 0.0404,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006651977717410773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.221985626220704,
      "state_query_norm": 1279.3619995117188,
      "state_relation_loss": 0.0006651977717410773,
      "step": 10060,
      "student_distance_mean": 1729.9762573242188,
      "teacher_distance_mean": 1.1567971587181092,
      "weighted_auxiliary_loss": 0.0006651977717410773,
      "weighted_relation_loss": 0.0006651977717410773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.4200187090739,
      "flow_matching_loss": 0.04382555969059467,
      "grad_norm": 0.14279533922672272,
      "learning_rate": 7.930447815718022e-05,
      "loss": 0.0444,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006215697154402733,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.310200119018553,
      "state_query_norm": 1282.4766845703125,
      "state_relation_loss": 0.0006215697154402733,
      "step": 10070,
      "student_distance_mean": 1731.3640502929688,
      "teacher_distance_mean": 1.1562809228897095,
      "weighted_auxiliary_loss": 0.0006215697154402733,
      "weighted_relation_loss": 0.0006215697154402733
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.429373246024323,
      "flow_matching_loss": 0.041139448434114455,
      "grad_norm": 0.12814755737781525,
      "learning_rate": 7.925980310375934e-05,
      "loss": 0.0418,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006581736088264734,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.910554885864258,
      "state_query_norm": 1307.8480224609375,
      "state_relation_loss": 0.0006581736088264734,
      "step": 10080,
      "student_distance_mean": 1764.585546875,
      "teacher_distance_mean": 1.1696949124336242,
      "weighted_auxiliary_loss": 0.0006581736088264734,
      "weighted_relation_loss": 0.0006581736088264734
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.438727782974743,
      "flow_matching_loss": 0.04244413785636425,
      "grad_norm": 0.15285919606685638,
      "learning_rate": 7.921509249691865e-05,
      "loss": 0.0431,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006450836837757379,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.01886672973633,
      "state_query_norm": 1309.7949584960938,
      "state_relation_loss": 0.0006450836837757379,
      "step": 10090,
      "student_distance_mean": 1765.976953125,
      "teacher_distance_mean": 1.1613009691238403,
      "weighted_auxiliary_loss": 0.0006450836837757379,
      "weighted_relation_loss": 0.0006450836837757379
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.448082319925163,
      "flow_matching_loss": 0.04237283393740654,
      "grad_norm": 0.13947828114032745,
      "learning_rate": 7.917034639098579e-05,
      "loss": 0.043,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006295037630479783,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.35902976989746,
      "state_query_norm": 1279.3210571289062,
      "state_relation_loss": 0.0006295037630479783,
      "step": 10100,
      "student_distance_mean": 1734.8801879882812,
      "teacher_distance_mean": 1.1424086093902588,
      "weighted_auxiliary_loss": 0.0006295037630479783,
      "weighted_relation_loss": 0.0006295037630479783
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.457436856875585,
      "flow_matching_loss": 0.04221285879611969,
      "grad_norm": 0.13034364581108093,
      "learning_rate": 7.912556484033146e-05,
      "loss": 0.0429,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006591480108909309,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.249790573120116,
      "state_query_norm": 1275.6341186523437,
      "state_relation_loss": 0.0006591480108909309,
      "step": 10110,
      "student_distance_mean": 1726.1838134765626,
      "teacher_distance_mean": 1.1502783656120301,
      "weighted_auxiliary_loss": 0.0006591480108909309,
      "weighted_relation_loss": 0.0006591480108909309
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.466791393826005,
      "flow_matching_loss": 0.04166276827454567,
      "grad_norm": 0.11813993752002716,
      "learning_rate": 7.908074789936952e-05,
      "loss": 0.0423,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006313531543128192,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.016475677490234,
      "state_query_norm": 1267.5428588867187,
      "state_relation_loss": 0.0006313531543128192,
      "step": 10120,
      "student_distance_mean": 1719.8674194335938,
      "teacher_distance_mean": 1.147378671169281,
      "weighted_auxiliary_loss": 0.0006313531543128192,
      "weighted_relation_loss": 0.0006313531543128192
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.476145930776427,
      "flow_matching_loss": 0.042073716223239896,
      "grad_norm": 0.1262069195508957,
      "learning_rate": 7.903589562255673e-05,
      "loss": 0.0427,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006082917912863195,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.528688812255858,
      "state_query_norm": 1293.0419921875,
      "state_relation_loss": 0.0006082917912863195,
      "step": 10130,
      "student_distance_mean": 1748.3514526367187,
      "teacher_distance_mean": 1.1633075952529908,
      "weighted_auxiliary_loss": 0.0006082917912863195,
      "weighted_relation_loss": 0.0006082917912863195
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.485500467726848,
      "flow_matching_loss": 0.04176427945494652,
      "grad_norm": 0.11291902512311935,
      "learning_rate": 7.899100806439285e-05,
      "loss": 0.0424,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006170383712742478,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.774099349975586,
      "state_query_norm": 1307.6486083984375,
      "state_relation_loss": 0.0006170383712742478,
      "step": 10140,
      "student_distance_mean": 1764.5574096679688,
      "teacher_distance_mean": 1.1714871644973754,
      "weighted_auxiliary_loss": 0.0006170383712742478,
      "weighted_relation_loss": 0.0006170383712742478
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.494855004677268,
      "flow_matching_loss": 0.04147126190364361,
      "grad_norm": 0.11788824945688248,
      "learning_rate": 7.894608527942049e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006410007423255593,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.34911518096924,
      "state_query_norm": 1267.3269653320312,
      "state_relation_loss": 0.0006410007423255593,
      "step": 10150,
      "student_distance_mean": 1729.860302734375,
      "teacher_distance_mean": 1.1532529950141908,
      "weighted_auxiliary_loss": 0.0006410007423255593,
      "weighted_relation_loss": 0.0006410007423255593
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.50420954162769,
      "flow_matching_loss": 0.040528813004493715,
      "grad_norm": 0.09953214228153229,
      "learning_rate": 7.89011273222251e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006290620018262416,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.31536998748779,
      "state_query_norm": 1278.326611328125,
      "state_relation_loss": 0.0006290620018262416,
      "step": 10160,
      "student_distance_mean": 1733.6296630859374,
      "teacher_distance_mean": 1.1532772183418274,
      "weighted_auxiliary_loss": 0.0006290620018262416,
      "weighted_relation_loss": 0.0006290620018262416
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.51356407857811,
      "flow_matching_loss": 0.041267399489879605,
      "grad_norm": 0.13067391514778137,
      "learning_rate": 7.885613424743482e-05,
      "loss": 0.0419,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006620978994760663,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.155234146118165,
      "state_query_norm": 1281.5680786132812,
      "state_relation_loss": 0.0006620978994760663,
      "step": 10170,
      "student_distance_mean": 1725.9500122070312,
      "teacher_distance_mean": 1.1537695050239563,
      "weighted_auxiliary_loss": 0.0006620978994760663,
      "weighted_relation_loss": 0.0006620978994760663
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.522918615528532,
      "flow_matching_loss": 0.04314519949257374,
      "grad_norm": 0.12472350150346756,
      "learning_rate": 7.881110610972044e-05,
      "loss": 0.0438,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006328294344712049,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.431147193908693,
      "state_query_norm": 1286.7685180664062,
      "state_relation_loss": 0.0006328294344712049,
      "step": 10180,
      "student_distance_mean": 1739.2405639648437,
      "teacher_distance_mean": 1.1620360970497132,
      "weighted_auxiliary_loss": 0.0006328294344712049,
      "weighted_relation_loss": 0.0006328294344712049
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.532273152478952,
      "flow_matching_loss": 0.04354848712682724,
      "grad_norm": 0.122655488550663,
      "learning_rate": 7.876604296379544e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006954208016395569,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.241101455688476,
      "state_query_norm": 1277.668896484375,
      "state_relation_loss": 0.0006954208016395569,
      "step": 10190,
      "student_distance_mean": 1728.7012084960938,
      "teacher_distance_mean": 1.1449374318122865,
      "weighted_auxiliary_loss": 0.0006954208016395569,
      "weighted_relation_loss": 0.0006954208016395569
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.541627689429372,
      "flow_matching_loss": 0.04398490861058235,
      "grad_norm": 0.1569060981273651,
      "learning_rate": 7.872094486441579e-05,
      "loss": 0.0446,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006601337809115648,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.55755500793457,
      "state_query_norm": 1294.285498046875,
      "state_relation_loss": 0.0006601337809115648,
      "step": 10200,
      "student_distance_mean": 1745.98466796875,
      "teacher_distance_mean": 1.1590432524681091,
      "weighted_auxiliary_loss": 0.0006601337809115648,
      "weighted_relation_loss": 0.0006601337809115648
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.550982226379794,
      "flow_matching_loss": 0.04114657491445541,
      "grad_norm": 0.18476110696792603,
      "learning_rate": 7.867581186637991e-05,
      "loss": 0.0418,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006301591172814369,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.325694274902343,
      "state_query_norm": 1285.40615234375,
      "state_relation_loss": 0.0006301591172814369,
      "step": 10210,
      "student_distance_mean": 1733.4518798828126,
      "teacher_distance_mean": 1.149384570121765,
      "weighted_auxiliary_loss": 0.0006301591172814369,
      "weighted_relation_loss": 0.0006301591172814369
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.560336763330215,
      "flow_matching_loss": 0.04144957475364208,
      "grad_norm": 0.20715515315532684,
      "learning_rate": 7.863064402452866e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006855035782791675,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 30.851447296142577,
      "state_query_norm": 1260.31796875,
      "state_relation_loss": 0.0006855035782791675,
      "step": 10220,
      "student_distance_mean": 1706.99794921875,
      "teacher_distance_mean": 1.1337745428085326,
      "weighted_auxiliary_loss": 0.0006855035782791675,
      "weighted_relation_loss": 0.0006855035782791675
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.569691300280637,
      "flow_matching_loss": 0.04193105064332485,
      "grad_norm": 0.12761512398719788,
      "learning_rate": 7.858544139374524e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006588174670469016,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.3855562210083,
      "state_query_norm": 1282.9675415039062,
      "state_relation_loss": 0.0006588174670469016,
      "step": 10230,
      "student_distance_mean": 1733.6654663085938,
      "teacher_distance_mean": 1.1523774981498718,
      "weighted_auxiliary_loss": 0.0006588174670469016,
      "weighted_relation_loss": 0.0006588174670469016
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.579045837231057,
      "flow_matching_loss": 0.04154014252126217,
      "grad_norm": 0.14216934144496918,
      "learning_rate": 7.854020402895508e-05,
      "loss": 0.0422,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006622325512580573,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.221036911010742,
      "state_query_norm": 1274.2528442382813,
      "state_relation_loss": 0.0006622325512580573,
      "step": 10240,
      "student_distance_mean": 1725.4147094726563,
      "teacher_distance_mean": 1.150016677379608,
      "weighted_auxiliary_loss": 0.0006622325512580573,
      "weighted_relation_loss": 0.0006622325512580573
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.588400374181479,
      "flow_matching_loss": 0.04200497977435589,
      "grad_norm": 0.10190018266439438,
      "learning_rate": 7.849493198512587e-05,
      "loss": 0.0427,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006458536081481725,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.0641731262207,
      "state_query_norm": 1308.5955078125,
      "state_relation_loss": 0.0006458536081481725,
      "step": 10250,
      "student_distance_mean": 1773.3108032226562,
      "teacher_distance_mean": 1.1758728742599487,
      "weighted_auxiliary_loss": 0.0006458536081481725,
      "weighted_relation_loss": 0.0006458536081481725
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.597754911131899,
      "flow_matching_loss": 0.0412391159683466,
      "grad_norm": 0.10866537690162659,
      "learning_rate": 7.84496253172674e-05,
      "loss": 0.0419,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006430888257455081,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.264809799194335,
      "state_query_norm": 1269.7765747070312,
      "state_relation_loss": 0.0006430888257455081,
      "step": 10260,
      "student_distance_mean": 1726.3481567382812,
      "teacher_distance_mean": 1.137280011177063,
      "weighted_auxiliary_loss": 0.0006430888257455081,
      "weighted_relation_loss": 0.0006430888257455081
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.60710944808232,
      "flow_matching_loss": 0.04004529044032097,
      "grad_norm": 0.10748057812452316,
      "learning_rate": 7.840428408043155e-05,
      "loss": 0.0407,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006479855510406196,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.145870018005372,
      "state_query_norm": 1273.2381225585937,
      "state_relation_loss": 0.0006479855510406196,
      "step": 10270,
      "student_distance_mean": 1725.15263671875,
      "teacher_distance_mean": 1.14161958694458,
      "weighted_auxiliary_loss": 0.0006479855510406196,
      "weighted_relation_loss": 0.0006479855510406196
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.616463985032741,
      "flow_matching_loss": 0.040558765456080434,
      "grad_norm": 0.16239692270755768,
      "learning_rate": 7.835890832971218e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006297546380665154,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.451310920715333,
      "state_query_norm": 1292.4769165039063,
      "state_relation_loss": 0.0006297546380665154,
      "step": 10280,
      "student_distance_mean": 1740.8303344726562,
      "teacher_distance_mean": 1.1535321950912476,
      "weighted_auxiliary_loss": 0.0006297546380665154,
      "weighted_relation_loss": 0.0006297546380665154
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.625818521983161,
      "flow_matching_loss": 0.041907258704304694,
      "grad_norm": 0.11704814434051514,
      "learning_rate": 7.831349812024512e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006709577399305999,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.1535737991333,
      "state_query_norm": 1276.8537475585938,
      "state_relation_loss": 0.0006709577399305999,
      "step": 10290,
      "student_distance_mean": 1720.419482421875,
      "teacher_distance_mean": 1.1355111956596375,
      "weighted_auxiliary_loss": 0.0006709577399305999,
      "weighted_relation_loss": 0.0006709577399305999
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.635173058933583,
      "flow_matching_loss": 0.04054743386805058,
      "grad_norm": 0.12472641468048096,
      "learning_rate": 7.826805350720807e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006510503124445676,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.41413459777832,
      "state_query_norm": 1323.8990600585937,
      "state_relation_loss": 0.0006510503124445676,
      "step": 10300,
      "student_distance_mean": 1789.6621704101562,
      "teacher_distance_mean": 1.1761026501655578,
      "weighted_auxiliary_loss": 0.0006510503124445676,
      "weighted_relation_loss": 0.0006510503124445676
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.644527595884004,
      "flow_matching_loss": 0.04294726587831974,
      "grad_norm": 0.12192953377962112,
      "learning_rate": 7.822257454582049e-05,
      "loss": 0.0436,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006551724800374359,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.811982345581054,
      "state_query_norm": 1303.4095336914063,
      "state_relation_loss": 0.0006551724800374359,
      "step": 10310,
      "student_distance_mean": 1760.0041748046874,
      "teacher_distance_mean": 1.1491197109222413,
      "weighted_auxiliary_loss": 0.0006551724800374359,
      "weighted_relation_loss": 0.0006551724800374359
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.653882132834426,
      "flow_matching_loss": 0.04149367585778237,
      "grad_norm": 0.128851979970932,
      "learning_rate": 7.817706129134363e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006481298769358546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.778060340881346,
      "state_query_norm": 1300.8087768554688,
      "state_relation_loss": 0.0006481298769358546,
      "step": 10320,
      "student_distance_mean": 1756.5107543945312,
      "teacher_distance_mean": 1.1575021147727966,
      "weighted_auxiliary_loss": 0.0006481298769358546,
      "weighted_relation_loss": 0.0006481298769358546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.663236669784846,
      "flow_matching_loss": 0.04320542849600315,
      "grad_norm": 0.11674122512340546,
      "learning_rate": 7.813151379908036e-05,
      "loss": 0.0438,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006395972508471459,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.162479209899903,
      "state_query_norm": 1270.990380859375,
      "state_relation_loss": 0.0006395972508471459,
      "step": 10330,
      "student_distance_mean": 1724.1553955078125,
      "teacher_distance_mean": 1.1410950541496276,
      "weighted_auxiliary_loss": 0.0006395972508471459,
      "weighted_relation_loss": 0.0006395972508471459
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.672591206735266,
      "flow_matching_loss": 0.04207397252321243,
      "grad_norm": 0.12833891808986664,
      "learning_rate": 7.808593212437523e-05,
      "loss": 0.0427,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006300419336184859,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.33722896575928,
      "state_query_norm": 1325.505419921875,
      "state_relation_loss": 0.0006300419336184859,
      "step": 10340,
      "student_distance_mean": 1792.9690795898437,
      "teacher_distance_mean": 1.1788505792617798,
      "weighted_auxiliary_loss": 0.0006300419336184859,
      "weighted_relation_loss": 0.0006300419336184859
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.681945743685688,
      "flow_matching_loss": 0.03952118456363678,
      "grad_norm": 0.13361041247844696,
      "learning_rate": 7.804031632261421e-05,
      "loss": 0.0402,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006387947243638337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.557225227355957,
      "state_query_norm": 1295.987646484375,
      "state_relation_loss": 0.0006387947243638337,
      "step": 10350,
      "student_distance_mean": 1746.8628295898438,
      "teacher_distance_mean": 1.1462768197059632,
      "weighted_auxiliary_loss": 0.0006387947243638337,
      "weighted_relation_loss": 0.0006387947243638337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.691300280636108,
      "flow_matching_loss": 0.04354546777904034,
      "grad_norm": 0.14177305996418,
      "learning_rate": 7.799466644922484e-05,
      "loss": 0.0442,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006399928883183747,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.03840503692627,
      "state_query_norm": 1305.0454711914062,
      "state_relation_loss": 0.0006399928883183747,
      "step": 10360,
      "student_distance_mean": 1769.4813598632813,
      "teacher_distance_mean": 1.1610773921012878,
      "weighted_auxiliary_loss": 0.0006399928883183747,
      "weighted_relation_loss": 0.0006399928883183747
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.70065481758653,
      "flow_matching_loss": 0.040121236443519594,
      "grad_norm": 0.1109156683087349,
      "learning_rate": 7.794898255967602e-05,
      "loss": 0.0408,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006976368313189596,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.24209289550781,
      "state_query_norm": 1307.8212036132813,
      "state_relation_loss": 0.0006976368313189596,
      "step": 10370,
      "student_distance_mean": 1778.96630859375,
      "teacher_distance_mean": 1.165699028968811,
      "weighted_auxiliary_loss": 0.0006976368313189596,
      "weighted_relation_loss": 0.0006976368313189596
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.71000935453695,
      "flow_matching_loss": 0.04182318225502968,
      "grad_norm": 0.10569429397583008,
      "learning_rate": 7.790326470947795e-05,
      "loss": 0.0424,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006231127888895571,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.165312576293946,
      "state_query_norm": 1283.2902465820312,
      "state_relation_loss": 0.0006231127888895571,
      "step": 10380,
      "student_distance_mean": 1733.0646118164063,
      "teacher_distance_mean": 1.1392376661300658,
      "weighted_auxiliary_loss": 0.0006231127888895571,
      "weighted_relation_loss": 0.0006231127888895571
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.71936389148737,
      "flow_matching_loss": 0.04159816056489944,
      "grad_norm": 0.09332901239395142,
      "learning_rate": 7.785751295418217e-05,
      "loss": 0.0422,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006242853414732963,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.039827156066895,
      "state_query_norm": 1261.9150390625,
      "state_relation_loss": 0.0006242853414732963,
      "step": 10390,
      "student_distance_mean": 1715.3943359375,
      "teacher_distance_mean": 1.1287451267242432,
      "weighted_auxiliary_loss": 0.0006242853414732963,
      "weighted_relation_loss": 0.0006242853414732963
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.728718428437793,
      "flow_matching_loss": 0.041894235834479335,
      "grad_norm": 0.10999548435211182,
      "learning_rate": 7.781172734938136e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006591218116227537,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.973739433288575,
      "state_query_norm": 1306.591748046875,
      "state_relation_loss": 0.0006591218116227537,
      "step": 10400,
      "student_distance_mean": 1766.0921875,
      "teacher_distance_mean": 1.1547564387321472,
      "weighted_auxiliary_loss": 0.0006591218116227537,
      "weighted_relation_loss": 0.0006591218116227537
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.738072965388213,
      "flow_matching_loss": 0.03895261734724045,
      "grad_norm": 0.10164802521467209,
      "learning_rate": 7.776590795070933e-05,
      "loss": 0.0396,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006336644757539034,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.30571880340576,
      "state_query_norm": 1284.3191040039062,
      "state_relation_loss": 0.0006336644757539034,
      "step": 10410,
      "student_distance_mean": 1730.6202270507813,
      "teacher_distance_mean": 1.1353694558143617,
      "weighted_auxiliary_loss": 0.0006336644757539034,
      "weighted_relation_loss": 0.0006336644757539034
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.747427502338635,
      "flow_matching_loss": 0.04089015945792198,
      "grad_norm": 0.1321098506450653,
      "learning_rate": 7.772005481384099e-05,
      "loss": 0.0416,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006644635752309114,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.955278968811037,
      "state_query_norm": 1305.243798828125,
      "state_relation_loss": 0.0006644635752309114,
      "step": 10420,
      "student_distance_mean": 1766.5645629882813,
      "teacher_distance_mean": 1.157050323486328,
      "weighted_auxiliary_loss": 0.0006644635752309114,
      "weighted_relation_loss": 0.0006644635752309114
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.756782039289055,
      "flow_matching_loss": 0.03921782299876213,
      "grad_norm": 0.11927176266908646,
      "learning_rate": 7.767416799449223e-05,
      "loss": 0.0398,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006196219241246581,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.64555606842041,
      "state_query_norm": 1295.849609375,
      "state_relation_loss": 0.0006196219241246581,
      "step": 10430,
      "student_distance_mean": 1750.2183471679687,
      "teacher_distance_mean": 1.1412288904190064,
      "weighted_auxiliary_loss": 0.0006196219241246581,
      "weighted_relation_loss": 0.0006196219241246581
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.766136576239477,
      "flow_matching_loss": 0.03995710276067257,
      "grad_norm": 0.13407523930072784,
      "learning_rate": 7.762824754841984e-05,
      "loss": 0.0406,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006175660935696214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.91155071258545,
      "state_query_norm": 1311.5152954101563,
      "state_relation_loss": 0.0006175660935696214,
      "step": 10440,
      "student_distance_mean": 1766.378857421875,
      "teacher_distance_mean": 1.1538745880126953,
      "weighted_auxiliary_loss": 0.0006175660935696214,
      "weighted_relation_loss": 0.0006175660935696214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.775491113189897,
      "flow_matching_loss": 0.041768182814121244,
      "grad_norm": 0.12208918482065201,
      "learning_rate": 7.758229353142152e-05,
      "loss": 0.0424,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006268832716159522,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.216654205322264,
      "state_query_norm": 1308.6754272460937,
      "state_relation_loss": 0.0006268832716159522,
      "step": 10450,
      "student_distance_mean": 1777.47255859375,
      "teacher_distance_mean": 1.1605093002319335,
      "weighted_auxiliary_loss": 0.0006268832716159522,
      "weighted_relation_loss": 0.0006268832716159522
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.784845650140317,
      "flow_matching_loss": 0.04139733389019966,
      "grad_norm": 0.144992396235466,
      "learning_rate": 7.753630599933572e-05,
      "loss": 0.042,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005875612434465439,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.730955505371092,
      "state_query_norm": 1294.2139770507813,
      "state_relation_loss": 0.0005875612434465439,
      "step": 10460,
      "student_distance_mean": 1755.38125,
      "teacher_distance_mean": 1.138983952999115,
      "weighted_auxiliary_loss": 0.0005875612434465439,
      "weighted_relation_loss": 0.0005875612434465439
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.79420018709074,
      "flow_matching_loss": 0.04139500185847282,
      "grad_norm": 0.14687618613243103,
      "learning_rate": 7.749028500804161e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006581513211131096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.88025188446045,
      "state_query_norm": 1338.00078125,
      "state_relation_loss": 0.0006581513211131096,
      "step": 10470,
      "student_distance_mean": 1812.4935302734375,
      "teacher_distance_mean": 1.17563294172287,
      "weighted_auxiliary_loss": 0.0006581513211131096,
      "weighted_relation_loss": 0.0006581513211131096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.80355472404116,
      "flow_matching_loss": 0.04065104871988297,
      "grad_norm": 0.117997907102108,
      "learning_rate": 7.744423061345906e-05,
      "loss": 0.0413,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000636624323669821,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.19087944030762,
      "state_query_norm": 1313.7830444335937,
      "state_relation_loss": 0.000636624323669821,
      "step": 10480,
      "student_distance_mean": 1779.41865234375,
      "teacher_distance_mean": 1.159523355960846,
      "weighted_auxiliary_loss": 0.000636624323669821,
      "weighted_relation_loss": 0.000636624323669821
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.812909260991582,
      "flow_matching_loss": 0.039955854415893555,
      "grad_norm": 0.1223173439502716,
      "learning_rate": 7.73981428715485e-05,
      "loss": 0.0406,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006311911391094327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.68668155670166,
      "state_query_norm": 1291.9761840820313,
      "state_relation_loss": 0.0006311911391094327,
      "step": 10490,
      "student_distance_mean": 1752.6285400390625,
      "teacher_distance_mean": 1.1472487449645996,
      "weighted_auxiliary_loss": 0.0006311911391094327,
      "weighted_relation_loss": 0.0006311911391094327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.822263797942002,
      "flow_matching_loss": 0.041949187591671946,
      "grad_norm": 0.1318552941083908,
      "learning_rate": 7.735202183831085e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006021426117513329,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.22711124420166,
      "state_query_norm": 1312.865966796875,
      "state_relation_loss": 0.0006021426117513329,
      "step": 10500,
      "student_distance_mean": 1781.8113403320312,
      "teacher_distance_mean": 1.1664007544517516,
      "weighted_auxiliary_loss": 0.0006021426117513329,
      "weighted_relation_loss": 0.0006021426117513329
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.831618334892422,
      "flow_matching_loss": 0.040326996520161626,
      "grad_norm": 0.16972698271274567,
      "learning_rate": 7.730586756978757e-05,
      "loss": 0.0409,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005991156329400837,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.072160148620604,
      "state_query_norm": 1311.7055908203124,
      "state_relation_loss": 0.0005991156329400837,
      "step": 10510,
      "student_distance_mean": 1775.3010620117188,
      "teacher_distance_mean": 1.1602731585502624,
      "weighted_auxiliary_loss": 0.0005991156329400837,
      "weighted_relation_loss": 0.0005991156329400837
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.840972871842844,
      "flow_matching_loss": 0.03936530686914921,
      "grad_norm": 0.11928362399339676,
      "learning_rate": 7.72596801220604e-05,
      "loss": 0.04,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006193833134602756,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.977351188659668,
      "state_query_norm": 1302.1486938476562,
      "state_relation_loss": 0.0006193833134602756,
      "step": 10520,
      "student_distance_mean": 1768.9321899414062,
      "teacher_distance_mean": 1.154518711566925,
      "weighted_auxiliary_loss": 0.0006193833134602756,
      "weighted_relation_loss": 0.0006193833134602756
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.850327408793264,
      "flow_matching_loss": 0.04094926826655865,
      "grad_norm": 0.16441474854946136,
      "learning_rate": 7.721345955125147e-05,
      "loss": 0.0416,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006081602245103568,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.810904884338377,
      "state_query_norm": 1297.18056640625,
      "state_relation_loss": 0.0006081602245103568,
      "step": 10530,
      "student_distance_mean": 1757.329345703125,
      "teacher_distance_mean": 1.1493411779403686,
      "weighted_auxiliary_loss": 0.0006081602245103568,
      "weighted_relation_loss": 0.0006081602245103568
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.859681945743686,
      "flow_matching_loss": 0.03942183088511229,
      "grad_norm": 0.16363969445228577,
      "learning_rate": 7.71672059135231e-05,
      "loss": 0.0401,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006553170329425484,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.57021560668945,
      "state_query_norm": 1291.7122924804687,
      "state_relation_loss": 0.0006553170329425484,
      "step": 10540,
      "student_distance_mean": 1747.4021606445312,
      "teacher_distance_mean": 1.1394750356674195,
      "weighted_auxiliary_loss": 0.0006553170329425484,
      "weighted_relation_loss": 0.0006553170329425484
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.869036482694106,
      "flow_matching_loss": 0.04279412366449833,
      "grad_norm": 0.11314281076192856,
      "learning_rate": 7.712091926507787e-05,
      "loss": 0.0434,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006241979892365635,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.838705635070802,
      "state_query_norm": 1300.2702514648438,
      "state_relation_loss": 0.0006241979892365635,
      "step": 10550,
      "student_distance_mean": 1765.032177734375,
      "teacher_distance_mean": 1.1514227271080018,
      "weighted_auxiliary_loss": 0.0006241979892365635,
      "weighted_relation_loss": 0.0006241979892365635
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.878391019644528,
      "flow_matching_loss": 0.03991547226905823,
      "grad_norm": 0.10829820483922958,
      "learning_rate": 7.70745996621584e-05,
      "loss": 0.0406,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006400393671356141,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.436387252807616,
      "state_query_norm": 1291.5543212890625,
      "state_relation_loss": 0.0006400393671356141,
      "step": 10560,
      "student_distance_mean": 1742.7757446289063,
      "teacher_distance_mean": 1.1370984196662903,
      "weighted_auxiliary_loss": 0.0006400393671356141,
      "weighted_relation_loss": 0.0006400393671356141
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.887745556594949,
      "flow_matching_loss": 0.040517131239175795,
      "grad_norm": 0.10681083798408508,
      "learning_rate": 7.702824716104735e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000652012781938538,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.50474395751953,
      "state_query_norm": 1330.2864379882812,
      "state_relation_loss": 0.000652012781938538,
      "step": 10570,
      "student_distance_mean": 1798.7700927734375,
      "teacher_distance_mean": 1.1669580459594726,
      "weighted_auxiliary_loss": 0.000652012781938538,
      "weighted_relation_loss": 0.000652012781938538
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.897100093545369,
      "flow_matching_loss": 0.04042074978351593,
      "grad_norm": 0.1602908819913864,
      "learning_rate": 7.698186181806743e-05,
      "loss": 0.0411,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000649187434464693,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.094628715515135,
      "state_query_norm": 1315.6038696289063,
      "state_relation_loss": 0.000649187434464693,
      "step": 10580,
      "student_distance_mean": 1774.7990844726562,
      "teacher_distance_mean": 1.1529664754867555,
      "weighted_auxiliary_loss": 0.000649187434464693,
      "weighted_relation_loss": 0.000649187434464693
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.90645463049579,
      "flow_matching_loss": 0.04200239591300488,
      "grad_norm": 0.0992458164691925,
      "learning_rate": 7.693544368958116e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006273226696066559,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.283734512329104,
      "state_query_norm": 1318.9952270507813,
      "state_relation_loss": 0.0006273226696066559,
      "step": 10590,
      "student_distance_mean": 1787.0385986328124,
      "teacher_distance_mean": 1.1594188332557678,
      "weighted_auxiliary_loss": 0.0006273226696066559,
      "weighted_relation_loss": 0.0006273226696066559
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.915809167446211,
      "flow_matching_loss": 0.04149624221026897,
      "grad_norm": 0.1325712352991104,
      "learning_rate": 7.688899283199096e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005893057445064187,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.905692863464356,
      "state_query_norm": 1301.6737915039062,
      "state_relation_loss": 0.0005893057445064187,
      "step": 10600,
      "student_distance_mean": 1763.27421875,
      "teacher_distance_mean": 1.1491888403892516,
      "weighted_auxiliary_loss": 0.0005893057445064187,
      "weighted_relation_loss": 0.0005893057445064187
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.925163704396633,
      "flow_matching_loss": 0.041959628090262414,
      "grad_norm": 0.1422644406557083,
      "learning_rate": 7.684250930173901e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006314549827948213,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.10905284881592,
      "state_query_norm": 1309.3348266601563,
      "state_relation_loss": 0.0006314549827948213,
      "step": 10610,
      "student_distance_mean": 1774.8059326171874,
      "teacher_distance_mean": 1.1527193546295167,
      "weighted_auxiliary_loss": 0.0006314549827948213,
      "weighted_relation_loss": 0.0006314549827948213
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.934518241347053,
      "flow_matching_loss": 0.04256726168096066,
      "grad_norm": 0.11735827475786209,
      "learning_rate": 7.679599315530718e-05,
      "loss": 0.0432,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006453758280258626,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.47696380615234,
      "state_query_norm": 1329.55146484375,
      "state_relation_loss": 0.0006453758280258626,
      "step": 10620,
      "student_distance_mean": 1796.90576171875,
      "teacher_distance_mean": 1.1709095120429993,
      "weighted_auxiliary_loss": 0.0006453758280258626,
      "weighted_relation_loss": 0.0006453758280258626
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.943872778297475,
      "flow_matching_loss": 0.041093287616968156,
      "grad_norm": 0.09917553514242172,
      "learning_rate": 7.674944444921695e-05,
      "loss": 0.0417,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006448095897212625,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.92528018951416,
      "state_query_norm": 1305.3300048828125,
      "state_relation_loss": 0.0006448095897212625,
      "step": 10630,
      "student_distance_mean": 1764.3740356445312,
      "teacher_distance_mean": 1.146644115447998,
      "weighted_auxiliary_loss": 0.0006448095897212625,
      "weighted_relation_loss": 0.0006448095897212625
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.953227315247895,
      "flow_matching_loss": 0.03970971331000328,
      "grad_norm": 0.11441371589899063,
      "learning_rate": 7.670286324002944e-05,
      "loss": 0.0403,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006200240924954415,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.959006118774415,
      "state_query_norm": 1301.5642822265625,
      "state_relation_loss": 0.0006200240924954415,
      "step": 10640,
      "student_distance_mean": 1769.5425537109375,
      "teacher_distance_mean": 1.1522873878479003,
      "weighted_auxiliary_loss": 0.0006200240924954415,
      "weighted_relation_loss": 0.0006200240924954415
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.962581852198316,
      "flow_matching_loss": 0.04219858534634113,
      "grad_norm": 0.11244741827249527,
      "learning_rate": 7.665624958434514e-05,
      "loss": 0.0428,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00063236128189601,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.402869606018065,
      "state_query_norm": 1328.0257446289063,
      "state_relation_loss": 0.00063236128189601,
      "step": 10650,
      "student_distance_mean": 1793.8348388671875,
      "teacher_distance_mean": 1.1671222686767577,
      "weighted_auxiliary_loss": 0.00063236128189601,
      "weighted_relation_loss": 0.00063236128189601
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.971936389148738,
      "flow_matching_loss": 0.04129285365343094,
      "grad_norm": 0.12534752488136292,
      "learning_rate": 7.66096035388041e-05,
      "loss": 0.0419,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006167670071590692,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.49059658050537,
      "state_query_norm": 1324.0688720703124,
      "state_relation_loss": 0.0006167670071590692,
      "step": 10660,
      "student_distance_mean": 1795.9934814453125,
      "teacher_distance_mean": 1.1546535968780518,
      "weighted_auxiliary_loss": 0.0006167670071590692,
      "weighted_relation_loss": 0.0006167670071590692
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.981290926099158,
      "flow_matching_loss": 0.039380477741360664,
      "grad_norm": 0.1448308825492859,
      "learning_rate": 7.656292516008563e-05,
      "loss": 0.04,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006188354920595885,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.623171997070315,
      "state_query_norm": 1335.0742309570312,
      "state_relation_loss": 0.0006188354920595885,
      "step": 10670,
      "student_distance_mean": 1803.9599243164062,
      "teacher_distance_mean": 1.1595815539360046,
      "weighted_auxiliary_loss": 0.0006188354920595885,
      "weighted_relation_loss": 0.0006188354920595885
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 9.99064546304958,
      "flow_matching_loss": 0.0402470413595438,
      "grad_norm": 0.10763208568096161,
      "learning_rate": 7.651621450490837e-05,
      "loss": 0.0409,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006344423047266901,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.13332805633545,
      "state_query_norm": 1311.2699829101562,
      "state_relation_loss": 0.0006344423047266901,
      "step": 10680,
      "student_distance_mean": 1775.2858520507812,
      "teacher_distance_mean": 1.1536160707473755,
      "weighted_auxiliary_loss": 0.0006344423047266901,
      "weighted_relation_loss": 0.0006344423047266901
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.0,
      "flow_matching_loss": 0.04034475386142731,
      "grad_norm": 0.21801261603832245,
      "learning_rate": 7.646947163003017e-05,
      "loss": 0.041,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0006233057996723801,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 32.165343284606934,
      "state_query_norm": 1304.5274169921875,
      "state_relation_loss": 0.0006233057996723801,
      "step": 10690,
      "student_distance_mean": 1776.0908325195312,
      "teacher_distance_mean": 1.1569382786750793,
      "weighted_auxiliary_loss": 0.0006233057996723801,
      "weighted_relation_loss": 0.0006233057996723801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.00935453695042,
      "flow_matching_loss": 0.03625116031616926,
      "grad_norm": 0.10848969221115112,
      "learning_rate": 7.642269659224803e-05,
      "loss": 0.0369,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006185989070218057,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.44955558776856,
      "state_query_norm": 1319.0060424804688,
      "state_relation_loss": 0.0006185989070218057,
      "step": 10700,
      "student_distance_mean": 1795.8676513671876,
      "teacher_distance_mean": 1.1617783665657044,
      "weighted_auxiliary_loss": 0.0006185989070218057,
      "weighted_relation_loss": 0.0006185989070218057
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.018709073900842,
      "flow_matching_loss": 0.03876518160104751,
      "grad_norm": 0.13104824721813202,
      "learning_rate": 7.637588944839803e-05,
      "loss": 0.0394,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006353891280014068,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.927086448669435,
      "state_query_norm": 1309.8094116210937,
      "state_relation_loss": 0.0006353891280014068,
      "step": 10710,
      "student_distance_mean": 1764.8716552734375,
      "teacher_distance_mean": 1.1375533103942872,
      "weighted_auxiliary_loss": 0.0006353891280014068,
      "weighted_relation_loss": 0.0006353891280014068
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.028063610851262,
      "flow_matching_loss": 0.03779454417526722,
      "grad_norm": 0.09607044607400894,
      "learning_rate": 7.632905025535528e-05,
      "loss": 0.0384,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006241845025215298,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.07418251037598,
      "state_query_norm": 1309.1974731445312,
      "state_relation_loss": 0.0006241845025215298,
      "step": 10720,
      "student_distance_mean": 1770.0993774414062,
      "teacher_distance_mean": 1.140472686290741,
      "weighted_auxiliary_loss": 0.0006241845025215298,
      "weighted_relation_loss": 0.0006241845025215298
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.037418147801684,
      "flow_matching_loss": 0.03984539471566677,
      "grad_norm": 0.12931886315345764,
      "learning_rate": 7.628217907003378e-05,
      "loss": 0.0405,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006413615425117314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.39500770568848,
      "state_query_norm": 1318.5500610351562,
      "state_relation_loss": 0.0006413615425117314,
      "step": 10730,
      "student_distance_mean": 1792.029736328125,
      "teacher_distance_mean": 1.150426721572876,
      "weighted_auxiliary_loss": 0.0006413615425117314,
      "weighted_relation_loss": 0.0006413615425117314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.046772684752105,
      "flow_matching_loss": 0.040055349469184875,
      "grad_norm": 0.09996743500232697,
      "learning_rate": 7.623527594938649e-05,
      "loss": 0.0407,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006808835722040385,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.72097282409668,
      "state_query_norm": 1326.3439086914063,
      "state_relation_loss": 0.0006808835722040385,
      "step": 10740,
      "student_distance_mean": 1803.6908081054687,
      "teacher_distance_mean": 1.1641142249107361,
      "weighted_auxiliary_loss": 0.0006808835722040385,
      "weighted_relation_loss": 0.0006808835722040385
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.056127221702527,
      "flow_matching_loss": 0.03850181959569454,
      "grad_norm": 0.13148055970668793,
      "learning_rate": 7.618834095040509e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000620168203022331,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.35361022949219,
      "state_query_norm": 1315.0986572265624,
      "state_relation_loss": 0.000620168203022331,
      "step": 10750,
      "student_distance_mean": 1787.6017944335938,
      "teacher_distance_mean": 1.151507019996643,
      "weighted_auxiliary_loss": 0.000620168203022331,
      "weighted_relation_loss": 0.000620168203022331
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.065481758652947,
      "flow_matching_loss": 0.038306872546672824,
      "grad_norm": 0.13943664729595184,
      "learning_rate": 7.614137413012e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006149599677883089,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.862034606933594,
      "state_query_norm": 1293.6012329101563,
      "state_relation_loss": 0.0006149599677883089,
      "step": 10760,
      "student_distance_mean": 1758.709326171875,
      "teacher_distance_mean": 1.140292227268219,
      "weighted_auxiliary_loss": 0.0006149599677883089,
      "weighted_relation_loss": 0.0006149599677883089
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.074836295603367,
      "flow_matching_loss": 0.04154169075191021,
      "grad_norm": 0.11569038778543472,
      "learning_rate": 7.609437554560042e-05,
      "loss": 0.0422,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006416173710022122,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.03567314147949,
      "state_query_norm": 1298.1238647460937,
      "state_relation_loss": 0.0006416173710022122,
      "step": 10770,
      "student_distance_mean": 1767.7070678710938,
      "teacher_distance_mean": 1.1529324054718018,
      "weighted_auxiliary_loss": 0.0006416173710022122,
      "weighted_relation_loss": 0.0006416173710022122
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.084190832553789,
      "flow_matching_loss": 0.03857877589762211,
      "grad_norm": 0.11638302356004715,
      "learning_rate": 7.604734525395398e-05,
      "loss": 0.0392,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006142984493635595,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.18787307739258,
      "state_query_norm": 1311.3630004882812,
      "state_relation_loss": 0.0006142984493635595,
      "step": 10780,
      "student_distance_mean": 1780.0155029296875,
      "teacher_distance_mean": 1.1510708570480346,
      "weighted_auxiliary_loss": 0.0006142984493635595,
      "weighted_relation_loss": 0.0006142984493635595
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.09354536950421,
      "flow_matching_loss": 0.03870654739439487,
      "grad_norm": 0.15246032178401947,
      "learning_rate": 7.600028331232699e-05,
      "loss": 0.0393,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000639828824205324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.51414260864258,
      "state_query_norm": 1320.375634765625,
      "state_relation_loss": 0.000639828824205324,
      "step": 10790,
      "student_distance_mean": 1795.2601196289063,
      "teacher_distance_mean": 1.156926155090332,
      "weighted_auxiliary_loss": 0.000639828824205324,
      "weighted_relation_loss": 0.000639828824205324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.102899906454631,
      "flow_matching_loss": 0.041577615216374396,
      "grad_norm": 0.0988016352057457,
      "learning_rate": 7.595318977790408e-05,
      "loss": 0.0422,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006003475224133581,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.303036308288576,
      "state_query_norm": 1321.7581665039063,
      "state_relation_loss": 0.0006003475224133581,
      "step": 10800,
      "student_distance_mean": 1785.4579956054688,
      "teacher_distance_mean": 1.1517270803451538,
      "weighted_auxiliary_loss": 0.0006003475224133581,
      "weighted_relation_loss": 0.0006003475224133581
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.112254443405051,
      "flow_matching_loss": 0.039018455520272256,
      "grad_norm": 0.11943338811397552,
      "learning_rate": 7.590606470790836e-05,
      "loss": 0.0397,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006412364484276623,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.129879760742185,
      "state_query_norm": 1306.637353515625,
      "state_relation_loss": 0.0006412364484276623,
      "step": 10810,
      "student_distance_mean": 1774.0209838867188,
      "teacher_distance_mean": 1.1469308495521546,
      "weighted_auxiliary_loss": 0.0006412364484276623,
      "weighted_relation_loss": 0.0006412364484276623
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.121608980355472,
      "flow_matching_loss": 0.04095222875475883,
      "grad_norm": 0.19456742703914642,
      "learning_rate": 7.585890815960125e-05,
      "loss": 0.0416,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006332523655146361,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.107134628295896,
      "state_query_norm": 1306.0511840820313,
      "state_relation_loss": 0.0006332523655146361,
      "step": 10820,
      "student_distance_mean": 1773.3173828125,
      "teacher_distance_mean": 1.1495704650878906,
      "weighted_auxiliary_loss": 0.0006332523655146361,
      "weighted_relation_loss": 0.0006332523655146361
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.130963517305894,
      "flow_matching_loss": 0.0393002200871706,
      "grad_norm": 0.10956449061632156,
      "learning_rate": 7.581172019028238e-05,
      "loss": 0.0399,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006464874837547541,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.91339111328125,
      "state_query_norm": 1303.2375854492188,
      "state_relation_loss": 0.0006464874837547541,
      "step": 10830,
      "student_distance_mean": 1762.5068481445312,
      "teacher_distance_mean": 1.1439801454544067,
      "weighted_auxiliary_loss": 0.0006464874837547541,
      "weighted_relation_loss": 0.0006464874837547541
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.140318054256314,
      "flow_matching_loss": 0.042294280230998994,
      "grad_norm": 0.13092736899852753,
      "learning_rate": 7.576450085728959e-05,
      "loss": 0.0429,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006368425150867551,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.45151348114014,
      "state_query_norm": 1328.37197265625,
      "state_relation_loss": 0.0006368425150867551,
      "step": 10840,
      "student_distance_mean": 1791.89072265625,
      "teacher_distance_mean": 1.1547876715660095,
      "weighted_auxiliary_loss": 0.0006368425150867551,
      "weighted_relation_loss": 0.0006368425150867551
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.149672591206736,
      "flow_matching_loss": 0.04053445905447006,
      "grad_norm": 0.1232939064502716,
      "learning_rate": 7.571725021799885e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006192546512465924,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.826217269897462,
      "state_query_norm": 1296.6679809570312,
      "state_relation_loss": 0.0006192546512465924,
      "step": 10850,
      "student_distance_mean": 1757.2197021484376,
      "teacher_distance_mean": 1.1346705675125122,
      "weighted_auxiliary_loss": 0.0006192546512465924,
      "weighted_relation_loss": 0.0006192546512465924
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.159027128157156,
      "flow_matching_loss": 0.040876633301377294,
      "grad_norm": 0.10074839740991592,
      "learning_rate": 7.566996832982409e-05,
      "loss": 0.0415,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006191153021063656,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.10943489074707,
      "state_query_norm": 1302.0766967773438,
      "state_relation_loss": 0.0006191153021063656,
      "step": 10860,
      "student_distance_mean": 1775.5081176757812,
      "teacher_distance_mean": 1.151190996170044,
      "weighted_auxiliary_loss": 0.0006191153021063656,
      "weighted_relation_loss": 0.0006191153021063656
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.168381665107578,
      "flow_matching_loss": 0.03785206712782383,
      "grad_norm": 0.10933589190244675,
      "learning_rate": 7.56226552502173e-05,
      "loss": 0.0385,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006581063033081591,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.36924095153809,
      "state_query_norm": 1319.639501953125,
      "state_relation_loss": 0.0006581063033081591,
      "step": 10870,
      "student_distance_mean": 1788.6258666992187,
      "teacher_distance_mean": 1.1631566047668458,
      "weighted_auxiliary_loss": 0.0006581063033081591,
      "weighted_relation_loss": 0.0006581063033081591
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.177736202057998,
      "flow_matching_loss": 0.03924522995948791,
      "grad_norm": 0.13344044983386993,
      "learning_rate": 7.557531103666833e-05,
      "loss": 0.0399,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006245000695344061,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.73621711730957,
      "state_query_norm": 1289.8177490234375,
      "state_relation_loss": 0.0006245000695344061,
      "step": 10880,
      "student_distance_mean": 1754.8051147460938,
      "teacher_distance_mean": 1.144853663444519,
      "weighted_auxiliary_loss": 0.0006245000695344061,
      "weighted_relation_loss": 0.0006245000695344061
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.187090739008418,
      "flow_matching_loss": 0.03949851281940937,
      "grad_norm": 0.09268897026777267,
      "learning_rate": 7.552793574670485e-05,
      "loss": 0.0402,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006736037379596382,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.77071323394775,
      "state_query_norm": 1327.3957275390626,
      "state_relation_loss": 0.0006736037379596382,
      "step": 10890,
      "student_distance_mean": 1809.9207275390625,
      "teacher_distance_mean": 1.1786571145057678,
      "weighted_auxiliary_loss": 0.0006736037379596382,
      "weighted_relation_loss": 0.0006736037379596382
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.19644527595884,
      "flow_matching_loss": 0.041480864956974985,
      "grad_norm": 0.14608193933963776,
      "learning_rate": 7.54805294378923e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006241878378205002,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.24484214782715,
      "state_query_norm": 1322.1504638671875,
      "state_relation_loss": 0.0006241878378205002,
      "step": 10900,
      "student_distance_mean": 1788.0068359375,
      "teacher_distance_mean": 1.152252471446991,
      "weighted_auxiliary_loss": 0.0006241878378205002,
      "weighted_relation_loss": 0.0006241878378205002
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.20579981290926,
      "flow_matching_loss": 0.04205524437129497,
      "grad_norm": 0.17865145206451416,
      "learning_rate": 7.543309216783384e-05,
      "loss": 0.0427,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000649600097676739,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.859818840026854,
      "state_query_norm": 1332.93251953125,
      "state_relation_loss": 0.000649600097676739,
      "step": 10910,
      "student_distance_mean": 1812.4627807617187,
      "teacher_distance_mean": 1.1740346074104309,
      "weighted_auxiliary_loss": 0.000649600097676739,
      "weighted_relation_loss": 0.000649600097676739
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.215154349859683,
      "flow_matching_loss": 0.039398154243826866,
      "grad_norm": 0.19132593274116516,
      "learning_rate": 7.53856239941702e-05,
      "loss": 0.04,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000631696410709992,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.056692886352536,
      "state_query_norm": 1309.5767211914062,
      "state_relation_loss": 0.000631696410709992,
      "step": 10920,
      "student_distance_mean": 1773.570654296875,
      "teacher_distance_mean": 1.1501778721809388,
      "weighted_auxiliary_loss": 0.000631696410709992,
      "weighted_relation_loss": 0.000631696410709992
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.224508886810103,
      "flow_matching_loss": 0.039257713779807094,
      "grad_norm": 0.13906319439411163,
      "learning_rate": 7.533812497457972e-05,
      "loss": 0.0399,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006234789558220655,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.091396522521975,
      "state_query_norm": 1314.4797729492188,
      "state_relation_loss": 0.0006234789558220655,
      "step": 10930,
      "student_distance_mean": 1775.1759887695312,
      "teacher_distance_mean": 1.1539475917816162,
      "weighted_auxiliary_loss": 0.0006234789558220655,
      "weighted_relation_loss": 0.0006234789558220655
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.233863423760523,
      "flow_matching_loss": 0.04020299166440964,
      "grad_norm": 0.1293744295835495,
      "learning_rate": 7.529059516677814e-05,
      "loss": 0.0408,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006030511809512973,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.42005252838135,
      "state_query_norm": 1309.88544921875,
      "state_relation_loss": 0.0006030511809512973,
      "step": 10940,
      "student_distance_mean": 1784.2647827148437,
      "teacher_distance_mean": 1.1587789535522461,
      "weighted_auxiliary_loss": 0.0006030511809512973,
      "weighted_relation_loss": 0.0006030511809512973
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.243217960710945,
      "flow_matching_loss": 0.04044427834451199,
      "grad_norm": 0.1178537905216217,
      "learning_rate": 7.524303462851872e-05,
      "loss": 0.0411,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006212995969690383,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.13322277069092,
      "state_query_norm": 1303.3524291992187,
      "state_relation_loss": 0.0006212995969690383,
      "step": 10950,
      "student_distance_mean": 1772.0900634765626,
      "teacher_distance_mean": 1.1471614003181458,
      "weighted_auxiliary_loss": 0.0006212995969690383,
      "weighted_relation_loss": 0.0006212995969690383
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.252572497661365,
      "flow_matching_loss": 0.041503554582595824,
      "grad_norm": 0.13660281896591187,
      "learning_rate": 7.519544341759192e-05,
      "loss": 0.0421,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006026717310305685,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.209073066711426,
      "state_query_norm": 1321.1980590820312,
      "state_relation_loss": 0.0006026717310305685,
      "step": 10960,
      "student_distance_mean": 1782.2005737304687,
      "teacher_distance_mean": 1.1533042192459106,
      "weighted_auxiliary_loss": 0.0006026717310305685,
      "weighted_relation_loss": 0.0006026717310305685
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.261927034611787,
      "flow_matching_loss": 0.04059881754219532,
      "grad_norm": 0.10006290674209595,
      "learning_rate": 7.514782159182562e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006312373618129641,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.02206192016602,
      "state_query_norm": 1306.5662719726563,
      "state_relation_loss": 0.0006312373618129641,
      "step": 10970,
      "student_distance_mean": 1770.7328247070313,
      "teacher_distance_mean": 1.149190080165863,
      "weighted_auxiliary_loss": 0.0006312373618129641,
      "weighted_relation_loss": 0.0006312373618129641
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.271281571562207,
      "flow_matching_loss": 0.040293794870376584,
      "grad_norm": 0.13579657673835754,
      "learning_rate": 7.510016920908481e-05,
      "loss": 0.0409,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005954036605544388,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.071106910705566,
      "state_query_norm": 1314.4744995117187,
      "state_relation_loss": 0.0005954036605544388,
      "step": 10980,
      "student_distance_mean": 1774.4116943359375,
      "teacher_distance_mean": 1.1510161757469177,
      "weighted_auxiliary_loss": 0.0005954036605544388,
      "weighted_relation_loss": 0.0005954036605544388
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.28063610851263,
      "flow_matching_loss": 0.040150325372815134,
      "grad_norm": 0.1358320713043213,
      "learning_rate": 7.505248632727158e-05,
      "loss": 0.0408,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006287571741268039,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.31113948822021,
      "state_query_norm": 1315.6558349609375,
      "state_relation_loss": 0.0006287571741268039,
      "step": 10990,
      "student_distance_mean": 1783.4719360351562,
      "teacher_distance_mean": 1.157043218612671,
      "weighted_auxiliary_loss": 0.0006287571741268039,
      "weighted_relation_loss": 0.0006287571741268039
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.28999064546305,
      "flow_matching_loss": 0.04086372368037701,
      "grad_norm": 0.15960614383220673,
      "learning_rate": 7.50047730043252e-05,
      "loss": 0.0415,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006607446644920856,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.37123832702637,
      "state_query_norm": 1315.6381469726562,
      "state_relation_loss": 0.0006607446644920856,
      "step": 11000,
      "student_distance_mean": 1785.3734008789063,
      "teacher_distance_mean": 1.1491452932357789,
      "weighted_auxiliary_loss": 0.0006607446644920856,
      "weighted_relation_loss": 0.0006607446644920856
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.29934518241347,
      "flow_matching_loss": 0.039303159713745116,
      "grad_norm": 0.1796140968799591,
      "learning_rate": 7.495702929822183e-05,
      "loss": 0.0399,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006015354010742158,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.088569450378415,
      "state_query_norm": 1347.9715942382813,
      "state_relation_loss": 0.0006015354010742158,
      "step": 11010,
      "student_distance_mean": 1829.4561767578125,
      "teacher_distance_mean": 1.1665374279022216,
      "weighted_auxiliary_loss": 0.0006015354010742158,
      "weighted_relation_loss": 0.0006015354010742158
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.308699719363892,
      "flow_matching_loss": 0.03827819712460041,
      "grad_norm": 0.10142295062541962,
      "learning_rate": 7.490925526697455e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000628141564084217,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.49404754638672,
      "state_query_norm": 1329.9228515625,
      "state_relation_loss": 0.000628141564084217,
      "step": 11020,
      "student_distance_mean": 1796.4864624023437,
      "teacher_distance_mean": 1.1489926815032958,
      "weighted_auxiliary_loss": 0.000628141564084217,
      "weighted_relation_loss": 0.000628141564084217
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.318054256314312,
      "flow_matching_loss": 0.04033961296081543,
      "grad_norm": 0.13051767647266388,
      "learning_rate": 7.486145096863334e-05,
      "loss": 0.041,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006360461935400963,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.58314838409424,
      "state_query_norm": 1324.66884765625,
      "state_relation_loss": 0.0006360461935400963,
      "step": 11030,
      "student_distance_mean": 1798.917919921875,
      "teacher_distance_mean": 1.150620973110199,
      "weighted_auxiliary_loss": 0.0006360461935400963,
      "weighted_relation_loss": 0.0006360461935400963
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.327408793264734,
      "flow_matching_loss": 0.04068263322114944,
      "grad_norm": 0.11796294152736664,
      "learning_rate": 7.481361646128491e-05,
      "loss": 0.0413,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005890757427550852,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.97723960876465,
      "state_query_norm": 1306.7515380859375,
      "state_relation_loss": 0.0005890757427550852,
      "step": 11040,
      "student_distance_mean": 1769.321728515625,
      "teacher_distance_mean": 1.1370363235473633,
      "weighted_auxiliary_loss": 0.0005890757427550852,
      "weighted_relation_loss": 0.0005890757427550852
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.336763330215154,
      "flow_matching_loss": 0.04178622700273991,
      "grad_norm": 0.12648551166057587,
      "learning_rate": 7.476575180305271e-05,
      "loss": 0.0424,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005979821376968175,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.185997581481935,
      "state_query_norm": 1325.934716796875,
      "state_relation_loss": 0.0005979821376968175,
      "step": 11050,
      "student_distance_mean": 1784.0624755859376,
      "teacher_distance_mean": 1.1490174889564515,
      "weighted_auxiliary_loss": 0.0005979821376968175,
      "weighted_relation_loss": 0.0005979821376968175
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.346117867165574,
      "flow_matching_loss": 0.03852497786283493,
      "grad_norm": 0.11336931586265564,
      "learning_rate": 7.471785705209682e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006174265814479441,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.24292697906494,
      "state_query_norm": 1310.1994506835938,
      "state_relation_loss": 0.0006174265814479441,
      "step": 11060,
      "student_distance_mean": 1783.2118286132813,
      "teacher_distance_mean": 1.1514106869697571,
      "weighted_auxiliary_loss": 0.0006174265814479441,
      "weighted_relation_loss": 0.0006174265814479441
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.355472404115996,
      "flow_matching_loss": 0.041197870299220084,
      "grad_norm": 0.12105278670787811,
      "learning_rate": 7.466993226661387e-05,
      "loss": 0.0418,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006281412148382515,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.123999977111815,
      "state_query_norm": 1314.5454467773438,
      "state_relation_loss": 0.0006281412148382515,
      "step": 11070,
      "student_distance_mean": 1774.4657592773438,
      "teacher_distance_mean": 1.1502950072288514,
      "weighted_auxiliary_loss": 0.0006281412148382515,
      "weighted_relation_loss": 0.0006281412148382515
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.364826941066417,
      "flow_matching_loss": 0.04027037434279919,
      "grad_norm": 0.10566837340593338,
      "learning_rate": 7.462197750483699e-05,
      "loss": 0.0409,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005867945845238864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.360374450683594,
      "state_query_norm": 1319.8800903320312,
      "state_relation_loss": 0.0005867945845238864,
      "step": 11080,
      "student_distance_mean": 1787.1699951171875,
      "teacher_distance_mean": 1.1532877802848815,
      "weighted_auxiliary_loss": 0.0005867945845238864,
      "weighted_relation_loss": 0.0005867945845238864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.374181478016839,
      "flow_matching_loss": 0.0380689661949873,
      "grad_norm": 0.10922476649284363,
      "learning_rate": 7.457399282503574e-05,
      "loss": 0.0387,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006194970977958292,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.984814453125,
      "state_query_norm": 1292.1987060546876,
      "state_relation_loss": 0.0006194970977958292,
      "step": 11090,
      "student_distance_mean": 1763.2296142578125,
      "teacher_distance_mean": 1.1433030486106872,
      "weighted_auxiliary_loss": 0.0006194970977958292,
      "weighted_relation_loss": 0.0006194970977958292
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.383536014967259,
      "flow_matching_loss": 0.04041159003973007,
      "grad_norm": 0.13355191051959991,
      "learning_rate": 7.452597828551604e-05,
      "loss": 0.041,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006090914190281183,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.41574192047119,
      "state_query_norm": 1313.9025146484375,
      "state_relation_loss": 0.0006090914190281183,
      "step": 11100,
      "student_distance_mean": 1788.0641479492188,
      "teacher_distance_mean": 1.1451019287109374,
      "weighted_auxiliary_loss": 0.0006090914190281183,
      "weighted_relation_loss": 0.0006090914190281183
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.39289055191768,
      "flow_matching_loss": 0.03797328770160675,
      "grad_norm": 0.13910764455795288,
      "learning_rate": 7.447793394462006e-05,
      "loss": 0.0386,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006065926281735301,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.966407775878906,
      "state_query_norm": 1343.018310546875,
      "state_relation_loss": 0.0006065926281735301,
      "step": 11110,
      "student_distance_mean": 1820.3564086914062,
      "teacher_distance_mean": 1.1599520325660706,
      "weighted_auxiliary_loss": 0.0006065926281735301,
      "weighted_relation_loss": 0.0006065926281735301
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.402245088868101,
      "flow_matching_loss": 0.038438502326607706,
      "grad_norm": 0.14037196338176727,
      "learning_rate": 7.442985986072624e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006343294109683484,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.68510093688965,
      "state_query_norm": 1336.2575927734374,
      "state_relation_loss": 0.0006343294109683484,
      "step": 11120,
      "student_distance_mean": 1806.4653930664062,
      "teacher_distance_mean": 1.1532615423202515,
      "weighted_auxiliary_loss": 0.0006343294109683484,
      "weighted_relation_loss": 0.0006343294109683484
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.411599625818521,
      "flow_matching_loss": 0.03827893659472466,
      "grad_norm": 0.15249691903591156,
      "learning_rate": 7.438175609224908e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006119226105511189,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.86874217987061,
      "state_query_norm": 1339.6226928710937,
      "state_relation_loss": 0.0006119226105511189,
      "step": 11130,
      "student_distance_mean": 1815.8636962890625,
      "teacher_distance_mean": 1.164570939540863,
      "weighted_auxiliary_loss": 0.0006119226105511189,
      "weighted_relation_loss": 0.0006119226105511189
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.420954162768943,
      "flow_matching_loss": 0.0413011197000742,
      "grad_norm": 0.1441648304462433,
      "learning_rate": 7.433362269763924e-05,
      "loss": 0.0419,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006277247914113105,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.80277328491211,
      "state_query_norm": 1328.2766967773437,
      "state_relation_loss": 0.0006277247914113105,
      "step": 11140,
      "student_distance_mean": 1807.7892456054688,
      "teacher_distance_mean": 1.157539701461792,
      "weighted_auxiliary_loss": 0.0006277247914113105,
      "weighted_relation_loss": 0.0006277247914113105
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.430308699719363,
      "flow_matching_loss": 0.04341215007007122,
      "grad_norm": 0.14208205044269562,
      "learning_rate": 7.428545973538329e-05,
      "loss": 0.044,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00059321807930246,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.24561958312988,
      "state_query_norm": 1321.910009765625,
      "state_relation_loss": 0.00059321807930246,
      "step": 11150,
      "student_distance_mean": 1782.1422485351563,
      "teacher_distance_mean": 1.1435362219810485,
      "weighted_auxiliary_loss": 0.00059321807930246,
      "weighted_relation_loss": 0.00059321807930246
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.439663236669785,
      "flow_matching_loss": 0.03824652694165707,
      "grad_norm": 0.10852847248315811,
      "learning_rate": 7.42372672640038e-05,
      "loss": 0.0388,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000578634423436597,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.42563591003418,
      "state_query_norm": 1325.5524658203126,
      "state_relation_loss": 0.000578634423436597,
      "step": 11160,
      "student_distance_mean": 1794.378955078125,
      "teacher_distance_mean": 1.1530118107795715,
      "weighted_auxiliary_loss": 0.000578634423436597,
      "weighted_relation_loss": 0.000578634423436597
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.449017773620206,
      "flow_matching_loss": 0.03878291640430689,
      "grad_norm": 0.1579413264989853,
      "learning_rate": 7.418904534205917e-05,
      "loss": 0.0394,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000595052121207118,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.907526206970214,
      "state_query_norm": 1300.4577758789062,
      "state_relation_loss": 0.000595052121207118,
      "step": 11170,
      "student_distance_mean": 1766.1909790039062,
      "teacher_distance_mean": 1.1359325408935548,
      "weighted_auxiliary_loss": 0.000595052121207118,
      "weighted_relation_loss": 0.000595052121207118
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.458372310570628,
      "flow_matching_loss": 0.04052675403654575,
      "grad_norm": 0.13586701452732086,
      "learning_rate": 7.414079402814356e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006306915485765785,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.7004753112793,
      "state_query_norm": 1328.9955688476562,
      "state_relation_loss": 0.0006306915485765785,
      "step": 11180,
      "student_distance_mean": 1803.6582397460938,
      "teacher_distance_mean": 1.1594458222389221,
      "weighted_auxiliary_loss": 0.0006306915485765785,
      "weighted_relation_loss": 0.0006306915485765785
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.467726847521048,
      "flow_matching_loss": 0.041933394968509674,
      "grad_norm": 0.12997666001319885,
      "learning_rate": 7.409251338088687e-05,
      "loss": 0.0426,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006287550204433501,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.71439781188965,
      "state_query_norm": 1327.5742431640624,
      "state_relation_loss": 0.0006287550204433501,
      "step": 11190,
      "student_distance_mean": 1807.9687622070312,
      "teacher_distance_mean": 1.160853672027588,
      "weighted_auxiliary_loss": 0.0006287550204433501,
      "weighted_relation_loss": 0.0006287550204433501
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.477081384471468,
      "flow_matching_loss": 0.039448685199022296,
      "grad_norm": 0.14108404517173767,
      "learning_rate": 7.404420345895467e-05,
      "loss": 0.0401,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006182811921462417,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.96528587341309,
      "state_query_norm": 1342.9066284179687,
      "state_relation_loss": 0.0006182811921462417,
      "step": 11200,
      "student_distance_mean": 1823.3795776367188,
      "teacher_distance_mean": 1.1645800471305847,
      "weighted_auxiliary_loss": 0.0006182811921462417,
      "weighted_relation_loss": 0.0006182811921462417
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.48643592142189,
      "flow_matching_loss": 0.03927184939384461,
      "grad_norm": 0.12467394769191742,
      "learning_rate": 7.399586432104804e-05,
      "loss": 0.0399,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006401025224477052,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.85596809387207,
      "state_query_norm": 1334.5124877929688,
      "state_relation_loss": 0.0006401025224477052,
      "step": 11210,
      "student_distance_mean": 1812.2829711914062,
      "teacher_distance_mean": 1.1618389010429382,
      "weighted_auxiliary_loss": 0.0006401025224477052,
      "weighted_relation_loss": 0.0006401025224477052
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.49579045837231,
      "flow_matching_loss": 0.03998149298131466,
      "grad_norm": 0.12726019322872162,
      "learning_rate": 7.394749602590358e-05,
      "loss": 0.0406,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006252871069591492,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.98017978668213,
      "state_query_norm": 1347.1694213867188,
      "state_relation_loss": 0.0006252871069591492,
      "step": 11220,
      "student_distance_mean": 1823.472900390625,
      "teacher_distance_mean": 1.1686756253242492,
      "weighted_auxiliary_loss": 0.0006252871069591492,
      "weighted_relation_loss": 0.0006252871069591492
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.505144995322732,
      "flow_matching_loss": 0.03851690068840981,
      "grad_norm": 0.13466042280197144,
      "learning_rate": 7.389909863229336e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006021493230946362,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.583896446228025,
      "state_query_norm": 1322.6400512695313,
      "state_relation_loss": 0.0006021493230946362,
      "step": 11230,
      "student_distance_mean": 1800.0660278320313,
      "teacher_distance_mean": 1.1478474736213684,
      "weighted_auxiliary_loss": 0.0006021493230946362,
      "weighted_relation_loss": 0.0006021493230946362
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.514499532273152,
      "flow_matching_loss": 0.04132068268954754,
      "grad_norm": 0.1244630292057991,
      "learning_rate": 7.385067219902477e-05,
      "loss": 0.0419,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005949047103058547,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.39200057983398,
      "state_query_norm": 1319.5898803710938,
      "state_relation_loss": 0.0005949047103058547,
      "step": 11240,
      "student_distance_mean": 1791.2288208007812,
      "teacher_distance_mean": 1.1421838521957397,
      "weighted_auxiliary_loss": 0.0005949047103058547,
      "weighted_relation_loss": 0.0005949047103058547
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.523854069223573,
      "flow_matching_loss": 0.03800697512924671,
      "grad_norm": 0.19883280992507935,
      "learning_rate": 7.380221678494049e-05,
      "loss": 0.0386,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000615769490832463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.853090858459474,
      "state_query_norm": 1337.1849609375,
      "state_relation_loss": 0.000615769490832463,
      "step": 11250,
      "student_distance_mean": 1815.444970703125,
      "teacher_distance_mean": 1.1571192264556884,
      "weighted_auxiliary_loss": 0.000615769490832463,
      "weighted_relation_loss": 0.000615769490832463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.533208606173995,
      "flow_matching_loss": 0.039378656446933745,
      "grad_norm": 0.14233997464179993,
      "learning_rate": 7.37537324489184e-05,
      "loss": 0.04,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006143902195617556,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.02563648223877,
      "state_query_norm": 1338.6469116210938,
      "state_relation_loss": 0.0006143902195617556,
      "step": 11260,
      "student_distance_mean": 1822.8794921875,
      "teacher_distance_mean": 1.162249791622162,
      "weighted_auxiliary_loss": 0.0006143902195617556,
      "weighted_relation_loss": 0.0006143902195617556
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.542563143124415,
      "flow_matching_loss": 0.03978653997182846,
      "grad_norm": 0.11951562762260437,
      "learning_rate": 7.370521924987155e-05,
      "loss": 0.0404,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000625148112885654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.87227878570557,
      "state_query_norm": 1336.0763916015626,
      "state_relation_loss": 0.000625148112885654,
      "step": 11270,
      "student_distance_mean": 1814.0029052734376,
      "teacher_distance_mean": 1.1530578970909118,
      "weighted_auxiliary_loss": 0.000625148112885654,
      "weighted_relation_loss": 0.000625148112885654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.551917680074837,
      "flow_matching_loss": 0.040224894881248474,
      "grad_norm": 0.1071835532784462,
      "learning_rate": 7.365667724674806e-05,
      "loss": 0.0408,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006240196875296533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.659830474853514,
      "state_query_norm": 1337.005419921875,
      "state_relation_loss": 0.0006240196875296533,
      "step": 11280,
      "student_distance_mean": 1806.9295043945312,
      "teacher_distance_mean": 1.1517713069915771,
      "weighted_auxiliary_loss": 0.0006240196875296533,
      "weighted_relation_loss": 0.0006240196875296533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.561272217025257,
      "flow_matching_loss": 0.040268654376268385,
      "grad_norm": 0.09335040301084518,
      "learning_rate": 7.360810649853105e-05,
      "loss": 0.0409,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006060501211322844,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.71412887573242,
      "state_query_norm": 1334.1845336914062,
      "state_relation_loss": 0.0006060501211322844,
      "step": 11290,
      "student_distance_mean": 1806.6835571289062,
      "teacher_distance_mean": 1.1482898473739624,
      "weighted_auxiliary_loss": 0.0006060501211322844,
      "weighted_relation_loss": 0.0006060501211322844
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.570626753975677,
      "flow_matching_loss": 0.03805658221244812,
      "grad_norm": 0.11750943958759308,
      "learning_rate": 7.355950706423853e-05,
      "loss": 0.0387,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005940737028140575,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 31.827689361572265,
      "state_query_norm": 1301.1280395507813,
      "state_relation_loss": 0.0005940737028140575,
      "step": 11300,
      "student_distance_mean": 1760.0403564453125,
      "teacher_distance_mean": 1.1302901268005372,
      "weighted_auxiliary_loss": 0.0005940737028140575,
      "weighted_relation_loss": 0.0005940737028140575
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.5799812909261,
      "flow_matching_loss": 0.03890969529747963,
      "grad_norm": 0.12060834467411041,
      "learning_rate": 7.351087900292342e-05,
      "loss": 0.0395,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006096239027101547,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.7908073425293,
      "state_query_norm": 1337.2298095703125,
      "state_relation_loss": 0.0006096239027101547,
      "step": 11310,
      "student_distance_mean": 1808.644580078125,
      "teacher_distance_mean": 1.1565472960472107,
      "weighted_auxiliary_loss": 0.0006096239027101547,
      "weighted_relation_loss": 0.0006096239027101547
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.58933582787652,
      "flow_matching_loss": 0.0383470419794321,
      "grad_norm": 0.12896645069122314,
      "learning_rate": 7.346222237367339e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005907474202103913,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.0720157623291,
      "state_query_norm": 1346.5433471679687,
      "state_relation_loss": 0.0005907474202103913,
      "step": 11320,
      "student_distance_mean": 1827.2170288085938,
      "teacher_distance_mean": 1.162272071838379,
      "weighted_auxiliary_loss": 0.0005907474202103913,
      "weighted_relation_loss": 0.0005907474202103913
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.598690364826941,
      "flow_matching_loss": 0.03913928493857384,
      "grad_norm": 0.0991445854306221,
      "learning_rate": 7.341353723561081e-05,
      "loss": 0.0397,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005911905376706272,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.23206481933594,
      "state_query_norm": 1358.81552734375,
      "state_relation_loss": 0.0005911905376706272,
      "step": 11330,
      "student_distance_mean": 1833.4388061523437,
      "teacher_distance_mean": 1.1644087433815002,
      "weighted_auxiliary_loss": 0.0005911905376706272,
      "weighted_relation_loss": 0.0005911905376706272
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.608044901777362,
      "flow_matching_loss": 0.03780097737908363,
      "grad_norm": 0.1004890725016594,
      "learning_rate": 7.336482364789277e-05,
      "loss": 0.0384,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006336582358926534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.23194770812988,
      "state_query_norm": 1343.7998779296875,
      "state_relation_loss": 0.0006336582358926534,
      "step": 11340,
      "student_distance_mean": 1830.7540405273437,
      "teacher_distance_mean": 1.1673271894454955,
      "weighted_auxiliary_loss": 0.0006336582358926534,
      "weighted_relation_loss": 0.0006336582358926534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.617399438727784,
      "flow_matching_loss": 0.038418065011501315,
      "grad_norm": 0.12553980946540833,
      "learning_rate": 7.331608166971082e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000648751575499773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.63772602081299,
      "state_query_norm": 1321.3235717773437,
      "state_relation_loss": 0.000648751575499773,
      "step": 11350,
      "student_distance_mean": 1800.288623046875,
      "teacher_distance_mean": 1.1507633566856383,
      "weighted_auxiliary_loss": 0.000648751575499773,
      "weighted_relation_loss": 0.000648751575499773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.626753975678204,
      "flow_matching_loss": 0.03773919381201267,
      "grad_norm": 0.1047990545630455,
      "learning_rate": 7.326731136029108e-05,
      "loss": 0.0384,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006263435701839626,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.83920059204102,
      "state_query_norm": 1339.2062866210938,
      "state_relation_loss": 0.0006263435701839626,
      "step": 11360,
      "student_distance_mean": 1814.6335083007812,
      "teacher_distance_mean": 1.160363757610321,
      "weighted_auxiliary_loss": 0.0006263435701839626,
      "weighted_relation_loss": 0.0006263435701839626
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.636108512628624,
      "flow_matching_loss": 0.03824208863079548,
      "grad_norm": 0.1602710336446762,
      "learning_rate": 7.321851277889408e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006180311902426183,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.06192588806152,
      "state_query_norm": 1339.5638427734375,
      "state_relation_loss": 0.0006180311902426183,
      "step": 11370,
      "student_distance_mean": 1819.8884399414062,
      "teacher_distance_mean": 1.1582100152969361,
      "weighted_auxiliary_loss": 0.0006180311902426183,
      "weighted_relation_loss": 0.0006180311902426183
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.645463049579046,
      "flow_matching_loss": 0.04060981534421444,
      "grad_norm": 0.12407921254634857,
      "learning_rate": 7.316968598481469e-05,
      "loss": 0.0412,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005971930513624102,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.6884880065918,
      "state_query_norm": 1331.3662841796875,
      "state_relation_loss": 0.0005971930513624102,
      "step": 11380,
      "student_distance_mean": 1802.2321044921875,
      "teacher_distance_mean": 1.1466181397438049,
      "weighted_auxiliary_loss": 0.0005971930513624102,
      "weighted_relation_loss": 0.0005971930513624102
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.654817586529466,
      "flow_matching_loss": 0.03923499882221222,
      "grad_norm": 0.10086424648761749,
      "learning_rate": 7.312083103738206e-05,
      "loss": 0.0398,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005833879113197327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.85066223144531,
      "state_query_norm": 1344.9959716796875,
      "state_relation_loss": 0.0005833879113197327,
      "step": 11390,
      "student_distance_mean": 1813.6611938476562,
      "teacher_distance_mean": 1.1513432502746581,
      "weighted_auxiliary_loss": 0.0005833879113197327,
      "weighted_relation_loss": 0.0005833879113197327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.664172123479888,
      "flow_matching_loss": 0.040205090492963794,
      "grad_norm": 0.13952462375164032,
      "learning_rate": 7.307194799595958e-05,
      "loss": 0.0408,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000594258785713464,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.418386459350586,
      "state_query_norm": 1335.41533203125,
      "state_relation_loss": 0.000594258785713464,
      "step": 11400,
      "student_distance_mean": 1795.0102905273438,
      "teacher_distance_mean": 1.1412037134170532,
      "weighted_auxiliary_loss": 0.000594258785713464,
      "weighted_relation_loss": 0.000594258785713464
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.673526660430309,
      "flow_matching_loss": 0.03825989291071892,
      "grad_norm": 0.13203980028629303,
      "learning_rate": 7.302303691994473e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006051460630260407,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.19418907165527,
      "state_query_norm": 1349.9439575195313,
      "state_relation_loss": 0.0006051460630260407,
      "step": 11410,
      "student_distance_mean": 1833.4255126953126,
      "teacher_distance_mean": 1.1645191669464112,
      "weighted_auxiliary_loss": 0.0006051460630260407,
      "weighted_relation_loss": 0.0006051460630260407
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.68288119738073,
      "flow_matching_loss": 0.03840640038251877,
      "grad_norm": 0.1405155211687088,
      "learning_rate": 7.29740978687691e-05,
      "loss": 0.039,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006131294823717326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.07929000854492,
      "state_query_norm": 1347.52548828125,
      "state_relation_loss": 0.0006131294823717326,
      "step": 11420,
      "student_distance_mean": 1828.25185546875,
      "teacher_distance_mean": 1.159994661808014,
      "weighted_auxiliary_loss": 0.0006131294823717326,
      "weighted_relation_loss": 0.0006131294823717326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.69223573433115,
      "flow_matching_loss": 0.037111349403858185,
      "grad_norm": 0.21637892723083496,
      "learning_rate": 7.292513090189825e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000609390449244529,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.650676536560056,
      "state_query_norm": 1331.0541625976562,
      "state_relation_loss": 0.000609390449244529,
      "step": 11430,
      "student_distance_mean": 1801.4759643554687,
      "teacher_distance_mean": 1.142202341556549,
      "weighted_auxiliary_loss": 0.000609390449244529,
      "weighted_relation_loss": 0.000609390449244529
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.701590271281571,
      "flow_matching_loss": 0.03968779146671295,
      "grad_norm": 0.14963126182556152,
      "learning_rate": 7.287613607883163e-05,
      "loss": 0.0403,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005903392797335982,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.88055896759033,
      "state_query_norm": 1334.0636962890626,
      "state_relation_loss": 0.0005903392797335982,
      "step": 11440,
      "student_distance_mean": 1817.6373901367188,
      "teacher_distance_mean": 1.150760328769684,
      "weighted_auxiliary_loss": 0.0005903392797335982,
      "weighted_relation_loss": 0.0005903392797335982
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.710944808231993,
      "flow_matching_loss": 0.03684267513453961,
      "grad_norm": 0.1119173914194107,
      "learning_rate": 7.282711345910263e-05,
      "loss": 0.0374,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000599049188895151,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.18198356628418,
      "state_query_norm": 1346.5725830078125,
      "state_relation_loss": 0.000599049188895151,
      "step": 11450,
      "student_distance_mean": 1830.8432006835938,
      "teacher_distance_mean": 1.1560906052589417,
      "weighted_auxiliary_loss": 0.000599049188895151,
      "weighted_relation_loss": 0.000599049188895151
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.720299345182413,
      "flow_matching_loss": 0.03709213137626648,
      "grad_norm": 0.11727901548147202,
      "learning_rate": 7.27780631022783e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005961054877843708,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.35331230163574,
      "state_query_norm": 1356.0940551757812,
      "state_relation_loss": 0.0005961054877843708,
      "step": 11460,
      "student_distance_mean": 1838.454296875,
      "teacher_distance_mean": 1.1615654468536376,
      "weighted_auxiliary_loss": 0.0005961054877843708,
      "weighted_relation_loss": 0.0005961054877843708
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.729653882132835,
      "flow_matching_loss": 0.036238780803978445,
      "grad_norm": 0.12535397708415985,
      "learning_rate": 7.272898506795948e-05,
      "loss": 0.0369,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000618835975183174,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.19822673797607,
      "state_query_norm": 1344.1770263671874,
      "state_relation_loss": 0.000618835975183174,
      "step": 11470,
      "student_distance_mean": 1830.9947143554687,
      "teacher_distance_mean": 1.15617516040802,
      "weighted_auxiliary_loss": 0.000618835975183174,
      "weighted_relation_loss": 0.000618835975183174
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.739008419083255,
      "flow_matching_loss": 0.036462200060486796,
      "grad_norm": 0.08854931592941284,
      "learning_rate": 7.267987941578058e-05,
      "loss": 0.0371,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006047348782885819,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.71125869750976,
      "state_query_norm": 1329.501513671875,
      "state_relation_loss": 0.0006047348782885819,
      "step": 11480,
      "student_distance_mean": 1804.9830444335937,
      "teacher_distance_mean": 1.142458987236023,
      "weighted_auxiliary_loss": 0.0006047348782885819,
      "weighted_relation_loss": 0.0006047348782885819
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.748362956033676,
      "flow_matching_loss": 0.03825057540088892,
      "grad_norm": 0.12831690907478333,
      "learning_rate": 7.263074620540962e-05,
      "loss": 0.0388,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000569373311009258,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.119095420837404,
      "state_query_norm": 1312.8804565429687,
      "state_relation_loss": 0.000569373311009258,
      "step": 11490,
      "student_distance_mean": 1776.668994140625,
      "teacher_distance_mean": 1.1295941591262817,
      "weighted_auxiliary_loss": 0.000569373311009258,
      "weighted_relation_loss": 0.000569373311009258
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.757717492984098,
      "flow_matching_loss": 0.03752472475171089,
      "grad_norm": 0.1392413228750229,
      "learning_rate": 7.25815854965481e-05,
      "loss": 0.0381,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005796431156340987,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.86040058135986,
      "state_query_norm": 1325.7513305664063,
      "state_relation_loss": 0.0005796431156340987,
      "step": 11500,
      "student_distance_mean": 1810.6853393554688,
      "teacher_distance_mean": 1.1526410222053527,
      "weighted_auxiliary_loss": 0.0005796431156340987,
      "weighted_relation_loss": 0.0005796431156340987
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.767072029934518,
      "flow_matching_loss": 0.038744830340147016,
      "grad_norm": 0.08988185971975327,
      "learning_rate": 7.253239734893089e-05,
      "loss": 0.0393,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005759820051025599,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.45398101806641,
      "state_query_norm": 1318.506884765625,
      "state_relation_loss": 0.0005759820051025599,
      "step": 11510,
      "student_distance_mean": 1789.3916015625,
      "teacher_distance_mean": 1.138198935985565,
      "weighted_auxiliary_loss": 0.0005759820051025599,
      "weighted_relation_loss": 0.0005759820051025599
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.77642656688494,
      "flow_matching_loss": 0.03841242920607328,
      "grad_norm": 0.13076692819595337,
      "learning_rate": 7.248318182232623e-05,
      "loss": 0.039,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005700546258594841,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.319353103637695,
      "state_query_norm": 1363.891552734375,
      "state_relation_loss": 0.0005700546258594841,
      "step": 11520,
      "student_distance_mean": 1844.5584594726563,
      "teacher_distance_mean": 1.1719534635543822,
      "weighted_auxiliary_loss": 0.0005700546258594841,
      "weighted_relation_loss": 0.0005700546258594841
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.78578110383536,
      "flow_matching_loss": 0.038593525812029836,
      "grad_norm": 0.08930203318595886,
      "learning_rate": 7.243393897653565e-05,
      "loss": 0.0392,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005762523389421404,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.720611572265625,
      "state_query_norm": 1332.6052001953126,
      "state_relation_loss": 0.0005762523389421404,
      "step": 11530,
      "student_distance_mean": 1810.1319213867187,
      "teacher_distance_mean": 1.149935817718506,
      "weighted_auxiliary_loss": 0.0005762523389421404,
      "weighted_relation_loss": 0.0005762523389421404
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.795135640785782,
      "flow_matching_loss": 0.036397726833820344,
      "grad_norm": 0.12862107157707214,
      "learning_rate": 7.23846688713938e-05,
      "loss": 0.037,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005898543167859316,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.73298664093018,
      "state_query_norm": 1327.4484497070312,
      "state_relation_loss": 0.0005898543167859316,
      "step": 11540,
      "student_distance_mean": 1804.841064453125,
      "teacher_distance_mean": 1.1469216108322144,
      "weighted_auxiliary_loss": 0.0005898543167859316,
      "weighted_relation_loss": 0.0005898543167859316
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.804490177736202,
      "flow_matching_loss": 0.03811969086527824,
      "grad_norm": 0.11035402119159698,
      "learning_rate": 7.233537156676854e-05,
      "loss": 0.0387,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005600003234576434,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.41546516418457,
      "state_query_norm": 1321.08095703125,
      "state_relation_loss": 0.0005600003234576434,
      "step": 11550,
      "student_distance_mean": 1792.5106323242187,
      "teacher_distance_mean": 1.1454281687736512,
      "weighted_auxiliary_loss": 0.0005600003234576434,
      "weighted_relation_loss": 0.0005600003234576434
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.813844714686622,
      "flow_matching_loss": 0.036017540097236636,
      "grad_norm": 0.12307785451412201,
      "learning_rate": 7.228604712256076e-05,
      "loss": 0.0366,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000591207982506603,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.83272933959961,
      "state_query_norm": 1343.0475219726563,
      "state_relation_loss": 0.000591207982506603,
      "step": 11560,
      "student_distance_mean": 1813.8900512695313,
      "teacher_distance_mean": 1.1546969056129455,
      "weighted_auxiliary_loss": 0.000591207982506603,
      "weighted_relation_loss": 0.000591207982506603
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.823199251637044,
      "flow_matching_loss": 0.037188243865966794,
      "grad_norm": 0.1180344820022583,
      "learning_rate": 7.223669559870426e-05,
      "loss": 0.0378,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006088348687626422,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.303412437438965,
      "state_query_norm": 1363.4956176757812,
      "state_relation_loss": 0.0006088348687626422,
      "step": 11570,
      "student_distance_mean": 1843.20361328125,
      "teacher_distance_mean": 1.164431869983673,
      "weighted_auxiliary_loss": 0.0006088348687626422,
      "weighted_relation_loss": 0.0006088348687626422
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.832553788587465,
      "flow_matching_loss": 0.037847939878702164,
      "grad_norm": 0.11396545171737671,
      "learning_rate": 7.218731705516585e-05,
      "loss": 0.0385,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006135001429356634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.41565589904785,
      "state_query_norm": 1349.7177124023438,
      "state_relation_loss": 0.0006135001429356634,
      "step": 11580,
      "student_distance_mean": 1837.5981323242188,
      "teacher_distance_mean": 1.1548387050628661,
      "weighted_auxiliary_loss": 0.0006135001429356634,
      "weighted_relation_loss": 0.0006135001429356634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.841908325537887,
      "flow_matching_loss": 0.03884459473192692,
      "grad_norm": 0.11814863234758377,
      "learning_rate": 7.21379115519451e-05,
      "loss": 0.0394,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005599216558039189,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.91262302398682,
      "state_query_norm": 1340.8570556640625,
      "state_relation_loss": 0.0005599216558039189,
      "step": 11590,
      "student_distance_mean": 1820.066259765625,
      "teacher_distance_mean": 1.1480937480926514,
      "weighted_auxiliary_loss": 0.0005599216558039189,
      "weighted_relation_loss": 0.0005599216558039189
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.851262862488307,
      "flow_matching_loss": 0.03877709619700909,
      "grad_norm": 0.1329997032880783,
      "learning_rate": 7.20884791490743e-05,
      "loss": 0.0394,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006105610402300954,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.00461902618408,
      "state_query_norm": 1334.5751098632813,
      "state_relation_loss": 0.0006105610402300954,
      "step": 11600,
      "student_distance_mean": 1818.6750366210938,
      "teacher_distance_mean": 1.1500265002250671,
      "weighted_auxiliary_loss": 0.0006105610402300954,
      "weighted_relation_loss": 0.0006105610402300954
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.860617399438727,
      "flow_matching_loss": 0.03834195397794247,
      "grad_norm": 0.16090092062950134,
      "learning_rate": 7.203901990661857e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005955820088274777,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.7951681137085,
      "state_query_norm": 1326.9329467773437,
      "state_relation_loss": 0.0005955820088274777,
      "step": 11610,
      "student_distance_mean": 1806.88681640625,
      "teacher_distance_mean": 1.1414756059646607,
      "weighted_auxiliary_loss": 0.0005955820088274777,
      "weighted_relation_loss": 0.0005955820088274777
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.869971936389149,
      "flow_matching_loss": 0.03781303279101848,
      "grad_norm": 0.1718490719795227,
      "learning_rate": 7.198953388467549e-05,
      "loss": 0.0384,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006187173188664019,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.982384490966794,
      "state_query_norm": 1335.6457641601562,
      "state_relation_loss": 0.0006187173188664019,
      "step": 11620,
      "student_distance_mean": 1818.7190795898437,
      "teacher_distance_mean": 1.1492295861244202,
      "weighted_auxiliary_loss": 0.0006187173188664019,
      "weighted_relation_loss": 0.0006187173188664019
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.87932647333957,
      "flow_matching_loss": 0.04017330892384052,
      "grad_norm": 0.17500418424606323,
      "learning_rate": 7.194002114337526e-05,
      "loss": 0.0408,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005790068302303552,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.55087127685547,
      "state_query_norm": 1358.4793579101563,
      "state_relation_loss": 0.0005790068302303552,
      "step": 11630,
      "student_distance_mean": 1849.8042724609375,
      "teacher_distance_mean": 1.160678219795227,
      "weighted_auxiliary_loss": 0.0005790068302303552,
      "weighted_relation_loss": 0.0005790068302303552
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.888681010289991,
      "flow_matching_loss": 0.03876215852797031,
      "grad_norm": 0.14666448533535004,
      "learning_rate": 7.189048174288054e-05,
      "loss": 0.0393,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005518063902854919,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.93777618408203,
      "state_query_norm": 1346.602783203125,
      "state_relation_loss": 0.0005518063902854919,
      "step": 11640,
      "student_distance_mean": 1820.6842895507812,
      "teacher_distance_mean": 1.1472142577171325,
      "weighted_auxiliary_loss": 0.0005518063902854919,
      "weighted_relation_loss": 0.0005518063902854919
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.898035547240411,
      "flow_matching_loss": 0.03846673835068941,
      "grad_norm": 0.10423547774553299,
      "learning_rate": 7.184091574338636e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006043320521712303,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.977538681030275,
      "state_query_norm": 1347.1610717773438,
      "state_relation_loss": 0.0006043320521712303,
      "step": 11650,
      "student_distance_mean": 1822.9620849609375,
      "teacher_distance_mean": 1.1519025921821595,
      "weighted_auxiliary_loss": 0.0006043320521712303,
      "weighted_relation_loss": 0.0006043320521712303
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.907390084190833,
      "flow_matching_loss": 0.036620819196105006,
      "grad_norm": 0.12959687411785126,
      "learning_rate": 7.179132320512009e-05,
      "loss": 0.0372,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005925834935624153,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.285416793823245,
      "state_query_norm": 1348.9777954101562,
      "state_relation_loss": 0.0005925834935624153,
      "step": 11660,
      "student_distance_mean": 1834.770947265625,
      "teacher_distance_mean": 1.1591361165046692,
      "weighted_auxiliary_loss": 0.0005925834935624153,
      "weighted_relation_loss": 0.0005925834935624153
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.916744621141254,
      "flow_matching_loss": 0.03716620579361916,
      "grad_norm": 0.11981341242790222,
      "learning_rate": 7.174170418834134e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000579246663255617,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.73158988952637,
      "state_query_norm": 1330.0796875,
      "state_relation_loss": 0.000579246663255617,
      "step": 11670,
      "student_distance_mean": 1808.4840209960937,
      "teacher_distance_mean": 1.1485079526901245,
      "weighted_auxiliary_loss": 0.000579246663255617,
      "weighted_relation_loss": 0.000579246663255617
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.926099158091674,
      "flow_matching_loss": 0.03851932175457477,
      "grad_norm": 0.11818456649780273,
      "learning_rate": 7.169205875334189e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005828278313856572,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.360253715515135,
      "state_query_norm": 1318.5932250976562,
      "state_relation_loss": 0.0005828278313856572,
      "step": 11680,
      "student_distance_mean": 1787.629345703125,
      "teacher_distance_mean": 1.1330806732177734,
      "weighted_auxiliary_loss": 0.0005828278313856572,
      "weighted_relation_loss": 0.0005828278313856572
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.935453695042096,
      "flow_matching_loss": 0.03553767167031765,
      "grad_norm": 0.12840498983860016,
      "learning_rate": 7.164238696044562e-05,
      "loss": 0.0361,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006046232825610787,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.96950092315674,
      "state_query_norm": 1345.5206787109375,
      "state_relation_loss": 0.0006046232825610787,
      "step": 11690,
      "student_distance_mean": 1821.9188110351563,
      "teacher_distance_mean": 1.155907428264618,
      "weighted_auxiliary_loss": 0.0006046232825610787,
      "weighted_relation_loss": 0.0006046232825610787
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.944808231992516,
      "flow_matching_loss": 0.03766228184103966,
      "grad_norm": 0.10645049065351486,
      "learning_rate": 7.159268887000846e-05,
      "loss": 0.0382,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005705074581783265,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.81083011627197,
      "state_query_norm": 1331.0469970703125,
      "state_relation_loss": 0.0005705074581783265,
      "step": 11700,
      "student_distance_mean": 1810.8115356445312,
      "teacher_distance_mean": 1.1493184089660644,
      "weighted_auxiliary_loss": 0.0005705074581783265,
      "weighted_relation_loss": 0.0005705074581783265
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.954162768942938,
      "flow_matching_loss": 0.038359807804226875,
      "grad_norm": 0.10656673461198807,
      "learning_rate": 7.154296454241827e-05,
      "loss": 0.039,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000595480651827529,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.82107849121094,
      "state_query_norm": 1341.3917602539063,
      "state_relation_loss": 0.000595480651827529,
      "step": 11710,
      "student_distance_mean": 1815.7088134765625,
      "teacher_distance_mean": 1.151562798023224,
      "weighted_auxiliary_loss": 0.000595480651827529,
      "weighted_relation_loss": 0.000595480651827529
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.963517305893358,
      "flow_matching_loss": 0.036675804667174815,
      "grad_norm": 0.12928803265094757,
      "learning_rate": 7.149321403809479e-05,
      "loss": 0.0373,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006208777311258018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.457457542419434,
      "state_query_norm": 1319.1034545898438,
      "state_relation_loss": 0.0006208777311258018,
      "step": 11720,
      "student_distance_mean": 1794.0840698242187,
      "teacher_distance_mean": 1.1443084359169007,
      "weighted_auxiliary_loss": 0.0006208777311258018,
      "weighted_relation_loss": 0.0006208777311258018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.97287184284378,
      "flow_matching_loss": 0.03668061047792435,
      "grad_norm": 0.1095198467373848,
      "learning_rate": 7.144343741748956e-05,
      "loss": 0.0373,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005766257003415376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.73807220458984,
      "state_query_norm": 1333.0663208007813,
      "state_relation_loss": 0.0005766257003415376,
      "step": 11730,
      "student_distance_mean": 1807.9685424804688,
      "teacher_distance_mean": 1.1475451350212098,
      "weighted_auxiliary_loss": 0.0005766257003415376,
      "weighted_relation_loss": 0.0005766257003415376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.9822263797942,
      "flow_matching_loss": 0.03828321695327759,
      "grad_norm": 0.09203948080539703,
      "learning_rate": 7.139363474108589e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005878641270101071,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.101569747924806,
      "state_query_norm": 1343.3185791015626,
      "state_relation_loss": 0.0005878641270101071,
      "step": 11740,
      "student_distance_mean": 1827.0706909179687,
      "teacher_distance_mean": 1.1555571556091309,
      "weighted_auxiliary_loss": 0.0005878641270101071,
      "weighted_relation_loss": 0.0005878641270101071
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 10.99158091674462,
      "flow_matching_loss": 0.03814950957894325,
      "grad_norm": 0.14825031161308289,
      "learning_rate": 7.134380606939874e-05,
      "loss": 0.0387,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005998710868880153,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.42201671600342,
      "state_query_norm": 1351.005224609375,
      "state_relation_loss": 0.0005998710868880153,
      "step": 11750,
      "student_distance_mean": 1839.676123046875,
      "teacher_distance_mean": 1.1665287017822266,
      "weighted_auxiliary_loss": 0.0005998710868880153,
      "weighted_relation_loss": 0.0005998710868880153
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.000935453695043,
      "flow_matching_loss": 0.038605314865708354,
      "grad_norm": 0.09483806043863297,
      "learning_rate": 7.12939514629746e-05,
      "loss": 0.0392,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0005795828474219888,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 32.999786949157716,
      "state_query_norm": 1351.22734375,
      "state_relation_loss": 0.0005795828474219888,
      "step": 11760,
      "student_distance_mean": 1827.0048828125,
      "teacher_distance_mean": 1.1562906742095946,
      "weighted_auxiliary_loss": 0.0005795828474219888,
      "weighted_relation_loss": 0.0005795828474219888
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.010289990645463,
      "flow_matching_loss": 0.035503331944346427,
      "grad_norm": 0.11861075460910797,
      "learning_rate": 7.124407098239155e-05,
      "loss": 0.0361,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005771478638052941,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.94417514801025,
      "state_query_norm": 1341.39287109375,
      "state_relation_loss": 0.0005771478638052941,
      "step": 11770,
      "student_distance_mean": 1819.0863403320313,
      "teacher_distance_mean": 1.1506293535232544,
      "weighted_auxiliary_loss": 0.0005771478638052941,
      "weighted_relation_loss": 0.0005771478638052941
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.019644527595885,
      "flow_matching_loss": 0.03718207962810993,
      "grad_norm": 0.1072457879781723,
      "learning_rate": 7.119416468825908e-05,
      "loss": 0.0378,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005761202308349311,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.83741092681885,
      "state_query_norm": 1334.8274169921874,
      "state_relation_loss": 0.0005761202308349311,
      "step": 11780,
      "student_distance_mean": 1812.2007080078124,
      "teacher_distance_mean": 1.1502953052520752,
      "weighted_auxiliary_loss": 0.0005761202308349311,
      "weighted_relation_loss": 0.0005761202308349311
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.028999064546305,
      "flow_matching_loss": 0.0377269234508276,
      "grad_norm": 0.10629136860370636,
      "learning_rate": 7.114423264121804e-05,
      "loss": 0.0383,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005730112257879228,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.32753238677979,
      "state_query_norm": 1324.1750854492188,
      "state_relation_loss": 0.0005730112257879228,
      "step": 11790,
      "student_distance_mean": 1787.9682006835938,
      "teacher_distance_mean": 1.1370904684066772,
      "weighted_auxiliary_loss": 0.0005730112257879228,
      "weighted_relation_loss": 0.0005730112257879228
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.038353601496725,
      "flow_matching_loss": 0.036923177912831305,
      "grad_norm": 0.11749958246946335,
      "learning_rate": 7.109427490194056e-05,
      "loss": 0.0375,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005907628452405334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.95109939575195,
      "state_query_norm": 1342.257421875,
      "state_relation_loss": 0.0005907628452405334,
      "step": 11800,
      "student_distance_mean": 1821.1018432617188,
      "teacher_distance_mean": 1.1574116468429565,
      "weighted_auxiliary_loss": 0.0005907628452405334,
      "weighted_relation_loss": 0.0005907628452405334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.047708138447147,
      "flow_matching_loss": 0.037547959759831426,
      "grad_norm": 0.1578947901725769,
      "learning_rate": 7.104429153113001e-05,
      "loss": 0.0381,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005962393945083023,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.81074485778809,
      "state_query_norm": 1344.2867797851563,
      "state_relation_loss": 0.0005962393945083023,
      "step": 11810,
      "student_distance_mean": 1811.6050170898438,
      "teacher_distance_mean": 1.1468742728233337,
      "weighted_auxiliary_loss": 0.0005962393945083023,
      "weighted_relation_loss": 0.0005962393945083023
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.057062675397567,
      "flow_matching_loss": 0.037872214615345,
      "grad_norm": 0.12344150990247726,
      "learning_rate": 7.099428258952092e-05,
      "loss": 0.0385,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006180890661198646,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.0636022567749,
      "state_query_norm": 1334.7038818359374,
      "state_relation_loss": 0.0006180890661198646,
      "step": 11820,
      "student_distance_mean": 1821.0881713867188,
      "teacher_distance_mean": 1.1481526136398315,
      "weighted_auxiliary_loss": 0.0006180890661198646,
      "weighted_relation_loss": 0.0006180890661198646
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.06641721234799,
      "flow_matching_loss": 0.037131091207265855,
      "grad_norm": 0.12008704245090485,
      "learning_rate": 7.094424813787883e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006160553370136767,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.56064300537109,
      "state_query_norm": 1350.4694091796875,
      "state_relation_loss": 0.0006160553370136767,
      "step": 11830,
      "student_distance_mean": 1845.9666015625,
      "teacher_distance_mean": 1.1651864171028137,
      "weighted_auxiliary_loss": 0.0006160553370136767,
      "weighted_relation_loss": 0.0006160553370136767
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.07577174929841,
      "flow_matching_loss": 0.037004686519503595,
      "grad_norm": 0.11400017142295837,
      "learning_rate": 7.089418823700034e-05,
      "loss": 0.0376,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006033843441400677,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.57881546020508,
      "state_query_norm": 1360.8776489257812,
      "state_relation_loss": 0.0006033843441400677,
      "step": 11840,
      "student_distance_mean": 1853.1382446289062,
      "teacher_distance_mean": 1.164235532283783,
      "weighted_auxiliary_loss": 0.0006033843441400677,
      "weighted_relation_loss": 0.0006033843441400677
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.085126286248832,
      "flow_matching_loss": 0.036880680918693544,
      "grad_norm": 0.09844685345888138,
      "learning_rate": 7.084410294771298e-05,
      "loss": 0.0375,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005693870247341692,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.526890182495116,
      "state_query_norm": 1362.7994384765625,
      "state_relation_loss": 0.0005693870247341692,
      "step": 11850,
      "student_distance_mean": 1849.4360107421876,
      "teacher_distance_mean": 1.1608956933021546,
      "weighted_auxiliary_loss": 0.0005693870247341692,
      "weighted_relation_loss": 0.0005693870247341692
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.094480823199252,
      "flow_matching_loss": 0.03599222656339407,
      "grad_norm": 0.1248091533780098,
      "learning_rate": 7.079399233087504e-05,
      "loss": 0.0365,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005426596704637632,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.23334102630615,
      "state_query_norm": 1359.21591796875,
      "state_relation_loss": 0.0005426596704637632,
      "step": 11860,
      "student_distance_mean": 1838.4322875976563,
      "teacher_distance_mean": 1.154888641834259,
      "weighted_auxiliary_loss": 0.0005426596704637632,
      "weighted_relation_loss": 0.0005426596704637632
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.103835360149672,
      "flow_matching_loss": 0.036464433558285235,
      "grad_norm": 0.09476277977228165,
      "learning_rate": 7.074385644737568e-05,
      "loss": 0.0371,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005975925945676863,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.97422866821289,
      "state_query_norm": 1343.4627319335937,
      "state_relation_loss": 0.0005975925945676863,
      "step": 11870,
      "student_distance_mean": 1823.174072265625,
      "teacher_distance_mean": 1.1545889139175416,
      "weighted_auxiliary_loss": 0.0005975925945676863,
      "weighted_relation_loss": 0.0005975925945676863
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.113189897100094,
      "flow_matching_loss": 0.03563220612704754,
      "grad_norm": 0.13511686027050018,
      "learning_rate": 7.069369535813473e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006127678614575415,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.91283111572265,
      "state_query_norm": 1337.270556640625,
      "state_relation_loss": 0.0006127678614575415,
      "step": 11880,
      "student_distance_mean": 1812.6080932617188,
      "teacher_distance_mean": 1.146063256263733,
      "weighted_auxiliary_loss": 0.0006127678614575415,
      "weighted_relation_loss": 0.0006127678614575415
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.122544434050514,
      "flow_matching_loss": 0.03626235481351614,
      "grad_norm": 0.11200501769781113,
      "learning_rate": 7.06435091241026e-05,
      "loss": 0.0368,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005763162043876946,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.95402488708496,
      "state_query_norm": 1342.7974365234375,
      "state_relation_loss": 0.0005763162043876946,
      "step": 11890,
      "student_distance_mean": 1821.1673217773437,
      "teacher_distance_mean": 1.147588300704956,
      "weighted_auxiliary_loss": 0.0005763162043876946,
      "weighted_relation_loss": 0.0005763162043876946
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.131898971000936,
      "flow_matching_loss": 0.03763900753110647,
      "grad_norm": 0.13534651696681976,
      "learning_rate": 7.059329780626034e-05,
      "loss": 0.0382,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006045410467777401,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.913478469848634,
      "state_query_norm": 1332.4522827148437,
      "state_relation_loss": 0.0006045410467777401,
      "step": 11900,
      "student_distance_mean": 1812.059912109375,
      "teacher_distance_mean": 1.145885968208313,
      "weighted_auxiliary_loss": 0.0006045410467777401,
      "weighted_relation_loss": 0.0006045410467777401
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.141253507951356,
      "flow_matching_loss": 0.037304094433784483,
      "grad_norm": 0.11889161169528961,
      "learning_rate": 7.054306146561944e-05,
      "loss": 0.0379,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005902739358134568,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.91165428161621,
      "state_query_norm": 1336.933984375,
      "state_relation_loss": 0.0005902739358134568,
      "step": 11910,
      "student_distance_mean": 1818.7865600585938,
      "teacher_distance_mean": 1.1485018372535705,
      "weighted_auxiliary_loss": 0.0005902739358134568,
      "weighted_relation_loss": 0.0005902739358134568
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.150608044901777,
      "flow_matching_loss": 0.038536280393600464,
      "grad_norm": 0.12603028118610382,
      "learning_rate": 7.049280016322178e-05,
      "loss": 0.0391,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005796754034236073,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.15050010681152,
      "state_query_norm": 1355.3123779296875,
      "state_relation_loss": 0.0005796754034236073,
      "step": 11920,
      "student_distance_mean": 1833.8585083007813,
      "teacher_distance_mean": 1.1588184714317322,
      "weighted_auxiliary_loss": 0.0005796754034236073,
      "weighted_relation_loss": 0.0005796754034236073
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.159962581852199,
      "flow_matching_loss": 0.035328819416463375,
      "grad_norm": 0.09896185994148254,
      "learning_rate": 7.044251396013957e-05,
      "loss": 0.0359,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005822770355734975,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.87343406677246,
      "state_query_norm": 1333.964013671875,
      "state_relation_loss": 0.0005822770355734975,
      "step": 11930,
      "student_distance_mean": 1813.2215698242187,
      "teacher_distance_mean": 1.1426878452301026,
      "weighted_auxiliary_loss": 0.0005822770355734975,
      "weighted_relation_loss": 0.0005822770355734975
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.169317118802619,
      "flow_matching_loss": 0.0364013534039259,
      "grad_norm": 0.10451795160770416,
      "learning_rate": 7.039220291747528e-05,
      "loss": 0.037,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006055345584172755,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.330227279663085,
      "state_query_norm": 1347.0424072265625,
      "state_relation_loss": 0.0006055345584172755,
      "step": 11940,
      "student_distance_mean": 1833.942041015625,
      "teacher_distance_mean": 1.1546423435211182,
      "weighted_auxiliary_loss": 0.0006055345584172755,
      "weighted_relation_loss": 0.0006055345584172755
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.17867165575304,
      "flow_matching_loss": 0.036707299202680586,
      "grad_norm": 0.10400336980819702,
      "learning_rate": 7.034186709636159e-05,
      "loss": 0.0373,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005978218337986618,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.94253940582276,
      "state_query_norm": 1340.7598876953125,
      "state_relation_loss": 0.0005978218337986618,
      "step": 11950,
      "student_distance_mean": 1816.52509765625,
      "teacher_distance_mean": 1.1429778575897216,
      "weighted_auxiliary_loss": 0.0005978218337986618,
      "weighted_relation_loss": 0.0005978218337986618
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.188026192703461,
      "flow_matching_loss": 0.03748004995286465,
      "grad_norm": 0.11414302885532379,
      "learning_rate": 7.029150655796129e-05,
      "loss": 0.038,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005696943553630263,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.576092910766604,
      "state_query_norm": 1354.4906372070313,
      "state_relation_loss": 0.0005696943553630263,
      "step": 11960,
      "student_distance_mean": 1848.9844360351562,
      "teacher_distance_mean": 1.1631649494171143,
      "weighted_auxiliary_loss": 0.0005696943553630263,
      "weighted_relation_loss": 0.0005696943553630263
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.197380729653883,
      "flow_matching_loss": 0.035868299379944804,
      "grad_norm": 0.12895242869853973,
      "learning_rate": 7.024112136346712e-05,
      "loss": 0.0364,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005728266260121018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.59539604187012,
      "state_query_norm": 1365.92021484375,
      "state_relation_loss": 0.0005728266260121018,
      "step": 11970,
      "student_distance_mean": 1855.991943359375,
      "teacher_distance_mean": 1.1637478828430177,
      "weighted_auxiliary_loss": 0.0005728266260121018,
      "weighted_relation_loss": 0.0005728266260121018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.206735266604303,
      "flow_matching_loss": 0.037167702428996564,
      "grad_norm": 0.16201408207416534,
      "learning_rate": 7.01907115741019e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005784562905319035,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.23604755401611,
      "state_query_norm": 1347.3102661132812,
      "state_relation_loss": 0.0005784562905319035,
      "step": 11980,
      "student_distance_mean": 1833.9552612304688,
      "teacher_distance_mean": 1.1515989899635315,
      "weighted_auxiliary_loss": 0.0005784562905319035,
      "weighted_relation_loss": 0.0005784562905319035
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.216089803554723,
      "flow_matching_loss": 0.038722839951515195,
      "grad_norm": 0.12312544882297516,
      "learning_rate": 7.014027725111825e-05,
      "loss": 0.0393,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005881704855710268,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.524942779541014,
      "state_query_norm": 1367.8122924804688,
      "state_relation_loss": 0.0005881704855710268,
      "step": 11990,
      "student_distance_mean": 1849.8872314453124,
      "teacher_distance_mean": 1.1605841755867004,
      "weighted_auxiliary_loss": 0.0005881704855710268,
      "weighted_relation_loss": 0.0005881704855710268
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.225444340505145,
      "flow_matching_loss": 0.03777748979628086,
      "grad_norm": 0.1313491314649582,
      "learning_rate": 7.008981845579864e-05,
      "loss": 0.0383,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005583077552728354,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.73524131774902,
      "state_query_norm": 1330.896826171875,
      "state_relation_loss": 0.0005583077552728354,
      "step": 12000,
      "student_distance_mean": 1808.6076293945312,
      "teacher_distance_mean": 1.1399704575538636,
      "weighted_auxiliary_loss": 0.0005583077552728354,
      "weighted_relation_loss": 0.0005583077552728354
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.234798877455566,
      "flow_matching_loss": 0.03566902317106724,
      "grad_norm": 0.08740539103746414,
      "learning_rate": 7.003933524945528e-05,
      "loss": 0.0363,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006052568147424609,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.4835205078125,
      "state_query_norm": 1357.6245971679687,
      "state_relation_loss": 0.0006052568147424609,
      "step": 12010,
      "student_distance_mean": 1843.85244140625,
      "teacher_distance_mean": 1.1563361525535583,
      "weighted_auxiliary_loss": 0.0006052568147424609,
      "weighted_relation_loss": 0.0006052568147424609
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.244153414405988,
      "flow_matching_loss": 0.03555892165750265,
      "grad_norm": 0.1289799064397812,
      "learning_rate": 6.998882769342998e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006105397420469672,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.84239044189453,
      "state_query_norm": 1362.2324584960938,
      "state_relation_loss": 0.0006105397420469672,
      "step": 12020,
      "student_distance_mean": 1862.7473876953125,
      "teacher_distance_mean": 1.1647058963775634,
      "weighted_auxiliary_loss": 0.0006105397420469672,
      "weighted_relation_loss": 0.0006105397420469672
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.253507951356408,
      "flow_matching_loss": 0.0350678900256753,
      "grad_norm": 0.11732620000839233,
      "learning_rate": 6.993829584909423e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005873170419363305,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.14358386993408,
      "state_query_norm": 1349.13662109375,
      "state_relation_loss": 0.0005873170419363305,
      "step": 12030,
      "student_distance_mean": 1830.0033935546876,
      "teacher_distance_mean": 1.1496987342834473,
      "weighted_auxiliary_loss": 0.0005873170419363305,
      "weighted_relation_loss": 0.0005873170419363305
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.262862488306828,
      "flow_matching_loss": 0.03629904575645924,
      "grad_norm": 0.08630228042602539,
      "learning_rate": 6.988773977784895e-05,
      "loss": 0.0369,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005794559605419636,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.455773735046385,
      "state_query_norm": 1353.54111328125,
      "state_relation_loss": 0.0005794559605419636,
      "step": 12040,
      "student_distance_mean": 1843.4083251953125,
      "teacher_distance_mean": 1.151231062412262,
      "weighted_auxiliary_loss": 0.0005794559605419636,
      "weighted_relation_loss": 0.0005794559605419636
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.27221702525725,
      "flow_matching_loss": 0.03797186687588692,
      "grad_norm": 0.17126360535621643,
      "learning_rate": 6.983715954112454e-05,
      "loss": 0.0385,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005597371375188232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.58224449157715,
      "state_query_norm": 1365.8233154296875,
      "state_relation_loss": 0.0005597371375188232,
      "step": 12050,
      "student_distance_mean": 1855.4976928710937,
      "teacher_distance_mean": 1.1584836602211,
      "weighted_auxiliary_loss": 0.0005597371375188232,
      "weighted_relation_loss": 0.0005597371375188232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.28157156220767,
      "flow_matching_loss": 0.03561049569398165,
      "grad_norm": 0.1352202594280243,
      "learning_rate": 6.978655520038079e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005806586646940559,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.83681468963623,
      "state_query_norm": 1337.2774658203125,
      "state_relation_loss": 0.0005806586646940559,
      "step": 12060,
      "student_distance_mean": 1811.9132568359375,
      "teacher_distance_mean": 1.1343070268630981,
      "weighted_auxiliary_loss": 0.0005806586646940559,
      "weighted_relation_loss": 0.0005806586646940559
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.290926099158092,
      "flow_matching_loss": 0.03576659969985485,
      "grad_norm": 0.12053266167640686,
      "learning_rate": 6.97359268171067e-05,
      "loss": 0.0364,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006137230549938977,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.12023773193359,
      "state_query_norm": 1346.1644287109375,
      "state_relation_loss": 0.0006137230549938977,
      "step": 12070,
      "student_distance_mean": 1826.2133666992188,
      "teacher_distance_mean": 1.141604483127594,
      "weighted_auxiliary_loss": 0.0006137230549938977,
      "weighted_relation_loss": 0.0006137230549938977
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.300280636108512,
      "flow_matching_loss": 0.03833158276975155,
      "grad_norm": 0.11195621639490128,
      "learning_rate": 6.968527445282056e-05,
      "loss": 0.0389,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005880456534214318,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.6516773223877,
      "state_query_norm": 1359.253125,
      "state_relation_loss": 0.0005880456534214318,
      "step": 12080,
      "student_distance_mean": 1854.8202758789062,
      "teacher_distance_mean": 1.158104169368744,
      "weighted_auxiliary_loss": 0.0005880456534214318,
      "weighted_relation_loss": 0.0005880456534214318
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.309635173058934,
      "flow_matching_loss": 0.037288551963865754,
      "grad_norm": 0.11087874323129654,
      "learning_rate": 6.963459816906977e-05,
      "loss": 0.0379,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005716359883081168,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.74763793945313,
      "state_query_norm": 1376.3099731445313,
      "state_relation_loss": 0.0005716359883081168,
      "step": 12090,
      "student_distance_mean": 1864.6233276367188,
      "teacher_distance_mean": 1.166266965866089,
      "weighted_auxiliary_loss": 0.0005716359883081168,
      "weighted_relation_loss": 0.0005716359883081168
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.318989710009355,
      "flow_matching_loss": 0.03601995445787907,
      "grad_norm": 0.13571599125862122,
      "learning_rate": 6.958389802743077e-05,
      "loss": 0.0366,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005893692083191127,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.50593147277832,
      "state_query_norm": 1367.2441162109376,
      "state_relation_loss": 0.0005893692083191127,
      "step": 12100,
      "student_distance_mean": 1849.097265625,
      "teacher_distance_mean": 1.1532559394836426,
      "weighted_auxiliary_loss": 0.0005893692083191127,
      "weighted_relation_loss": 0.0005893692083191127
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.328344246959775,
      "flow_matching_loss": 0.036398640088737014,
      "grad_norm": 0.11415980011224747,
      "learning_rate": 6.953317408950903e-05,
      "loss": 0.037,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006038161634933203,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.618567657470706,
      "state_query_norm": 1360.4841186523438,
      "state_relation_loss": 0.0006038161634933203,
      "step": 12110,
      "student_distance_mean": 1856.1393432617188,
      "teacher_distance_mean": 1.155755305290222,
      "weighted_auxiliary_loss": 0.0006038161634933203,
      "weighted_relation_loss": 0.0006038161634933203
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.337698783910197,
      "flow_matching_loss": 0.03577216044068336,
      "grad_norm": 0.09582963585853577,
      "learning_rate": 6.94824264169389e-05,
      "loss": 0.0363,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000562852865550667,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.54526557922363,
      "state_query_norm": 1365.5800415039062,
      "state_relation_loss": 0.000562852865550667,
      "step": 12120,
      "student_distance_mean": 1851.9490234375,
      "teacher_distance_mean": 1.1544859170913697,
      "weighted_auxiliary_loss": 0.000562852865550667,
      "weighted_relation_loss": 0.000562852865550667
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.347053320860617,
      "flow_matching_loss": 0.03640547096729278,
      "grad_norm": 0.0995601937174797,
      "learning_rate": 6.94316550713836e-05,
      "loss": 0.037,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005713323713280261,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.29370536804199,
      "state_query_norm": 1352.783154296875,
      "state_relation_loss": 0.0005713323713280261,
      "step": 12130,
      "student_distance_mean": 1838.8593627929688,
      "teacher_distance_mean": 1.1499710917472838,
      "weighted_auxiliary_loss": 0.0005713323713280261,
      "weighted_relation_loss": 0.0005713323713280261
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.356407857811039,
      "flow_matching_loss": 0.038247577100992206,
      "grad_norm": 0.1563831865787506,
      "learning_rate": 6.938086011453513e-05,
      "loss": 0.0388,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005621685588266701,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.51338882446289,
      "state_query_norm": 1337.902880859375,
      "state_relation_loss": 0.0005621685588266701,
      "step": 12140,
      "student_distance_mean": 1799.57705078125,
      "teacher_distance_mean": 1.1323517441749573,
      "weighted_auxiliary_loss": 0.0005621685588266701,
      "weighted_relation_loss": 0.0005621685588266701
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.36576239476146,
      "flow_matching_loss": 0.03690120130777359,
      "grad_norm": 0.10222438722848892,
      "learning_rate": 6.93300416081141e-05,
      "loss": 0.0375,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005989527504425496,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.76190452575683,
      "state_query_norm": 1359.5233276367187,
      "state_relation_loss": 0.0005989527504425496,
      "step": 12150,
      "student_distance_mean": 1859.1090209960937,
      "teacher_distance_mean": 1.1628489017486572,
      "weighted_auxiliary_loss": 0.0005989527504425496,
      "weighted_relation_loss": 0.0005989527504425496
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.37511693171188,
      "flow_matching_loss": 0.03734973706305027,
      "grad_norm": 0.1101393774151802,
      "learning_rate": 6.927919961386984e-05,
      "loss": 0.0379,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005842196114826947,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.36784343719482,
      "state_query_norm": 1350.8869873046874,
      "state_relation_loss": 0.0005842196114826947,
      "step": 12160,
      "student_distance_mean": 1839.0812622070312,
      "teacher_distance_mean": 1.1490462660789489,
      "weighted_auxiliary_loss": 0.0005842196114826947,
      "weighted_relation_loss": 0.0005842196114826947
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.384471468662301,
      "flow_matching_loss": 0.03809322863817215,
      "grad_norm": 0.1259794682264328,
      "learning_rate": 6.922833419358013e-05,
      "loss": 0.0387,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005823196901474148,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.565406799316406,
      "state_query_norm": 1350.79453125,
      "state_relation_loss": 0.0005823196901474148,
      "step": 12170,
      "student_distance_mean": 1844.4821533203126,
      "teacher_distance_mean": 1.1508134245872497,
      "weighted_auxiliary_loss": 0.0005823196901474148,
      "weighted_relation_loss": 0.0005823196901474148
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.393826005612722,
      "flow_matching_loss": 0.03492360580712557,
      "grad_norm": 0.16744592785835266,
      "learning_rate": 6.917744540905124e-05,
      "loss": 0.0355,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005467816255986691,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.49696502685547,
      "state_query_norm": 1365.1553833007813,
      "state_relation_loss": 0.0005467816255986691,
      "step": 12180,
      "student_distance_mean": 1849.4323974609374,
      "teacher_distance_mean": 1.149213469028473,
      "weighted_auxiliary_loss": 0.0005467816255986691,
      "weighted_relation_loss": 0.0005467816255986691
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.403180542563144,
      "flow_matching_loss": 0.03642249088734388,
      "grad_norm": 0.16706880927085876,
      "learning_rate": 6.912653332211787e-05,
      "loss": 0.037,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005720151064451783,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.41479568481445,
      "state_query_norm": 1357.0515869140625,
      "state_relation_loss": 0.0005720151064451783,
      "step": 12190,
      "student_distance_mean": 1847.5150146484375,
      "teacher_distance_mean": 1.1477446556091309,
      "weighted_auxiliary_loss": 0.0005720151064451783,
      "weighted_relation_loss": 0.0005720151064451783
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.412535079513564,
      "flow_matching_loss": 0.03695141077041626,
      "grad_norm": 0.1198100671172142,
      "learning_rate": 6.9075597994643e-05,
      "loss": 0.0375,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005741430621128529,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.2039077758789,
      "state_query_norm": 1340.449560546875,
      "state_relation_loss": 0.0005741430621128529,
      "step": 12200,
      "student_distance_mean": 1831.4192016601562,
      "teacher_distance_mean": 1.1436524629592895,
      "weighted_auxiliary_loss": 0.0005741430621128529,
      "weighted_relation_loss": 0.0005741430621128529
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.421889616463986,
      "flow_matching_loss": 0.03924427479505539,
      "grad_norm": 0.1252996176481247,
      "learning_rate": 6.902463948851787e-05,
      "loss": 0.0398,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005756645346991718,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.68631172180176,
      "state_query_norm": 1329.9357421875,
      "state_relation_loss": 0.0005756645346991718,
      "step": 12210,
      "student_distance_mean": 1805.6413940429688,
      "teacher_distance_mean": 1.1307814717292786,
      "weighted_auxiliary_loss": 0.0005756645346991718,
      "weighted_relation_loss": 0.0005756645346991718
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.431244153414406,
      "flow_matching_loss": 0.03501833230257034,
      "grad_norm": 0.10791108012199402,
      "learning_rate": 6.897365786566184e-05,
      "loss": 0.0356,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005784040607977658,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.77261047363281,
      "state_query_norm": 1365.8448486328125,
      "state_relation_loss": 0.0005784040607977658,
      "step": 12220,
      "student_distance_mean": 1859.263525390625,
      "teacher_distance_mean": 1.1601322889328003,
      "weighted_auxiliary_loss": 0.0005784040607977658,
      "weighted_relation_loss": 0.0005784040607977658
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.440598690364826,
      "flow_matching_loss": 0.03692080806940794,
      "grad_norm": 0.1338200867176056,
      "learning_rate": 6.892265318802242e-05,
      "loss": 0.0375,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005816537654027343,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.25838623046875,
      "state_query_norm": 1353.7561645507812,
      "state_relation_loss": 0.0005816537654027343,
      "step": 12230,
      "student_distance_mean": 1835.8669311523438,
      "teacher_distance_mean": 1.1462692260742187,
      "weighted_auxiliary_loss": 0.0005816537654027343,
      "weighted_relation_loss": 0.0005816537654027343
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.449953227315248,
      "flow_matching_loss": 0.03518035877496004,
      "grad_norm": 0.10640835016965866,
      "learning_rate": 6.887162551757507e-05,
      "loss": 0.0358,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006116884003859013,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.87471160888672,
      "state_query_norm": 1372.3354248046876,
      "state_relation_loss": 0.0006116884003859013,
      "step": 12240,
      "student_distance_mean": 1865.1002685546875,
      "teacher_distance_mean": 1.1618174314498901,
      "weighted_auxiliary_loss": 0.0006116884003859013,
      "weighted_relation_loss": 0.0006116884003859013
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.459307764265668,
      "flow_matching_loss": 0.03507534470409155,
      "grad_norm": 0.13128221035003662,
      "learning_rate": 6.882057491632326e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005794930388219654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.00560398101807,
      "state_query_norm": 1341.0808837890625,
      "state_relation_loss": 0.0005794930388219654,
      "step": 12250,
      "student_distance_mean": 1824.6908935546876,
      "teacher_distance_mean": 1.1290571093559265,
      "weighted_auxiliary_loss": 0.0005794930388219654,
      "weighted_relation_loss": 0.0005794930388219654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.46866230121609,
      "flow_matching_loss": 0.03686704635620117,
      "grad_norm": 0.16690868139266968,
      "learning_rate": 6.876950144629824e-05,
      "loss": 0.0374,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000582456385018304,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.7514087677002,
      "state_query_norm": 1365.4378295898437,
      "state_relation_loss": 0.000582456385018304,
      "step": 12260,
      "student_distance_mean": 1857.6552490234376,
      "teacher_distance_mean": 1.160180139541626,
      "weighted_auxiliary_loss": 0.000582456385018304,
      "weighted_relation_loss": 0.000582456385018304
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.47801683816651,
      "flow_matching_loss": 0.036800182238221166,
      "grad_norm": 0.13252723217010498,
      "learning_rate": 6.87184051695591e-05,
      "loss": 0.0374,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005622039287118241,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.80252456665039,
      "state_query_norm": 1374.91533203125,
      "state_relation_loss": 0.0005622039287118241,
      "step": 12270,
      "student_distance_mean": 1867.1119506835937,
      "teacher_distance_mean": 1.1608374476432801,
      "weighted_auxiliary_loss": 0.0005622039287118241,
      "weighted_relation_loss": 0.0005622039287118241
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.487371375116933,
      "flow_matching_loss": 0.03660837225615978,
      "grad_norm": 0.12651491165161133,
      "learning_rate": 6.866728614819268e-05,
      "loss": 0.0372,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000589041376952082,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.70977249145508,
      "state_query_norm": 1359.3578002929687,
      "state_relation_loss": 0.000589041376952082,
      "step": 12280,
      "student_distance_mean": 1856.0057495117187,
      "teacher_distance_mean": 1.1582109689712525,
      "weighted_auxiliary_loss": 0.000589041376952082,
      "weighted_relation_loss": 0.000589041376952082
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.496725912067353,
      "flow_matching_loss": 0.035218802839517595,
      "grad_norm": 0.11327149718999863,
      "learning_rate": 6.861614444431337e-05,
      "loss": 0.0358,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005693639686796814,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.72383117675781,
      "state_query_norm": 1365.6266479492188,
      "state_relation_loss": 0.0005693639686796814,
      "step": 12290,
      "student_distance_mean": 1857.94755859375,
      "teacher_distance_mean": 1.1636810660362245,
      "weighted_auxiliary_loss": 0.0005693639686796814,
      "weighted_relation_loss": 0.0005693639686796814
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.506080449017773,
      "flow_matching_loss": 0.03592791073024273,
      "grad_norm": 0.10680588334798813,
      "learning_rate": 6.856498012006318e-05,
      "loss": 0.0365,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005643134179990738,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.43390159606933,
      "state_query_norm": 1360.2671875,
      "state_relation_loss": 0.0005643134179990738,
      "step": 12300,
      "student_distance_mean": 1845.0918579101562,
      "teacher_distance_mean": 1.1573362350463867,
      "weighted_auxiliary_loss": 0.0005643134179990738,
      "weighted_relation_loss": 0.0005643134179990738
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.515434985968195,
      "flow_matching_loss": 0.03733082264661789,
      "grad_norm": 0.09176202118396759,
      "learning_rate": 6.851379323761157e-05,
      "loss": 0.0379,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000586013006977737,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.42162857055664,
      "state_query_norm": 1358.0430419921875,
      "state_relation_loss": 0.000586013006977737,
      "step": 12310,
      "student_distance_mean": 1842.1856811523437,
      "teacher_distance_mean": 1.1533532857894897,
      "weighted_auxiliary_loss": 0.000586013006977737,
      "weighted_relation_loss": 0.000586013006977737
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.524789522918615,
      "flow_matching_loss": 0.035129771754145625,
      "grad_norm": 0.14413800835609436,
      "learning_rate": 6.846258385915545e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005574326438363642,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.21198081970215,
      "state_query_norm": 1346.8402221679687,
      "state_relation_loss": 0.0005574326438363642,
      "step": 12320,
      "student_distance_mean": 1833.1306274414062,
      "teacher_distance_mean": 1.1461400985717773,
      "weighted_auxiliary_loss": 0.0005574326438363642,
      "weighted_relation_loss": 0.0005574326438363642
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.534144059869037,
      "flow_matching_loss": 0.03763006739318371,
      "grad_norm": 0.13748377561569214,
      "learning_rate": 6.841135204691902e-05,
      "loss": 0.0382,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005726700183004141,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.469922637939455,
      "state_query_norm": 1355.7967041015625,
      "state_relation_loss": 0.0005726700183004141,
      "step": 12330,
      "student_distance_mean": 1842.2819702148438,
      "teacher_distance_mean": 1.1480190753936768,
      "weighted_auxiliary_loss": 0.0005726700183004141,
      "weighted_relation_loss": 0.0005726700183004141
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.543498596819457,
      "flow_matching_loss": 0.03554878793656826,
      "grad_norm": 0.10692846029996872,
      "learning_rate": 6.836009786315377e-05,
      "loss": 0.0361,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005911804328206927,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.43843765258789,
      "state_query_norm": 1398.4853881835938,
      "state_relation_loss": 0.0005911804328206927,
      "step": 12340,
      "student_distance_mean": 1900.6298706054688,
      "teacher_distance_mean": 1.1903822779655457,
      "weighted_auxiliary_loss": 0.0005911804328206927,
      "weighted_relation_loss": 0.0005911804328206927
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.552853133769878,
      "flow_matching_loss": 0.036563499830663204,
      "grad_norm": 0.08729652315378189,
      "learning_rate": 6.830882137013839e-05,
      "loss": 0.0371,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005820845661219209,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.97601547241211,
      "state_query_norm": 1339.9850708007812,
      "state_relation_loss": 0.0005820845661219209,
      "step": 12350,
      "student_distance_mean": 1821.1704223632812,
      "teacher_distance_mean": 1.1402613401412964,
      "weighted_auxiliary_loss": 0.0005820845661219209,
      "weighted_relation_loss": 0.0005820845661219209
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.5622076707203,
      "flow_matching_loss": 0.03449519220739603,
      "grad_norm": 0.12104158848524094,
      "learning_rate": 6.825752263017863e-05,
      "loss": 0.0351,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006375793571351096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.581275367736815,
      "state_query_norm": 1365.10458984375,
      "state_relation_loss": 0.0006375793571351096,
      "step": 12360,
      "student_distance_mean": 1851.23857421875,
      "teacher_distance_mean": 1.1516862988471985,
      "weighted_auxiliary_loss": 0.0006375793571351096,
      "weighted_relation_loss": 0.0006375793571351096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.57156220767072,
      "flow_matching_loss": 0.03607784025371075,
      "grad_norm": 0.1092216819524765,
      "learning_rate": 6.820620170560731e-05,
      "loss": 0.0366,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005686248361598701,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.75364456176758,
      "state_query_norm": 1368.7096069335937,
      "state_relation_loss": 0.0005686248361598701,
      "step": 12370,
      "student_distance_mean": 1862.6590209960937,
      "teacher_distance_mean": 1.1589561104774475,
      "weighted_auxiliary_loss": 0.0005686248361598701,
      "weighted_relation_loss": 0.0005686248361598701
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.580916744621142,
      "flow_matching_loss": 0.035666166432201864,
      "grad_norm": 0.17209836840629578,
      "learning_rate": 6.815485865878418e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005731490906327963,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.561476135253905,
      "state_query_norm": 1361.2606079101563,
      "state_relation_loss": 0.0005731490906327963,
      "step": 12380,
      "student_distance_mean": 1849.4031372070312,
      "teacher_distance_mean": 1.1490334630012513,
      "weighted_auxiliary_loss": 0.0005731490906327963,
      "weighted_relation_loss": 0.0005731490906327963
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.590271281571562,
      "flow_matching_loss": 0.03437215834856033,
      "grad_norm": 0.11555169522762299,
      "learning_rate": 6.810349355209588e-05,
      "loss": 0.035,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005835686693899334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.537516021728514,
      "state_query_norm": 1370.4538696289062,
      "state_relation_loss": 0.0005835686693899334,
      "step": 12390,
      "student_distance_mean": 1852.4758666992188,
      "teacher_distance_mean": 1.1472628712654114,
      "weighted_auxiliary_loss": 0.0005835686693899334,
      "weighted_relation_loss": 0.0005835686693899334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.599625818521982,
      "flow_matching_loss": 0.03693577479571104,
      "grad_norm": 0.13905596733093262,
      "learning_rate": 6.805210644795588e-05,
      "loss": 0.0375,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005526900466065853,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.52898979187012,
      "state_query_norm": 1369.46328125,
      "state_relation_loss": 0.0005526900466065853,
      "step": 12400,
      "student_distance_mean": 1853.4406860351562,
      "teacher_distance_mean": 1.1557700991630555,
      "weighted_auxiliary_loss": 0.0005526900466065853,
      "weighted_relation_loss": 0.0005526900466065853
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.608980355472404,
      "flow_matching_loss": 0.03565676733851433,
      "grad_norm": 0.12124288827180862,
      "learning_rate": 6.800069740880433e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005422830523457378,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.22309799194336,
      "state_query_norm": 1345.3828979492187,
      "state_relation_loss": 0.0005422830523457378,
      "step": 12410,
      "student_distance_mean": 1835.4330200195313,
      "teacher_distance_mean": 1.150134027004242,
      "weighted_auxiliary_loss": 0.0005422830523457378,
      "weighted_relation_loss": 0.0005422830523457378
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.618334892422824,
      "flow_matching_loss": 0.03613934814929962,
      "grad_norm": 0.1411714404821396,
      "learning_rate": 6.794926649710807e-05,
      "loss": 0.0367,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000563288334524259,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.89028053283691,
      "state_query_norm": 1370.4348876953125,
      "state_relation_loss": 0.000563288334524259,
      "step": 12420,
      "student_distance_mean": 1869.32451171875,
      "teacher_distance_mean": 1.1616907835006713,
      "weighted_auxiliary_loss": 0.000563288334524259,
      "weighted_relation_loss": 0.000563288334524259
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.627689429373246,
      "flow_matching_loss": 0.03566476907581091,
      "grad_norm": 0.1230865865945816,
      "learning_rate": 6.789781377536052e-05,
      "loss": 0.0363,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006053125485777855,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.96860809326172,
      "state_query_norm": 1379.0621826171875,
      "state_relation_loss": 0.0006053125485777855,
      "step": 12430,
      "student_distance_mean": 1872.14423828125,
      "teacher_distance_mean": 1.1595074892044068,
      "weighted_auxiliary_loss": 0.0006053125485777855,
      "weighted_relation_loss": 0.0006053125485777855
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.637043966323667,
      "flow_matching_loss": 0.03645188361406326,
      "grad_norm": 0.14621052145957947,
      "learning_rate": 6.784633930608158e-05,
      "loss": 0.037,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005710627185180784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.544619750976565,
      "state_query_norm": 1362.8012084960938,
      "state_relation_loss": 0.0005710627185180784,
      "step": 12440,
      "student_distance_mean": 1847.0786987304687,
      "teacher_distance_mean": 1.1501505613327025,
      "weighted_auxiliary_loss": 0.0005710627185180784,
      "weighted_relation_loss": 0.0005710627185180784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.646398503274089,
      "flow_matching_loss": 0.0360083257779479,
      "grad_norm": 0.11409767717123032,
      "learning_rate": 6.779484315181759e-05,
      "loss": 0.0366,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005714225408155471,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.025109100341794,
      "state_query_norm": 1379.84306640625,
      "state_relation_loss": 0.0005714225408155471,
      "step": 12450,
      "student_distance_mean": 1876.6791625976562,
      "teacher_distance_mean": 1.1657433271408082,
      "weighted_auxiliary_loss": 0.0005714225408155471,
      "weighted_relation_loss": 0.0005714225408155471
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.655753040224509,
      "flow_matching_loss": 0.03731766436249018,
      "grad_norm": 0.13230445981025696,
      "learning_rate": 6.774332537514122e-05,
      "loss": 0.0379,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005780868290457874,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.31931495666504,
      "state_query_norm": 1355.8522705078126,
      "state_relation_loss": 0.0005780868290457874,
      "step": 12460,
      "student_distance_mean": 1837.9883056640624,
      "teacher_distance_mean": 1.1402006030082703,
      "weighted_auxiliary_loss": 0.0005780868290457874,
      "weighted_relation_loss": 0.0005780868290457874
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.665107577174929,
      "flow_matching_loss": 0.03616773169487715,
      "grad_norm": 0.12483086436986923,
      "learning_rate": 6.769178603865143e-05,
      "loss": 0.0368,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005998044623993337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.53567733764648,
      "state_query_norm": 1354.8619506835937,
      "state_relation_loss": 0.0005998044623993337,
      "step": 12470,
      "student_distance_mean": 1846.5596557617187,
      "teacher_distance_mean": 1.146471667289734,
      "weighted_auxiliary_loss": 0.0005998044623993337,
      "weighted_relation_loss": 0.0005998044623993337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.674462114125351,
      "flow_matching_loss": 0.03624122515320778,
      "grad_norm": 0.08930571377277374,
      "learning_rate": 6.764022520497337e-05,
      "loss": 0.0368,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005346383026335388,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.59272155761719,
      "state_query_norm": 1366.9939086914062,
      "state_relation_loss": 0.0005346383026335388,
      "step": 12480,
      "student_distance_mean": 1856.4327026367187,
      "teacher_distance_mean": 1.1522682785987854,
      "weighted_auxiliary_loss": 0.0005346383026335388,
      "weighted_relation_loss": 0.0005346383026335388
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.683816651075771,
      "flow_matching_loss": 0.03574953898787499,
      "grad_norm": 0.09472864866256714,
      "learning_rate": 6.758864293675833e-05,
      "loss": 0.0363,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000575500976992771,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.65440826416015,
      "state_query_norm": 1362.8283569335938,
      "state_relation_loss": 0.000575500976992771,
      "step": 12490,
      "student_distance_mean": 1856.75087890625,
      "teacher_distance_mean": 1.1606373190879822,
      "weighted_auxiliary_loss": 0.000575500976992771,
      "weighted_relation_loss": 0.000575500976992771
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.693171188026193,
      "flow_matching_loss": 0.037147707864642145,
      "grad_norm": 0.09888338297605515,
      "learning_rate": 6.753703929668362e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000531443563522771,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.414320373535155,
      "state_query_norm": 1364.9112060546875,
      "state_relation_loss": 0.000531443563522771,
      "step": 12500,
      "student_distance_mean": 1846.5870361328125,
      "teacher_distance_mean": 1.1519935965538024,
      "weighted_auxiliary_loss": 0.000531443563522771,
      "weighted_relation_loss": 0.000531443563522771
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.702525724976613,
      "flow_matching_loss": 0.03405373748391867,
      "grad_norm": 0.13267703354358673,
      "learning_rate": 6.748541434745253e-05,
      "loss": 0.0346,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005932072992436588,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.33694267272949,
      "state_query_norm": 1357.82939453125,
      "state_relation_loss": 0.0005932072992436588,
      "step": 12510,
      "student_distance_mean": 1840.6262329101562,
      "teacher_distance_mean": 1.1521630764007569,
      "weighted_auxiliary_loss": 0.0005932072992436588,
      "weighted_relation_loss": 0.0005932072992436588
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.711880261927035,
      "flow_matching_loss": 0.035780958645045755,
      "grad_norm": 0.12182711809873581,
      "learning_rate": 6.743376815179424e-05,
      "loss": 0.0364,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005903974932152777,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.873600769042966,
      "state_query_norm": 1373.8980102539062,
      "state_relation_loss": 0.0005903974932152777,
      "step": 12520,
      "student_distance_mean": 1867.5789184570312,
      "teacher_distance_mean": 1.1653945803642274,
      "weighted_auxiliary_loss": 0.0005903974932152777,
      "weighted_relation_loss": 0.0005903974932152777
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.721234798877456,
      "flow_matching_loss": 0.03718687426298857,
      "grad_norm": 0.11353226006031036,
      "learning_rate": 6.738210077246376e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005482165230205283,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.597882461547854,
      "state_query_norm": 1363.93876953125,
      "state_relation_loss": 0.0005482165230205283,
      "step": 12530,
      "student_distance_mean": 1855.1415405273438,
      "teacher_distance_mean": 1.1534024357795716,
      "weighted_auxiliary_loss": 0.0005482165230205283,
      "weighted_relation_loss": 0.0005482165230205283
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.730589335827876,
      "flow_matching_loss": 0.03736855816096067,
      "grad_norm": 0.11798331141471863,
      "learning_rate": 6.733041227224181e-05,
      "loss": 0.038,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005889530526474118,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.170247650146486,
      "state_query_norm": 1338.2449829101563,
      "state_relation_loss": 0.0005889530526474118,
      "step": 12540,
      "student_distance_mean": 1826.127685546875,
      "teacher_distance_mean": 1.135881245136261,
      "weighted_auxiliary_loss": 0.0005889530526474118,
      "weighted_relation_loss": 0.0005889530526474118
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.739943872778298,
      "flow_matching_loss": 0.036096010357141495,
      "grad_norm": 0.1175999790430069,
      "learning_rate": 6.72787027139348e-05,
      "loss": 0.0367,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005720017245039344,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.25355186462402,
      "state_query_norm": 1372.9888671875,
      "state_relation_loss": 0.0005720017245039344,
      "step": 12550,
      "student_distance_mean": 1882.031787109375,
      "teacher_distance_mean": 1.170319426059723,
      "weighted_auxiliary_loss": 0.0005720017245039344,
      "weighted_relation_loss": 0.0005720017245039344
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.749298409728718,
      "flow_matching_loss": 0.038063842430710794,
      "grad_norm": 0.12166787683963776,
      "learning_rate": 6.72269721603747e-05,
      "loss": 0.0386,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005455699167214334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.73709335327148,
      "state_query_norm": 1367.2124145507812,
      "state_relation_loss": 0.0005455699167214334,
      "step": 12560,
      "student_distance_mean": 1859.9200317382813,
      "teacher_distance_mean": 1.1557153701782226,
      "weighted_auxiliary_loss": 0.0005455699167214334,
      "weighted_relation_loss": 0.0005455699167214334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.75865294667914,
      "flow_matching_loss": 0.033820306323468684,
      "grad_norm": 0.12416642159223557,
      "learning_rate": 6.717522067441904e-05,
      "loss": 0.0344,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005767833616118878,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.51130294799805,
      "state_query_norm": 1358.0139526367188,
      "state_relation_loss": 0.0005767833616118878,
      "step": 12570,
      "student_distance_mean": 1847.2100219726562,
      "teacher_distance_mean": 1.1535323977470398,
      "weighted_auxiliary_loss": 0.0005767833616118878,
      "weighted_relation_loss": 0.0005767833616118878
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.76800748362956,
      "flow_matching_loss": 0.03590912856161595,
      "grad_norm": 0.0956411138176918,
      "learning_rate": 6.712344831895074e-05,
      "loss": 0.0365,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005544707208173349,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.9112247467041,
      "state_query_norm": 1376.7552124023437,
      "state_relation_loss": 0.0005544707208173349,
      "step": 12580,
      "student_distance_mean": 1872.2649780273437,
      "teacher_distance_mean": 1.1679606795310975,
      "weighted_auxiliary_loss": 0.0005544707208173349,
      "weighted_relation_loss": 0.0005544707208173349
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.77736202057998,
      "flow_matching_loss": 0.03600184638053179,
      "grad_norm": 0.09943156689405441,
      "learning_rate": 6.707165515687811e-05,
      "loss": 0.0366,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005486560461577028,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.480690002441406,
      "state_query_norm": 1354.5141479492188,
      "state_relation_loss": 0.0005486560461577028,
      "step": 12590,
      "student_distance_mean": 1844.9993408203125,
      "teacher_distance_mean": 1.14703049659729,
      "weighted_auxiliary_loss": 0.0005486560461577028,
      "weighted_relation_loss": 0.0005486560461577028
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.786716557530402,
      "flow_matching_loss": 0.032744640670716765,
      "grad_norm": 0.10653050243854523,
      "learning_rate": 6.70198412511347e-05,
      "loss": 0.0333,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005656989233102649,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.28150901794434,
      "state_query_norm": 1350.8995727539063,
      "state_relation_loss": 0.0005656989233102649,
      "step": 12600,
      "student_distance_mean": 1838.1571166992187,
      "teacher_distance_mean": 1.1434422492980958,
      "weighted_auxiliary_loss": 0.0005656989233102649,
      "weighted_relation_loss": 0.0005656989233102649
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.796071094480823,
      "flow_matching_loss": 0.03555931206792593,
      "grad_norm": 0.11022868007421494,
      "learning_rate": 6.69680066646793e-05,
      "loss": 0.0361,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005785801622550935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.76442375183105,
      "state_query_norm": 1365.8920043945313,
      "state_relation_loss": 0.0005785801622550935,
      "step": 12610,
      "student_distance_mean": 1860.2368408203124,
      "teacher_distance_mean": 1.1569823384284974,
      "weighted_auxiliary_loss": 0.0005785801622550935,
      "weighted_relation_loss": 0.0005785801622550935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.805425631431245,
      "flow_matching_loss": 0.035566216334700584,
      "grad_norm": 0.11904102563858032,
      "learning_rate": 6.691615146049583e-05,
      "loss": 0.0361,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005538199358852581,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.99657859802246,
      "state_query_norm": 1381.0834716796876,
      "state_relation_loss": 0.0005538199358852581,
      "step": 12620,
      "student_distance_mean": 1876.6083129882813,
      "teacher_distance_mean": 1.1651317596435546,
      "weighted_auxiliary_loss": 0.0005538199358852581,
      "weighted_relation_loss": 0.0005538199358852581
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.814780168381665,
      "flow_matching_loss": 0.03662833068519831,
      "grad_norm": 0.13180987536907196,
      "learning_rate": 6.686427570159324e-05,
      "loss": 0.0372,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005718684813473373,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.13032150268555,
      "state_query_norm": 1378.9765380859376,
      "state_relation_loss": 0.0005718684813473373,
      "step": 12630,
      "student_distance_mean": 1878.27939453125,
      "teacher_distance_mean": 1.1694061756134033,
      "weighted_auxiliary_loss": 0.0005718684813473373,
      "weighted_relation_loss": 0.0005718684813473373
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.824134705332087,
      "flow_matching_loss": 0.03437716439366341,
      "grad_norm": 0.11649379134178162,
      "learning_rate": 6.681237945100548e-05,
      "loss": 0.0349,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005429137090686709,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 32.84069480895996,
      "state_query_norm": 1333.251513671875,
      "state_relation_loss": 0.0005429137090686709,
      "step": 12640,
      "student_distance_mean": 1811.0252563476563,
      "teacher_distance_mean": 1.1329307794570922,
      "weighted_auxiliary_loss": 0.0005429137090686709,
      "weighted_relation_loss": 0.0005429137090686709
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.833489242282507,
      "flow_matching_loss": 0.03514089435338974,
      "grad_norm": 0.1416541188955307,
      "learning_rate": 6.676046277179139e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005632155283819884,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.43947982788086,
      "state_query_norm": 1360.00439453125,
      "state_relation_loss": 0.0005632155283819884,
      "step": 12650,
      "student_distance_mean": 1847.7461059570312,
      "teacher_distance_mean": 1.1522767305374146,
      "weighted_auxiliary_loss": 0.0005632155283819884,
      "weighted_relation_loss": 0.0005632155283819884
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.842843779232927,
      "flow_matching_loss": 0.03649671785533428,
      "grad_norm": 0.13214458525180817,
      "learning_rate": 6.670852572703462e-05,
      "loss": 0.0371,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005648219783324748,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.55160636901856,
      "state_query_norm": 1351.9226440429688,
      "state_relation_loss": 0.0005648219783324748,
      "step": 12660,
      "student_distance_mean": 1847.2429931640625,
      "teacher_distance_mean": 1.1458010792732238,
      "weighted_auxiliary_loss": 0.0005648219783324748,
      "weighted_relation_loss": 0.0005648219783324748
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.85219831618335,
      "flow_matching_loss": 0.03879108820110559,
      "grad_norm": 0.15815088152885437,
      "learning_rate": 6.665656837984359e-05,
      "loss": 0.0393,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005451306409668177,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.234628677368164,
      "state_query_norm": 1345.6316772460937,
      "state_relation_loss": 0.0005451306409668177,
      "step": 12670,
      "student_distance_mean": 1834.9382934570312,
      "teacher_distance_mean": 1.1364343166351318,
      "weighted_auxiliary_loss": 0.0005451306409668177,
      "weighted_relation_loss": 0.0005451306409668177
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.86155285313377,
      "flow_matching_loss": 0.03356841709464788,
      "grad_norm": 0.16273042559623718,
      "learning_rate": 6.660459079335135e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005816413118736819,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.10062370300293,
      "state_query_norm": 1346.540234375,
      "state_relation_loss": 0.0005816413118736819,
      "step": 12680,
      "student_distance_mean": 1827.1184814453125,
      "teacher_distance_mean": 1.132398295402527,
      "weighted_auxiliary_loss": 0.0005816413118736819,
      "weighted_relation_loss": 0.0005816413118736819
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.870907390084191,
      "flow_matching_loss": 0.035036871768534185,
      "grad_norm": 0.09507856518030167,
      "learning_rate": 6.655259303071558e-05,
      "loss": 0.0356,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005853319365996867,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.374297523498534,
      "state_query_norm": 1351.6513793945312,
      "state_relation_loss": 0.0005853319365996867,
      "step": 12690,
      "student_distance_mean": 1837.2143676757812,
      "teacher_distance_mean": 1.1390497803688049,
      "weighted_auxiliary_loss": 0.0005853319365996867,
      "weighted_relation_loss": 0.0005853319365996867
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.880261927034612,
      "flow_matching_loss": 0.03497213441878557,
      "grad_norm": 0.11261899024248123,
      "learning_rate": 6.650057515511848e-05,
      "loss": 0.0355,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005614889523712918,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.89778099060059,
      "state_query_norm": 1373.2008911132812,
      "state_relation_loss": 0.0005614889523712918,
      "step": 12700,
      "student_distance_mean": 1872.5613525390625,
      "teacher_distance_mean": 1.1631970167160035,
      "weighted_auxiliary_loss": 0.0005614889523712918,
      "weighted_relation_loss": 0.0005614889523712918
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.889616463985032,
      "flow_matching_loss": 0.03714600652456283,
      "grad_norm": 0.13571004569530487,
      "learning_rate": 6.644853722976667e-05,
      "loss": 0.0377,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005664770607836545,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.70574188232422,
      "state_query_norm": 1365.5141967773438,
      "state_relation_loss": 0.0005664770607836545,
      "step": 12710,
      "student_distance_mean": 1857.2890258789062,
      "teacher_distance_mean": 1.157141101360321,
      "weighted_auxiliary_loss": 0.0005664770607836545,
      "weighted_relation_loss": 0.0005664770607836545
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.898971000935454,
      "flow_matching_loss": 0.03470824975520372,
      "grad_norm": 0.16510802507400513,
      "learning_rate": 6.639647931789113e-05,
      "loss": 0.0353,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005792287411168218,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.73932723999023,
      "state_query_norm": 1369.0328735351563,
      "state_relation_loss": 0.0005792287411168218,
      "step": 12720,
      "student_distance_mean": 1860.0516967773438,
      "teacher_distance_mean": 1.14924556016922,
      "weighted_auxiliary_loss": 0.0005792287411168218,
      "weighted_relation_loss": 0.0005792287411168218
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.908325537885874,
      "flow_matching_loss": 0.03521374482661486,
      "grad_norm": 0.13762477040290833,
      "learning_rate": 6.634440148274713e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005353428685339167,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.134491729736325,
      "state_query_norm": 1348.9369873046876,
      "state_relation_loss": 0.0005353428685339167,
      "step": 12730,
      "student_distance_mean": 1829.4330444335938,
      "teacher_distance_mean": 1.1339012026786803,
      "weighted_auxiliary_loss": 0.0005353428685339167,
      "weighted_relation_loss": 0.0005353428685339167
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.917680074836296,
      "flow_matching_loss": 0.03529563322663307,
      "grad_norm": 0.10759080946445465,
      "learning_rate": 6.629230378761415e-05,
      "loss": 0.0359,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005728635936975479,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.23334922790527,
      "state_query_norm": 1382.569921875,
      "state_relation_loss": 0.0005728635936975479,
      "step": 12740,
      "student_distance_mean": 1883.5114379882812,
      "teacher_distance_mean": 1.171689760684967,
      "weighted_auxiliary_loss": 0.0005728635936975479,
      "weighted_relation_loss": 0.0005728635936975479
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.927034611786716,
      "flow_matching_loss": 0.033340159058570865,
      "grad_norm": 0.10475928336381912,
      "learning_rate": 6.624018629579581e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005990748701151461,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.7564582824707,
      "state_query_norm": 1372.0756591796876,
      "state_relation_loss": 0.0005990748701151461,
      "step": 12750,
      "student_distance_mean": 1864.5121337890625,
      "teacher_distance_mean": 1.1524662733078004,
      "weighted_auxiliary_loss": 0.0005990748701151461,
      "weighted_relation_loss": 0.0005990748701151461
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.936389148737138,
      "flow_matching_loss": 0.036721831932663915,
      "grad_norm": 0.12848839163780212,
      "learning_rate": 6.618804907061976e-05,
      "loss": 0.0373,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005511769792065025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.79001235961914,
      "state_query_norm": 1360.2188598632813,
      "state_relation_loss": 0.0005511769792065025,
      "step": 12760,
      "student_distance_mean": 1861.7540161132813,
      "teacher_distance_mean": 1.149601697921753,
      "weighted_auxiliary_loss": 0.0005511769792065025,
      "weighted_relation_loss": 0.0005511769792065025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.945743685687559,
      "flow_matching_loss": 0.03513306248933077,
      "grad_norm": 0.11237502843141556,
      "learning_rate": 6.613589217543766e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005978651577606797,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.22255630493164,
      "state_query_norm": 1377.0712890625,
      "state_relation_loss": 0.0005978651577606797,
      "step": 12770,
      "student_distance_mean": 1882.767041015625,
      "teacher_distance_mean": 1.159646451473236,
      "weighted_auxiliary_loss": 0.0005978651577606797,
      "weighted_relation_loss": 0.0005978651577606797
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.955098222637979,
      "flow_matching_loss": 0.035635972023010255,
      "grad_norm": 0.11181820183992386,
      "learning_rate": 6.608371567362504e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005592737637925893,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.93188247680664,
      "state_query_norm": 1370.7254760742187,
      "state_relation_loss": 0.0005592737637925893,
      "step": 12780,
      "student_distance_mean": 1869.2089111328125,
      "teacher_distance_mean": 1.1531413793563843,
      "weighted_auxiliary_loss": 0.0005592737637925893,
      "weighted_relation_loss": 0.0005592737637925893
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.9644527595884,
      "flow_matching_loss": 0.03508018534630537,
      "grad_norm": 0.13361968100070953,
      "learning_rate": 6.60315196285813e-05,
      "loss": 0.0356,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005495488381711766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.52645206451416,
      "state_query_norm": 1369.9174072265625,
      "state_relation_loss": 0.0005495488381711766,
      "step": 12790,
      "student_distance_mean": 1853.6896850585938,
      "teacher_distance_mean": 1.1474790453910828,
      "weighted_auxiliary_loss": 0.0005495488381711766,
      "weighted_relation_loss": 0.0005495488381711766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.973807296538821,
      "flow_matching_loss": 0.03457036726176739,
      "grad_norm": 0.16147759556770325,
      "learning_rate": 6.597930410372954e-05,
      "loss": 0.0351,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005297895375406369,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.69515113830566,
      "state_query_norm": 1368.1798095703125,
      "state_relation_loss": 0.0005297895375406369,
      "step": 12800,
      "student_distance_mean": 1858.2239013671874,
      "teacher_distance_mean": 1.1511239767074586,
      "weighted_auxiliary_loss": 0.0005297895375406369,
      "weighted_relation_loss": 0.0005297895375406369
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.983161833489243,
      "flow_matching_loss": 0.034954659081995486,
      "grad_norm": 0.12014152109622955,
      "learning_rate": 6.592706916251653e-05,
      "loss": 0.0355,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005884874670300632,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.76570434570313,
      "state_query_norm": 1362.4538818359374,
      "state_relation_loss": 0.0005884874670300632,
      "step": 12810,
      "student_distance_mean": 1857.9187744140625,
      "teacher_distance_mean": 1.157417929172516,
      "weighted_auxiliary_loss": 0.0005884874670300632,
      "weighted_relation_loss": 0.0005884874670300632
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 11.992516370439663,
      "flow_matching_loss": 0.03765426017343998,
      "grad_norm": 0.14062358438968658,
      "learning_rate": 6.587481486841267e-05,
      "loss": 0.0382,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005697099259123206,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.80589904785156,
      "state_query_norm": 1366.1884887695312,
      "state_relation_loss": 0.0005697099259123206,
      "step": 12820,
      "student_distance_mean": 1864.261328125,
      "teacher_distance_mean": 1.1580115675926208,
      "weighted_auxiliary_loss": 0.0005697099259123206,
      "weighted_relation_loss": 0.0005697099259123206
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.001870907390083,
      "flow_matching_loss": 0.034772060625255105,
      "grad_norm": 0.10271519422531128,
      "learning_rate": 6.582254128491184e-05,
      "loss": 0.0353,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0005557939293794334,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 33.73814811706543,
      "state_query_norm": 1361.0812377929688,
      "state_relation_loss": 0.0005557939293794334,
      "step": 12830,
      "student_distance_mean": 1859.3590454101563,
      "teacher_distance_mean": 1.15314040184021,
      "weighted_auxiliary_loss": 0.0005557939293794334,
      "weighted_relation_loss": 0.0005557939293794334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.011225444340505,
      "flow_matching_loss": 0.033892786130309105,
      "grad_norm": 0.1090911328792572,
      "learning_rate": 6.577024847553139e-05,
      "loss": 0.0344,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005533031770028173,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.55682811737061,
      "state_query_norm": 1364.5905639648438,
      "state_relation_loss": 0.0005533031770028173,
      "step": 12840,
      "student_distance_mean": 1851.2525512695313,
      "teacher_distance_mean": 1.147624683380127,
      "weighted_auxiliary_loss": 0.0005533031770028173,
      "weighted_relation_loss": 0.0005533031770028173
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.020579981290926,
      "flow_matching_loss": 0.03319545891135931,
      "grad_norm": 0.14106890559196472,
      "learning_rate": 6.571793650381201e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005532786482945085,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.90981101989746,
      "state_query_norm": 1376.7589965820312,
      "state_relation_loss": 0.0005532786482945085,
      "step": 12850,
      "student_distance_mean": 1872.8334106445313,
      "teacher_distance_mean": 1.15635085105896,
      "weighted_auxiliary_loss": 0.0005532786482945085,
      "weighted_relation_loss": 0.0005532786482945085
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.029934518241348,
      "flow_matching_loss": 0.034874028712511065,
      "grad_norm": 0.17551274597644806,
      "learning_rate": 6.566560543331766e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005320591997588053,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.576181411743164,
      "state_query_norm": 1357.4920166015625,
      "state_relation_loss": 0.0005320591997588053,
      "step": 12860,
      "student_distance_mean": 1853.81083984375,
      "teacher_distance_mean": 1.143765389919281,
      "weighted_auxiliary_loss": 0.0005320591997588053,
      "weighted_relation_loss": 0.0005320591997588053
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.039289055191768,
      "flow_matching_loss": 0.03400576841086149,
      "grad_norm": 0.08719103038311005,
      "learning_rate": 6.561325532763554e-05,
      "loss": 0.0346,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006341487460304052,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.937330627441405,
      "state_query_norm": 1375.8712036132813,
      "state_relation_loss": 0.0006341487460304052,
      "step": 12870,
      "student_distance_mean": 1869.099462890625,
      "teacher_distance_mean": 1.1513893723487854,
      "weighted_auxiliary_loss": 0.0006341487460304052,
      "weighted_relation_loss": 0.0006341487460304052
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.04864359214219,
      "flow_matching_loss": 0.035335732996463774,
      "grad_norm": 0.12250352650880814,
      "learning_rate": 6.556088625037597e-05,
      "loss": 0.0359,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005602369841653853,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.27734375,
      "state_query_norm": 1381.5155151367187,
      "state_relation_loss": 0.0005602369841653853,
      "step": 12880,
      "student_distance_mean": 1882.7899291992187,
      "teacher_distance_mean": 1.1553545355796815,
      "weighted_auxiliary_loss": 0.0005602369841653853,
      "weighted_relation_loss": 0.0005602369841653853
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.05799812909261,
      "flow_matching_loss": 0.033786560967564584,
      "grad_norm": 0.08792028576135635,
      "learning_rate": 6.55084982651723e-05,
      "loss": 0.0343,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000557114533148706,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.8239086151123,
      "state_query_norm": 1371.1111572265625,
      "state_relation_loss": 0.000557114533148706,
      "step": 12890,
      "student_distance_mean": 1862.85654296875,
      "teacher_distance_mean": 1.141075575351715,
      "weighted_auxiliary_loss": 0.000557114533148706,
      "weighted_relation_loss": 0.000557114533148706
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.06735266604303,
      "flow_matching_loss": 0.034456795640289785,
      "grad_norm": 0.08119146525859833,
      "learning_rate": 6.54560914356809e-05,
      "loss": 0.035,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005624204175546765,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.17794342041016,
      "state_query_norm": 1382.9705322265625,
      "state_relation_loss": 0.0005624204175546765,
      "step": 12900,
      "student_distance_mean": 1882.8597778320313,
      "teacher_distance_mean": 1.1584792971611022,
      "weighted_auxiliary_loss": 0.0005624204175546765,
      "weighted_relation_loss": 0.0005624204175546765
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.076707202993452,
      "flow_matching_loss": 0.03320622090250254,
      "grad_norm": 0.1134096160531044,
      "learning_rate": 6.540366582558097e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005385913536883891,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.72477226257324,
      "state_query_norm": 1363.1209106445312,
      "state_relation_loss": 0.0005385913536883891,
      "step": 12910,
      "student_distance_mean": 1855.7213256835937,
      "teacher_distance_mean": 1.14308420419693,
      "weighted_auxiliary_loss": 0.0005385913536883891,
      "weighted_relation_loss": 0.0005385913536883891
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.086061739943872,
      "flow_matching_loss": 0.038226840645074846,
      "grad_norm": 0.10593322664499283,
      "learning_rate": 6.53512214985746e-05,
      "loss": 0.0388,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005650152277667075,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.97078971862793,
      "state_query_norm": 1380.708837890625,
      "state_relation_loss": 0.0005650152277667075,
      "step": 12920,
      "student_distance_mean": 1874.12490234375,
      "teacher_distance_mean": 1.153869366645813,
      "weighted_auxiliary_loss": 0.0005650152277667075,
      "weighted_relation_loss": 0.0005650152277667075
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.095416276894294,
      "flow_matching_loss": 0.03481939937919378,
      "grad_norm": 0.09844761341810226,
      "learning_rate": 6.529875851838658e-05,
      "loss": 0.0353,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005190639320062473,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.47433738708496,
      "state_query_norm": 1359.2846923828124,
      "state_relation_loss": 0.0005190639320062473,
      "step": 12930,
      "student_distance_mean": 1847.652294921875,
      "teacher_distance_mean": 1.1399267077445985,
      "weighted_auxiliary_loss": 0.0005190639320062473,
      "weighted_relation_loss": 0.0005190639320062473
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.104770813844715,
      "flow_matching_loss": 0.03461345247924328,
      "grad_norm": 0.1131996437907219,
      "learning_rate": 6.52462769487644e-05,
      "loss": 0.0352,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000549538159975782,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.362042617797854,
      "state_query_norm": 1353.7807861328124,
      "state_relation_loss": 0.000549538159975782,
      "step": 12940,
      "student_distance_mean": 1839.9131713867187,
      "teacher_distance_mean": 1.1296639561653137,
      "weighted_auxiliary_loss": 0.000549538159975782,
      "weighted_relation_loss": 0.000549538159975782
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.114125350795135,
      "flow_matching_loss": 0.03285321742296219,
      "grad_norm": 0.10778869688510895,
      "learning_rate": 6.519377685347807e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005452973127830774,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.27435569763183,
      "state_query_norm": 1393.6546142578125,
      "state_relation_loss": 0.0005452973127830774,
      "step": 12950,
      "student_distance_mean": 1892.0359130859374,
      "teacher_distance_mean": 1.161909532546997,
      "weighted_auxiliary_loss": 0.0005452973127830774,
      "weighted_relation_loss": 0.0005452973127830774
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.123479887745557,
      "flow_matching_loss": 0.03235692586749792,
      "grad_norm": 0.11958201229572296,
      "learning_rate": 6.514125829632021e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005539102421607822,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.11605339050293,
      "state_query_norm": 1388.6450073242188,
      "state_relation_loss": 0.0005539102421607822,
      "step": 12960,
      "student_distance_mean": 1882.4400634765625,
      "teacher_distance_mean": 1.1512606859207153,
      "weighted_auxiliary_loss": 0.0005539102421607822,
      "weighted_relation_loss": 0.0005539102421607822
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.132834424695977,
      "flow_matching_loss": 0.03271275609731674,
      "grad_norm": 0.13435110449790955,
      "learning_rate": 6.508872134110577e-05,
      "loss": 0.0333,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005732890509534627,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.16745147705078,
      "state_query_norm": 1392.6512084960937,
      "state_relation_loss": 0.0005732890509534627,
      "step": 12970,
      "student_distance_mean": 1886.9719482421874,
      "teacher_distance_mean": 1.159158504009247,
      "weighted_auxiliary_loss": 0.0005732890509534627,
      "weighted_relation_loss": 0.0005732890509534627
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.142188961646399,
      "flow_matching_loss": 0.034949944354593754,
      "grad_norm": 0.10045909881591797,
      "learning_rate": 6.503616605167213e-05,
      "loss": 0.0355,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005599845317192376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.86581573486328,
      "state_query_norm": 1374.67177734375,
      "state_relation_loss": 0.0005599845317192376,
      "step": 12980,
      "student_distance_mean": 1867.0248291015625,
      "teacher_distance_mean": 1.149847412109375,
      "weighted_auxiliary_loss": 0.0005599845317192376,
      "weighted_relation_loss": 0.0005599845317192376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.15154349859682,
      "flow_matching_loss": 0.03381097782403231,
      "grad_norm": 0.1345335692167282,
      "learning_rate": 6.498359249187893e-05,
      "loss": 0.0344,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005745138390921056,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.6768497467041,
      "state_query_norm": 1364.223974609375,
      "state_relation_loss": 0.0005745138390921056,
      "step": 12990,
      "student_distance_mean": 1856.1904174804688,
      "teacher_distance_mean": 1.142370593547821,
      "weighted_auxiliary_loss": 0.0005745138390921056,
      "weighted_relation_loss": 0.0005745138390921056
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.160898035547241,
      "flow_matching_loss": 0.033117355592548844,
      "grad_norm": 0.12224041670560837,
      "learning_rate": 6.493100072560799e-05,
      "loss": 0.0336,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005296484625432641,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.29188137054443,
      "state_query_norm": 1354.0962890625,
      "state_relation_loss": 0.0005296484625432641,
      "step": 13000,
      "student_distance_mean": 1838.2722778320312,
      "teacher_distance_mean": 1.1320282459259032,
      "weighted_auxiliary_loss": 0.0005296484625432641,
      "weighted_relation_loss": 0.0005296484625432641
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.170252572497661,
      "flow_matching_loss": 0.03362664002925157,
      "grad_norm": 0.09428923577070236,
      "learning_rate": 6.487839081676327e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005316452152328566,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.342921829223634,
      "state_query_norm": 1354.7130126953125,
      "state_relation_loss": 0.0005316452152328566,
      "step": 13010,
      "student_distance_mean": 1838.1850219726562,
      "teacher_distance_mean": 1.132324182987213,
      "weighted_auxiliary_loss": 0.0005316452152328566,
      "weighted_relation_loss": 0.0005316452152328566
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.179607109448082,
      "flow_matching_loss": 0.033337872847914694,
      "grad_norm": 0.1355864554643631,
      "learning_rate": 6.482576282927076e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005515164579264819,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.05134544372559,
      "state_query_norm": 1385.2092529296874,
      "state_relation_loss": 0.0005515164579264819,
      "step": 13020,
      "student_distance_mean": 1877.75322265625,
      "teacher_distance_mean": 1.1570387959480286,
      "weighted_auxiliary_loss": 0.0005515164579264819,
      "weighted_relation_loss": 0.0005515164579264819
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.188961646398504,
      "flow_matching_loss": 0.0327477166429162,
      "grad_norm": 0.14350780844688416,
      "learning_rate": 6.477311682707845e-05,
      "loss": 0.0333,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005371134058805183,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.8284912109375,
      "state_query_norm": 1376.5744140625,
      "state_relation_loss": 0.0005371134058805183,
      "step": 13030,
      "student_distance_mean": 1868.9299072265626,
      "teacher_distance_mean": 1.1541677117347717,
      "weighted_auxiliary_loss": 0.0005371134058805183,
      "weighted_relation_loss": 0.0005371134058805183
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.198316183348924,
      "flow_matching_loss": 0.03483604677021503,
      "grad_norm": 0.11971364915370941,
      "learning_rate": 6.472045287415616e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000550277266302146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.12805595397949,
      "state_query_norm": 1379.132958984375,
      "state_relation_loss": 0.000550277266302146,
      "step": 13040,
      "student_distance_mean": 1879.3067626953125,
      "teacher_distance_mean": 1.1557359099388123,
      "weighted_auxiliary_loss": 0.000550277266302146,
      "weighted_relation_loss": 0.000550277266302146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.207670720299346,
      "flow_matching_loss": 0.03250618875026703,
      "grad_norm": 0.15681952238082886,
      "learning_rate": 6.466777103449558e-05,
      "loss": 0.033,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005368267360609025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.74906578063965,
      "state_query_norm": 1365.763818359375,
      "state_relation_loss": 0.0005368267360609025,
      "step": 13050,
      "student_distance_mean": 1860.1061889648438,
      "teacher_distance_mean": 1.1452162146568299,
      "weighted_auxiliary_loss": 0.0005368267360609025,
      "weighted_relation_loss": 0.0005368267360609025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.217025257249766,
      "flow_matching_loss": 0.03318279609084129,
      "grad_norm": 0.14397287368774414,
      "learning_rate": 6.461507137211012e-05,
      "loss": 0.0338,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006015186721924692,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.13413009643555,
      "state_query_norm": 1367.3827026367187,
      "state_relation_loss": 0.0006015186721924692,
      "step": 13060,
      "student_distance_mean": 1875.1450561523438,
      "teacher_distance_mean": 1.1598275303840637,
      "weighted_auxiliary_loss": 0.0006015186721924692,
      "weighted_relation_loss": 0.0006015186721924692
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.226379794200188,
      "flow_matching_loss": 0.03448639046400785,
      "grad_norm": 0.11872731149196625,
      "learning_rate": 6.456235395103483e-05,
      "loss": 0.0351,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005639901239192114,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.232535934448244,
      "state_query_norm": 1382.875,
      "state_relation_loss": 0.0005639901239192114,
      "step": 13070,
      "student_distance_mean": 1885.7710327148438,
      "teacher_distance_mean": 1.1594314575195312,
      "weighted_auxiliary_loss": 0.0005639901239192114,
      "weighted_relation_loss": 0.0005639901239192114
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.235734331150608,
      "flow_matching_loss": 0.03247980047017336,
      "grad_norm": 0.14004994928836823,
      "learning_rate": 6.450961883532634e-05,
      "loss": 0.0331,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000576568953692913,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.854953384399415,
      "state_query_norm": 1363.731689453125,
      "state_relation_loss": 0.000576568953692913,
      "step": 13080,
      "student_distance_mean": 1865.62451171875,
      "teacher_distance_mean": 1.1483208417892456,
      "weighted_auxiliary_loss": 0.000576568953692913,
      "weighted_relation_loss": 0.000576568953692913
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.245088868101028,
      "flow_matching_loss": 0.03370347209274769,
      "grad_norm": 0.16698436439037323,
      "learning_rate": 6.445686608906283e-05,
      "loss": 0.0343,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005900574033148587,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.40676898956299,
      "state_query_norm": 1352.5059448242187,
      "state_relation_loss": 0.0005900574033148587,
      "step": 13090,
      "student_distance_mean": 1842.48173828125,
      "teacher_distance_mean": 1.1397994399070739,
      "weighted_auxiliary_loss": 0.0005900574033148587,
      "weighted_relation_loss": 0.0005900574033148587
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.25444340505145,
      "flow_matching_loss": 0.03494586814194918,
      "grad_norm": 0.13882362842559814,
      "learning_rate": 6.44040957763438e-05,
      "loss": 0.0355,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005446932394988835,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.720158386230466,
      "state_query_norm": 1365.875732421875,
      "state_relation_loss": 0.0005446932394988835,
      "step": 13100,
      "student_distance_mean": 1857.6526733398437,
      "teacher_distance_mean": 1.1459946870803832,
      "weighted_auxiliary_loss": 0.0005446932394988835,
      "weighted_relation_loss": 0.0005446932394988835
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.26379794200187,
      "flow_matching_loss": 0.035193195939064024,
      "grad_norm": 0.14792633056640625,
      "learning_rate": 6.435130796129018e-05,
      "loss": 0.0358,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0006293455022387207,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.690056228637694,
      "state_query_norm": 1401.8294677734375,
      "state_relation_loss": 0.0006293455022387207,
      "step": 13110,
      "student_distance_mean": 1911.5973876953126,
      "teacher_distance_mean": 1.174564492702484,
      "weighted_auxiliary_loss": 0.0006293455022387207,
      "weighted_relation_loss": 0.0006293455022387207
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.273152478952293,
      "flow_matching_loss": 0.035901119001209736,
      "grad_norm": 0.11669114232063293,
      "learning_rate": 6.429850270804416e-05,
      "loss": 0.0365,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005812334071379155,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.08191452026367,
      "state_query_norm": 1390.1608276367188,
      "state_relation_loss": 0.0005812334071379155,
      "step": 13120,
      "student_distance_mean": 1882.88955078125,
      "teacher_distance_mean": 1.155846154689789,
      "weighted_auxiliary_loss": 0.0005812334071379155,
      "weighted_relation_loss": 0.0005812334071379155
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.282507015902713,
      "flow_matching_loss": 0.035667655803263185,
      "grad_norm": 0.13011442124843597,
      "learning_rate": 6.424568008076909e-05,
      "loss": 0.0362,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005479200422996655,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.030416870117186,
      "state_query_norm": 1380.767919921875,
      "state_relation_loss": 0.0005479200422996655,
      "step": 13130,
      "student_distance_mean": 1877.1023315429688,
      "teacher_distance_mean": 1.1527645349502564,
      "weighted_auxiliary_loss": 0.0005479200422996655,
      "weighted_relation_loss": 0.0005479200422996655
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.291861552853133,
      "flow_matching_loss": 0.034208504110574724,
      "grad_norm": 0.11689786612987518,
      "learning_rate": 6.419284014364943e-05,
      "loss": 0.0347,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000541208015056327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.940642547607425,
      "state_query_norm": 1375.9851440429688,
      "state_relation_loss": 0.000541208015056327,
      "step": 13140,
      "student_distance_mean": 1873.5859985351562,
      "teacher_distance_mean": 1.1450601935386657,
      "weighted_auxiliary_loss": 0.000541208015056327,
      "weighted_relation_loss": 0.000541208015056327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.301216089803555,
      "flow_matching_loss": 0.03278195150196552,
      "grad_norm": 0.10834316164255142,
      "learning_rate": 6.41399829608907e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005747166171204299,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.54288291931152,
      "state_query_norm": 1389.7053466796874,
      "state_relation_loss": 0.0005747166171204299,
      "step": 13150,
      "student_distance_mean": 1901.76015625,
      "teacher_distance_mean": 1.1664836525917053,
      "weighted_auxiliary_loss": 0.0005747166171204299,
      "weighted_relation_loss": 0.0005747166171204299
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.310570626753975,
      "flow_matching_loss": 0.03336934614926577,
      "grad_norm": 0.13356667757034302,
      "learning_rate": 6.408710859671938e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005696055217413232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.68972930908203,
      "state_query_norm": 1394.9583374023437,
      "state_relation_loss": 0.0005696055217413232,
      "step": 13160,
      "student_distance_mean": 1908.4475708007812,
      "teacher_distance_mean": 1.1650811910629273,
      "weighted_auxiliary_loss": 0.0005696055217413232,
      "weighted_relation_loss": 0.0005696055217413232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.319925163704397,
      "flow_matching_loss": 0.03659386727958917,
      "grad_norm": 0.1699393391609192,
      "learning_rate": 6.403421711538278e-05,
      "loss": 0.0371,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000534723253804259,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.1967155456543,
      "state_query_norm": 1383.3874389648438,
      "state_relation_loss": 0.000534723253804259,
      "step": 13170,
      "student_distance_mean": 1883.88486328125,
      "teacher_distance_mean": 1.1503763914108276,
      "weighted_auxiliary_loss": 0.000534723253804259,
      "weighted_relation_loss": 0.000534723253804259
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.329279700654817,
      "flow_matching_loss": 0.03660158701241016,
      "grad_norm": 0.13761229813098907,
      "learning_rate": 6.398130858114903e-05,
      "loss": 0.0371,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005416233267169446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.102351379394534,
      "state_query_norm": 1379.465625,
      "state_relation_loss": 0.0005416233267169446,
      "step": 13180,
      "student_distance_mean": 1879.324267578125,
      "teacher_distance_mean": 1.146336591243744,
      "weighted_auxiliary_loss": 0.0005416233267169446,
      "weighted_relation_loss": 0.0005416233267169446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.33863423760524,
      "flow_matching_loss": 0.035388001799583436,
      "grad_norm": 0.16514933109283447,
      "learning_rate": 6.392838305830701e-05,
      "loss": 0.0359,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005454462196212261,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.003611373901364,
      "state_query_norm": 1382.6993286132813,
      "state_relation_loss": 0.0005454462196212261,
      "step": 13190,
      "student_distance_mean": 1877.0467529296875,
      "teacher_distance_mean": 1.1460875868797302,
      "weighted_auxiliary_loss": 0.0005454462196212261,
      "weighted_relation_loss": 0.0005454462196212261
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.34798877455566,
      "flow_matching_loss": 0.035088539123535156,
      "grad_norm": 0.14880415797233582,
      "learning_rate": 6.387544061116621e-05,
      "loss": 0.0356,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005518334510270506,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.37104263305664,
      "state_query_norm": 1387.080859375,
      "state_relation_loss": 0.0005518334510270506,
      "step": 13200,
      "student_distance_mean": 1890.9112670898437,
      "teacher_distance_mean": 1.1581108093261718,
      "weighted_auxiliary_loss": 0.0005518334510270506,
      "weighted_relation_loss": 0.0005518334510270506
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.35734331150608,
      "flow_matching_loss": 0.03284466098994017,
      "grad_norm": 0.13160958886146545,
      "learning_rate": 6.382248130405671e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005342674645362422,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.783613204956055,
      "state_query_norm": 1371.5785766601562,
      "state_relation_loss": 0.0005342674645362422,
      "step": 13210,
      "student_distance_mean": 1864.6288696289062,
      "teacher_distance_mean": 1.1359113216400147,
      "weighted_auxiliary_loss": 0.0005342674645362422,
      "weighted_relation_loss": 0.0005342674645362422
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.366697848456502,
      "flow_matching_loss": 0.032998408749699594,
      "grad_norm": 0.09196089953184128,
      "learning_rate": 6.376950520132905e-05,
      "loss": 0.0335,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005460461892653256,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.23490982055664,
      "state_query_norm": 1386.2718383789063,
      "state_relation_loss": 0.0005460461892653256,
      "step": 13220,
      "student_distance_mean": 1883.737939453125,
      "teacher_distance_mean": 1.1511877655982972,
      "weighted_auxiliary_loss": 0.0005460461892653256,
      "weighted_relation_loss": 0.0005460461892653256
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.376052385406922,
      "flow_matching_loss": 0.03310132622718811,
      "grad_norm": 0.09132980555295944,
      "learning_rate": 6.371651236735418e-05,
      "loss": 0.0336,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000545147669618018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.4242977142334,
      "state_query_norm": 1357.7833618164063,
      "state_relation_loss": 0.000545147669618018,
      "step": 13230,
      "student_distance_mean": 1844.0717163085938,
      "teacher_distance_mean": 1.128819990158081,
      "weighted_auxiliary_loss": 0.000545147669618018,
      "weighted_relation_loss": 0.000545147669618018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.385406922357344,
      "flow_matching_loss": 0.03319888021796942,
      "grad_norm": 0.08153130114078522,
      "learning_rate": 6.366350286652341e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000524572454742156,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.0714527130127,
      "state_query_norm": 1384.0266357421874,
      "state_relation_loss": 0.000524572454742156,
      "step": 13240,
      "student_distance_mean": 1880.4246704101563,
      "teacher_distance_mean": 1.1516956567764283,
      "weighted_auxiliary_loss": 0.000524572454742156,
      "weighted_relation_loss": 0.000524572454742156
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.394761459307764,
      "flow_matching_loss": 0.034250630810856816,
      "grad_norm": 0.11776544153690338,
      "learning_rate": 6.361047676324827e-05,
      "loss": 0.0348,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005599541589617729,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.878316497802736,
      "state_query_norm": 1379.4487426757812,
      "state_relation_loss": 0.0005599541589617729,
      "step": 13250,
      "student_distance_mean": 1869.4467895507812,
      "teacher_distance_mean": 1.1509859085083007,
      "weighted_auxiliary_loss": 0.0005599541589617729,
      "weighted_relation_loss": 0.0005599541589617729
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.404115996258184,
      "flow_matching_loss": 0.032102739438414574,
      "grad_norm": 0.09432778507471085,
      "learning_rate": 6.355743412196047e-05,
      "loss": 0.0326,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005402406968642026,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.742156982421875,
      "state_query_norm": 1361.6160278320312,
      "state_relation_loss": 0.0005402406968642026,
      "step": 13260,
      "student_distance_mean": 1860.27529296875,
      "teacher_distance_mean": 1.1466607093811034,
      "weighted_auxiliary_loss": 0.0005402406968642026,
      "weighted_relation_loss": 0.0005402406968642026
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.413470533208606,
      "flow_matching_loss": 0.0350306648761034,
      "grad_norm": 0.10729626566171646,
      "learning_rate": 6.350437500711184e-05,
      "loss": 0.0356,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005390191334299743,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.81477851867676,
      "state_query_norm": 1368.0591918945313,
      "state_relation_loss": 0.0005390191334299743,
      "step": 13270,
      "student_distance_mean": 1864.90615234375,
      "teacher_distance_mean": 1.1456146121025086,
      "weighted_auxiliary_loss": 0.0005390191334299743,
      "weighted_relation_loss": 0.0005390191334299743
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.422825070159027,
      "flow_matching_loss": 0.0333600539714098,
      "grad_norm": 0.1213686540722847,
      "learning_rate": 6.345129948317418e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005506511661224067,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.84481506347656,
      "state_query_norm": 1370.465380859375,
      "state_relation_loss": 0.0005506511661224067,
      "step": 13280,
      "student_distance_mean": 1865.1518432617188,
      "teacher_distance_mean": 1.1434121489524842,
      "weighted_auxiliary_loss": 0.0005506511661224067,
      "weighted_relation_loss": 0.0005506511661224067
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.432179607109449,
      "flow_matching_loss": 0.03284583874046802,
      "grad_norm": 0.09097135812044144,
      "learning_rate": 6.33982076146393e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005380821356084197,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.104065322875975,
      "state_query_norm": 1379.4584228515625,
      "state_relation_loss": 0.0005380821356084197,
      "step": 13290,
      "student_distance_mean": 1877.2346801757812,
      "teacher_distance_mean": 1.146778655052185,
      "weighted_auxiliary_loss": 0.0005380821356084197,
      "weighted_relation_loss": 0.0005380821356084197
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.441534144059869,
      "flow_matching_loss": 0.03360303435474634,
      "grad_norm": 0.08523769676685333,
      "learning_rate": 6.334509946601879e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005525556014617905,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.986406326293945,
      "state_query_norm": 1375.1765869140625,
      "state_relation_loss": 0.0005525556014617905,
      "step": 13300,
      "student_distance_mean": 1876.8108764648437,
      "teacher_distance_mean": 1.1478230834007264,
      "weighted_auxiliary_loss": 0.0005525556014617905,
      "weighted_relation_loss": 0.0005525556014617905
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.45088868101029,
      "flow_matching_loss": 0.03330127000808716,
      "grad_norm": 0.11617980152368546,
      "learning_rate": 6.329197510184406e-05,
      "loss": 0.0338,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005335125606507063,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.47025566101074,
      "state_query_norm": 1403.4163696289063,
      "state_relation_loss": 0.0005335125606507063,
      "step": 13310,
      "student_distance_mean": 1900.9107421875,
      "teacher_distance_mean": 1.1610281229019166,
      "weighted_auxiliary_loss": 0.0005335125606507063,
      "weighted_relation_loss": 0.0005335125606507063
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.460243217960711,
      "flow_matching_loss": 0.033531700074672696,
      "grad_norm": 0.11262551695108414,
      "learning_rate": 6.323883458666624e-05,
      "loss": 0.0341,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005452338402392343,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.44798316955566,
      "state_query_norm": 1397.1473022460937,
      "state_relation_loss": 0.0005452338402392343,
      "step": 13320,
      "student_distance_mean": 1898.2490966796875,
      "teacher_distance_mean": 1.1549588918685914,
      "weighted_auxiliary_loss": 0.0005452338402392343,
      "weighted_relation_loss": 0.0005452338402392343
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.469597754911131,
      "flow_matching_loss": 0.032767017185688016,
      "grad_norm": 0.12213599681854248,
      "learning_rate": 6.318567798505605e-05,
      "loss": 0.0333,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005572636669967324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.67438869476318,
      "state_query_norm": 1412.6241088867187,
      "state_relation_loss": 0.0005572636669967324,
      "step": 13330,
      "student_distance_mean": 1915.4588989257813,
      "teacher_distance_mean": 1.1646840453147889,
      "weighted_auxiliary_loss": 0.0005572636669967324,
      "weighted_relation_loss": 0.0005572636669967324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.478952291861553,
      "flow_matching_loss": 0.03600866515189409,
      "grad_norm": 0.10937313735485077,
      "learning_rate": 6.313250536160378e-05,
      "loss": 0.0366,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005860624893102795,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.468074798583984,
      "state_query_norm": 1386.0278076171876,
      "state_relation_loss": 0.0005860624893102795,
      "step": 13340,
      "student_distance_mean": 1898.4372192382812,
      "teacher_distance_mean": 1.1580319046974181,
      "weighted_auxiliary_loss": 0.0005860624893102795,
      "weighted_relation_loss": 0.0005860624893102795
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.488306828811973,
      "flow_matching_loss": 0.03381382301449776,
      "grad_norm": 0.1244410052895546,
      "learning_rate": 6.307931678091917e-05,
      "loss": 0.0344,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005391663697082549,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.85632438659668,
      "state_query_norm": 1370.0120239257812,
      "state_relation_loss": 0.0005391663697082549,
      "step": 13350,
      "student_distance_mean": 1867.303173828125,
      "teacher_distance_mean": 1.1413192987442016,
      "weighted_auxiliary_loss": 0.0005391663697082549,
      "weighted_relation_loss": 0.0005391663697082549
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.497661365762395,
      "flow_matching_loss": 0.03297571018338204,
      "grad_norm": 0.12982654571533203,
      "learning_rate": 6.302611230763138e-05,
      "loss": 0.0335,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005526331049622967,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.47392234802246,
      "state_query_norm": 1392.3347534179688,
      "state_relation_loss": 0.0005526331049622967,
      "step": 13360,
      "student_distance_mean": 1898.4016723632812,
      "teacher_distance_mean": 1.1599775552749634,
      "weighted_auxiliary_loss": 0.0005526331049622967,
      "weighted_relation_loss": 0.0005526331049622967
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.507015902712816,
      "flow_matching_loss": 0.03360382299870253,
      "grad_norm": 0.11642047017812729,
      "learning_rate": 6.297289200638887e-05,
      "loss": 0.0341,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005424060014775023,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.05122718811035,
      "state_query_norm": 1381.0431762695312,
      "state_relation_loss": 0.0005424060014775023,
      "step": 13370,
      "student_distance_mean": 1877.2505859375,
      "teacher_distance_mean": 1.1416566729545594,
      "weighted_auxiliary_loss": 0.0005424060014775023,
      "weighted_relation_loss": 0.0005424060014775023
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.516370439663238,
      "flow_matching_loss": 0.03164322916418314,
      "grad_norm": 0.14548014104366302,
      "learning_rate": 6.29196559418593e-05,
      "loss": 0.0322,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005503008287632838,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.71284446716309,
      "state_query_norm": 1348.1709838867187,
      "state_relation_loss": 0.0005503008287632838,
      "step": 13380,
      "student_distance_mean": 1850.1348754882813,
      "teacher_distance_mean": 1.1312272548675537,
      "weighted_auxiliary_loss": 0.0005503008287632838,
      "weighted_relation_loss": 0.0005503008287632838
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.525724976613658,
      "flow_matching_loss": 0.03339714240282774,
      "grad_norm": 0.1496352255344391,
      "learning_rate": 6.286640417872951e-05,
      "loss": 0.034,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005741410481277853,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.32539367675781,
      "state_query_norm": 1380.6368774414063,
      "state_relation_loss": 0.0005741410481277853,
      "step": 13390,
      "student_distance_mean": 1888.0544799804688,
      "teacher_distance_mean": 1.1554317355155945,
      "weighted_auxiliary_loss": 0.0005741410481277853,
      "weighted_relation_loss": 0.0005741410481277853
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.535079513564078,
      "flow_matching_loss": 0.033548619411885736,
      "grad_norm": 0.08483465760946274,
      "learning_rate": 6.281313678170542e-05,
      "loss": 0.0341,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005627329577691853,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.43641891479492,
      "state_query_norm": 1395.4680053710938,
      "state_relation_loss": 0.0005627329577691853,
      "step": 13400,
      "student_distance_mean": 1897.6017944335938,
      "teacher_distance_mean": 1.1590094923973084,
      "weighted_auxiliary_loss": 0.0005627329577691853,
      "weighted_relation_loss": 0.0005627329577691853
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.5444340505145,
      "flow_matching_loss": 0.034673265367746356,
      "grad_norm": 0.10665778815746307,
      "learning_rate": 6.275985381551194e-05,
      "loss": 0.0352,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005685615120455623,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.21014060974121,
      "state_query_norm": 1385.4419921875,
      "state_relation_loss": 0.0005685615120455623,
      "step": 13410,
      "student_distance_mean": 1884.6203979492188,
      "teacher_distance_mean": 1.152994692325592,
      "weighted_auxiliary_loss": 0.0005685615120455623,
      "weighted_relation_loss": 0.0005685615120455623
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.55378858746492,
      "flow_matching_loss": 0.03299186266958713,
      "grad_norm": 0.08229345083236694,
      "learning_rate": 6.270655534489292e-05,
      "loss": 0.0336,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005657576897647232,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.51127052307129,
      "state_query_norm": 1394.2329711914062,
      "state_relation_loss": 0.0005657576897647232,
      "step": 13420,
      "student_distance_mean": 1901.7635620117187,
      "teacher_distance_mean": 1.1587224841117858,
      "weighted_auxiliary_loss": 0.0005657576897647232,
      "weighted_relation_loss": 0.0005657576897647232
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.563143124415342,
      "flow_matching_loss": 0.03450034316629171,
      "grad_norm": 0.12892356514930725,
      "learning_rate": 6.265324143461098e-05,
      "loss": 0.035,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000532440611277707,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.2330020904541,
      "state_query_norm": 1380.4371215820313,
      "state_relation_loss": 0.000532440611277707,
      "step": 13430,
      "student_distance_mean": 1884.3700439453125,
      "teacher_distance_mean": 1.1494407296180724,
      "weighted_auxiliary_loss": 0.000532440611277707,
      "weighted_relation_loss": 0.000532440611277707
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.572497661365762,
      "flow_matching_loss": 0.03482176251709461,
      "grad_norm": 0.1256273090839386,
      "learning_rate": 6.259991214944758e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005295324081089348,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.5142147064209,
      "state_query_norm": 1401.5420776367187,
      "state_relation_loss": 0.0005295324081089348,
      "step": 13440,
      "student_distance_mean": 1907.498193359375,
      "teacher_distance_mean": 1.16168270111084,
      "weighted_auxiliary_loss": 0.0005295324081089348,
      "weighted_relation_loss": 0.0005295324081089348
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.581852198316183,
      "flow_matching_loss": 0.03313762787729502,
      "grad_norm": 0.12541663646697998,
      "learning_rate": 6.254656755420283e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005542933795368299,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.18975944519043,
      "state_query_norm": 1388.7183349609375,
      "state_relation_loss": 0.0005542933795368299,
      "step": 13450,
      "student_distance_mean": 1883.8481567382812,
      "teacher_distance_mean": 1.1536863565444946,
      "weighted_auxiliary_loss": 0.0005542933795368299,
      "weighted_relation_loss": 0.0005542933795368299
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.591206735266605,
      "flow_matching_loss": 0.03416298683732748,
      "grad_norm": 0.10202483832836151,
      "learning_rate": 6.249320771369545e-05,
      "loss": 0.0347,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005322250508470461,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.100049591064455,
      "state_query_norm": 1374.6580932617187,
      "state_relation_loss": 0.0005322250508470461,
      "step": 13460,
      "student_distance_mean": 1878.937109375,
      "teacher_distance_mean": 1.1486444354057312,
      "weighted_auxiliary_loss": 0.0005322250508470461,
      "weighted_relation_loss": 0.0005322250508470461
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.600561272217025,
      "flow_matching_loss": 0.03219859972596169,
      "grad_norm": 0.10109372437000275,
      "learning_rate": 6.243983269276263e-05,
      "loss": 0.0328,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000563535062246956,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.53380699157715,
      "state_query_norm": 1390.2109375,
      "state_relation_loss": 0.000563535062246956,
      "step": 13470,
      "student_distance_mean": 1901.2616943359376,
      "teacher_distance_mean": 1.1590612173080443,
      "weighted_auxiliary_loss": 0.000563535062246956,
      "weighted_relation_loss": 0.000563535062246956
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.609915809167447,
      "flow_matching_loss": 0.03371468316763639,
      "grad_norm": 0.11326862871646881,
      "learning_rate": 6.238644255626012e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005240302183665335,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.634144973754886,
      "state_query_norm": 1398.6341796875,
      "state_relation_loss": 0.0005240302183665335,
      "step": 13480,
      "student_distance_mean": 1911.1739868164063,
      "teacher_distance_mean": 1.161111843585968,
      "weighted_auxiliary_loss": 0.0005240302183665335,
      "weighted_relation_loss": 0.0005240302183665335
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.619270346117867,
      "flow_matching_loss": 0.03573120981454849,
      "grad_norm": 0.13981163501739502,
      "learning_rate": 6.233303736906193e-05,
      "loss": 0.0363,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005398235691245646,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.427925491333006,
      "state_query_norm": 1385.0215576171875,
      "state_relation_loss": 0.0005398235691245646,
      "step": 13490,
      "student_distance_mean": 1895.4877685546876,
      "teacher_distance_mean": 1.1572656393051148,
      "weighted_auxiliary_loss": 0.0005398235691245646,
      "weighted_relation_loss": 0.0005398235691245646
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.628624883068287,
      "flow_matching_loss": 0.035001777671277524,
      "grad_norm": 0.12962032854557037,
      "learning_rate": 6.227961719606039e-05,
      "loss": 0.0355,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005186440132092685,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.23730964660645,
      "state_query_norm": 1389.5552978515625,
      "state_relation_loss": 0.0005186440132092685,
      "step": 13500,
      "student_distance_mean": 1890.1695434570313,
      "teacher_distance_mean": 1.1500919818878175,
      "weighted_auxiliary_loss": 0.0005186440132092685,
      "weighted_relation_loss": 0.0005186440132092685
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.63797942001871,
      "flow_matching_loss": 0.03487109150737524,
      "grad_norm": 0.11731258034706116,
      "learning_rate": 6.222618210216608e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005398560926551,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.62374153137207,
      "state_query_norm": 1396.36220703125,
      "state_relation_loss": 0.0005398560926551,
      "step": 13510,
      "student_distance_mean": 1906.2914306640625,
      "teacher_distance_mean": 1.1626537680625915,
      "weighted_auxiliary_loss": 0.0005398560926551,
      "weighted_relation_loss": 0.0005398560926551
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.64733395696913,
      "flow_matching_loss": 0.03317648731172085,
      "grad_norm": 0.1101493164896965,
      "learning_rate": 6.217273215230767e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005306571052642539,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.378224563598636,
      "state_query_norm": 1385.356298828125,
      "state_relation_loss": 0.0005306571052642539,
      "step": 13520,
      "student_distance_mean": 1896.6200927734376,
      "teacher_distance_mean": 1.1572535514831543,
      "weighted_auxiliary_loss": 0.0005306571052642539,
      "weighted_relation_loss": 0.0005306571052642539
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.656688493919551,
      "flow_matching_loss": 0.03350574634969235,
      "grad_norm": 0.09322633594274521,
      "learning_rate": 6.211926741143188e-05,
      "loss": 0.0341,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005524354113731533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.302738952636716,
      "state_query_norm": 1390.7280517578124,
      "state_relation_loss": 0.0005524354113731533,
      "step": 13530,
      "student_distance_mean": 1892.3844970703126,
      "teacher_distance_mean": 1.1509055137634276,
      "weighted_auxiliary_loss": 0.0005524354113731533,
      "weighted_relation_loss": 0.0005524354113731533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.666043030869972,
      "flow_matching_loss": 0.03470739386975765,
      "grad_norm": 0.10867495834827423,
      "learning_rate": 6.20657879445034e-05,
      "loss": 0.0353,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000565448624547571,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.66987075805664,
      "state_query_norm": 1392.2470092773438,
      "state_relation_loss": 0.000565448624547571,
      "step": 13540,
      "student_distance_mean": 1908.0598754882812,
      "teacher_distance_mean": 1.1615307211875916,
      "weighted_auxiliary_loss": 0.000565448624547571,
      "weighted_relation_loss": 0.000565448624547571
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.675397567820394,
      "flow_matching_loss": 0.03369074501097202,
      "grad_norm": 0.10389673709869385,
      "learning_rate": 6.201229381650485e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005524031032109634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.56692276000977,
      "state_query_norm": 1394.4934448242188,
      "state_relation_loss": 0.0005524031032109634,
      "step": 13550,
      "student_distance_mean": 1904.6327514648438,
      "teacher_distance_mean": 1.1534531474113465,
      "weighted_auxiliary_loss": 0.0005524031032109634,
      "weighted_relation_loss": 0.0005524031032109634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.684752104770814,
      "flow_matching_loss": 0.032348154485225676,
      "grad_norm": 0.1137913316488266,
      "learning_rate": 6.195878509243661e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005422569054644554,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.863487243652344,
      "state_query_norm": 1398.05166015625,
      "state_relation_loss": 0.0005422569054644554,
      "step": 13560,
      "student_distance_mean": 1916.177978515625,
      "teacher_distance_mean": 1.1662582993507384,
      "weighted_auxiliary_loss": 0.0005422569054644554,
      "weighted_relation_loss": 0.0005422569054644554
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.694106641721234,
      "flow_matching_loss": 0.03485742751508951,
      "grad_norm": 0.09583588689565659,
      "learning_rate": 6.190526183731685e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005441154935397208,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.52503967285156,
      "state_query_norm": 1394.57138671875,
      "state_relation_loss": 0.0005441154935397208,
      "step": 13570,
      "student_distance_mean": 1903.0041870117188,
      "teacher_distance_mean": 1.16360844373703,
      "weighted_auxiliary_loss": 0.0005441154935397208,
      "weighted_relation_loss": 0.0005441154935397208
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.703461178671656,
      "flow_matching_loss": 0.03312217332422733,
      "grad_norm": 0.09545975178480148,
      "learning_rate": 6.185172411618138e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005555808020289987,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.59837188720703,
      "state_query_norm": 1395.5277099609375,
      "state_relation_loss": 0.0005555808020289987,
      "step": 13580,
      "student_distance_mean": 1902.6173217773437,
      "teacher_distance_mean": 1.159988784790039,
      "weighted_auxiliary_loss": 0.0005555808020289987,
      "weighted_relation_loss": 0.0005555808020289987
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.712815715622076,
      "flow_matching_loss": 0.033423728123307225,
      "grad_norm": 0.16869091987609863,
      "learning_rate": 6.179817199408355e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000520964662428014,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.456173706054685,
      "state_query_norm": 1400.3824584960937,
      "state_relation_loss": 0.000520964662428014,
      "step": 13590,
      "student_distance_mean": 1901.3250732421875,
      "teacher_distance_mean": 1.153759729862213,
      "weighted_auxiliary_loss": 0.000520964662428014,
      "weighted_relation_loss": 0.000520964662428014
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.722170252572498,
      "flow_matching_loss": 0.03507211096584797,
      "grad_norm": 0.1279313564300537,
      "learning_rate": 6.174460553609426e-05,
      "loss": 0.0356,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005292721529258415,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.027320861816406,
      "state_query_norm": 1372.9777954101562,
      "state_relation_loss": 0.0005292721529258415,
      "step": 13600,
      "student_distance_mean": 1870.939794921875,
      "teacher_distance_mean": 1.1415174841880797,
      "weighted_auxiliary_loss": 0.0005292721529258415,
      "weighted_relation_loss": 0.0005292721529258415
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.731524789522918,
      "flow_matching_loss": 0.03199467398226261,
      "grad_norm": 0.08097405731678009,
      "learning_rate": 6.16910248073018e-05,
      "loss": 0.0325,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005304126592818647,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.96663398742676,
      "state_query_norm": 1388.181396484375,
      "state_relation_loss": 0.0005304126592818647,
      "step": 13610,
      "student_distance_mean": 1876.5815551757812,
      "teacher_distance_mean": 1.1444597005844117,
      "weighted_auxiliary_loss": 0.0005304126592818647,
      "weighted_relation_loss": 0.0005304126592818647
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.74087932647334,
      "flow_matching_loss": 0.03368698544800282,
      "grad_norm": 0.09401670843362808,
      "learning_rate": 6.16374298728118e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005438303574919701,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.25295181274414,
      "state_query_norm": 1387.5388793945312,
      "state_relation_loss": 0.0005438303574919701,
      "step": 13620,
      "student_distance_mean": 1885.6225830078124,
      "teacher_distance_mean": 1.1465088367462157,
      "weighted_auxiliary_loss": 0.0005438303574919701,
      "weighted_relation_loss": 0.0005438303574919701
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.75023386342376,
      "flow_matching_loss": 0.035117927566170694,
      "grad_norm": 0.12417878210544586,
      "learning_rate": 6.158382079774716e-05,
      "loss": 0.0357,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005382886505685746,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.484823608398436,
      "state_query_norm": 1395.2960083007813,
      "state_relation_loss": 0.0005382886505685746,
      "step": 13630,
      "student_distance_mean": 1900.8241455078125,
      "teacher_distance_mean": 1.1568788886070251,
      "weighted_auxiliary_loss": 0.0005382886505685746,
      "weighted_relation_loss": 0.0005382886505685746
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.75958840037418,
      "flow_matching_loss": 0.032542500644922256,
      "grad_norm": 0.18986958265304565,
      "learning_rate": 6.153019764724799e-05,
      "loss": 0.0331,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005384816962759942,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.34064979553223,
      "state_query_norm": 1396.74833984375,
      "state_relation_loss": 0.0005384816962759942,
      "step": 13640,
      "student_distance_mean": 1895.0714111328125,
      "teacher_distance_mean": 1.1561917901039123,
      "weighted_auxiliary_loss": 0.0005384816962759942,
      "weighted_relation_loss": 0.0005384816962759942
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.768942937324603,
      "flow_matching_loss": 0.03298244718462229,
      "grad_norm": 0.1409192532300949,
      "learning_rate": 6.147656048647144e-05,
      "loss": 0.0335,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005350369436200708,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.105308532714844,
      "state_query_norm": 1378.292333984375,
      "state_relation_loss": 0.0005350369436200708,
      "step": 13650,
      "student_distance_mean": 1881.39951171875,
      "teacher_distance_mean": 1.1464359045028687,
      "weighted_auxiliary_loss": 0.0005350369436200708,
      "weighted_relation_loss": 0.0005350369436200708
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.778297474275023,
      "flow_matching_loss": 0.03390056993812322,
      "grad_norm": 0.11483512818813324,
      "learning_rate": 6.142290938059173e-05,
      "loss": 0.0345,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005546171916648746,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.033678436279295,
      "state_query_norm": 1413.1066528320312,
      "state_relation_loss": 0.0005546171916648746,
      "step": 13660,
      "student_distance_mean": 1931.8253784179688,
      "teacher_distance_mean": 1.1747312664985656,
      "weighted_auxiliary_loss": 0.0005546171916648746,
      "weighted_relation_loss": 0.0005546171916648746
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.787652011225445,
      "flow_matching_loss": 0.033456556126475336,
      "grad_norm": 0.11844764649868011,
      "learning_rate": 6.136924439480001e-05,
      "loss": 0.034,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005465435184305534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.95448799133301,
      "state_query_norm": 1368.32412109375,
      "state_relation_loss": 0.0005465435184305534,
      "step": 13670,
      "student_distance_mean": 1869.089208984375,
      "teacher_distance_mean": 1.1396848320961,
      "weighted_auxiliary_loss": 0.0005465435184305534,
      "weighted_relation_loss": 0.0005465435184305534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.797006548175865,
      "flow_matching_loss": 0.03266341853886843,
      "grad_norm": 0.10164126008749008,
      "learning_rate": 6.13155655943043e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005764340690802783,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.566491317749026,
      "state_query_norm": 1407.6701782226562,
      "state_relation_loss": 0.0005764340690802783,
      "step": 13680,
      "student_distance_mean": 1906.1975708007812,
      "teacher_distance_mean": 1.1644186139106751,
      "weighted_auxiliary_loss": 0.0005764340690802783,
      "weighted_relation_loss": 0.0005764340690802783
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.806361085126285,
      "flow_matching_loss": 0.03154821041971445,
      "grad_norm": 0.1306605488061905,
      "learning_rate": 6.126187304432941e-05,
      "loss": 0.0321,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005432857200503349,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.67407112121582,
      "state_query_norm": 1401.9472534179688,
      "state_relation_loss": 0.0005432857200503349,
      "step": 13690,
      "student_distance_mean": 1910.7853637695312,
      "teacher_distance_mean": 1.1602435469627381,
      "weighted_auxiliary_loss": 0.0005432857200503349,
      "weighted_relation_loss": 0.0005432857200503349
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.815715622076707,
      "flow_matching_loss": 0.03442786671221256,
      "grad_norm": 0.09746714681386948,
      "learning_rate": 6.120816681011684e-05,
      "loss": 0.035,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005241564416792244,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.45975303649902,
      "state_query_norm": 1393.027392578125,
      "state_relation_loss": 0.0005241564416792244,
      "step": 13700,
      "student_distance_mean": 1900.0667602539063,
      "teacher_distance_mean": 1.1565873622894287,
      "weighted_auxiliary_loss": 0.0005241564416792244,
      "weighted_relation_loss": 0.0005241564416792244
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.825070159027128,
      "flow_matching_loss": 0.033316869474947455,
      "grad_norm": 0.12726327776908875,
      "learning_rate": 6.115444695692474e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005499839753611013,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.67368659973145,
      "state_query_norm": 1400.4337158203125,
      "state_relation_loss": 0.0005499839753611013,
      "step": 13710,
      "student_distance_mean": 1908.850927734375,
      "teacher_distance_mean": 1.1653329968452453,
      "weighted_auxiliary_loss": 0.0005499839753611013,
      "weighted_relation_loss": 0.0005499839753611013
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.83442469597755,
      "flow_matching_loss": 0.032629146985709664,
      "grad_norm": 0.139921635389328,
      "learning_rate": 6.110071355002779e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005756566708441824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.122402572631835,
      "state_query_norm": 1377.2815551757812,
      "state_relation_loss": 0.0005756566708441824,
      "step": 13720,
      "student_distance_mean": 1877.8743896484375,
      "teacher_distance_mean": 1.1451637983322143,
      "weighted_auxiliary_loss": 0.0005756566708441824,
      "weighted_relation_loss": 0.0005756566708441824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.84377923292797,
      "flow_matching_loss": 0.0327863996848464,
      "grad_norm": 0.14799706637859344,
      "learning_rate": 6.104696665471714e-05,
      "loss": 0.0333,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004924784268951043,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.72327575683594,
      "state_query_norm": 1365.1491577148438,
      "state_relation_loss": 0.0004924784268951043,
      "step": 13730,
      "student_distance_mean": 1860.0655517578125,
      "teacher_distance_mean": 1.1345826268196106,
      "weighted_auxiliary_loss": 0.0004924784268951043,
      "weighted_relation_loss": 0.0004924784268951043
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.853133769878392,
      "flow_matching_loss": 0.03408075235784054,
      "grad_norm": 0.10738544166088104,
      "learning_rate": 6.099320633630036e-05,
      "loss": 0.0347,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005748536088503898,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.34475898742676,
      "state_query_norm": 1390.9739501953125,
      "state_relation_loss": 0.0005748536088503898,
      "step": 13740,
      "student_distance_mean": 1891.5529663085938,
      "teacher_distance_mean": 1.1558114290237427,
      "weighted_auxiliary_loss": 0.0005748536088503898,
      "weighted_relation_loss": 0.0005748536088503898
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.862488306828812,
      "flow_matching_loss": 0.03302907347679138,
      "grad_norm": 0.09742942452430725,
      "learning_rate": 6.093943266010128e-05,
      "loss": 0.0336,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005270468624075875,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.289455032348634,
      "state_query_norm": 1388.799462890625,
      "state_relation_loss": 0.0005270468624075875,
      "step": 13750,
      "student_distance_mean": 1890.1619506835937,
      "teacher_distance_mean": 1.1477652549743653,
      "weighted_auxiliary_loss": 0.0005270468624075875,
      "weighted_relation_loss": 0.0005270468624075875
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.871842843779232,
      "flow_matching_loss": 0.03243463337421417,
      "grad_norm": 0.13705430924892426,
      "learning_rate": 6.088564569146e-05,
      "loss": 0.033,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005434807681012899,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.43415412902832,
      "state_query_norm": 1394.0090454101562,
      "state_relation_loss": 0.0005434807681012899,
      "step": 13760,
      "student_distance_mean": 1896.1379272460938,
      "teacher_distance_mean": 1.1489315152168273,
      "weighted_auxiliary_loss": 0.0005434807681012899,
      "weighted_relation_loss": 0.0005434807681012899
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.881197380729654,
      "flow_matching_loss": 0.033368743397295474,
      "grad_norm": 0.12682446837425232,
      "learning_rate": 6.083184549573274e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005249623121926561,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.78755569458008,
      "state_query_norm": 1402.0190063476562,
      "state_relation_loss": 0.0005249623121926561,
      "step": 13770,
      "student_distance_mean": 1912.8386474609374,
      "teacher_distance_mean": 1.1574182391166687,
      "weighted_auxiliary_loss": 0.0005249623121926561,
      "weighted_relation_loss": 0.0005249623121926561
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.890551917680074,
      "flow_matching_loss": 0.032620908506214616,
      "grad_norm": 0.11133243143558502,
      "learning_rate": 6.077803213829184e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005464928777655587,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.26187744140625,
      "state_query_norm": 1384.9140869140624,
      "state_relation_loss": 0.0005464928777655587,
      "step": 13780,
      "student_distance_mean": 1887.4204956054687,
      "teacher_distance_mean": 1.1460216522216797,
      "weighted_auxiliary_loss": 0.0005464928777655587,
      "weighted_relation_loss": 0.0005464928777655587
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.899906454630496,
      "flow_matching_loss": 0.032303431257605555,
      "grad_norm": 0.11927957832813263,
      "learning_rate": 6.0724205684525594e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005494961951626465,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.395642852783205,
      "state_query_norm": 1397.9416137695312,
      "state_relation_loss": 0.0005494961951626465,
      "step": 13790,
      "student_distance_mean": 1898.0562377929687,
      "teacher_distance_mean": 1.1529868721961976,
      "weighted_auxiliary_loss": 0.0005494961951626465,
      "weighted_relation_loss": 0.0005494961951626465
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.909260991580917,
      "flow_matching_loss": 0.030052952468395233,
      "grad_norm": 0.10715391486883163,
      "learning_rate": 6.0670366199838215e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005757268809247762,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.0746509552002,
      "state_query_norm": 1409.3701049804688,
      "state_relation_loss": 0.0005757268809247762,
      "step": 13800,
      "student_distance_mean": 1929.0399536132813,
      "teacher_distance_mean": 1.1692697286605835,
      "weighted_auxiliary_loss": 0.0005757268809247762,
      "weighted_relation_loss": 0.0005757268809247762
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.918615528531337,
      "flow_matching_loss": 0.03468919601291418,
      "grad_norm": 0.12628395855426788,
      "learning_rate": 6.061651374964974e-05,
      "loss": 0.0352,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000540907189133577,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.478244400024415,
      "state_query_norm": 1396.8562377929688,
      "state_relation_loss": 0.000540907189133577,
      "step": 13810,
      "student_distance_mean": 1900.2897338867188,
      "teacher_distance_mean": 1.1557516932487488,
      "weighted_auxiliary_loss": 0.000540907189133577,
      "weighted_relation_loss": 0.000540907189133577
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.927970065481759,
      "flow_matching_loss": 0.03213654868304729,
      "grad_norm": 0.13456816971302032,
      "learning_rate": 6.0562648399396004e-05,
      "loss": 0.0327,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005503784777829424,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.49231185913086,
      "state_query_norm": 1395.8599487304687,
      "state_relation_loss": 0.0005503784777829424,
      "step": 13820,
      "student_distance_mean": 1900.1598754882812,
      "teacher_distance_mean": 1.154983401298523,
      "weighted_auxiliary_loss": 0.0005503784777829424,
      "weighted_relation_loss": 0.0005503784777829424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.937324602432179,
      "flow_matching_loss": 0.03403727076947689,
      "grad_norm": 0.11910425126552582,
      "learning_rate": 6.050877021452845e-05,
      "loss": 0.0345,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004970022768247872,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.6588623046875,
      "state_query_norm": 1404.4010375976563,
      "state_relation_loss": 0.0004970022768247872,
      "step": 13830,
      "student_distance_mean": 1911.71962890625,
      "teacher_distance_mean": 1.16303608417511,
      "weighted_auxiliary_loss": 0.0004970022768247872,
      "weighted_relation_loss": 0.0004970022768247872
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.946679139382601,
      "flow_matching_loss": 0.032420163974165916,
      "grad_norm": 0.1493932008743286,
      "learning_rate": 6.04548792605142e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005288797343382612,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.27728385925293,
      "state_query_norm": 1384.220458984375,
      "state_relation_loss": 0.0005288797343382612,
      "step": 13840,
      "student_distance_mean": 1889.9333984375,
      "teacher_distance_mean": 1.1488384842872619,
      "weighted_auxiliary_loss": 0.0005288797343382612,
      "weighted_relation_loss": 0.0005288797343382612
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.956033676333021,
      "flow_matching_loss": 0.03300311043858528,
      "grad_norm": 0.10837456583976746,
      "learning_rate": 6.04009756028358e-05,
      "loss": 0.0336,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005633495195070282,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.77805290222168,
      "state_query_norm": 1400.705615234375,
      "state_relation_loss": 0.0005633495195070282,
      "step": 13850,
      "student_distance_mean": 1914.2687866210938,
      "teacher_distance_mean": 1.1566583156585692,
      "weighted_auxiliary_loss": 0.0005633495195070282,
      "weighted_relation_loss": 0.0005633495195070282
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.965388213283443,
      "flow_matching_loss": 0.03487980850040913,
      "grad_norm": 0.09746478497982025,
      "learning_rate": 6.0347059306991294e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005484352237544954,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.9784481048584,
      "state_query_norm": 1422.9750366210938,
      "state_relation_loss": 0.0005484352237544954,
      "step": 13860,
      "student_distance_mean": 1932.6275512695313,
      "teacher_distance_mean": 1.164771056175232,
      "weighted_auxiliary_loss": 0.0005484352237544954,
      "weighted_relation_loss": 0.0005484352237544954
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.974742750233863,
      "flow_matching_loss": 0.03218608871102333,
      "grad_norm": 0.10146404802799225,
      "learning_rate": 6.029313043849407e-05,
      "loss": 0.0327,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005435119586763903,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.94282684326172,
      "state_query_norm": 1412.45615234375,
      "state_relation_loss": 0.0005435119586763903,
      "step": 13870,
      "student_distance_mean": 1922.4589965820312,
      "teacher_distance_mean": 1.1621479988098145,
      "weighted_auxiliary_loss": 0.0005435119586763903,
      "weighted_relation_loss": 0.0005435119586763903
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.984097287184284,
      "flow_matching_loss": 0.03091550339013338,
      "grad_norm": 0.10588055104017258,
      "learning_rate": 6.023918906287279e-05,
      "loss": 0.0315,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005511105642654002,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.65207366943359,
      "state_query_norm": 1406.4809326171876,
      "state_relation_loss": 0.0005511105642654002,
      "step": 13880,
      "student_distance_mean": 1910.8892944335937,
      "teacher_distance_mean": 1.1554826855659486,
      "weighted_auxiliary_loss": 0.0005511105642654002,
      "weighted_relation_loss": 0.0005511105642654002
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 12.993451824134706,
      "flow_matching_loss": 0.03319801222532988,
      "grad_norm": 0.12661443650722504,
      "learning_rate": 6.01852352456713e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005238308105617762,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.598928451538086,
      "state_query_norm": 1396.5537109375,
      "state_relation_loss": 0.0005238308105617762,
      "step": 13890,
      "student_distance_mean": 1907.9316040039062,
      "teacher_distance_mean": 1.1497752904891967,
      "weighted_auxiliary_loss": 0.0005238308105617762,
      "weighted_relation_loss": 0.0005238308105617762
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.002806361085126,
      "flow_matching_loss": 0.03196770492941141,
      "grad_norm": 0.12356781214475632,
      "learning_rate": 6.013126905244858e-05,
      "loss": 0.0325,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0005482969281729311,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 34.779344177246095,
      "state_query_norm": 1410.66484375,
      "state_relation_loss": 0.0005482969281729311,
      "step": 13900,
      "student_distance_mean": 1918.4099731445312,
      "teacher_distance_mean": 1.1583380103111267,
      "weighted_auxiliary_loss": 0.0005482969281729311,
      "weighted_relation_loss": 0.0005482969281729311
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.012160898035548,
      "flow_matching_loss": 0.03204029221087694,
      "grad_norm": 0.1126258596777916,
      "learning_rate": 6.007729054877864e-05,
      "loss": 0.0326,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005147124786162749,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.561946105957034,
      "state_query_norm": 1396.570556640625,
      "state_relation_loss": 0.0005147124786162749,
      "step": 13910,
      "student_distance_mean": 1903.4619750976562,
      "teacher_distance_mean": 1.1480053901672362,
      "weighted_auxiliary_loss": 0.0005147124786162749,
      "weighted_relation_loss": 0.0005147124786162749
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.021515434985968,
      "flow_matching_loss": 0.033937524631619455,
      "grad_norm": 0.11193940043449402,
      "learning_rate": 6.0023299800250466e-05,
      "loss": 0.0345,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005348286009393633,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.619554901123045,
      "state_query_norm": 1400.1881225585937,
      "state_relation_loss": 0.0005348286009393633,
      "step": 13920,
      "student_distance_mean": 1908.9783325195312,
      "teacher_distance_mean": 1.1522584676742553,
      "weighted_auxiliary_loss": 0.0005348286009393633,
      "weighted_relation_loss": 0.0005348286009393633
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.030869971936388,
      "flow_matching_loss": 0.03202498182654381,
      "grad_norm": 0.13188159465789795,
      "learning_rate": 5.9969296872467894e-05,
      "loss": 0.0325,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000506153964670375,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.80047378540039,
      "state_query_norm": 1406.77890625,
      "state_relation_loss": 0.000506153964670375,
      "step": 13930,
      "student_distance_mean": 1918.4799438476562,
      "teacher_distance_mean": 1.1599281191825868,
      "weighted_auxiliary_loss": 0.000506153964670375,
      "weighted_relation_loss": 0.000506153964670375
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.04022450888681,
      "flow_matching_loss": 0.03169989548623562,
      "grad_norm": 0.13533416390419006,
      "learning_rate": 5.9915281831049585e-05,
      "loss": 0.0322,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000540474581066519,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.42340354919433,
      "state_query_norm": 1388.283544921875,
      "state_relation_loss": 0.000540474581066519,
      "step": 13940,
      "student_distance_mean": 1895.657373046875,
      "teacher_distance_mean": 1.1527531027793885,
      "weighted_auxiliary_loss": 0.000540474581066519,
      "weighted_relation_loss": 0.000540474581066519
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.04957904583723,
      "flow_matching_loss": 0.03301959615200758,
      "grad_norm": 0.1311134397983551,
      "learning_rate": 5.98612547416289e-05,
      "loss": 0.0335,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005210603238083422,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.40558204650879,
      "state_query_norm": 1393.9940795898438,
      "state_relation_loss": 0.0005210603238083422,
      "step": 13950,
      "student_distance_mean": 1898.0704223632813,
      "teacher_distance_mean": 1.154924499988556,
      "weighted_auxiliary_loss": 0.0005210603238083422,
      "weighted_relation_loss": 0.0005210603238083422
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.058933582787652,
      "flow_matching_loss": 0.03205749709159136,
      "grad_norm": 0.10869352519512177,
      "learning_rate": 5.9807215669853855e-05,
      "loss": 0.0326,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005320568103343249,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.36697196960449,
      "state_query_norm": 1386.8344482421876,
      "state_relation_loss": 0.0005320568103343249,
      "step": 13960,
      "student_distance_mean": 1891.4990966796875,
      "teacher_distance_mean": 1.1527235627174377,
      "weighted_auxiliary_loss": 0.0005320568103343249,
      "weighted_relation_loss": 0.0005320568103343249
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.068288119738073,
      "flow_matching_loss": 0.03216214794665575,
      "grad_norm": 0.11041998118162155,
      "learning_rate": 5.9753164681387e-05,
      "loss": 0.0327,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005293023306876421,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.6403076171875,
      "state_query_norm": 1403.4865356445312,
      "state_relation_loss": 0.0005293023306876421,
      "step": 13970,
      "student_distance_mean": 1908.7446655273438,
      "teacher_distance_mean": 1.1627517461776733,
      "weighted_auxiliary_loss": 0.0005293023306876421,
      "weighted_relation_loss": 0.0005293023306876421
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.077642656688495,
      "flow_matching_loss": 0.0318219643086195,
      "grad_norm": 0.1024637296795845,
      "learning_rate": 5.969910184190539e-05,
      "loss": 0.0324,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005501281120814383,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.596993637084964,
      "state_query_norm": 1396.4313598632812,
      "state_relation_loss": 0.0005501281120814383,
      "step": 13980,
      "student_distance_mean": 1903.9748657226562,
      "teacher_distance_mean": 1.1544382095336914,
      "weighted_auxiliary_loss": 0.0005501281120814383,
      "weighted_relation_loss": 0.0005501281120814383
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.086997193638915,
      "flow_matching_loss": 0.032567649334669116,
      "grad_norm": 0.1301603764295578,
      "learning_rate": 5.9645027217100477e-05,
      "loss": 0.0331,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005336785310646519,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.60844459533691,
      "state_query_norm": 1392.639111328125,
      "state_relation_loss": 0.0005336785310646519,
      "step": 13990,
      "student_distance_mean": 1903.4241088867188,
      "teacher_distance_mean": 1.1508394479751587,
      "weighted_auxiliary_loss": 0.0005336785310646519,
      "weighted_relation_loss": 0.0005336785310646519
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.096351730589335,
      "flow_matching_loss": 0.03199775442481041,
      "grad_norm": 0.10948044061660767,
      "learning_rate": 5.959094087267804e-05,
      "loss": 0.0325,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005459819454699755,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.22368965148926,
      "state_query_norm": 1382.9305908203125,
      "state_relation_loss": 0.0005459819454699755,
      "step": 14000,
      "student_distance_mean": 1887.7874267578125,
      "teacher_distance_mean": 1.1396316885948181,
      "weighted_auxiliary_loss": 0.0005459819454699755,
      "weighted_relation_loss": 0.0005459819454699755
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.105706267539757,
      "flow_matching_loss": 0.031147138960659505,
      "grad_norm": 0.1349301040172577,
      "learning_rate": 5.9536842874358065e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005412704660557211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.83172416687012,
      "state_query_norm": 1392.8545532226562,
      "state_relation_loss": 0.0005412704660557211,
      "step": 14010,
      "student_distance_mean": 1912.047998046875,
      "teacher_distance_mean": 1.157570207118988,
      "weighted_auxiliary_loss": 0.0005412704660557211,
      "weighted_relation_loss": 0.0005412704660557211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.115060804490177,
      "flow_matching_loss": 0.033687447942793367,
      "grad_norm": 0.12978443503379822,
      "learning_rate": 5.9482733287874734e-05,
      "loss": 0.0342,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005258930090349167,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.3708740234375,
      "state_query_norm": 1389.0108520507813,
      "state_relation_loss": 0.0005258930090349167,
      "step": 14020,
      "student_distance_mean": 1895.4716430664062,
      "teacher_distance_mean": 1.150964879989624,
      "weighted_auxiliary_loss": 0.0005258930090349167,
      "weighted_relation_loss": 0.0005258930090349167
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.1244153414406,
      "flow_matching_loss": 0.033756344206631185,
      "grad_norm": 0.11654628813266754,
      "learning_rate": 5.942861217897631e-05,
      "loss": 0.0343,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005339105206076056,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.69223899841309,
      "state_query_norm": 1402.0001342773437,
      "state_relation_loss": 0.0005339105206076056,
      "step": 14030,
      "student_distance_mean": 1911.8396484375,
      "teacher_distance_mean": 1.1570422410964967,
      "weighted_auxiliary_loss": 0.0005339105206076056,
      "weighted_relation_loss": 0.0005339105206076056
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.13376987839102,
      "flow_matching_loss": 0.03267286103218794,
      "grad_norm": 0.12466054409742355,
      "learning_rate": 5.9374479613425e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005153297912329436,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.48764877319336,
      "state_query_norm": 1397.716259765625,
      "state_relation_loss": 0.0005153297912329436,
      "step": 14040,
      "student_distance_mean": 1903.6175170898437,
      "teacher_distance_mean": 1.1479774117469788,
      "weighted_auxiliary_loss": 0.0005153297912329436,
      "weighted_relation_loss": 0.0005153297912329436
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.14312441534144,
      "flow_matching_loss": 0.033132814429700376,
      "grad_norm": 0.1432574987411499,
      "learning_rate": 5.932033565699704e-05,
      "loss": 0.0337,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005424101458629593,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.58504791259766,
      "state_query_norm": 1399.1640991210938,
      "state_relation_loss": 0.0005424101458629593,
      "step": 14050,
      "student_distance_mean": 1903.977099609375,
      "teacher_distance_mean": 1.1465811252593994,
      "weighted_auxiliary_loss": 0.0005424101458629593,
      "weighted_relation_loss": 0.0005424101458629593
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.152478952291862,
      "flow_matching_loss": 0.0326421232894063,
      "grad_norm": 0.13084429502487183,
      "learning_rate": 5.926618037548237e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005208574351854622,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.86601943969727,
      "state_query_norm": 1412.875537109375,
      "state_relation_loss": 0.0005208574351854622,
      "step": 14060,
      "student_distance_mean": 1923.1748413085938,
      "teacher_distance_mean": 1.1591957807540894,
      "weighted_auxiliary_loss": 0.0005208574351854622,
      "weighted_relation_loss": 0.0005208574351854622
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.161833489242282,
      "flow_matching_loss": 0.03295076489448547,
      "grad_norm": 0.13598084449768066,
      "learning_rate": 5.9212013834684824e-05,
      "loss": 0.0335,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005022460827603936,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.120734786987306,
      "state_query_norm": 1381.3417358398438,
      "state_relation_loss": 0.0005022460827603936,
      "step": 14070,
      "student_distance_mean": 1878.1395629882813,
      "teacher_distance_mean": 1.1314165115356445,
      "weighted_auxiliary_loss": 0.0005022460827603936,
      "weighted_relation_loss": 0.0005022460827603936
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.171188026192704,
      "flow_matching_loss": 0.032581894844770434,
      "grad_norm": 0.10035275667905807,
      "learning_rate": 5.9157836100421824e-05,
      "loss": 0.0331,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005267072090646252,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.47439689636231,
      "state_query_norm": 1399.4652465820313,
      "state_relation_loss": 0.0005267072090646252,
      "step": 14080,
      "student_distance_mean": 1900.589794921875,
      "teacher_distance_mean": 1.1471306204795837,
      "weighted_auxiliary_loss": 0.0005267072090646252,
      "weighted_relation_loss": 0.0005267072090646252
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.180542563143124,
      "flow_matching_loss": 0.034798615239560605,
      "grad_norm": 0.11190597712993622,
      "learning_rate": 5.910364723852444e-05,
      "loss": 0.0354,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005530361901037395,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.39845771789551,
      "state_query_norm": 1398.7236572265624,
      "state_relation_loss": 0.0005530361901037395,
      "step": 14090,
      "student_distance_mean": 1900.3258544921875,
      "teacher_distance_mean": 1.1516893029212951,
      "weighted_auxiliary_loss": 0.0005530361901037395,
      "weighted_relation_loss": 0.0005530361901037395
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.189897100093546,
      "flow_matching_loss": 0.032818616926670076,
      "grad_norm": 0.13087336719036102,
      "learning_rate": 5.904944731483723e-05,
      "loss": 0.0333,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005219846323598176,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.475888061523435,
      "state_query_norm": 1398.3803833007812,
      "state_relation_loss": 0.0005219846323598176,
      "step": 14100,
      "student_distance_mean": 1901.406494140625,
      "teacher_distance_mean": 1.1582246899604798,
      "weighted_auxiliary_loss": 0.0005219846323598176,
      "weighted_relation_loss": 0.0005219846323598176
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.199251637043966,
      "flow_matching_loss": 0.033322599157691,
      "grad_norm": 0.1390313059091568,
      "learning_rate": 5.899523639521825e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005492930911714211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.60477905273437,
      "state_query_norm": 1398.4520263671875,
      "state_relation_loss": 0.0005492930911714211,
      "step": 14110,
      "student_distance_mean": 1904.1680297851562,
      "teacher_distance_mean": 1.1600934863090515,
      "weighted_auxiliary_loss": 0.0005492930911714211,
      "weighted_relation_loss": 0.0005492930911714211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.208606173994387,
      "flow_matching_loss": 0.032368212938308716,
      "grad_norm": 0.10993418097496033,
      "learning_rate": 5.894101454553883e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004975287389243022,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.941839599609374,
      "state_query_norm": 1373.3646240234375,
      "state_relation_loss": 0.0004975287389243022,
      "step": 14120,
      "student_distance_mean": 1869.975244140625,
      "teacher_distance_mean": 1.1368232727050782,
      "weighted_auxiliary_loss": 0.0004975287389243022,
      "weighted_relation_loss": 0.0004975287389243022
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.217960710944809,
      "flow_matching_loss": 0.032526956498622896,
      "grad_norm": 0.1301192045211792,
      "learning_rate": 5.888678183168368e-05,
      "loss": 0.033,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004943695297697559,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.090568923950194,
      "state_query_norm": 1377.4078857421875,
      "state_relation_loss": 0.0004943695297697559,
      "step": 14130,
      "student_distance_mean": 1877.015380859375,
      "teacher_distance_mean": 1.1417164921760559,
      "weighted_auxiliary_loss": 0.0004943695297697559,
      "weighted_relation_loss": 0.0004943695297697559
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.227315247895229,
      "flow_matching_loss": 0.03176140990108252,
      "grad_norm": 0.09703110158443451,
      "learning_rate": 5.883253831955061e-05,
      "loss": 0.0323,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005286654166411608,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.49817886352539,
      "state_query_norm": 1387.6971923828125,
      "state_relation_loss": 0.0005286654166411608,
      "step": 14140,
      "student_distance_mean": 1898.0706909179687,
      "teacher_distance_mean": 1.1498955845832826,
      "weighted_auxiliary_loss": 0.0005286654166411608,
      "weighted_relation_loss": 0.0005286654166411608
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.23666978484565,
      "flow_matching_loss": 0.03261035885661841,
      "grad_norm": 0.10852611064910889,
      "learning_rate": 5.8778284075050625e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005400923982961103,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.105381393432616,
      "state_query_norm": 1415.5824584960938,
      "state_relation_loss": 0.0005400923982961103,
      "step": 14150,
      "student_distance_mean": 1931.8792846679687,
      "teacher_distance_mean": 1.1674872994422913,
      "weighted_auxiliary_loss": 0.0005400923982961103,
      "weighted_relation_loss": 0.0005400923982961103
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.246024321796071,
      "flow_matching_loss": 0.03157592322677374,
      "grad_norm": 0.10040384531021118,
      "learning_rate": 5.872401916410777e-05,
      "loss": 0.0321,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005212021264014765,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.75793800354004,
      "state_query_norm": 1364.0524291992188,
      "state_relation_loss": 0.0005212021264014765,
      "step": 14160,
      "student_distance_mean": 1860.1022094726563,
      "teacher_distance_mean": 1.1249784588813783,
      "weighted_auxiliary_loss": 0.0005212021264014765,
      "weighted_relation_loss": 0.0005212021264014765
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.255378858746493,
      "flow_matching_loss": 0.031663541495800016,
      "grad_norm": 0.13057145476341248,
      "learning_rate": 5.8669743652659014e-05,
      "loss": 0.0322,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005113383871503174,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.543461608886716,
      "state_query_norm": 1402.1962890625,
      "state_relation_loss": 0.0005113383871503174,
      "step": 14170,
      "student_distance_mean": 1906.1783081054687,
      "teacher_distance_mean": 1.1525553584098815,
      "weighted_auxiliary_loss": 0.0005113383871503174,
      "weighted_relation_loss": 0.0005113383871503174
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.264733395696913,
      "flow_matching_loss": 0.03234426006674766,
      "grad_norm": 0.10782869905233383,
      "learning_rate": 5.86154576066542e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000531794669223018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.66151885986328,
      "state_query_norm": 1406.1981323242187,
      "state_relation_loss": 0.000531794669223018,
      "step": 14180,
      "student_distance_mean": 1913.9346557617187,
      "teacher_distance_mean": 1.1583324790000915,
      "weighted_auxiliary_loss": 0.000531794669223018,
      "weighted_relation_loss": 0.000531794669223018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.274087932647333,
      "flow_matching_loss": 0.030342999659478665,
      "grad_norm": 0.09788646548986435,
      "learning_rate": 5.856116109205602e-05,
      "loss": 0.0309,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005447408475447446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.74055938720703,
      "state_query_norm": 1400.898193359375,
      "state_relation_loss": 0.0005447408475447446,
      "step": 14190,
      "student_distance_mean": 1914.2500244140624,
      "teacher_distance_mean": 1.155655312538147,
      "weighted_auxiliary_loss": 0.0005447408475447446,
      "weighted_relation_loss": 0.0005447408475447446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.283442469597755,
      "flow_matching_loss": 0.03061546627432108,
      "grad_norm": 0.11079820990562439,
      "learning_rate": 5.850685417483983e-05,
      "loss": 0.0312,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005499797640368343,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.137868118286136,
      "state_query_norm": 1419.4030029296875,
      "state_relation_loss": 0.0005499797640368343,
      "step": 14200,
      "student_distance_mean": 1933.620166015625,
      "teacher_distance_mean": 1.1668232917785644,
      "weighted_auxiliary_loss": 0.0005499797640368343,
      "weighted_relation_loss": 0.0005499797640368343
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.292797006548176,
      "flow_matching_loss": 0.030399496853351592,
      "grad_norm": 0.09615973383188248,
      "learning_rate": 5.8452536920993684e-05,
      "loss": 0.0309,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005267815606202931,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.37488555908203,
      "state_query_norm": 1396.1420532226562,
      "state_relation_loss": 0.0005267815606202931,
      "step": 14210,
      "student_distance_mean": 1898.251171875,
      "teacher_distance_mean": 1.1483625650405884,
      "weighted_auxiliary_loss": 0.0005267815606202931,
      "weighted_relation_loss": 0.0005267815606202931
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.302151543498598,
      "flow_matching_loss": 0.03128492441028356,
      "grad_norm": 0.09325330704450607,
      "learning_rate": 5.839820939651817e-05,
      "loss": 0.0318,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005092687817523256,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.65338401794433,
      "state_query_norm": 1405.8589965820313,
      "state_relation_loss": 0.0005092687817523256,
      "step": 14220,
      "student_distance_mean": 1912.5442138671874,
      "teacher_distance_mean": 1.1607335090637207,
      "weighted_auxiliary_loss": 0.0005092687817523256,
      "weighted_relation_loss": 0.0005092687817523256
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.311506080449018,
      "flow_matching_loss": 0.03117577116936445,
      "grad_norm": 0.10725940763950348,
      "learning_rate": 5.834387166742632e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005335791065590456,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01981964111328,
      "state_query_norm": 1412.7903442382812,
      "state_relation_loss": 0.0005335791065590456,
      "step": 14230,
      "student_distance_mean": 1925.6764404296875,
      "teacher_distance_mean": 1.1633991837501525,
      "weighted_auxiliary_loss": 0.0005335791065590456,
      "weighted_relation_loss": 0.0005335791065590456
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.320860617399438,
      "flow_matching_loss": 0.030017045326530933,
      "grad_norm": 0.09043452888727188,
      "learning_rate": 5.828952379974364e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005569119850406424,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.960152053833006,
      "state_query_norm": 1408.3194946289063,
      "state_relation_loss": 0.0005569119850406424,
      "step": 14240,
      "student_distance_mean": 1921.7130493164063,
      "teacher_distance_mean": 1.1605963587760926,
      "weighted_auxiliary_loss": 0.0005569119850406424,
      "weighted_relation_loss": 0.0005569119850406424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.33021515434986,
      "flow_matching_loss": 0.03232261408120394,
      "grad_norm": 0.09425558149814606,
      "learning_rate": 5.8235165859507864e-05,
      "loss": 0.0328,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005183558998396621,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.47771110534668,
      "state_query_norm": 1394.3366333007812,
      "state_relation_loss": 0.0005183558998396621,
      "step": 14250,
      "student_distance_mean": 1899.7880981445312,
      "teacher_distance_mean": 1.149073100090027,
      "weighted_auxiliary_loss": 0.0005183558998396621,
      "weighted_relation_loss": 0.0005183558998396621
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.33956969130028,
      "flow_matching_loss": 0.031140468455851077,
      "grad_norm": 0.08785107731819153,
      "learning_rate": 5.818079791276907e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005250211630482227,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.12560386657715,
      "state_query_norm": 1418.323779296875,
      "state_relation_loss": 0.0005250211630482227,
      "step": 14260,
      "student_distance_mean": 1933.7395874023437,
      "teacher_distance_mean": 1.1644944190979003,
      "weighted_auxiliary_loss": 0.0005250211630482227,
      "weighted_relation_loss": 0.0005250211630482227
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.348924228250702,
      "flow_matching_loss": 0.0318374790251255,
      "grad_norm": 0.09919789433479309,
      "learning_rate": 5.8126420025589415e-05,
      "loss": 0.0324,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005264393519610166,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.47562522888184,
      "state_query_norm": 1392.527978515625,
      "state_relation_loss": 0.0005264393519610166,
      "step": 14270,
      "student_distance_mean": 1899.1462768554688,
      "teacher_distance_mean": 1.1474181890487671,
      "weighted_auxiliary_loss": 0.0005264393519610166,
      "weighted_relation_loss": 0.0005264393519610166
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.358278765201122,
      "flow_matching_loss": 0.02882980667054653,
      "grad_norm": 0.11193954199552536,
      "learning_rate": 5.807203226404313e-05,
      "loss": 0.0294,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000551202252972871,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.5051082611084,
      "state_query_norm": 1388.83544921875,
      "state_relation_loss": 0.000551202252972871,
      "step": 14280,
      "student_distance_mean": 1895.8282104492187,
      "teacher_distance_mean": 1.1452180862426757,
      "weighted_auxiliary_loss": 0.000551202252972871,
      "weighted_relation_loss": 0.000551202252972871
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.367633302151544,
      "flow_matching_loss": 0.03221933227032423,
      "grad_norm": 0.13162216544151306,
      "learning_rate": 5.801763469421652e-05,
      "loss": 0.0327,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005282554047880694,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01985549926758,
      "state_query_norm": 1420.9359741210938,
      "state_relation_loss": 0.0005282554047880694,
      "step": 14290,
      "student_distance_mean": 1929.7979858398437,
      "teacher_distance_mean": 1.163319170475006,
      "weighted_auxiliary_loss": 0.0005282554047880694,
      "weighted_relation_loss": 0.0005282554047880694
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.376987839101965,
      "flow_matching_loss": 0.0316745188087225,
      "grad_norm": 0.12134785205125809,
      "learning_rate": 5.796322738220774e-05,
      "loss": 0.0322,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005051586631452665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.28419532775879,
      "state_query_norm": 1389.1819091796874,
      "state_relation_loss": 0.0005051586631452665,
      "step": 14300,
      "student_distance_mean": 1888.8304809570313,
      "teacher_distance_mean": 1.1394506812095642,
      "weighted_auxiliary_loss": 0.0005051586631452665,
      "weighted_relation_loss": 0.0005051586631452665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.386342376052385,
      "flow_matching_loss": 0.03168839644640684,
      "grad_norm": 0.11647326499223709,
      "learning_rate": 5.79088103941268e-05,
      "loss": 0.0322,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005206337926210835,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16462516784668,
      "state_query_norm": 1427.1555786132812,
      "state_relation_loss": 0.0005206337926210835,
      "step": 14310,
      "student_distance_mean": 1940.7050903320312,
      "teacher_distance_mean": 1.1710740804672242,
      "weighted_auxiliary_loss": 0.0005206337926210835,
      "weighted_relation_loss": 0.0005206337926210835
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.395696913002807,
      "flow_matching_loss": 0.03336600083857775,
      "grad_norm": 0.14571334421634674,
      "learning_rate": 5.785438379609549e-05,
      "loss": 0.0339,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004916496953228488,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.39441337585449,
      "state_query_norm": 1393.0530883789063,
      "state_relation_loss": 0.0004916496953228488,
      "step": 14320,
      "student_distance_mean": 1895.4891845703125,
      "teacher_distance_mean": 1.1439847826957703,
      "weighted_auxiliary_loss": 0.0004916496953228488,
      "weighted_relation_loss": 0.0004916496953228488
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.405051449953227,
      "flow_matching_loss": 0.03381791803985834,
      "grad_norm": 0.12420082092285156,
      "learning_rate": 5.779994765424724e-05,
      "loss": 0.0343,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005284321727231145,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.86567420959473,
      "state_query_norm": 1411.9907470703124,
      "state_relation_loss": 0.0005284321727231145,
      "step": 14330,
      "student_distance_mean": 1922.4715576171875,
      "teacher_distance_mean": 1.1654776692390443,
      "weighted_auxiliary_loss": 0.0005284321727231145,
      "weighted_relation_loss": 0.0005284321727231145
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.414405986903649,
      "flow_matching_loss": 0.03089578039944172,
      "grad_norm": 0.11325478553771973,
      "learning_rate": 5.7745502034727114e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000522173976060003,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.37833595275879,
      "state_query_norm": 1383.0357299804687,
      "state_relation_loss": 0.000522173976060003,
      "step": 14340,
      "student_distance_mean": 1891.5174560546875,
      "teacher_distance_mean": 1.1429972171783447,
      "weighted_auxiliary_loss": 0.000522173976060003,
      "weighted_relation_loss": 0.000522173976060003
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.42376052385407,
      "flow_matching_loss": 0.03246153499931097,
      "grad_norm": 0.12082149088382721,
      "learning_rate": 5.769104700369164e-05,
      "loss": 0.033,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000515218946384266,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.72316551208496,
      "state_query_norm": 1400.6119018554687,
      "state_relation_loss": 0.000515218946384266,
      "step": 14350,
      "student_distance_mean": 1913.3178100585938,
      "teacher_distance_mean": 1.159884798526764,
      "weighted_auxiliary_loss": 0.000515218946384266,
      "weighted_relation_loss": 0.000515218946384266
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.43311506080449,
      "flow_matching_loss": 0.03238554485142231,
      "grad_norm": 0.11497372388839722,
      "learning_rate": 5.7636582627308854e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000529269824619405,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.05562286376953,
      "state_query_norm": 1411.6459838867188,
      "state_relation_loss": 0.000529269824619405,
      "step": 14360,
      "student_distance_mean": 1931.0407348632812,
      "teacher_distance_mean": 1.1682556748390198,
      "weighted_auxiliary_loss": 0.000529269824619405,
      "weighted_relation_loss": 0.000529269824619405
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.442469597754911,
      "flow_matching_loss": 0.03197714164853096,
      "grad_norm": 0.1116294264793396,
      "learning_rate": 5.7582108971758095e-05,
      "loss": 0.0325,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005551019625272602,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.815896606445314,
      "state_query_norm": 1400.7028198242188,
      "state_relation_loss": 0.0005551019625272602,
      "step": 14370,
      "student_distance_mean": 1915.6691040039063,
      "teacher_distance_mean": 1.1566801190376281,
      "weighted_auxiliary_loss": 0.0005551019625272602,
      "weighted_relation_loss": 0.0005551019625272602
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.451824134705332,
      "flow_matching_loss": 0.03209725245833397,
      "grad_norm": 0.1267816722393036,
      "learning_rate": 5.752762610322999e-05,
      "loss": 0.0326,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000536058732541278,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.54374351501465,
      "state_query_norm": 1396.769189453125,
      "state_relation_loss": 0.000536058732541278,
      "step": 14380,
      "student_distance_mean": 1902.6752319335938,
      "teacher_distance_mean": 1.1469375133514403,
      "weighted_auxiliary_loss": 0.000536058732541278,
      "weighted_relation_loss": 0.000536058732541278
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.461178671655754,
      "flow_matching_loss": 0.029994426295161247,
      "grad_norm": 0.08943482488393784,
      "learning_rate": 5.747313408792636e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005017850431613624,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92125930786133,
      "state_query_norm": 1410.9958984375,
      "state_relation_loss": 0.0005017850431613624,
      "step": 14390,
      "student_distance_mean": 1922.722802734375,
      "teacher_distance_mean": 1.1603152394294738,
      "weighted_auxiliary_loss": 0.0005017850431613624,
      "weighted_relation_loss": 0.0005017850431613624
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.470533208606174,
      "flow_matching_loss": 0.03212872613221407,
      "grad_norm": 0.12920507788658142,
      "learning_rate": 5.741863299206014e-05,
      "loss": 0.0327,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005445174989290536,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.84836692810059,
      "state_query_norm": 1404.3213134765624,
      "state_relation_loss": 0.0005445174989290536,
      "step": 14400,
      "student_distance_mean": 1915.9831909179688,
      "teacher_distance_mean": 1.162662136554718,
      "weighted_auxiliary_loss": 0.0005445174989290536,
      "weighted_relation_loss": 0.0005445174989290536
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.479887745556596,
      "flow_matching_loss": 0.029915069602429865,
      "grad_norm": 0.11857248097658157,
      "learning_rate": 5.73641228818553e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005503486492671073,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.6362865447998,
      "state_query_norm": 1399.1230590820312,
      "state_relation_loss": 0.0005503486492671073,
      "step": 14410,
      "student_distance_mean": 1906.4553100585938,
      "teacher_distance_mean": 1.1505883574485778,
      "weighted_auxiliary_loss": 0.0005503486492671073,
      "weighted_relation_loss": 0.0005503486492671073
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.489242282507016,
      "flow_matching_loss": 0.031147191673517226,
      "grad_norm": 0.11263471841812134,
      "learning_rate": 5.730960382354676e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005442978377686814,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.93339157104492,
      "state_query_norm": 1405.7404174804688,
      "state_relation_loss": 0.0005442978377686814,
      "step": 14420,
      "student_distance_mean": 1920.8416015625,
      "teacher_distance_mean": 1.159424614906311,
      "weighted_auxiliary_loss": 0.0005442978377686814,
      "weighted_relation_loss": 0.0005442978377686814
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.498596819457436,
      "flow_matching_loss": 0.030030342750251293,
      "grad_norm": 0.09541060030460358,
      "learning_rate": 5.725507588338035e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005417415231931955,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.58741302490235,
      "state_query_norm": 1391.6698974609376,
      "state_relation_loss": 0.0005417415231931955,
      "step": 14430,
      "student_distance_mean": 1902.1404907226563,
      "teacher_distance_mean": 1.1471142172813416,
      "weighted_auxiliary_loss": 0.0005417415231931955,
      "weighted_relation_loss": 0.0005417415231931955
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.507951356407858,
      "flow_matching_loss": 0.029907610081136227,
      "grad_norm": 0.11345064640045166,
      "learning_rate": 5.7200539127612604e-05,
      "loss": 0.0304,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005373840220272541,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.8680233001709,
      "state_query_norm": 1395.3101440429687,
      "state_relation_loss": 0.0005373840220272541,
      "step": 14440,
      "student_distance_mean": 1913.7195434570312,
      "teacher_distance_mean": 1.1541860699653625,
      "weighted_auxiliary_loss": 0.0005373840220272541,
      "weighted_relation_loss": 0.0005373840220272541
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.517305893358278,
      "flow_matching_loss": 0.032709677889943126,
      "grad_norm": 0.11497396975755692,
      "learning_rate": 5.714599362251087e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005086762190330774,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.51672477722168,
      "state_query_norm": 1399.8864624023438,
      "state_relation_loss": 0.0005086762190330774,
      "step": 14450,
      "student_distance_mean": 1900.8067016601562,
      "teacher_distance_mean": 1.1459233641624451,
      "weighted_auxiliary_loss": 0.0005086762190330774,
      "weighted_relation_loss": 0.0005086762190330774
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.5266604303087,
      "flow_matching_loss": 0.03224496506154537,
      "grad_norm": 0.11250512301921844,
      "learning_rate": 5.709143943435307e-05,
      "loss": 0.0328,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005454430269310251,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92039794921875,
      "state_query_norm": 1406.780078125,
      "state_relation_loss": 0.0005454430269310251,
      "step": 14460,
      "student_distance_mean": 1920.4893920898437,
      "teacher_distance_mean": 1.164042627811432,
      "weighted_auxiliary_loss": 0.0005454430269310251,
      "weighted_relation_loss": 0.0005454430269310251
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.53601496725912,
      "flow_matching_loss": 0.03022490330040455,
      "grad_norm": 0.12264734506607056,
      "learning_rate": 5.7036876629427646e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005264568200800568,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.77876243591309,
      "state_query_norm": 1404.4021362304688,
      "state_relation_loss": 0.0005264568200800568,
      "step": 14470,
      "student_distance_mean": 1915.17431640625,
      "teacher_distance_mean": 1.1633962869644165,
      "weighted_auxiliary_loss": 0.0005264568200800568,
      "weighted_relation_loss": 0.0005264568200800568
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.545369504209543,
      "flow_matching_loss": 0.03008168060332537,
      "grad_norm": 0.11471951752901077,
      "learning_rate": 5.698230527403361e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005055473826359958,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.376399993896484,
      "state_query_norm": 1386.5828491210937,
      "state_relation_loss": 0.0005055473826359958,
      "step": 14480,
      "student_distance_mean": 1893.72490234375,
      "teacher_distance_mean": 1.1469022274017333,
      "weighted_auxiliary_loss": 0.0005055473826359958,
      "weighted_relation_loss": 0.0005055473826359958
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.554724041159963,
      "flow_matching_loss": 0.03240417409688234,
      "grad_norm": 0.12249372154474258,
      "learning_rate": 5.6927725434480274e-05,
      "loss": 0.0329,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005107820819830522,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.64059715270996,
      "state_query_norm": 1396.1142333984376,
      "state_relation_loss": 0.0005107820819830522,
      "step": 14490,
      "student_distance_mean": 1905.5545776367187,
      "teacher_distance_mean": 1.1533278226852417,
      "weighted_auxiliary_loss": 0.0005107820819830522,
      "weighted_relation_loss": 0.0005107820819830522
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.564078578110383,
      "flow_matching_loss": 0.03087939918041229,
      "grad_norm": 0.10958384722471237,
      "learning_rate": 5.687313717708728e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004919895087368787,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06252899169922,
      "state_query_norm": 1420.2677001953125,
      "state_relation_loss": 0.0004919895087368787,
      "step": 14500,
      "student_distance_mean": 1935.9966430664062,
      "teacher_distance_mean": 1.1635178565979003,
      "weighted_auxiliary_loss": 0.0004919895087368787,
      "weighted_relation_loss": 0.0004919895087368787
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.573433115060805,
      "flow_matching_loss": 0.03060981221497059,
      "grad_norm": 0.09325350075960159,
      "learning_rate": 5.681854056818453e-05,
      "loss": 0.0311,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005230049340752885,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21940956115723,
      "state_query_norm": 1412.8913452148438,
      "state_relation_loss": 0.0005230049340752885,
      "step": 14510,
      "student_distance_mean": 1936.1469604492188,
      "teacher_distance_mean": 1.164084267616272,
      "weighted_auxiliary_loss": 0.0005230049340752885,
      "weighted_relation_loss": 0.0005230049340752885
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.582787652011225,
      "flow_matching_loss": 0.031232811324298382,
      "grad_norm": 0.12278344482183456,
      "learning_rate": 5.676393567411205e-05,
      "loss": 0.0318,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005195653007831425,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.45541877746582,
      "state_query_norm": 1386.605322265625,
      "state_relation_loss": 0.0005195653007831425,
      "step": 14520,
      "student_distance_mean": 1897.347314453125,
      "teacher_distance_mean": 1.146454405784607,
      "weighted_auxiliary_loss": 0.0005195653007831425,
      "weighted_relation_loss": 0.0005195653007831425
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.592142188961647,
      "flow_matching_loss": 0.032660482451319695,
      "grad_norm": 0.11075996607542038,
      "learning_rate": 5.670932256121991e-05,
      "loss": 0.0332,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005278902943246066,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.881202697753906,
      "state_query_norm": 1409.5228271484375,
      "state_relation_loss": 0.0005278902943246066,
      "step": 14530,
      "student_distance_mean": 1921.4225830078126,
      "teacher_distance_mean": 1.1629879236221314,
      "weighted_auxiliary_loss": 0.0005278902943246066,
      "weighted_relation_loss": 0.0005278902943246066
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.601496725912067,
      "flow_matching_loss": 0.030856646411120892,
      "grad_norm": 0.09072528779506683,
      "learning_rate": 5.6654701295868216e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005433033162262291,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.27006454467774,
      "state_query_norm": 1381.8222778320312,
      "state_relation_loss": 0.0005433033162262291,
      "step": 14540,
      "student_distance_mean": 1887.60166015625,
      "teacher_distance_mean": 1.1412256002426147,
      "weighted_auxiliary_loss": 0.0005433033162262291,
      "weighted_relation_loss": 0.0005433033162262291
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.610851262862488,
      "flow_matching_loss": 0.03071105219423771,
      "grad_norm": 0.12178077548742294,
      "learning_rate": 5.660007194442697e-05,
      "loss": 0.0312,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047936570190358906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.84416618347168,
      "state_query_norm": 1413.2736572265626,
      "state_relation_loss": 0.00047936570190358906,
      "step": 14550,
      "student_distance_mean": 1924.073779296875,
      "teacher_distance_mean": 1.1633808135986328,
      "weighted_auxiliary_loss": 0.00047936570190358906,
      "weighted_relation_loss": 0.00047936570190358906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.62020579981291,
      "flow_matching_loss": 0.03166011776775122,
      "grad_norm": 0.09954968094825745,
      "learning_rate": 5.6545434573275946e-05,
      "loss": 0.0322,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000509094342123717,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.41736335754395,
      "state_query_norm": 1392.8340209960938,
      "state_relation_loss": 0.000509094342123717,
      "step": 14560,
      "student_distance_mean": 1897.1435180664062,
      "teacher_distance_mean": 1.150464081764221,
      "weighted_auxiliary_loss": 0.000509094342123717,
      "weighted_relation_loss": 0.000509094342123717
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.62956033676333,
      "flow_matching_loss": 0.031465863808989525,
      "grad_norm": 0.12240505963563919,
      "learning_rate": 5.649078924880472e-05,
      "loss": 0.032,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004895301477517933,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.32405662536621,
      "state_query_norm": 1388.8585815429688,
      "state_relation_loss": 0.0004895301477517933,
      "step": 14570,
      "student_distance_mean": 1895.89677734375,
      "teacher_distance_mean": 1.1496530532836915,
      "weighted_auxiliary_loss": 0.0004895301477517933,
      "weighted_relation_loss": 0.0004895301477517933
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.638914873713752,
      "flow_matching_loss": 0.02879276517778635,
      "grad_norm": 0.09029863029718399,
      "learning_rate": 5.643613603741252e-05,
      "loss": 0.0293,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005520884820725769,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75345115661621,
      "state_query_norm": 1408.3099609375,
      "state_relation_loss": 0.0005520884820725769,
      "step": 14580,
      "student_distance_mean": 1916.3349365234376,
      "teacher_distance_mean": 1.1612351655960083,
      "weighted_auxiliary_loss": 0.0005520884820725769,
      "weighted_relation_loss": 0.0005520884820725769
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.648269410664172,
      "flow_matching_loss": 0.03129908349364996,
      "grad_norm": 0.08803227543830872,
      "learning_rate": 5.6381475005508154e-05,
      "loss": 0.0318,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005117610940942541,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.42708396911621,
      "state_query_norm": 1383.0629028320313,
      "state_relation_loss": 0.0005117610940942541,
      "step": 14590,
      "student_distance_mean": 1892.1171875,
      "teacher_distance_mean": 1.1444736003875733,
      "weighted_auxiliary_loss": 0.0005117610940942541,
      "weighted_relation_loss": 0.0005117610940942541
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.657623947614592,
      "flow_matching_loss": 0.03090284429490566,
      "grad_norm": 0.10260441154241562,
      "learning_rate": 5.632680621950992e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005231933988397941,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.25823440551758,
      "state_query_norm": 1386.7938842773438,
      "state_relation_loss": 0.0005231933988397941,
      "step": 14600,
      "student_distance_mean": 1885.76923828125,
      "teacher_distance_mean": 1.1454500555992126,
      "weighted_auxiliary_loss": 0.0005231933988397941,
      "weighted_relation_loss": 0.0005231933988397941
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.666978484565014,
      "flow_matching_loss": 0.03074359055608511,
      "grad_norm": 0.1051361933350563,
      "learning_rate": 5.627212974584555e-05,
      "loss": 0.0313,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005164250556845218,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75860404968262,
      "state_query_norm": 1404.3410034179688,
      "state_relation_loss": 0.0005164250556845218,
      "step": 14610,
      "student_distance_mean": 1913.80673828125,
      "teacher_distance_mean": 1.156864094734192,
      "weighted_auxiliary_loss": 0.0005164250556845218,
      "weighted_relation_loss": 0.0005164250556845218
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.676333021515434,
      "flow_matching_loss": 0.030123813077807426,
      "grad_norm": 0.13298830389976501,
      "learning_rate": 5.62174456509521e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005206044064834714,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.11765365600586,
      "state_query_norm": 1371.9775634765624,
      "state_relation_loss": 0.0005206044064834714,
      "step": 14620,
      "student_distance_mean": 1874.6308227539062,
      "teacher_distance_mean": 1.132139790058136,
      "weighted_auxiliary_loss": 0.0005206044064834714,
      "weighted_relation_loss": 0.0005206044064834714
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.685687558465856,
      "flow_matching_loss": 0.030912642367184162,
      "grad_norm": 0.09747862815856934,
      "learning_rate": 5.616275400127594e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005199534789426252,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.89708137512207,
      "state_query_norm": 1402.2614868164062,
      "state_relation_loss": 0.0005199534789426252,
      "step": 14630,
      "student_distance_mean": 1919.1753540039062,
      "teacher_distance_mean": 1.1585686326026916,
      "weighted_auxiliary_loss": 0.0005199534789426252,
      "weighted_relation_loss": 0.0005199534789426252
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.695042095416277,
      "flow_matching_loss": 0.030387284606695174,
      "grad_norm": 0.10074277967214584,
      "learning_rate": 5.610805486327254e-05,
      "loss": 0.0309,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005265351413981989,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07393836975098,
      "state_query_norm": 1414.4754028320312,
      "state_relation_loss": 0.0005265351413981989,
      "step": 14640,
      "student_distance_mean": 1930.337939453125,
      "teacher_distance_mean": 1.156795620918274,
      "weighted_auxiliary_loss": 0.0005265351413981989,
      "weighted_relation_loss": 0.0005265351413981989
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.704396632366699,
      "flow_matching_loss": 0.03290506694465876,
      "grad_norm": 0.10317864269018173,
      "learning_rate": 5.6053348303406536e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005100184178445488,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.4894905090332,
      "state_query_norm": 1390.505712890625,
      "state_relation_loss": 0.0005100184178445488,
      "step": 14650,
      "student_distance_mean": 1899.3779174804688,
      "teacher_distance_mean": 1.1405957341194153,
      "weighted_auxiliary_loss": 0.0005100184178445488,
      "weighted_relation_loss": 0.0005100184178445488
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.713751169317119,
      "flow_matching_loss": 0.03209081590175629,
      "grad_norm": 0.08964700996875763,
      "learning_rate": 5.599863438815156e-05,
      "loss": 0.0326,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005110982077894732,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.95045967102051,
      "state_query_norm": 1376.3241088867187,
      "state_relation_loss": 0.0005110982077894732,
      "step": 14660,
      "student_distance_mean": 1871.7492431640626,
      "teacher_distance_mean": 1.1243355512619018,
      "weighted_auxiliary_loss": 0.0005110982077894732,
      "weighted_relation_loss": 0.0005110982077894732
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.723105706267539,
      "flow_matching_loss": 0.030785508640110492,
      "grad_norm": 0.08263196796178818,
      "learning_rate": 5.594391318399017e-05,
      "loss": 0.0313,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005112896411446855,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.638196563720705,
      "state_query_norm": 1408.1723754882812,
      "state_relation_loss": 0.0005112896411446855,
      "step": 14670,
      "student_distance_mean": 1913.9364990234376,
      "teacher_distance_mean": 1.154208493232727,
      "weighted_auxiliary_loss": 0.0005112896411446855,
      "weighted_relation_loss": 0.0005112896411446855
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.732460243217961,
      "flow_matching_loss": 0.03161311633884907,
      "grad_norm": 0.11108902841806412,
      "learning_rate": 5.588918475741377e-05,
      "loss": 0.0321,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005217165715293958,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14051513671875,
      "state_query_norm": 1408.2748291015625,
      "state_relation_loss": 0.0005217165715293958,
      "step": 14680,
      "student_distance_mean": 1930.6342895507812,
      "teacher_distance_mean": 1.1641688466072082,
      "weighted_auxiliary_loss": 0.0005217165715293958,
      "weighted_relation_loss": 0.0005217165715293958
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.741814780168381,
      "flow_matching_loss": 0.03004490491002798,
      "grad_norm": 0.15819260478019714,
      "learning_rate": 5.5834449174922586e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005189412331674248,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.63135185241699,
      "state_query_norm": 1395.1336303710937,
      "state_relation_loss": 0.0005189412331674248,
      "step": 14690,
      "student_distance_mean": 1903.2725708007813,
      "teacher_distance_mean": 1.1483052730560304,
      "weighted_auxiliary_loss": 0.0005189412331674248,
      "weighted_relation_loss": 0.0005189412331674248
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.751169317118803,
      "flow_matching_loss": 0.03034581393003464,
      "grad_norm": 0.12155045568943024,
      "learning_rate": 5.57797065030255e-05,
      "loss": 0.0309,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000515188230201602,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.71951141357422,
      "state_query_norm": 1401.7069702148438,
      "state_relation_loss": 0.000515188230201602,
      "step": 14700,
      "student_distance_mean": 1914.70166015625,
      "teacher_distance_mean": 1.1524893164634704,
      "weighted_auxiliary_loss": 0.000515188230201602,
      "weighted_relation_loss": 0.000515188230201602
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.760523854069223,
      "flow_matching_loss": 0.03137691486626863,
      "grad_norm": 0.1277357041835785,
      "learning_rate": 5.5724956808240016e-05,
      "loss": 0.0319,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005100453301565722,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.90410461425781,
      "state_query_norm": 1399.24912109375,
      "state_relation_loss": 0.0005100453301565722,
      "step": 14710,
      "student_distance_mean": 1916.3939575195313,
      "teacher_distance_mean": 1.1563261866569519,
      "weighted_auxiliary_loss": 0.0005100453301565722,
      "weighted_relation_loss": 0.0005100453301565722
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.769878391019645,
      "flow_matching_loss": 0.030512894876301287,
      "grad_norm": 0.11029785871505737,
      "learning_rate": 5.567020015709219e-05,
      "loss": 0.031,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005347769270883873,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.86905174255371,
      "state_query_norm": 1395.693115234375,
      "state_relation_loss": 0.0005347769270883873,
      "step": 14720,
      "student_distance_mean": 1914.631396484375,
      "teacher_distance_mean": 1.1580737471580504,
      "weighted_auxiliary_loss": 0.0005347769270883873,
      "weighted_relation_loss": 0.0005347769270883873
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.779232927970066,
      "flow_matching_loss": 0.02996850199997425,
      "grad_norm": 0.11653972417116165,
      "learning_rate": 5.561543661611649e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005281735706375912,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.72723121643067,
      "state_query_norm": 1395.7696044921875,
      "state_relation_loss": 0.0005281735706375912,
      "step": 14730,
      "student_distance_mean": 1910.577685546875,
      "teacher_distance_mean": 1.1523231148719788,
      "weighted_auxiliary_loss": 0.0005281735706375912,
      "weighted_relation_loss": 0.0005281735706375912
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.788587464920486,
      "flow_matching_loss": 0.03065912760794163,
      "grad_norm": 0.11292613297700882,
      "learning_rate": 5.556066625185583e-05,
      "loss": 0.0312,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005263657891191542,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.348799514770505,
      "state_query_norm": 1394.4364868164062,
      "state_relation_loss": 0.0005263657891191542,
      "step": 14740,
      "student_distance_mean": 1892.28203125,
      "teacher_distance_mean": 1.1418089747428894,
      "weighted_auxiliary_loss": 0.0005263657891191542,
      "weighted_relation_loss": 0.0005263657891191542
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.797942001870908,
      "flow_matching_loss": 0.03180690836161375,
      "grad_norm": 0.11873485893011093,
      "learning_rate": 5.550588913086131e-05,
      "loss": 0.0324,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000560399709502235,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.19277801513672,
      "state_query_norm": 1415.5801635742187,
      "state_relation_loss": 0.000560399709502235,
      "step": 14750,
      "student_distance_mean": 1935.1149780273438,
      "teacher_distance_mean": 1.1667075157165527,
      "weighted_auxiliary_loss": 0.000560399709502235,
      "weighted_relation_loss": 0.000560399709502235
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.807296538821328,
      "flow_matching_loss": 0.029439136572182178,
      "grad_norm": 0.1074109748005867,
      "learning_rate": 5.545110531969234e-05,
      "loss": 0.03,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005193701683310792,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14898414611817,
      "state_query_norm": 1419.71875,
      "state_relation_loss": 0.0005193701683310792,
      "step": 14760,
      "student_distance_mean": 1933.9716552734376,
      "teacher_distance_mean": 1.1661041498184204,
      "weighted_auxiliary_loss": 0.0005193701683310792,
      "weighted_relation_loss": 0.0005193701683310792
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.81665107577175,
      "flow_matching_loss": 0.03114200718700886,
      "grad_norm": 0.09183630347251892,
      "learning_rate": 5.539631488491641e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005419359600637108,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.05319938659668,
      "state_query_norm": 1411.6538208007812,
      "state_relation_loss": 0.0005419359600637108,
      "step": 14770,
      "student_distance_mean": 1929.2259033203125,
      "teacher_distance_mean": 1.1568189978599548,
      "weighted_auxiliary_loss": 0.0005419359600637108,
      "weighted_relation_loss": 0.0005419359600637108
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.82600561272217,
      "flow_matching_loss": 0.030640014819800852,
      "grad_norm": 0.09835146367549896,
      "learning_rate": 5.534151789310904e-05,
      "loss": 0.0312,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005294702539686113,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.725909042358396,
      "state_query_norm": 1402.0327758789062,
      "state_relation_loss": 0.0005294702539686113,
      "step": 14780,
      "student_distance_mean": 1913.2979614257813,
      "teacher_distance_mean": 1.1466465830802917,
      "weighted_auxiliary_loss": 0.0005294702539686113,
      "weighted_relation_loss": 0.0005294702539686113
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.83536014967259,
      "flow_matching_loss": 0.029316445998847484,
      "grad_norm": 0.09496776759624481,
      "learning_rate": 5.528671441085376e-05,
      "loss": 0.0298,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004975391755579039,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.70524673461914,
      "state_query_norm": 1403.4667846679688,
      "state_relation_loss": 0.0004975391755579039,
      "step": 14790,
      "student_distance_mean": 1912.7277709960938,
      "teacher_distance_mean": 1.1462891697883606,
      "weighted_auxiliary_loss": 0.0004975391755579039,
      "weighted_relation_loss": 0.0004975391755579039
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.844714686623012,
      "flow_matching_loss": 0.03126263841986656,
      "grad_norm": 0.09895890951156616,
      "learning_rate": 5.523190450474197e-05,
      "loss": 0.0318,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000526127201737836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.73093147277832,
      "state_query_norm": 1412.412451171875,
      "state_relation_loss": 0.000526127201737836,
      "step": 14800,
      "student_distance_mean": 1915.2481201171875,
      "teacher_distance_mean": 1.1473785042762756,
      "weighted_auxiliary_loss": 0.000526127201737836,
      "weighted_relation_loss": 0.000526127201737836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.854069223573433,
      "flow_matching_loss": 0.034817043133080004,
      "grad_norm": 0.10791011899709702,
      "learning_rate": 5.5177088241372865e-05,
      "loss": 0.0353,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004966050270013511,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.14770584106445,
      "state_query_norm": 1376.7625244140625,
      "state_relation_loss": 0.0004966050270013511,
      "step": 14810,
      "student_distance_mean": 1879.4143920898437,
      "teacher_distance_mean": 1.1306310653686524,
      "weighted_auxiliary_loss": 0.0004966050270013511,
      "weighted_relation_loss": 0.0004966050270013511
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.863423760523855,
      "flow_matching_loss": 0.029582327604293822,
      "grad_norm": 0.09139677882194519,
      "learning_rate": 5.512226568735338e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005368595477193594,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.80619735717774,
      "state_query_norm": 1414.5451416015626,
      "state_relation_loss": 0.0005368595477193594,
      "step": 14820,
      "student_distance_mean": 1921.7036987304687,
      "teacher_distance_mean": 1.1547073245048523,
      "weighted_auxiliary_loss": 0.0005368595477193594,
      "weighted_relation_loss": 0.0005368595477193594
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.872778297474275,
      "flow_matching_loss": 0.031093208678066732,
      "grad_norm": 0.1238292008638382,
      "learning_rate": 5.506743690929809e-05,
      "loss": 0.0316,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005215523298829794,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.71064720153809,
      "state_query_norm": 1404.572802734375,
      "state_relation_loss": 0.0005215523298829794,
      "step": 14830,
      "student_distance_mean": 1912.945703125,
      "teacher_distance_mean": 1.1507321357727052,
      "weighted_auxiliary_loss": 0.0005215523298829794,
      "weighted_relation_loss": 0.0005215523298829794
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.882132834424697,
      "flow_matching_loss": 0.030847899429500104,
      "grad_norm": 0.09575125575065613,
      "learning_rate": 5.501260197382912e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005150208191480488,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.505191040039065,
      "state_query_norm": 1396.949267578125,
      "state_relation_loss": 0.0005150208191480488,
      "step": 14840,
      "student_distance_mean": 1902.9080688476563,
      "teacher_distance_mean": 1.1426801681518555,
      "weighted_auxiliary_loss": 0.0005150208191480488,
      "weighted_relation_loss": 0.0005150208191480488
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.891487371375117,
      "flow_matching_loss": 0.03219699114561081,
      "grad_norm": 0.09864028543233871,
      "learning_rate": 5.49577609475761e-05,
      "loss": 0.0327,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005325388017809018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.907666778564455,
      "state_query_norm": 1415.7823364257813,
      "state_relation_loss": 0.0005325388017809018,
      "step": 14850,
      "student_distance_mean": 1925.1624755859375,
      "teacher_distance_mean": 1.1543274998664856,
      "weighted_auxiliary_loss": 0.0005325388017809018,
      "weighted_relation_loss": 0.0005325388017809018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.900841908325537,
      "flow_matching_loss": 0.03183191511780024,
      "grad_norm": 0.1196773573756218,
      "learning_rate": 5.490291389717603e-05,
      "loss": 0.0324,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005191097501665354,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14190483093262,
      "state_query_norm": 1415.9607177734374,
      "state_relation_loss": 0.0005191097501665354,
      "step": 14860,
      "student_distance_mean": 1933.2081298828125,
      "teacher_distance_mean": 1.1566410779953002,
      "weighted_auxiliary_loss": 0.0005191097501665354,
      "weighted_relation_loss": 0.0005191097501665354
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.91019644527596,
      "flow_matching_loss": 0.03295112494379282,
      "grad_norm": 0.1342335045337677,
      "learning_rate": 5.484806088927329e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004939343692967668,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.442827987670896,
      "state_query_norm": 1422.0317504882812,
      "state_relation_loss": 0.0004939343692967668,
      "step": 14870,
      "student_distance_mean": 1948.7151489257812,
      "teacher_distance_mean": 1.1678338050842285,
      "weighted_auxiliary_loss": 0.0004939343692967668,
      "weighted_relation_loss": 0.0004939343692967668
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.91955098222638,
      "flow_matching_loss": 0.02958434522151947,
      "grad_norm": 0.09210960566997528,
      "learning_rate": 5.479320199051942e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004896039463346824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.671186447143555,
      "state_query_norm": 1414.7972412109375,
      "state_relation_loss": 0.0004896039463346824,
      "step": 14880,
      "student_distance_mean": 1917.255126953125,
      "teacher_distance_mean": 1.1463170766830444,
      "weighted_auxiliary_loss": 0.0004896039463346824,
      "weighted_relation_loss": 0.0004896039463346824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.928905519176801,
      "flow_matching_loss": 0.03181314859539271,
      "grad_norm": 0.1404903382062912,
      "learning_rate": 5.473833726757314e-05,
      "loss": 0.0323,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005218200763920322,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.66536979675293,
      "state_query_norm": 1402.662548828125,
      "state_relation_loss": 0.0005218200763920322,
      "step": 14890,
      "student_distance_mean": 1908.462548828125,
      "teacher_distance_mean": 1.144977080821991,
      "weighted_auxiliary_loss": 0.0005218200763920322,
      "weighted_relation_loss": 0.0005218200763920322
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.938260056127222,
      "flow_matching_loss": 0.03126527201384306,
      "grad_norm": 0.11434825509786606,
      "learning_rate": 5.4683466787100304e-05,
      "loss": 0.0318,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005077320791315287,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.212063217163085,
      "state_query_norm": 1388.5364868164063,
      "state_relation_loss": 0.0005077320791315287,
      "step": 14900,
      "student_distance_mean": 1886.7309936523438,
      "teacher_distance_mean": 1.1362902641296386,
      "weighted_auxiliary_loss": 0.0005077320791315287,
      "weighted_relation_loss": 0.0005077320791315287
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.947614593077642,
      "flow_matching_loss": 0.03144807927310467,
      "grad_norm": 0.12671439349651337,
      "learning_rate": 5.462859061577369e-05,
      "loss": 0.032,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005032402055803687,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.727898025512694,
      "state_query_norm": 1410.9100952148438,
      "state_relation_loss": 0.0005032402055803687,
      "step": 14910,
      "student_distance_mean": 1915.6049438476562,
      "teacher_distance_mean": 1.149138832092285,
      "weighted_auxiliary_loss": 0.0005032402055803687,
      "weighted_relation_loss": 0.0005032402055803687
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.956969130028064,
      "flow_matching_loss": 0.030880552902817725,
      "grad_norm": 0.11911438405513763,
      "learning_rate": 5.4573708820273026e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004958467005053535,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.762051010131835,
      "state_query_norm": 1410.2621337890625,
      "state_relation_loss": 0.0004958467005053535,
      "step": 14920,
      "student_distance_mean": 1919.276904296875,
      "teacher_distance_mean": 1.1480269312858582,
      "weighted_auxiliary_loss": 0.0004958467005053535,
      "weighted_relation_loss": 0.0004958467005053535
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.966323666978484,
      "flow_matching_loss": 0.030885962210595606,
      "grad_norm": 0.1502220332622528,
      "learning_rate": 5.451882146728488e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004964430059771985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.38945350646973,
      "state_query_norm": 1398.7922119140626,
      "state_relation_loss": 0.0004964430059771985,
      "step": 14930,
      "student_distance_mean": 1895.9024780273437,
      "teacher_distance_mean": 1.14001407623291,
      "weighted_auxiliary_loss": 0.0004964430059771985,
      "weighted_relation_loss": 0.0004964430059771985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.975678203928906,
      "flow_matching_loss": 0.02909268718212843,
      "grad_norm": 0.10923094302415848,
      "learning_rate": 5.446392862350255e-05,
      "loss": 0.0296,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005103676143335178,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.97321090698242,
      "state_query_norm": 1410.2104858398438,
      "state_relation_loss": 0.0005103676143335178,
      "step": 14940,
      "student_distance_mean": 1926.2955200195313,
      "teacher_distance_mean": 1.1552883982658386,
      "weighted_auxiliary_loss": 0.0005103676143335178,
      "weighted_relation_loss": 0.0005103676143335178
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.985032740879326,
      "flow_matching_loss": 0.03004141114652157,
      "grad_norm": 0.09990005195140839,
      "learning_rate": 5.440903035562603e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005189803370740265,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.22299461364746,
      "state_query_norm": 1416.6159423828126,
      "state_relation_loss": 0.0005189803370740265,
      "step": 14950,
      "student_distance_mean": 1939.7451293945312,
      "teacher_distance_mean": 1.1641566395759582,
      "weighted_auxiliary_loss": 0.0005189803370740265,
      "weighted_relation_loss": 0.0005189803370740265
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 13.994387277829748,
      "flow_matching_loss": 0.03057870306074619,
      "grad_norm": 0.08504373580217361,
      "learning_rate": 5.435412673036188e-05,
      "loss": 0.0311,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005034585075918585,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.99600410461426,
      "state_query_norm": 1409.8173828125,
      "state_relation_loss": 0.0005034585075918585,
      "step": 14960,
      "student_distance_mean": 1928.0307006835938,
      "teacher_distance_mean": 1.1531848311424255,
      "weighted_auxiliary_loss": 0.0005034585075918585,
      "weighted_relation_loss": 0.0005034585075918585
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.003741814780168,
      "flow_matching_loss": 0.03215644210577011,
      "grad_norm": 0.10868648439645767,
      "learning_rate": 5.429921781442318e-05,
      "loss": 0.0326,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.000488613496418111,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 34.92235679626465,
      "state_query_norm": 1411.7588012695312,
      "state_relation_loss": 0.000488613496418111,
      "step": 14970,
      "student_distance_mean": 1927.701708984375,
      "teacher_distance_mean": 1.1555552363395691,
      "weighted_auxiliary_loss": 0.000488613496418111,
      "weighted_relation_loss": 0.000488613496418111
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.013096351730589,
      "flow_matching_loss": 0.030011369101703166,
      "grad_norm": 0.10961216688156128,
      "learning_rate": 5.424430367452945e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005173817538889125,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.365616989135745,
      "state_query_norm": 1390.1106323242188,
      "state_relation_loss": 0.0005173817538889125,
      "step": 14980,
      "student_distance_mean": 1896.2966552734374,
      "teacher_distance_mean": 1.1401427030563354,
      "weighted_auxiliary_loss": 0.0005173817538889125,
      "weighted_relation_loss": 0.0005173817538889125
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.02245088868101,
      "flow_matching_loss": 0.030755656771361828,
      "grad_norm": 0.08115260303020477,
      "learning_rate": 5.418938437740655e-05,
      "loss": 0.0313,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005175142723601311,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.58124885559082,
      "state_query_norm": 1398.5540161132812,
      "state_relation_loss": 0.0005175142723601311,
      "step": 14990,
      "student_distance_mean": 1908.475830078125,
      "teacher_distance_mean": 1.1518951058387756,
      "weighted_auxiliary_loss": 0.0005175142723601311,
      "weighted_relation_loss": 0.0005175142723601311
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.03180542563143,
      "flow_matching_loss": 0.029730971530079843,
      "grad_norm": 0.0955217182636261,
      "learning_rate": 5.413445998978658e-05,
      "loss": 0.0302,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000503368696081452,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.82408256530762,
      "state_query_norm": 1403.5124267578126,
      "state_relation_loss": 0.000503368696081452,
      "step": 15000,
      "student_distance_mean": 1920.2193725585937,
      "teacher_distance_mean": 1.1586827993392945,
      "weighted_auxiliary_loss": 0.000503368696081452,
      "weighted_relation_loss": 0.000503368696081452
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.041159962581853,
      "flow_matching_loss": 0.03027486950159073,
      "grad_norm": 0.09902361780405045,
      "learning_rate": 5.407953057840789e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004916729230899364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.04905967712402,
      "state_query_norm": 1415.2931762695312,
      "state_relation_loss": 0.0004916729230899364,
      "step": 15010,
      "student_distance_mean": 1930.6441528320313,
      "teacher_distance_mean": 1.1610694646835327,
      "weighted_auxiliary_loss": 0.0004916729230899364,
      "weighted_relation_loss": 0.0004916729230899364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.050514499532273,
      "flow_matching_loss": 0.031180380284786223,
      "grad_norm": 0.10470470786094666,
      "learning_rate": 5.4024596210014853e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047165320138446986,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 33.787323760986325,
      "state_query_norm": 1364.1370361328125,
      "state_relation_loss": 0.00047165320138446986,
      "step": 15020,
      "student_distance_mean": 1862.8042114257812,
      "teacher_distance_mean": 1.120703899860382,
      "weighted_auxiliary_loss": 0.00047165320138446986,
      "weighted_relation_loss": 0.00047165320138446986
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.059869036482693,
      "flow_matching_loss": 0.030546599812805653,
      "grad_norm": 0.10923433303833008,
      "learning_rate": 5.3969656951357935e-05,
      "loss": 0.031,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048159913276322184,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.79874267578125,
      "state_query_norm": 1410.4601684570312,
      "state_relation_loss": 0.00048159913276322184,
      "step": 15030,
      "student_distance_mean": 1919.3352783203125,
      "teacher_distance_mean": 1.1506827831268311,
      "weighted_auxiliary_loss": 0.00048159913276322184,
      "weighted_relation_loss": 0.00048159913276322184
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.069223573433115,
      "flow_matching_loss": 0.02880274914205074,
      "grad_norm": 0.08564979583024979,
      "learning_rate": 5.3914712869193504e-05,
      "loss": 0.0293,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005348969018086791,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.01635627746582,
      "state_query_norm": 1440.14609375,
      "state_relation_loss": 0.0005348969018086791,
      "step": 15040,
      "student_distance_mean": 1977.5071166992188,
      "teacher_distance_mean": 1.1855180859565735,
      "weighted_auxiliary_loss": 0.0005348969018086791,
      "weighted_relation_loss": 0.0005348969018086791
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.078578110383535,
      "flow_matching_loss": 0.028261867351830006,
      "grad_norm": 0.09236641228199005,
      "learning_rate": 5.385976403028381e-05,
      "loss": 0.0287,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004695435520261526,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.56954917907715,
      "state_query_norm": 1401.9941772460938,
      "state_relation_loss": 0.0004695435520261526,
      "step": 15050,
      "student_distance_mean": 1909.1221557617187,
      "teacher_distance_mean": 1.1444910407066344,
      "weighted_auxiliary_loss": 0.0004695435520261526,
      "weighted_relation_loss": 0.0004695435520261526
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.087932647333957,
      "flow_matching_loss": 0.029532517306506634,
      "grad_norm": 0.10206694900989532,
      "learning_rate": 5.3804810501396875e-05,
      "loss": 0.03,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048006768338382244,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.667541122436525,
      "state_query_norm": 1391.198583984375,
      "state_relation_loss": 0.00048006768338382244,
      "step": 15060,
      "student_distance_mean": 1906.26513671875,
      "teacher_distance_mean": 1.1445042490959167,
      "weighted_auxiliary_loss": 0.00048006768338382244,
      "weighted_relation_loss": 0.00048006768338382244
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.097287184284378,
      "flow_matching_loss": 0.030179411731660367,
      "grad_norm": 0.11691036075353622,
      "learning_rate": 5.374985234930639e-05,
      "loss": 0.0307,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005225153232458979,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.44324989318848,
      "state_query_norm": 1386.26162109375,
      "state_relation_loss": 0.0005225153232458979,
      "step": 15070,
      "student_distance_mean": 1897.1788208007813,
      "teacher_distance_mean": 1.141649568080902,
      "weighted_auxiliary_loss": 0.0005225153232458979,
      "weighted_relation_loss": 0.0005225153232458979
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.1066417212348,
      "flow_matching_loss": 0.02999521270394325,
      "grad_norm": 0.11125710606575012,
      "learning_rate": 5.3694889640791724e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005567678599618375,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15751953125,
      "state_query_norm": 1413.553955078125,
      "state_relation_loss": 0.0005567678599618375,
      "step": 15080,
      "student_distance_mean": 1934.6848754882812,
      "teacher_distance_mean": 1.1630894064903259,
      "weighted_auxiliary_loss": 0.0005567678599618375,
      "weighted_relation_loss": 0.0005567678599618375
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.11599625818522,
      "flow_matching_loss": 0.03030704241245985,
      "grad_norm": 0.10824770480394363,
      "learning_rate": 5.3639922442637736e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005084735981654376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1126708984375,
      "state_query_norm": 1415.9488647460937,
      "state_relation_loss": 0.0005084735981654376,
      "step": 15090,
      "student_distance_mean": 1933.2463012695312,
      "teacher_distance_mean": 1.1636129856109618,
      "weighted_auxiliary_loss": 0.0005084735981654376,
      "weighted_relation_loss": 0.0005084735981654376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.12535079513564,
      "flow_matching_loss": 0.031462276726961134,
      "grad_norm": 0.1374729424715042,
      "learning_rate": 5.358495082163476e-05,
      "loss": 0.032,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005089632380986586,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.140322875976565,
      "state_query_norm": 1418.6879272460938,
      "state_relation_loss": 0.0005089632380986586,
      "step": 15100,
      "student_distance_mean": 1934.1110107421875,
      "teacher_distance_mean": 1.1601808071136475,
      "weighted_auxiliary_loss": 0.0005089632380986586,
      "weighted_relation_loss": 0.0005089632380986586
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.134705332086062,
      "flow_matching_loss": 0.03201490268111229,
      "grad_norm": 0.09962832927703857,
      "learning_rate": 5.35299748445785e-05,
      "loss": 0.0325,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000494835470453836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.732598876953126,
      "state_query_norm": 1408.3238159179687,
      "state_relation_loss": 0.000494835470453836,
      "step": 15110,
      "student_distance_mean": 1917.678369140625,
      "teacher_distance_mean": 1.1480042219161988,
      "weighted_auxiliary_loss": 0.000494835470453836,
      "weighted_relation_loss": 0.000494835470453836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.144059869036482,
      "flow_matching_loss": 0.03121595289558172,
      "grad_norm": 0.10626835376024246,
      "learning_rate": 5.3474994578269944e-05,
      "loss": 0.0317,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000514187224325724,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.76657943725586,
      "state_query_norm": 1398.013623046875,
      "state_relation_loss": 0.000514187224325724,
      "step": 15120,
      "student_distance_mean": 1911.3594970703125,
      "teacher_distance_mean": 1.147790014743805,
      "weighted_auxiliary_loss": 0.000514187224325724,
      "weighted_relation_loss": 0.000514187224325724
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.153414405986904,
      "flow_matching_loss": 0.030006755143404007,
      "grad_norm": 0.12100424617528915,
      "learning_rate": 5.34200100895153e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005129582161316649,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.95225563049316,
      "state_query_norm": 1406.4889404296875,
      "state_relation_loss": 0.0005129582161316649,
      "step": 15130,
      "student_distance_mean": 1922.3712158203125,
      "teacher_distance_mean": 1.1530055403709412,
      "weighted_auxiliary_loss": 0.0005129582161316649,
      "weighted_relation_loss": 0.0005129582161316649
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.162768942937324,
      "flow_matching_loss": 0.0318567743524909,
      "grad_norm": 0.12685076892375946,
      "learning_rate": 5.3365021445125916e-05,
      "loss": 0.0324,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005030815023928881,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.94217376708984,
      "state_query_norm": 1407.5499877929688,
      "state_relation_loss": 0.0005030815023928881,
      "step": 15140,
      "student_distance_mean": 1922.94013671875,
      "teacher_distance_mean": 1.1525442838668822,
      "weighted_auxiliary_loss": 0.0005030815023928881,
      "weighted_relation_loss": 0.0005030815023928881
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.172123479887745,
      "flow_matching_loss": 0.029279070533812047,
      "grad_norm": 0.11011385917663574,
      "learning_rate": 5.331002871191817e-05,
      "loss": 0.0298,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004983074177289382,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.873877334594724,
      "state_query_norm": 1418.5657470703125,
      "state_relation_loss": 0.0004983074177289382,
      "step": 15150,
      "student_distance_mean": 1923.209912109375,
      "teacher_distance_mean": 1.1436536073684693,
      "weighted_auxiliary_loss": 0.0004983074177289382,
      "weighted_relation_loss": 0.0004983074177289382
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.181478016838167,
      "flow_matching_loss": 0.03029635827988386,
      "grad_norm": 0.0900127962231636,
      "learning_rate": 5.3255031956713443e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048785629915073516,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.83389549255371,
      "state_query_norm": 1415.3177124023437,
      "state_relation_loss": 0.00048785629915073516,
      "step": 15160,
      "student_distance_mean": 1921.5660400390625,
      "teacher_distance_mean": 1.1489205598831176,
      "weighted_auxiliary_loss": 0.00048785629915073516,
      "weighted_relation_loss": 0.00048785629915073516
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.190832553788587,
      "flow_matching_loss": 0.02962898537516594,
      "grad_norm": 0.10422029346227646,
      "learning_rate": 5.320003124633795e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005045562662417069,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.845572280883786,
      "state_query_norm": 1408.01611328125,
      "state_relation_loss": 0.0005045562662417069,
      "step": 15170,
      "student_distance_mean": 1918.494482421875,
      "teacher_distance_mean": 1.1497650861740112,
      "weighted_auxiliary_loss": 0.0005045562662417069,
      "weighted_relation_loss": 0.0005045562662417069
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.200187090739009,
      "flow_matching_loss": 0.028831185400485994,
      "grad_norm": 0.10280055552721024,
      "learning_rate": 5.314502664762274e-05,
      "loss": 0.0293,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004995128692826256,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.9090389251709,
      "state_query_norm": 1406.4497314453124,
      "state_relation_loss": 0.0004995128692826256,
      "step": 15180,
      "student_distance_mean": 1921.7605102539062,
      "teacher_distance_mean": 1.155085825920105,
      "weighted_auxiliary_loss": 0.0004995128692826256,
      "weighted_relation_loss": 0.0004995128692826256
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.209541627689429,
      "flow_matching_loss": 0.02858731374144554,
      "grad_norm": 0.10741622000932693,
      "learning_rate": 5.3090018227403604e-05,
      "loss": 0.0291,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004700054763816297,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.785094833374025,
      "state_query_norm": 1416.2339233398438,
      "state_relation_loss": 0.0004700054763816297,
      "step": 15190,
      "student_distance_mean": 1921.9924194335938,
      "teacher_distance_mean": 1.1541968703269958,
      "weighted_auxiliary_loss": 0.0004700054763816297,
      "weighted_relation_loss": 0.0004700054763816297
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.218896164639851,
      "flow_matching_loss": 0.030720766447484494,
      "grad_norm": 0.09502332657575607,
      "learning_rate": 5.3035006052520955e-05,
      "loss": 0.0312,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004988628352293745,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.64305419921875,
      "state_query_norm": 1393.5543090820313,
      "state_relation_loss": 0.0004988628352293745,
      "step": 15200,
      "student_distance_mean": 1906.6689208984376,
      "teacher_distance_mean": 1.145170772075653,
      "weighted_auxiliary_loss": 0.0004988628352293745,
      "weighted_relation_loss": 0.0004988628352293745
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.228250701590271,
      "flow_matching_loss": 0.030321907624602316,
      "grad_norm": 0.09346864372491837,
      "learning_rate": 5.297999018981976e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005037467344664037,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.51238021850586,
      "state_query_norm": 1401.1035888671875,
      "state_relation_loss": 0.0005037467344664037,
      "step": 15210,
      "student_distance_mean": 1905.024560546875,
      "teacher_distance_mean": 1.1437161326408387,
      "weighted_auxiliary_loss": 0.0005037467344664037,
      "weighted_relation_loss": 0.0005037467344664037
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.237605238540691,
      "flow_matching_loss": 0.030445844680070878,
      "grad_norm": 0.11604288220405579,
      "learning_rate": 5.2924970706149505e-05,
      "loss": 0.031,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005328291590558365,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.218027114868164,
      "state_query_norm": 1421.8275512695313,
      "state_relation_loss": 0.0005328291590558365,
      "step": 15220,
      "student_distance_mean": 1939.4019897460937,
      "teacher_distance_mean": 1.1630156993865968,
      "weighted_auxiliary_loss": 0.0005328291590558365,
      "weighted_relation_loss": 0.0005328291590558365
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.246959775491113,
      "flow_matching_loss": 0.028322920948266984,
      "grad_norm": 0.12010113894939423,
      "learning_rate": 5.2869947668364014e-05,
      "loss": 0.0288,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005019122210796923,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16527671813965,
      "state_query_norm": 1416.9679077148437,
      "state_relation_loss": 0.0005019122210796923,
      "step": 15230,
      "student_distance_mean": 1933.649267578125,
      "teacher_distance_mean": 1.1537871718406678,
      "weighted_auxiliary_loss": 0.0005019122210796923,
      "weighted_relation_loss": 0.0005019122210796923
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.256314312441534,
      "flow_matching_loss": 0.03008501660078764,
      "grad_norm": 0.10186545550823212,
      "learning_rate": 5.2814921143321504e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005176113860215992,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.97702865600586,
      "state_query_norm": 1412.4092895507813,
      "state_relation_loss": 0.0005176113860215992,
      "step": 15240,
      "student_distance_mean": 1924.1152099609376,
      "teacher_distance_mean": 1.1489405751228332,
      "weighted_auxiliary_loss": 0.0005176113860215992,
      "weighted_relation_loss": 0.0005176113860215992
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.265668849391956,
      "flow_matching_loss": 0.030374188721179963,
      "grad_norm": 0.09834747016429901,
      "learning_rate": 5.275989119788436e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047230547352228314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.78898887634277,
      "state_query_norm": 1404.8881591796876,
      "state_relation_loss": 0.00047230547352228314,
      "step": 15250,
      "student_distance_mean": 1913.99873046875,
      "teacher_distance_mean": 1.1433351397514344,
      "weighted_auxiliary_loss": 0.00047230547352228314,
      "weighted_relation_loss": 0.00047230547352228314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.275023386342376,
      "flow_matching_loss": 0.027723702415823937,
      "grad_norm": 0.11194396018981934,
      "learning_rate": 5.270485789891919e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005080526461824775,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.02738304138184,
      "state_query_norm": 1415.9224487304687,
      "state_relation_loss": 0.0005080526461824775,
      "step": 15260,
      "student_distance_mean": 1930.7583251953124,
      "teacher_distance_mean": 1.1527809858322144,
      "weighted_auxiliary_loss": 0.0005080526461824775,
      "weighted_relation_loss": 0.0005080526461824775
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.284377923292798,
      "flow_matching_loss": 0.031869385950267316,
      "grad_norm": 0.11509303003549576,
      "learning_rate": 5.264982131329661e-05,
      "loss": 0.0323,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047538324142806234,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.111196517944336,
      "state_query_norm": 1415.1144897460938,
      "state_relation_loss": 0.00047538324142806234,
      "step": 15270,
      "student_distance_mean": 1931.4581787109375,
      "teacher_distance_mean": 1.1533755660057068,
      "weighted_auxiliary_loss": 0.00047538324142806234,
      "weighted_relation_loss": 0.00047538324142806234
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.293732460243218,
      "flow_matching_loss": 0.028915499150753022,
      "grad_norm": 0.11274449527263641,
      "learning_rate": 5.259478150789128e-05,
      "loss": 0.0295,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005431739060441032,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39264106750488,
      "state_query_norm": 1424.5829345703125,
      "state_relation_loss": 0.0005431739060441032,
      "step": 15280,
      "student_distance_mean": 1945.5957885742187,
      "teacher_distance_mean": 1.1654162764549256,
      "weighted_auxiliary_loss": 0.0005431739060441032,
      "weighted_relation_loss": 0.0005431739060441032
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.303086997193638,
      "flow_matching_loss": 0.028445976600050927,
      "grad_norm": 0.09937837719917297,
      "learning_rate": 5.253973854958173e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004968489025486633,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.00603408813477,
      "state_query_norm": 1411.062548828125,
      "state_relation_loss": 0.0004968489025486633,
      "step": 15290,
      "student_distance_mean": 1927.9616943359374,
      "teacher_distance_mean": 1.1496195077896119,
      "weighted_auxiliary_loss": 0.0004968489025486633,
      "weighted_relation_loss": 0.0004968489025486633
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.31244153414406,
      "flow_matching_loss": 0.02985303904861212,
      "grad_norm": 0.10110580176115036,
      "learning_rate": 5.2484692505250376e-05,
      "loss": 0.0304,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005365473189158365,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16015090942383,
      "state_query_norm": 1419.36787109375,
      "state_relation_loss": 0.0005365473189158365,
      "step": 15300,
      "student_distance_mean": 1936.9547973632812,
      "teacher_distance_mean": 1.15882465839386,
      "weighted_auxiliary_loss": 0.0005365473189158365,
      "weighted_relation_loss": 0.0005365473189158365
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.32179607109448,
      "flow_matching_loss": 0.030153154768049716,
      "grad_norm": 0.1170104369521141,
      "learning_rate": 5.242964344178333e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004935046890750527,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.86519775390625,
      "state_query_norm": 1410.8985595703125,
      "state_relation_loss": 0.0004935046890750527,
      "step": 15310,
      "student_distance_mean": 1922.6455932617187,
      "teacher_distance_mean": 1.1525447249412537,
      "weighted_auxiliary_loss": 0.0004935046890750527,
      "weighted_relation_loss": 0.0004935046890750527
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.331150608044902,
      "flow_matching_loss": 0.031105165369808674,
      "grad_norm": 0.11227284371852875,
      "learning_rate": 5.237459142607041e-05,
      "loss": 0.0316,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048513893270865085,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14977188110352,
      "state_query_norm": 1413.461669921875,
      "state_relation_loss": 0.00048513893270865085,
      "step": 15320,
      "student_distance_mean": 1931.5445190429687,
      "teacher_distance_mean": 1.157228207588196,
      "weighted_auxiliary_loss": 0.00048513893270865085,
      "weighted_relation_loss": 0.00048513893270865085
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.340505144995323,
      "flow_matching_loss": 0.028996261581778527,
      "grad_norm": 0.09015260636806488,
      "learning_rate": 5.231953652500498e-05,
      "loss": 0.0295,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005021126416977495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.82811279296875,
      "state_query_norm": 1412.477099609375,
      "state_relation_loss": 0.0005021126416977495,
      "step": 15330,
      "student_distance_mean": 1920.3253540039063,
      "teacher_distance_mean": 1.150925385951996,
      "weighted_auxiliary_loss": 0.0005021126416977495,
      "weighted_relation_loss": 0.0005021126416977495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.349859681945743,
      "flow_matching_loss": 0.028873172029852866,
      "grad_norm": 0.0801805779337883,
      "learning_rate": 5.2264478805483976e-05,
      "loss": 0.0294,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005030822823755444,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06595115661621,
      "state_query_norm": 1410.9969360351563,
      "state_relation_loss": 0.0005030822823755444,
      "step": 15340,
      "student_distance_mean": 1929.554248046875,
      "teacher_distance_mean": 1.153645646572113,
      "weighted_auxiliary_loss": 0.0005030822823755444,
      "weighted_relation_loss": 0.0005030822823755444
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.359214218896165,
      "flow_matching_loss": 0.030203698575496672,
      "grad_norm": 0.09889304637908936,
      "learning_rate": 5.220941833440768e-05,
      "loss": 0.0307,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005109716119477525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6077163696289,
      "state_query_norm": 1433.50595703125,
      "state_relation_loss": 0.0005109716119477525,
      "step": 15350,
      "student_distance_mean": 1958.9514282226562,
      "teacher_distance_mean": 1.1699068546295166,
      "weighted_auxiliary_loss": 0.0005109716119477525,
      "weighted_relation_loss": 0.0005109716119477525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.368568755846585,
      "flow_matching_loss": 0.02788886595517397,
      "grad_norm": 0.1034122109413147,
      "learning_rate": 5.215435517867977e-05,
      "loss": 0.0284,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004814657615497708,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50236625671387,
      "state_query_norm": 1424.6681030273437,
      "state_relation_loss": 0.0004814657615497708,
      "step": 15360,
      "student_distance_mean": 1952.7824951171874,
      "teacher_distance_mean": 1.162202537059784,
      "weighted_auxiliary_loss": 0.0004814657615497708,
      "weighted_relation_loss": 0.0004814657615497708
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.377923292797007,
      "flow_matching_loss": 0.029082771204411984,
      "grad_norm": 0.11142238974571228,
      "learning_rate": 5.209928940520719e-05,
      "loss": 0.0296,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005092241684906184,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23207740783691,
      "state_query_norm": 1420.3800170898437,
      "state_relation_loss": 0.0005092241684906184,
      "step": 15370,
      "student_distance_mean": 1940.6756103515625,
      "teacher_distance_mean": 1.1551206231117248,
      "weighted_auxiliary_loss": 0.0005092241684906184,
      "weighted_relation_loss": 0.0005092241684906184
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.387277829747427,
      "flow_matching_loss": 0.02952213864773512,
      "grad_norm": 0.11683907359838486,
      "learning_rate": 5.204422108090004e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005279860750306397,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.569631576538086,
      "state_query_norm": 1386.58349609375,
      "state_relation_loss": 0.0005279860750306397,
      "step": 15380,
      "student_distance_mean": 1897.7547973632813,
      "teacher_distance_mean": 1.138661301136017,
      "weighted_auxiliary_loss": 0.0005279860750306397,
      "weighted_relation_loss": 0.0005279860750306397
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.39663236669785,
      "flow_matching_loss": 0.029017531499266624,
      "grad_norm": 0.10756175220012665,
      "learning_rate": 5.19891502726715e-05,
      "loss": 0.0295,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004962458973750472,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.71723747253418,
      "state_query_norm": 1406.2850219726563,
      "state_relation_loss": 0.0004962458973750472,
      "step": 15390,
      "student_distance_mean": 1917.0038330078125,
      "teacher_distance_mean": 1.1514901161193847,
      "weighted_auxiliary_loss": 0.0004962458973750472,
      "weighted_relation_loss": 0.0004962458973750472
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.40598690364827,
      "flow_matching_loss": 0.030863374285399914,
      "grad_norm": 0.09889647364616394,
      "learning_rate": 5.193407704743782e-05,
      "loss": 0.0314,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005110959900775924,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.00811424255371,
      "state_query_norm": 1411.4171875,
      "state_relation_loss": 0.0005110959900775924,
      "step": 15400,
      "student_distance_mean": 1926.139208984375,
      "teacher_distance_mean": 1.1544872403144837,
      "weighted_auxiliary_loss": 0.0005110959900775924,
      "weighted_relation_loss": 0.0005110959900775924
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.41534144059869,
      "flow_matching_loss": 0.028818936832249165,
      "grad_norm": 0.09709315001964569,
      "learning_rate": 5.187900147211815e-05,
      "loss": 0.0293,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004859813256189227,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.69355926513672,
      "state_query_norm": 1405.0420532226562,
      "state_relation_loss": 0.0004859813256189227,
      "step": 15410,
      "student_distance_mean": 1911.7913452148437,
      "teacher_distance_mean": 1.1427545547485352,
      "weighted_auxiliary_loss": 0.0004859813256189227,
      "weighted_relation_loss": 0.0004859813256189227
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.424695977549112,
      "flow_matching_loss": 0.026397704891860486,
      "grad_norm": 0.0946725457906723,
      "learning_rate": 5.182392361363453e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005318298644851893,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.321998596191406,
      "state_query_norm": 1416.1275634765625,
      "state_relation_loss": 0.0005318298644851893,
      "step": 15420,
      "student_distance_mean": 1940.5600463867188,
      "teacher_distance_mean": 1.1601795077323913,
      "weighted_auxiliary_loss": 0.0005318298644851893,
      "weighted_relation_loss": 0.0005318298644851893
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.434050514499532,
      "flow_matching_loss": 0.028408313915133477,
      "grad_norm": 0.08594322949647903,
      "learning_rate": 5.176884353891172e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005210011411691084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.220838165283205,
      "state_query_norm": 1422.0258178710938,
      "state_relation_loss": 0.0005210011411691084,
      "step": 15430,
      "student_distance_mean": 1943.482666015625,
      "teacher_distance_mean": 1.1581999897956847,
      "weighted_auxiliary_loss": 0.0005210011411691084,
      "weighted_relation_loss": 0.0005210011411691084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.443405051449954,
      "flow_matching_loss": 0.028103701025247573,
      "grad_norm": 0.1021907776594162,
      "learning_rate": 5.171376131487722e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004991327354218811,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.591953659057616,
      "state_query_norm": 1427.4261840820313,
      "state_relation_loss": 0.0004991327354218811,
      "step": 15440,
      "student_distance_mean": 1954.7130859375,
      "teacher_distance_mean": 1.1641003727912902,
      "weighted_auxiliary_loss": 0.0004991327354218811,
      "weighted_relation_loss": 0.0004991327354218811
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.452759588400374,
      "flow_matching_loss": 0.029573314264416693,
      "grad_norm": 0.10126682370901108,
      "learning_rate": 5.165867700846113e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005261143640382215,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.138314437866214,
      "state_query_norm": 1419.3939453125,
      "state_relation_loss": 0.0005261143640382215,
      "step": 15450,
      "student_distance_mean": 1935.9645141601563,
      "teacher_distance_mean": 1.1536445021629333,
      "weighted_auxiliary_loss": 0.0005261143640382215,
      "weighted_relation_loss": 0.0005261143640382215
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.462114125350794,
      "flow_matching_loss": 0.027958675846457483,
      "grad_norm": 0.09495944529771805,
      "learning_rate": 5.160359068659606e-05,
      "loss": 0.0285,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005287501408020034,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.118228149414065,
      "state_query_norm": 1410.1981811523438,
      "state_relation_loss": 0.0005287501408020034,
      "step": 15460,
      "student_distance_mean": 1929.831884765625,
      "teacher_distance_mean": 1.1568171024322509,
      "weighted_auxiliary_loss": 0.0005287501408020034,
      "weighted_relation_loss": 0.0005287501408020034
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.471468662301216,
      "flow_matching_loss": 0.03294053766876459,
      "grad_norm": 0.10180478543043137,
      "learning_rate": 5.154850241621712e-05,
      "loss": 0.0334,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004680739832110703,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47433586120606,
      "state_query_norm": 1426.8451904296876,
      "state_relation_loss": 0.0004680739832110703,
      "step": 15470,
      "student_distance_mean": 1951.6462646484374,
      "teacher_distance_mean": 1.1683720588684081,
      "weighted_auxiliary_loss": 0.0004680739832110703,
      "weighted_relation_loss": 0.0004680739832110703
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.480823199251637,
      "flow_matching_loss": 0.03021280262619257,
      "grad_norm": 0.10821860283613205,
      "learning_rate": 5.149341226426172e-05,
      "loss": 0.0307,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005036507965996862,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.661533737182616,
      "state_query_norm": 1394.2851318359376,
      "state_relation_loss": 0.0005036507965996862,
      "step": 15480,
      "student_distance_mean": 1907.046826171875,
      "teacher_distance_mean": 1.1462918996810914,
      "weighted_auxiliary_loss": 0.0005036507965996862,
      "weighted_relation_loss": 0.0005036507965996862
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.490177736202059,
      "flow_matching_loss": 0.027580533362925052,
      "grad_norm": 0.11159384250640869,
      "learning_rate": 5.14383202976696e-05,
      "loss": 0.0281,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000516942355898209,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.31347808837891,
      "state_query_norm": 1392.3165161132813,
      "state_relation_loss": 0.000516942355898209,
      "step": 15490,
      "student_distance_mean": 1891.8364013671876,
      "teacher_distance_mean": 1.1383407950401305,
      "weighted_auxiliary_loss": 0.000516942355898209,
      "weighted_relation_loss": 0.000516942355898209
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.499532273152479,
      "flow_matching_loss": 0.029400079138576986,
      "grad_norm": 0.08202087879180908,
      "learning_rate": 5.1383226583382684e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004916879028314724,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.93290252685547,
      "state_query_norm": 1408.3034912109374,
      "state_relation_loss": 0.0004916879028314724,
      "step": 15500,
      "student_distance_mean": 1922.8682495117187,
      "teacher_distance_mean": 1.1541371941566467,
      "weighted_auxiliary_loss": 0.0004916879028314724,
      "weighted_relation_loss": 0.0004916879028314724
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.5088868101029,
      "flow_matching_loss": 0.027858006209135054,
      "grad_norm": 0.1023976057767868,
      "learning_rate": 5.132813118834504e-05,
      "loss": 0.0284,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005064132390543818,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.114728927612305,
      "state_query_norm": 1415.0633056640625,
      "state_relation_loss": 0.0005064132390543818,
      "step": 15510,
      "student_distance_mean": 1932.6267333984374,
      "teacher_distance_mean": 1.156298077106476,
      "weighted_auxiliary_loss": 0.0005064132390543818,
      "weighted_relation_loss": 0.0005064132390543818
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.518241347053321,
      "flow_matching_loss": 0.029986845143139363,
      "grad_norm": 0.0981561690568924,
      "learning_rate": 5.1273034179502777e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004844902228796855,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.430049133300784,
      "state_query_norm": 1392.0513793945313,
      "state_relation_loss": 0.0004844902228796855,
      "step": 15520,
      "student_distance_mean": 1898.4030639648438,
      "teacher_distance_mean": 1.135592770576477,
      "weighted_auxiliary_loss": 0.0004844902228796855,
      "weighted_relation_loss": 0.0004844902228796855
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.527595884003741,
      "flow_matching_loss": 0.029559214413166047,
      "grad_norm": 0.0987362340092659,
      "learning_rate": 5.1217935623803945e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004932410432957113,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17287139892578,
      "state_query_norm": 1410.6183837890626,
      "state_relation_loss": 0.0004932410432957113,
      "step": 15530,
      "student_distance_mean": 1933.5575927734376,
      "teacher_distance_mean": 1.160859751701355,
      "weighted_auxiliary_loss": 0.0004932410432957113,
      "weighted_relation_loss": 0.0004932410432957113
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.536950420954163,
      "flow_matching_loss": 0.028703846596181394,
      "grad_norm": 0.11774498969316483,
      "learning_rate": 5.1162835588198476e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005119471898069605,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52677268981934,
      "state_query_norm": 1426.29384765625,
      "state_relation_loss": 0.0005119471898069605,
      "step": 15540,
      "student_distance_mean": 1954.8991943359374,
      "teacher_distance_mean": 1.171349585056305,
      "weighted_auxiliary_loss": 0.0005119471898069605,
      "weighted_relation_loss": 0.0005119471898069605
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.546304957904583,
      "flow_matching_loss": 0.027406495250761508,
      "grad_norm": 0.09084219485521317,
      "learning_rate": 5.110773413963813e-05,
      "loss": 0.0279,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005119027948239819,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.713341522216794,
      "state_query_norm": 1410.9000610351563,
      "state_relation_loss": 0.0005119027948239819,
      "step": 15550,
      "student_distance_mean": 1912.6762939453124,
      "teacher_distance_mean": 1.1496511697769165,
      "weighted_auxiliary_loss": 0.0005119027948239819,
      "weighted_relation_loss": 0.0005119027948239819
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.555659494855005,
      "flow_matching_loss": 0.029686659947037695,
      "grad_norm": 0.11051397025585175,
      "learning_rate": 5.105263134507636e-05,
      "loss": 0.0302,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048240744799841193,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.86102142333984,
      "state_query_norm": 1406.7699340820313,
      "state_relation_loss": 0.00048240744799841193,
      "step": 15560,
      "student_distance_mean": 1920.9094970703125,
      "teacher_distance_mean": 1.1484879732131958,
      "weighted_auxiliary_loss": 0.00048240744799841193,
      "weighted_relation_loss": 0.00048240744799841193
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.565014031805426,
      "flow_matching_loss": 0.02936923522502184,
      "grad_norm": 0.12032745033502579,
      "learning_rate": 5.099752727146824e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005194042110815644,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.71427879333496,
      "state_query_norm": 1400.9252197265625,
      "state_relation_loss": 0.0005194042110815644,
      "step": 15570,
      "student_distance_mean": 1908.4614624023438,
      "teacher_distance_mean": 1.1444331645965575,
      "weighted_auxiliary_loss": 0.0005194042110815644,
      "weighted_relation_loss": 0.0005194042110815644
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.574368568755848,
      "flow_matching_loss": 0.030560107715427876,
      "grad_norm": 0.10731317102909088,
      "learning_rate": 5.094242198577042e-05,
      "loss": 0.031,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048762050282675774,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.59017105102539,
      "state_query_norm": 1403.289794921875,
      "state_relation_loss": 0.00048762050282675774,
      "step": 15580,
      "student_distance_mean": 1905.3894287109374,
      "teacher_distance_mean": 1.1393146753311156,
      "weighted_auxiliary_loss": 0.00048762050282675774,
      "weighted_relation_loss": 0.00048762050282675774
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.583723105706268,
      "flow_matching_loss": 0.0282513078302145,
      "grad_norm": 0.12221001088619232,
      "learning_rate": 5.088731555494102e-05,
      "loss": 0.0288,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004999447060981766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.292121505737306,
      "state_query_norm": 1419.3356201171875,
      "state_relation_loss": 0.0004999447060981766,
      "step": 15590,
      "student_distance_mean": 1937.4001586914062,
      "teacher_distance_mean": 1.158518671989441,
      "weighted_auxiliary_loss": 0.0004999447060981766,
      "weighted_relation_loss": 0.0004999447060981766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.593077642656688,
      "flow_matching_loss": 0.031022657081484793,
      "grad_norm": 0.11211685091257095,
      "learning_rate": 5.0832208045939556e-05,
      "loss": 0.0315,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004818949237233028,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.6815803527832,
      "state_query_norm": 1411.8895141601563,
      "state_relation_loss": 0.0004818949237233028,
      "step": 15600,
      "student_distance_mean": 1914.253662109375,
      "teacher_distance_mean": 1.1438122987747192,
      "weighted_auxiliary_loss": 0.0004818949237233028,
      "weighted_relation_loss": 0.0004818949237233028
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.60243217960711,
      "flow_matching_loss": 0.02916528657078743,
      "grad_norm": 0.10636256635189056,
      "learning_rate": 5.0777099525726844e-05,
      "loss": 0.0296,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047226015594787897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.4984375,
      "state_query_norm": 1393.5841552734375,
      "state_relation_loss": 0.00047226015594787897,
      "step": 15610,
      "student_distance_mean": 1900.483984375,
      "teacher_distance_mean": 1.1340259075164796,
      "weighted_auxiliary_loss": 0.00047226015594787897,
      "weighted_relation_loss": 0.00047226015594787897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.61178671655753,
      "flow_matching_loss": 0.030441381596028806,
      "grad_norm": 0.16680976748466492,
      "learning_rate": 5.072199006126494e-05,
      "loss": 0.0309,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048821979144122454,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.93726692199707,
      "state_query_norm": 1415.9601318359375,
      "state_relation_loss": 0.00048821979144122454,
      "step": 15620,
      "student_distance_mean": 1923.3476440429688,
      "teacher_distance_mean": 1.1517734050750732,
      "weighted_auxiliary_loss": 0.00048821979144122454,
      "weighted_relation_loss": 0.00048821979144122454
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.621141253507952,
      "flow_matching_loss": 0.029585442692041396,
      "grad_norm": 0.11013878881931305,
      "learning_rate": 5.0666879719517026e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005447904404718429,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56687507629395,
      "state_query_norm": 1428.0460693359375,
      "state_relation_loss": 0.0005447904404718429,
      "step": 15630,
      "student_distance_mean": 1955.6920043945313,
      "teacher_distance_mean": 1.1698583364486694,
      "weighted_auxiliary_loss": 0.0005447904404718429,
      "weighted_relation_loss": 0.0005447904404718429
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.630495790458372,
      "flow_matching_loss": 0.030237434804439543,
      "grad_norm": 0.09619257599115372,
      "learning_rate": 5.061176856744737e-05,
      "loss": 0.0307,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005019851116230712,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41848526000977,
      "state_query_norm": 1423.4529296875,
      "state_relation_loss": 0.0005019851116230712,
      "step": 15640,
      "student_distance_mean": 1947.0416625976563,
      "teacher_distance_mean": 1.1592581272125244,
      "weighted_auxiliary_loss": 0.0005019851116230712,
      "weighted_relation_loss": 0.0005019851116230712
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.639850327408793,
      "flow_matching_loss": 0.02920453194528818,
      "grad_norm": 0.11347932368516922,
      "learning_rate": 5.0556656672021205e-05,
      "loss": 0.0297,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004843726521357894,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32641754150391,
      "state_query_norm": 1431.1461181640625,
      "state_relation_loss": 0.0004843726521357894,
      "step": 15650,
      "student_distance_mean": 1949.0948852539063,
      "teacher_distance_mean": 1.1587128520011902,
      "weighted_auxiliary_loss": 0.0004843726521357894,
      "weighted_relation_loss": 0.0004843726521357894
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.649204864359215,
      "flow_matching_loss": 0.027689754962921143,
      "grad_norm": 0.10587657243013382,
      "learning_rate": 5.050154410020472e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004853570193517953,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.54941539764404,
      "state_query_norm": 1398.5412353515626,
      "state_relation_loss": 0.0004853570193517953,
      "step": 15660,
      "student_distance_mean": 1904.677783203125,
      "teacher_distance_mean": 1.1383970141410829,
      "weighted_auxiliary_loss": 0.0004853570193517953,
      "weighted_relation_loss": 0.0004853570193517953
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.658559401309635,
      "flow_matching_loss": 0.029699463024735452,
      "grad_norm": 0.11034484952688217,
      "learning_rate": 5.044643091896485e-05,
      "loss": 0.0302,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000503942885552533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52883033752441,
      "state_query_norm": 1430.2916870117188,
      "state_relation_loss": 0.000503942885552533,
      "step": 15670,
      "student_distance_mean": 1956.5153198242188,
      "teacher_distance_mean": 1.166893470287323,
      "weighted_auxiliary_loss": 0.000503942885552533,
      "weighted_relation_loss": 0.000503942885552533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.667913938260057,
      "flow_matching_loss": 0.029075587913393975,
      "grad_norm": 0.08800309896469116,
      "learning_rate": 5.039131719526932e-05,
      "loss": 0.0296,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005175191967282444,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07431488037109,
      "state_query_norm": 1413.6445678710938,
      "state_relation_loss": 0.0005175191967282444,
      "step": 15680,
      "student_distance_mean": 1930.5009033203125,
      "teacher_distance_mean": 1.152797818183899,
      "weighted_auxiliary_loss": 0.0005175191967282444,
      "weighted_relation_loss": 0.0005175191967282444
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.677268475210477,
      "flow_matching_loss": 0.028698442317545415,
      "grad_norm": 0.12062518298625946,
      "learning_rate": 5.03362029960865e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004881957051111385,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.046187210083005,
      "state_query_norm": 1413.0301635742187,
      "state_relation_loss": 0.0004881957051111385,
      "step": 15690,
      "student_distance_mean": 1931.2420532226563,
      "teacher_distance_mean": 1.1507092714309692,
      "weighted_auxiliary_loss": 0.0004881957051111385,
      "weighted_relation_loss": 0.0004881957051111385
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.686623012160897,
      "flow_matching_loss": 0.03012269027531147,
      "grad_norm": 0.09984938055276871,
      "learning_rate": 5.028108838838533e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005167022522073239,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.238304138183594,
      "state_query_norm": 1417.3067016601562,
      "state_relation_loss": 0.0005167022522073239,
      "step": 15700,
      "student_distance_mean": 1936.1853393554688,
      "teacher_distance_mean": 1.1486157178878784,
      "weighted_auxiliary_loss": 0.0005167022522073239,
      "weighted_relation_loss": 0.0005167022522073239
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.69597754911132,
      "flow_matching_loss": 0.029538415372371674,
      "grad_norm": 0.0956389307975769,
      "learning_rate": 5.0225973439135276e-05,
      "loss": 0.03,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005055422487203031,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.35080795288086,
      "state_query_norm": 1430.4026733398437,
      "state_relation_loss": 0.0005055422487203031,
      "step": 15710,
      "student_distance_mean": 1947.4407958984375,
      "teacher_distance_mean": 1.1560262322425843,
      "weighted_auxiliary_loss": 0.0005055422487203031,
      "weighted_relation_loss": 0.0005055422487203031
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.70533208606174,
      "flow_matching_loss": 0.029508467577397824,
      "grad_norm": 0.12107519805431366,
      "learning_rate": 5.017085821530617e-05,
      "loss": 0.03,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000493010293575935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38307876586914,
      "state_query_norm": 1432.9578491210937,
      "state_relation_loss": 0.000493010293575935,
      "step": 15720,
      "student_distance_mean": 1948.2628295898437,
      "teacher_distance_mean": 1.1573720932006837,
      "weighted_auxiliary_loss": 0.000493010293575935,
      "weighted_relation_loss": 0.000493010293575935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.714686623012161,
      "flow_matching_loss": 0.028747762739658355,
      "grad_norm": 0.10488234460353851,
      "learning_rate": 5.011574278386822e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047978111542761325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24213180541992,
      "state_query_norm": 1419.6253784179687,
      "state_relation_loss": 0.00047978111542761325,
      "step": 15730,
      "student_distance_mean": 1936.4722412109375,
      "teacher_distance_mean": 1.153946876525879,
      "weighted_auxiliary_loss": 0.00047978111542761325,
      "weighted_relation_loss": 0.00047978111542761325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.724041159962582,
      "flow_matching_loss": 0.029918996058404444,
      "grad_norm": 0.10036367177963257,
      "learning_rate": 5.006062721179189e-05,
      "loss": 0.0304,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004789121594512835,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.576164627075194,
      "state_query_norm": 1391.19345703125,
      "state_relation_loss": 0.0004789121594512835,
      "step": 15740,
      "student_distance_mean": 1902.5292236328125,
      "teacher_distance_mean": 1.1343261361122132,
      "weighted_auxiliary_loss": 0.0004789121594512835,
      "weighted_relation_loss": 0.0004789121594512835
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.733395696913004,
      "flow_matching_loss": 0.029094578698277473,
      "grad_norm": 0.0936993658542633,
      "learning_rate": 5.000551156604777e-05,
      "loss": 0.0296,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004830614838283509,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.054696273803714,
      "state_query_norm": 1416.9931762695312,
      "state_relation_loss": 0.0004830614838283509,
      "step": 15750,
      "student_distance_mean": 1929.41181640625,
      "teacher_distance_mean": 1.1541453361511231,
      "weighted_auxiliary_loss": 0.0004830614838283509,
      "weighted_relation_loss": 0.0004830614838283509
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.742750233863424,
      "flow_matching_loss": 0.030615204758942128,
      "grad_norm": 0.1183866485953331,
      "learning_rate": 4.9950395913606594e-05,
      "loss": 0.0311,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004878121020738035,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75368118286133,
      "state_query_norm": 1397.4573608398437,
      "state_relation_loss": 0.0004878121020738035,
      "step": 15760,
      "student_distance_mean": 1913.5091674804687,
      "teacher_distance_mean": 1.145671784877777,
      "weighted_auxiliary_loss": 0.0004878121020738035,
      "weighted_relation_loss": 0.0004878121020738035
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.752104770813844,
      "flow_matching_loss": 0.029432549700140954,
      "grad_norm": 0.10033584386110306,
      "learning_rate": 4.989528032143903e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005001712619559839,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.921648025512695,
      "state_query_norm": 1409.977734375,
      "state_relation_loss": 0.0005001712619559839,
      "step": 15770,
      "student_distance_mean": 1919.706689453125,
      "teacher_distance_mean": 1.146583092212677,
      "weighted_auxiliary_loss": 0.0005001712619559839,
      "weighted_relation_loss": 0.0005001712619559839
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.761459307764266,
      "flow_matching_loss": 0.028378243185579778,
      "grad_norm": 0.1277676373720169,
      "learning_rate": 4.984016485651578e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000547155091771856,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.213639450073245,
      "state_query_norm": 1417.6322265625,
      "state_relation_loss": 0.000547155091771856,
      "step": 15780,
      "student_distance_mean": 1936.5590087890625,
      "teacher_distance_mean": 1.1550719022750855,
      "weighted_auxiliary_loss": 0.000547155091771856,
      "weighted_relation_loss": 0.000547155091771856
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.770813844714686,
      "flow_matching_loss": 0.029167911596596242,
      "grad_norm": 0.10223197937011719,
      "learning_rate": 4.9785049585807274e-05,
      "loss": 0.0297,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004936375480610877,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.296983337402345,
      "state_query_norm": 1421.7226806640624,
      "state_relation_loss": 0.0004936375480610877,
      "step": 15790,
      "student_distance_mean": 1942.1890991210937,
      "teacher_distance_mean": 1.1548388600349426,
      "weighted_auxiliary_loss": 0.0004936375480610877,
      "weighted_relation_loss": 0.0004936375480610877
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.780168381665108,
      "flow_matching_loss": 0.030774985812604426,
      "grad_norm": 0.10505787283182144,
      "learning_rate": 4.972993457628381e-05,
      "loss": 0.0313,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005409920384408906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64654655456543,
      "state_query_norm": 1432.1051513671875,
      "state_relation_loss": 0.0005409920384408906,
      "step": 15800,
      "student_distance_mean": 1957.6954467773437,
      "teacher_distance_mean": 1.1660652995109557,
      "weighted_auxiliary_loss": 0.0005409920384408906,
      "weighted_relation_loss": 0.0005409920384408906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.789522918615528,
      "flow_matching_loss": 0.029643517173826694,
      "grad_norm": 0.11742287874221802,
      "learning_rate": 4.9674819894915306e-05,
      "loss": 0.0302,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005308237799908966,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56661262512207,
      "state_query_norm": 1429.301123046875,
      "state_relation_loss": 0.0005308237799908966,
      "step": 15810,
      "student_distance_mean": 1955.094873046875,
      "teacher_distance_mean": 1.1630871772766114,
      "weighted_auxiliary_loss": 0.0005308237799908966,
      "weighted_relation_loss": 0.0005308237799908966
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.79887745556595,
      "flow_matching_loss": 0.03053261935710907,
      "grad_norm": 0.12150681763887405,
      "learning_rate": 4.9619705608671265e-05,
      "loss": 0.0311,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005242740415269509,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45210266113281,
      "state_query_norm": 1424.6639770507813,
      "state_relation_loss": 0.0005242740415269509,
      "step": 15820,
      "student_distance_mean": 1948.8415893554688,
      "teacher_distance_mean": 1.1591346502304076,
      "weighted_auxiliary_loss": 0.0005242740415269509,
      "weighted_relation_loss": 0.0005242740415269509
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.80823199251637,
      "flow_matching_loss": 0.028738947585225105,
      "grad_norm": 0.1096685603260994,
      "learning_rate": 4.956459178452079e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004878844367340207,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.90839767456055,
      "state_query_norm": 1412.292626953125,
      "state_relation_loss": 0.0004878844367340207,
      "step": 15830,
      "student_distance_mean": 1922.3176147460938,
      "teacher_distance_mean": 1.143571925163269,
      "weighted_auxiliary_loss": 0.0004878844367340207,
      "weighted_relation_loss": 0.0004878844367340207
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.81758652946679,
      "flow_matching_loss": 0.028647392615675928,
      "grad_norm": 0.09362609684467316,
      "learning_rate": 4.950947848943235e-05,
      "loss": 0.0291,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004985990497516468,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28780632019043,
      "state_query_norm": 1427.1060913085937,
      "state_relation_loss": 0.0004985990497516468,
      "step": 15840,
      "student_distance_mean": 1945.6337036132813,
      "teacher_distance_mean": 1.1586223721504212,
      "weighted_auxiliary_loss": 0.0004985990497516468,
      "weighted_relation_loss": 0.0004985990497516468
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.826941066417213,
      "flow_matching_loss": 0.0288264038041234,
      "grad_norm": 0.13880769908428192,
      "learning_rate": 4.9454365790373805e-05,
      "loss": 0.0293,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004920547187794,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.8038932800293,
      "state_query_norm": 1400.8635864257812,
      "state_relation_loss": 0.0004920547187794,
      "step": 15850,
      "student_distance_mean": 1912.3990112304687,
      "teacher_distance_mean": 1.1419121146202087,
      "weighted_auxiliary_loss": 0.0004920547187794,
      "weighted_relation_loss": 0.0004920547187794
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.836295603367633,
      "flow_matching_loss": 0.029910944402217865,
      "grad_norm": 0.10125421732664108,
      "learning_rate": 4.939925375431226e-05,
      "loss": 0.0304,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005077362264273688,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.022271728515626,
      "state_query_norm": 1417.3751342773437,
      "state_relation_loss": 0.0005077362264273688,
      "step": 15860,
      "student_distance_mean": 1931.29130859375,
      "teacher_distance_mean": 1.150398027896881,
      "weighted_auxiliary_loss": 0.0005077362264273688,
      "weighted_relation_loss": 0.0005077362264273688
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.845650140318055,
      "flow_matching_loss": 0.028947926685214044,
      "grad_norm": 0.14147168397903442,
      "learning_rate": 4.9344142448214046e-05,
      "loss": 0.0294,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004968585388269275,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.447096252441405,
      "state_query_norm": 1426.852001953125,
      "state_relation_loss": 0.0004968585388269275,
      "step": 15870,
      "student_distance_mean": 1948.7900024414062,
      "teacher_distance_mean": 1.1553750395774842,
      "weighted_auxiliary_loss": 0.0004968585388269275,
      "weighted_relation_loss": 0.0004968585388269275
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.855004677268475,
      "flow_matching_loss": 0.027657574974000454,
      "grad_norm": 0.09706772118806839,
      "learning_rate": 4.928903193904461e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005030509229982272,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.02061920166015,
      "state_query_norm": 1419.1754272460937,
      "state_relation_loss": 0.0005030509229982272,
      "step": 15880,
      "student_distance_mean": 1932.5878051757813,
      "teacher_distance_mean": 1.1435465693473816,
      "weighted_auxiliary_loss": 0.0005030509229982272,
      "weighted_relation_loss": 0.0005030509229982272
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.864359214218895,
      "flow_matching_loss": 0.028923235088586807,
      "grad_norm": 0.11123327165842056,
      "learning_rate": 4.923392229376842e-05,
      "loss": 0.0294,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004949870897689834,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.78589286804199,
      "state_query_norm": 1407.0119506835938,
      "state_relation_loss": 0.0004949870897689834,
      "step": 15890,
      "student_distance_mean": 1916.7390869140625,
      "teacher_distance_mean": 1.1377480506896973,
      "weighted_auxiliary_loss": 0.0004949870897689834,
      "weighted_relation_loss": 0.0004949870897689834
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.873713751169317,
      "flow_matching_loss": 0.028654466569423675,
      "grad_norm": 0.09466113150119781,
      "learning_rate": 4.9178813579348865e-05,
      "loss": 0.0291,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048476835072506217,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.119314956665036,
      "state_query_norm": 1409.1743774414062,
      "state_relation_loss": 0.00048476835072506217,
      "step": 15900,
      "student_distance_mean": 1929.1981689453125,
      "teacher_distance_mean": 1.1447405695915223,
      "weighted_auxiliary_loss": 0.00048476835072506217,
      "weighted_relation_loss": 0.00048476835072506217
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.883068288119738,
      "flow_matching_loss": 0.027463018521666528,
      "grad_norm": 0.1039196252822876,
      "learning_rate": 4.9123705862748253e-05,
      "loss": 0.028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000530710507882759,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23333930969238,
      "state_query_norm": 1415.94619140625,
      "state_relation_loss": 0.000530710507882759,
      "step": 15910,
      "student_distance_mean": 1937.6813720703126,
      "teacher_distance_mean": 1.15247585773468,
      "weighted_auxiliary_loss": 0.000530710507882759,
      "weighted_relation_loss": 0.000530710507882759
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.89242282507016,
      "flow_matching_loss": 0.028222352638840674,
      "grad_norm": 0.0891762301325798,
      "learning_rate": 4.9068599210927627e-05,
      "loss": 0.0287,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005255201569525525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42467994689942,
      "state_query_norm": 1429.2515502929687,
      "state_relation_loss": 0.0005255201569525525,
      "step": 15920,
      "student_distance_mean": 1948.1635986328124,
      "teacher_distance_mean": 1.1610270619392395,
      "weighted_auxiliary_loss": 0.0005255201569525525,
      "weighted_relation_loss": 0.0005255201569525525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.90177736202058,
      "flow_matching_loss": 0.029985067434608936,
      "grad_norm": 0.10331597924232483,
      "learning_rate": 4.901349369084681e-05,
      "loss": 0.0305,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048776143521536144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.071284484863284,
      "state_query_norm": 1408.0977416992187,
      "state_relation_loss": 0.00048776143521536144,
      "step": 15930,
      "student_distance_mean": 1931.7836791992188,
      "teacher_distance_mean": 1.1513140201568604,
      "weighted_auxiliary_loss": 0.00048776143521536144,
      "weighted_relation_loss": 0.00048776143521536144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.911131898971,
      "flow_matching_loss": 0.02889448180794716,
      "grad_norm": 0.11339342594146729,
      "learning_rate": 4.8958389369464156e-05,
      "loss": 0.0294,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000518247700529173,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06737442016602,
      "state_query_norm": 1422.8368408203125,
      "state_relation_loss": 0.000518247700529173,
      "step": 15940,
      "student_distance_mean": 1932.9853637695312,
      "teacher_distance_mean": 1.1491940140724182,
      "weighted_auxiliary_loss": 0.000518247700529173,
      "weighted_relation_loss": 0.000518247700529173
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.920486435921422,
      "flow_matching_loss": 0.030333887599408627,
      "grad_norm": 0.12547039985656738,
      "learning_rate": 4.890328631373666e-05,
      "loss": 0.0309,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005266385822324082,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27965774536133,
      "state_query_norm": 1417.5209228515625,
      "state_relation_loss": 0.0005266385822324082,
      "step": 15950,
      "student_distance_mean": 1941.3046752929688,
      "teacher_distance_mean": 1.153541338443756,
      "weighted_auxiliary_loss": 0.0005266385822324082,
      "weighted_relation_loss": 0.0005266385822324082
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.929840972871842,
      "flow_matching_loss": 0.028975820168852806,
      "grad_norm": 0.08714404702186584,
      "learning_rate": 4.88481845906197e-05,
      "loss": 0.0295,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005019273259676993,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37142066955566,
      "state_query_norm": 1426.3163696289062,
      "state_relation_loss": 0.0005019273259676993,
      "step": 15960,
      "student_distance_mean": 1946.0395874023438,
      "teacher_distance_mean": 1.1621647715568542,
      "weighted_auxiliary_loss": 0.0005019273259676993,
      "weighted_relation_loss": 0.0005019273259676993
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.939195509822264,
      "flow_matching_loss": 0.029646438732743263,
      "grad_norm": 0.11171609908342361,
      "learning_rate": 4.879308426706707e-05,
      "loss": 0.0301,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004988841857993976,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.701779174804685,
      "state_query_norm": 1401.2440307617187,
      "state_relation_loss": 0.0004988841857993976,
      "step": 15970,
      "student_distance_mean": 1909.8053588867188,
      "teacher_distance_mean": 1.1432103157043456,
      "weighted_auxiliary_loss": 0.0004988841857993976,
      "weighted_relation_loss": 0.0004988841857993976
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.948550046772684,
      "flow_matching_loss": 0.029405831545591354,
      "grad_norm": 0.082963727414608,
      "learning_rate": 4.873798541003084e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048411593015771357,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.03878936767578,
      "state_query_norm": 1414.1253540039063,
      "state_relation_loss": 0.00048411593015771357,
      "step": 15980,
      "student_distance_mean": 1931.362353515625,
      "teacher_distance_mean": 1.152388048171997,
      "weighted_auxiliary_loss": 0.00048411593015771357,
      "weighted_relation_loss": 0.00048411593015771357
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.957904583723106,
      "flow_matching_loss": 0.03009659368544817,
      "grad_norm": 0.134246826171875,
      "learning_rate": 4.868288808646136e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000500800961162895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.88277359008789,
      "state_query_norm": 1403.3466796875,
      "state_relation_loss": 0.000500800961162895,
      "step": 15990,
      "student_distance_mean": 1919.6982299804688,
      "teacher_distance_mean": 1.14659526348114,
      "weighted_auxiliary_loss": 0.000500800961162895,
      "weighted_relation_loss": 0.000500800961162895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.967259120673527,
      "flow_matching_loss": 0.027842212840914725,
      "grad_norm": 0.11800261586904526,
      "learning_rate": 4.862779236330705e-05,
      "loss": 0.0284,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005183569941436872,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83064193725586,
      "state_query_norm": 1449.2681884765625,
      "state_relation_loss": 0.0005183569941436872,
      "step": 16000,
      "student_distance_mean": 1973.6837280273437,
      "teacher_distance_mean": 1.1734771966934203,
      "weighted_auxiliary_loss": 0.0005183569941436872,
      "weighted_relation_loss": 0.0005183569941436872
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.976613657623947,
      "flow_matching_loss": 0.02986608799546957,
      "grad_norm": 0.12171231210231781,
      "learning_rate": 4.8572698307514395e-05,
      "loss": 0.0303,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004748862440465018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92613296508789,
      "state_query_norm": 1407.1295532226563,
      "state_relation_loss": 0.0004748862440465018,
      "step": 16010,
      "student_distance_mean": 1921.2642578125,
      "teacher_distance_mean": 1.14381023645401,
      "weighted_auxiliary_loss": 0.0004748862440465018,
      "weighted_relation_loss": 0.0004748862440465018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.985968194574369,
      "flow_matching_loss": 0.027164280228316785,
      "grad_norm": 0.10542885959148407,
      "learning_rate": 4.85176059860279e-05,
      "loss": 0.0277,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000510314924758859,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17578353881836,
      "state_query_norm": 1423.8810791015626,
      "state_relation_loss": 0.000510314924758859,
      "step": 16020,
      "student_distance_mean": 1936.5999389648437,
      "teacher_distance_mean": 1.1535048484802246,
      "weighted_auxiliary_loss": 0.000510314924758859,
      "weighted_relation_loss": 0.000510314924758859
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 14.995322731524789,
      "flow_matching_loss": 0.03034239634871483,
      "grad_norm": 0.11019602417945862,
      "learning_rate": 4.846251546578989e-05,
      "loss": 0.0308,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004843092930968851,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.122778701782224,
      "state_query_norm": 1419.487158203125,
      "state_relation_loss": 0.0004843092930968851,
      "step": 16030,
      "student_distance_mean": 1933.9932861328125,
      "teacher_distance_mean": 1.1509062051773071,
      "weighted_auxiliary_loss": 0.0004843092930968851,
      "weighted_relation_loss": 0.0004843092930968851
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.004677268475211,
      "flow_matching_loss": 0.02864073868840933,
      "grad_norm": 0.12332554906606674,
      "learning_rate": 4.840742681374058e-05,
      "loss": 0.0291,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00048383501125499607,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.54776878356934,
      "state_query_norm": 1435.3823364257812,
      "state_relation_loss": 0.00048383501125499607,
      "step": 16040,
      "student_distance_mean": 1958.1702514648437,
      "teacher_distance_mean": 1.1614103198051453,
      "weighted_auxiliary_loss": 0.00048383501125499607,
      "weighted_relation_loss": 0.00048383501125499607
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.014031805425631,
      "flow_matching_loss": 0.027041305601596833,
      "grad_norm": 0.11820006370544434,
      "learning_rate": 4.835234009681787e-05,
      "loss": 0.0275,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004970794718246907,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.98852310180664,
      "state_query_norm": 1406.5675537109375,
      "state_relation_loss": 0.0004970794718246907,
      "step": 16050,
      "student_distance_mean": 1925.263232421875,
      "teacher_distance_mean": 1.1452187895774841,
      "weighted_auxiliary_loss": 0.0004970794718246907,
      "weighted_relation_loss": 0.0004970794718246907
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.023386342376053,
      "flow_matching_loss": 0.030729323998093604,
      "grad_norm": 0.10001420974731445,
      "learning_rate": 4.8297255381957296e-05,
      "loss": 0.0312,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045973363448865713,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.788540267944335,
      "state_query_norm": 1409.9435913085938,
      "state_relation_loss": 0.00045973363448865713,
      "step": 16060,
      "student_distance_mean": 1917.6470825195313,
      "teacher_distance_mean": 1.146164345741272,
      "weighted_auxiliary_loss": 0.00045973363448865713,
      "weighted_relation_loss": 0.00045973363448865713
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.032740879326473,
      "flow_matching_loss": 0.028109118714928626,
      "grad_norm": 0.11700980365276337,
      "learning_rate": 4.824217273609198e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046875928528606893,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30515251159668,
      "state_query_norm": 1428.9193237304687,
      "state_relation_loss": 0.00046875928528606893,
      "step": 16070,
      "student_distance_mean": 1947.4598022460937,
      "teacher_distance_mean": 1.1629253387451173,
      "weighted_auxiliary_loss": 0.00046875928528606893,
      "weighted_relation_loss": 0.00046875928528606893
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.042095416276894,
      "flow_matching_loss": 0.027853473089635372,
      "grad_norm": 0.11571644246578217,
      "learning_rate": 4.8187092226152555e-05,
      "loss": 0.0283,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004890984535450115,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.64442863464355,
      "state_query_norm": 1391.5280517578126,
      "state_relation_loss": 0.0004890984535450115,
      "step": 16080,
      "student_distance_mean": 1905.5910888671874,
      "teacher_distance_mean": 1.1378436923027038,
      "weighted_auxiliary_loss": 0.0004890984535450115,
      "weighted_relation_loss": 0.0004890984535450115
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.051449953227316,
      "flow_matching_loss": 0.027742341347038745,
      "grad_norm": 0.09448558837175369,
      "learning_rate": 4.8132013919067016e-05,
      "loss": 0.0283,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000515123238437809,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.79027404785156,
      "state_query_norm": 1438.5301879882813,
      "state_relation_loss": 0.000515123238437809,
      "step": 16090,
      "student_distance_mean": 1965.5259399414062,
      "teacher_distance_mean": 1.1719324231147765,
      "weighted_auxiliary_loss": 0.000515123238437809,
      "weighted_relation_loss": 0.000515123238437809
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.060804490177736,
      "flow_matching_loss": 0.02793050054460764,
      "grad_norm": 0.09350695461034775,
      "learning_rate": 4.807693788176071e-05,
      "loss": 0.0284,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000466463845805265,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.59001922607422,
      "state_query_norm": 1406.0145263671875,
      "state_relation_loss": 0.000466463845805265,
      "step": 16100,
      "student_distance_mean": 1910.5456420898438,
      "teacher_distance_mean": 1.1389084696769713,
      "weighted_auxiliary_loss": 0.000466463845805265,
      "weighted_relation_loss": 0.000466463845805265
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.070159027128158,
      "flow_matching_loss": 0.027999729476869108,
      "grad_norm": 0.10329002141952515,
      "learning_rate": 4.8021864181156214e-05,
      "loss": 0.0285,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047654664958827195,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.979051971435545,
      "state_query_norm": 1420.465771484375,
      "state_relation_loss": 0.00047654664958827195,
      "step": 16110,
      "student_distance_mean": 1927.1418212890626,
      "teacher_distance_mean": 1.1468547701835632,
      "weighted_auxiliary_loss": 0.00047654664958827195,
      "weighted_relation_loss": 0.00047654664958827195
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.079513564078578,
      "flow_matching_loss": 0.028436879627406597,
      "grad_norm": 0.09673386067152023,
      "learning_rate": 4.796679288417326e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004924085573293268,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.403421783447264,
      "state_query_norm": 1426.5679443359375,
      "state_relation_loss": 0.0004924085573293268,
      "step": 16120,
      "student_distance_mean": 1944.6192260742187,
      "teacher_distance_mean": 1.1609402656555177,
      "weighted_auxiliary_loss": 0.0004924085573293268,
      "weighted_relation_loss": 0.0004924085573293268
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.088868101028998,
      "flow_matching_loss": 0.026272977143526076,
      "grad_norm": 0.09822241961956024,
      "learning_rate": 4.791172405772866e-05,
      "loss": 0.0268,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005437055166112259,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56694793701172,
      "state_query_norm": 1425.8004028320313,
      "state_relation_loss": 0.0005437055166112259,
      "step": 16130,
      "student_distance_mean": 1954.2975463867188,
      "teacher_distance_mean": 1.160403025150299,
      "weighted_auxiliary_loss": 0.0005437055166112259,
      "weighted_relation_loss": 0.0005437055166112259
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.09822263797942,
      "flow_matching_loss": 0.026379364356398583,
      "grad_norm": 0.10228253901004791,
      "learning_rate": 4.785665776873625e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004854892933508381,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45245513916016,
      "state_query_norm": 1423.8022827148438,
      "state_relation_loss": 0.0004854892933508381,
      "step": 16140,
      "student_distance_mean": 1948.9113159179688,
      "teacher_distance_mean": 1.158354914188385,
      "weighted_auxiliary_loss": 0.0004854892933508381,
      "weighted_relation_loss": 0.0004854892933508381
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.10757717492984,
      "flow_matching_loss": 0.02941932138055563,
      "grad_norm": 0.09168631583452225,
      "learning_rate": 4.7801594084106763e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004976420954335481,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59146499633789,
      "state_query_norm": 1438.830712890625,
      "state_relation_loss": 0.0004976420954335481,
      "step": 16150,
      "student_distance_mean": 1958.7694946289062,
      "teacher_distance_mean": 1.163679575920105,
      "weighted_auxiliary_loss": 0.0004976420954335481,
      "weighted_relation_loss": 0.0004976420954335481
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.116931711880262,
      "flow_matching_loss": 0.02808550652116537,
      "grad_norm": 0.09768249839544296,
      "learning_rate": 4.774653307074775e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047368168889079244,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44244117736817,
      "state_query_norm": 1432.0854858398438,
      "state_relation_loss": 0.00047368168889079244,
      "step": 16160,
      "student_distance_mean": 1954.1743286132812,
      "teacher_distance_mean": 1.1600636839866638,
      "weighted_auxiliary_loss": 0.00047368168889079244,
      "weighted_relation_loss": 0.00047368168889079244
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.126286248830683,
      "flow_matching_loss": 0.02582202199846506,
      "grad_norm": 0.12732793390750885,
      "learning_rate": 4.769147479556355e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047940344957169144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53505897521973,
      "state_query_norm": 1428.5037963867187,
      "state_relation_loss": 0.00047940344957169144,
      "step": 16170,
      "student_distance_mean": 1956.3953369140625,
      "teacher_distance_mean": 1.1611915230751038,
      "weighted_auxiliary_loss": 0.00047940344957169144,
      "weighted_relation_loss": 0.00047940344957169144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.135640785781105,
      "flow_matching_loss": 0.02814140245318413,
      "grad_norm": 0.11996657401323318,
      "learning_rate": 4.763641932545515e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004533927538432181,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.874077224731444,
      "state_query_norm": 1411.2532958984375,
      "state_relation_loss": 0.0004533927538432181,
      "step": 16180,
      "student_distance_mean": 1920.2768310546876,
      "teacher_distance_mean": 1.1410731315612792,
      "weighted_auxiliary_loss": 0.0004533927538432181,
      "weighted_relation_loss": 0.0004533927538432181
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.144995322731525,
      "flow_matching_loss": 0.02821698673069477,
      "grad_norm": 0.11359795182943344,
      "learning_rate": 4.758136672732013e-05,
      "loss": 0.0287,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004895390098681674,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.91112937927246,
      "state_query_norm": 1410.0628662109375,
      "state_relation_loss": 0.0004895390098681674,
      "step": 16190,
      "student_distance_mean": 1921.0631103515625,
      "teacher_distance_mean": 1.1460665464401245,
      "weighted_auxiliary_loss": 0.0004895390098681674,
      "weighted_relation_loss": 0.0004895390098681674
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.154349859681945,
      "flow_matching_loss": 0.027124506793916224,
      "grad_norm": 0.12248203158378601,
      "learning_rate": 4.752631706805261e-05,
      "loss": 0.0276,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004690652625868097,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.469526290893555,
      "state_query_norm": 1398.0611328125,
      "state_relation_loss": 0.0004690652625868097,
      "step": 16200,
      "student_distance_mean": 1901.3312255859375,
      "teacher_distance_mean": 1.1308271288871765,
      "weighted_auxiliary_loss": 0.0004690652625868097,
      "weighted_relation_loss": 0.0004690652625868097
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.163704396632367,
      "flow_matching_loss": 0.02830943949520588,
      "grad_norm": 0.10511212795972824,
      "learning_rate": 4.7471270414543114e-05,
      "loss": 0.0288,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004627555987099186,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.99526824951172,
      "state_query_norm": 1415.567333984375,
      "state_relation_loss": 0.0004627555987099186,
      "step": 16210,
      "student_distance_mean": 1927.8362182617188,
      "teacher_distance_mean": 1.146773111820221,
      "weighted_auxiliary_loss": 0.0004627555987099186,
      "weighted_relation_loss": 0.0004627555987099186
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.173058933582787,
      "flow_matching_loss": 0.02794403750449419,
      "grad_norm": 0.12625710666179657,
      "learning_rate": 4.741622683367849e-05,
      "loss": 0.0284,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004992816626327113,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69033279418945,
      "state_query_norm": 1427.7889282226563,
      "state_relation_loss": 0.0004992816626327113,
      "step": 16220,
      "student_distance_mean": 1960.13193359375,
      "teacher_distance_mean": 1.1666299819946289,
      "weighted_auxiliary_loss": 0.0004992816626327113,
      "weighted_relation_loss": 0.0004992816626327113
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.18241347053321,
      "flow_matching_loss": 0.029153533093631268,
      "grad_norm": 0.08321970701217651,
      "learning_rate": 4.736118639234191e-05,
      "loss": 0.0296,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004712043621111661,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10036087036133,
      "state_query_norm": 1411.6328125,
      "state_relation_loss": 0.0004712043621111661,
      "step": 16230,
      "student_distance_mean": 1932.0209838867188,
      "teacher_distance_mean": 1.1513285517692566,
      "weighted_auxiliary_loss": 0.0004712043621111661,
      "weighted_relation_loss": 0.0004712043621111661
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.19176800748363,
      "flow_matching_loss": 0.024454656429588793,
      "grad_norm": 0.08925404399633408,
      "learning_rate": 4.730614915741267e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005063443648396059,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.027863311767575,
      "state_query_norm": 1412.0048828125,
      "state_relation_loss": 0.0005063443648396059,
      "step": 16240,
      "student_distance_mean": 1927.740673828125,
      "teacher_distance_mean": 1.1508715510368348,
      "weighted_auxiliary_loss": 0.0005063443648396059,
      "weighted_relation_loss": 0.0005063443648396059
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.20112254443405,
      "flow_matching_loss": 0.027617665007710457,
      "grad_norm": 0.09883245080709457,
      "learning_rate": 4.7251115195766234e-05,
      "loss": 0.0281,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005158983869478107,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.19669647216797,
      "state_query_norm": 1414.30537109375,
      "state_relation_loss": 0.0005158983869478107,
      "step": 16250,
      "student_distance_mean": 1932.944091796875,
      "teacher_distance_mean": 1.1553025841712952,
      "weighted_auxiliary_loss": 0.0005158983869478107,
      "weighted_relation_loss": 0.0005158983869478107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.210477081384472,
      "flow_matching_loss": 0.027371766045689583,
      "grad_norm": 0.09988981485366821,
      "learning_rate": 4.719608457427404e-05,
      "loss": 0.0279,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004883991932729259,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26255569458008,
      "state_query_norm": 1424.241845703125,
      "state_relation_loss": 0.0004883991932729259,
      "step": 16260,
      "student_distance_mean": 1942.9244506835937,
      "teacher_distance_mean": 1.1582910656929015,
      "weighted_auxiliary_loss": 0.0004883991932729259,
      "weighted_relation_loss": 0.0004883991932729259
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.219831618334892,
      "flow_matching_loss": 0.02875081393867731,
      "grad_norm": 0.08991867303848267,
      "learning_rate": 4.714105735980347e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004641151143005118,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.753503036499026,
      "state_query_norm": 1403.7644287109374,
      "state_relation_loss": 0.0004641151143005118,
      "step": 16270,
      "student_distance_mean": 1916.8699340820312,
      "teacher_distance_mean": 1.1398293614387511,
      "weighted_auxiliary_loss": 0.0004641151143005118,
      "weighted_relation_loss": 0.0004641151143005118
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.229186155285314,
      "flow_matching_loss": 0.02717972230166197,
      "grad_norm": 0.11629611998796463,
      "learning_rate": 4.70860336192178e-05,
      "loss": 0.0277,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004970023466739804,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59835357666016,
      "state_query_norm": 1432.2297729492188,
      "state_relation_loss": 0.0004970023466739804,
      "step": 16280,
      "student_distance_mean": 1957.8679809570312,
      "teacher_distance_mean": 1.1674896359443665,
      "weighted_auxiliary_loss": 0.0004970023466739804,
      "weighted_relation_loss": 0.0004970023466739804
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.238540692235734,
      "flow_matching_loss": 0.02570637073367834,
      "grad_norm": 0.10492872446775436,
      "learning_rate": 4.7031013419376034e-05,
      "loss": 0.0262,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048786171537358315,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10681686401367,
      "state_query_norm": 1419.1729248046875,
      "state_relation_loss": 0.00048786171537358315,
      "step": 16290,
      "student_distance_mean": 1933.1083251953125,
      "teacher_distance_mean": 1.1566582083702088,
      "weighted_auxiliary_loss": 0.00048786171537358315,
      "weighted_relation_loss": 0.00048786171537358315
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.247895229186156,
      "flow_matching_loss": 0.02559781037271023,
      "grad_norm": 0.10093951225280762,
      "learning_rate": 4.6975996827132926e-05,
      "loss": 0.0261,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004763749369885772,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45155792236328,
      "state_query_norm": 1423.7398681640625,
      "state_relation_loss": 0.0004763749369885772,
      "step": 16300,
      "student_distance_mean": 1949.9459716796875,
      "teacher_distance_mean": 1.1660139203071593,
      "weighted_auxiliary_loss": 0.0004763749369885772,
      "weighted_relation_loss": 0.0004763749369885772
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.257249766136576,
      "flow_matching_loss": 0.02725777346640825,
      "grad_norm": 0.10084733366966248,
      "learning_rate": 4.692098390933883e-05,
      "loss": 0.0277,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045944626326672734,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01633491516113,
      "state_query_norm": 1411.1366088867187,
      "state_relation_loss": 0.00045944626326672734,
      "step": 16310,
      "student_distance_mean": 1929.0833862304687,
      "teacher_distance_mean": 1.1495417594909667,
      "weighted_auxiliary_loss": 0.00045944626326672734,
      "weighted_relation_loss": 0.00045944626326672734
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.266604303086996,
      "flow_matching_loss": 0.02925680186599493,
      "grad_norm": 0.08933261781930923,
      "learning_rate": 4.6865974732839615e-05,
      "loss": 0.0298,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004995889175916091,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28987693786621,
      "state_query_norm": 1419.8323486328125,
      "state_relation_loss": 0.0004995889175916091,
      "step": 16320,
      "student_distance_mean": 1941.4595092773438,
      "teacher_distance_mean": 1.158387267589569,
      "weighted_auxiliary_loss": 0.0004995889175916091,
      "weighted_relation_loss": 0.0004995889175916091
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.275958840037418,
      "flow_matching_loss": 0.029532848671078683,
      "grad_norm": 0.10827236622571945,
      "learning_rate": 4.6810969364476615e-05,
      "loss": 0.03,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046643943642266095,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.62420806884766,
      "state_query_norm": 1402.431005859375,
      "state_relation_loss": 0.00046643943642266095,
      "step": 16330,
      "student_distance_mean": 1909.0571166992188,
      "teacher_distance_mean": 1.1387216687202453,
      "weighted_auxiliary_loss": 0.00046643943642266095,
      "weighted_relation_loss": 0.00046643943642266095
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.285313376987839,
      "flow_matching_loss": 0.02920148931443691,
      "grad_norm": 0.0985669270157814,
      "learning_rate": 4.675596787108653e-05,
      "loss": 0.0297,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00049897680291906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23982315063476,
      "state_query_norm": 1417.5864013671876,
      "state_relation_loss": 0.00049897680291906,
      "step": 16340,
      "student_distance_mean": 1936.4914428710938,
      "teacher_distance_mean": 1.1530326843261718,
      "weighted_auxiliary_loss": 0.00049897680291906,
      "weighted_relation_loss": 0.00049897680291906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.29466791393826,
      "flow_matching_loss": 0.027544601261615752,
      "grad_norm": 0.09245248138904572,
      "learning_rate": 4.670097031950139e-05,
      "loss": 0.028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004921308573102579,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.8006591796875,
      "state_query_norm": 1406.2022216796875,
      "state_relation_loss": 0.0004921308573102579,
      "step": 16350,
      "student_distance_mean": 1917.9889038085937,
      "teacher_distance_mean": 1.141370689868927,
      "weighted_auxiliary_loss": 0.0004921308573102579,
      "weighted_relation_loss": 0.0004921308573102579
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.30402245088868,
      "flow_matching_loss": 0.02710518892854452,
      "grad_norm": 0.11500230431556702,
      "learning_rate": 4.664597677654838e-05,
      "loss": 0.0276,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004510085680522025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.958824157714844,
      "state_query_norm": 1413.9307495117187,
      "state_relation_loss": 0.0004510085680522025,
      "step": 16360,
      "student_distance_mean": 1924.8819946289063,
      "teacher_distance_mean": 1.140930426120758,
      "weighted_auxiliary_loss": 0.0004510085680522025,
      "weighted_relation_loss": 0.0004510085680522025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.313376987839103,
      "flow_matching_loss": 0.03004648368805647,
      "grad_norm": 0.12841852009296417,
      "learning_rate": 4.659098730904986e-05,
      "loss": 0.0306,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005062178854132072,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76037826538086,
      "state_query_norm": 1440.6818237304688,
      "state_relation_loss": 0.0005062178854132072,
      "step": 16370,
      "student_distance_mean": 1965.22802734375,
      "teacher_distance_mean": 1.165741539001465,
      "weighted_auxiliary_loss": 0.0005062178854132072,
      "weighted_relation_loss": 0.0005062178854132072
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.322731524789523,
      "flow_matching_loss": 0.028252633288502693,
      "grad_norm": 0.11199397593736649,
      "learning_rate": 4.6536001983823204e-05,
      "loss": 0.0287,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004840938840061426,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.77850761413574,
      "state_query_norm": 1441.9125,
      "state_relation_loss": 0.0004840938840061426,
      "step": 16380,
      "student_distance_mean": 1969.5108276367187,
      "teacher_distance_mean": 1.172071623802185,
      "weighted_auxiliary_loss": 0.0004840938840061426,
      "weighted_relation_loss": 0.0004840938840061426
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.332086061739943,
      "flow_matching_loss": 0.026054799370467664,
      "grad_norm": 0.10098925232887268,
      "learning_rate": 4.648102086768078e-05,
      "loss": 0.0266,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004961096419719979,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36753387451172,
      "state_query_norm": 1414.530029296875,
      "state_relation_loss": 0.0004961096419719979,
      "step": 16390,
      "student_distance_mean": 1939.613720703125,
      "teacher_distance_mean": 1.154972243309021,
      "weighted_auxiliary_loss": 0.0004961096419719979,
      "weighted_relation_loss": 0.0004961096419719979
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.341440598690365,
      "flow_matching_loss": 0.027158686891198158,
      "grad_norm": 0.10406485944986343,
      "learning_rate": 4.6426044027429786e-05,
      "loss": 0.0276,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004848242475418374,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.784262084960936,
      "state_query_norm": 1440.575390625,
      "state_relation_loss": 0.0004848242475418374,
      "step": 16400,
      "student_distance_mean": 1967.2169799804688,
      "teacher_distance_mean": 1.1684903025627136,
      "weighted_auxiliary_loss": 0.0004848242475418374,
      "weighted_relation_loss": 0.0004848242475418374
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.350795135640785,
      "flow_matching_loss": 0.026528405398130415,
      "grad_norm": 0.1062859445810318,
      "learning_rate": 4.637107152987234e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004966790380422026,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.97831840515137,
      "state_query_norm": 1409.1032104492188,
      "state_relation_loss": 0.0004966790380422026,
      "step": 16410,
      "student_distance_mean": 1926.5969848632812,
      "teacher_distance_mean": 1.146743154525757,
      "weighted_auxiliary_loss": 0.0004966790380422026,
      "weighted_relation_loss": 0.0004966790380422026
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.360149672591207,
      "flow_matching_loss": 0.028940534219145775,
      "grad_norm": 0.12715229392051697,
      "learning_rate": 4.631610344180515e-05,
      "loss": 0.0294,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004642328480258584,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75350799560547,
      "state_query_norm": 1406.2587036132813,
      "state_relation_loss": 0.0004642328480258584,
      "step": 16420,
      "student_distance_mean": 1916.66279296875,
      "teacher_distance_mean": 1.1430370569229127,
      "weighted_auxiliary_loss": 0.0004642328480258584,
      "weighted_relation_loss": 0.0004642328480258584
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.369504209541628,
      "flow_matching_loss": 0.02603644262999296,
      "grad_norm": 0.11175481230020523,
      "learning_rate": 4.626113983001965e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004966487787896768,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.268904495239255,
      "state_query_norm": 1421.5750610351563,
      "state_relation_loss": 0.0004966487787896768,
      "step": 16430,
      "student_distance_mean": 1941.3872314453124,
      "teacher_distance_mean": 1.154387855529785,
      "weighted_auxiliary_loss": 0.0004966487787896768,
      "weighted_relation_loss": 0.0004966487787896768
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.378858746492048,
      "flow_matching_loss": 0.02842337340116501,
      "grad_norm": 0.10902811586856842,
      "learning_rate": 4.620618076130182e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004758758615935221,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.35956153869629,
      "state_query_norm": 1426.3990478515625,
      "state_relation_loss": 0.0004758758615935221,
      "step": 16440,
      "student_distance_mean": 1943.7812866210938,
      "teacher_distance_mean": 1.154762077331543,
      "weighted_auxiliary_loss": 0.0004758758615935221,
      "weighted_relation_loss": 0.0004758758615935221
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.38821328344247,
      "flow_matching_loss": 0.028372754529118537,
      "grad_norm": 0.1076693907380104,
      "learning_rate": 4.6151226302432074e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004835124971577898,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30622444152832,
      "state_query_norm": 1419.4232177734375,
      "state_relation_loss": 0.0004835124971577898,
      "step": 16450,
      "student_distance_mean": 1941.8928833007812,
      "teacher_distance_mean": 1.1528219699859619,
      "weighted_auxiliary_loss": 0.0004835124971577898,
      "weighted_relation_loss": 0.0004835124971577898
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.39756782039289,
      "flow_matching_loss": 0.02944575697183609,
      "grad_norm": 0.09897798299789429,
      "learning_rate": 4.6096276520185304e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000466052841511555,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.679928588867185,
      "state_query_norm": 1439.1095092773437,
      "state_relation_loss": 0.000466052841511555,
      "step": 16460,
      "student_distance_mean": 1965.2119873046875,
      "teacher_distance_mean": 1.1674578547477723,
      "weighted_auxiliary_loss": 0.000466052841511555,
      "weighted_relation_loss": 0.000466052841511555
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.406922357343312,
      "flow_matching_loss": 0.02772866450250149,
      "grad_norm": 0.09397000074386597,
      "learning_rate": 4.604133148133066e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048792262386996297,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47524909973144,
      "state_query_norm": 1431.5010375976562,
      "state_relation_loss": 0.00048792262386996297,
      "step": 16470,
      "student_distance_mean": 1949.8180053710937,
      "teacher_distance_mean": 1.1533262372016906,
      "weighted_auxiliary_loss": 0.00048792262386996297,
      "weighted_relation_loss": 0.00048792262386996297
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.416276894293732,
      "flow_matching_loss": 0.027027287147939207,
      "grad_norm": 0.12232068926095963,
      "learning_rate": 4.598639125263155e-05,
      "loss": 0.0275,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046828159247525036,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30022087097168,
      "state_query_norm": 1419.2633422851563,
      "state_relation_loss": 0.00046828159247525036,
      "step": 16480,
      "student_distance_mean": 1940.7206176757813,
      "teacher_distance_mean": 1.1499439477920532,
      "weighted_auxiliary_loss": 0.00046828159247525036,
      "weighted_relation_loss": 0.00046828159247525036
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.425631431244154,
      "flow_matching_loss": 0.028693922236561774,
      "grad_norm": 0.10584106296300888,
      "learning_rate": 4.593145590084553e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004746172053273767,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21888694763184,
      "state_query_norm": 1413.0806518554687,
      "state_relation_loss": 0.0004746172053273767,
      "step": 16490,
      "student_distance_mean": 1937.9864624023437,
      "teacher_distance_mean": 1.1519448637962342,
      "weighted_auxiliary_loss": 0.0004746172053273767,
      "weighted_relation_loss": 0.0004746172053273767
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.434985968194574,
      "flow_matching_loss": 0.02651284709572792,
      "grad_norm": 0.12014210969209671,
      "learning_rate": 4.58765254927242e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046736565127503127,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.155586624145506,
      "state_query_norm": 1418.9953491210938,
      "state_relation_loss": 0.00046736565127503127,
      "step": 16500,
      "student_distance_mean": 1936.0252807617187,
      "teacher_distance_mean": 1.1561397314071655,
      "weighted_auxiliary_loss": 0.00046736565127503127,
      "weighted_relation_loss": 0.00046736565127503127
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.444340505144995,
      "flow_matching_loss": 0.026734892278909683,
      "grad_norm": 0.09174565225839615,
      "learning_rate": 4.582160009501323e-05,
      "loss": 0.0273,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005232435592915863,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.84248504638672,
      "state_query_norm": 1441.0531372070313,
      "state_relation_loss": 0.0005232435592915863,
      "step": 16510,
      "student_distance_mean": 1969.1461303710937,
      "teacher_distance_mean": 1.1728513479232787,
      "weighted_auxiliary_loss": 0.0005232435592915863,
      "weighted_relation_loss": 0.0005232435592915863
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.453695042095417,
      "flow_matching_loss": 0.02655385136604309,
      "grad_norm": 0.11706030368804932,
      "learning_rate": 4.5766679774452143e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004889951116638259,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.001041412353516,
      "state_query_norm": 1397.7968383789062,
      "state_relation_loss": 0.0004889951116638259,
      "step": 16520,
      "student_distance_mean": 1918.939697265625,
      "teacher_distance_mean": 1.1425561547279357,
      "weighted_auxiliary_loss": 0.0004889951116638259,
      "weighted_relation_loss": 0.0004889951116638259
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.463049579045837,
      "flow_matching_loss": 0.02779872193932533,
      "grad_norm": 0.1025669202208519,
      "learning_rate": 4.571176459777431e-05,
      "loss": 0.0283,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005046817532274872,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43141174316406,
      "state_query_norm": 1423.6820922851562,
      "state_relation_loss": 0.0005046817532274872,
      "step": 16530,
      "student_distance_mean": 1948.096533203125,
      "teacher_distance_mean": 1.155639111995697,
      "weighted_auxiliary_loss": 0.0005046817532274872,
      "weighted_relation_loss": 0.0005046817532274872
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.472404115996259,
      "flow_matching_loss": 0.02633440513163805,
      "grad_norm": 0.09310147166252136,
      "learning_rate": 4.5656854631706846e-05,
      "loss": 0.0268,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004889381350949406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10065460205078,
      "state_query_norm": 1417.8393310546876,
      "state_relation_loss": 0.0004889381350949406,
      "step": 16540,
      "student_distance_mean": 1933.052294921875,
      "teacher_distance_mean": 1.1468280911445619,
      "weighted_auxiliary_loss": 0.0004889381350949406,
      "weighted_relation_loss": 0.0004889381350949406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.481758652946679,
      "flow_matching_loss": 0.027997365035116673,
      "grad_norm": 0.12012093514204025,
      "learning_rate": 4.560194994297054e-05,
      "loss": 0.0285,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004945487395161763,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23609504699707,
      "state_query_norm": 1421.755078125,
      "state_relation_loss": 0.0004945487395161763,
      "step": 16550,
      "student_distance_mean": 1936.9737915039063,
      "teacher_distance_mean": 1.1502900123596191,
      "weighted_auxiliary_loss": 0.0004945487395161763,
      "weighted_relation_loss": 0.0004945487395161763
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.4911131898971,
      "flow_matching_loss": 0.02569089587777853,
      "grad_norm": 0.10332410782575607,
      "learning_rate": 4.554705059827974e-05,
      "loss": 0.0262,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005094875203212723,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.519944763183595,
      "state_query_norm": 1433.1651611328125,
      "state_relation_loss": 0.0005094875203212723,
      "step": 16560,
      "student_distance_mean": 1956.4108520507812,
      "teacher_distance_mean": 1.1637948036193848,
      "weighted_auxiliary_loss": 0.0005094875203212723,
      "weighted_relation_loss": 0.0005094875203212723
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.500467726847521,
      "flow_matching_loss": 0.027699398435652257,
      "grad_norm": 0.10845125466585159,
      "learning_rate": 4.5492156664342365e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004905667447019368,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75603675842285,
      "state_query_norm": 1405.9871459960937,
      "state_relation_loss": 0.0004905667447019368,
      "step": 16570,
      "student_distance_mean": 1913.62255859375,
      "teacher_distance_mean": 1.1373990178108215,
      "weighted_auxiliary_loss": 0.0004905667447019368,
      "weighted_relation_loss": 0.0004905667447019368
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.509822263797941,
      "flow_matching_loss": 0.02844732888042927,
      "grad_norm": 0.10019126534461975,
      "learning_rate": 4.5437268207859695e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047998307272791865,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20859794616699,
      "state_query_norm": 1418.7725952148437,
      "state_relation_loss": 0.00047998307272791865,
      "step": 16580,
      "student_distance_mean": 1935.8361938476562,
      "teacher_distance_mean": 1.150102400779724,
      "weighted_auxiliary_loss": 0.00047998307272791865,
      "weighted_relation_loss": 0.00047998307272791865
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.519176800748363,
      "flow_matching_loss": 0.027364652976393698,
      "grad_norm": 0.1108899936079979,
      "learning_rate": 4.538238529552641e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004540461057331413,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.22314262390137,
      "state_query_norm": 1417.924658203125,
      "state_relation_loss": 0.0004540461057331413,
      "step": 16590,
      "student_distance_mean": 1941.3905639648438,
      "teacher_distance_mean": 1.1533909559249877,
      "weighted_auxiliary_loss": 0.0004540461057331413,
      "weighted_relation_loss": 0.0004540461057331413
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.528531337698784,
      "flow_matching_loss": 0.02730795219540596,
      "grad_norm": 0.09934724867343903,
      "learning_rate": 4.5327507994030394e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004708988621132448,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.05461044311524,
      "state_query_norm": 1414.9246826171875,
      "state_relation_loss": 0.0004708988621132448,
      "step": 16600,
      "student_distance_mean": 1934.233544921875,
      "teacher_distance_mean": 1.1507779717445374,
      "weighted_auxiliary_loss": 0.0004708988621132448,
      "weighted_relation_loss": 0.0004708988621132448
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.537885874649206,
      "flow_matching_loss": 0.026821867935359477,
      "grad_norm": 0.10915609449148178,
      "learning_rate": 4.527263637005273e-05,
      "loss": 0.0273,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004785770346643403,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.82359275817871,
      "state_query_norm": 1402.5071411132812,
      "state_relation_loss": 0.0004785770346643403,
      "step": 16610,
      "student_distance_mean": 1916.1007690429688,
      "teacher_distance_mean": 1.1435265064239502,
      "weighted_auxiliary_loss": 0.0004785770346643403,
      "weighted_relation_loss": 0.0004785770346643403
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.547240411599626,
      "flow_matching_loss": 0.026656661182641983,
      "grad_norm": 0.11916787177324295,
      "learning_rate": 4.521777049026766e-05,
      "loss": 0.0271,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047791696269996466,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.075304412841795,
      "state_query_norm": 1410.8244140625,
      "state_relation_loss": 0.00047791696269996466,
      "step": 16620,
      "student_distance_mean": 1931.3340454101562,
      "teacher_distance_mean": 1.1473357796669006,
      "weighted_auxiliary_loss": 0.00047791696269996466,
      "weighted_relation_loss": 0.00047791696269996466
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.556594948550046,
      "flow_matching_loss": 0.02518205586820841,
      "grad_norm": 0.10094986855983734,
      "learning_rate": 4.5162910421342386e-05,
      "loss": 0.0257,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004785074241226539,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72498664855957,
      "state_query_norm": 1438.7509033203125,
      "state_relation_loss": 0.0004785074241226539,
      "step": 16630,
      "student_distance_mean": 1965.4760375976562,
      "teacher_distance_mean": 1.166924774646759,
      "weighted_auxiliary_loss": 0.0004785074241226539,
      "weighted_relation_loss": 0.0004785074241226539
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.565949485500468,
      "flow_matching_loss": 0.02577617596834898,
      "grad_norm": 0.12982051074504852,
      "learning_rate": 4.510805622993706e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047550116141792387,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37839889526367,
      "state_query_norm": 1420.8001342773437,
      "state_relation_loss": 0.00047550116141792387,
      "step": 16640,
      "student_distance_mean": 1946.6740600585938,
      "teacher_distance_mean": 1.156261718273163,
      "weighted_auxiliary_loss": 0.00047550116141792387,
      "weighted_relation_loss": 0.00047550116141792387
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.575304022450888,
      "flow_matching_loss": 0.02923212256282568,
      "grad_norm": 0.10240999609231949,
      "learning_rate": 4.505320798270467e-05,
      "loss": 0.0297,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004714891081675887,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.885044479370116,
      "state_query_norm": 1406.3807006835937,
      "state_relation_loss": 0.0004714891081675887,
      "step": 16650,
      "student_distance_mean": 1922.6778076171875,
      "teacher_distance_mean": 1.1448862552642822,
      "weighted_auxiliary_loss": 0.0004714891081675887,
      "weighted_relation_loss": 0.0004714891081675887
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.58465855940131,
      "flow_matching_loss": 0.026392163895070552,
      "grad_norm": 0.0982380211353302,
      "learning_rate": 4.499836574629105e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046424577012658117,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.205431365966795,
      "state_query_norm": 1421.9219604492187,
      "state_relation_loss": 0.00046424577012658117,
      "step": 16660,
      "student_distance_mean": 1939.8169555664062,
      "teacher_distance_mean": 1.1546054124832152,
      "weighted_auxiliary_loss": 0.00046424577012658117,
      "weighted_relation_loss": 0.00046424577012658117
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.59401309635173,
      "flow_matching_loss": 0.02611463237553835,
      "grad_norm": 0.09590192139148712,
      "learning_rate": 4.494352958733465e-05,
      "loss": 0.0266,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045532141521107403,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10866851806641,
      "state_query_norm": 1419.0064453125,
      "state_relation_loss": 0.00045532141521107403,
      "step": 16670,
      "student_distance_mean": 1933.2171264648437,
      "teacher_distance_mean": 1.1497984528541565,
      "weighted_auxiliary_loss": 0.00045532141521107403,
      "weighted_relation_loss": 0.00045532141521107403
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.603367633302152,
      "flow_matching_loss": 0.028080925345420837,
      "grad_norm": 0.10120841860771179,
      "learning_rate": 4.488869957246663e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004692783346399665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.981445693969725,
      "state_query_norm": 1407.1753295898438,
      "state_relation_loss": 0.0004692783346399665,
      "step": 16680,
      "student_distance_mean": 1926.2090576171875,
      "teacher_distance_mean": 1.1471120953559875,
      "weighted_auxiliary_loss": 0.0004692783346399665,
      "weighted_relation_loss": 0.0004692783346399665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.612722170252573,
      "flow_matching_loss": 0.026117364503443243,
      "grad_norm": 0.11622750014066696,
      "learning_rate": 4.4833875768310575e-05,
      "loss": 0.0266,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004613897006493062,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.491059112548825,
      "state_query_norm": 1427.2302001953126,
      "state_relation_loss": 0.0004613897006493062,
      "step": 16690,
      "student_distance_mean": 1953.844970703125,
      "teacher_distance_mean": 1.1637803316116333,
      "weighted_auxiliary_loss": 0.0004613897006493062,
      "weighted_relation_loss": 0.0004613897006493062
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.622076707202993,
      "flow_matching_loss": 0.02824672907590866,
      "grad_norm": 0.09619995951652527,
      "learning_rate": 4.47790582414826e-05,
      "loss": 0.0287,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004948770336341112,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.22027816772461,
      "state_query_norm": 1406.5613647460937,
      "state_relation_loss": 0.0004948770336341112,
      "step": 16700,
      "student_distance_mean": 1933.5933837890625,
      "teacher_distance_mean": 1.147594428062439,
      "weighted_auxiliary_loss": 0.0004948770336341112,
      "weighted_relation_loss": 0.0004948770336341112
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.631431244153415,
      "flow_matching_loss": 0.027311120927333832,
      "grad_norm": 0.11999446153640747,
      "learning_rate": 4.472424705859114e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047652852663304657,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24603462219238,
      "state_query_norm": 1421.9307373046875,
      "state_relation_loss": 0.00047652852663304657,
      "step": 16710,
      "student_distance_mean": 1941.8898071289063,
      "teacher_distance_mean": 1.1499051570892334,
      "weighted_auxiliary_loss": 0.00047652852663304657,
      "weighted_relation_loss": 0.00047652852663304657
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.640785781103835,
      "flow_matching_loss": 0.02750708982348442,
      "grad_norm": 0.09234517067670822,
      "learning_rate": 4.466944228623701e-05,
      "loss": 0.028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046115923032630235,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.46530380249023,
      "state_query_norm": 1435.8057739257813,
      "state_relation_loss": 0.00046115923032630235,
      "step": 16720,
      "student_distance_mean": 1954.425390625,
      "teacher_distance_mean": 1.1533153057098389,
      "weighted_auxiliary_loss": 0.00046115923032630235,
      "weighted_relation_loss": 0.00046115923032630235
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.650140318054257,
      "flow_matching_loss": 0.026921110041439534,
      "grad_norm": 0.09391214698553085,
      "learning_rate": 4.461464399101312e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048183774924837053,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62521209716797,
      "state_query_norm": 1443.2185668945312,
      "state_relation_loss": 0.00048183774924837053,
      "step": 16730,
      "student_distance_mean": 1964.35,
      "teacher_distance_mean": 1.162059760093689,
      "weighted_auxiliary_loss": 0.00048183774924837053,
      "weighted_relation_loss": 0.00048183774924837053
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.659494855004677,
      "flow_matching_loss": 0.028473165072500705,
      "grad_norm": 0.13941523432731628,
      "learning_rate": 4.45598522395046e-05,
      "loss": 0.0289,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004581331973895431,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.056948852539065,
      "state_query_norm": 1419.16630859375,
      "state_relation_loss": 0.0004581331973895431,
      "step": 16740,
      "student_distance_mean": 1932.294970703125,
      "teacher_distance_mean": 1.1499064803123473,
      "weighted_auxiliary_loss": 0.0004581331973895431,
      "weighted_relation_loss": 0.0004581331973895431
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.668849391955098,
      "flow_matching_loss": 0.028069942817091943,
      "grad_norm": 0.09468138217926025,
      "learning_rate": 4.4505067098288577e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000491209066240117,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.873677825927736,
      "state_query_norm": 1408.5323364257813,
      "state_relation_loss": 0.000491209066240117,
      "step": 16750,
      "student_distance_mean": 1921.6710205078125,
      "teacher_distance_mean": 1.1417760252952576,
      "weighted_auxiliary_loss": 0.000491209066240117,
      "weighted_relation_loss": 0.000491209066240117
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.67820392890552,
      "flow_matching_loss": 0.0281132647767663,
      "grad_norm": 0.09269970655441284,
      "learning_rate": 4.445028863393417e-05,
      "loss": 0.0286,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005077928595710546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50634689331055,
      "state_query_norm": 1424.16689453125,
      "state_relation_loss": 0.0005077928595710546,
      "step": 16760,
      "student_distance_mean": 1951.6378540039063,
      "teacher_distance_mean": 1.159598970413208,
      "weighted_auxiliary_loss": 0.0005077928595710546,
      "weighted_relation_loss": 0.0005077928595710546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.68755846585594,
      "flow_matching_loss": 0.026147176325321198,
      "grad_norm": 0.11238562315702438,
      "learning_rate": 4.439551691300236e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005277105694403872,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33796768188476,
      "state_query_norm": 1410.9058227539062,
      "state_relation_loss": 0.0005277105694403872,
      "step": 16770,
      "student_distance_mean": 1935.8529541015625,
      "teacher_distance_mean": 1.1491026520729064,
      "weighted_auxiliary_loss": 0.0005277105694403872,
      "weighted_relation_loss": 0.0005277105694403872
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.696913002806362,
      "flow_matching_loss": 0.026251154579222204,
      "grad_norm": 0.11288226395845413,
      "learning_rate": 4.434075200204599e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048076706007122995,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.96907196044922,
      "state_query_norm": 1400.3418090820312,
      "state_relation_loss": 0.00048076706007122995,
      "step": 16780,
      "student_distance_mean": 1917.3283081054688,
      "teacher_distance_mean": 1.1343889236450195,
      "weighted_auxiliary_loss": 0.00048076706007122995,
      "weighted_relation_loss": 0.00048076706007122995
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.706267539756782,
      "flow_matching_loss": 0.02605482190847397,
      "grad_norm": 0.09869743138551712,
      "learning_rate": 4.428599396760956e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004943473846651614,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.97977294921875,
      "state_query_norm": 1406.242138671875,
      "state_relation_loss": 0.0004943473846651614,
      "step": 16790,
      "student_distance_mean": 1923.3873168945313,
      "teacher_distance_mean": 1.1348998427391053,
      "weighted_auxiliary_loss": 0.0004943473846651614,
      "weighted_relation_loss": 0.0004943473846651614
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.715622076707202,
      "flow_matching_loss": 0.027942271344363688,
      "grad_norm": 0.10860493779182434,
      "learning_rate": 4.423124287622926e-05,
      "loss": 0.0285,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005195200821617619,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36260719299317,
      "state_query_norm": 1423.7664306640625,
      "state_relation_loss": 0.0005195200821617619,
      "step": 16800,
      "student_distance_mean": 1943.3969604492188,
      "teacher_distance_mean": 1.1543697834014892,
      "weighted_auxiliary_loss": 0.0005195200821617619,
      "weighted_relation_loss": 0.0005195200821617619
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.724976613657624,
      "flow_matching_loss": 0.027866128273308276,
      "grad_norm": 0.09739476442337036,
      "learning_rate": 4.417649879443282e-05,
      "loss": 0.0283,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004637299454770982,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.456796646118164,
      "state_query_norm": 1421.1920654296875,
      "state_relation_loss": 0.0004637299454770982,
      "step": 16810,
      "student_distance_mean": 1949.6922973632813,
      "teacher_distance_mean": 1.1556536078453064,
      "weighted_auxiliary_loss": 0.0004637299454770982,
      "weighted_relation_loss": 0.0004637299454770982
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.734331150608044,
      "flow_matching_loss": 0.02723053954541683,
      "grad_norm": 0.08422712981700897,
      "learning_rate": 4.4121761788739444e-05,
      "loss": 0.0277,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044096701894886793,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.84223670959473,
      "state_query_norm": 1408.6034545898438,
      "state_relation_loss": 0.00044096701894886793,
      "step": 16820,
      "student_distance_mean": 1920.5159423828125,
      "teacher_distance_mean": 1.136957848072052,
      "weighted_auxiliary_loss": 0.00044096701894886793,
      "weighted_relation_loss": 0.00044096701894886793
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.743685687558466,
      "flow_matching_loss": 0.02727147117257118,
      "grad_norm": 0.10520348697900772,
      "learning_rate": 4.4067031925659805e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048752062139101326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.083295059204104,
      "state_query_norm": 1407.6163208007813,
      "state_relation_loss": 0.00048752062139101326,
      "step": 16830,
      "student_distance_mean": 1927.6156005859375,
      "teacher_distance_mean": 1.1440247535705566,
      "weighted_auxiliary_loss": 0.00048752062139101326,
      "weighted_relation_loss": 0.00048752062139101326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.753040224508887,
      "flow_matching_loss": 0.025659366883337496,
      "grad_norm": 0.0837072804570198,
      "learning_rate": 4.4012309271695816e-05,
      "loss": 0.0261,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004743056953884661,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.88743858337402,
      "state_query_norm": 1408.096044921875,
      "state_relation_loss": 0.0004743056953884661,
      "step": 16840,
      "student_distance_mean": 1922.8065185546875,
      "teacher_distance_mean": 1.1380167841911315,
      "weighted_auxiliary_loss": 0.0004743056953884661,
      "weighted_relation_loss": 0.0004743056953884661
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.762394761459309,
      "flow_matching_loss": 0.026733864471316336,
      "grad_norm": 0.10175158083438873,
      "learning_rate": 4.395759389334067e-05,
      "loss": 0.0272,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004891966527793556,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31498222351074,
      "state_query_norm": 1411.0296020507812,
      "state_relation_loss": 0.0004891966527793556,
      "step": 16850,
      "student_distance_mean": 1939.8148803710938,
      "teacher_distance_mean": 1.1505845189094543,
      "weighted_auxiliary_loss": 0.0004891966527793556,
      "weighted_relation_loss": 0.0004891966527793556
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.771749298409729,
      "flow_matching_loss": 0.025354549661278723,
      "grad_norm": 0.1042860820889473,
      "learning_rate": 4.3902885857078685e-05,
      "loss": 0.0258,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004863739915890619,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.18510856628418,
      "state_query_norm": 1412.0576538085938,
      "state_relation_loss": 0.0004863739915890619,
      "step": 16860,
      "student_distance_mean": 1934.466552734375,
      "teacher_distance_mean": 1.1438093423843383,
      "weighted_auxiliary_loss": 0.0004863739915890619,
      "weighted_relation_loss": 0.0004863739915890619
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.781103835360149,
      "flow_matching_loss": 0.02710583433508873,
      "grad_norm": 0.08514823019504547,
      "learning_rate": 4.384818522938531e-05,
      "loss": 0.0276,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004761402873555198,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30075187683106,
      "state_query_norm": 1417.6604858398437,
      "state_relation_loss": 0.0004761402873555198,
      "step": 16870,
      "student_distance_mean": 1941.3593139648438,
      "teacher_distance_mean": 1.1508571982383728,
      "weighted_auxiliary_loss": 0.0004761402873555198,
      "weighted_relation_loss": 0.0004761402873555198
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.790458372310571,
      "flow_matching_loss": 0.028638345934450627,
      "grad_norm": 0.10226753354072571,
      "learning_rate": 4.379349207672695e-05,
      "loss": 0.0291,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004990400979295373,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.521897888183595,
      "state_query_norm": 1421.3147705078125,
      "state_relation_loss": 0.0004990400979295373,
      "step": 16880,
      "student_distance_mean": 1948.289501953125,
      "teacher_distance_mean": 1.155624222755432,
      "weighted_auxiliary_loss": 0.0004990400979295373,
      "weighted_relation_loss": 0.0004990400979295373
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.799812909260991,
      "flow_matching_loss": 0.026681144163012503,
      "grad_norm": 0.10053278505802155,
      "learning_rate": 4.373880646556098e-05,
      "loss": 0.0272,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047270387294702233,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.946342086791994,
      "state_query_norm": 1401.6696411132812,
      "state_relation_loss": 0.00047270387294702233,
      "step": 16890,
      "student_distance_mean": 1918.9556518554687,
      "teacher_distance_mean": 1.1389520049095154,
      "weighted_auxiliary_loss": 0.00047270387294702233,
      "weighted_relation_loss": 0.00047270387294702233
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.809167446211413,
      "flow_matching_loss": 0.02769189141690731,
      "grad_norm": 0.12794527411460876,
      "learning_rate": 4.3684128462335533e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048139661375898866,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.85088119506836,
      "state_query_norm": 1440.7739990234375,
      "state_relation_loss": 0.00048139661375898866,
      "step": 16900,
      "student_distance_mean": 1971.798828125,
      "teacher_distance_mean": 1.1672091364860535,
      "weighted_auxiliary_loss": 0.00048139661375898866,
      "weighted_relation_loss": 0.00048139661375898866
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.818521983161833,
      "flow_matching_loss": 0.026536021567881107,
      "grad_norm": 0.09425388276576996,
      "learning_rate": 4.362945813348955e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004807418677955866,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.015916442871095,
      "state_query_norm": 1449.7959106445312,
      "state_relation_loss": 0.0004807418677955866,
      "step": 16910,
      "student_distance_mean": 1982.1763671875,
      "teacher_distance_mean": 1.1752781987190246,
      "weighted_auxiliary_loss": 0.0004807418677955866,
      "weighted_relation_loss": 0.0004807418677955866
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.827876520112255,
      "flow_matching_loss": 0.026673195138573646,
      "grad_norm": 0.09182874113321304,
      "learning_rate": 4.357479554545262e-05,
      "loss": 0.0271,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004666960361646488,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07184028625488,
      "state_query_norm": 1408.7615234375,
      "state_relation_loss": 0.0004666960361646488,
      "step": 16920,
      "student_distance_mean": 1928.5308715820313,
      "teacher_distance_mean": 1.1443806767463685,
      "weighted_auxiliary_loss": 0.0004666960361646488,
      "weighted_relation_loss": 0.0004666960361646488
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.837231057062676,
      "flow_matching_loss": 0.02624445892870426,
      "grad_norm": 0.10825401544570923,
      "learning_rate": 4.352014076464499e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005006428429624066,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65310173034668,
      "state_query_norm": 1427.999267578125,
      "state_relation_loss": 0.0005006428429624066,
      "step": 16930,
      "student_distance_mean": 1958.8088989257812,
      "teacher_distance_mean": 1.161344337463379,
      "weighted_auxiliary_loss": 0.0005006428429624066,
      "weighted_relation_loss": 0.0005006428429624066
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.846585594013096,
      "flow_matching_loss": 0.027081149630248547,
      "grad_norm": 0.08687853813171387,
      "learning_rate": 4.346549385747734e-05,
      "loss": 0.0275,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046588968543801455,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.999878311157225,
      "state_query_norm": 1408.8021606445313,
      "state_relation_loss": 0.00046588968543801455,
      "step": 16940,
      "student_distance_mean": 1926.9239501953125,
      "teacher_distance_mean": 1.1435373783111573,
      "weighted_auxiliary_loss": 0.00046588968543801455,
      "weighted_relation_loss": 0.00046588968543801455
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.855940130963518,
      "flow_matching_loss": 0.02743476238101721,
      "grad_norm": 0.09018834680318832,
      "learning_rate": 4.34108548903508e-05,
      "loss": 0.0279,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004642992716981098,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.508063888549806,
      "state_query_norm": 1421.7062744140626,
      "state_relation_loss": 0.0004642992716981098,
      "step": 16950,
      "student_distance_mean": 1950.1194091796874,
      "teacher_distance_mean": 1.1591758251190185,
      "weighted_auxiliary_loss": 0.0004642992716981098,
      "weighted_relation_loss": 0.0004642992716981098
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.865294667913938,
      "flow_matching_loss": 0.02689584195613861,
      "grad_norm": 0.10430695116519928,
      "learning_rate": 4.335622392965689e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004649858601624146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25412483215332,
      "state_query_norm": 1424.2978637695312,
      "state_relation_loss": 0.0004649858601624146,
      "step": 16960,
      "student_distance_mean": 1941.3201538085937,
      "teacher_distance_mean": 1.1534260153770446,
      "weighted_auxiliary_loss": 0.0004649858601624146,
      "weighted_relation_loss": 0.0004649858601624146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.87464920486436,
      "flow_matching_loss": 0.027281994372606276,
      "grad_norm": 0.13454677164554596,
      "learning_rate": 4.330160104177738e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048020587128121405,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70509643554688,
      "state_query_norm": 1431.7064086914063,
      "state_relation_loss": 0.00048020587128121405,
      "step": 16970,
      "student_distance_mean": 1959.5803833007812,
      "teacher_distance_mean": 1.1610945343971253,
      "weighted_auxiliary_loss": 0.00048020587128121405,
      "weighted_relation_loss": 0.00048020587128121405
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.88400374181478,
      "flow_matching_loss": 0.026241894625127316,
      "grad_norm": 0.10234763473272324,
      "learning_rate": 4.32469862930842e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004829337471164763,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56903190612793,
      "state_query_norm": 1423.6755493164062,
      "state_relation_loss": 0.0004829337471164763,
      "step": 16980,
      "student_distance_mean": 1955.6451171875,
      "teacher_distance_mean": 1.158940041065216,
      "weighted_auxiliary_loss": 0.0004829337471164763,
      "weighted_relation_loss": 0.0004829337471164763
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.8933582787652,
      "flow_matching_loss": 0.026932806335389615,
      "grad_norm": 0.08606424927711487,
      "learning_rate": 4.319237974993946e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045627198414877057,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.370195388793945,
      "state_query_norm": 1432.2994873046875,
      "state_relation_loss": 0.00045627198414877057,
      "step": 16990,
      "student_distance_mean": 1949.2732177734374,
      "teacher_distance_mean": 1.1534960746765137,
      "weighted_auxiliary_loss": 0.00045627198414877057,
      "weighted_relation_loss": 0.00045627198414877057
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.902712815715622,
      "flow_matching_loss": 0.02742397002875805,
      "grad_norm": 0.10207490622997284,
      "learning_rate": 4.3137781478695236e-05,
      "loss": 0.0279,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004793499858351424,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59263610839844,
      "state_query_norm": 1430.0381103515624,
      "state_relation_loss": 0.0004793499858351424,
      "step": 17000,
      "student_distance_mean": 1954.4015991210938,
      "teacher_distance_mean": 1.1565014719963074,
      "weighted_auxiliary_loss": 0.0004793499858351424,
      "weighted_relation_loss": 0.0004793499858351424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.912067352666043,
      "flow_matching_loss": 0.027564539387822152,
      "grad_norm": 0.09463971108198166,
      "learning_rate": 4.308319154569358e-05,
      "loss": 0.028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004834114166442305,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.79600601196289,
      "state_query_norm": 1432.891357421875,
      "state_relation_loss": 0.0004834114166442305,
      "step": 17010,
      "student_distance_mean": 1966.2002319335938,
      "teacher_distance_mean": 1.1651937127113343,
      "weighted_auxiliary_loss": 0.0004834114166442305,
      "weighted_relation_loss": 0.0004834114166442305
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.921421889616465,
      "flow_matching_loss": 0.026216147653758525,
      "grad_norm": 0.11586642265319824,
      "learning_rate": 4.302861001726641e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004905176319880411,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40723991394043,
      "state_query_norm": 1429.0789428710937,
      "state_relation_loss": 0.0004905176319880411,
      "step": 17020,
      "student_distance_mean": 1943.5403686523437,
      "teacher_distance_mean": 1.151347291469574,
      "weighted_auxiliary_loss": 0.0004905176319880411,
      "weighted_relation_loss": 0.0004905176319880411
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.930776426566885,
      "flow_matching_loss": 0.027262000553309917,
      "grad_norm": 0.08492411673069,
      "learning_rate": 4.2974036959735415e-05,
      "loss": 0.0277,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048517986433580516,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92822418212891,
      "state_query_norm": 1413.3576293945312,
      "state_relation_loss": 0.00048517986433580516,
      "step": 17030,
      "student_distance_mean": 1925.2909912109376,
      "teacher_distance_mean": 1.1416895270347596,
      "weighted_auxiliary_loss": 0.00048517986433580516,
      "weighted_relation_loss": 0.00048517986433580516
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.940130963517305,
      "flow_matching_loss": 0.028720393404364587,
      "grad_norm": 0.09487222135066986,
      "learning_rate": 4.2919472439412034e-05,
      "loss": 0.0292,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004778259695740417,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65908660888672,
      "state_query_norm": 1425.5368896484374,
      "state_relation_loss": 0.0004778259695740417,
      "step": 17040,
      "student_distance_mean": 1957.1572631835938,
      "teacher_distance_mean": 1.157763922214508,
      "weighted_auxiliary_loss": 0.0004778259695740417,
      "weighted_relation_loss": 0.0004778259695740417
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.949485500467727,
      "flow_matching_loss": 0.026011029444634914,
      "grad_norm": 0.11172056198120117,
      "learning_rate": 4.286491652259729e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004747801023768261,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.871377563476564,
      "state_query_norm": 1398.8843017578124,
      "state_relation_loss": 0.0004747801023768261,
      "step": 17050,
      "student_distance_mean": 1914.7743041992187,
      "teacher_distance_mean": 1.1336967706680299,
      "weighted_auxiliary_loss": 0.0004747801023768261,
      "weighted_relation_loss": 0.0004747801023768261
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.958840037418147,
      "flow_matching_loss": 0.029250426031649113,
      "grad_norm": 0.07528439909219742,
      "learning_rate": 4.281036927558178e-05,
      "loss": 0.0297,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045710925187449904,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.333849334716795,
      "state_query_norm": 1418.8399658203125,
      "state_relation_loss": 0.00045710925187449904,
      "step": 17060,
      "student_distance_mean": 1941.4526000976562,
      "teacher_distance_mean": 1.1532222986221314,
      "weighted_auxiliary_loss": 0.00045710925187449904,
      "weighted_relation_loss": 0.00045710925187449904
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.96819457436857,
      "flow_matching_loss": 0.0259177865460515,
      "grad_norm": 0.07991310209035873,
      "learning_rate": 4.275583076464552e-05,
      "loss": 0.0264,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046569357509724797,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.35622978210449,
      "state_query_norm": 1420.4080200195312,
      "state_relation_loss": 0.00046569357509724797,
      "step": 17070,
      "student_distance_mean": 1942.7026123046876,
      "teacher_distance_mean": 1.1590778946876525,
      "weighted_auxiliary_loss": 0.00046569357509724797,
      "weighted_relation_loss": 0.00046569357509724797
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.97754911131899,
      "flow_matching_loss": 0.025658343732357026,
      "grad_norm": 0.10065171122550964,
      "learning_rate": 4.270130105605794e-05,
      "loss": 0.0261,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004610238393070176,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.888095474243165,
      "state_query_norm": 1392.8903686523438,
      "state_relation_loss": 0.0004610238393070176,
      "step": 17080,
      "student_distance_mean": 1914.2291748046875,
      "teacher_distance_mean": 1.140708827972412,
      "weighted_auxiliary_loss": 0.0004610238393070176,
      "weighted_relation_loss": 0.0004610238393070176
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.986903648269411,
      "flow_matching_loss": 0.02688749022781849,
      "grad_norm": 0.09462389349937439,
      "learning_rate": 4.264678021607782e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004671245493227616,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.89083633422852,
      "state_query_norm": 1407.5030029296875,
      "state_relation_loss": 0.0004671245493227616,
      "step": 17090,
      "student_distance_mean": 1921.36806640625,
      "teacher_distance_mean": 1.1419973254203797,
      "weighted_auxiliary_loss": 0.0004671245493227616,
      "weighted_relation_loss": 0.0004671245493227616
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 15.996258185219832,
      "flow_matching_loss": 0.0269005523994565,
      "grad_norm": 0.12755008041858673,
      "learning_rate": 4.25922683109531e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000485382997430861,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.068965911865234,
      "state_query_norm": 1418.9020263671875,
      "state_relation_loss": 0.000485382997430861,
      "step": 17100,
      "student_distance_mean": 1934.1544921875,
      "teacher_distance_mean": 1.1479458332061767,
      "weighted_auxiliary_loss": 0.000485382997430861,
      "weighted_relation_loss": 0.000485382997430861
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.00561272217025,
      "flow_matching_loss": 0.025430831126868725,
      "grad_norm": 0.09309795498847961,
      "learning_rate": 4.25377654069209e-05,
      "loss": 0.0259,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0005083804979221895,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.56695327758789,
      "state_query_norm": 1429.0149658203125,
      "state_relation_loss": 0.0005083804979221895,
      "step": 17110,
      "student_distance_mean": 1955.01884765625,
      "teacher_distance_mean": 1.1573339462280274,
      "weighted_auxiliary_loss": 0.0005083804979221895,
      "weighted_relation_loss": 0.0005083804979221895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.014967259120674,
      "flow_matching_loss": 0.029435175657272338,
      "grad_norm": 0.11160080879926682,
      "learning_rate": 4.248327157020737e-05,
      "loss": 0.0299,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004617040103767067,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.434427642822264,
      "state_query_norm": 1426.4861938476563,
      "state_relation_loss": 0.0004617040103767067,
      "step": 17120,
      "student_distance_mean": 1947.8347534179688,
      "teacher_distance_mean": 1.1517340898513795,
      "weighted_auxiliary_loss": 0.0004617040103767067,
      "weighted_relation_loss": 0.0004617040103767067
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.024321796071096,
      "flow_matching_loss": 0.025426238030195236,
      "grad_norm": 0.12348717451095581,
      "learning_rate": 4.242878686702763e-05,
      "loss": 0.0259,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004713010479463264,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.292145538330075,
      "state_query_norm": 1419.109326171875,
      "state_relation_loss": 0.0004713010479463264,
      "step": 17130,
      "student_distance_mean": 1940.1156494140625,
      "teacher_distance_mean": 1.150244677066803,
      "weighted_auxiliary_loss": 0.0004713010479463264,
      "weighted_relation_loss": 0.0004713010479463264
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.033676333021514,
      "flow_matching_loss": 0.0255392512306571,
      "grad_norm": 0.08231665939092636,
      "learning_rate": 4.23743113635858e-05,
      "loss": 0.026,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004717652074759826,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32428398132324,
      "state_query_norm": 1415.4622802734375,
      "state_relation_loss": 0.0004717652074759826,
      "step": 17140,
      "student_distance_mean": 1939.2572387695313,
      "teacher_distance_mean": 1.1485309720039367,
      "weighted_auxiliary_loss": 0.0004717652074759826,
      "weighted_relation_loss": 0.0004717652074759826
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.043030869971936,
      "flow_matching_loss": 0.02639251258224249,
      "grad_norm": 0.10163533687591553,
      "learning_rate": 4.2319845126074705e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000489522743737325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.8532829284668,
      "state_query_norm": 1427.947998046875,
      "state_relation_loss": 0.000489522743737325,
      "step": 17150,
      "student_distance_mean": 1966.5174560546875,
      "teacher_distance_mean": 1.158972442150116,
      "weighted_auxiliary_loss": 0.000489522743737325,
      "weighted_relation_loss": 0.000489522743737325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.052385406922358,
      "flow_matching_loss": 0.023863197676837443,
      "grad_norm": 0.10239783674478531,
      "learning_rate": 4.226538822067598e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047607927117496727,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.20289268493652,
      "state_query_norm": 1454.6594970703125,
      "state_relation_loss": 0.00047607927117496727,
      "step": 17160,
      "student_distance_mean": 1992.7757568359375,
      "teacher_distance_mean": 1.1760244369506836,
      "weighted_auxiliary_loss": 0.00047607927117496727,
      "weighted_relation_loss": 0.00047607927117496727
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.061739943872777,
      "flow_matching_loss": 0.02561186496168375,
      "grad_norm": 0.09270285815000534,
      "learning_rate": 4.2210940713559896e-05,
      "loss": 0.0261,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004969056026311591,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28361854553223,
      "state_query_norm": 1422.411669921875,
      "state_relation_loss": 0.0004969056026311591,
      "step": 17170,
      "student_distance_mean": 1942.5509765625,
      "teacher_distance_mean": 1.1465669393539428,
      "weighted_auxiliary_loss": 0.0004969056026311591,
      "weighted_relation_loss": 0.0004969056026311591
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.0710944808232,
      "flow_matching_loss": 0.026246995665133,
      "grad_norm": 0.1085028126835823,
      "learning_rate": 4.21565026708853e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000467489639413543,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54767990112305,
      "state_query_norm": 1423.500390625,
      "state_relation_loss": 0.000467489639413543,
      "step": 17180,
      "student_distance_mean": 1949.9606567382812,
      "teacher_distance_mean": 1.1546217203140259,
      "weighted_auxiliary_loss": 0.000467489639413543,
      "weighted_relation_loss": 0.000467489639413543
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.08044901777362,
      "flow_matching_loss": 0.024034996703267097,
      "grad_norm": 0.11014683544635773,
      "learning_rate": 4.2102074158799535e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044458870543166994,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.09235038757324,
      "state_query_norm": 1420.2230102539063,
      "state_relation_loss": 0.00044458870543166994,
      "step": 17190,
      "student_distance_mean": 1933.4618530273438,
      "teacher_distance_mean": 1.1460125684738158,
      "weighted_auxiliary_loss": 0.00044458870543166994,
      "weighted_relation_loss": 0.00044458870543166994
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.089803554724043,
      "flow_matching_loss": 0.026175388507544993,
      "grad_norm": 0.10825898498296738,
      "learning_rate": 4.204765524343841e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004912200820399448,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54505729675293,
      "state_query_norm": 1431.0161865234375,
      "state_relation_loss": 0.0004912200820399448,
      "step": 17200,
      "student_distance_mean": 1953.919189453125,
      "teacher_distance_mean": 1.1591298699378967,
      "weighted_auxiliary_loss": 0.0004912200820399448,
      "weighted_relation_loss": 0.0004912200820399448
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.09915809167446,
      "flow_matching_loss": 0.026247984915971755,
      "grad_norm": 0.10954645276069641,
      "learning_rate": 4.1993245990926036e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004783134470926598,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07699203491211,
      "state_query_norm": 1409.6928466796876,
      "state_relation_loss": 0.0004783134470926598,
      "step": 17210,
      "student_distance_mean": 1929.410546875,
      "teacher_distance_mean": 1.140795886516571,
      "weighted_auxiliary_loss": 0.0004783134470926598,
      "weighted_relation_loss": 0.0004783134470926598
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.108512628624883,
      "flow_matching_loss": 0.02731255367398262,
      "grad_norm": 0.10848137736320496,
      "learning_rate": 4.193884646737474e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004915502708172425,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6798641204834,
      "state_query_norm": 1441.7828857421875,
      "state_relation_loss": 0.0004915502708172425,
      "step": 17220,
      "student_distance_mean": 1964.869580078125,
      "teacher_distance_mean": 1.1613136768341064,
      "weighted_auxiliary_loss": 0.0004915502708172425,
      "weighted_relation_loss": 0.0004915502708172425
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.117867165575305,
      "flow_matching_loss": 0.02645583972334862,
      "grad_norm": 0.1358696073293686,
      "learning_rate": 4.1884456738885124e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004502131429035217,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.403469467163085,
      "state_query_norm": 1426.2753784179688,
      "state_relation_loss": 0.0004502131429035217,
      "step": 17230,
      "student_distance_mean": 1948.9291015625,
      "teacher_distance_mean": 1.1515798091888427,
      "weighted_auxiliary_loss": 0.0004502131429035217,
      "weighted_relation_loss": 0.0004502131429035217
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.127221702525723,
      "flow_matching_loss": 0.02542509324848652,
      "grad_norm": 0.09500589966773987,
      "learning_rate": 4.1830076871545795e-05,
      "loss": 0.0259,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043457722349558023,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.76542930603027,
      "state_query_norm": 1397.8807739257813,
      "state_relation_loss": 0.00043457722349558023,
      "step": 17240,
      "student_distance_mean": 1915.993017578125,
      "teacher_distance_mean": 1.1353341221809388,
      "weighted_auxiliary_loss": 0.00043457722349558023,
      "weighted_relation_loss": 0.00043457722349558023
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.136576239476145,
      "flow_matching_loss": 0.02725066225975752,
      "grad_norm": 0.08486031740903854,
      "learning_rate": 4.177570693143346e-05,
      "loss": 0.0277,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046061673492658883,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40812950134277,
      "state_query_norm": 1416.9026123046874,
      "state_relation_loss": 0.00046061673492658883,
      "step": 17250,
      "student_distance_mean": 1943.8240234375,
      "teacher_distance_mean": 1.1534655928611754,
      "weighted_auxiliary_loss": 0.00046061673492658883,
      "weighted_relation_loss": 0.00046061673492658883
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.145930776426567,
      "flow_matching_loss": 0.026641812175512314,
      "grad_norm": 0.12348467111587524,
      "learning_rate": 4.172134698461271e-05,
      "loss": 0.0271,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043564910884015263,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.640423965454104,
      "state_query_norm": 1394.2808959960937,
      "state_relation_loss": 0.00043564910884015263,
      "step": 17260,
      "student_distance_mean": 1906.96865234375,
      "teacher_distance_mean": 1.1331014156341552,
      "weighted_auxiliary_loss": 0.00043564910884015263,
      "weighted_relation_loss": 0.00043564910884015263
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.15528531337699,
      "flow_matching_loss": 0.02605611626058817,
      "grad_norm": 0.09743509441614151,
      "learning_rate": 4.166699709713599e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048437128716614095,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53664360046387,
      "state_query_norm": 1432.0607666015626,
      "state_relation_loss": 0.00048437128716614095,
      "step": 17270,
      "student_distance_mean": 1954.7405395507812,
      "teacher_distance_mean": 1.1617482781410218,
      "weighted_auxiliary_loss": 0.00048437128716614095,
      "weighted_relation_loss": 0.00048437128716614095
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.164639850327408,
      "flow_matching_loss": 0.026751613058149815,
      "grad_norm": 0.10186860710382462,
      "learning_rate": 4.1612657335043555e-05,
      "loss": 0.0272,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004579265398206189,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.453491592407225,
      "state_query_norm": 1425.3834350585937,
      "state_relation_loss": 0.0004579265398206189,
      "step": 17280,
      "student_distance_mean": 1950.1429443359375,
      "teacher_distance_mean": 1.1556312918663025,
      "weighted_auxiliary_loss": 0.0004579265398206189,
      "weighted_relation_loss": 0.0004579265398206189
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.17399438727783,
      "flow_matching_loss": 0.026250083371996878,
      "grad_norm": 0.11462225019931793,
      "learning_rate": 4.155832776436331e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043962499767076225,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.135372924804685,
      "state_query_norm": 1410.7528198242187,
      "state_relation_loss": 0.00043962499767076225,
      "step": 17290,
      "student_distance_mean": 1929.8781494140626,
      "teacher_distance_mean": 1.1406642198562622,
      "weighted_auxiliary_loss": 0.00043962499767076225,
      "weighted_relation_loss": 0.00043962499767076225
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.18334892422825,
      "flow_matching_loss": 0.027850969694554806,
      "grad_norm": 0.1218729093670845,
      "learning_rate": 4.1504008451110845e-05,
      "loss": 0.0283,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004713998758234084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.9647834777832,
      "state_query_norm": 1407.53056640625,
      "state_relation_loss": 0.0004713998758234084,
      "step": 17300,
      "student_distance_mean": 1925.3919921875,
      "teacher_distance_mean": 1.140070104598999,
      "weighted_auxiliary_loss": 0.0004713998758234084,
      "weighted_relation_loss": 0.0004713998758234084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.19270346117867,
      "flow_matching_loss": 0.02664525415748358,
      "grad_norm": 0.1524474024772644,
      "learning_rate": 4.1449699461289225e-05,
      "loss": 0.0271,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004419378121383488,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.198435592651364,
      "state_query_norm": 1417.8185791015626,
      "state_relation_loss": 0.0004419378121383488,
      "step": 17310,
      "student_distance_mean": 1935.9852783203125,
      "teacher_distance_mean": 1.1519413590431213,
      "weighted_auxiliary_loss": 0.0004419378121383488,
      "weighted_relation_loss": 0.0004419378121383488
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.202057998129092,
      "flow_matching_loss": 0.026780514046549797,
      "grad_norm": 0.10359971225261688,
      "learning_rate": 4.139540086088901e-05,
      "loss": 0.0272,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004539079440291971,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.937744522094725,
      "state_query_norm": 1408.0518676757813,
      "state_relation_loss": 0.0004539079440291971,
      "step": 17320,
      "student_distance_mean": 1922.4228881835938,
      "teacher_distance_mean": 1.1430615186691284,
      "weighted_auxiliary_loss": 0.0004539079440291971,
      "weighted_relation_loss": 0.0004539079440291971
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.211412535079514,
      "flow_matching_loss": 0.02687714099884033,
      "grad_norm": 0.09197387844324112,
      "learning_rate": 4.1341112715888106e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004813584178918973,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.81581497192383,
      "state_query_norm": 1435.2104248046876,
      "state_relation_loss": 0.0004813584178918973,
      "step": 17330,
      "student_distance_mean": 1966.727099609375,
      "teacher_distance_mean": 1.1653208136558533,
      "weighted_auxiliary_loss": 0.0004813584178918973,
      "weighted_relation_loss": 0.0004813584178918973
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.220767072029936,
      "flow_matching_loss": 0.02569994777441025,
      "grad_norm": 0.10607657581567764,
      "learning_rate": 4.128683509225172e-05,
      "loss": 0.0262,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004561432986520231,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.068446731567384,
      "state_query_norm": 1417.567919921875,
      "state_relation_loss": 0.0004561432986520231,
      "step": 17340,
      "student_distance_mean": 1933.1230102539062,
      "teacher_distance_mean": 1.1445582985877991,
      "weighted_auxiliary_loss": 0.0004561432986520231,
      "weighted_relation_loss": 0.0004561432986520231
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.230121608980355,
      "flow_matching_loss": 0.02739247474819422,
      "grad_norm": 0.10464959591627121,
      "learning_rate": 4.123256805593231e-05,
      "loss": 0.0279,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048497562238480896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59687957763672,
      "state_query_norm": 1425.946484375,
      "state_relation_loss": 0.00048497562238480896,
      "step": 17350,
      "student_distance_mean": 1954.5660278320313,
      "teacher_distance_mean": 1.1605339169502258,
      "weighted_auxiliary_loss": 0.00048497562238480896,
      "weighted_relation_loss": 0.00048497562238480896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.239476145930777,
      "flow_matching_loss": 0.025535299070179464,
      "grad_norm": 0.0937177985906601,
      "learning_rate": 4.117831167286943e-05,
      "loss": 0.026,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004849369142903015,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.00125694274902,
      "state_query_norm": 1415.4834350585938,
      "state_relation_loss": 0.0004849369142903015,
      "step": 17360,
      "student_distance_mean": 1926.58447265625,
      "teacher_distance_mean": 1.1399014949798585,
      "weighted_auxiliary_loss": 0.0004849369142903015,
      "weighted_relation_loss": 0.0004849369142903015
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.2488306828812,
      "flow_matching_loss": 0.026847061514854432,
      "grad_norm": 0.12016505002975464,
      "learning_rate": 4.1124066008989685e-05,
      "loss": 0.0273,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004809411766473204,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.077994918823244,
      "state_query_norm": 1408.0325073242188,
      "state_relation_loss": 0.0004809411766473204,
      "step": 17370,
      "student_distance_mean": 1929.5863159179687,
      "teacher_distance_mean": 1.1430765032768249,
      "weighted_auxiliary_loss": 0.0004809411766473204,
      "weighted_relation_loss": 0.0004809411766473204
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.258185219831617,
      "flow_matching_loss": 0.026211777329444887,
      "grad_norm": 0.10486596822738647,
      "learning_rate": 4.106983113020669e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004809810576261953,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66891708374023,
      "state_query_norm": 1429.223974609375,
      "state_relation_loss": 0.0004809810576261953,
      "step": 17380,
      "student_distance_mean": 1961.5290283203126,
      "teacher_distance_mean": 1.1575623154640198,
      "weighted_auxiliary_loss": 0.0004809810576261953,
      "weighted_relation_loss": 0.0004809810576261953
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.26753975678204,
      "flow_matching_loss": 0.02659259717911482,
      "grad_norm": 0.08657396584749222,
      "learning_rate": 4.101560710242094e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044805360375903546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.315687561035155,
      "state_query_norm": 1419.0910888671874,
      "state_relation_loss": 0.00044805360375903546,
      "step": 17390,
      "student_distance_mean": 1942.6280639648437,
      "teacher_distance_mean": 1.148245346546173,
      "weighted_auxiliary_loss": 0.00044805360375903546,
      "weighted_relation_loss": 0.00044805360375903546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.27689429373246,
      "flow_matching_loss": 0.02648698091506958,
      "grad_norm": 0.1278873234987259,
      "learning_rate": 4.096139399151971e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047929572174325584,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70345802307129,
      "state_query_norm": 1443.94990234375,
      "state_relation_loss": 0.00047929572174325584,
      "step": 17400,
      "student_distance_mean": 1966.3974365234376,
      "teacher_distance_mean": 1.1638074040412902,
      "weighted_auxiliary_loss": 0.00047929572174325584,
      "weighted_relation_loss": 0.00047929572174325584
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.28624883068288,
      "flow_matching_loss": 0.025721002742648123,
      "grad_norm": 0.09524491429328918,
      "learning_rate": 4.090719186337709e-05,
      "loss": 0.0262,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004471270425710827,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.61076889038086,
      "state_query_norm": 1437.3932373046875,
      "state_relation_loss": 0.0004471270425710827,
      "step": 17410,
      "student_distance_mean": 1959.4943481445312,
      "teacher_distance_mean": 1.16002459526062,
      "weighted_auxiliary_loss": 0.0004471270425710827,
      "weighted_relation_loss": 0.0004471270425710827
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.2956033676333,
      "flow_matching_loss": 0.026672909036278725,
      "grad_norm": 0.11861873418092728,
      "learning_rate": 4.0853000783853746e-05,
      "loss": 0.0271,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004628507129382342,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29119110107422,
      "state_query_norm": 1423.6096069335938,
      "state_relation_loss": 0.0004628507129382342,
      "step": 17420,
      "student_distance_mean": 1943.60146484375,
      "teacher_distance_mean": 1.1506464004516601,
      "weighted_auxiliary_loss": 0.0004628507129382342,
      "weighted_relation_loss": 0.0004628507129382342
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.304957904583723,
      "flow_matching_loss": 0.0260317025706172,
      "grad_norm": 0.11811134219169617,
      "learning_rate": 4.0798820818796956e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045238524326123296,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60368118286133,
      "state_query_norm": 1436.0197509765626,
      "state_relation_loss": 0.00045238524326123296,
      "step": 17430,
      "student_distance_mean": 1961.735791015625,
      "teacher_distance_mean": 1.1654262661933898,
      "weighted_auxiliary_loss": 0.00045238524326123296,
      "weighted_relation_loss": 0.00045238524326123296
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.314312441534145,
      "flow_matching_loss": 0.0246027123183012,
      "grad_norm": 0.10216368734836578,
      "learning_rate": 4.074465203404048e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000469712755875662,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37176856994629,
      "state_query_norm": 1419.7419067382812,
      "state_relation_loss": 0.000469712755875662,
      "step": 17440,
      "student_distance_mean": 1944.3271728515624,
      "teacher_distance_mean": 1.150997793674469,
      "weighted_auxiliary_loss": 0.000469712755875662,
      "weighted_relation_loss": 0.000469712755875662
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.323666978484564,
      "flow_matching_loss": 0.028574160672724246,
      "grad_norm": 0.10434962809085846,
      "learning_rate": 4.0690494495404495e-05,
      "loss": 0.029,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045606321364175526,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65799560546875,
      "state_query_norm": 1429.267236328125,
      "state_relation_loss": 0.00045606321364175526,
      "step": 17450,
      "student_distance_mean": 1959.343115234375,
      "teacher_distance_mean": 1.1607264161109925,
      "weighted_auxiliary_loss": 0.00045606321364175526,
      "weighted_relation_loss": 0.00045606321364175526
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.333021515434986,
      "flow_matching_loss": 0.024604210257530214,
      "grad_norm": 0.0907665491104126,
      "learning_rate": 4.0636348268695535e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004736816103104502,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.224609375,
      "state_query_norm": 1412.7440673828125,
      "state_relation_loss": 0.0004736816103104502,
      "step": 17460,
      "student_distance_mean": 1936.442529296875,
      "teacher_distance_mean": 1.1431588768959045,
      "weighted_auxiliary_loss": 0.0004736816103104502,
      "weighted_relation_loss": 0.0004736816103104502
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.342376052385408,
      "flow_matching_loss": 0.02646790388971567,
      "grad_norm": 0.10359698534011841,
      "learning_rate": 4.058221341970637e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000474859683890827,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72904739379883,
      "state_query_norm": 1438.5145629882813,
      "state_relation_loss": 0.000474859683890827,
      "step": 17470,
      "student_distance_mean": 1964.977001953125,
      "teacher_distance_mean": 1.1620474576950073,
      "weighted_auxiliary_loss": 0.000474859683890827,
      "weighted_relation_loss": 0.000474859683890827
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.351730589335826,
      "flow_matching_loss": 0.027005262859165668,
      "grad_norm": 0.08109405636787415,
      "learning_rate": 4.0528090014215945e-05,
      "loss": 0.0275,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004722789191873744,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.61608352661133,
      "state_query_norm": 1438.1048095703125,
      "state_relation_loss": 0.0004722789191873744,
      "step": 17480,
      "student_distance_mean": 1960.4675659179688,
      "teacher_distance_mean": 1.157143783569336,
      "weighted_auxiliary_loss": 0.0004722789191873744,
      "weighted_relation_loss": 0.0004722789191873744
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.36108512628625,
      "flow_matching_loss": 0.02666181456297636,
      "grad_norm": 0.1069079264998436,
      "learning_rate": 4.047397811798929e-05,
      "loss": 0.0272,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004905923386104404,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9788875579834,
      "state_query_norm": 1445.2355712890626,
      "state_relation_loss": 0.0004905923386104404,
      "step": 17490,
      "student_distance_mean": 1974.4257080078125,
      "teacher_distance_mean": 1.168589675426483,
      "weighted_auxiliary_loss": 0.0004905923386104404,
      "weighted_relation_loss": 0.0004905923386104404
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.37043966323667,
      "flow_matching_loss": 0.026008028350770474,
      "grad_norm": 0.08891189843416214,
      "learning_rate": 4.041987779677745e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004567981086438522,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17832145690918,
      "state_query_norm": 1415.2130981445312,
      "state_relation_loss": 0.0004567981086438522,
      "step": 17500,
      "student_distance_mean": 1934.1777465820312,
      "teacher_distance_mean": 1.1462344646453857,
      "weighted_auxiliary_loss": 0.0004567981086438522,
      "weighted_relation_loss": 0.0004567981086438522
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.379794200187092,
      "flow_matching_loss": 0.0275316609069705,
      "grad_norm": 0.09911579638719559,
      "learning_rate": 4.036578911631745e-05,
      "loss": 0.028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004703102953499183,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2723747253418,
      "state_query_norm": 1415.6921875,
      "state_relation_loss": 0.0004703102953499183,
      "step": 17510,
      "student_distance_mean": 1937.965185546875,
      "teacher_distance_mean": 1.146395742893219,
      "weighted_auxiliary_loss": 0.0004703102953499183,
      "weighted_relation_loss": 0.0004703102953499183
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.38914873713751,
      "flow_matching_loss": 0.02611028477549553,
      "grad_norm": 0.08891413360834122,
      "learning_rate": 4.031171214233211e-05,
      "loss": 0.0266,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045847919536754487,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54378051757813,
      "state_query_norm": 1434.8127807617188,
      "state_relation_loss": 0.00045847919536754487,
      "step": 17520,
      "student_distance_mean": 1955.9057861328124,
      "teacher_distance_mean": 1.1582070112228393,
      "weighted_auxiliary_loss": 0.00045847919536754487,
      "weighted_relation_loss": 0.00045847919536754487
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.398503274087933,
      "flow_matching_loss": 0.024930459447205067,
      "grad_norm": 0.11074787378311157,
      "learning_rate": 4.025764694053008e-05,
      "loss": 0.0254,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004932482581352815,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70010604858398,
      "state_query_norm": 1427.9374877929688,
      "state_relation_loss": 0.0004932482581352815,
      "step": 17530,
      "student_distance_mean": 1958.68427734375,
      "teacher_distance_mean": 1.1556935787200928,
      "weighted_auxiliary_loss": 0.0004932482581352815,
      "weighted_relation_loss": 0.0004932482581352815
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.407857811038355,
      "flow_matching_loss": 0.02410729955881834,
      "grad_norm": 0.11238878965377808,
      "learning_rate": 4.020359357660566e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000493719961377792,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.571349716186525,
      "state_query_norm": 1425.4622924804687,
      "state_relation_loss": 0.000493719961377792,
      "step": 17540,
      "student_distance_mean": 1952.6633911132812,
      "teacher_distance_mean": 1.1505059599876404,
      "weighted_auxiliary_loss": 0.000493719961377792,
      "weighted_relation_loss": 0.000493719961377792
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.417212347988773,
      "flow_matching_loss": 0.026937047392129897,
      "grad_norm": 0.12112317979335785,
      "learning_rate": 4.014955211623875e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004860461951466277,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37884864807129,
      "state_query_norm": 1421.1999633789062,
      "state_relation_loss": 0.0004860461951466277,
      "step": 17550,
      "student_distance_mean": 1944.5177490234375,
      "teacher_distance_mean": 1.1463823080062867,
      "weighted_auxiliary_loss": 0.0004860461951466277,
      "weighted_relation_loss": 0.0004860461951466277
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.426566884939195,
      "flow_matching_loss": 0.02412171717733145,
      "grad_norm": 0.09536584466695786,
      "learning_rate": 4.0095522625094895e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047780367312952874,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.311919403076175,
      "state_query_norm": 1415.6462158203126,
      "state_relation_loss": 0.00047780367312952874,
      "step": 17560,
      "student_distance_mean": 1943.4215209960937,
      "teacher_distance_mean": 1.1471429347991944,
      "weighted_auxiliary_loss": 0.00047780367312952874,
      "weighted_relation_loss": 0.00047780367312952874
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.435921421889617,
      "flow_matching_loss": 0.026775444485247137,
      "grad_norm": 0.11796770989894867,
      "learning_rate": 4.004150516882497e-05,
      "loss": 0.0273,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004887014540145174,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.244818115234374,
      "state_query_norm": 1417.1753295898438,
      "state_relation_loss": 0.0004887014540145174,
      "step": 17570,
      "student_distance_mean": 1937.3421508789063,
      "teacher_distance_mean": 1.141517734527588,
      "weighted_auxiliary_loss": 0.0004887014540145174,
      "weighted_relation_loss": 0.0004887014540145174
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.44527595884004,
      "flow_matching_loss": 0.02734985314309597,
      "grad_norm": 0.1136782094836235,
      "learning_rate": 3.99874998130653e-05,
      "loss": 0.0278,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045958056580275297,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66898956298828,
      "state_query_norm": 1425.947314453125,
      "state_relation_loss": 0.00045958056580275297,
      "step": 17580,
      "student_distance_mean": 1958.4292846679687,
      "teacher_distance_mean": 1.151310431957245,
      "weighted_auxiliary_loss": 0.00045958056580275297,
      "weighted_relation_loss": 0.00045958056580275297
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.454630495790457,
      "flow_matching_loss": 0.02626985274255276,
      "grad_norm": 0.10284408926963806,
      "learning_rate": 3.9933506623437455e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004720859637018293,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73882293701172,
      "state_query_norm": 1437.6134033203125,
      "state_relation_loss": 0.0004720859637018293,
      "step": 17590,
      "student_distance_mean": 1966.5823364257812,
      "teacher_distance_mean": 1.1535961031913757,
      "weighted_auxiliary_loss": 0.0004720859637018293,
      "weighted_relation_loss": 0.0004720859637018293
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.46398503274088,
      "flow_matching_loss": 0.02556659039109945,
      "grad_norm": 0.08977187424898148,
      "learning_rate": 3.9879525665548276e-05,
      "loss": 0.026,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000448356251581572,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64653205871582,
      "state_query_norm": 1430.785791015625,
      "state_relation_loss": 0.000448356251581572,
      "step": 17600,
      "student_distance_mean": 1959.3526977539063,
      "teacher_distance_mean": 1.1553288221359252,
      "weighted_auxiliary_loss": 0.000448356251581572,
      "weighted_relation_loss": 0.000448356251581572
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.4733395696913,
      "flow_matching_loss": 0.025555245578289032,
      "grad_norm": 0.08212367445230484,
      "learning_rate": 3.9825557004989704e-05,
      "loss": 0.026,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004797848727321252,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71399154663086,
      "state_query_norm": 1432.9062744140624,
      "state_relation_loss": 0.0004797848727321252,
      "step": 17610,
      "student_distance_mean": 1964.3406982421875,
      "teacher_distance_mean": 1.1590773820877076,
      "weighted_auxiliary_loss": 0.0004797848727321252,
      "weighted_relation_loss": 0.0004797848727321252
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.48269410664172,
      "flow_matching_loss": 0.026201015524566173,
      "grad_norm": 0.12156550586223602,
      "learning_rate": 3.977160070733877e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004870195174589753,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.871266555786136,
      "state_query_norm": 1447.1392578125,
      "state_relation_loss": 0.0004870195174589753,
      "step": 17620,
      "student_distance_mean": 1975.7235107421875,
      "teacher_distance_mean": 1.1653268933296204,
      "weighted_auxiliary_loss": 0.0004870195174589753,
      "weighted_relation_loss": 0.0004870195174589753
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.492048643592142,
      "flow_matching_loss": 0.027766731195151807,
      "grad_norm": 0.10809215158224106,
      "learning_rate": 3.971765683815746e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004623621003702283,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83943099975586,
      "state_query_norm": 1434.6323852539062,
      "state_relation_loss": 0.0004623621003702283,
      "step": 17630,
      "student_distance_mean": 1971.8809204101562,
      "teacher_distance_mean": 1.162138617038727,
      "weighted_auxiliary_loss": 0.0004623621003702283,
      "weighted_relation_loss": 0.0004623621003702283
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.501403180542564,
      "flow_matching_loss": 0.02584350425750017,
      "grad_norm": 0.10280336439609528,
      "learning_rate": 3.966372546299265e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004572619451209903,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.701420593261716,
      "state_query_norm": 1434.1125366210938,
      "state_relation_loss": 0.0004572619451209903,
      "step": 17640,
      "student_distance_mean": 1962.5517456054688,
      "teacher_distance_mean": 1.1583784699440003,
      "weighted_auxiliary_loss": 0.0004572619451209903,
      "weighted_relation_loss": 0.0004572619451209903
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.510757717492986,
      "flow_matching_loss": 0.02497897855937481,
      "grad_norm": 0.08675793558359146,
      "learning_rate": 3.960980664737604e-05,
      "loss": 0.0254,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004580443783197552,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.611823654174806,
      "state_query_norm": 1427.0740478515625,
      "state_relation_loss": 0.0004580443783197552,
      "step": 17650,
      "student_distance_mean": 1953.5184204101563,
      "teacher_distance_mean": 1.1563022136688232,
      "weighted_auxiliary_loss": 0.0004580443783197552,
      "weighted_relation_loss": 0.0004580443783197552
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.520112254443404,
      "flow_matching_loss": 0.026222556456923486,
      "grad_norm": 0.09564707428216934,
      "learning_rate": 3.955590045682408e-05,
      "loss": 0.0267,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004682076192693785,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.781157684326175,
      "state_query_norm": 1431.1288696289062,
      "state_relation_loss": 0.0004682076192693785,
      "step": 17660,
      "student_distance_mean": 1966.5049682617187,
      "teacher_distance_mean": 1.1640317797660829,
      "weighted_auxiliary_loss": 0.0004682076192693785,
      "weighted_relation_loss": 0.0004682076192693785
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.529466791393826,
      "flow_matching_loss": 0.024507811665534972,
      "grad_norm": 0.10143011808395386,
      "learning_rate": 3.950200695683788e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045003368577454237,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.70270881652832,
      "state_query_norm": 1391.4845825195312,
      "state_relation_loss": 0.00045003368577454237,
      "step": 17670,
      "student_distance_mean": 1908.3104370117187,
      "teacher_distance_mean": 1.1292422771453858,
      "weighted_auxiliary_loss": 0.00045003368577454237,
      "weighted_relation_loss": 0.00045003368577454237
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.53882132834425,
      "flow_matching_loss": 0.025126741081476212,
      "grad_norm": 0.11587351560592651,
      "learning_rate": 3.944812621290314e-05,
      "loss": 0.0256,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047697462723590435,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1350830078125,
      "state_query_norm": 1408.6113037109376,
      "state_relation_loss": 0.00047697462723590435,
      "step": 17680,
      "student_distance_mean": 1930.6518920898438,
      "teacher_distance_mean": 1.1402418732643127,
      "weighted_auxiliary_loss": 0.00047697462723590435,
      "weighted_relation_loss": 0.00047697462723590435
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.548175865294667,
      "flow_matching_loss": 0.025801443494856358,
      "grad_norm": 0.1319570690393448,
      "learning_rate": 3.939425829049002e-05,
      "loss": 0.0262,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044505408441182227,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28074493408203,
      "state_query_norm": 1418.9878051757812,
      "state_relation_loss": 0.00044505408441182227,
      "step": 17690,
      "student_distance_mean": 1943.0169555664063,
      "teacher_distance_mean": 1.146464228630066,
      "weighted_auxiliary_loss": 0.00044505408441182227,
      "weighted_relation_loss": 0.00044505408441182227
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.55753040224509,
      "flow_matching_loss": 0.026295449770987033,
      "grad_norm": 0.14064238965511322,
      "learning_rate": 3.9340403255053135e-05,
      "loss": 0.0268,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004595472157234326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.14499092102051,
      "state_query_norm": 1443.7344970703125,
      "state_relation_loss": 0.0004595472157234326,
      "step": 17700,
      "student_distance_mean": 1985.8203979492187,
      "teacher_distance_mean": 1.1701397895812988,
      "weighted_auxiliary_loss": 0.0004595472157234326,
      "weighted_relation_loss": 0.0004595472157234326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.56688493919551,
      "flow_matching_loss": 0.024518206901848315,
      "grad_norm": 0.1123705729842186,
      "learning_rate": 3.9286561172031407e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004982365557225421,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28287925720215,
      "state_query_norm": 1415.6834228515625,
      "state_relation_loss": 0.0004982365557225421,
      "step": 17710,
      "student_distance_mean": 1940.6005004882813,
      "teacher_distance_mean": 1.1400323390960694,
      "weighted_auxiliary_loss": 0.0004982365557225421,
      "weighted_relation_loss": 0.0004982365557225421
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.57623947614593,
      "flow_matching_loss": 0.025878985226154328,
      "grad_norm": 0.11667481809854507,
      "learning_rate": 3.923273210684809e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045836686622351407,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.563411712646484,
      "state_query_norm": 1426.6072265625,
      "state_relation_loss": 0.00045836686622351407,
      "step": 17720,
      "student_distance_mean": 1956.5767211914062,
      "teacher_distance_mean": 1.1528878808021545,
      "weighted_auxiliary_loss": 0.00045836686622351407,
      "weighted_relation_loss": 0.00045836686622351407
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.58559401309635,
      "flow_matching_loss": 0.025235387869179247,
      "grad_norm": 0.08183809369802475,
      "learning_rate": 3.9178916124910555e-05,
      "loss": 0.0257,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044513482716865835,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.30181884765625,
      "state_query_norm": 1463.056298828125,
      "state_relation_loss": 0.00044513482716865835,
      "step": 17730,
      "student_distance_mean": 1998.1217407226563,
      "teacher_distance_mean": 1.1750327110290528,
      "weighted_auxiliary_loss": 0.00044513482716865835,
      "weighted_relation_loss": 0.00044513482716865835
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.594948550046773,
      "flow_matching_loss": 0.025283741392195226,
      "grad_norm": 0.09583637863397598,
      "learning_rate": 3.9125113291610274e-05,
      "loss": 0.0258,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048723666986916214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.85024719238281,
      "state_query_norm": 1440.7636596679688,
      "state_relation_loss": 0.00048723666986916214,
      "step": 17740,
      "student_distance_mean": 1970.9275268554688,
      "teacher_distance_mean": 1.1606030225753785,
      "weighted_auxiliary_loss": 0.00048723666986916214,
      "weighted_relation_loss": 0.00048723666986916214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.604303086997195,
      "flow_matching_loss": 0.02607169169932604,
      "grad_norm": 0.1053847000002861,
      "learning_rate": 3.907132367232278e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004487291007535532,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.97606735229492,
      "state_query_norm": 1440.5256469726562,
      "state_relation_loss": 0.0004487291007535532,
      "step": 17750,
      "student_distance_mean": 1978.8807250976563,
      "teacher_distance_mean": 1.1656548976898193,
      "weighted_auxiliary_loss": 0.0004487291007535532,
      "weighted_relation_loss": 0.0004487291007535532
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.613657623947613,
      "flow_matching_loss": 0.026013453863561152,
      "grad_norm": 0.10004038363695145,
      "learning_rate": 3.901754733240752e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045891945483163,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.11771774291992,
      "state_query_norm": 1417.8657836914062,
      "state_relation_loss": 0.00045891945483163,
      "step": 17760,
      "student_distance_mean": 1933.1405517578125,
      "teacher_distance_mean": 1.1384516477584838,
      "weighted_auxiliary_loss": 0.00045891945483163,
      "weighted_relation_loss": 0.00045891945483163
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.623012160898035,
      "flow_matching_loss": 0.024316149950027465,
      "grad_norm": 0.09551914036273956,
      "learning_rate": 3.896378433720785e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004491008352488279,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1514949798584,
      "state_query_norm": 1413.649658203125,
      "state_relation_loss": 0.0004491008352488279,
      "step": 17770,
      "student_distance_mean": 1933.2014770507812,
      "teacher_distance_mean": 1.1391473412513733,
      "weighted_auxiliary_loss": 0.0004491008352488279,
      "weighted_relation_loss": 0.0004491008352488279
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.632366697848457,
      "flow_matching_loss": 0.025201172940433024,
      "grad_norm": 0.10439801961183548,
      "learning_rate": 3.891003475205086e-05,
      "loss": 0.0256,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044328604999464003,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.84527740478516,
      "state_query_norm": 1442.9891723632813,
      "state_relation_loss": 0.00044328604999464003,
      "step": 17780,
      "student_distance_mean": 1971.987939453125,
      "teacher_distance_mean": 1.1626195073127747,
      "weighted_auxiliary_loss": 0.00044328604999464003,
      "weighted_relation_loss": 0.00044328604999464003
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.641721234798876,
      "flow_matching_loss": 0.026942967809736727,
      "grad_norm": 0.09264792501926422,
      "learning_rate": 3.885629864224736e-05,
      "loss": 0.0274,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046166396350599825,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.860704803466795,
      "state_query_norm": 1445.3531127929687,
      "state_relation_loss": 0.00046166396350599825,
      "step": 17790,
      "student_distance_mean": 1975.6240112304688,
      "teacher_distance_mean": 1.161345410346985,
      "weighted_auxiliary_loss": 0.00046166396350599825,
      "weighted_relation_loss": 0.00046166396350599825
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.651075771749298,
      "flow_matching_loss": 0.025616683810949326,
      "grad_norm": 0.10315798223018646,
      "learning_rate": 3.880257607309178e-05,
      "loss": 0.0261,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046975809382274745,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.617499923706056,
      "state_query_norm": 1422.2439086914062,
      "state_relation_loss": 0.00046975809382274745,
      "step": 17800,
      "student_distance_mean": 1956.1642333984375,
      "teacher_distance_mean": 1.1517693161964417,
      "weighted_auxiliary_loss": 0.00046975809382274745,
      "weighted_relation_loss": 0.00046975809382274745
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.66043030869972,
      "flow_matching_loss": 0.025692068785429,
      "grad_norm": 0.09204109758138657,
      "learning_rate": 3.874886710986213e-05,
      "loss": 0.0262,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004590693512000144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6593879699707,
      "state_query_norm": 1434.562353515625,
      "state_relation_loss": 0.0004590693512000144,
      "step": 17810,
      "student_distance_mean": 1961.8639038085937,
      "teacher_distance_mean": 1.1585220575332642,
      "weighted_auxiliary_loss": 0.0004590693512000144,
      "weighted_relation_loss": 0.0004590693512000144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.669784845650142,
      "flow_matching_loss": 0.025847806967794894,
      "grad_norm": 0.10484706610441208,
      "learning_rate": 3.8695171817819834e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044824774377048013,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.091732788085935,
      "state_query_norm": 1409.9194946289062,
      "state_relation_loss": 0.00044824774377048013,
      "step": 17820,
      "student_distance_mean": 1926.9836303710938,
      "teacher_distance_mean": 1.135651743412018,
      "weighted_auxiliary_loss": 0.00044824774377048013,
      "weighted_relation_loss": 0.00044824774377048013
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.67913938260056,
      "flow_matching_loss": 0.02428485807031393,
      "grad_norm": 0.07831662893295288,
      "learning_rate": 3.8641490262209774e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004950265923980623,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58389434814453,
      "state_query_norm": 1422.8003295898438,
      "state_relation_loss": 0.0004950265923980623,
      "step": 17830,
      "student_distance_mean": 1951.6474609375,
      "teacher_distance_mean": 1.150382971763611,
      "weighted_auxiliary_loss": 0.0004950265923980623,
      "weighted_relation_loss": 0.0004950265923980623
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.688493919550982,
      "flow_matching_loss": 0.0242977486923337,
      "grad_norm": 0.10287725925445557,
      "learning_rate": 3.8587822508260084e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045989628706593065,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76371421813965,
      "state_query_norm": 1435.35380859375,
      "state_relation_loss": 0.00045989628706593065,
      "step": 17840,
      "student_distance_mean": 1966.7063110351562,
      "teacher_distance_mean": 1.1577349185943604,
      "weighted_auxiliary_loss": 0.00045989628706593065,
      "weighted_relation_loss": 0.00045989628706593065
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.697848456501404,
      "flow_matching_loss": 0.02432235740125179,
      "grad_norm": 0.08334685117006302,
      "learning_rate": 3.853416862118214e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005025326565373689,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.92843246459961,
      "state_query_norm": 1443.5927368164062,
      "state_relation_loss": 0.0005025326565373689,
      "step": 17850,
      "student_distance_mean": 1971.048583984375,
      "teacher_distance_mean": 1.1585273265838623,
      "weighted_auxiliary_loss": 0.0005025326565373689,
      "weighted_relation_loss": 0.0005025326565373689
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.707202993451823,
      "flow_matching_loss": 0.02581180240958929,
      "grad_norm": 0.10486574470996857,
      "learning_rate": 3.848052866617049e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004748897219542414,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75742874145508,
      "state_query_norm": 1443.0087646484376,
      "state_relation_loss": 0.0004748897219542414,
      "step": 17860,
      "student_distance_mean": 1966.1217407226563,
      "teacher_distance_mean": 1.155632781982422,
      "weighted_auxiliary_loss": 0.0004748897219542414,
      "weighted_relation_loss": 0.0004748897219542414
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.716557530402245,
      "flow_matching_loss": 0.026027235388755798,
      "grad_norm": 0.08156966418027878,
      "learning_rate": 3.84269027084027e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047558427904732526,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.01638221740723,
      "state_query_norm": 1442.6871704101563,
      "state_relation_loss": 0.00047558427904732526,
      "step": 17870,
      "student_distance_mean": 1976.9416870117188,
      "teacher_distance_mean": 1.165072536468506,
      "weighted_auxiliary_loss": 0.00047558427904732526,
      "weighted_relation_loss": 0.00047558427904732526
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.725912067352667,
      "flow_matching_loss": 0.02498995028436184,
      "grad_norm": 0.09233343601226807,
      "learning_rate": 3.83732908130394e-05,
      "loss": 0.0255,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004670249443734065,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74124374389648,
      "state_query_norm": 1440.8712158203125,
      "state_relation_loss": 0.0004670249443734065,
      "step": 17880,
      "student_distance_mean": 1963.1558959960937,
      "teacher_distance_mean": 1.157432770729065,
      "weighted_auxiliary_loss": 0.0004670249443734065,
      "weighted_relation_loss": 0.0004670249443734065
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.73526660430309,
      "flow_matching_loss": 0.02546946294605732,
      "grad_norm": 0.10640843957662582,
      "learning_rate": 3.831969304522409e-05,
      "loss": 0.0259,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004393999435706064,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42456283569336,
      "state_query_norm": 1422.9197631835937,
      "state_relation_loss": 0.0004393999435706064,
      "step": 17890,
      "student_distance_mean": 1950.7713134765625,
      "teacher_distance_mean": 1.1499917030334472,
      "weighted_auxiliary_loss": 0.0004393999435706064,
      "weighted_relation_loss": 0.0004393999435706064
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.744621141253507,
      "flow_matching_loss": 0.024934927001595497,
      "grad_norm": 0.09191147983074188,
      "learning_rate": 3.826610947008312e-05,
      "loss": 0.0254,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004490107938181609,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.134952545166016,
      "state_query_norm": 1418.0166870117187,
      "state_relation_loss": 0.0004490107938181609,
      "step": 17900,
      "student_distance_mean": 1933.6182495117187,
      "teacher_distance_mean": 1.1419794678688049,
      "weighted_auxiliary_loss": 0.0004490107938181609,
      "weighted_relation_loss": 0.0004490107938181609
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.75397567820393,
      "flow_matching_loss": 0.025342732854187488,
      "grad_norm": 0.09482937306165695,
      "learning_rate": 3.821254015272559e-05,
      "loss": 0.0258,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00048216791765298694,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40193977355957,
      "state_query_norm": 1425.3963623046875,
      "state_relation_loss": 0.00048216791765298694,
      "step": 17910,
      "student_distance_mean": 1947.3782348632812,
      "teacher_distance_mean": 1.150840198993683,
      "weighted_auxiliary_loss": 0.00048216791765298694,
      "weighted_relation_loss": 0.00048216791765298694
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.76333021515435,
      "flow_matching_loss": 0.023808674700558186,
      "grad_norm": 0.0976683720946312,
      "learning_rate": 3.8158985158243214e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0005069481761893257,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62403564453125,
      "state_query_norm": 1427.2447875976563,
      "state_relation_loss": 0.0005069481761893257,
      "step": 17920,
      "student_distance_mean": 1957.2265380859376,
      "teacher_distance_mean": 1.1526023626327515,
      "weighted_auxiliary_loss": 0.0005069481761893257,
      "weighted_relation_loss": 0.0005069481761893257
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.77268475210477,
      "flow_matching_loss": 0.026531165838241576,
      "grad_norm": 0.1388310343027115,
      "learning_rate": 3.810544455171044e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004578903957735747,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.395009231567386,
      "state_query_norm": 1428.6867919921874,
      "state_relation_loss": 0.0004578903957735747,
      "step": 17930,
      "student_distance_mean": 1949.522802734375,
      "teacher_distance_mean": 1.1499904155731202,
      "weighted_auxiliary_loss": 0.0004578903957735747,
      "weighted_relation_loss": 0.0004578903957735747
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.78203928905519,
      "flow_matching_loss": 0.025854507833719252,
      "grad_norm": 0.0909944400191307,
      "learning_rate": 3.8051918398184114e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004698516830103472,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21583442687988,
      "state_query_norm": 1428.7389282226563,
      "state_relation_loss": 0.0004698516830103472,
      "step": 17940,
      "student_distance_mean": 1941.8628295898438,
      "teacher_distance_mean": 1.1473511099815368,
      "weighted_auxiliary_loss": 0.0004698516830103472,
      "weighted_relation_loss": 0.0004698516830103472
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.791393826005613,
      "flow_matching_loss": 0.024980066530406474,
      "grad_norm": 0.11232192069292068,
      "learning_rate": 3.799840676270356e-05,
      "loss": 0.0254,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043838704004883766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50567626953125,
      "state_query_norm": 1424.2079711914062,
      "state_relation_loss": 0.00043838704004883766,
      "step": 17950,
      "student_distance_mean": 1953.2367553710938,
      "teacher_distance_mean": 1.1535358786582948,
      "weighted_auxiliary_loss": 0.00043838704004883766,
      "weighted_relation_loss": 0.00043838704004883766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.800748362956035,
      "flow_matching_loss": 0.024435599520802497,
      "grad_norm": 0.10781247168779373,
      "learning_rate": 3.794490971029048e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004607261507771909,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33447685241699,
      "state_query_norm": 1420.4677368164062,
      "state_relation_loss": 0.0004607261507771909,
      "step": 17960,
      "student_distance_mean": 1944.329931640625,
      "teacher_distance_mean": 1.1465398788452148,
      "weighted_auxiliary_loss": 0.0004607261507771909,
      "weighted_relation_loss": 0.0004607261507771909
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.810102899906454,
      "flow_matching_loss": 0.0240764282643795,
      "grad_norm": 0.09390221536159515,
      "learning_rate": 3.789142730594881e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042719003395177426,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.249477005004884,
      "state_query_norm": 1412.12099609375,
      "state_relation_loss": 0.00042719003395177426,
      "step": 17970,
      "student_distance_mean": 1938.05537109375,
      "teacher_distance_mean": 1.1431795954704285,
      "weighted_auxiliary_loss": 0.00042719003395177426,
      "weighted_relation_loss": 0.00042719003395177426
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.819457436856876,
      "flow_matching_loss": 0.025897657312452794,
      "grad_norm": 0.12406150251626968,
      "learning_rate": 3.783795961466471e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004396556760184467,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63401412963867,
      "state_query_norm": 1437.975537109375,
      "state_relation_loss": 0.0004396556760184467,
      "step": 17980,
      "student_distance_mean": 1962.407958984375,
      "teacher_distance_mean": 1.1571165919303894,
      "weighted_auxiliary_loss": 0.0004396556760184467,
      "weighted_relation_loss": 0.0004396556760184467
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.828811973807298,
      "flow_matching_loss": 0.02424110546708107,
      "grad_norm": 0.09265416860580444,
      "learning_rate": 3.778450670140651e-05,
      "loss": 0.0247,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004520003392826766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74956130981445,
      "state_query_norm": 1433.2080322265624,
      "state_relation_loss": 0.0004520003392826766,
      "step": 17990,
      "student_distance_mean": 1961.8975830078125,
      "teacher_distance_mean": 1.1568586587905885,
      "weighted_auxiliary_loss": 0.0004520003392826766,
      "weighted_relation_loss": 0.0004520003392826766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.838166510757716,
      "flow_matching_loss": 0.023547262884676456,
      "grad_norm": 0.10683523118495941,
      "learning_rate": 3.773106863112451e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043887936626560984,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.881612396240236,
      "state_query_norm": 1446.0593627929688,
      "state_relation_loss": 0.00043887936626560984,
      "step": 18000,
      "student_distance_mean": 1975.1098388671876,
      "teacher_distance_mean": 1.1615525007247924,
      "weighted_auxiliary_loss": 0.00043887936626560984,
      "weighted_relation_loss": 0.00043887936626560984
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.84752104770814,
      "flow_matching_loss": 0.026906380802392958,
      "grad_norm": 0.10307062417268753,
      "learning_rate": 3.7677645468751e-05,
      "loss": 0.0273,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043709104065783324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.017007446289064,
      "state_query_norm": 1409.8010131835938,
      "state_relation_loss": 0.00043709104065783324,
      "step": 18010,
      "student_distance_mean": 1926.1152709960938,
      "teacher_distance_mean": 1.1340132236480713,
      "weighted_auxiliary_loss": 0.00043709104065783324,
      "weighted_relation_loss": 0.00043709104065783324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.85687558465856,
      "flow_matching_loss": 0.02773743439465761,
      "grad_norm": 0.09977930039167404,
      "learning_rate": 3.762423727920018e-05,
      "loss": 0.0282,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004621527623385191,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2064868927002,
      "state_query_norm": 1414.6328979492187,
      "state_relation_loss": 0.0004621527623385191,
      "step": 18020,
      "student_distance_mean": 1935.761181640625,
      "teacher_distance_mean": 1.1411838293075562,
      "weighted_auxiliary_loss": 0.0004621527623385191,
      "weighted_relation_loss": 0.0004621527623385191
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.86623012160898,
      "flow_matching_loss": 0.02439666073769331,
      "grad_norm": 0.12228803336620331,
      "learning_rate": 3.7570844127367995e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045470221375580876,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71107559204101,
      "state_query_norm": 1433.185400390625,
      "state_relation_loss": 0.00045470221375580876,
      "step": 18030,
      "student_distance_mean": 1963.3133666992187,
      "teacher_distance_mean": 1.1566615223884582,
      "weighted_auxiliary_loss": 0.00045470221375580876,
      "weighted_relation_loss": 0.00045470221375580876
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.8755846585594,
      "flow_matching_loss": 0.02602888997644186,
      "grad_norm": 0.12305399775505066,
      "learning_rate": 3.751746607813221e-05,
      "loss": 0.0265,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004905378475086764,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.88287773132324,
      "state_query_norm": 1451.2679565429687,
      "state_relation_loss": 0.0004905378475086764,
      "step": 18040,
      "student_distance_mean": 1976.7694702148438,
      "teacher_distance_mean": 1.1656098365783691,
      "weighted_auxiliary_loss": 0.0004905378475086764,
      "weighted_relation_loss": 0.0004905378475086764
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.884939195509823,
      "flow_matching_loss": 0.025609470903873444,
      "grad_norm": 0.10657911747694016,
      "learning_rate": 3.746410319635217e-05,
      "loss": 0.0261,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004573799786157906,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3544921875,
      "state_query_norm": 1421.24033203125,
      "state_relation_loss": 0.0004573799786157906,
      "step": 18050,
      "student_distance_mean": 1946.2619262695312,
      "teacher_distance_mean": 1.1500407338142395,
      "weighted_auxiliary_loss": 0.0004573799786157906,
      "weighted_relation_loss": 0.0004573799786157906
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.894293732460245,
      "flow_matching_loss": 0.024888195656239988,
      "grad_norm": 0.10783076286315918,
      "learning_rate": 3.7410755546868806e-05,
      "loss": 0.0254,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004706650070147589,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.430366134643556,
      "state_query_norm": 1427.7216430664062,
      "state_relation_loss": 0.0004706650070147589,
      "step": 18060,
      "student_distance_mean": 1948.2921508789063,
      "teacher_distance_mean": 1.151029074192047,
      "weighted_auxiliary_loss": 0.0004706650070147589,
      "weighted_relation_loss": 0.0004706650070147589
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.903648269410663,
      "flow_matching_loss": 0.02471495382487774,
      "grad_norm": 0.10017422586679459,
      "learning_rate": 3.735742319450454e-05,
      "loss": 0.0252,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004746848891954869,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83949241638184,
      "state_query_norm": 1436.0317016601562,
      "state_relation_loss": 0.0004746848891954869,
      "step": 18070,
      "student_distance_mean": 1968.1341064453125,
      "teacher_distance_mean": 1.1595834136009215,
      "weighted_auxiliary_loss": 0.0004746848891954869,
      "weighted_relation_loss": 0.0004746848891954869
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.913002806361085,
      "flow_matching_loss": 0.025825254060328005,
      "grad_norm": 0.10493586957454681,
      "learning_rate": 3.7304106204063184e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045708905381616207,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.577980041503906,
      "state_query_norm": 1423.8816650390625,
      "state_relation_loss": 0.00045708905381616207,
      "step": 18080,
      "student_distance_mean": 1955.6878173828125,
      "teacher_distance_mean": 1.1513850092887878,
      "weighted_auxiliary_loss": 0.00045708905381616207,
      "weighted_relation_loss": 0.00045708905381616207
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.922357343311507,
      "flow_matching_loss": 0.02581757362931967,
      "grad_norm": 0.1203056126832962,
      "learning_rate": 3.725080464032996e-05,
      "loss": 0.0263,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004548653465462849,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.482543563842775,
      "state_query_norm": 1422.7873901367188,
      "state_relation_loss": 0.0004548653465462849,
      "step": 18090,
      "student_distance_mean": 1950.5248046875,
      "teacher_distance_mean": 1.1538751602172852,
      "weighted_auxiliary_loss": 0.0004548653465462849,
      "weighted_relation_loss": 0.0004548653465462849
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.931711880261926,
      "flow_matching_loss": 0.02453257292509079,
      "grad_norm": 0.12273011356592178,
      "learning_rate": 3.719751856807125e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043133940489497035,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.84125518798828,
      "state_query_norm": 1406.4674560546875,
      "state_relation_loss": 0.00043133940489497035,
      "step": 18100,
      "student_distance_mean": 1920.9718872070312,
      "teacher_distance_mean": 1.1374045968055726,
      "weighted_auxiliary_loss": 0.00043133940489497035,
      "weighted_relation_loss": 0.00043133940489497035
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.941066417212348,
      "flow_matching_loss": 0.02482763435691595,
      "grad_norm": 0.13882136344909668,
      "learning_rate": 3.71442480520347e-05,
      "loss": 0.0253,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046046749339438974,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23820114135742,
      "state_query_norm": 1415.0777709960937,
      "state_relation_loss": 0.00046046749339438974,
      "step": 18110,
      "student_distance_mean": 1936.3600952148438,
      "teacher_distance_mean": 1.1461135149002075,
      "weighted_auxiliary_loss": 0.00046046749339438974,
      "weighted_relation_loss": 0.00046046749339438974
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.95042095416277,
      "flow_matching_loss": 0.02751009352505207,
      "grad_norm": 0.09166863560676575,
      "learning_rate": 3.709099315694897e-05,
      "loss": 0.028,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004755183996167034,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.165034484863284,
      "state_query_norm": 1408.52939453125,
      "state_relation_loss": 0.0004755183996167034,
      "step": 18120,
      "student_distance_mean": 1931.1409301757812,
      "teacher_distance_mean": 1.143651533126831,
      "weighted_auxiliary_loss": 0.0004755183996167034,
      "weighted_relation_loss": 0.0004755183996167034
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.95977549111319,
      "flow_matching_loss": 0.02654099799692631,
      "grad_norm": 0.09450274705886841,
      "learning_rate": 3.703775394752378e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047381784534081815,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93426704406738,
      "state_query_norm": 1444.0201538085937,
      "state_relation_loss": 0.00047381784534081815,
      "step": 18130,
      "student_distance_mean": 1975.2807250976562,
      "teacher_distance_mean": 1.1678680896759033,
      "weighted_auxiliary_loss": 0.00047381784534081815,
      "weighted_relation_loss": 0.00047381784534081815
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.96913002806361,
      "flow_matching_loss": 0.023911355622112752,
      "grad_norm": 0.10968271642923355,
      "learning_rate": 3.698453048844983e-05,
      "loss": 0.0244,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044231143838260325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.204749298095706,
      "state_query_norm": 1424.8680786132813,
      "state_relation_loss": 0.00044231143838260325,
      "step": 18140,
      "student_distance_mean": 1941.4071411132813,
      "teacher_distance_mean": 1.1421587586402893,
      "weighted_auxiliary_loss": 0.00044231143838260325,
      "weighted_relation_loss": 0.00044231143838260325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.978484565014032,
      "flow_matching_loss": 0.02626944687217474,
      "grad_norm": 0.09205832332372665,
      "learning_rate": 3.693132284439861e-05,
      "loss": 0.0268,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004856087325606495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.87345199584961,
      "state_query_norm": 1434.3337036132812,
      "state_relation_loss": 0.0004856087325606495,
      "step": 18150,
      "student_distance_mean": 1970.621826171875,
      "teacher_distance_mean": 1.159554922580719,
      "weighted_auxiliary_loss": 0.0004856087325606495,
      "weighted_relation_loss": 0.0004856087325606495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.987839101964454,
      "flow_matching_loss": 0.025524971820414067,
      "grad_norm": 0.10765504091978073,
      "learning_rate": 3.6878131080022414e-05,
      "loss": 0.026,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004599798092385754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.11190071105957,
      "state_query_norm": 1416.9582763671874,
      "state_relation_loss": 0.0004599798092385754,
      "step": 18160,
      "student_distance_mean": 1929.964306640625,
      "teacher_distance_mean": 1.135447871685028,
      "weighted_auxiliary_loss": 0.0004599798092385754,
      "weighted_relation_loss": 0.0004599798092385754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 16.997193638914872,
      "flow_matching_loss": 0.02617518659681082,
      "grad_norm": 0.12340135127305984,
      "learning_rate": 3.682495525995429e-05,
      "loss": 0.0266,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004494707682169974,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45505599975586,
      "state_query_norm": 1422.4884887695312,
      "state_relation_loss": 0.0004494707682169974,
      "step": 18170,
      "student_distance_mean": 1948.88662109375,
      "teacher_distance_mean": 1.1478163838386535,
      "weighted_auxiliary_loss": 0.0004494707682169974,
      "weighted_relation_loss": 0.0004494707682169974
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.006548175865294,
      "flow_matching_loss": 0.027048039622604846,
      "grad_norm": 0.08916699886322021,
      "learning_rate": 3.6771795448807846e-05,
      "loss": 0.0275,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00045066914753988385,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.86108207702637,
      "state_query_norm": 1446.89267578125,
      "state_relation_loss": 0.00045066914753988385,
      "step": 18180,
      "student_distance_mean": 1974.60810546875,
      "teacher_distance_mean": 1.1641815304756165,
      "weighted_auxiliary_loss": 0.00045066914753988385,
      "weighted_relation_loss": 0.00045066914753988385
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.015902712815716,
      "flow_matching_loss": 0.02314926665276289,
      "grad_norm": 0.08973290771245956,
      "learning_rate": 3.671865171117724e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047999608505051584,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5819206237793,
      "state_query_norm": 1420.7800537109374,
      "state_relation_loss": 0.00047999608505051584,
      "step": 18190,
      "student_distance_mean": 1952.1621826171875,
      "teacher_distance_mean": 1.151285219192505,
      "weighted_auxiliary_loss": 0.00047999608505051584,
      "weighted_relation_loss": 0.00047999608505051584
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.02525724976614,
      "flow_matching_loss": 0.02478539776057005,
      "grad_norm": 0.09623076766729355,
      "learning_rate": 3.666552411163718e-05,
      "loss": 0.0252,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004533616360276937,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62743225097656,
      "state_query_norm": 1433.7060791015624,
      "state_relation_loss": 0.0004533616360276937,
      "step": 18200,
      "student_distance_mean": 1960.0926879882813,
      "teacher_distance_mean": 1.1563848614692689,
      "weighted_auxiliary_loss": 0.0004533616360276937,
      "weighted_relation_loss": 0.0004533616360276937
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.034611786716557,
      "flow_matching_loss": 0.02486314531415701,
      "grad_norm": 0.10274534672498703,
      "learning_rate": 3.661241271474269e-05,
      "loss": 0.0253,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004668939422117546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.21105842590332,
      "state_query_norm": 1449.4581298828125,
      "state_relation_loss": 0.0004668939422117546,
      "step": 18210,
      "student_distance_mean": 1994.066455078125,
      "teacher_distance_mean": 1.174748134613037,
      "weighted_auxiliary_loss": 0.0004668939422117546,
      "weighted_relation_loss": 0.0004668939422117546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.04396632366698,
      "flow_matching_loss": 0.024767574667930604,
      "grad_norm": 0.10820547491312027,
      "learning_rate": 3.6559317585029116e-05,
      "loss": 0.0252,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004550700803520158,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78206596374512,
      "state_query_norm": 1432.7493041992188,
      "state_relation_loss": 0.0004550700803520158,
      "step": 18220,
      "student_distance_mean": 1964.6561279296875,
      "teacher_distance_mean": 1.1584587454795838,
      "weighted_auxiliary_loss": 0.0004550700803520158,
      "weighted_relation_loss": 0.0004550700803520158
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.0533208606174,
      "flow_matching_loss": 0.02408674154430628,
      "grad_norm": 0.08325772732496262,
      "learning_rate": 3.650623878701204e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044411413255147637,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24767990112305,
      "state_query_norm": 1416.5205200195312,
      "state_relation_loss": 0.00044411413255147637,
      "step": 18230,
      "student_distance_mean": 1941.1532470703125,
      "teacher_distance_mean": 1.1462987899780273,
      "weighted_auxiliary_loss": 0.00044411413255147637,
      "weighted_relation_loss": 0.00044411413255147637
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.06267539756782,
      "flow_matching_loss": 0.025147431530058384,
      "grad_norm": 0.10616082698106766,
      "learning_rate": 3.645317638518721e-05,
      "loss": 0.0256,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004560787288937718,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72601890563965,
      "state_query_norm": 1440.4235473632812,
      "state_relation_loss": 0.0004560787288937718,
      "step": 18240,
      "student_distance_mean": 1966.0611572265625,
      "teacher_distance_mean": 1.1607818007469177,
      "weighted_auxiliary_loss": 0.0004560787288937718,
      "weighted_relation_loss": 0.0004560787288937718
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.07202993451824,
      "flow_matching_loss": 0.025344842299818992,
      "grad_norm": 0.11770261824131012,
      "learning_rate": 3.640013044403046e-05,
      "loss": 0.0258,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004472658154554665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.99038124084473,
      "state_query_norm": 1402.4990234375,
      "state_relation_loss": 0.0004472658154554665,
      "step": 18250,
      "student_distance_mean": 1920.2218872070312,
      "teacher_distance_mean": 1.1383649706840515,
      "weighted_auxiliary_loss": 0.0004472658154554665,
      "weighted_relation_loss": 0.0004472658154554665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.081384471468663,
      "flow_matching_loss": 0.02589299567043781,
      "grad_norm": 0.1005103588104248,
      "learning_rate": 3.634710102799761e-05,
      "loss": 0.0264,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004708111344370991,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.466526412963866,
      "state_query_norm": 1423.4253051757812,
      "state_relation_loss": 0.0004708111344370991,
      "step": 18260,
      "student_distance_mean": 1950.0247192382812,
      "teacher_distance_mean": 1.1501754760742187,
      "weighted_auxiliary_loss": 0.0004708111344370991,
      "weighted_relation_loss": 0.0004708111344370991
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.09073900841908,
      "flow_matching_loss": 0.023975908197462557,
      "grad_norm": 0.08219588547945023,
      "learning_rate": 3.6294088201524394e-05,
      "loss": 0.0244,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004576159379212186,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2961124420166,
      "state_query_norm": 1427.3299194335937,
      "state_relation_loss": 0.0004576159379212186,
      "step": 18270,
      "student_distance_mean": 1945.8540161132812,
      "teacher_distance_mean": 1.1472936034202577,
      "weighted_auxiliary_loss": 0.0004576159379212186,
      "weighted_relation_loss": 0.0004576159379212186
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.100093545369504,
      "flow_matching_loss": 0.02552518416196108,
      "grad_norm": 0.10421691089868546,
      "learning_rate": 3.62410920290264e-05,
      "loss": 0.026,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004694633942563087,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69861907958985,
      "state_query_norm": 1434.509033203125,
      "state_relation_loss": 0.0004694633942563087,
      "step": 18280,
      "student_distance_mean": 1963.9030029296875,
      "teacher_distance_mean": 1.1583327293395995,
      "weighted_auxiliary_loss": 0.0004694633942563087,
      "weighted_relation_loss": 0.0004694633942563087
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.109448082319926,
      "flow_matching_loss": 0.02363262865692377,
      "grad_norm": 0.10720483958721161,
      "learning_rate": 3.6188112574898954e-05,
      "loss": 0.0241,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004838289372855797,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.284030532836915,
      "state_query_norm": 1422.3151611328126,
      "state_relation_loss": 0.0004838289372855797,
      "step": 18290,
      "student_distance_mean": 1943.6254760742188,
      "teacher_distance_mean": 1.150248110294342,
      "weighted_auxiliary_loss": 0.0004838289372855797,
      "weighted_relation_loss": 0.0004838289372855797
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.118802619270348,
      "flow_matching_loss": 0.026484057307243347,
      "grad_norm": 0.09831542521715164,
      "learning_rate": 3.6135149903517115e-05,
      "loss": 0.027,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047329614171758293,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74210777282715,
      "state_query_norm": 1436.3439453125,
      "state_relation_loss": 0.00047329614171758293,
      "step": 18300,
      "student_distance_mean": 1963.8767578125,
      "teacher_distance_mean": 1.1592155337333678,
      "weighted_auxiliary_loss": 0.00047329614171758293,
      "weighted_relation_loss": 0.00047329614171758293
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.128157156220766,
      "flow_matching_loss": 0.02673819027841091,
      "grad_norm": 0.11479221284389496,
      "learning_rate": 3.608220407923552e-05,
      "loss": 0.0272,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044728150533046573,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21851272583008,
      "state_query_norm": 1419.8011962890625,
      "state_relation_loss": 0.00044728150533046573,
      "step": 18310,
      "student_distance_mean": 1938.7418090820313,
      "teacher_distance_mean": 1.1439246654510498,
      "weighted_auxiliary_loss": 0.00044728150533046573,
      "weighted_relation_loss": 0.00044728150533046573
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.137511693171188,
      "flow_matching_loss": 0.02515985406935215,
      "grad_norm": 0.11616653949022293,
      "learning_rate": 3.602927516638833e-05,
      "loss": 0.0256,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004577958083245903,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89670372009277,
      "state_query_norm": 1436.9931274414062,
      "state_relation_loss": 0.0004577958083245903,
      "step": 18320,
      "student_distance_mean": 1975.1707153320312,
      "teacher_distance_mean": 1.1605148434638977,
      "weighted_auxiliary_loss": 0.0004577958083245903,
      "weighted_relation_loss": 0.0004577958083245903
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.14686623012161,
      "flow_matching_loss": 0.022484372556209564,
      "grad_norm": 0.0998481959104538,
      "learning_rate": 3.5976363229289165e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045664201606996355,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14257926940918,
      "state_query_norm": 1405.2241821289062,
      "state_relation_loss": 0.00045664201606996355,
      "step": 18330,
      "student_distance_mean": 1931.3656616210938,
      "teacher_distance_mean": 1.1377094745635987,
      "weighted_auxiliary_loss": 0.00045664201606996355,
      "weighted_relation_loss": 0.00045664201606996355
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.15622076707203,
      "flow_matching_loss": 0.02412975449115038,
      "grad_norm": 0.10780130326747894,
      "learning_rate": 3.5923468332231e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042688141693361106,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57499694824219,
      "state_query_norm": 1432.9242431640625,
      "state_relation_loss": 0.00042688141693361106,
      "step": 18340,
      "student_distance_mean": 1958.80224609375,
      "teacher_distance_mean": 1.1563632607460022,
      "weighted_auxiliary_loss": 0.00042688141693361106,
      "weighted_relation_loss": 0.00042688141693361106
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.16557530402245,
      "flow_matching_loss": 0.02382282018661499,
      "grad_norm": 0.08276782184839249,
      "learning_rate": 3.587059053948616e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044965056877117605,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23249435424805,
      "state_query_norm": 1424.1899047851562,
      "state_relation_loss": 0.00044965056877117605,
      "step": 18350,
      "student_distance_mean": 1939.391015625,
      "teacher_distance_mean": 1.1460908770561218,
      "weighted_auxiliary_loss": 0.00044965056877117605,
      "weighted_relation_loss": 0.00044965056877117605
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.174929840972872,
      "flow_matching_loss": 0.025139984115958212,
      "grad_norm": 0.10325220972299576,
      "learning_rate": 3.5817729915306134e-05,
      "loss": 0.0256,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045978880662005394,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37620582580566,
      "state_query_norm": 1426.9658203125,
      "state_relation_loss": 0.00045978880662005394,
      "step": 18360,
      "student_distance_mean": 1947.5982177734375,
      "teacher_distance_mean": 1.1504396557807923,
      "weighted_auxiliary_loss": 0.00045978880662005394,
      "weighted_relation_loss": 0.00045978880662005394
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.184284377923294,
      "flow_matching_loss": 0.02323669958859682,
      "grad_norm": 0.1114719808101654,
      "learning_rate": 3.5764886523921566e-05,
      "loss": 0.0237,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004478724644286558,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.24678497314453,
      "state_query_norm": 1457.8828857421875,
      "state_relation_loss": 0.0004478724644286558,
      "step": 18370,
      "student_distance_mean": 1992.732958984375,
      "teacher_distance_mean": 1.1744998455047608,
      "weighted_auxiliary_loss": 0.0004478724644286558,
      "weighted_relation_loss": 0.0004478724644286558
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.193638914873713,
      "flow_matching_loss": 0.025379634462296963,
      "grad_norm": 0.09423746168613434,
      "learning_rate": 3.571206042954214e-05,
      "loss": 0.0258,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004405144165502861,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45344200134277,
      "state_query_norm": 1424.2176391601563,
      "state_relation_loss": 0.0004405144165502861,
      "step": 18380,
      "student_distance_mean": 1946.1296264648438,
      "teacher_distance_mean": 1.1461900234222413,
      "weighted_auxiliary_loss": 0.0004405144165502861,
      "weighted_relation_loss": 0.0004405144165502861
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.202993451824135,
      "flow_matching_loss": 0.02412466686218977,
      "grad_norm": 0.1251329630613327,
      "learning_rate": 3.5659251696356566e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043163554510101677,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.06615219116211,
      "state_query_norm": 1446.2598510742187,
      "state_relation_loss": 0.00043163554510101677,
      "step": 18390,
      "student_distance_mean": 1982.1432495117188,
      "teacher_distance_mean": 1.1660974621772766,
      "weighted_auxiliary_loss": 0.00043163554510101677,
      "weighted_relation_loss": 0.00043163554510101677
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.212347988774557,
      "flow_matching_loss": 0.023323920927941798,
      "grad_norm": 0.09463100880384445,
      "learning_rate": 3.56064603885324e-05,
      "loss": 0.0238,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004538924840744585,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55924072265625,
      "state_query_norm": 1429.8558715820313,
      "state_relation_loss": 0.0004538924840744585,
      "step": 18400,
      "student_distance_mean": 1956.9921752929688,
      "teacher_distance_mean": 1.1537066102027893,
      "weighted_auxiliary_loss": 0.0004538924840744585,
      "weighted_relation_loss": 0.0004538924840744585
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.221702525724975,
      "flow_matching_loss": 0.024579946137964725,
      "grad_norm": 0.12050296366214752,
      "learning_rate": 3.5553686570216115e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004336801444878802,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.05971755981445,
      "state_query_norm": 1415.9565795898438,
      "state_relation_loss": 0.0004336801444878802,
      "step": 18410,
      "student_distance_mean": 1932.374365234375,
      "teacher_distance_mean": 1.1397736191749572,
      "weighted_auxiliary_loss": 0.0004336801444878802,
      "weighted_relation_loss": 0.0004336801444878802
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.231057062675397,
      "flow_matching_loss": 0.02467960920184851,
      "grad_norm": 0.08587899804115295,
      "learning_rate": 3.550093030553284e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044322625326458364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38816375732422,
      "state_query_norm": 1415.2931274414063,
      "state_relation_loss": 0.00044322625326458364,
      "step": 18420,
      "student_distance_mean": 1942.2908203125,
      "teacher_distance_mean": 1.1505951046943665,
      "weighted_auxiliary_loss": 0.00044322625326458364,
      "weighted_relation_loss": 0.00044322625326458364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.24041159962582,
      "flow_matching_loss": 0.024016373604536057,
      "grad_norm": 0.0886932834982872,
      "learning_rate": 3.544819165858642e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044399850303307177,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.46553039550781,
      "state_query_norm": 1432.4512451171875,
      "state_relation_loss": 0.00044399850303307177,
      "step": 18430,
      "student_distance_mean": 1951.8328002929688,
      "teacher_distance_mean": 1.1590349435806275,
      "weighted_auxiliary_loss": 0.00044399850303307177,
      "weighted_relation_loss": 0.00044399850303307177
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.24976613657624,
      "flow_matching_loss": 0.023985212482511998,
      "grad_norm": 0.11014848947525024,
      "learning_rate": 3.539547069345926e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004935015604132786,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.666312789916994,
      "state_query_norm": 1427.0599365234375,
      "state_relation_loss": 0.0004935015604132786,
      "step": 18440,
      "student_distance_mean": 1958.8509155273437,
      "teacher_distance_mean": 1.16240496635437,
      "weighted_auxiliary_loss": 0.0004935015604132786,
      "weighted_relation_loss": 0.0004935015604132786
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.25912067352666,
      "flow_matching_loss": 0.02427287455648184,
      "grad_norm": 0.09891027957201004,
      "learning_rate": 3.5342767474212344e-05,
      "loss": 0.0247,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004580709879519418,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5388988494873,
      "state_query_norm": 1425.399658203125,
      "state_relation_loss": 0.0004580709879519418,
      "step": 18450,
      "student_distance_mean": 1951.1733642578124,
      "teacher_distance_mean": 1.152447235584259,
      "weighted_auxiliary_loss": 0.0004580709879519418,
      "weighted_relation_loss": 0.0004580709879519418
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.26847521047708,
      "flow_matching_loss": 0.023566041700541975,
      "grad_norm": 0.11104917526245117,
      "learning_rate": 3.529008206488502e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044725184270646425,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63502426147461,
      "state_query_norm": 1425.5024047851562,
      "state_relation_loss": 0.00044725184270646425,
      "step": 18460,
      "student_distance_mean": 1955.6742431640625,
      "teacher_distance_mean": 1.154809331893921,
      "weighted_auxiliary_loss": 0.00044725184270646425,
      "weighted_relation_loss": 0.00044725184270646425
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.277829747427504,
      "flow_matching_loss": 0.022953847609460353,
      "grad_norm": 0.10874193161725998,
      "learning_rate": 3.5237414529495055e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043152540747541933,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.143012619018556,
      "state_query_norm": 1414.8161865234374,
      "state_relation_loss": 0.00043152540747541933,
      "step": 18470,
      "student_distance_mean": 1932.4696899414062,
      "teacher_distance_mean": 1.141460382938385,
      "weighted_auxiliary_loss": 0.00043152540747541933,
      "weighted_relation_loss": 0.00043152540747541933
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.287184284377922,
      "flow_matching_loss": 0.02180613111704588,
      "grad_norm": 0.09387461841106415,
      "learning_rate": 3.5184764932038454e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042830177990254014,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.137799453735354,
      "state_query_norm": 1420.2514404296876,
      "state_relation_loss": 0.00042830177990254014,
      "step": 18480,
      "student_distance_mean": 1938.050341796875,
      "teacher_distance_mean": 1.148571741580963,
      "weighted_auxiliary_loss": 0.00042830177990254014,
      "weighted_relation_loss": 0.00042830177990254014
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.296538821328344,
      "flow_matching_loss": 0.02383471969515085,
      "grad_norm": 0.08724801242351532,
      "learning_rate": 3.513213333648945e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045083812437951565,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45682373046875,
      "state_query_norm": 1424.1859252929687,
      "state_relation_loss": 0.00045083812437951565,
      "step": 18490,
      "student_distance_mean": 1951.1650756835938,
      "teacher_distance_mean": 1.1545805215835572,
      "weighted_auxiliary_loss": 0.00045083812437951565,
      "weighted_relation_loss": 0.00045083812437951565
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.305893358278766,
      "flow_matching_loss": 0.0223475307226181,
      "grad_norm": 0.09026773273944855,
      "learning_rate": 3.5079519806800374e-05,
      "loss": 0.0228,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004913479584502056,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74207038879395,
      "state_query_norm": 1435.9214965820313,
      "state_relation_loss": 0.0004913479584502056,
      "step": 18500,
      "student_distance_mean": 1965.0671020507812,
      "teacher_distance_mean": 1.160312008857727,
      "weighted_auxiliary_loss": 0.0004913479584502056,
      "weighted_relation_loss": 0.0004913479584502056
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.315247895229184,
      "flow_matching_loss": 0.025025430880486966,
      "grad_norm": 0.11476868391036987,
      "learning_rate": 3.502692440690165e-05,
      "loss": 0.0255,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004518000816460699,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5467586517334,
      "state_query_norm": 1432.7132202148437,
      "state_relation_loss": 0.0004518000816460699,
      "step": 18510,
      "student_distance_mean": 1957.9797485351562,
      "teacher_distance_mean": 1.1546109795570374,
      "weighted_auxiliary_loss": 0.0004518000816460699,
      "weighted_relation_loss": 0.0004518000816460699
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.324602432179606,
      "flow_matching_loss": 0.025348583050072194,
      "grad_norm": 0.08977547287940979,
      "learning_rate": 3.497434720070165e-05,
      "loss": 0.0258,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043671741150319576,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.178167724609374,
      "state_query_norm": 1419.362646484375,
      "state_relation_loss": 0.00043671741150319576,
      "step": 18520,
      "student_distance_mean": 1938.471142578125,
      "teacher_distance_mean": 1.1463314414024353,
      "weighted_auxiliary_loss": 0.00043671741150319576,
      "weighted_relation_loss": 0.00043671741150319576
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.33395696913003,
      "flow_matching_loss": 0.024358388595283033,
      "grad_norm": 0.1049223244190216,
      "learning_rate": 3.4921788252086616e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043349903135094794,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.19289588928223,
      "state_query_norm": 1416.5526733398438,
      "state_relation_loss": 0.00043349903135094794,
      "step": 18530,
      "student_distance_mean": 1938.3824584960937,
      "teacher_distance_mean": 1.148036503791809,
      "weighted_auxiliary_loss": 0.00043349903135094794,
      "weighted_relation_loss": 0.00043349903135094794
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.34331150608045,
      "flow_matching_loss": 0.024353972263634206,
      "grad_norm": 0.11047115176916122,
      "learning_rate": 3.486924762492065e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004517317865975201,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.632018661499025,
      "state_query_norm": 1420.2680908203124,
      "state_relation_loss": 0.0004517317865975201,
      "step": 18540,
      "student_distance_mean": 1954.8230590820312,
      "teacher_distance_mean": 1.1603929281234742,
      "weighted_auxiliary_loss": 0.0004517317865975201,
      "weighted_relation_loss": 0.0004517317865975201
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.35266604303087,
      "flow_matching_loss": 0.02359877061098814,
      "grad_norm": 0.11334653943777084,
      "learning_rate": 3.4816725383045534e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004416007170220837,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.563622665405276,
      "state_query_norm": 1433.846142578125,
      "state_relation_loss": 0.0004416007170220837,
      "step": 18550,
      "student_distance_mean": 1957.9921020507813,
      "teacher_distance_mean": 1.1594677925109864,
      "weighted_auxiliary_loss": 0.0004416007170220837,
      "weighted_relation_loss": 0.0004416007170220837
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.36202057998129,
      "flow_matching_loss": 0.0250297699123621,
      "grad_norm": 0.12021221220493317,
      "learning_rate": 3.476422159028079e-05,
      "loss": 0.0255,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004577439045533538,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.411936569213864,
      "state_query_norm": 1419.0456787109374,
      "state_relation_loss": 0.0004577439045533538,
      "step": 18560,
      "student_distance_mean": 1944.8318725585937,
      "teacher_distance_mean": 1.1498892545700072,
      "weighted_auxiliary_loss": 0.0004577439045533538,
      "weighted_relation_loss": 0.0004577439045533538
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.371375116931713,
      "flow_matching_loss": 0.023551493138074874,
      "grad_norm": 0.11751633882522583,
      "learning_rate": 3.471173631042345e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004575780068989843,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57504539489746,
      "state_query_norm": 1424.8475341796875,
      "state_relation_loss": 0.0004575780068989843,
      "step": 18570,
      "student_distance_mean": 1955.2678100585938,
      "teacher_distance_mean": 1.1540685176849366,
      "weighted_auxiliary_loss": 0.0004575780068989843,
      "weighted_relation_loss": 0.0004575780068989843
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.38072965388213,
      "flow_matching_loss": 0.023859753832221033,
      "grad_norm": 0.11354189366102219,
      "learning_rate": 3.465926960724808e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004818598012207076,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57909660339355,
      "state_query_norm": 1425.220849609375,
      "state_relation_loss": 0.0004818598012207076,
      "step": 18580,
      "student_distance_mean": 1951.9111328125,
      "teacher_distance_mean": 1.1556706309318543,
      "weighted_auxiliary_loss": 0.0004818598012207076,
      "weighted_relation_loss": 0.0004818598012207076
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.390084190832553,
      "flow_matching_loss": 0.023528219759464265,
      "grad_norm": 0.11432374268770218,
      "learning_rate": 3.460682154450666e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043203112145420166,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33531951904297,
      "state_query_norm": 1422.8295043945313,
      "state_relation_loss": 0.00043203112145420166,
      "step": 18590,
      "student_distance_mean": 1944.235693359375,
      "teacher_distance_mean": 1.1493026733398437,
      "weighted_auxiliary_loss": 0.00043203112145420166,
      "weighted_relation_loss": 0.00043203112145420166
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.399438727782975,
      "flow_matching_loss": 0.024626407772302628,
      "grad_norm": 0.09782634675502777,
      "learning_rate": 3.4554392185928564e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004453396541066468,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.46917724609375,
      "state_query_norm": 1422.76767578125,
      "state_relation_loss": 0.0004453396541066468,
      "step": 18600,
      "student_distance_mean": 1949.0794921875,
      "teacher_distance_mean": 1.1521679878234863,
      "weighted_auxiliary_loss": 0.0004453396541066468,
      "weighted_relation_loss": 0.0004453396541066468
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.408793264733397,
      "flow_matching_loss": 0.024991273880004883,
      "grad_norm": 0.08566123247146606,
      "learning_rate": 3.450198159522037e-05,
      "loss": 0.0254,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004397951386636123,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.490987396240236,
      "state_query_norm": 1424.68564453125,
      "state_relation_loss": 0.0004397951386636123,
      "step": 18610,
      "student_distance_mean": 1950.3048583984375,
      "teacher_distance_mean": 1.1506227016448975,
      "weighted_auxiliary_loss": 0.0004397951386636123,
      "weighted_relation_loss": 0.0004397951386636123
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.418147801683816,
      "flow_matching_loss": 0.022679393552243708,
      "grad_norm": 0.08633440732955933,
      "learning_rate": 3.444958983606592e-05,
      "loss": 0.0231,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043230046576354654,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.11821022033691,
      "state_query_norm": 1417.3385620117188,
      "state_relation_loss": 0.00043230046576354654,
      "step": 18620,
      "student_distance_mean": 1936.9840576171875,
      "teacher_distance_mean": 1.1405609846115112,
      "weighted_auxiliary_loss": 0.00043230046576354654,
      "weighted_relation_loss": 0.00043230046576354654
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.427502338634238,
      "flow_matching_loss": 0.023741403222084047,
      "grad_norm": 0.10576868057250977,
      "learning_rate": 3.439721697212612e-05,
      "loss": 0.0242,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045546208857558665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5770824432373,
      "state_query_norm": 1431.0753295898437,
      "state_relation_loss": 0.00045546208857558665,
      "step": 18630,
      "student_distance_mean": 1956.7584716796875,
      "teacher_distance_mean": 1.1579137086868285,
      "weighted_auxiliary_loss": 0.00045546208857558665,
      "weighted_relation_loss": 0.00045546208857558665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.43685687558466,
      "flow_matching_loss": 0.023025083355605602,
      "grad_norm": 0.09609894454479218,
      "learning_rate": 3.4344863067038954e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004583915724651888,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.82570457458496,
      "state_query_norm": 1430.7389038085937,
      "state_relation_loss": 0.0004583915724651888,
      "step": 18640,
      "student_distance_mean": 1966.3295776367188,
      "teacher_distance_mean": 1.1610333323478699,
      "weighted_auxiliary_loss": 0.0004583915724651888,
      "weighted_relation_loss": 0.0004583915724651888
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.446211412535078,
      "flow_matching_loss": 0.02429684139788151,
      "grad_norm": 0.09474320709705353,
      "learning_rate": 3.429252818441935e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004787914251210168,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.600182342529294,
      "state_query_norm": 1428.3475830078125,
      "state_relation_loss": 0.0004787914251210168,
      "step": 18650,
      "student_distance_mean": 1954.8498046875,
      "teacher_distance_mean": 1.1492127537727357,
      "weighted_auxiliary_loss": 0.0004787914251210168,
      "weighted_relation_loss": 0.0004787914251210168
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.4555659494855,
      "flow_matching_loss": 0.024468685127794744,
      "grad_norm": 0.0933467447757721,
      "learning_rate": 3.4240212387859094e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004477586568100378,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41925163269043,
      "state_query_norm": 1433.5406372070313,
      "state_relation_loss": 0.0004477586568100378,
      "step": 18660,
      "student_distance_mean": 1952.1905029296875,
      "teacher_distance_mean": 1.1488799095153808,
      "weighted_auxiliary_loss": 0.0004477586568100378,
      "weighted_relation_loss": 0.0004477586568100378
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.464920486435922,
      "flow_matching_loss": 0.025005827844142913,
      "grad_norm": 0.09944648295640945,
      "learning_rate": 3.418791574092686e-05,
      "loss": 0.0255,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004868320800596848,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59060096740723,
      "state_query_norm": 1422.5891479492188,
      "state_relation_loss": 0.0004868320800596848,
      "step": 18670,
      "student_distance_mean": 1954.9452270507813,
      "teacher_distance_mean": 1.1490787863731384,
      "weighted_auxiliary_loss": 0.0004868320800596848,
      "weighted_relation_loss": 0.0004868320800596848
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.474275023386344,
      "flow_matching_loss": 0.02316603697836399,
      "grad_norm": 0.08281790465116501,
      "learning_rate": 3.413563830716796e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045564963656943294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73857040405274,
      "state_query_norm": 1438.4364135742187,
      "state_relation_loss": 0.00045564963656943294,
      "step": 18680,
      "student_distance_mean": 1966.8310791015624,
      "teacher_distance_mean": 1.1523037791252135,
      "weighted_auxiliary_loss": 0.00045564963656943294,
      "weighted_relation_loss": 0.00045564963656943294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.483629560336762,
      "flow_matching_loss": 0.02483172658830881,
      "grad_norm": 0.11036733537912369,
      "learning_rate": 3.408338015010445e-05,
      "loss": 0.0253,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004530435166088864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50761260986328,
      "state_query_norm": 1423.7480346679688,
      "state_relation_loss": 0.0004530435166088864,
      "step": 18690,
      "student_distance_mean": 1948.5179809570313,
      "teacher_distance_mean": 1.1420190453529357,
      "weighted_auxiliary_loss": 0.0004530435166088864,
      "weighted_relation_loss": 0.0004530435166088864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.492984097287184,
      "flow_matching_loss": 0.024582302756607534,
      "grad_norm": 0.11255720257759094,
      "learning_rate": 3.40311413332349e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004629806237062439,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.84055595397949,
      "state_query_norm": 1433.3119995117188,
      "state_relation_loss": 0.0004629806237062439,
      "step": 18700,
      "student_distance_mean": 1968.4115600585938,
      "teacher_distance_mean": 1.1571585416793824,
      "weighted_auxiliary_loss": 0.0004629806237062439,
      "weighted_relation_loss": 0.0004629806237062439
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.502338634237606,
      "flow_matching_loss": 0.023279201239347458,
      "grad_norm": 0.08120032399892807,
      "learning_rate": 3.3978921920034365e-05,
      "loss": 0.0237,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042220155883114784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60043067932129,
      "state_query_norm": 1429.8369995117187,
      "state_relation_loss": 0.00042220155883114784,
      "step": 18710,
      "student_distance_mean": 1955.86533203125,
      "teacher_distance_mean": 1.1548179984092712,
      "weighted_auxiliary_loss": 0.00042220155883114784,
      "weighted_relation_loss": 0.00042220155883114784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.511693171188025,
      "flow_matching_loss": 0.022730806656181813,
      "grad_norm": 0.136189267039299,
      "learning_rate": 3.392672197395441e-05,
      "loss": 0.0232,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004906903428491205,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.805655288696286,
      "state_query_norm": 1441.901513671875,
      "state_relation_loss": 0.0004906903428491205,
      "step": 18720,
      "student_distance_mean": 1967.2992431640625,
      "teacher_distance_mean": 1.16203533411026,
      "weighted_auxiliary_loss": 0.0004906903428491205,
      "weighted_relation_loss": 0.0004906903428491205
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.521047708138447,
      "flow_matching_loss": 0.023143376410007476,
      "grad_norm": 0.06744664162397385,
      "learning_rate": 3.387454155842287e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045718888868577777,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.345694351196286,
      "state_query_norm": 1417.4571166992187,
      "state_relation_loss": 0.00045718888868577777,
      "step": 18730,
      "student_distance_mean": 1941.4659057617187,
      "teacher_distance_mean": 1.1444382548332215,
      "weighted_auxiliary_loss": 0.00045718888868577777,
      "weighted_relation_loss": 0.00045718888868577777
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.53040224508887,
      "flow_matching_loss": 0.023493818193674087,
      "grad_norm": 0.09332380443811417,
      "learning_rate": 3.382238073684386e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042459835240151735,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.530386734008786,
      "state_query_norm": 1428.91220703125,
      "state_relation_loss": 0.00042459835240151735,
      "step": 18740,
      "student_distance_mean": 1956.5786987304687,
      "teacher_distance_mean": 1.150855052471161,
      "weighted_auxiliary_loss": 0.00042459835240151735,
      "weighted_relation_loss": 0.00042459835240151735
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.53975678203929,
      "flow_matching_loss": 0.022556521371006964,
      "grad_norm": 0.08771321177482605,
      "learning_rate": 3.377023957259771e-05,
      "loss": 0.023,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004543542687315494,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.765814208984374,
      "state_query_norm": 1436.0126098632813,
      "state_relation_loss": 0.0004543542687315494,
      "step": 18750,
      "student_distance_mean": 1964.648486328125,
      "teacher_distance_mean": 1.1568164825439453,
      "weighted_auxiliary_loss": 0.0004543542687315494,
      "weighted_relation_loss": 0.0004543542687315494
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.54911131898971,
      "flow_matching_loss": 0.02295084614306688,
      "grad_norm": 0.07980167865753174,
      "learning_rate": 3.3718118129040835e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044908525596838446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39391746520996,
      "state_query_norm": 1422.603857421875,
      "state_relation_loss": 0.00044908525596838446,
      "step": 18760,
      "student_distance_mean": 1948.5948608398437,
      "teacher_distance_mean": 1.1503297328948974,
      "weighted_auxiliary_loss": 0.00044908525596838446,
      "weighted_relation_loss": 0.00044908525596838446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.55846585594013,
      "flow_matching_loss": 0.022432247549295424,
      "grad_norm": 0.08531299978494644,
      "learning_rate": 3.3666016469505724e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045264510554261507,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.404018783569335,
      "state_query_norm": 1420.5040283203125,
      "state_relation_loss": 0.00045264510554261507,
      "step": 18770,
      "student_distance_mean": 1947.7769287109375,
      "teacher_distance_mean": 1.146893608570099,
      "weighted_auxiliary_loss": 0.00045264510554261507,
      "weighted_relation_loss": 0.00045264510554261507
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.567820392890553,
      "flow_matching_loss": 0.02269340939819813,
      "grad_norm": 0.07782671600580215,
      "learning_rate": 3.361393465730079e-05,
      "loss": 0.0232,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047142715484369544,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02120475769043,
      "state_query_norm": 1439.9436767578125,
      "state_relation_loss": 0.00047142715484369544,
      "step": 18780,
      "student_distance_mean": 1976.9666870117187,
      "teacher_distance_mean": 1.1659506797790526,
      "weighted_auxiliary_loss": 0.00047142715484369544,
      "weighted_relation_loss": 0.00047142715484369544
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.57717492984097,
      "flow_matching_loss": 0.026452822797000407,
      "grad_norm": 0.1037171334028244,
      "learning_rate": 3.3561872755710366e-05,
      "loss": 0.0269,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043427709024399517,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.986975479125974,
      "state_query_norm": 1436.9858154296876,
      "state_relation_loss": 0.00043427709024399517,
      "step": 18790,
      "student_distance_mean": 1977.9382568359374,
      "teacher_distance_mean": 1.1643980264663696,
      "weighted_auxiliary_loss": 0.00043427709024399517,
      "weighted_relation_loss": 0.00043427709024399517
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.586529466791394,
      "flow_matching_loss": 0.022963646054267883,
      "grad_norm": 0.09125208109617233,
      "learning_rate": 3.350983082799456e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004418820928549394,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44854888916016,
      "state_query_norm": 1431.6588012695313,
      "state_relation_loss": 0.0004418820928549394,
      "step": 18800,
      "student_distance_mean": 1952.1275634765625,
      "teacher_distance_mean": 1.1482037425041198,
      "weighted_auxiliary_loss": 0.0004418820928549394,
      "weighted_relation_loss": 0.0004418820928549394
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.595884003741816,
      "flow_matching_loss": 0.02221533115953207,
      "grad_norm": 0.08562559634447098,
      "learning_rate": 3.34578089373892e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004502954776398838,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.497886657714844,
      "state_query_norm": 1423.5458984375,
      "state_relation_loss": 0.0004502954776398838,
      "step": 18810,
      "student_distance_mean": 1951.6310913085938,
      "teacher_distance_mean": 1.1475991129875183,
      "weighted_auxiliary_loss": 0.0004502954776398838,
      "weighted_relation_loss": 0.0004502954776398838
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.605238540692234,
      "flow_matching_loss": 0.02312650326639414,
      "grad_norm": 0.09672902524471283,
      "learning_rate": 3.340580714710581e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046687057183589784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.926962280273436,
      "state_query_norm": 1442.4323852539062,
      "state_relation_loss": 0.00046687057183589784,
      "step": 18820,
      "student_distance_mean": 1976.8596801757812,
      "teacher_distance_mean": 1.165028941631317,
      "weighted_auxiliary_loss": 0.00046687057183589784,
      "weighted_relation_loss": 0.00046687057183589784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.614593077642656,
      "flow_matching_loss": 0.024433102272450925,
      "grad_norm": 0.10080300271511078,
      "learning_rate": 3.3353825520331466e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045659384923055766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26157569885254,
      "state_query_norm": 1418.3610229492188,
      "state_relation_loss": 0.00045659384923055766,
      "step": 18830,
      "student_distance_mean": 1942.1156005859375,
      "teacher_distance_mean": 1.1459642410278321,
      "weighted_auxiliary_loss": 0.00045659384923055766,
      "weighted_relation_loss": 0.00045659384923055766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.623947614593078,
      "flow_matching_loss": 0.02312835119664669,
      "grad_norm": 0.09231606870889664,
      "learning_rate": 3.330186412022876e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044382684864103795,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63477630615235,
      "state_query_norm": 1434.114208984375,
      "state_relation_loss": 0.00044382684864103795,
      "step": 18840,
      "student_distance_mean": 1961.2345336914063,
      "teacher_distance_mean": 1.1561706066131592,
      "weighted_auxiliary_loss": 0.00044382684864103795,
      "weighted_relation_loss": 0.00044382684864103795
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.6333021515435,
      "flow_matching_loss": 0.0215968469157815,
      "grad_norm": 0.07451960444450378,
      "learning_rate": 3.324992300993568e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004465438920306042,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23622283935547,
      "state_query_norm": 1419.5544311523438,
      "state_relation_loss": 0.0004465438920306042,
      "step": 18850,
      "student_distance_mean": 1937.4822998046875,
      "teacher_distance_mean": 1.1401342511177064,
      "weighted_auxiliary_loss": 0.0004465438920306042,
      "weighted_relation_loss": 0.0004465438920306042
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.64265668849392,
      "flow_matching_loss": 0.022881416045129298,
      "grad_norm": 0.0936121791601181,
      "learning_rate": 3.319800225256556e-05,
      "loss": 0.0233,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004326285154093057,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33250350952149,
      "state_query_norm": 1418.0036743164062,
      "state_relation_loss": 0.0004326285154093057,
      "step": 18860,
      "student_distance_mean": 1943.107470703125,
      "teacher_distance_mean": 1.146496868133545,
      "weighted_auxiliary_loss": 0.0004326285154093057,
      "weighted_relation_loss": 0.0004326285154093057
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.65201122544434,
      "flow_matching_loss": 0.02376386392861605,
      "grad_norm": 0.11007814109325409,
      "learning_rate": 3.314610191120702e-05,
      "loss": 0.0242,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004349776281742379,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.703560256958006,
      "state_query_norm": 1438.355810546875,
      "state_relation_loss": 0.0004349776281742379,
      "step": 18870,
      "student_distance_mean": 1964.6969848632812,
      "teacher_distance_mean": 1.1581339478492736,
      "weighted_auxiliary_loss": 0.0004349776281742379,
      "weighted_relation_loss": 0.0004349776281742379
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.661365762394762,
      "flow_matching_loss": 0.024588933773338796,
      "grad_norm": 0.11791609972715378,
      "learning_rate": 3.30942220489239e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046118074678815903,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44591217041015,
      "state_query_norm": 1421.6659545898438,
      "state_relation_loss": 0.00046118074678815903,
      "step": 18880,
      "student_distance_mean": 1947.2970458984375,
      "teacher_distance_mean": 1.1479864239692688,
      "weighted_auxiliary_loss": 0.00046118074678815903,
      "weighted_relation_loss": 0.00046118074678815903
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.67072029934518,
      "flow_matching_loss": 0.02429213225841522,
      "grad_norm": 0.09673197567462921,
      "learning_rate": 3.3042362728755086e-05,
      "loss": 0.0247,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042674657015595585,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.939408493041995,
      "state_query_norm": 1404.9400756835937,
      "state_relation_loss": 0.00042674657015595585,
      "step": 18890,
      "student_distance_mean": 1922.7830078125,
      "teacher_distance_mean": 1.128597366809845,
      "weighted_auxiliary_loss": 0.00042674657015595585,
      "weighted_relation_loss": 0.00042674657015595585
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.680074836295603,
      "flow_matching_loss": 0.024684666097164153,
      "grad_norm": 0.1087881326675415,
      "learning_rate": 3.299052401371456e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004416290728840977,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.012176513671875,
      "state_query_norm": 1448.436865234375,
      "state_relation_loss": 0.0004416290728840977,
      "step": 18900,
      "student_distance_mean": 1982.0482666015625,
      "teacher_distance_mean": 1.1622509837150574,
      "weighted_auxiliary_loss": 0.0004416290728840977,
      "weighted_relation_loss": 0.0004416290728840977
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.689429373246025,
      "flow_matching_loss": 0.022994365729391574,
      "grad_norm": 0.10306032001972198,
      "learning_rate": 3.293870596679125e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046636342885904015,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55757064819336,
      "state_query_norm": 1431.4812255859374,
      "state_relation_loss": 0.00046636342885904015,
      "step": 18910,
      "student_distance_mean": 1955.9091552734376,
      "teacher_distance_mean": 1.1469114780426026,
      "weighted_auxiliary_loss": 0.00046636342885904015,
      "weighted_relation_loss": 0.00046636342885904015
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.698783910196447,
      "flow_matching_loss": 0.025053122825920583,
      "grad_norm": 0.11721289157867432,
      "learning_rate": 3.288690865094895e-05,
      "loss": 0.0255,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042623362969607115,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48696517944336,
      "state_query_norm": 1426.9837646484375,
      "state_relation_loss": 0.00042623362969607115,
      "step": 18920,
      "student_distance_mean": 1953.9951538085938,
      "teacher_distance_mean": 1.150583553314209,
      "weighted_auxiliary_loss": 0.00042623362969607115,
      "weighted_relation_loss": 0.00042623362969607115
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.708138447146865,
      "flow_matching_loss": 0.024786622636020182,
      "grad_norm": 0.09031296521425247,
      "learning_rate": 3.283513212912632e-05,
      "loss": 0.0253,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004736245144158602,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65818634033203,
      "state_query_norm": 1423.6779418945312,
      "state_relation_loss": 0.0004736245144158602,
      "step": 18930,
      "student_distance_mean": 1957.2802978515624,
      "teacher_distance_mean": 1.155176305770874,
      "weighted_auxiliary_loss": 0.0004736245144158602,
      "weighted_relation_loss": 0.0004736245144158602
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.717492984097287,
      "flow_matching_loss": 0.023512225411832333,
      "grad_norm": 0.10304932296276093,
      "learning_rate": 3.2783376464236684e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043695627537090333,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.18582611083984,
      "state_query_norm": 1411.18642578125,
      "state_relation_loss": 0.00043695627537090333,
      "step": 18940,
      "student_distance_mean": 1933.2917358398438,
      "teacher_distance_mean": 1.1378998041152955,
      "weighted_auxiliary_loss": 0.00043695627537090333,
      "weighted_relation_loss": 0.00043695627537090333
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.72684752104771,
      "flow_matching_loss": 0.023003225959837437,
      "grad_norm": 0.11343738436698914,
      "learning_rate": 3.273164171916806e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043829624773934485,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.426458740234374,
      "state_query_norm": 1423.66123046875,
      "state_relation_loss": 0.00043829624773934485,
      "step": 18950,
      "student_distance_mean": 1951.221044921875,
      "teacher_distance_mean": 1.1468315124511719,
      "weighted_auxiliary_loss": 0.00043829624773934485,
      "weighted_relation_loss": 0.00043829624773934485
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.736202057998128,
      "flow_matching_loss": 0.02452117521315813,
      "grad_norm": 0.08694101870059967,
      "learning_rate": 3.267992795678306e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004412324517033994,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.862309265136716,
      "state_query_norm": 1438.2438354492188,
      "state_relation_loss": 0.0004412324517033994,
      "step": 18960,
      "student_distance_mean": 1971.1115478515626,
      "teacher_distance_mean": 1.1568910121917724,
      "weighted_auxiliary_loss": 0.0004412324517033994,
      "weighted_relation_loss": 0.0004412324517033994
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.74555659494855,
      "flow_matching_loss": 0.024160480313003063,
      "grad_norm": 0.09054136276245117,
      "learning_rate": 3.2628235239918744e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042538428388070313,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.369777297973634,
      "state_query_norm": 1431.9008178710938,
      "state_relation_loss": 0.00042538428388070313,
      "step": 18970,
      "student_distance_mean": 1950.3848266601562,
      "teacher_distance_mean": 1.144698441028595,
      "weighted_auxiliary_loss": 0.00042538428388070313,
      "weighted_relation_loss": 0.00042538428388070313
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.75491113189897,
      "flow_matching_loss": 0.023890754207968712,
      "grad_norm": 0.09629461914300919,
      "learning_rate": 3.2576563631386695e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004310375079512596,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55576705932617,
      "state_query_norm": 1420.89384765625,
      "state_relation_loss": 0.0004310375079512596,
      "step": 18980,
      "student_distance_mean": 1951.5536499023438,
      "teacher_distance_mean": 1.1488758206367493,
      "weighted_auxiliary_loss": 0.0004310375079512596,
      "weighted_relation_loss": 0.0004310375079512596
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.764265668849394,
      "flow_matching_loss": 0.02287479769438505,
      "grad_norm": 0.1568981409072876,
      "learning_rate": 3.252491319397275e-05,
      "loss": 0.0233,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004471987922443077,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.644147872924805,
      "state_query_norm": 1437.8864868164062,
      "state_relation_loss": 0.0004471987922443077,
      "step": 18990,
      "student_distance_mean": 1961.6354125976563,
      "teacher_distance_mean": 1.157749879360199,
      "weighted_auxiliary_loss": 0.0004471987922443077,
      "weighted_relation_loss": 0.0004471987922443077
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.773620205799812,
      "flow_matching_loss": 0.024115212820470334,
      "grad_norm": 0.10023009032011032,
      "learning_rate": 3.247328399043706e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004455500893527642,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32620697021484,
      "state_query_norm": 1413.8021484375,
      "state_relation_loss": 0.0004455500893527642,
      "step": 19000,
      "student_distance_mean": 1940.0298095703124,
      "teacher_distance_mean": 1.1423198342323304,
      "weighted_auxiliary_loss": 0.0004455500893527642,
      "weighted_relation_loss": 0.0004455500893527642
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.782974742750234,
      "flow_matching_loss": 0.02413985375314951,
      "grad_norm": 0.09858041256666183,
      "learning_rate": 3.242167608351399e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004390028858324513,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.938546371459964,
      "state_query_norm": 1446.47158203125,
      "state_relation_loss": 0.0004390028858324513,
      "step": 19010,
      "student_distance_mean": 1979.5849609375,
      "teacher_distance_mean": 1.1627779364585877,
      "weighted_auxiliary_loss": 0.0004390028858324513,
      "weighted_relation_loss": 0.0004390028858324513
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.792329279700656,
      "flow_matching_loss": 0.02398003302514553,
      "grad_norm": 0.08140344172716141,
      "learning_rate": 3.2370089535911986e-05,
      "loss": 0.0244,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044006036478094757,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76416854858398,
      "state_query_norm": 1433.1067016601562,
      "state_relation_loss": 0.00044006036478094757,
      "step": 19020,
      "student_distance_mean": 1967.8393310546876,
      "teacher_distance_mean": 1.1555923581123353,
      "weighted_auxiliary_loss": 0.00044006036478094757,
      "weighted_relation_loss": 0.00044006036478094757
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.801683816651074,
      "flow_matching_loss": 0.02406855672597885,
      "grad_norm": 0.1078798770904541,
      "learning_rate": 3.23185244103136e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004626736947102472,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.33250427246094,
      "state_query_norm": 1459.559228515625,
      "state_relation_loss": 0.0004626736947102472,
      "step": 19030,
      "student_distance_mean": 1994.7691528320313,
      "teacher_distance_mean": 1.1725727319717407,
      "weighted_auxiliary_loss": 0.0004626736947102472,
      "weighted_relation_loss": 0.0004626736947102472
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.811038353601496,
      "flow_matching_loss": 0.02328080926090479,
      "grad_norm": 0.09147075563669205,
      "learning_rate": 3.22669807693753e-05,
      "loss": 0.0237,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004400184436235577,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52254028320313,
      "state_query_norm": 1414.1541748046875,
      "state_relation_loss": 0.0004400184436235577,
      "step": 19040,
      "student_distance_mean": 1947.31806640625,
      "teacher_distance_mean": 1.147283959388733,
      "weighted_auxiliary_loss": 0.0004400184436235577,
      "weighted_relation_loss": 0.0004400184436235577
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.82039289055192,
      "flow_matching_loss": 0.026834774389863016,
      "grad_norm": 0.11383505165576935,
      "learning_rate": 3.2215458675727495e-05,
      "loss": 0.0273,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046627827978227285,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.221923446655275,
      "state_query_norm": 1452.8258666992188,
      "state_relation_loss": 0.00046627827978227285,
      "step": 19050,
      "student_distance_mean": 1990.0823852539063,
      "teacher_distance_mean": 1.1705835103988647,
      "weighted_auxiliary_loss": 0.00046627827978227285,
      "weighted_relation_loss": 0.00046627827978227285
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.82974742750234,
      "flow_matching_loss": 0.024633790738880633,
      "grad_norm": 0.1075667068362236,
      "learning_rate": 3.216395819197438e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045328159467317164,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.132342147827146,
      "state_query_norm": 1411.6294799804687,
      "state_relation_loss": 0.00045328159467317164,
      "step": 19060,
      "student_distance_mean": 1929.4957275390625,
      "teacher_distance_mean": 1.130681300163269,
      "weighted_auxiliary_loss": 0.00045328159467317164,
      "weighted_relation_loss": 0.00045328159467317164
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.83910196445276,
      "flow_matching_loss": 0.02301290687173605,
      "grad_norm": 0.11062897741794586,
      "learning_rate": 3.211247938069387e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046361042477656157,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.815483474731444,
      "state_query_norm": 1439.72255859375,
      "state_relation_loss": 0.00046361042477656157,
      "step": 19070,
      "student_distance_mean": 1966.4054809570312,
      "teacher_distance_mean": 1.1512354731559753,
      "weighted_auxiliary_loss": 0.00046361042477656157,
      "weighted_relation_loss": 0.00046361042477656157
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.84845650140318,
      "flow_matching_loss": 0.024252121895551683,
      "grad_norm": 0.09292544424533844,
      "learning_rate": 3.206102230443759e-05,
      "loss": 0.0247,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004648620903026313,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.79136085510254,
      "state_query_norm": 1434.1391723632812,
      "state_relation_loss": 0.0004648620903026313,
      "step": 19080,
      "student_distance_mean": 1962.89208984375,
      "teacher_distance_mean": 1.1506526470184326,
      "weighted_auxiliary_loss": 0.0004648620903026313,
      "weighted_relation_loss": 0.0004648620903026313
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.857811038353603,
      "flow_matching_loss": 0.023765083588659763,
      "grad_norm": 0.10142325609922409,
      "learning_rate": 3.2009587025730764e-05,
      "loss": 0.0242,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004616844089468941,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70081825256348,
      "state_query_norm": 1426.74296875,
      "state_relation_loss": 0.0004616844089468941,
      "step": 19090,
      "student_distance_mean": 1958.7812866210938,
      "teacher_distance_mean": 1.1498946666717529,
      "weighted_auxiliary_loss": 0.0004616844089468941,
      "weighted_relation_loss": 0.0004616844089468941
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.86716557530402,
      "flow_matching_loss": 0.023447991348803043,
      "grad_norm": 0.10550285130739212,
      "learning_rate": 3.195817360707207e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041329418017994614,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.12326850891113,
      "state_query_norm": 1406.6400146484375,
      "state_relation_loss": 0.00041329418017994614,
      "step": 19100,
      "student_distance_mean": 1929.102734375,
      "teacher_distance_mean": 1.136446237564087,
      "weighted_auxiliary_loss": 0.00041329418017994614,
      "weighted_relation_loss": 0.00041329418017994614
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.876520112254443,
      "flow_matching_loss": 0.021839995868504047,
      "grad_norm": 0.08371321111917496,
      "learning_rate": 3.19067821109337e-05,
      "loss": 0.0223,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042724616068881007,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.456411361694336,
      "state_query_norm": 1425.3513549804688,
      "state_relation_loss": 0.00042724616068881007,
      "step": 19110,
      "student_distance_mean": 1954.3729370117187,
      "teacher_distance_mean": 1.151511311531067,
      "weighted_auxiliary_loss": 0.00042724616068881007,
      "weighted_relation_loss": 0.00042724616068881007
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.885874649204865,
      "flow_matching_loss": 0.02302711047232151,
      "grad_norm": 0.07903695851564407,
      "learning_rate": 3.185541259976114e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004312020289944485,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.402273941040036,
      "state_query_norm": 1422.980029296875,
      "state_relation_loss": 0.0004312020289944485,
      "step": 19120,
      "student_distance_mean": 1946.9736206054688,
      "teacher_distance_mean": 1.1469911456108093,
      "weighted_auxiliary_loss": 0.0004312020289944485,
      "weighted_relation_loss": 0.0004312020289944485
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.895229186155284,
      "flow_matching_loss": 0.02243843525648117,
      "grad_norm": 0.0822557583451271,
      "learning_rate": 3.180406513597316e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004243404313456267,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76103401184082,
      "state_query_norm": 1438.7900634765624,
      "state_relation_loss": 0.0004243404313456267,
      "step": 19130,
      "student_distance_mean": 1970.1833618164062,
      "teacher_distance_mean": 1.1573589205741883,
      "weighted_auxiliary_loss": 0.0004243404313456267,
      "weighted_relation_loss": 0.0004243404313456267
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.904583723105706,
      "flow_matching_loss": 0.02451081071048975,
      "grad_norm": 0.11186874657869339,
      "learning_rate": 3.1752739781961835e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000423586941906251,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.588168716430665,
      "state_query_norm": 1429.70146484375,
      "state_relation_loss": 0.000423586941906251,
      "step": 19140,
      "student_distance_mean": 1959.5495239257812,
      "teacher_distance_mean": 1.153748106956482,
      "weighted_auxiliary_loss": 0.000423586941906251,
      "weighted_relation_loss": 0.000423586941906251
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.913938260056128,
      "flow_matching_loss": 0.024369333870708942,
      "grad_norm": 0.10560034960508347,
      "learning_rate": 3.170143660009228e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004249733843607828,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.801479721069335,
      "state_query_norm": 1431.765576171875,
      "state_relation_loss": 0.0004249733843607828,
      "step": 19150,
      "student_distance_mean": 1966.0018676757813,
      "teacher_distance_mean": 1.159640109539032,
      "weighted_auxiliary_loss": 0.0004249733843607828,
      "weighted_relation_loss": 0.0004249733843607828
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.92329279700655,
      "flow_matching_loss": 0.02239114884287119,
      "grad_norm": 0.10536859184503555,
      "learning_rate": 3.16501556527027e-05,
      "loss": 0.0228,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044278379064053295,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52894325256348,
      "state_query_norm": 1424.9595458984375,
      "state_relation_loss": 0.00044278379064053295,
      "step": 19160,
      "student_distance_mean": 1952.811474609375,
      "teacher_distance_mean": 1.149669361114502,
      "weighted_auxiliary_loss": 0.00044278379064053295,
      "weighted_relation_loss": 0.00044278379064053295
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.932647333956968,
      "flow_matching_loss": 0.02318353820592165,
      "grad_norm": 0.10975121706724167,
      "learning_rate": 3.1598897002104265e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004382750892546028,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.738034057617185,
      "state_query_norm": 1438.3583618164062,
      "state_relation_loss": 0.0004382750892546028,
      "step": 19170,
      "student_distance_mean": 1964.9800170898438,
      "teacher_distance_mean": 1.156813144683838,
      "weighted_auxiliary_loss": 0.0004382750892546028,
      "weighted_relation_loss": 0.0004382750892546028
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.94200187090739,
      "flow_matching_loss": 0.022931687347590924,
      "grad_norm": 0.0915084257721901,
      "learning_rate": 3.154766071058108e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004428898449987173,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.630639266967776,
      "state_query_norm": 1427.522314453125,
      "state_relation_loss": 0.0004428898449987173,
      "step": 19180,
      "student_distance_mean": 1956.9303100585937,
      "teacher_distance_mean": 1.1525111436843871,
      "weighted_auxiliary_loss": 0.0004428898449987173,
      "weighted_relation_loss": 0.0004428898449987173
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.951356407857812,
      "flow_matching_loss": 0.023446985334157944,
      "grad_norm": 0.12926359474658966,
      "learning_rate": 3.149644684039008e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004416134412167594,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.46626853942871,
      "state_query_norm": 1429.5560668945313,
      "state_relation_loss": 0.0004416134412167594,
      "step": 19190,
      "student_distance_mean": 1953.9515502929687,
      "teacher_distance_mean": 1.1488316178321838,
      "weighted_auxiliary_loss": 0.0004416134412167594,
      "weighted_relation_loss": 0.0004416134412167594
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.96071094480823,
      "flow_matching_loss": 0.0244621517136693,
      "grad_norm": 0.09217926114797592,
      "learning_rate": 3.144525545376095e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004239240428432822,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63132553100586,
      "state_query_norm": 1424.1295776367188,
      "state_relation_loss": 0.0004239240428432822,
      "step": 19200,
      "student_distance_mean": 1953.9993530273437,
      "teacher_distance_mean": 1.1495337724685668,
      "weighted_auxiliary_loss": 0.0004239240428432822,
      "weighted_relation_loss": 0.0004239240428432822
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.970065481758652,
      "flow_matching_loss": 0.022725704312324523,
      "grad_norm": 0.08901015669107437,
      "learning_rate": 3.139408661289603e-05,
      "loss": 0.0232,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045406763383653014,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70603523254395,
      "state_query_norm": 1441.1211547851562,
      "state_relation_loss": 0.00045406763383653014,
      "step": 19210,
      "student_distance_mean": 1964.7003540039063,
      "teacher_distance_mean": 1.1548479795455933,
      "weighted_auxiliary_loss": 0.00045406763383653014,
      "weighted_relation_loss": 0.00045406763383653014
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.979420018709074,
      "flow_matching_loss": 0.02272942364215851,
      "grad_norm": 0.11754494905471802,
      "learning_rate": 3.134294037997032e-05,
      "loss": 0.0232,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000459216398303397,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.207851791381835,
      "state_query_norm": 1454.1455322265624,
      "state_relation_loss": 0.000459216398303397,
      "step": 19220,
      "student_distance_mean": 1991.8541259765625,
      "teacher_distance_mean": 1.167848241329193,
      "weighted_auxiliary_loss": 0.000459216398303397,
      "weighted_relation_loss": 0.000459216398303397
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.988774555659496,
      "flow_matching_loss": 0.022712318785488607,
      "grad_norm": 0.09566942602396011,
      "learning_rate": 3.129181681713127e-05,
      "loss": 0.0232,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000449765645316802,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.877363967895505,
      "state_query_norm": 1441.458056640625,
      "state_relation_loss": 0.000449765645316802,
      "step": 19230,
      "student_distance_mean": 1967.3297241210937,
      "teacher_distance_mean": 1.1523297071456908,
      "weighted_auxiliary_loss": 0.000449765645316802,
      "weighted_relation_loss": 0.000449765645316802
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 17.998129092609915,
      "flow_matching_loss": 0.02399444915354252,
      "grad_norm": 0.113755963742733,
      "learning_rate": 3.1240715986498855e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046909902011975645,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.098435974121095,
      "state_query_norm": 1447.28408203125,
      "state_relation_loss": 0.00046909902011975645,
      "step": 19240,
      "student_distance_mean": 1984.8538452148437,
      "teacher_distance_mean": 1.1640180349349976,
      "weighted_auxiliary_loss": 0.00046909902011975645,
      "weighted_relation_loss": 0.00046909902011975645
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.007483629560337,
      "flow_matching_loss": 0.023700945451855658,
      "grad_norm": 0.11455757915973663,
      "learning_rate": 3.1189637950165394e-05,
      "loss": 0.0241,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00044081425585318357,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.20696201324463,
      "state_query_norm": 1412.7440307617187,
      "state_relation_loss": 0.00044081425585318357,
      "step": 19250,
      "student_distance_mean": 1936.743896484375,
      "teacher_distance_mean": 1.1388206899166107,
      "weighted_auxiliary_loss": 0.00044081425585318357,
      "weighted_relation_loss": 0.00044081425585318357
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.01683816651076,
      "flow_matching_loss": 0.023446000926196576,
      "grad_norm": 0.10081830620765686,
      "learning_rate": 3.1138582770195544e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004291532590286806,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.820469284057616,
      "state_query_norm": 1436.3566284179688,
      "state_relation_loss": 0.0004291532590286806,
      "step": 19260,
      "student_distance_mean": 1969.6755737304688,
      "teacher_distance_mean": 1.155126941204071,
      "weighted_auxiliary_loss": 0.0004291532590286806,
      "weighted_relation_loss": 0.0004291532590286806
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.026192703461177,
      "flow_matching_loss": 0.024045209027826787,
      "grad_norm": 0.08716703951358795,
      "learning_rate": 3.108755050862615e-05,
      "loss": 0.0245,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004200065101031214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36808319091797,
      "state_query_norm": 1427.7682373046875,
      "state_relation_loss": 0.0004200065101031214,
      "step": 19270,
      "student_distance_mean": 1945.8622436523438,
      "teacher_distance_mean": 1.1407508492469787,
      "weighted_auxiliary_loss": 0.0004200065101031214,
      "weighted_relation_loss": 0.0004200065101031214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.0355472404116,
      "flow_matching_loss": 0.024660172313451766,
      "grad_norm": 0.1360342800617218,
      "learning_rate": 3.1036541227466204e-05,
      "loss": 0.0251,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044954423501621934,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.634250259399415,
      "state_query_norm": 1430.3946655273437,
      "state_relation_loss": 0.00044954423501621934,
      "step": 19280,
      "student_distance_mean": 1956.8198486328124,
      "teacher_distance_mean": 1.1491422772407531,
      "weighted_auxiliary_loss": 0.00044954423501621934,
      "weighted_relation_loss": 0.00044954423501621934
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.04490177736202,
      "flow_matching_loss": 0.02340396959334612,
      "grad_norm": 0.09025245904922485,
      "learning_rate": 3.0985554988696786e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004566450079437345,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.34840621948242,
      "state_query_norm": 1456.7023559570312,
      "state_relation_loss": 0.0004566450079437345,
      "step": 19290,
      "student_distance_mean": 1996.9682250976562,
      "teacher_distance_mean": 1.1731620907783509,
      "weighted_auxiliary_loss": 0.0004566450079437345,
      "weighted_relation_loss": 0.0004566450079437345
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.054256314312443,
      "flow_matching_loss": 0.02386721521615982,
      "grad_norm": 0.08754987269639969,
      "learning_rate": 3.093459185427102e-05,
      "loss": 0.0243,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041822110360953955,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50487365722656,
      "state_query_norm": 1428.665087890625,
      "state_relation_loss": 0.00041822110360953955,
      "step": 19300,
      "student_distance_mean": 1953.161962890625,
      "teacher_distance_mean": 1.1469062924385072,
      "weighted_auxiliary_loss": 0.00041822110360953955,
      "weighted_relation_loss": 0.00041822110360953955
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.06361085126286,
      "flow_matching_loss": 0.02154442947357893,
      "grad_norm": 0.09668449312448502,
      "learning_rate": 3.0883651886113905e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000445561675587669,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5456958770752,
      "state_query_norm": 1422.896826171875,
      "state_relation_loss": 0.000445561675587669,
      "step": 19310,
      "student_distance_mean": 1953.82607421875,
      "teacher_distance_mean": 1.1486690163612365,
      "weighted_auxiliary_loss": 0.000445561675587669,
      "weighted_relation_loss": 0.000445561675587669
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.072965388213284,
      "flow_matching_loss": 0.0235036825761199,
      "grad_norm": 0.09522147476673126,
      "learning_rate": 3.0832735146122294e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004422549187438563,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48119125366211,
      "state_query_norm": 1428.2600219726562,
      "state_relation_loss": 0.0004422549187438563,
      "step": 19320,
      "student_distance_mean": 1949.3217895507812,
      "teacher_distance_mean": 1.144619059562683,
      "weighted_auxiliary_loss": 0.0004422549187438563,
      "weighted_relation_loss": 0.0004422549187438563
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.082319925163706,
      "flow_matching_loss": 0.023378534615039824,
      "grad_norm": 0.09377121925354004,
      "learning_rate": 3.078184169616485e-05,
      "loss": 0.0238,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043812985240947454,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.668648910522464,
      "state_query_norm": 1439.3854370117188,
      "state_relation_loss": 0.00043812985240947454,
      "step": 19330,
      "student_distance_mean": 1966.0394653320313,
      "teacher_distance_mean": 1.155152678489685,
      "weighted_auxiliary_loss": 0.00043812985240947454,
      "weighted_relation_loss": 0.00043812985240947454
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.091674462114124,
      "flow_matching_loss": 0.0223197091370821,
      "grad_norm": 0.08762556314468384,
      "learning_rate": 3.073097159808187e-05,
      "loss": 0.0228,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004438200616277754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.273695373535155,
      "state_query_norm": 1424.4321655273438,
      "state_relation_loss": 0.0004438200616277754,
      "step": 19340,
      "student_distance_mean": 1941.7835815429687,
      "teacher_distance_mean": 1.1413243889808655,
      "weighted_auxiliary_loss": 0.0004438200616277754,
      "weighted_relation_loss": 0.0004438200616277754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.101028999064546,
      "flow_matching_loss": 0.021523830108344556,
      "grad_norm": 0.09741681814193726,
      "learning_rate": 3.068012491368537e-05,
      "loss": 0.0219,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004044102446641773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27342529296875,
      "state_query_norm": 1425.4918090820313,
      "state_relation_loss": 0.0004044102446641773,
      "step": 19350,
      "student_distance_mean": 1943.7343505859376,
      "teacher_distance_mean": 1.1435635685920715,
      "weighted_auxiliary_loss": 0.0004044102446641773,
      "weighted_relation_loss": 0.0004044102446641773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.110383536014968,
      "flow_matching_loss": 0.023168745636940002,
      "grad_norm": 0.0737762376666069,
      "learning_rate": 3.062930170475885e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042102925945073364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2141056060791,
      "state_query_norm": 1409.8736206054687,
      "state_relation_loss": 0.00042102925945073364,
      "step": 19360,
      "student_distance_mean": 1932.3625122070312,
      "teacher_distance_mean": 1.1374761939048768,
      "weighted_auxiliary_loss": 0.00042102925945073364,
      "weighted_relation_loss": 0.00042102925945073364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.119738072965387,
      "flow_matching_loss": 0.022931081429123878,
      "grad_norm": 0.09330438077449799,
      "learning_rate": 3.057850203305729e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044483655074145645,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.05999755859375,
      "state_query_norm": 1454.977880859375,
      "state_relation_loss": 0.00044483655074145645,
      "step": 19370,
      "student_distance_mean": 1987.8482299804687,
      "teacher_distance_mean": 1.1690119981765748,
      "weighted_auxiliary_loss": 0.00044483655074145645,
      "weighted_relation_loss": 0.00044483655074145645
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.12909260991581,
      "flow_matching_loss": 0.02313823066651821,
      "grad_norm": 0.08391957730054855,
      "learning_rate": 3.052772596030708e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045454660139512273,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75265693664551,
      "state_query_norm": 1436.4197143554688,
      "state_relation_loss": 0.00045454660139512273,
      "step": 19380,
      "student_distance_mean": 1967.4961181640624,
      "teacher_distance_mean": 1.1563062906265258,
      "weighted_auxiliary_loss": 0.00045454660139512273,
      "weighted_relation_loss": 0.00045454660139512273
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.13844714686623,
      "flow_matching_loss": 0.021855852752923965,
      "grad_norm": 0.10712108761072159,
      "learning_rate": 3.0476973548205943e-05,
      "loss": 0.0223,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042441900295671077,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.09097900390625,
      "state_query_norm": 1451.0133178710937,
      "state_relation_loss": 0.00042441900295671077,
      "step": 19390,
      "student_distance_mean": 1986.782666015625,
      "teacher_distance_mean": 1.1648466467857361,
      "weighted_auxiliary_loss": 0.00042441900295671077,
      "weighted_relation_loss": 0.00042441900295671077
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.147801683816652,
      "flow_matching_loss": 0.023664357885718346,
      "grad_norm": 0.10133563727140427,
      "learning_rate": 3.0426244858422847e-05,
      "loss": 0.0241,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004650620569009334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3983642578125,
      "state_query_norm": 1427.54697265625,
      "state_relation_loss": 0.0004650620569009334,
      "step": 19400,
      "student_distance_mean": 1948.17119140625,
      "teacher_distance_mean": 1.1398693442344665,
      "weighted_auxiliary_loss": 0.0004650620569009334,
      "weighted_relation_loss": 0.0004650620569009334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.15715622076707,
      "flow_matching_loss": 0.022432894818484782,
      "grad_norm": 0.09918950498104095,
      "learning_rate": 3.0375539952597943e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004261747468262911,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30848350524902,
      "state_query_norm": 1412.867626953125,
      "state_relation_loss": 0.0004261747468262911,
      "step": 19410,
      "student_distance_mean": 1941.8145751953125,
      "teacher_distance_mean": 1.1367022514343261,
      "weighted_auxiliary_loss": 0.0004261747468262911,
      "weighted_relation_loss": 0.0004261747468262911
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.166510757717493,
      "flow_matching_loss": 0.020391126442700625,
      "grad_norm": 0.08068405836820602,
      "learning_rate": 3.0324858892342468e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004276283027138561,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.257910537719724,
      "state_query_norm": 1451.7086547851563,
      "state_relation_loss": 0.0004276283027138561,
      "step": 19420,
      "student_distance_mean": 1991.6162719726562,
      "teacher_distance_mean": 1.169867706298828,
      "weighted_auxiliary_loss": 0.0004276283027138561,
      "weighted_relation_loss": 0.0004276283027138561
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.175865294667915,
      "flow_matching_loss": 0.02354109864681959,
      "grad_norm": 0.08479461818933487,
      "learning_rate": 3.0274201739238672e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044178064272273334,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14425621032715,
      "state_query_norm": 1410.8207641601562,
      "state_relation_loss": 0.00044178064272273334,
      "step": 19430,
      "student_distance_mean": 1932.034326171875,
      "teacher_distance_mean": 1.135272753238678,
      "weighted_auxiliary_loss": 0.00044178064272273334,
      "weighted_relation_loss": 0.00044178064272273334
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.185219831618333,
      "flow_matching_loss": 0.023167849332094193,
      "grad_norm": 0.1036895364522934,
      "learning_rate": 3.0223568554839786e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043200590589549394,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.12509002685547,
      "state_query_norm": 1412.8073974609374,
      "state_relation_loss": 0.00043200590589549394,
      "step": 19440,
      "student_distance_mean": 1933.3016723632813,
      "teacher_distance_mean": 1.1350882530212403,
      "weighted_auxiliary_loss": 0.00043200590589549394,
      "weighted_relation_loss": 0.00043200590589549394
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.194574368568755,
      "flow_matching_loss": 0.02169874180108309,
      "grad_norm": 0.07751186937093735,
      "learning_rate": 3.0172959400669886e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004382745333714411,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.61168746948242,
      "state_query_norm": 1434.8772827148437,
      "state_relation_loss": 0.0004382745333714411,
      "step": 19450,
      "student_distance_mean": 1960.8481689453124,
      "teacher_distance_mean": 1.153514802455902,
      "weighted_auxiliary_loss": 0.0004382745333714411,
      "weighted_relation_loss": 0.0004382745333714411
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.203928905519177,
      "flow_matching_loss": 0.022016852907836438,
      "grad_norm": 0.07567251473665237,
      "learning_rate": 3.0122374338223902e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004568925651255995,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6063793182373,
      "state_query_norm": 1426.42451171875,
      "state_relation_loss": 0.0004568925651255995,
      "step": 19460,
      "student_distance_mean": 1954.01123046875,
      "teacher_distance_mean": 1.147110390663147,
      "weighted_auxiliary_loss": 0.0004568925651255995,
      "weighted_relation_loss": 0.0004568925651255995
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.2132834424696,
      "flow_matching_loss": 0.023037800192832948,
      "grad_norm": 0.07387875765562057,
      "learning_rate": 3.0071813428967427e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041027681436389685,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36318130493164,
      "state_query_norm": 1422.8333862304687,
      "state_relation_loss": 0.00041027681436389685,
      "step": 19470,
      "student_distance_mean": 1945.32822265625,
      "teacher_distance_mean": 1.1403997659683227,
      "weighted_auxiliary_loss": 0.00041027681436389685,
      "weighted_relation_loss": 0.00041027681436389685
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.222637979420018,
      "flow_matching_loss": 0.022401431016623972,
      "grad_norm": 0.10059657692909241,
      "learning_rate": 3.0021276734336746e-05,
      "loss": 0.0228,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004442043078597635,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.982191467285155,
      "state_query_norm": 1431.1924926757813,
      "state_relation_loss": 0.0004442043078597635,
      "step": 19480,
      "student_distance_mean": 1971.8927856445312,
      "teacher_distance_mean": 1.1588834881782533,
      "weighted_auxiliary_loss": 0.0004442043078597635,
      "weighted_relation_loss": 0.0004442043078597635
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.23199251637044,
      "flow_matching_loss": 0.021789156924933197,
      "grad_norm": 0.09197717905044556,
      "learning_rate": 2.997076431573871e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000433342281030491,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40934524536133,
      "state_query_norm": 1417.6021484375,
      "state_relation_loss": 0.000433342281030491,
      "step": 19490,
      "student_distance_mean": 1946.3875610351563,
      "teacher_distance_mean": 1.1469476580619813,
      "weighted_auxiliary_loss": 0.000433342281030491,
      "weighted_relation_loss": 0.000433342281030491
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.24134705332086,
      "flow_matching_loss": 0.022933086566627025,
      "grad_norm": 0.12251059710979462,
      "learning_rate": 2.9920276234550636e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004392462782561779,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56985511779785,
      "state_query_norm": 1420.4631469726562,
      "state_relation_loss": 0.0004392462782561779,
      "step": 19500,
      "student_distance_mean": 1951.3506591796875,
      "teacher_distance_mean": 1.1497413635253906,
      "weighted_auxiliary_loss": 0.0004392462782561779,
      "weighted_relation_loss": 0.0004392462782561779
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.25070159027128,
      "flow_matching_loss": 0.023804819397628307,
      "grad_norm": 0.10805927217006683,
      "learning_rate": 2.986981255212035e-05,
      "loss": 0.0242,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041608543251641094,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.68348350524902,
      "state_query_norm": 1438.796484375,
      "state_relation_loss": 0.00041608543251641094,
      "step": 19510,
      "student_distance_mean": 1961.7455688476562,
      "teacher_distance_mean": 1.153423798084259,
      "weighted_auxiliary_loss": 0.00041608543251641094,
      "weighted_relation_loss": 0.00041608543251641094
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.260056127221702,
      "flow_matching_loss": 0.02419789545238018,
      "grad_norm": 0.09950251877307892,
      "learning_rate": 2.9819373329765977e-05,
      "loss": 0.0246,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004343213833635673,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37412528991699,
      "state_query_norm": 1420.935498046875,
      "state_relation_loss": 0.0004343213833635673,
      "step": 19520,
      "student_distance_mean": 1942.5418701171875,
      "teacher_distance_mean": 1.1412137746810913,
      "weighted_auxiliary_loss": 0.0004343213833635673,
      "weighted_relation_loss": 0.0004343213833635673
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.269410664172124,
      "flow_matching_loss": 0.021114393323659896,
      "grad_norm": 0.09431704133749008,
      "learning_rate": 2.9768958628775902e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042674565047491343,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.67012977600098,
      "state_query_norm": 1435.572314453125,
      "state_relation_loss": 0.00042674565047491343,
      "step": 19530,
      "student_distance_mean": 1962.91240234375,
      "teacher_distance_mean": 1.1508735418319702,
      "weighted_auxiliary_loss": 0.00042674565047491343,
      "weighted_relation_loss": 0.00042674565047491343
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.278765201122546,
      "flow_matching_loss": 0.021267240680754185,
      "grad_norm": 0.08417901396751404,
      "learning_rate": 2.9718568510408762e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004365530563518405,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48724632263183,
      "state_query_norm": 1420.3006591796875,
      "state_relation_loss": 0.0004365530563518405,
      "step": 19540,
      "student_distance_mean": 1951.9695922851563,
      "teacher_distance_mean": 1.1452989816665649,
      "weighted_auxiliary_loss": 0.0004365530563518405,
      "weighted_relation_loss": 0.0004365530563518405
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.288119738072965,
      "flow_matching_loss": 0.023981555178761483,
      "grad_norm": 0.09329771995544434,
      "learning_rate": 2.9668203035893272e-05,
      "loss": 0.0244,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044727190397679805,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.98392028808594,
      "state_query_norm": 1442.0637939453125,
      "state_relation_loss": 0.00044727190397679805,
      "step": 19550,
      "student_distance_mean": 1977.43359375,
      "teacher_distance_mean": 1.1606116890907288,
      "weighted_auxiliary_loss": 0.00044727190397679805,
      "weighted_relation_loss": 0.00044727190397679805
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.297474275023387,
      "flow_matching_loss": 0.023057936690747738,
      "grad_norm": 0.09711845219135284,
      "learning_rate": 2.9617862266428287e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045129346544854344,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.568169021606444,
      "state_query_norm": 1431.4047973632812,
      "state_relation_loss": 0.00045129346544854344,
      "step": 19560,
      "student_distance_mean": 1955.1051879882812,
      "teacher_distance_mean": 1.1481936573982239,
      "weighted_auxiliary_loss": 0.00045129346544854344,
      "weighted_relation_loss": 0.00045129346544854344
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.30682881197381,
      "flow_matching_loss": 0.024593112245202064,
      "grad_norm": 0.1050640121102333,
      "learning_rate": 2.9567546263182556e-05,
      "loss": 0.025,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041722812165971844,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.34177169799805,
      "state_query_norm": 1428.2595581054688,
      "state_relation_loss": 0.00041722812165971844,
      "step": 19570,
      "student_distance_mean": 1946.9332275390625,
      "teacher_distance_mean": 1.1438597798347474,
      "weighted_auxiliary_loss": 0.00041722812165971844,
      "weighted_relation_loss": 0.00041722812165971844
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.316183348924227,
      "flow_matching_loss": 0.024388913065195084,
      "grad_norm": 0.0923658162355423,
      "learning_rate": 2.951725508729476e-05,
      "loss": 0.0248,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041257409611716866,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10595550537109,
      "state_query_norm": 1403.3796264648438,
      "state_relation_loss": 0.00041257409611716866,
      "step": 19580,
      "student_distance_mean": 1925.2079956054688,
      "teacher_distance_mean": 1.1304075002670289,
      "weighted_auxiliary_loss": 0.00041257409611716866,
      "weighted_relation_loss": 0.00041257409611716866
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.32553788587465,
      "flow_matching_loss": 0.022997333854436874,
      "grad_norm": 0.11450913548469543,
      "learning_rate": 2.946698879987344e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043098409369122235,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65659446716309,
      "state_query_norm": 1438.9171020507813,
      "state_relation_loss": 0.00043098409369122235,
      "step": 19590,
      "student_distance_mean": 1961.5380249023438,
      "teacher_distance_mean": 1.1507329940795898,
      "weighted_auxiliary_loss": 0.00043098409369122235,
      "weighted_relation_loss": 0.00043098409369122235
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.33489242282507,
      "flow_matching_loss": 0.02347810473293066,
      "grad_norm": 0.10347412526607513,
      "learning_rate": 2.9416747461996853e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004387174732983112,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.899148559570314,
      "state_query_norm": 1439.9143676757812,
      "state_relation_loss": 0.0004387174732983112,
      "step": 19600,
      "student_distance_mean": 1972.3553955078125,
      "teacher_distance_mean": 1.1597410082817077,
      "weighted_auxiliary_loss": 0.0004387174732983112,
      "weighted_relation_loss": 0.0004387174732983112
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.34424695977549,
      "flow_matching_loss": 0.02246218379586935,
      "grad_norm": 0.10560975223779678,
      "learning_rate": 2.9366531134712973e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045563491003122183,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73838920593262,
      "state_query_norm": 1439.646240234375,
      "state_relation_loss": 0.00045563491003122183,
      "step": 19610,
      "student_distance_mean": 1963.9595092773438,
      "teacher_distance_mean": 1.157033348083496,
      "weighted_auxiliary_loss": 0.00045563491003122183,
      "weighted_relation_loss": 0.00045563491003122183
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.35360149672591,
      "flow_matching_loss": 0.02201376836746931,
      "grad_norm": 0.09233306348323822,
      "learning_rate": 2.9316339879039367e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004497943038586527,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83689956665039,
      "state_query_norm": 1429.3379272460938,
      "state_relation_loss": 0.0004497943038586527,
      "step": 19620,
      "student_distance_mean": 1963.9442260742187,
      "teacher_distance_mean": 1.1573862910270691,
      "weighted_auxiliary_loss": 0.0004497943038586527,
      "weighted_relation_loss": 0.0004497943038586527
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.362956033676333,
      "flow_matching_loss": 0.021807936765253545,
      "grad_norm": 0.1041455790400505,
      "learning_rate": 2.9266173755963167e-05,
      "loss": 0.0223,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004620144987711683,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.563287353515626,
      "state_query_norm": 1422.4547973632812,
      "state_relation_loss": 0.0004620144987711683,
      "step": 19630,
      "student_distance_mean": 1950.57880859375,
      "teacher_distance_mean": 1.1489898324012757,
      "weighted_auxiliary_loss": 0.0004620144987711683,
      "weighted_relation_loss": 0.0004620144987711683
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.372310570626755,
      "flow_matching_loss": 0.02089825514703989,
      "grad_norm": 0.08984200656414032,
      "learning_rate": 2.9216032826440926e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040923260676208886,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4060245513916,
      "state_query_norm": 1431.10810546875,
      "state_relation_loss": 0.00040923260676208886,
      "step": 19640,
      "student_distance_mean": 1950.5205810546875,
      "teacher_distance_mean": 1.1457509398460388,
      "weighted_auxiliary_loss": 0.00040923260676208886,
      "weighted_relation_loss": 0.00040923260676208886
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.381665107577174,
      "flow_matching_loss": 0.02228556089103222,
      "grad_norm": 0.10502047836780548,
      "learning_rate": 2.9165917151398592e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004409635585034266,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52392425537109,
      "state_query_norm": 1427.3090087890625,
      "state_relation_loss": 0.0004409635585034266,
      "step": 19650,
      "student_distance_mean": 1956.5154907226563,
      "teacher_distance_mean": 1.1490105032920837,
      "weighted_auxiliary_loss": 0.0004409635585034266,
      "weighted_relation_loss": 0.0004409635585034266
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.391019644527596,
      "flow_matching_loss": 0.02323397360742092,
      "grad_norm": 0.08419494330883026,
      "learning_rate": 2.9115826791731428e-05,
      "loss": 0.0236,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004133320937398821,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.51340789794922,
      "state_query_norm": 1433.0724243164063,
      "state_relation_loss": 0.0004133320937398821,
      "step": 19660,
      "student_distance_mean": 1956.6207641601563,
      "teacher_distance_mean": 1.149193787574768,
      "weighted_auxiliary_loss": 0.0004133320937398821,
      "weighted_relation_loss": 0.0004133320937398821
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.400374181478018,
      "flow_matching_loss": 0.02297088038176298,
      "grad_norm": 0.08143506944179535,
      "learning_rate": 2.906576180830398e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047708862693980337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.21165542602539,
      "state_query_norm": 1451.1313354492188,
      "state_relation_loss": 0.00047708862693980337,
      "step": 19670,
      "student_distance_mean": 1986.913134765625,
      "teacher_distance_mean": 1.16826993227005,
      "weighted_auxiliary_loss": 0.00047708862693980337,
      "weighted_relation_loss": 0.00047708862693980337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.409728718428436,
      "flow_matching_loss": 0.021770037524402142,
      "grad_norm": 0.09051306545734406,
      "learning_rate": 2.9015722261949917e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044164345890749244,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.909641647338866,
      "state_query_norm": 1404.990625,
      "state_relation_loss": 0.00044164345890749244,
      "step": 19680,
      "student_distance_mean": 1920.6935424804688,
      "teacher_distance_mean": 1.1293594241142273,
      "weighted_auxiliary_loss": 0.00044164345890749244,
      "weighted_relation_loss": 0.00044164345890749244
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.419083255378858,
      "flow_matching_loss": 0.02301437109708786,
      "grad_norm": 0.090920589864254,
      "learning_rate": 2.8965708213471986e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043444863113109025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.614275741577146,
      "state_query_norm": 1427.4453491210938,
      "state_relation_loss": 0.00043444863113109025,
      "step": 19690,
      "student_distance_mean": 1956.9560180664062,
      "teacher_distance_mean": 1.1480433344841003,
      "weighted_auxiliary_loss": 0.00043444863113109025,
      "weighted_relation_loss": 0.00043444863113109025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.42843779232928,
      "flow_matching_loss": 0.02282893843948841,
      "grad_norm": 0.12074663490056992,
      "learning_rate": 2.8915719723641975e-05,
      "loss": 0.0233,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044596544466912746,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37588272094727,
      "state_query_norm": 1420.9412231445312,
      "state_relation_loss": 0.00044596544466912746,
      "step": 19700,
      "student_distance_mean": 1942.7202392578124,
      "teacher_distance_mean": 1.1427301168441772,
      "weighted_auxiliary_loss": 0.00044596544466912746,
      "weighted_relation_loss": 0.00044596544466912746
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.437792329279702,
      "flow_matching_loss": 0.022720020450651645,
      "grad_norm": 0.09843320399522781,
      "learning_rate": 2.8865756853200604e-05,
      "loss": 0.0231,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004190952255157754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.06239128112793,
      "state_query_norm": 1447.9623657226562,
      "state_relation_loss": 0.0004190952255157754,
      "step": 19710,
      "student_distance_mean": 1982.8808837890624,
      "teacher_distance_mean": 1.1672532558441162,
      "weighted_auxiliary_loss": 0.0004190952255157754,
      "weighted_relation_loss": 0.0004190952255157754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.44714686623012,
      "flow_matching_loss": 0.020962894521653654,
      "grad_norm": 0.11317937821149826,
      "learning_rate": 2.8815819662857503e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045368408318609,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.8629955291748,
      "state_query_norm": 1449.135107421875,
      "state_relation_loss": 0.00045368408318609,
      "step": 19720,
      "student_distance_mean": 1974.3534790039062,
      "teacher_distance_mean": 1.1602143406867982,
      "weighted_auxiliary_loss": 0.00045368408318609,
      "weighted_relation_loss": 0.00045368408318609
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.456501403180543,
      "flow_matching_loss": 0.023476056382060052,
      "grad_norm": 0.09381115436553955,
      "learning_rate": 2.876590821329105e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004299707623431459,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.200277709960936,
      "state_query_norm": 1452.4807861328125,
      "state_relation_loss": 0.0004299707623431459,
      "step": 19730,
      "student_distance_mean": 1990.4473266601562,
      "teacher_distance_mean": 1.170286202430725,
      "weighted_auxiliary_loss": 0.0004299707623431459,
      "weighted_relation_loss": 0.0004299707623431459
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.465855940130965,
      "flow_matching_loss": 0.023934518545866014,
      "grad_norm": 0.09094925969839096,
      "learning_rate": 2.871602256514836e-05,
      "loss": 0.0244,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000459467907785438,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.86282081604004,
      "state_query_norm": 1432.8873291015625,
      "state_relation_loss": 0.000459467907785438,
      "step": 19740,
      "student_distance_mean": 1968.92705078125,
      "teacher_distance_mean": 1.159776759147644,
      "weighted_auxiliary_loss": 0.000459467907785438,
      "weighted_relation_loss": 0.000459467907785438
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.475210477081383,
      "flow_matching_loss": 0.020868977159261705,
      "grad_norm": 0.0909474715590477,
      "learning_rate": 2.8666162779045203e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004451543762115762,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83991394042969,
      "state_query_norm": 1439.835546875,
      "state_relation_loss": 0.0004451543762115762,
      "step": 19750,
      "student_distance_mean": 1972.3186157226562,
      "teacher_distance_mean": 1.1608579516410829,
      "weighted_auxiliary_loss": 0.0004451543762115762,
      "weighted_relation_loss": 0.0004451543762115762
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.484565014031805,
      "flow_matching_loss": 0.022999608889222145,
      "grad_norm": 0.09462639689445496,
      "learning_rate": 2.8616328915565904e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042498805560171605,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.51874046325683,
      "state_query_norm": 1424.4799072265625,
      "state_relation_loss": 0.00042498805560171605,
      "step": 19760,
      "student_distance_mean": 1952.6970458984374,
      "teacher_distance_mean": 1.1477242350578307,
      "weighted_auxiliary_loss": 0.00042498805560171605,
      "weighted_relation_loss": 0.00042498805560171605
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.493919550982227,
      "flow_matching_loss": 0.02444870937615633,
      "grad_norm": 0.09403280168771744,
      "learning_rate": 2.856652103526334e-05,
      "loss": 0.0249,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004426078841788694,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.736994934082034,
      "state_query_norm": 1433.5208984375,
      "state_relation_loss": 0.0004426078841788694,
      "step": 19770,
      "student_distance_mean": 1966.8568115234375,
      "teacher_distance_mean": 1.1531392455101013,
      "weighted_auxiliary_loss": 0.0004426078841788694,
      "weighted_relation_loss": 0.0004426078841788694
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.50327408793265,
      "flow_matching_loss": 0.021676833368837835,
      "grad_norm": 0.10825800895690918,
      "learning_rate": 2.8516739198658755e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004720359167549759,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02941398620605,
      "state_query_norm": 1439.1311401367188,
      "state_relation_loss": 0.0004720359167549759,
      "step": 19780,
      "student_distance_mean": 1976.1091796875,
      "teacher_distance_mean": 1.1561468601226808,
      "weighted_auxiliary_loss": 0.0004720359167549759,
      "weighted_relation_loss": 0.0004720359167549759
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.512628624883067,
      "flow_matching_loss": 0.022045471519231797,
      "grad_norm": 0.09245184808969498,
      "learning_rate": 2.846698346624177e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045384149125311525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.7106201171875,
      "state_query_norm": 1428.0088745117187,
      "state_relation_loss": 0.00045384149125311525,
      "step": 19790,
      "student_distance_mean": 1963.7706420898437,
      "teacher_distance_mean": 1.1503678798675536,
      "weighted_auxiliary_loss": 0.00045384149125311525,
      "weighted_relation_loss": 0.00045384149125311525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.52198316183349,
      "flow_matching_loss": 0.020619645435363054,
      "grad_norm": 0.11384798586368561,
      "learning_rate": 2.841725389847032e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043788448092527685,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.458391189575195,
      "state_query_norm": 1417.9205200195313,
      "state_relation_loss": 0.00043788448092527685,
      "step": 19800,
      "student_distance_mean": 1947.6927001953125,
      "teacher_distance_mean": 1.143550741672516,
      "weighted_auxiliary_loss": 0.00043788448092527685,
      "weighted_relation_loss": 0.00043788448092527685
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.53133769878391,
      "flow_matching_loss": 0.02204678803682327,
      "grad_norm": 0.09210432320833206,
      "learning_rate": 2.8367550555770506e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043051929387729617,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89368019104004,
      "state_query_norm": 1449.9532592773437,
      "state_relation_loss": 0.00043051929387729617,
      "step": 19810,
      "student_distance_mean": 1975.4224365234375,
      "teacher_distance_mean": 1.158829355239868,
      "weighted_auxiliary_loss": 0.00043051929387729617,
      "weighted_relation_loss": 0.00043051929387729617
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.54069223573433,
      "flow_matching_loss": 0.021819988545030356,
      "grad_norm": 0.09454932808876038,
      "learning_rate": 2.831787349853655e-05,
      "loss": 0.0223,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044068997376598416,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.574911499023436,
      "state_query_norm": 1429.1994384765626,
      "state_relation_loss": 0.00044068997376598416,
      "step": 19820,
      "student_distance_mean": 1959.2037719726563,
      "teacher_distance_mean": 1.1507365107536316,
      "weighted_auxiliary_loss": 0.00044068997376598416,
      "weighted_relation_loss": 0.00044068997376598416
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.55004677268475,
      "flow_matching_loss": 0.021591518074274063,
      "grad_norm": 0.08377154916524887,
      "learning_rate": 2.8268222787130806e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004363334010122344,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43776245117188,
      "state_query_norm": 1419.1810668945313,
      "state_relation_loss": 0.0004363334010122344,
      "step": 19830,
      "student_distance_mean": 1948.415380859375,
      "teacher_distance_mean": 1.144080603122711,
      "weighted_auxiliary_loss": 0.0004363334010122344,
      "weighted_relation_loss": 0.0004363334010122344
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.559401309635174,
      "flow_matching_loss": 0.022992237098515035,
      "grad_norm": 0.10550450533628464,
      "learning_rate": 2.821859848188355e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004656575241824612,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.24167594909668,
      "state_query_norm": 1455.3766723632812,
      "state_relation_loss": 0.0004656575241824612,
      "step": 19840,
      "student_distance_mean": 1992.143017578125,
      "teacher_distance_mean": 1.1710538744926453,
      "weighted_auxiliary_loss": 0.0004656575241824612,
      "weighted_relation_loss": 0.0004656575241824612
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.568755846585596,
      "flow_matching_loss": 0.023060440830886365,
      "grad_norm": 0.1038372591137886,
      "learning_rate": 2.816900064309299e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044418959296308456,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.590087509155275,
      "state_query_norm": 1424.4721557617188,
      "state_relation_loss": 0.00044418959296308456,
      "step": 19850,
      "student_distance_mean": 1953.7103759765625,
      "teacher_distance_mean": 1.152282428741455,
      "weighted_auxiliary_loss": 0.00044418959296308456,
      "weighted_relation_loss": 0.00044418959296308456
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.578110383536014,
      "flow_matching_loss": 0.02427854686975479,
      "grad_norm": 0.0918184146285057,
      "learning_rate": 2.811942933102517e-05,
      "loss": 0.0247,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042556276312097905,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70900115966797,
      "state_query_norm": 1421.619091796875,
      "state_relation_loss": 0.00042556276312097905,
      "step": 19860,
      "student_distance_mean": 1959.5312866210938,
      "teacher_distance_mean": 1.1527849793434144,
      "weighted_auxiliary_loss": 0.00042556276312097905,
      "weighted_relation_loss": 0.00042556276312097905
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.587464920486436,
      "flow_matching_loss": 0.021669172681868076,
      "grad_norm": 0.08551231771707535,
      "learning_rate": 2.806988460591391e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042683498177211734,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.568247604370114,
      "state_query_norm": 1430.5875366210937,
      "state_relation_loss": 0.00042683498177211734,
      "step": 19870,
      "student_distance_mean": 1957.5374145507812,
      "teacher_distance_mean": 1.1516285300254823,
      "weighted_auxiliary_loss": 0.00042683498177211734,
      "weighted_relation_loss": 0.00042683498177211734
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.596819457436858,
      "flow_matching_loss": 0.022467583417892456,
      "grad_norm": 0.09344275295734406,
      "learning_rate": 2.802036652796074e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004554986284347251,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.95030288696289,
      "state_query_norm": 1427.4105102539063,
      "state_relation_loss": 0.0004554986284347251,
      "step": 19880,
      "student_distance_mean": 1968.330517578125,
      "teacher_distance_mean": 1.1557223796844482,
      "weighted_auxiliary_loss": 0.0004554986284347251,
      "weighted_relation_loss": 0.0004554986284347251
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.606173994387277,
      "flow_matching_loss": 0.021773768216371538,
      "grad_norm": 0.10757362097501755,
      "learning_rate": 2.797087515733478e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041977055952884256,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.96923828125,
      "state_query_norm": 1438.37802734375,
      "state_relation_loss": 0.00041977055952884256,
      "step": 19890,
      "student_distance_mean": 1974.1867919921874,
      "teacher_distance_mean": 1.1601866960525513,
      "weighted_auxiliary_loss": 0.00041977055952884256,
      "weighted_relation_loss": 0.00041977055952884256
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.6155285313377,
      "flow_matching_loss": 0.022966335900127886,
      "grad_norm": 0.0973171815276146,
      "learning_rate": 2.7921410554172723e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004236530337948352,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.06150665283203,
      "state_query_norm": 1444.4575317382812,
      "state_relation_loss": 0.0004236530337948352,
      "step": 19900,
      "student_distance_mean": 1981.9937377929687,
      "teacher_distance_mean": 1.165799045562744,
      "weighted_auxiliary_loss": 0.0004236530337948352,
      "weighted_relation_loss": 0.0004236530337948352
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.62488306828812,
      "flow_matching_loss": 0.023479557409882546,
      "grad_norm": 0.09152315557003021,
      "learning_rate": 2.787197277857871e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047675831010565164,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.82662353515625,
      "state_query_norm": 1442.1802368164062,
      "state_relation_loss": 0.00047675831010565164,
      "step": 19910,
      "student_distance_mean": 1968.4143310546874,
      "teacher_distance_mean": 1.158952569961548,
      "weighted_auxiliary_loss": 0.00047675831010565164,
      "weighted_relation_loss": 0.00047675831010565164
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.63423760523854,
      "flow_matching_loss": 0.023008232936263084,
      "grad_norm": 0.1019541472196579,
      "learning_rate": 2.782256189062429e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000425196680589579,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.847886657714845,
      "state_query_norm": 1443.2919799804688,
      "state_relation_loss": 0.000425196680589579,
      "step": 19920,
      "student_distance_mean": 1972.4529663085937,
      "teacher_distance_mean": 1.1601792097091674,
      "weighted_auxiliary_loss": 0.000425196680589579,
      "weighted_relation_loss": 0.000425196680589579
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.64359214218896,
      "flow_matching_loss": 0.021979104727506638,
      "grad_norm": 0.08383341133594513,
      "learning_rate": 2.777317795034839e-05,
      "loss": 0.0224,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004272007179679349,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.685987091064455,
      "state_query_norm": 1435.5212890625,
      "state_relation_loss": 0.0004272007179679349,
      "step": 19930,
      "student_distance_mean": 1960.2549682617187,
      "teacher_distance_mean": 1.1559913635253907,
      "weighted_auxiliary_loss": 0.0004272007179679349,
      "weighted_relation_loss": 0.0004272007179679349
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.652946679139383,
      "flow_matching_loss": 0.022456517815589903,
      "grad_norm": 0.0893591120839119,
      "learning_rate": 2.7723821017757112e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045613789407070724,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89372100830078,
      "state_query_norm": 1437.195556640625,
      "state_relation_loss": 0.00045613789407070724,
      "step": 19940,
      "student_distance_mean": 1973.4327392578125,
      "teacher_distance_mean": 1.1615132927894591,
      "weighted_auxiliary_loss": 0.00045613789407070724,
      "weighted_relation_loss": 0.00045613789407070724
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.662301216089805,
      "flow_matching_loss": 0.022700381837785243,
      "grad_norm": 0.1041826382279396,
      "learning_rate": 2.7674491152823822e-05,
      "loss": 0.0231,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004352135874796659,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.580980682373045,
      "state_query_norm": 1427.394384765625,
      "state_relation_loss": 0.0004352135874796659,
      "step": 19950,
      "student_distance_mean": 1956.1308959960938,
      "teacher_distance_mean": 1.1509402632713317,
      "weighted_auxiliary_loss": 0.0004352135874796659,
      "weighted_relation_loss": 0.0004352135874796659
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.671655753040223,
      "flow_matching_loss": 0.02077010553330183,
      "grad_norm": 0.1005278006196022,
      "learning_rate": 2.7625188415488944e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004542579757981002,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50789604187012,
      "state_query_norm": 1424.7282592773438,
      "state_relation_loss": 0.0004542579757981002,
      "step": 19960,
      "student_distance_mean": 1950.4021484375,
      "teacher_distance_mean": 1.1478953957557678,
      "weighted_auxiliary_loss": 0.0004542579757981002,
      "weighted_relation_loss": 0.0004542579757981002
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.681010289990645,
      "flow_matching_loss": 0.02248491421341896,
      "grad_norm": 0.0830153226852417,
      "learning_rate": 2.7575912865659924e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004584131034789607,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.00307159423828,
      "state_query_norm": 1444.1861938476563,
      "state_relation_loss": 0.0004584131034789607,
      "step": 19970,
      "student_distance_mean": 1979.506396484375,
      "teacher_distance_mean": 1.1620549082756042,
      "weighted_auxiliary_loss": 0.0004584131034789607,
      "weighted_relation_loss": 0.0004584131034789607
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.690364826941067,
      "flow_matching_loss": 0.023596758022904396,
      "grad_norm": 0.08329524099826813,
      "learning_rate": 2.7526664563211245e-05,
      "loss": 0.024,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004229955549817532,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.042222595214845,
      "state_query_norm": 1411.6442993164062,
      "state_relation_loss": 0.0004229955549817532,
      "step": 19980,
      "student_distance_mean": 1930.298779296875,
      "teacher_distance_mean": 1.1367037415504455,
      "weighted_auxiliary_loss": 0.0004229955549817532,
      "weighted_relation_loss": 0.0004229955549817532
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.699719363891486,
      "flow_matching_loss": 0.022200710698962213,
      "grad_norm": 0.10725296288728714,
      "learning_rate": 2.7477443567984224e-05,
      "loss": 0.0226,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043984225194435564,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72009162902832,
      "state_query_norm": 1432.0904663085937,
      "state_relation_loss": 0.00043984225194435564,
      "step": 19990,
      "student_distance_mean": 1963.2627685546875,
      "teacher_distance_mean": 1.1561924815177917,
      "weighted_auxiliary_loss": 0.00043984225194435564,
      "weighted_relation_loss": 0.00043984225194435564
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.709073900841908,
      "flow_matching_loss": 0.022189498599618673,
      "grad_norm": 0.09010961651802063,
      "learning_rate": 2.7428249939787e-05,
      "loss": 0.0226,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044092470197938384,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.126597213745114,
      "state_query_norm": 1411.7754272460938,
      "state_relation_loss": 0.00044092470197938384,
      "step": 20000,
      "student_distance_mean": 1930.5001098632813,
      "teacher_distance_mean": 1.136007261276245,
      "weighted_auxiliary_loss": 0.00044092470197938384,
      "weighted_relation_loss": 0.00044092470197938384
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.71842843779233,
      "flow_matching_loss": 0.022105163894593716,
      "grad_norm": 0.10779523849487305,
      "learning_rate": 2.7379083738394484e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042517077235970646,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50794372558594,
      "state_query_norm": 1418.6109497070313,
      "state_relation_loss": 0.00042517077235970646,
      "step": 20010,
      "student_distance_mean": 1950.2390502929688,
      "teacher_distance_mean": 1.1474936485290528,
      "weighted_auxiliary_loss": 0.00042517077235970646,
      "weighted_relation_loss": 0.00042517077235970646
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.72778297474275,
      "flow_matching_loss": 0.023487192019820212,
      "grad_norm": 0.11687915027141571,
      "learning_rate": 2.732994502354823e-05,
      "loss": 0.0239,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004192212683847174,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.544013595581056,
      "state_query_norm": 1433.947119140625,
      "state_relation_loss": 0.0004192212683847174,
      "step": 20020,
      "student_distance_mean": 1956.3167236328125,
      "teacher_distance_mean": 1.1478186249732971,
      "weighted_auxiliary_loss": 0.0004192212683847174,
      "weighted_relation_loss": 0.0004192212683847174
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.73713751169317,
      "flow_matching_loss": 0.02273787818849087,
      "grad_norm": 0.09688515216112137,
      "learning_rate": 2.72808338549564e-05,
      "loss": 0.0232,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004422883270308375,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56672973632813,
      "state_query_norm": 1419.8002319335938,
      "state_relation_loss": 0.0004422883270308375,
      "step": 20030,
      "student_distance_mean": 1952.9364990234376,
      "teacher_distance_mean": 1.1482035398483277,
      "weighted_auxiliary_loss": 0.0004422883270308375,
      "weighted_relation_loss": 0.0004422883270308375
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.746492048643592,
      "flow_matching_loss": 0.021777637675404547,
      "grad_norm": 0.09783261269330978,
      "learning_rate": 2.723175029229374e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004510641505476087,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.80303039550781,
      "state_query_norm": 1434.9868896484375,
      "state_relation_loss": 0.0004510641505476087,
      "step": 20040,
      "student_distance_mean": 1964.79765625,
      "teacher_distance_mean": 1.1535323143005372,
      "weighted_auxiliary_loss": 0.0004510641505476087,
      "weighted_relation_loss": 0.0004510641505476087
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.755846585594014,
      "flow_matching_loss": 0.021884677372872828,
      "grad_norm": 0.132609561085701,
      "learning_rate": 2.718269439520138e-05,
      "loss": 0.0223,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004242841736413538,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.09945220947266,
      "state_query_norm": 1449.7330932617188,
      "state_relation_loss": 0.0004242841736413538,
      "step": 20050,
      "student_distance_mean": 1984.8373901367188,
      "teacher_distance_mean": 1.1664398312568665,
      "weighted_auxiliary_loss": 0.0004242841736413538,
      "weighted_relation_loss": 0.0004242841736413538
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.765201122544433,
      "flow_matching_loss": 0.02209530845284462,
      "grad_norm": 0.09468677639961243,
      "learning_rate": 2.7133666223286858e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004329241521190852,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.638262557983396,
      "state_query_norm": 1429.9912231445312,
      "state_relation_loss": 0.0004329241521190852,
      "step": 20060,
      "student_distance_mean": 1959.1277465820312,
      "teacher_distance_mean": 1.1539734959602357,
      "weighted_auxiliary_loss": 0.0004329241521190852,
      "weighted_relation_loss": 0.0004329241521190852
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.774555659494855,
      "flow_matching_loss": 0.021645511500537395,
      "grad_norm": 0.10311980545520782,
      "learning_rate": 2.7084665836124006e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004229689249768853,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42437553405762,
      "state_query_norm": 1423.9870849609374,
      "state_relation_loss": 0.0004229689249768853,
      "step": 20070,
      "student_distance_mean": 1948.0997314453125,
      "teacher_distance_mean": 1.1497246026992798,
      "weighted_auxiliary_loss": 0.0004229689249768853,
      "weighted_relation_loss": 0.0004229689249768853
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.783910196445277,
      "flow_matching_loss": 0.020901303738355637,
      "grad_norm": 0.09292861819267273,
      "learning_rate": 2.703569329325296e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004455144633539021,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.151176071166994,
      "state_query_norm": 1412.89375,
      "state_relation_loss": 0.0004455144633539021,
      "step": 20080,
      "student_distance_mean": 1932.7185546875,
      "teacher_distance_mean": 1.1404999852180482,
      "weighted_auxiliary_loss": 0.0004455144633539021,
      "weighted_relation_loss": 0.0004455144633539021
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.7932647333957,
      "flow_matching_loss": 0.021055596880614758,
      "grad_norm": 0.0986914336681366,
      "learning_rate": 2.698674865417994e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004099330079043284,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2976806640625,
      "state_query_norm": 1412.4542846679688,
      "state_relation_loss": 0.0004099330079043284,
      "step": 20090,
      "student_distance_mean": 1938.6297729492187,
      "teacher_distance_mean": 1.14238383769989,
      "weighted_auxiliary_loss": 0.0004099330079043284,
      "weighted_relation_loss": 0.0004099330079043284
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.802619270346117,
      "flow_matching_loss": 0.022271514311432837,
      "grad_norm": 0.08775071799755096,
      "learning_rate": 2.6937831978377325e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043491309334058313,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.717813873291014,
      "state_query_norm": 1429.820703125,
      "state_relation_loss": 0.00043491309334058313,
      "step": 20100,
      "student_distance_mean": 1963.7599365234375,
      "teacher_distance_mean": 1.1547802090644836,
      "weighted_auxiliary_loss": 0.00043491309334058313,
      "weighted_relation_loss": 0.00043491309334058313
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.81197380729654,
      "flow_matching_loss": 0.022912821359932424,
      "grad_norm": 0.09880825877189636,
      "learning_rate": 2.6888943325283485e-05,
      "loss": 0.0233,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041687640477903185,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43003273010254,
      "state_query_norm": 1421.0276611328125,
      "state_relation_loss": 0.00041687640477903185,
      "step": 20110,
      "student_distance_mean": 1945.0477783203125,
      "teacher_distance_mean": 1.1454174041748046,
      "weighted_auxiliary_loss": 0.00041687640477903185,
      "weighted_relation_loss": 0.00041687640477903185
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.82132834424696,
      "flow_matching_loss": 0.02328628785908222,
      "grad_norm": 0.08354329317808151,
      "learning_rate": 2.6840082754302733e-05,
      "loss": 0.0237,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004169164865743369,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.80071029663086,
      "state_query_norm": 1431.6110229492188,
      "state_relation_loss": 0.0004169164865743369,
      "step": 20120,
      "student_distance_mean": 1966.5615234375,
      "teacher_distance_mean": 1.155654203891754,
      "weighted_auxiliary_loss": 0.0004169164865743369,
      "weighted_relation_loss": 0.0004169164865743369
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.83068288119738,
      "flow_matching_loss": 0.022493963502347468,
      "grad_norm": 0.07856076955795288,
      "learning_rate": 2.679125032480525e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004280763416318223,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.356317138671876,
      "state_query_norm": 1419.2420166015625,
      "state_relation_loss": 0.0004280763416318223,
      "step": 20130,
      "student_distance_mean": 1943.8346313476563,
      "teacher_distance_mean": 1.1438826322555542,
      "weighted_auxiliary_loss": 0.0004280763416318223,
      "weighted_relation_loss": 0.0004280763416318223
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.8400374181478,
      "flow_matching_loss": 0.021905098855495454,
      "grad_norm": 0.10536885261535645,
      "learning_rate": 2.674244609612708e-05,
      "loss": 0.0224,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045367774146143347,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.595743560791014,
      "state_query_norm": 1425.2864379882812,
      "state_relation_loss": 0.00045367774146143347,
      "step": 20140,
      "student_distance_mean": 1952.3216064453125,
      "teacher_distance_mean": 1.1505702972412108,
      "weighted_auxiliary_loss": 0.00045367774146143347,
      "weighted_relation_loss": 0.00045367774146143347
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.849391955098223,
      "flow_matching_loss": 0.021186469681560995,
      "grad_norm": 0.08820692449808121,
      "learning_rate": 2.6693670127569958e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004446185805136338,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45170974731445,
      "state_query_norm": 1415.8725952148438,
      "state_relation_loss": 0.0004446185805136338,
      "step": 20150,
      "student_distance_mean": 1948.8259399414062,
      "teacher_distance_mean": 1.1446474194526672,
      "weighted_auxiliary_loss": 0.0004446185805136338,
      "weighted_relation_loss": 0.0004446185805136338
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.858746492048645,
      "flow_matching_loss": 0.021630227752029895,
      "grad_norm": 0.08212868124246597,
      "learning_rate": 2.6644922478401268e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004244474897859618,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.03255653381348,
      "state_query_norm": 1442.4369384765625,
      "state_relation_loss": 0.0004244474897859618,
      "step": 20160,
      "student_distance_mean": 1977.1018798828125,
      "teacher_distance_mean": 1.1574064135551452,
      "weighted_auxiliary_loss": 0.0004244474897859618,
      "weighted_relation_loss": 0.0004244474897859618
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.868101028999064,
      "flow_matching_loss": 0.02261482961475849,
      "grad_norm": 0.08920196443796158,
      "learning_rate": 2.6596203207854004e-05,
      "loss": 0.023,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042444517312105744,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9964714050293,
      "state_query_norm": 1444.0743896484375,
      "state_relation_loss": 0.00042444517312105744,
      "step": 20170,
      "student_distance_mean": 1979.3107421875,
      "teacher_distance_mean": 1.1591273665428161,
      "weighted_auxiliary_loss": 0.00042444517312105744,
      "weighted_relation_loss": 0.00042444517312105744
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.877455565949486,
      "flow_matching_loss": 0.02216370962560177,
      "grad_norm": 0.08014394342899323,
      "learning_rate": 2.654751237512666e-05,
      "loss": 0.0226,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004337664198828861,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.96912841796875,
      "state_query_norm": 1441.7728149414063,
      "state_relation_loss": 0.0004337664198828861,
      "step": 20180,
      "student_distance_mean": 1974.2352172851563,
      "teacher_distance_mean": 1.1596093893051147,
      "weighted_auxiliary_loss": 0.0004337664198828861,
      "weighted_relation_loss": 0.0004337664198828861
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.886810102899908,
      "flow_matching_loss": 0.02259868700057268,
      "grad_norm": 0.11345767229795456,
      "learning_rate": 2.649885003938323e-05,
      "loss": 0.023,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042876376246567813,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.68648071289063,
      "state_query_norm": 1434.1241577148437,
      "state_relation_loss": 0.00042876376246567813,
      "step": 20190,
      "student_distance_mean": 1963.114697265625,
      "teacher_distance_mean": 1.153586757183075,
      "weighted_auxiliary_loss": 0.00042876376246567813,
      "weighted_relation_loss": 0.00042876376246567813
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.896164639850326,
      "flow_matching_loss": 0.021240400709211828,
      "grad_norm": 0.11649660021066666,
      "learning_rate": 2.6450216259753e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045560793369077147,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.21306800842285,
      "state_query_norm": 1450.7766967773437,
      "state_relation_loss": 0.00045560793369077147,
      "step": 20200,
      "student_distance_mean": 1987.5506469726563,
      "teacher_distance_mean": 1.168973696231842,
      "weighted_auxiliary_loss": 0.00045560793369077147,
      "weighted_relation_loss": 0.00045560793369077147
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.90551917680075,
      "flow_matching_loss": 0.020436868630349637,
      "grad_norm": 0.10305656492710114,
      "learning_rate": 2.640161109533063e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004619028593879193,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.06364097595215,
      "state_query_norm": 1444.1727905273438,
      "state_relation_loss": 0.0004619028593879193,
      "step": 20210,
      "student_distance_mean": 1981.7297729492188,
      "teacher_distance_mean": 1.1666238903999329,
      "weighted_auxiliary_loss": 0.0004619028593879193,
      "weighted_relation_loss": 0.0004619028593879193
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.91487371375117,
      "flow_matching_loss": 0.022435393929481507,
      "grad_norm": 0.09263049066066742,
      "learning_rate": 2.6353034605175936e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045052209461573514,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.857429122924806,
      "state_query_norm": 1432.3333740234375,
      "state_relation_loss": 0.00045052209461573514,
      "step": 20220,
      "student_distance_mean": 1967.4838256835938,
      "teacher_distance_mean": 1.159666609764099,
      "weighted_auxiliary_loss": 0.00045052209461573514,
      "weighted_relation_loss": 0.00045052209461573514
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.92422825070159,
      "flow_matching_loss": 0.025271885097026825,
      "grad_norm": 0.1040286272764206,
      "learning_rate": 2.630448684831398e-05,
      "loss": 0.0257,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004405067098559812,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.726430892944336,
      "state_query_norm": 1436.3772338867188,
      "state_relation_loss": 0.0004405067098559812,
      "step": 20230,
      "student_distance_mean": 1964.0148071289063,
      "teacher_distance_mean": 1.1548561215400697,
      "weighted_auxiliary_loss": 0.0004405067098559812,
      "weighted_relation_loss": 0.0004405067098559812
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.93358278765201,
      "flow_matching_loss": 0.023097680322825907,
      "grad_norm": 0.0988512933254242,
      "learning_rate": 2.625596788373482e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004443445533979684,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.913913345336915,
      "state_query_norm": 1440.78330078125,
      "state_relation_loss": 0.0004443445533979684,
      "step": 20240,
      "student_distance_mean": 1972.1102172851563,
      "teacher_distance_mean": 1.1585362195968627,
      "weighted_auxiliary_loss": 0.0004443445533979684,
      "weighted_relation_loss": 0.0004443445533979684
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.942937324602433,
      "flow_matching_loss": 0.023660659790039062,
      "grad_norm": 0.09017632156610489,
      "learning_rate": 2.620747777039363e-05,
      "loss": 0.0241,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004259456385625526,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72413024902344,
      "state_query_norm": 1432.2296752929688,
      "state_relation_loss": 0.0004259456385625526,
      "step": 20250,
      "student_distance_mean": 1959.6786987304688,
      "teacher_distance_mean": 1.1524141669273376,
      "weighted_auxiliary_loss": 0.0004259456385625526,
      "weighted_relation_loss": 0.0004259456385625526
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.952291861552855,
      "flow_matching_loss": 0.021132526639848947,
      "grad_norm": 0.12269952893257141,
      "learning_rate": 2.6159016567210424e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004259400506271049,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.05441818237305,
      "state_query_norm": 1434.6928344726562,
      "state_relation_loss": 0.0004259400506271049,
      "step": 20260,
      "student_distance_mean": 1977.0939819335938,
      "teacher_distance_mean": 1.1611486792564392,
      "weighted_auxiliary_loss": 0.0004259400506271049,
      "weighted_relation_loss": 0.0004259400506271049
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.961646398503273,
      "flow_matching_loss": 0.020986322406679393,
      "grad_norm": 0.09895160794258118,
      "learning_rate": 2.6110584333070153e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004330871714046225,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3221321105957,
      "state_query_norm": 1412.3081298828124,
      "state_relation_loss": 0.0004330871714046225,
      "step": 20270,
      "student_distance_mean": 1936.549658203125,
      "teacher_distance_mean": 1.1391912579536438,
      "weighted_auxiliary_loss": 0.0004330871714046225,
      "weighted_relation_loss": 0.0004330871714046225
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.971000935453695,
      "flow_matching_loss": 0.02170682568103075,
      "grad_norm": 0.09331203997135162,
      "learning_rate": 2.6062181126822537e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004140110162552446,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.77368392944336,
      "state_query_norm": 1435.256689453125,
      "state_relation_loss": 0.0004140110162552446,
      "step": 20280,
      "student_distance_mean": 1967.7529296875,
      "teacher_distance_mean": 1.158127450942993,
      "weighted_auxiliary_loss": 0.0004140110162552446,
      "weighted_relation_loss": 0.0004140110162552446
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.980355472404117,
      "flow_matching_loss": 0.023062854818999768,
      "grad_norm": 0.11673582345247269,
      "learning_rate": 2.601380700728203e-05,
      "loss": 0.0235,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045370080915745347,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56613006591797,
      "state_query_norm": 1427.7769897460937,
      "state_relation_loss": 0.00045370080915745347,
      "step": 20290,
      "student_distance_mean": 1956.122509765625,
      "teacher_distance_mean": 1.150258433818817,
      "weighted_auxiliary_loss": 0.00045370080915745347,
      "weighted_relation_loss": 0.00045370080915745347
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.989710009354535,
      "flow_matching_loss": 0.022217386588454246,
      "grad_norm": 0.08446664363145828,
      "learning_rate": 2.596546203322777e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043528863170649856,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.029645156860354,
      "state_query_norm": 1448.9143798828125,
      "state_relation_loss": 0.00043528863170649856,
      "step": 20300,
      "student_distance_mean": 1978.52744140625,
      "teacher_distance_mean": 1.1663417100906373,
      "weighted_auxiliary_loss": 0.00043528863170649856,
      "weighted_relation_loss": 0.00043528863170649856
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 18.999064546304957,
      "flow_matching_loss": 0.020726574584841728,
      "grad_norm": 0.10811800509691238,
      "learning_rate": 2.5917146263403457e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004157427698373795,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64724273681641,
      "state_query_norm": 1434.1935913085938,
      "state_relation_loss": 0.0004157427698373795,
      "step": 20310,
      "student_distance_mean": 1963.6401489257812,
      "teacher_distance_mean": 1.1589830994606019,
      "weighted_auxiliary_loss": 0.0004157427698373795,
      "weighted_relation_loss": 0.0004157427698373795
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.00841908325538,
      "flow_matching_loss": 0.020493587665259837,
      "grad_norm": 0.08034662157297134,
      "learning_rate": 2.5868859756517296e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00042145546758547424,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 36.10210952758789,
      "state_query_norm": 1442.4717407226562,
      "state_relation_loss": 0.00042145546758547424,
      "step": 20320,
      "student_distance_mean": 1986.5201171875,
      "teacher_distance_mean": 1.171131181716919,
      "weighted_auxiliary_loss": 0.00042145546758547424,
      "weighted_relation_loss": 0.00042145546758547424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.0177736202058,
      "flow_matching_loss": 0.022103326208889485,
      "grad_norm": 0.09342877566814423,
      "learning_rate": 2.5820602571241947e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004232876730384305,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.342643356323244,
      "state_query_norm": 1422.53828125,
      "state_relation_loss": 0.0004232876730384305,
      "step": 20330,
      "student_distance_mean": 1941.7692749023438,
      "teacher_distance_mean": 1.1449314832687378,
      "weighted_auxiliary_loss": 0.0004232876730384305,
      "weighted_relation_loss": 0.0004232876730384305
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.02712815715622,
      "flow_matching_loss": 0.02121577113866806,
      "grad_norm": 0.09125294536352158,
      "learning_rate": 2.577237476621442e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004061169398482889,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57159309387207,
      "state_query_norm": 1430.6864868164062,
      "state_relation_loss": 0.0004061169398482889,
      "step": 20340,
      "student_distance_mean": 1955.6594970703125,
      "teacher_distance_mean": 1.1514529824256896,
      "weighted_auxiliary_loss": 0.0004061169398482889,
      "weighted_relation_loss": 0.0004061169398482889
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.036482694106642,
      "flow_matching_loss": 0.02183180581778288,
      "grad_norm": 0.087084099650383,
      "learning_rate": 2.572417640003609e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000414484363864176,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33670768737793,
      "state_query_norm": 1425.62958984375,
      "state_relation_loss": 0.000414484363864176,
      "step": 20350,
      "student_distance_mean": 1945.6008056640626,
      "teacher_distance_mean": 1.148086941242218,
      "weighted_auxiliary_loss": 0.000414484363864176,
      "weighted_relation_loss": 0.000414484363864176
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.045837231057064,
      "flow_matching_loss": 0.020228358916938305,
      "grad_norm": 0.09193696081638336,
      "learning_rate": 2.5676007531272473e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044257829140406104,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55017776489258,
      "state_query_norm": 1430.7638549804688,
      "state_relation_loss": 0.00044257829140406104,
      "step": 20360,
      "student_distance_mean": 1953.0585083007813,
      "teacher_distance_mean": 1.1515903711318969,
      "weighted_auxiliary_loss": 0.00044257829140406104,
      "weighted_relation_loss": 0.00044257829140406104
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.055191768007482,
      "flow_matching_loss": 0.022198157384991646,
      "grad_norm": 0.09631630778312683,
      "learning_rate": 2.5627868218453332e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004813929583178833,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.67874984741211,
      "state_query_norm": 1430.9571044921875,
      "state_relation_loss": 0.0004813929583178833,
      "step": 20370,
      "student_distance_mean": 1957.3591674804688,
      "teacher_distance_mean": 1.1539870142936706,
      "weighted_auxiliary_loss": 0.0004813929583178833,
      "weighted_relation_loss": 0.0004813929583178833
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.064546304957904,
      "flow_matching_loss": 0.020848206244409084,
      "grad_norm": 0.08184283971786499,
      "learning_rate": 2.5579758520072443e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044609459582716225,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.179339218139646,
      "state_query_norm": 1438.4068237304687,
      "state_relation_loss": 0.00044609459582716225,
      "step": 20380,
      "student_distance_mean": 1980.697265625,
      "teacher_distance_mean": 1.1655359625816346,
      "weighted_auxiliary_loss": 0.00044609459582716225,
      "weighted_relation_loss": 0.00044609459582716225
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.073900841908326,
      "flow_matching_loss": 0.0216487642377615,
      "grad_norm": 0.09856909513473511,
      "learning_rate": 2.5531678494587615e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042111900984309614,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4484058380127,
      "state_query_norm": 1418.0369506835937,
      "state_relation_loss": 0.00042111900984309614,
      "step": 20390,
      "student_distance_mean": 1944.7737548828125,
      "teacher_distance_mean": 1.1405545592308044,
      "weighted_auxiliary_loss": 0.00042111900984309614,
      "weighted_relation_loss": 0.00042111900984309614
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.08325537885875,
      "flow_matching_loss": 0.021663113869726656,
      "grad_norm": 0.11067566275596619,
      "learning_rate": 2.5483628200420644e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004268776247045025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.87265625,
      "state_query_norm": 1425.2997436523438,
      "state_relation_loss": 0.0004268776247045025,
      "step": 20400,
      "student_distance_mean": 1964.1377319335938,
      "teacher_distance_mean": 1.15323349237442,
      "weighted_auxiliary_loss": 0.0004268776247045025,
      "weighted_relation_loss": 0.0004268776247045025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.092609915809167,
      "flow_matching_loss": 0.021541488729417323,
      "grad_norm": 0.10944323241710663,
      "learning_rate": 2.5435607695957154e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004611879849107936,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.40448341369629,
      "state_query_norm": 1460.4990234375,
      "state_relation_loss": 0.0004611879849107936,
      "step": 20410,
      "student_distance_mean": 2000.4401000976563,
      "teacher_distance_mean": 1.1718899130821228,
      "weighted_auxiliary_loss": 0.0004611879849107936,
      "weighted_relation_loss": 0.0004611879849107936
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.10196445275959,
      "flow_matching_loss": 0.021096340101212262,
      "grad_norm": 0.11387477815151215,
      "learning_rate": 2.5387617039546586e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043828105553984643,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69890785217285,
      "state_query_norm": 1426.9642822265625,
      "state_relation_loss": 0.00043828105553984643,
      "step": 20420,
      "student_distance_mean": 1961.384765625,
      "teacher_distance_mean": 1.1485399961471559,
      "weighted_auxiliary_loss": 0.00043828105553984643,
      "weighted_relation_loss": 0.00043828105553984643
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.11131898971001,
      "flow_matching_loss": 0.02149054203182459,
      "grad_norm": 0.07904481887817383,
      "learning_rate": 2.5339656289502106e-05,
      "loss": 0.0219,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043120758782606573,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.286682510375975,
      "state_query_norm": 1423.675341796875,
      "state_relation_loss": 0.00043120758782606573,
      "step": 20430,
      "student_distance_mean": 1943.6473388671875,
      "teacher_distance_mean": 1.1399120211601257,
      "weighted_auxiliary_loss": 0.00043120758782606573,
      "weighted_relation_loss": 0.00043120758782606573
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.12067352666043,
      "flow_matching_loss": 0.021076073311269282,
      "grad_norm": 0.09328348934650421,
      "learning_rate": 2.529172550410056e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043466851639095694,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72276153564453,
      "state_query_norm": 1429.5169311523437,
      "state_relation_loss": 0.00043466851639095694,
      "step": 20440,
      "student_distance_mean": 1960.0082641601562,
      "teacher_distance_mean": 1.1510069489479064,
      "weighted_auxiliary_loss": 0.00043466851639095694,
      "weighted_relation_loss": 0.00043466851639095694
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.13002806361085,
      "flow_matching_loss": 0.021482239477336407,
      "grad_norm": 0.09572256356477737,
      "learning_rate": 2.524382474158234e-05,
      "loss": 0.0219,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042521646537352353,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.703745651245114,
      "state_query_norm": 1433.2956787109374,
      "state_relation_loss": 0.00042521646537352353,
      "step": 20450,
      "student_distance_mean": 1962.8995361328125,
      "teacher_distance_mean": 1.150545024871826,
      "weighted_auxiliary_loss": 0.00042521646537352353,
      "weighted_relation_loss": 0.00042521646537352353
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.139382600561273,
      "flow_matching_loss": 0.02216683980077505,
      "grad_norm": 0.09316612035036087,
      "learning_rate": 2.5195954060151433e-05,
      "loss": 0.0226,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045351564476732166,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.84363212585449,
      "state_query_norm": 1438.6463012695312,
      "state_relation_loss": 0.00045351564476732166,
      "step": 20460,
      "student_distance_mean": 1972.6158325195313,
      "teacher_distance_mean": 1.1574030756950378,
      "weighted_auxiliary_loss": 0.00045351564476732166,
      "weighted_relation_loss": 0.00045351564476732166
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.14873713751169,
      "flow_matching_loss": 0.02204828318208456,
      "grad_norm": 0.09283851832151413,
      "learning_rate": 2.5148113517975212e-05,
      "loss": 0.0225,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004225797776598483,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.23483085632324,
      "state_query_norm": 1455.613623046875,
      "state_relation_loss": 0.0004225797776598483,
      "step": 20470,
      "student_distance_mean": 1995.0330078125,
      "teacher_distance_mean": 1.1733244061470032,
      "weighted_auxiliary_loss": 0.0004225797776598483,
      "weighted_relation_loss": 0.0004225797776598483
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.158091674462113,
      "flow_matching_loss": 0.022110424749553204,
      "grad_norm": 0.08479534834623337,
      "learning_rate": 2.5100303173184447e-05,
      "loss": 0.0226,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00046154591836966574,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91964263916016,
      "state_query_norm": 1433.7571533203125,
      "state_relation_loss": 0.00046154591836966574,
      "step": 20480,
      "student_distance_mean": 1970.2745239257813,
      "teacher_distance_mean": 1.1576037287712098,
      "weighted_auxiliary_loss": 0.00046154591836966574,
      "weighted_relation_loss": 0.00046154591836966574
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.167446211412535,
      "flow_matching_loss": 0.020243562292307615,
      "grad_norm": 0.09269844740629196,
      "learning_rate": 2.5052523083873196e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043806496541947125,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5947681427002,
      "state_query_norm": 1419.7127685546875,
      "state_relation_loss": 0.00043806496541947125,
      "step": 20490,
      "student_distance_mean": 1951.1612548828125,
      "teacher_distance_mean": 1.1458788633346557,
      "weighted_auxiliary_loss": 0.00043806496541947125,
      "weighted_relation_loss": 0.00043806496541947125
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.176800748362957,
      "flow_matching_loss": 0.02134052198380232,
      "grad_norm": 0.08393654227256775,
      "learning_rate": 2.500477330809881e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038943246763665227,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55375442504883,
      "state_query_norm": 1438.5375854492188,
      "state_relation_loss": 0.00038943246763665227,
      "step": 20500,
      "student_distance_mean": 1959.57138671875,
      "teacher_distance_mean": 1.1479400992393494,
      "weighted_auxiliary_loss": 0.00038943246763665227,
      "weighted_relation_loss": 0.00038943246763665227
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.186155285313376,
      "flow_matching_loss": 0.021895773988217116,
      "grad_norm": 0.08043701201677322,
      "learning_rate": 2.4957053903881733e-05,
      "loss": 0.0223,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044383720960468055,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.92420921325684,
      "state_query_norm": 1438.3491455078124,
      "state_relation_loss": 0.00044383720960468055,
      "step": 20510,
      "student_distance_mean": 1971.7893920898437,
      "teacher_distance_mean": 1.1592821717262267,
      "weighted_auxiliary_loss": 0.00044383720960468055,
      "weighted_relation_loss": 0.00044383720960468055
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.195509822263798,
      "flow_matching_loss": 0.02390819024294615,
      "grad_norm": 0.08101746439933777,
      "learning_rate": 2.4909364929205576e-05,
      "loss": 0.0244,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044188241881784054,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91615600585938,
      "state_query_norm": 1425.210986328125,
      "state_relation_loss": 0.00044188241881784054,
      "step": 20520,
      "student_distance_mean": 1969.5210205078124,
      "teacher_distance_mean": 1.1604096651077271,
      "weighted_auxiliary_loss": 0.00044188241881784054,
      "weighted_relation_loss": 0.00044188241881784054
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.20486435921422,
      "flow_matching_loss": 0.020849496126174927,
      "grad_norm": 0.07235271483659744,
      "learning_rate": 2.4861706442016923e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004173050896497443,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.811153411865234,
      "state_query_norm": 1447.053173828125,
      "state_relation_loss": 0.0004173050896497443,
      "step": 20530,
      "student_distance_mean": 1971.2384765625,
      "teacher_distance_mean": 1.1614549279212951,
      "weighted_auxiliary_loss": 0.0004173050896497443,
      "weighted_relation_loss": 0.0004173050896497443
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.21421889616464,
      "flow_matching_loss": 0.021025602612644433,
      "grad_norm": 0.08243236690759659,
      "learning_rate": 2.481407850022533e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004454807989532128,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.455406188964844,
      "state_query_norm": 1422.8951416015625,
      "state_relation_loss": 0.0004454807989532128,
      "step": 20540,
      "student_distance_mean": 1947.3497680664063,
      "teacher_distance_mean": 1.145005226135254,
      "weighted_auxiliary_loss": 0.0004454807989532128,
      "weighted_relation_loss": 0.0004454807989532128
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.22357343311506,
      "flow_matching_loss": 0.02153839487582445,
      "grad_norm": 0.08926943689584732,
      "learning_rate": 2.4766481161703213e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043052364198956636,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.92690582275391,
      "state_query_norm": 1433.2900024414062,
      "state_relation_loss": 0.00043052364198956636,
      "step": 20550,
      "student_distance_mean": 1968.3467407226562,
      "teacher_distance_mean": 1.1565911769866943,
      "weighted_auxiliary_loss": 0.00043052364198956636,
      "weighted_relation_loss": 0.00043052364198956636
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.232927970065482,
      "flow_matching_loss": 0.021153487823903562,
      "grad_norm": 0.07518612593412399,
      "learning_rate": 2.4718914484285876e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043114393774885685,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6827751159668,
      "state_query_norm": 1429.3173950195312,
      "state_relation_loss": 0.00043114393774885685,
      "step": 20560,
      "student_distance_mean": 1958.8400024414063,
      "teacher_distance_mean": 1.1487705945968627,
      "weighted_auxiliary_loss": 0.00043114393774885685,
      "weighted_relation_loss": 0.00043114393774885685
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.242282507015904,
      "flow_matching_loss": 0.02002690825611353,
      "grad_norm": 0.07406957447528839,
      "learning_rate": 2.467137852577129e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044336619321256877,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.706607818603516,
      "state_query_norm": 1439.0003540039063,
      "state_relation_loss": 0.00044336619321256877,
      "step": 20570,
      "student_distance_mean": 1965.0773803710938,
      "teacher_distance_mean": 1.151814329624176,
      "weighted_auxiliary_loss": 0.00044336619321256877,
      "weighted_relation_loss": 0.00044336619321256877
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.251637043966323,
      "flow_matching_loss": 0.021298754960298538,
      "grad_norm": 0.08998668938875198,
      "learning_rate": 2.4623873343920123e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000442117583588697,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78932991027832,
      "state_query_norm": 1436.8270141601563,
      "state_relation_loss": 0.000442117583588697,
      "step": 20580,
      "student_distance_mean": 1966.581982421875,
      "teacher_distance_mean": 1.1520977854728698,
      "weighted_auxiliary_loss": 0.000442117583588697,
      "weighted_relation_loss": 0.000442117583588697
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.260991580916745,
      "flow_matching_loss": 0.020884406007826328,
      "grad_norm": 0.09065330028533936,
      "learning_rate": 2.4576398996455656e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042992563685402275,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.633343887329104,
      "state_query_norm": 1429.2108154296875,
      "state_relation_loss": 0.00042992563685402275,
      "step": 20590,
      "student_distance_mean": 1959.0492919921876,
      "teacher_distance_mean": 1.1472734332084655,
      "weighted_auxiliary_loss": 0.00042992563685402275,
      "weighted_relation_loss": 0.00042992563685402275
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.270346117867167,
      "flow_matching_loss": 0.022688659001141788,
      "grad_norm": 0.10462542623281479,
      "learning_rate": 2.452895554106368e-05,
      "loss": 0.0231,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042468207830097525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.482658767700194,
      "state_query_norm": 1428.97392578125,
      "state_relation_loss": 0.00042468207830097525,
      "step": 20600,
      "student_distance_mean": 1956.0336059570313,
      "teacher_distance_mean": 1.1481256008148193,
      "weighted_auxiliary_loss": 0.00042468207830097525,
      "weighted_relation_loss": 0.00042468207830097525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.279700654817585,
      "flow_matching_loss": 0.022496120445430277,
      "grad_norm": 0.08239777386188507,
      "learning_rate": 2.4481543035392506e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004225206415867433,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.87680015563965,
      "state_query_norm": 1437.8319213867187,
      "state_relation_loss": 0.0004225206415867433,
      "step": 20610,
      "student_distance_mean": 1970.7569091796875,
      "teacher_distance_mean": 1.1571882367134094,
      "weighted_auxiliary_loss": 0.0004225206415867433,
      "weighted_relation_loss": 0.0004225206415867433
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.289055191768007,
      "flow_matching_loss": 0.021210342459380627,
      "grad_norm": 0.1597922146320343,
      "learning_rate": 2.4434161537052774e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042881716217380017,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.817112350463866,
      "state_query_norm": 1438.4334350585937,
      "state_relation_loss": 0.00042881716217380017,
      "step": 20620,
      "student_distance_mean": 1968.7430786132813,
      "teacher_distance_mean": 1.154709815979004,
      "weighted_auxiliary_loss": 0.00042881716217380017,
      "weighted_relation_loss": 0.00042881716217380017
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.29840972871843,
      "flow_matching_loss": 0.020803554356098174,
      "grad_norm": 0.07953793555498123,
      "learning_rate": 2.4386811103617474e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004259461944457144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.204569244384764,
      "state_query_norm": 1421.3129760742188,
      "state_relation_loss": 0.0004259461944457144,
      "step": 20630,
      "student_distance_mean": 1941.3323486328125,
      "teacher_distance_mean": 1.1378289580345153,
      "weighted_auxiliary_loss": 0.0004259461944457144,
      "weighted_relation_loss": 0.0004259461944457144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.30776426566885,
      "flow_matching_loss": 0.02104618400335312,
      "grad_norm": 0.08283626288175583,
      "learning_rate": 2.4339491792621833e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041505180124659093,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.87092666625976,
      "state_query_norm": 1437.62939453125,
      "state_relation_loss": 0.00041505180124659093,
      "step": 20640,
      "student_distance_mean": 1971.9259765625,
      "teacher_distance_mean": 1.1560017943382264,
      "weighted_auxiliary_loss": 0.00041505180124659093,
      "weighted_relation_loss": 0.00041505180124659093
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.31711880261927,
      "flow_matching_loss": 0.020174005441367625,
      "grad_norm": 0.10200018435716629,
      "learning_rate": 2.4292203661563312e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003958025306928903,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54910011291504,
      "state_query_norm": 1427.3993530273438,
      "state_relation_loss": 0.0003958025306928903,
      "step": 20650,
      "student_distance_mean": 1956.0828247070312,
      "teacher_distance_mean": 1.148757517337799,
      "weighted_auxiliary_loss": 0.0003958025306928903,
      "weighted_relation_loss": 0.0003958025306928903
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.32647333956969,
      "flow_matching_loss": 0.021224090829491615,
      "grad_norm": 0.10808536410331726,
      "learning_rate": 2.424494676790141e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004498336085816845,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5659740447998,
      "state_query_norm": 1413.02578125,
      "state_relation_loss": 0.0004498336085816845,
      "step": 20660,
      "student_distance_mean": 1948.2982055664063,
      "teacher_distance_mean": 1.1457985401153565,
      "weighted_auxiliary_loss": 0.0004498336085816845,
      "weighted_relation_loss": 0.0004498336085816845
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.335827876520113,
      "flow_matching_loss": 0.021992223151028157,
      "grad_norm": 0.09173881262540817,
      "learning_rate": 2.4197721169057745e-05,
      "loss": 0.0224,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043176143954042344,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.680695343017575,
      "state_query_norm": 1432.6069213867188,
      "state_relation_loss": 0.00043176143954042344,
      "step": 20670,
      "student_distance_mean": 1962.4808471679687,
      "teacher_distance_mean": 1.1561513543128967,
      "weighted_auxiliary_loss": 0.00043176143954042344,
      "weighted_relation_loss": 0.00043176143954042344
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.345182413470532,
      "flow_matching_loss": 0.021245461516082288,
      "grad_norm": 0.09273330122232437,
      "learning_rate": 2.4150526922415857e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004297037550713867,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.239253234863284,
      "state_query_norm": 1452.6664916992188,
      "state_relation_loss": 0.0004297037550713867,
      "step": 20680,
      "student_distance_mean": 1991.7958984375,
      "teacher_distance_mean": 1.1690298199653626,
      "weighted_auxiliary_loss": 0.0004297037550713867,
      "weighted_relation_loss": 0.0004297037550713867
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.354536950420954,
      "flow_matching_loss": 0.022458056174218654,
      "grad_norm": 0.0887150689959526,
      "learning_rate": 2.41033640853212e-05,
      "loss": 0.0229,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004232678416883573,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.95812759399414,
      "state_query_norm": 1436.3773071289063,
      "state_relation_loss": 0.0004232678416883573,
      "step": 20690,
      "student_distance_mean": 1973.945654296875,
      "teacher_distance_mean": 1.1571617364883422,
      "weighted_auxiliary_loss": 0.0004232678416883573,
      "weighted_relation_loss": 0.0004232678416883573
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.363891487371376,
      "flow_matching_loss": 0.019416504818946124,
      "grad_norm": 0.07430626451969147,
      "learning_rate": 2.405623271508108e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040687666041776537,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.577580261230466,
      "state_query_norm": 1425.9459594726563,
      "state_relation_loss": 0.00040687666041776537,
      "step": 20700,
      "student_distance_mean": 1955.7852783203125,
      "teacher_distance_mean": 1.1480209589004517,
      "weighted_auxiliary_loss": 0.00040687666041776537,
      "weighted_relation_loss": 0.00040687666041776537
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.373246024321794,
      "flow_matching_loss": 0.019880281202495097,
      "grad_norm": 0.0875932052731514,
      "learning_rate": 2.4009132868964522e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004206758429063484,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.953558349609374,
      "state_query_norm": 1452.6976806640625,
      "state_relation_loss": 0.0004206758429063484,
      "step": 20710,
      "student_distance_mean": 1976.3118530273437,
      "teacher_distance_mean": 1.1603771448135376,
      "weighted_auxiliary_loss": 0.0004206758429063484,
      "weighted_relation_loss": 0.0004206758429063484
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.382600561272216,
      "flow_matching_loss": 0.02223132476210594,
      "grad_norm": 0.08685635775327682,
      "learning_rate": 2.3962064604202327e-05,
      "loss": 0.0226,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003967222030041739,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.60643615722656,
      "state_query_norm": 1390.0320434570312,
      "state_relation_loss": 0.0003967222030041739,
      "step": 20720,
      "student_distance_mean": 1900.9099975585937,
      "teacher_distance_mean": 1.1168883919715882,
      "weighted_auxiliary_loss": 0.0003967222030041739,
      "weighted_relation_loss": 0.0003967222030041739
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.39195509822264,
      "flow_matching_loss": 0.02011332754045725,
      "grad_norm": 0.0932726040482521,
      "learning_rate": 2.3915027977986858e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041464649839326737,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.08872299194336,
      "state_query_norm": 1445.738818359375,
      "state_relation_loss": 0.00041464649839326737,
      "step": 20730,
      "student_distance_mean": 1984.5538452148437,
      "teacher_distance_mean": 1.1688037037849426,
      "weighted_auxiliary_loss": 0.00041464649839326737,
      "weighted_relation_loss": 0.00041464649839326737
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.40130963517306,
      "flow_matching_loss": 0.022918580938130616,
      "grad_norm": 0.10659631341695786,
      "learning_rate": 2.386802304747205e-05,
      "loss": 0.0234,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004429561551660299,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75425071716309,
      "state_query_norm": 1426.3478759765626,
      "state_relation_loss": 0.0004429561551660299,
      "step": 20740,
      "student_distance_mean": 1959.7942993164063,
      "teacher_distance_mean": 1.1544932961463927,
      "weighted_auxiliary_loss": 0.0004429561551660299,
      "weighted_relation_loss": 0.0004429561551660299
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.41066417212348,
      "flow_matching_loss": 0.021644631028175355,
      "grad_norm": 0.08666390180587769,
      "learning_rate": 2.382104986977332e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040620857034809885,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30900764465332,
      "state_query_norm": 1415.57568359375,
      "state_relation_loss": 0.00040620857034809885,
      "step": 20750,
      "student_distance_mean": 1939.621826171875,
      "teacher_distance_mean": 1.140894103050232,
      "weighted_auxiliary_loss": 0.00040620857034809885,
      "weighted_relation_loss": 0.00040620857034809885
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.4200187090739,
      "flow_matching_loss": 0.021142880897969007,
      "grad_norm": 0.09050948917865753,
      "learning_rate": 2.377410850196749e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041290353401564063,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.871307373046875,
      "state_query_norm": 1437.961328125,
      "state_relation_loss": 0.00041290353401564063,
      "step": 20760,
      "student_distance_mean": 1971.7520629882813,
      "teacher_distance_mean": 1.1586854577064514,
      "weighted_auxiliary_loss": 0.00041290353401564063,
      "weighted_relation_loss": 0.00041290353401564063
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.429373246024323,
      "flow_matching_loss": 0.02190953018143773,
      "grad_norm": 0.0679941177368164,
      "learning_rate": 2.3727199001092797e-05,
      "loss": 0.0224,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044076947669964285,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73890838623047,
      "state_query_norm": 1432.4300903320313,
      "state_relation_loss": 0.00044076947669964285,
      "step": 20770,
      "student_distance_mean": 1962.8573486328125,
      "teacher_distance_mean": 1.1544355750083923,
      "weighted_auxiliary_loss": 0.00044076947669964285,
      "weighted_relation_loss": 0.00044076947669964285
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.43872778297474,
      "flow_matching_loss": 0.019777033291757107,
      "grad_norm": 0.1132146492600441,
      "learning_rate": 2.3680321424148677e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000428602218744345,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.008535385131836,
      "state_query_norm": 1441.340380859375,
      "state_relation_loss": 0.000428602218744345,
      "step": 20780,
      "student_distance_mean": 1976.3916870117187,
      "teacher_distance_mean": 1.163469660282135,
      "weighted_auxiliary_loss": 0.000428602218744345,
      "weighted_relation_loss": 0.000428602218744345
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.448082319925163,
      "flow_matching_loss": 0.021547385677695274,
      "grad_norm": 0.09195184707641602,
      "learning_rate": 2.3633475828095793e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004108112188987434,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3340560913086,
      "state_query_norm": 1420.364453125,
      "state_relation_loss": 0.0004108112188987434,
      "step": 20790,
      "student_distance_mean": 1944.4390258789062,
      "teacher_distance_mean": 1.147400403022766,
      "weighted_auxiliary_loss": 0.0004108112188987434,
      "weighted_relation_loss": 0.0004108112188987434
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.457436856875585,
      "flow_matching_loss": 0.02182821128517389,
      "grad_norm": 0.07497496902942657,
      "learning_rate": 2.358666226985599e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040359254053328185,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.82848472595215,
      "state_query_norm": 1399.299267578125,
      "state_relation_loss": 0.00040359254053328185,
      "step": 20800,
      "student_distance_mean": 1916.0705322265626,
      "teacher_distance_mean": 1.1305116295814515,
      "weighted_auxiliary_loss": 0.00040359254053328185,
      "weighted_relation_loss": 0.00040359254053328185
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.466791393826007,
      "flow_matching_loss": 0.021345415338873863,
      "grad_norm": 0.0965869277715683,
      "learning_rate": 2.353988080631213e-05,
      "loss": 0.0218,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042588405485730616,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.322794342041014,
      "state_query_norm": 1449.2098022460937,
      "state_relation_loss": 0.00042588405485730616,
      "step": 20810,
      "student_distance_mean": 1994.9728759765626,
      "teacher_distance_mean": 1.1730070948600768,
      "weighted_auxiliary_loss": 0.00042588405485730616,
      "weighted_relation_loss": 0.00042588405485730616
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.476145930776426,
      "flow_matching_loss": 0.02086438275873661,
      "grad_norm": 0.08413335680961609,
      "learning_rate": 2.3493131494308142e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042814478802029045,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.097836303710935,
      "state_query_norm": 1449.6334838867188,
      "state_relation_loss": 0.00042814478802029045,
      "step": 20820,
      "student_distance_mean": 1984.7832153320312,
      "teacher_distance_mean": 1.1636195540428163,
      "weighted_auxiliary_loss": 0.00042814478802029045,
      "weighted_relation_loss": 0.00042814478802029045
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.485500467726848,
      "flow_matching_loss": 0.020760471001267433,
      "grad_norm": 0.08487585186958313,
      "learning_rate": 2.344641439064881e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042782779491972177,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.22540473937988,
      "state_query_norm": 1445.402880859375,
      "state_relation_loss": 0.00042782779491972177,
      "step": 20830,
      "student_distance_mean": 1986.1060180664062,
      "teacher_distance_mean": 1.166843056678772,
      "weighted_auxiliary_loss": 0.00042782779491972177,
      "weighted_relation_loss": 0.00042782779491972177
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.49485500467727,
      "flow_matching_loss": 0.020643503963947298,
      "grad_norm": 0.0915507972240448,
      "learning_rate": 2.339972955209984e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044356215803418306,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57441749572754,
      "state_query_norm": 1420.3703735351562,
      "state_relation_loss": 0.00044356215803418306,
      "step": 20840,
      "student_distance_mean": 1950.6268310546875,
      "teacher_distance_mean": 1.147682499885559,
      "weighted_auxiliary_loss": 0.00044356215803418306,
      "weighted_relation_loss": 0.00044356215803418306
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.504209541627688,
      "flow_matching_loss": 0.023776224441826345,
      "grad_norm": 0.11239451915025711,
      "learning_rate": 2.335307703538771e-05,
      "loss": 0.0242,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044391855772119015,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.15394477844238,
      "state_query_norm": 1451.4317016601562,
      "state_relation_loss": 0.00044391855772119015,
      "step": 20850,
      "student_distance_mean": 1987.521875,
      "teacher_distance_mean": 1.1674729466438294,
      "weighted_auxiliary_loss": 0.00044391855772119015,
      "weighted_relation_loss": 0.00044391855772119015
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.51356407857811,
      "flow_matching_loss": 0.022330452874302863,
      "grad_norm": 0.09799055755138397,
      "learning_rate": 2.330645689719962e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004066696448717266,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37548294067383,
      "state_query_norm": 1425.5586669921875,
      "state_relation_loss": 0.0004066696448717266,
      "step": 20860,
      "student_distance_mean": 1948.5154541015625,
      "teacher_distance_mean": 1.1441692113876343,
      "weighted_auxiliary_loss": 0.0004066696448717266,
      "weighted_relation_loss": 0.0004066696448717266
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.522918615528532,
      "flow_matching_loss": 0.020893635228276253,
      "grad_norm": 0.09762012213468552,
      "learning_rate": 2.3259869194183414e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004184998135315254,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.722449111938474,
      "state_query_norm": 1433.5736083984375,
      "state_relation_loss": 0.0004184998135315254,
      "step": 20870,
      "student_distance_mean": 1965.0948364257813,
      "teacher_distance_mean": 1.1532073855400085,
      "weighted_auxiliary_loss": 0.0004184998135315254,
      "weighted_relation_loss": 0.0004184998135315254
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.532273152478954,
      "flow_matching_loss": 0.01969177033752203,
      "grad_norm": 0.0993252545595169,
      "learning_rate": 2.321331398294759e-05,
      "loss": 0.0201,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042581711895763876,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.815629196166995,
      "state_query_norm": 1435.771435546875,
      "state_relation_loss": 0.00042581711895763876,
      "step": 20880,
      "student_distance_mean": 1969.8756469726563,
      "teacher_distance_mean": 1.1595298528671265,
      "weighted_auxiliary_loss": 0.00042581711895763876,
      "weighted_relation_loss": 0.00042581711895763876
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.541627689429372,
      "flow_matching_loss": 0.020615063793957233,
      "grad_norm": 0.0987248420715332,
      "learning_rate": 2.3166791320061094e-05,
      "loss": 0.021,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042392194154672325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.590693664550784,
      "state_query_norm": 1433.716455078125,
      "state_relation_loss": 0.00042392194154672325,
      "step": 20890,
      "student_distance_mean": 1958.1618530273438,
      "teacher_distance_mean": 1.1547520756721497,
      "weighted_auxiliary_loss": 0.00042392194154672325,
      "weighted_relation_loss": 0.00042392194154672325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.550982226379794,
      "flow_matching_loss": 0.020290450006723405,
      "grad_norm": 0.09281088411808014,
      "learning_rate": 2.312030126205335e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040595303871668873,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.178002166748044,
      "state_query_norm": 1419.7956787109374,
      "state_relation_loss": 0.00040595303871668873,
      "step": 20900,
      "student_distance_mean": 1940.3746948242188,
      "teacher_distance_mean": 1.1444516777992249,
      "weighted_auxiliary_loss": 0.00040595303871668873,
      "weighted_relation_loss": 0.00040595303871668873
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.560336763330216,
      "flow_matching_loss": 0.01953007047995925,
      "grad_norm": 0.13498494029045105,
      "learning_rate": 2.3073843865414162e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040210509323515,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64670295715332,
      "state_query_norm": 1426.276171875,
      "state_relation_loss": 0.00040210509323515,
      "step": 20910,
      "student_distance_mean": 1956.7959106445312,
      "teacher_distance_mean": 1.1540740132331848,
      "weighted_auxiliary_loss": 0.00040210509323515,
      "weighted_relation_loss": 0.00040210509323515
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.569691300280635,
      "flow_matching_loss": 0.020887059159576893,
      "grad_norm": 0.08877113461494446,
      "learning_rate": 2.302741918659363e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004148503619944677,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.7378646850586,
      "state_query_norm": 1427.3119995117188,
      "state_relation_loss": 0.0004148503619944677,
      "step": 20920,
      "student_distance_mean": 1962.7043334960938,
      "teacher_distance_mean": 1.158509612083435,
      "weighted_auxiliary_loss": 0.0004148503619944677,
      "weighted_relation_loss": 0.0004148503619944677
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.579045837231057,
      "flow_matching_loss": 0.020499145798385143,
      "grad_norm": 0.09940534830093384,
      "learning_rate": 2.2981027282002154e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004279283282812685,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.971426010131836,
      "state_query_norm": 1441.5478149414062,
      "state_relation_loss": 0.0004279283282812685,
      "step": 20930,
      "student_distance_mean": 1974.6314453125,
      "teacher_distance_mean": 1.1673761844635009,
      "weighted_auxiliary_loss": 0.0004279283282812685,
      "weighted_relation_loss": 0.0004279283282812685
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.58840037418148,
      "flow_matching_loss": 0.020404800586402417,
      "grad_norm": 0.09283951669931412,
      "learning_rate": 2.2934668208010236e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043717956286855043,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.719059753417966,
      "state_query_norm": 1426.0059448242187,
      "state_relation_loss": 0.00043717956286855043,
      "step": 20940,
      "student_distance_mean": 1958.484375,
      "teacher_distance_mean": 1.1580948233604431,
      "weighted_auxiliary_loss": 0.00043717956286855043,
      "weighted_relation_loss": 0.00043717956286855043
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.5977549111319,
      "flow_matching_loss": 0.02167564295232296,
      "grad_norm": 0.09477055072784424,
      "learning_rate": 2.2888342020948555e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004272017133189365,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28567657470703,
      "state_query_norm": 1414.8721801757813,
      "state_relation_loss": 0.0004272017133189365,
      "step": 20950,
      "student_distance_mean": 1938.2021240234376,
      "teacher_distance_mean": 1.145753514766693,
      "weighted_auxiliary_loss": 0.0004272017133189365,
      "weighted_relation_loss": 0.0004272017133189365
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.60710944808232,
      "flow_matching_loss": 0.02017021840438247,
      "grad_norm": 0.09488248825073242,
      "learning_rate": 2.2842048777107783e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003931212850147858,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40334625244141,
      "state_query_norm": 1422.1455200195312,
      "state_relation_loss": 0.0003931212850147858,
      "step": 20960,
      "student_distance_mean": 1945.8593139648438,
      "teacher_distance_mean": 1.1465836405754088,
      "weighted_auxiliary_loss": 0.0003931212850147858,
      "weighted_relation_loss": 0.0003931212850147858
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.61646398503274,
      "flow_matching_loss": 0.019535733200609685,
      "grad_norm": 0.08078215271234512,
      "learning_rate": 2.2795788532738553e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041504394612275065,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57052764892578,
      "state_query_norm": 1429.3909790039063,
      "state_relation_loss": 0.00041504394612275065,
      "step": 20970,
      "student_distance_mean": 1955.51181640625,
      "teacher_distance_mean": 1.1524155259132385,
      "weighted_auxiliary_loss": 0.00041504394612275065,
      "weighted_relation_loss": 0.00041504394612275065
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.625818521983163,
      "flow_matching_loss": 0.02112702075392008,
      "grad_norm": 0.09697897732257843,
      "learning_rate": 2.2749561344051468e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040462442266289147,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9185432434082,
      "state_query_norm": 1433.4157836914062,
      "state_relation_loss": 0.00040462442266289147,
      "step": 20980,
      "student_distance_mean": 1968.22568359375,
      "teacher_distance_mean": 1.1622592687606812,
      "weighted_auxiliary_loss": 0.00040462442266289147,
      "weighted_relation_loss": 0.00040462442266289147
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.63517305893358,
      "flow_matching_loss": 0.02119158934801817,
      "grad_norm": 0.0991135686635971,
      "learning_rate": 2.2703367267216896e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042378554353490473,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.308563995361325,
      "state_query_norm": 1411.76005859375,
      "state_relation_loss": 0.00042378554353490473,
      "step": 20990,
      "student_distance_mean": 1937.1893798828125,
      "teacher_distance_mean": 1.1442638158798217,
      "weighted_auxiliary_loss": 0.00042378554353490473,
      "weighted_relation_loss": 0.00042378554353490473
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.644527595884004,
      "flow_matching_loss": 0.0201511749997735,
      "grad_norm": 0.0938463807106018,
      "learning_rate": 2.2657206358364997e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041453749872744083,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.312538146972656,
      "state_query_norm": 1410.7131469726562,
      "state_relation_loss": 0.00041453749872744083,
      "step": 21000,
      "student_distance_mean": 1938.9373168945312,
      "teacher_distance_mean": 1.1431682467460633,
      "weighted_auxiliary_loss": 0.00041453749872744083,
      "weighted_relation_loss": 0.00041453749872744083
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.653882132834426,
      "flow_matching_loss": 0.021287361346185207,
      "grad_norm": 0.11739172041416168,
      "learning_rate": 2.261107867358563e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041151965560857204,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.922781753540036,
      "state_query_norm": 1398.2951782226562,
      "state_relation_loss": 0.00041151965560857204,
      "step": 21010,
      "student_distance_mean": 1917.4811401367188,
      "teacher_distance_mean": 1.1303439378738402,
      "weighted_auxiliary_loss": 0.00041151965560857204,
      "weighted_relation_loss": 0.00041151965560857204
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.663236669784844,
      "flow_matching_loss": 0.020806404761970043,
      "grad_norm": 0.11707611382007599,
      "learning_rate": 2.2564984268928263e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004223514173645526,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.27179374694824,
      "state_query_norm": 1453.1285278320313,
      "state_relation_loss": 0.0004223514173645526,
      "step": 21020,
      "student_distance_mean": 1992.3879150390626,
      "teacher_distance_mean": 1.175363540649414,
      "weighted_auxiliary_loss": 0.0004223514173645526,
      "weighted_relation_loss": 0.0004223514173645526
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.672591206735266,
      "flow_matching_loss": 0.023312056064605714,
      "grad_norm": 0.10585108399391174,
      "learning_rate": 2.251892320040198e-05,
      "loss": 0.0237,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004193367291009054,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72747039794922,
      "state_query_norm": 1434.6779541015626,
      "state_relation_loss": 0.0004193367291009054,
      "step": 21030,
      "student_distance_mean": 1961.4270385742188,
      "teacher_distance_mean": 1.1557212471961975,
      "weighted_auxiliary_loss": 0.0004193367291009054,
      "weighted_relation_loss": 0.0004193367291009054
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.681945743685688,
      "flow_matching_loss": 0.021034293062984943,
      "grad_norm": 0.10832735896110535,
      "learning_rate": 2.2472895523975312e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041896110924426465,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.82142639160156,
      "state_query_norm": 1438.58642578125,
      "state_relation_loss": 0.00041896110924426465,
      "step": 21040,
      "student_distance_mean": 1971.2970703125,
      "teacher_distance_mean": 1.1585636854171752,
      "weighted_auxiliary_loss": 0.00041896110924426465,
      "weighted_relation_loss": 0.00041896110924426465
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.69130028063611,
      "flow_matching_loss": 0.02042006840929389,
      "grad_norm": 0.08585165441036224,
      "learning_rate": 2.2426901295576212e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042955917306244375,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.957230758666995,
      "state_query_norm": 1435.0173828125,
      "state_relation_loss": 0.00042955917306244375,
      "step": 21050,
      "student_distance_mean": 1972.9079956054688,
      "teacher_distance_mean": 1.1614521026611329,
      "weighted_auxiliary_loss": 0.00042955917306244375,
      "weighted_relation_loss": 0.00042955917306244375
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.70065481758653,
      "flow_matching_loss": 0.02047356739640236,
      "grad_norm": 0.09439396113157272,
      "learning_rate": 2.2380940571092013e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042512712243478743,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41506996154785,
      "state_query_norm": 1427.6300537109375,
      "state_relation_loss": 0.00042512712243478743,
      "step": 21060,
      "student_distance_mean": 1948.870556640625,
      "teacher_distance_mean": 1.1450516343116761,
      "weighted_auxiliary_loss": 0.00042512712243478743,
      "weighted_relation_loss": 0.00042512712243478743
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.71000935453695,
      "flow_matching_loss": 0.022347495518624784,
      "grad_norm": 0.08882807195186615,
      "learning_rate": 2.2335013406369302e-05,
      "loss": 0.0228,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042013979400508107,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9863998413086,
      "state_query_norm": 1444.532275390625,
      "state_relation_loss": 0.00042013979400508107,
      "step": 21070,
      "student_distance_mean": 1978.9910766601563,
      "teacher_distance_mean": 1.1640485882759095,
      "weighted_auxiliary_loss": 0.00042013979400508107,
      "weighted_relation_loss": 0.00042013979400508107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.719363891487372,
      "flow_matching_loss": 0.020828415267169477,
      "grad_norm": 0.08607202768325806,
      "learning_rate": 2.228911985721397e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041246365290135143,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.81530227661133,
      "state_query_norm": 1441.5007446289062,
      "state_relation_loss": 0.00041246365290135143,
      "step": 21080,
      "student_distance_mean": 1970.0235961914063,
      "teacher_distance_mean": 1.1598050117492675,
      "weighted_auxiliary_loss": 0.00041246365290135143,
      "weighted_relation_loss": 0.00041246365290135143
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.72871842843779,
      "flow_matching_loss": 0.019750831834971904,
      "grad_norm": 0.09852144122123718,
      "learning_rate": 2.224325997939095e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041928946448024364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76503448486328,
      "state_query_norm": 1428.2433227539063,
      "state_relation_loss": 0.00041928946448024364,
      "step": 21090,
      "student_distance_mean": 1965.9008422851562,
      "teacher_distance_mean": 1.157501721382141,
      "weighted_auxiliary_loss": 0.00041928946448024364,
      "weighted_relation_loss": 0.00041928946448024364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.738072965388213,
      "flow_matching_loss": 0.021764455176889897,
      "grad_norm": 0.10084209591150284,
      "learning_rate": 2.219743382862437e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040193303721025586,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75259857177734,
      "state_query_norm": 1435.6905883789063,
      "state_relation_loss": 0.00040193303721025586,
      "step": 21100,
      "student_distance_mean": 1965.3524169921875,
      "teacher_distance_mean": 1.1581291437149048,
      "weighted_auxiliary_loss": 0.00040193303721025586,
      "weighted_relation_loss": 0.00040193303721025586
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.747427502338635,
      "flow_matching_loss": 0.020364532992243767,
      "grad_norm": 0.0949479267001152,
      "learning_rate": 2.2151641460597294e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041068673599511387,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71767120361328,
      "state_query_norm": 1439.219482421875,
      "state_relation_loss": 0.00041068673599511387,
      "step": 21110,
      "student_distance_mean": 1964.393212890625,
      "teacher_distance_mean": 1.1594526052474976,
      "weighted_auxiliary_loss": 0.00041068673599511387,
      "weighted_relation_loss": 0.00041068673599511387
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.756782039289057,
      "flow_matching_loss": 0.018247134890407322,
      "grad_norm": 0.10024049878120422,
      "learning_rate": 2.210588293095177e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004110617970582098,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23545379638672,
      "state_query_norm": 1412.074853515625,
      "state_relation_loss": 0.0004110617970582098,
      "step": 21120,
      "student_distance_mean": 1937.6706420898438,
      "teacher_distance_mean": 1.1387727856636047,
      "weighted_auxiliary_loss": 0.0004110617970582098,
      "weighted_relation_loss": 0.0004110617970582098
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.766136576239475,
      "flow_matching_loss": 0.02148104105144739,
      "grad_norm": 0.11017029732465744,
      "learning_rate": 2.2060158295288714e-05,
      "loss": 0.0219,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004202480224194005,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.553165435791016,
      "state_query_norm": 1421.5777465820313,
      "state_relation_loss": 0.0004202480224194005,
      "step": 21130,
      "student_distance_mean": 1953.5626098632813,
      "teacher_distance_mean": 1.1503183960914611,
      "weighted_auxiliary_loss": 0.0004202480224194005,
      "weighted_relation_loss": 0.0004202480224194005
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.775491113189897,
      "flow_matching_loss": 0.020891869626939296,
      "grad_norm": 0.09105567634105682,
      "learning_rate": 2.2014467609167905e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004288921132683754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.413926696777345,
      "state_query_norm": 1422.9772827148438,
      "state_relation_loss": 0.0004288921132683754,
      "step": 21140,
      "student_distance_mean": 1947.3532592773438,
      "teacher_distance_mean": 1.1478655099868775,
      "weighted_auxiliary_loss": 0.0004288921132683754,
      "weighted_relation_loss": 0.0004288921132683754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.78484565014032,
      "flow_matching_loss": 0.02115653995424509,
      "grad_norm": 0.11912190914154053,
      "learning_rate": 2.1968810928107806e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004356877150712535,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.593288040161134,
      "state_query_norm": 1429.562939453125,
      "state_relation_loss": 0.0004356877150712535,
      "step": 21150,
      "student_distance_mean": 1952.6751342773437,
      "teacher_distance_mean": 1.1509549736976623,
      "weighted_auxiliary_loss": 0.0004356877150712535,
      "weighted_relation_loss": 0.0004356877150712535
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.794200187090738,
      "flow_matching_loss": 0.01970757283270359,
      "grad_norm": 0.09531106054782867,
      "learning_rate": 2.1923188307585606e-05,
      "loss": 0.0201,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004300985630834475,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78310432434082,
      "state_query_norm": 1442.1567260742188,
      "state_relation_loss": 0.0004300985630834475,
      "step": 21160,
      "student_distance_mean": 1967.8532104492188,
      "teacher_distance_mean": 1.1577593207359314,
      "weighted_auxiliary_loss": 0.0004300985630834475,
      "weighted_relation_loss": 0.0004300985630834475
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.80355472404116,
      "flow_matching_loss": 0.02076269928365946,
      "grad_norm": 0.10737159848213196,
      "learning_rate": 2.187759980303708e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043024574406445024,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93507461547851,
      "state_query_norm": 1440.0280029296875,
      "state_relation_loss": 0.00043024574406445024,
      "step": 21170,
      "student_distance_mean": 1976.6678344726563,
      "teacher_distance_mean": 1.1606349229812623,
      "weighted_auxiliary_loss": 0.00043024574406445024,
      "weighted_relation_loss": 0.00043024574406445024
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.81290926099158,
      "flow_matching_loss": 0.021804564259946348,
      "grad_norm": 0.11070258170366287,
      "learning_rate": 2.1832045469856545e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004263587383320555,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33599624633789,
      "state_query_norm": 1417.1385131835937,
      "state_relation_loss": 0.0004263587383320555,
      "step": 21180,
      "student_distance_mean": 1941.23271484375,
      "teacher_distance_mean": 1.1408563137054444,
      "weighted_auxiliary_loss": 0.0004263587383320555,
      "weighted_relation_loss": 0.0004263587383320555
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.822263797942004,
      "flow_matching_loss": 0.020205684937536716,
      "grad_norm": 0.11217175424098969,
      "learning_rate": 2.1786525363396844e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004081970022525638,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38504981994629,
      "state_query_norm": 1423.11865234375,
      "state_relation_loss": 0.0004081970022525638,
      "step": 21190,
      "student_distance_mean": 1945.6221069335938,
      "teacher_distance_mean": 1.1450764894485475,
      "weighted_auxiliary_loss": 0.0004081970022525638,
      "weighted_relation_loss": 0.0004081970022525638
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.831618334892422,
      "flow_matching_loss": 0.021509956382215024,
      "grad_norm": 0.10385577380657196,
      "learning_rate": 2.1741039538969183e-05,
      "loss": 0.0219,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042821805109269915,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.604125595092775,
      "state_query_norm": 1429.2474975585938,
      "state_relation_loss": 0.00042821805109269915,
      "step": 21200,
      "student_distance_mean": 1955.8207763671876,
      "teacher_distance_mean": 1.1489396572113038,
      "weighted_auxiliary_loss": 0.00042821805109269915,
      "weighted_relation_loss": 0.00042821805109269915
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.840972871842844,
      "flow_matching_loss": 0.021655003167688848,
      "grad_norm": 0.10890497267246246,
      "learning_rate": 2.169558805184313e-05,
      "loss": 0.0221,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000406031621969305,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23360252380371,
      "state_query_norm": 1415.581640625,
      "state_relation_loss": 0.000406031621969305,
      "step": 21210,
      "student_distance_mean": 1937.0408203125,
      "teacher_distance_mean": 1.1374158024787904,
      "weighted_auxiliary_loss": 0.000406031621969305,
      "weighted_relation_loss": 0.000406031621969305
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.850327408793266,
      "flow_matching_loss": 0.02196764890104532,
      "grad_norm": 0.09517153352499008,
      "learning_rate": 2.165017095724651e-05,
      "loss": 0.0224,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004527453653281555,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54690322875977,
      "state_query_norm": 1426.6788940429688,
      "state_relation_loss": 0.0004527453653281555,
      "step": 21220,
      "student_distance_mean": 1953.42275390625,
      "teacher_distance_mean": 1.149513554573059,
      "weighted_auxiliary_loss": 0.0004527453653281555,
      "weighted_relation_loss": 0.0004527453653281555
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.859681945743684,
      "flow_matching_loss": 0.019994460977613926,
      "grad_norm": 0.0838131457567215,
      "learning_rate": 2.1604788310365405e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043382116127759217,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70509300231934,
      "state_query_norm": 1430.3559326171876,
      "state_relation_loss": 0.00043382116127759217,
      "step": 21230,
      "student_distance_mean": 1961.1233032226562,
      "teacher_distance_mean": 1.1531928420066833,
      "weighted_auxiliary_loss": 0.00043382116127759217,
      "weighted_relation_loss": 0.00043382116127759217
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.869036482694106,
      "flow_matching_loss": 0.019912222772836684,
      "grad_norm": 0.08407162129878998,
      "learning_rate": 2.155944016634398e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000404081103624776,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6023136138916,
      "state_query_norm": 1424.6737670898438,
      "state_relation_loss": 0.000404081103624776,
      "step": 21240,
      "student_distance_mean": 1956.0955688476563,
      "teacher_distance_mean": 1.150467872619629,
      "weighted_auxiliary_loss": 0.000404081103624776,
      "weighted_relation_loss": 0.000404081103624776
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.87839101964453,
      "flow_matching_loss": 0.01920459158718586,
      "grad_norm": 0.10473321378231049,
      "learning_rate": 2.1514126580284538e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044263603922445327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38394889831543,
      "state_query_norm": 1416.1908081054687,
      "state_relation_loss": 0.00044263603922445327,
      "step": 21250,
      "student_distance_mean": 1943.1766235351563,
      "teacher_distance_mean": 1.1420790314674378,
      "weighted_auxiliary_loss": 0.00044263603922445327,
      "weighted_relation_loss": 0.00044263603922445327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.88774555659495,
      "flow_matching_loss": 0.01931488011032343,
      "grad_norm": 0.09308557957410812,
      "learning_rate": 2.1468847607247344e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040445870836265386,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27773323059082,
      "state_query_norm": 1418.8029663085938,
      "state_relation_loss": 0.00040445870836265386,
      "step": 21260,
      "student_distance_mean": 1944.32978515625,
      "teacher_distance_mean": 1.1427452445030213,
      "weighted_auxiliary_loss": 0.00040445870836265386,
      "weighted_relation_loss": 0.00040445870836265386
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.89710009354537,
      "flow_matching_loss": 0.02019874583929777,
      "grad_norm": 0.08330627530813217,
      "learning_rate": 2.1423603302250623e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004334018361987546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.305485534667966,
      "state_query_norm": 1416.783544921875,
      "state_relation_loss": 0.0004334018361987546,
      "step": 21270,
      "student_distance_mean": 1943.06552734375,
      "teacher_distance_mean": 1.1451091289520263,
      "weighted_auxiliary_loss": 0.0004334018361987546,
      "weighted_relation_loss": 0.0004334018361987546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.90645463049579,
      "flow_matching_loss": 0.021980502270162107,
      "grad_norm": 0.10325576364994049,
      "learning_rate": 2.1378393720270468e-05,
      "loss": 0.0224,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004279034212231636,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89740142822266,
      "state_query_norm": 1435.0923095703124,
      "state_relation_loss": 0.0004279034212231636,
      "step": 21280,
      "student_distance_mean": 1970.8606689453125,
      "teacher_distance_mean": 1.1591018557548523,
      "weighted_auxiliary_loss": 0.0004279034212231636,
      "weighted_relation_loss": 0.0004279034212231636
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.915809167446213,
      "flow_matching_loss": 0.02118203230202198,
      "grad_norm": 0.09457907825708389,
      "learning_rate": 2.133321891624076e-05,
      "loss": 0.0216,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004301780427340418,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.34913330078125,
      "state_query_norm": 1426.3615478515626,
      "state_relation_loss": 0.0004301780427340418,
      "step": 21290,
      "student_distance_mean": 1945.28505859375,
      "teacher_distance_mean": 1.1449546337127685,
      "weighted_auxiliary_loss": 0.0004301780427340418,
      "weighted_relation_loss": 0.0004301780427340418
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.92516370439663,
      "flow_matching_loss": 0.020010099560022355,
      "grad_norm": 0.09575878828763962,
      "learning_rate": 2.1288078945053193e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004232210572808981,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.223331451416016,
      "state_query_norm": 1413.0984130859374,
      "state_relation_loss": 0.0004232210572808981,
      "step": 21300,
      "student_distance_mean": 1936.0202026367188,
      "teacher_distance_mean": 1.139652669429779,
      "weighted_auxiliary_loss": 0.0004232210572808981,
      "weighted_relation_loss": 0.0004232210572808981
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.934518241347053,
      "flow_matching_loss": 0.019248932786285877,
      "grad_norm": 0.0893864557147026,
      "learning_rate": 2.1242973861557064e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041865036473609507,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62695426940918,
      "state_query_norm": 1426.2035888671876,
      "state_relation_loss": 0.00041865036473609507,
      "step": 21310,
      "student_distance_mean": 1954.25205078125,
      "teacher_distance_mean": 1.149118685722351,
      "weighted_auxiliary_loss": 0.00041865036473609507,
      "weighted_relation_loss": 0.00041865036473609507
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.943872778297475,
      "flow_matching_loss": 0.01944501670077443,
      "grad_norm": 0.10237549245357513,
      "learning_rate": 2.1197903720559304e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004193066793959588,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.275217056274414,
      "state_query_norm": 1416.0207397460938,
      "state_relation_loss": 0.0004193066793959588,
      "step": 21320,
      "student_distance_mean": 1938.6699340820312,
      "teacher_distance_mean": 1.1411813020706176,
      "weighted_auxiliary_loss": 0.0004193066793959588,
      "weighted_relation_loss": 0.0004193066793959588
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.953227315247894,
      "flow_matching_loss": 0.02067727902904153,
      "grad_norm": 0.10261514782905579,
      "learning_rate": 2.115286857682438e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004376913100713864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.80188980102539,
      "state_query_norm": 1439.0378173828126,
      "state_relation_loss": 0.0004376913100713864,
      "step": 21330,
      "student_distance_mean": 1967.0817749023438,
      "teacher_distance_mean": 1.1570326328277587,
      "weighted_auxiliary_loss": 0.0004376913100713864,
      "weighted_relation_loss": 0.0004376913100713864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.962581852198316,
      "flow_matching_loss": 0.021611707285046577,
      "grad_norm": 0.1067795604467392,
      "learning_rate": 2.1107868485074228e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000388705130899325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.206520080566406,
      "state_query_norm": 1411.6401733398438,
      "state_relation_loss": 0.000388705130899325,
      "step": 21340,
      "student_distance_mean": 1934.875244140625,
      "teacher_distance_mean": 1.1383009433746338,
      "weighted_auxiliary_loss": 0.000388705130899325,
      "weighted_relation_loss": 0.000388705130899325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.971936389148738,
      "flow_matching_loss": 0.022291818261146547,
      "grad_norm": 0.1044643372297287,
      "learning_rate": 2.1062903499988234e-05,
      "loss": 0.0227,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004108169116079807,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.780587387084964,
      "state_query_norm": 1420.0578369140626,
      "state_relation_loss": 0.0004108169116079807,
      "step": 21350,
      "student_distance_mean": 1960.0463256835938,
      "teacher_distance_mean": 1.152437114715576,
      "weighted_auxiliary_loss": 0.0004108169116079807,
      "weighted_relation_loss": 0.0004108169116079807
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.98129092609916,
      "flow_matching_loss": 0.019428374152630566,
      "grad_norm": 0.08668190240859985,
      "learning_rate": 2.101797367620308e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041259240242652595,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59553909301758,
      "state_query_norm": 1431.0984375,
      "state_relation_loss": 0.00041259240242652595,
      "step": 21360,
      "student_distance_mean": 1961.474853515625,
      "teacher_distance_mean": 1.1542434573173523,
      "weighted_auxiliary_loss": 0.00041259240242652595,
      "weighted_relation_loss": 0.00041259240242652595
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 19.990645463049578,
      "flow_matching_loss": 0.020301276817917824,
      "grad_norm": 0.08366125822067261,
      "learning_rate": 2.0973079068312713e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041627456666901705,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.865400314331055,
      "state_query_norm": 1440.5025390625,
      "state_relation_loss": 0.00041627456666901705,
      "step": 21370,
      "student_distance_mean": 1973.905078125,
      "teacher_distance_mean": 1.1614471912384032,
      "weighted_auxiliary_loss": 0.00041627456666901705,
      "weighted_relation_loss": 0.00041627456666901705
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.0,
      "flow_matching_loss": 0.02181390207260847,
      "grad_norm": 0.18757344782352448,
      "learning_rate": 2.0928219730868358e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0004125225153984502,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.34740715026855,
      "state_query_norm": 1419.77578125,
      "state_relation_loss": 0.0004125225153984502,
      "step": 21380,
      "student_distance_mean": 1947.3506958007813,
      "teacher_distance_mean": 1.1482684135437011,
      "weighted_auxiliary_loss": 0.0004125225153984502,
      "weighted_relation_loss": 0.0004125225153984502
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.009354536950422,
      "flow_matching_loss": 0.0213278129696846,
      "grad_norm": 0.09877947717905045,
      "learning_rate": 2.0883395718378302e-05,
      "loss": 0.0218,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004551370075205341,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.080755615234374,
      "state_query_norm": 1444.5507690429688,
      "state_relation_loss": 0.0004551370075205341,
      "step": 21390,
      "student_distance_mean": 1981.2475463867188,
      "teacher_distance_mean": 1.1668720841407776,
      "weighted_auxiliary_loss": 0.0004551370075205341,
      "weighted_relation_loss": 0.0004551370075205341
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.01870907390084,
      "flow_matching_loss": 0.01940121641382575,
      "grad_norm": 0.09652135521173477,
      "learning_rate": 2.083860708530798e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044152057089377197,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.20934143066406,
      "state_query_norm": 1441.4175415039062,
      "state_relation_loss": 0.00044152057089377197,
      "step": 21400,
      "student_distance_mean": 1984.6005126953125,
      "teacher_distance_mean": 1.167410945892334,
      "weighted_auxiliary_loss": 0.00044152057089377197,
      "weighted_relation_loss": 0.00044152057089377197
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.028063610851262,
      "flow_matching_loss": 0.021295149996876716,
      "grad_norm": 0.11145325005054474,
      "learning_rate": 2.0793853886079796e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004253608960425481,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.86016654968262,
      "state_query_norm": 1435.9421752929688,
      "state_relation_loss": 0.0004253608960425481,
      "step": 21410,
      "student_distance_mean": 1969.4416015625,
      "teacher_distance_mean": 1.1592958569526672,
      "weighted_auxiliary_loss": 0.0004253608960425481,
      "weighted_relation_loss": 0.0004253608960425481
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.037418147801684,
      "flow_matching_loss": 0.01888517215847969,
      "grad_norm": 0.08769909292459488,
      "learning_rate": 2.074913617507309e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004459662188310176,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.82967071533203,
      "state_query_norm": 1430.9106689453124,
      "state_relation_loss": 0.0004459662188310176,
      "step": 21420,
      "student_distance_mean": 1965.95986328125,
      "teacher_distance_mean": 1.1567677140235901,
      "weighted_auxiliary_loss": 0.0004459662188310176,
      "weighted_relation_loss": 0.0004459662188310176
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.046772684752106,
      "flow_matching_loss": 0.0210593368858099,
      "grad_norm": 0.09075486660003662,
      "learning_rate": 2.070445400662411e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039366064011119305,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.714233779907225,
      "state_query_norm": 1425.8648193359375,
      "state_relation_loss": 0.00039366064011119305,
      "step": 21430,
      "student_distance_mean": 1960.7337036132812,
      "teacher_distance_mean": 1.1524110674858092,
      "weighted_auxiliary_loss": 0.00039366064011119305,
      "weighted_relation_loss": 0.00039366064011119305
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.056127221702525,
      "flow_matching_loss": 0.019658203423023223,
      "grad_norm": 0.08772553503513336,
      "learning_rate": 2.0659807435025907e-05,
      "loss": 0.0201,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042924100998789074,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.82424430847168,
      "state_query_norm": 1439.9062255859376,
      "state_relation_loss": 0.00042924100998789074,
      "step": 21440,
      "student_distance_mean": 1970.5307739257812,
      "teacher_distance_mean": 1.156968927383423,
      "weighted_auxiliary_loss": 0.00042924100998789074,
      "weighted_relation_loss": 0.00042924100998789074
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.065481758652947,
      "flow_matching_loss": 0.01898327236995101,
      "grad_norm": 0.08874478936195374,
      "learning_rate": 2.061519651452825e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003981955029303208,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10103759765625,
      "state_query_norm": 1407.7173950195313,
      "state_relation_loss": 0.0003981955029303208,
      "step": 21450,
      "student_distance_mean": 1930.0045166015625,
      "teacher_distance_mean": 1.1342428088188172,
      "weighted_auxiliary_loss": 0.0003981955029303208,
      "weighted_relation_loss": 0.0003981955029303208
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.07483629560337,
      "flow_matching_loss": 0.01979834893718362,
      "grad_norm": 0.08488988876342773,
      "learning_rate": 2.0570621299337656e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004190764535451308,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.04127998352051,
      "state_query_norm": 1437.980322265625,
      "state_relation_loss": 0.0004190764535451308,
      "step": 21460,
      "student_distance_mean": 1977.9606811523438,
      "teacher_distance_mean": 1.1634943962097168,
      "weighted_auxiliary_loss": 0.0004190764535451308,
      "weighted_relation_loss": 0.0004190764535451308
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.084190832553787,
      "flow_matching_loss": 0.020987721905112267,
      "grad_norm": 0.09279318898916245,
      "learning_rate": 2.0526081843617183e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042499966220930217,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56123046875,
      "state_query_norm": 1426.2958740234376,
      "state_relation_loss": 0.00042499966220930217,
      "step": 21470,
      "student_distance_mean": 1957.5225708007813,
      "teacher_distance_mean": 1.1517478466033935,
      "weighted_auxiliary_loss": 0.00042499966220930217,
      "weighted_relation_loss": 0.00042499966220930217
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.09354536950421,
      "flow_matching_loss": 0.020800324622541666,
      "grad_norm": 0.06940946727991104,
      "learning_rate": 2.0481578201486485e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040803807496558875,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63227729797363,
      "state_query_norm": 1426.4311157226562,
      "state_relation_loss": 0.00040803807496558875,
      "step": 21480,
      "student_distance_mean": 1956.1270263671875,
      "teacher_distance_mean": 1.1497734427452087,
      "weighted_auxiliary_loss": 0.00040803807496558875,
      "weighted_relation_loss": 0.00040803807496558875
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.10289990645463,
      "flow_matching_loss": 0.018571352120488882,
      "grad_norm": 0.08882363140583038,
      "learning_rate": 2.0437110427021677e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040120030171237887,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.923173141479495,
      "state_query_norm": 1446.6167358398438,
      "state_relation_loss": 0.00040120030171237887,
      "step": 21490,
      "student_distance_mean": 1977.8372924804687,
      "teacher_distance_mean": 1.1620641589164733,
      "weighted_auxiliary_loss": 0.00040120030171237887,
      "weighted_relation_loss": 0.00040120030171237887
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.112254443405053,
      "flow_matching_loss": 0.018921063281595707,
      "grad_norm": 0.08191407471895218,
      "learning_rate": 2.0392678574255282e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044326119823381307,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.708380126953124,
      "state_query_norm": 1420.594970703125,
      "state_relation_loss": 0.00044326119823381307,
      "step": 21500,
      "student_distance_mean": 1955.9772705078126,
      "teacher_distance_mean": 1.148896288871765,
      "weighted_auxiliary_loss": 0.00044326119823381307,
      "weighted_relation_loss": 0.00044326119823381307
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.12160898035547,
      "flow_matching_loss": 0.02041557803750038,
      "grad_norm": 0.09077673405408859,
      "learning_rate": 2.0348282697176217e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042532056977506727,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.868350982666016,
      "state_query_norm": 1440.4233032226562,
      "state_relation_loss": 0.00042532056977506727,
      "step": 21510,
      "student_distance_mean": 1970.886474609375,
      "teacher_distance_mean": 1.1568387866020202,
      "weighted_auxiliary_loss": 0.00042532056977506727,
      "weighted_relation_loss": 0.00042532056977506727
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.130963517305894,
      "flow_matching_loss": 0.01995114255696535,
      "grad_norm": 0.08861702680587769,
      "learning_rate": 2.030392284972964e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004174046480329707,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.570452499389646,
      "state_query_norm": 1433.6854614257813,
      "state_relation_loss": 0.0004174046480329707,
      "step": 21520,
      "student_distance_mean": 1957.3817138671875,
      "teacher_distance_mean": 1.150858724117279,
      "weighted_auxiliary_loss": 0.0004174046480329707,
      "weighted_relation_loss": 0.0004174046480329707
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.140318054256316,
      "flow_matching_loss": 0.019694506004452706,
      "grad_norm": 0.09424436092376709,
      "learning_rate": 2.0259599085816973e-05,
      "loss": 0.0201,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004099067795323208,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.86478614807129,
      "state_query_norm": 1440.9202270507812,
      "state_relation_loss": 0.0004099067795323208,
      "step": 21530,
      "student_distance_mean": 1969.82353515625,
      "teacher_distance_mean": 1.1579963088035583,
      "weighted_auxiliary_loss": 0.0004099067795323208,
      "weighted_relation_loss": 0.0004099067795323208
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.149672591206734,
      "flow_matching_loss": 0.020116546563804148,
      "grad_norm": 0.08562038838863373,
      "learning_rate": 2.0215311459295755e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040844630275387316,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69969444274902,
      "state_query_norm": 1428.935009765625,
      "state_relation_loss": 0.00040844630275387316,
      "step": 21540,
      "student_distance_mean": 1961.544775390625,
      "teacher_distance_mean": 1.1545437812805175,
      "weighted_auxiliary_loss": 0.00040844630275387316,
      "weighted_relation_loss": 0.00040844630275387316
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.159027128157156,
      "flow_matching_loss": 0.021532247588038445,
      "grad_norm": 0.09383109956979752,
      "learning_rate": 2.0171060023979606e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045676703157369045,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.20527114868164,
      "state_query_norm": 1446.5469482421875,
      "state_relation_loss": 0.00045676703157369045,
      "step": 21550,
      "student_distance_mean": 1986.7755493164063,
      "teacher_distance_mean": 1.169465970993042,
      "weighted_auxiliary_loss": 0.00045676703157369045,
      "weighted_relation_loss": 0.00045676703157369045
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.168381665107578,
      "flow_matching_loss": 0.02004684116691351,
      "grad_norm": 0.11176791787147522,
      "learning_rate": 2.012684483363823e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000422781246015802,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.161530303955075,
      "state_query_norm": 1413.8322387695312,
      "state_relation_loss": 0.000422781246015802,
      "step": 21560,
      "student_distance_mean": 1934.6289428710938,
      "teacher_distance_mean": 1.1391904592514037,
      "weighted_auxiliary_loss": 0.000422781246015802,
      "weighted_relation_loss": 0.000422781246015802
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.177736202057996,
      "flow_matching_loss": 0.018768921587616207,
      "grad_norm": 0.09964433312416077,
      "learning_rate": 2.0082665941997237e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043494971178006383,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.736290740966794,
      "state_query_norm": 1429.7734985351562,
      "state_relation_loss": 0.00043494971178006383,
      "step": 21570,
      "student_distance_mean": 1960.051171875,
      "teacher_distance_mean": 1.1567259550094604,
      "weighted_auxiliary_loss": 0.00043494971178006383,
      "weighted_relation_loss": 0.00043494971178006383
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.18709073900842,
      "flow_matching_loss": 0.01960594207048416,
      "grad_norm": 0.102793388068676,
      "learning_rate": 2.0038523402738146e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004202620650175959,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60685997009277,
      "state_query_norm": 1436.0532348632812,
      "state_relation_loss": 0.0004202620650175959,
      "step": 21580,
      "student_distance_mean": 1959.527294921875,
      "teacher_distance_mean": 1.156237828731537,
      "weighted_auxiliary_loss": 0.0004202620650175959,
      "weighted_relation_loss": 0.0004202620650175959
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.19644527595884,
      "flow_matching_loss": 0.019620682671666147,
      "grad_norm": 0.10749094933271408,
      "learning_rate": 1.99944172694983e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004115490039112046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69649772644043,
      "state_query_norm": 1427.1650512695312,
      "state_relation_loss": 0.0004115490039112046,
      "step": 21590,
      "student_distance_mean": 1959.9822509765625,
      "teacher_distance_mean": 1.1548641562461852,
      "weighted_auxiliary_loss": 0.0004115490039112046,
      "weighted_relation_loss": 0.0004115490039112046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.205799812909262,
      "flow_matching_loss": 0.02077921889722347,
      "grad_norm": 0.08598215132951736,
      "learning_rate": 1.99503475958708e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004188837134279311,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42659339904785,
      "state_query_norm": 1419.2034912109375,
      "state_relation_loss": 0.0004188837134279311,
      "step": 21600,
      "student_distance_mean": 1948.2902099609375,
      "teacher_distance_mean": 1.1486634373664857,
      "weighted_auxiliary_loss": 0.0004188837134279311,
      "weighted_relation_loss": 0.0004188837134279311
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.21515434985968,
      "flow_matching_loss": 0.019607076048851015,
      "grad_norm": 0.08526260405778885,
      "learning_rate": 1.9906314435404484e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038845803937874733,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.681585693359374,
      "state_query_norm": 1430.48203125,
      "state_relation_loss": 0.00038845803937874733,
      "step": 21610,
      "student_distance_mean": 1963.3773559570313,
      "teacher_distance_mean": 1.1568202137947083,
      "weighted_auxiliary_loss": 0.00038845803937874733,
      "weighted_relation_loss": 0.00038845803937874733
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.224508886810103,
      "flow_matching_loss": 0.019495511427521706,
      "grad_norm": 0.07291775196790695,
      "learning_rate": 1.986231784160378e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039324980753008276,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.94655914306641,
      "state_query_norm": 1407.2851928710938,
      "state_relation_loss": 0.00039324980753008276,
      "step": 21620,
      "student_distance_mean": 1923.8178100585938,
      "teacher_distance_mean": 1.1330351829528809,
      "weighted_auxiliary_loss": 0.00039324980753008276,
      "weighted_relation_loss": 0.00039324980753008276
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.233863423760525,
      "flow_matching_loss": 0.02050482966005802,
      "grad_norm": 0.09792355448007584,
      "learning_rate": 1.9818357867928693e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004036239697597921,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4237678527832,
      "state_query_norm": 1418.3863159179687,
      "state_relation_loss": 0.0004036239697597921,
      "step": 21630,
      "student_distance_mean": 1945.7267700195312,
      "teacher_distance_mean": 1.1476650357246398,
      "weighted_auxiliary_loss": 0.0004036239697597921,
      "weighted_relation_loss": 0.0004036239697597921
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.243217960710943,
      "flow_matching_loss": 0.021620112098753454,
      "grad_norm": 0.08671607077121735,
      "learning_rate": 1.977443456779474e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040734688518568874,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25709571838379,
      "state_query_norm": 1410.7541625976562,
      "state_relation_loss": 0.00040734688518568874,
      "step": 21640,
      "student_distance_mean": 1936.1116333007812,
      "teacher_distance_mean": 1.1413050055503846,
      "weighted_auxiliary_loss": 0.00040734688518568874,
      "weighted_relation_loss": 0.00040734688518568874
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.252572497661365,
      "flow_matching_loss": 0.021008280757814645,
      "grad_norm": 0.09576963633298874,
      "learning_rate": 1.9730547994572858e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004250996484188363,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.38014030456543,
      "state_query_norm": 1456.7916381835937,
      "state_relation_loss": 0.0004250996484188363,
      "step": 21650,
      "student_distance_mean": 1998.4711791992188,
      "teacher_distance_mean": 1.1780013203620912,
      "weighted_auxiliary_loss": 0.0004250996484188363,
      "weighted_relation_loss": 0.0004250996484188363
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.261927034611787,
      "flow_matching_loss": 0.020236326195299626,
      "grad_norm": 0.10272816568613052,
      "learning_rate": 1.9686698201589393e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042265398951712996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.04587326049805,
      "state_query_norm": 1448.3796020507812,
      "state_relation_loss": 0.00042265398951712996,
      "step": 21660,
      "student_distance_mean": 1981.5884399414062,
      "teacher_distance_mean": 1.165529441833496,
      "weighted_auxiliary_loss": 0.00042265398951712996,
      "weighted_relation_loss": 0.00042265398951712996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.27128157156221,
      "flow_matching_loss": 0.021272911317646505,
      "grad_norm": 0.0912717878818512,
      "learning_rate": 1.9642885242125964e-05,
      "loss": 0.0217,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004220357455778867,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.88357696533203,
      "state_query_norm": 1442.1979858398438,
      "state_relation_loss": 0.0004220357455778867,
      "step": 21670,
      "student_distance_mean": 1972.7124389648438,
      "teacher_distance_mean": 1.159134316444397,
      "weighted_auxiliary_loss": 0.0004220357455778867,
      "weighted_relation_loss": 0.0004220357455778867
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.280636108512628,
      "flow_matching_loss": 0.019778078608214855,
      "grad_norm": 0.07853221148252487,
      "learning_rate": 1.9599109169419466e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003994964063167572,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47224502563476,
      "state_query_norm": 1426.0809692382813,
      "state_relation_loss": 0.0003994964063167572,
      "step": 21680,
      "student_distance_mean": 1948.3956176757813,
      "teacher_distance_mean": 1.146430492401123,
      "weighted_auxiliary_loss": 0.0003994964063167572,
      "weighted_relation_loss": 0.0003994964063167572
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.28999064546305,
      "flow_matching_loss": 0.019912870600819588,
      "grad_norm": 0.08979696780443192,
      "learning_rate": 1.9555370036661948e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041367335361428557,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.365128707885745,
      "state_query_norm": 1412.0678466796876,
      "state_relation_loss": 0.00041367335361428557,
      "step": 21690,
      "student_distance_mean": 1944.9433349609376,
      "teacher_distance_mean": 1.1439485549926758,
      "weighted_auxiliary_loss": 0.00041367335361428557,
      "weighted_relation_loss": 0.00041367335361428557
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.29934518241347,
      "flow_matching_loss": 0.019400020129978657,
      "grad_norm": 0.09925320744514465,
      "learning_rate": 1.9511667897000574e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043784986191894857,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43740577697754,
      "state_query_norm": 1420.9582397460938,
      "state_relation_loss": 0.00043784986191894857,
      "step": 21700,
      "student_distance_mean": 1947.9462646484376,
      "teacher_distance_mean": 1.1468451976776124,
      "weighted_auxiliary_loss": 0.00043784986191894857,
      "weighted_relation_loss": 0.00043784986191894857
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.30869971936389,
      "flow_matching_loss": 0.019975552055984735,
      "grad_norm": 0.10206180810928345,
      "learning_rate": 1.946800280353755e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004254706436768174,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4373592376709,
      "state_query_norm": 1419.1535888671874,
      "state_relation_loss": 0.0004254706436768174,
      "step": 21710,
      "student_distance_mean": 1944.3936645507813,
      "teacher_distance_mean": 1.1444799542427062,
      "weighted_auxiliary_loss": 0.0004254706436768174,
      "weighted_relation_loss": 0.0004254706436768174
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.318054256314312,
      "flow_matching_loss": 0.018591862916946412,
      "grad_norm": 0.09236975759267807,
      "learning_rate": 1.9424374809330114e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039965841569937765,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9306941986084,
      "state_query_norm": 1440.2152099609375,
      "state_relation_loss": 0.00039965841569937765,
      "step": 21720,
      "student_distance_mean": 1974.6268310546875,
      "teacher_distance_mean": 1.1605141878128051,
      "weighted_auxiliary_loss": 0.00039965841569937765,
      "weighted_relation_loss": 0.00039965841569937765
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.327408793264734,
      "flow_matching_loss": 0.020451825857162476,
      "grad_norm": 0.09130612015724182,
      "learning_rate": 1.9380783967390382e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041848404798656703,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.467146682739255,
      "state_query_norm": 1421.4778686523437,
      "state_relation_loss": 0.00041848404798656703,
      "step": 21730,
      "student_distance_mean": 1948.9257690429688,
      "teacher_distance_mean": 1.1439336299896241,
      "weighted_auxiliary_loss": 0.00041848404798656703,
      "weighted_relation_loss": 0.00041848404798656703
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.336763330215156,
      "flow_matching_loss": 0.019342876598238946,
      "grad_norm": 0.1014087125658989,
      "learning_rate": 1.933723033068533e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004335224599344656,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.03125114440918,
      "state_query_norm": 1442.7123046875,
      "state_relation_loss": 0.0004335224599344656,
      "step": 21740,
      "student_distance_mean": 1977.5016845703126,
      "teacher_distance_mean": 1.1615310668945313,
      "weighted_auxiliary_loss": 0.0004335224599344656,
      "weighted_relation_loss": 0.0004335224599344656
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.346117867165574,
      "flow_matching_loss": 0.019736251421272756,
      "grad_norm": 0.10891277343034744,
      "learning_rate": 1.929371395213674e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004428241809364408,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.118764114379886,
      "state_query_norm": 1440.28994140625,
      "state_relation_loss": 0.0004428241809364408,
      "step": 21750,
      "student_distance_mean": 1981.6887451171874,
      "teacher_distance_mean": 1.1637823820114135,
      "weighted_auxiliary_loss": 0.0004428241809364408,
      "weighted_relation_loss": 0.0004428241809364408
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.355472404115996,
      "flow_matching_loss": 0.019072137586772443,
      "grad_norm": 0.09548115730285645,
      "learning_rate": 1.9250234884621092e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003821356687694788,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3759048461914,
      "state_query_norm": 1422.8083374023438,
      "state_relation_loss": 0.0003821356687694788,
      "step": 21760,
      "student_distance_mean": 1949.1090576171875,
      "teacher_distance_mean": 1.1449273705482483,
      "weighted_auxiliary_loss": 0.0003821356687694788,
      "weighted_relation_loss": 0.0003821356687694788
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.36482694106642,
      "flow_matching_loss": 0.018690996710211037,
      "grad_norm": 0.09195856004953384,
      "learning_rate": 1.920679318096959e-05,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041458997584413736,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43285789489746,
      "state_query_norm": 1417.2728271484375,
      "state_relation_loss": 0.00041458997584413736,
      "step": 21770,
      "student_distance_mean": 1948.1655395507812,
      "teacher_distance_mean": 1.1455368399620056,
      "weighted_auxiliary_loss": 0.00041458997584413736,
      "weighted_relation_loss": 0.00041458997584413736
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.374181478016837,
      "flow_matching_loss": 0.020278090611100197,
      "grad_norm": 0.09170466661453247,
      "learning_rate": 1.9163388893967982e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004081076360307634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.527717971801756,
      "state_query_norm": 1425.0486572265625,
      "state_relation_loss": 0.0004081076360307634,
      "step": 21780,
      "student_distance_mean": 1954.4899658203126,
      "teacher_distance_mean": 1.1530342102050781,
      "weighted_auxiliary_loss": 0.0004081076360307634,
      "weighted_relation_loss": 0.0004081076360307634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.38353601496726,
      "flow_matching_loss": 0.02002969915047288,
      "grad_norm": 0.09532414376735687,
      "learning_rate": 1.9120022076356575e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042239610047545284,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.417321395874026,
      "state_query_norm": 1419.8826049804688,
      "state_relation_loss": 0.00042239610047545284,
      "step": 21790,
      "student_distance_mean": 1945.5074096679687,
      "teacher_distance_mean": 1.148290503025055,
      "weighted_auxiliary_loss": 0.00042239610047545284,
      "weighted_relation_loss": 0.00042239610047545284
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.39289055191768,
      "flow_matching_loss": 0.02010071147233248,
      "grad_norm": 0.07830968499183655,
      "learning_rate": 1.9076692780830114e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042553023959044365,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78748817443848,
      "state_query_norm": 1440.5042236328125,
      "state_relation_loss": 0.00042553023959044365,
      "step": 21800,
      "student_distance_mean": 1969.2500244140624,
      "teacher_distance_mean": 1.163852608203888,
      "weighted_auxiliary_loss": 0.00042553023959044365,
      "weighted_relation_loss": 0.00042553023959044365
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.4022450888681,
      "flow_matching_loss": 0.019254919700324537,
      "grad_norm": 0.08542263507843018,
      "learning_rate": 1.903340106003782e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004043293447466567,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53669738769531,
      "state_query_norm": 1424.4356689453125,
      "state_relation_loss": 0.0004043293447466567,
      "step": 21810,
      "student_distance_mean": 1953.5571411132812,
      "teacher_distance_mean": 1.1540380835533142,
      "weighted_auxiliary_loss": 0.0004043293447466567,
      "weighted_relation_loss": 0.0004043293447466567
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.41159962581852,
      "flow_matching_loss": 0.0197261281311512,
      "grad_norm": 0.08745252341032028,
      "learning_rate": 1.8990146966583183e-05,
      "loss": 0.0201,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004075527569511905,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.576511764526366,
      "state_query_norm": 1429.6465576171875,
      "state_relation_loss": 0.0004075527569511905,
      "step": 21820,
      "student_distance_mean": 1958.19287109375,
      "teacher_distance_mean": 1.1555558800697328,
      "weighted_auxiliary_loss": 0.0004075527569511905,
      "weighted_relation_loss": 0.0004075527569511905
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.420954162768943,
      "flow_matching_loss": 0.020884506590664386,
      "grad_norm": 0.09436040371656418,
      "learning_rate": 1.8946930553024035e-05,
      "loss": 0.0213,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004284773953258991,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.509748458862305,
      "state_query_norm": 1422.9090576171875,
      "state_relation_loss": 0.0004284773953258991,
      "step": 21830,
      "student_distance_mean": 1954.6236938476563,
      "teacher_distance_mean": 1.1503262758255004,
      "weighted_auxiliary_loss": 0.0004284773953258991,
      "weighted_relation_loss": 0.0004284773953258991
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.430308699719365,
      "flow_matching_loss": 0.020632929541170596,
      "grad_norm": 0.08260061591863632,
      "learning_rate": 1.8903751871872378e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043273355404380707,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02491302490235,
      "state_query_norm": 1438.360107421875,
      "state_relation_loss": 0.00043273355404380707,
      "step": 21840,
      "student_distance_mean": 1978.9207397460937,
      "teacher_distance_mean": 1.1669582724571228,
      "weighted_auxiliary_loss": 0.00043273355404380707,
      "weighted_relation_loss": 0.00043273355404380707
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.439663236669784,
      "flow_matching_loss": 0.02013145051896572,
      "grad_norm": 0.09851430356502533,
      "learning_rate": 1.8860610975594382e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004153589630732313,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.913817977905275,
      "state_query_norm": 1403.2624633789062,
      "state_relation_loss": 0.0004153589630732313,
      "step": 21850,
      "student_distance_mean": 1917.9973876953125,
      "teacher_distance_mean": 1.1335824728012085,
      "weighted_auxiliary_loss": 0.0004153589630732313,
      "weighted_relation_loss": 0.0004153589630732313
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.449017773620206,
      "flow_matching_loss": 0.019617915339767932,
      "grad_norm": 0.10969299077987671,
      "learning_rate": 1.8817507916610307e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004218600282911211,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71484794616699,
      "state_query_norm": 1422.6197387695313,
      "state_relation_loss": 0.0004218600282911211,
      "step": 21860,
      "student_distance_mean": 1959.1329467773437,
      "teacher_distance_mean": 1.1569467067718506,
      "weighted_auxiliary_loss": 0.0004218600282911211,
      "weighted_relation_loss": 0.0004218600282911211
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.458372310570628,
      "flow_matching_loss": 0.0209446769207716,
      "grad_norm": 0.10637139528989792,
      "learning_rate": 1.8774442747294407e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041969474405050277,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60622634887695,
      "state_query_norm": 1433.0281982421875,
      "state_relation_loss": 0.00041969474405050277,
      "step": 21870,
      "student_distance_mean": 1958.3597412109375,
      "teacher_distance_mean": 1.1542296767234803,
      "weighted_auxiliary_loss": 0.00041969474405050277,
      "weighted_relation_loss": 0.00041969474405050277
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.467726847521046,
      "flow_matching_loss": 0.021117728017270566,
      "grad_norm": 0.103034608066082,
      "learning_rate": 1.8731415519974965e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039718739863019436,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.648824310302736,
      "state_query_norm": 1423.2951049804688,
      "state_relation_loss": 0.00039718739863019436,
      "step": 21880,
      "student_distance_mean": 1955.0586303710938,
      "teacher_distance_mean": 1.152480125427246,
      "weighted_auxiliary_loss": 0.00039718739863019436,
      "weighted_relation_loss": 0.00039718739863019436
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.477081384471468,
      "flow_matching_loss": 0.019588895514607428,
      "grad_norm": 0.0904603973031044,
      "learning_rate": 1.86884262869341e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040298920939676466,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52832221984863,
      "state_query_norm": 1420.6610229492187,
      "state_relation_loss": 0.00040298920939676466,
      "step": 21890,
      "student_distance_mean": 1952.6066040039063,
      "teacher_distance_mean": 1.1518671274185182,
      "weighted_auxiliary_loss": 0.00040298920939676466,
      "weighted_relation_loss": 0.00040298920939676466
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.48643592142189,
      "flow_matching_loss": 0.019916440453380348,
      "grad_norm": 0.08018354326486588,
      "learning_rate": 1.8645475100407788e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004326326015871018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26859321594238,
      "state_query_norm": 1401.424609375,
      "state_relation_loss": 0.0004326326015871018,
      "step": 21900,
      "student_distance_mean": 1932.2747802734375,
      "teacher_distance_mean": 1.1400046586990356,
      "weighted_auxiliary_loss": 0.0004326326015871018,
      "weighted_relation_loss": 0.0004326326015871018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.495790458372312,
      "flow_matching_loss": 0.01953153247013688,
      "grad_norm": 0.09588772058486938,
      "learning_rate": 1.8602562012585768e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039808344445191324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93794631958008,
      "state_query_norm": 1442.4239013671875,
      "state_relation_loss": 0.00039808344445191324,
      "step": 21910,
      "student_distance_mean": 1980.4713134765625,
      "teacher_distance_mean": 1.165862536430359,
      "weighted_auxiliary_loss": 0.00039808344445191324,
      "weighted_relation_loss": 0.00039808344445191324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.50514499532273,
      "flow_matching_loss": 0.020826187171041965,
      "grad_norm": 0.08954042196273804,
      "learning_rate": 1.8559687075611466e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004045652283821255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.765371322631836,
      "state_query_norm": 1431.578857421875,
      "state_relation_loss": 0.0004045652283821255,
      "step": 21920,
      "student_distance_mean": 1966.5037353515625,
      "teacher_distance_mean": 1.1578776717185975,
      "weighted_auxiliary_loss": 0.0004045652283821255,
      "weighted_relation_loss": 0.0004045652283821255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.514499532273152,
      "flow_matching_loss": 0.021446841396391393,
      "grad_norm": 0.09198414534330368,
      "learning_rate": 1.8516850341582017e-05,
      "loss": 0.0219,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004088369634700939,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.455946350097655,
      "state_query_norm": 1422.9268920898437,
      "state_relation_loss": 0.0004088369634700939,
      "step": 21930,
      "student_distance_mean": 1947.542041015625,
      "teacher_distance_mean": 1.1488121032714844,
      "weighted_auxiliary_loss": 0.0004088369634700939,
      "weighted_relation_loss": 0.0004088369634700939
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.523854069223574,
      "flow_matching_loss": 0.01859359396621585,
      "grad_norm": 0.08786307275295258,
      "learning_rate": 1.847405186254807e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040872188401408494,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.515296173095706,
      "state_query_norm": 1419.5885009765625,
      "state_relation_loss": 0.00040872188401408494,
      "step": 21940,
      "student_distance_mean": 1950.9708251953125,
      "teacher_distance_mean": 1.1504173636436463,
      "weighted_auxiliary_loss": 0.00040872188401408494,
      "weighted_relation_loss": 0.00040872188401408494
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.533208606173993,
      "flow_matching_loss": 0.01893730666488409,
      "grad_norm": 0.0794115737080574,
      "learning_rate": 1.843129169051379e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003880108444718644,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.373694229125974,
      "state_query_norm": 1417.1713745117188,
      "state_relation_loss": 0.0003880108444718644,
      "step": 21950,
      "student_distance_mean": 1945.1654663085938,
      "teacher_distance_mean": 1.147025465965271,
      "weighted_auxiliary_loss": 0.0003880108444718644,
      "weighted_relation_loss": 0.0003880108444718644
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.542563143124415,
      "flow_matching_loss": 0.01870002634823322,
      "grad_norm": 0.09866633266210556,
      "learning_rate": 1.8388569877436863e-05,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004100340564036742,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.129441833496095,
      "state_query_norm": 1406.3957275390626,
      "state_relation_loss": 0.0004100340564036742,
      "step": 21960,
      "student_distance_mean": 1930.0489990234375,
      "teacher_distance_mean": 1.1397948503494262,
      "weighted_auxiliary_loss": 0.0004100340564036742,
      "weighted_relation_loss": 0.0004100340564036742
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.551917680074837,
      "flow_matching_loss": 0.019168464373797177,
      "grad_norm": 0.08403124660253525,
      "learning_rate": 1.834588647522828e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004302311077481136,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74436264038086,
      "state_query_norm": 1430.726708984375,
      "state_relation_loss": 0.0004302311077481136,
      "step": 21970,
      "student_distance_mean": 1963.817041015625,
      "teacher_distance_mean": 1.1608460307121278,
      "weighted_auxiliary_loss": 0.0004302311077481136,
      "weighted_relation_loss": 0.0004302311077481136
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.56127221702526,
      "flow_matching_loss": 0.018818878382444382,
      "grad_norm": 0.08041311055421829,
      "learning_rate": 1.8303241535752435e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004092844756087288,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91987915039063,
      "state_query_norm": 1439.3708374023438,
      "state_relation_loss": 0.0004092844756087288,
      "step": 21980,
      "student_distance_mean": 1970.9377075195312,
      "teacher_distance_mean": 1.1633723258972168,
      "weighted_auxiliary_loss": 0.0004092844756087288,
      "weighted_relation_loss": 0.0004092844756087288
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.570626753975677,
      "flow_matching_loss": 0.019311492517590522,
      "grad_norm": 0.09446320682764053,
      "learning_rate": 1.8260635110826934e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000402512660366483,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56980171203613,
      "state_query_norm": 1427.4715087890625,
      "state_relation_loss": 0.000402512660366483,
      "step": 21990,
      "student_distance_mean": 1956.5146850585938,
      "teacher_distance_mean": 1.155755603313446,
      "weighted_auxiliary_loss": 0.000402512660366483,
      "weighted_relation_loss": 0.000402512660366483
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.5799812909261,
      "flow_matching_loss": 0.019589032884687184,
      "grad_norm": 0.09887721389532089,
      "learning_rate": 1.8218067252222597e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041211359784938396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.456509017944335,
      "state_query_norm": 1413.2602905273438,
      "state_relation_loss": 0.00041211359784938396,
      "step": 22000,
      "student_distance_mean": 1944.195166015625,
      "teacher_distance_mean": 1.1482904434204102,
      "weighted_auxiliary_loss": 0.00041211359784938396,
      "weighted_relation_loss": 0.00041211359784938396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.58933582787652,
      "flow_matching_loss": 0.01918247388675809,
      "grad_norm": 0.08820609003305435,
      "learning_rate": 1.817553801166339e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003808502951869741,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.640451431274414,
      "state_query_norm": 1390.8106201171875,
      "state_relation_loss": 0.0003808502951869741,
      "step": 22010,
      "student_distance_mean": 1907.2851806640624,
      "teacher_distance_mean": 1.1279271125793457,
      "weighted_auxiliary_loss": 0.0003808502951869741,
      "weighted_relation_loss": 0.0003808502951869741
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.59869036482694,
      "flow_matching_loss": 0.020405846275389196,
      "grad_norm": 0.12757527828216553,
      "learning_rate": 1.8133047440826333e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004125491948798299,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89010391235352,
      "state_query_norm": 1436.6020141601562,
      "state_relation_loss": 0.0004125491948798299,
      "step": 22020,
      "student_distance_mean": 1972.5739013671875,
      "teacher_distance_mean": 1.1670910835266113,
      "weighted_auxiliary_loss": 0.0004125491948798299,
      "weighted_relation_loss": 0.0004125491948798299
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.60804490177736,
      "flow_matching_loss": 0.01936224214732647,
      "grad_norm": 0.10721830278635025,
      "learning_rate": 1.809059559134151e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041260113939642904,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52779426574707,
      "state_query_norm": 1430.2746215820312,
      "state_relation_loss": 0.00041260113939642904,
      "step": 22030,
      "student_distance_mean": 1953.0099487304688,
      "teacher_distance_mean": 1.154317057132721,
      "weighted_auxiliary_loss": 0.00041260113939642904,
      "weighted_relation_loss": 0.00041260113939642904
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.617399438727784,
      "flow_matching_loss": 0.020994810108095407,
      "grad_norm": 0.09294102340936661,
      "learning_rate": 1.80481825147919e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041332838009111583,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5947940826416,
      "state_query_norm": 1435.5796264648438,
      "state_relation_loss": 0.00041332838009111583,
      "step": 22040,
      "student_distance_mean": 1960.707958984375,
      "teacher_distance_mean": 1.1574702262878418,
      "weighted_auxiliary_loss": 0.00041332838009111583,
      "weighted_relation_loss": 0.00041332838009111583
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.626753975678206,
      "flow_matching_loss": 0.019942357018589974,
      "grad_norm": 0.09243395179510117,
      "learning_rate": 1.8005808262713396e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044674680393654855,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.07829208374024,
      "state_query_norm": 1447.7900512695312,
      "state_relation_loss": 0.00044674680393654855,
      "step": 22050,
      "student_distance_mean": 1981.6223266601562,
      "teacher_distance_mean": 1.1694442510604859,
      "weighted_auxiliary_loss": 0.00044674680393654855,
      "weighted_relation_loss": 0.00044674680393654855
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.636108512628624,
      "flow_matching_loss": 0.01890548150986433,
      "grad_norm": 0.08121906965970993,
      "learning_rate": 1.7963472886594714e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040190621803049,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30415000915527,
      "state_query_norm": 1420.8260009765625,
      "state_relation_loss": 0.00040190621803049,
      "step": 22060,
      "student_distance_mean": 1941.9752685546875,
      "teacher_distance_mean": 1.1469595909118653,
      "weighted_auxiliary_loss": 0.00040190621803049,
      "weighted_relation_loss": 0.00040190621803049
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.645463049579046,
      "flow_matching_loss": 0.01898764297366142,
      "grad_norm": 0.09219682961702347,
      "learning_rate": 1.7921176437877302e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042545271280687303,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.665972518920896,
      "state_query_norm": 1433.8483642578126,
      "state_relation_loss": 0.00042545271280687303,
      "step": 22070,
      "student_distance_mean": 1960.7248046875,
      "teacher_distance_mean": 1.1585610508918762,
      "weighted_auxiliary_loss": 0.00042545271280687303,
      "weighted_relation_loss": 0.00042545271280687303
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.654817586529468,
      "flow_matching_loss": 0.020335515215992927,
      "grad_norm": 0.10025990009307861,
      "learning_rate": 1.7878918967955366e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003973514743847772,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58010368347168,
      "state_query_norm": 1439.1655883789062,
      "state_relation_loss": 0.0003973514743847772,
      "step": 22080,
      "student_distance_mean": 1960.3577880859375,
      "teacher_distance_mean": 1.157813835144043,
      "weighted_auxiliary_loss": 0.0003973514743847772,
      "weighted_relation_loss": 0.0003973514743847772
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.664172123479887,
      "flow_matching_loss": 0.02002861201763153,
      "grad_norm": 0.08449923992156982,
      "learning_rate": 1.7836700528175693e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043110744445584714,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.540825271606444,
      "state_query_norm": 1425.5327270507812,
      "state_relation_loss": 0.00043110744445584714,
      "step": 22090,
      "student_distance_mean": 1952.0140014648437,
      "teacher_distance_mean": 1.1520352125167848,
      "weighted_auxiliary_loss": 0.00043110744445584714,
      "weighted_relation_loss": 0.00043110744445584714
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.67352666043031,
      "flow_matching_loss": 0.020396862365305423,
      "grad_norm": 0.10966524481773376,
      "learning_rate": 1.779452116983769e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004320498264860362,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24769821166992,
      "state_query_norm": 1412.9966186523438,
      "state_relation_loss": 0.0004320498264860362,
      "step": 22100,
      "student_distance_mean": 1935.8155395507813,
      "teacher_distance_mean": 1.1426887154579162,
      "weighted_auxiliary_loss": 0.0004320498264860362,
      "weighted_relation_loss": 0.0004320498264860362
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.68288119738073,
      "flow_matching_loss": 0.020020827651023865,
      "grad_norm": 0.09962622076272964,
      "learning_rate": 1.775238094419325e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004296004248317331,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60824432373047,
      "state_query_norm": 1427.58232421875,
      "state_relation_loss": 0.0004296004248317331,
      "step": 22110,
      "student_distance_mean": 1957.3355224609375,
      "teacher_distance_mean": 1.1538795351982116,
      "weighted_auxiliary_loss": 0.0004296004248317331,
      "weighted_relation_loss": 0.0004296004248317331
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.69223573433115,
      "flow_matching_loss": 0.019835127890110014,
      "grad_norm": 0.10573001950979233,
      "learning_rate": 1.7710279902446708e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044018562766723335,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.998600006103516,
      "state_query_norm": 1438.1834838867187,
      "state_relation_loss": 0.00044018562766723335,
      "step": 22120,
      "student_distance_mean": 1974.3853393554687,
      "teacher_distance_mean": 1.1630098342895507,
      "weighted_auxiliary_loss": 0.00044018562766723335,
      "weighted_relation_loss": 0.00044018562766723335
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.70159027128157,
      "flow_matching_loss": 0.019331551250070335,
      "grad_norm": 0.08130060136318207,
      "learning_rate": 1.7668218095754795e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041412699210923164,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39188575744629,
      "state_query_norm": 1422.4056762695313,
      "state_relation_loss": 0.00041412699210923164,
      "step": 22130,
      "student_distance_mean": 1946.76220703125,
      "teacher_distance_mean": 1.1455454230308533,
      "weighted_auxiliary_loss": 0.00041412699210923164,
      "weighted_relation_loss": 0.00041412699210923164
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.710944808231993,
      "flow_matching_loss": 0.019389417953789234,
      "grad_norm": 0.08385573327541351,
      "learning_rate": 1.7626195575226594e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004010469448985532,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5280258178711,
      "state_query_norm": 1417.1221435546875,
      "state_relation_loss": 0.0004010469448985532,
      "step": 22140,
      "student_distance_mean": 1953.2021118164062,
      "teacher_distance_mean": 1.1515507459640504,
      "weighted_auxiliary_loss": 0.0004010469448985532,
      "weighted_relation_loss": 0.0004010469448985532
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.720299345182415,
      "flow_matching_loss": 0.02043528649955988,
      "grad_norm": 0.09418348222970963,
      "learning_rate": 1.758421239192343e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042690217960625885,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.483486557006835,
      "state_query_norm": 1427.4242065429687,
      "state_relation_loss": 0.00042690217960625885,
      "step": 22150,
      "student_distance_mean": 1950.2301879882812,
      "teacher_distance_mean": 1.150213623046875,
      "weighted_auxiliary_loss": 0.00042690217960625885,
      "weighted_relation_loss": 0.00042690217960625885
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.729653882132833,
      "flow_matching_loss": 0.021096504665911196,
      "grad_norm": 0.10366913676261902,
      "learning_rate": 1.7542268596858814e-05,
      "loss": 0.0215,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040364640299230816,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93473129272461,
      "state_query_norm": 1450.4652587890625,
      "state_relation_loss": 0.00040364640299230816,
      "step": 22160,
      "student_distance_mean": 1977.827099609375,
      "teacher_distance_mean": 1.1653516292572021,
      "weighted_auxiliary_loss": 0.00040364640299230816,
      "weighted_relation_loss": 0.00040364640299230816
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.739008419083255,
      "flow_matching_loss": 0.02007630905136466,
      "grad_norm": 0.08100463449954987,
      "learning_rate": 1.750036424099841e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040528615936636926,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26126251220703,
      "state_query_norm": 1425.7283569335937,
      "state_relation_loss": 0.00040528615936636926,
      "step": 22170,
      "student_distance_mean": 1944.7724365234376,
      "teacher_distance_mean": 1.144778025150299,
      "weighted_auxiliary_loss": 0.00040528615936636926,
      "weighted_relation_loss": 0.00040528615936636926
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.748362956033677,
      "flow_matching_loss": 0.0189526510424912,
      "grad_norm": 0.08578820526599884,
      "learning_rate": 1.7458499375259956e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000417945280787535,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.701501083374026,
      "state_query_norm": 1423.38525390625,
      "state_relation_loss": 0.000417945280787535,
      "step": 22180,
      "student_distance_mean": 1958.8695068359375,
      "teacher_distance_mean": 1.155505645275116,
      "weighted_auxiliary_loss": 0.000417945280787535,
      "weighted_relation_loss": 0.000417945280787535
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.757717492984096,
      "flow_matching_loss": 0.020035756565630437,
      "grad_norm": 0.08747043460607529,
      "learning_rate": 1.7416674050513245e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004344867542386055,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.282706832885744,
      "state_query_norm": 1441.5719116210937,
      "state_relation_loss": 0.0004344867542386055,
      "step": 22190,
      "student_distance_mean": 1989.682861328125,
      "teacher_distance_mean": 1.1730599284172059,
      "weighted_auxiliary_loss": 0.0004344867542386055,
      "weighted_relation_loss": 0.0004344867542386055
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.767072029934518,
      "flow_matching_loss": 0.02066900338977575,
      "grad_norm": 0.0855056419968605,
      "learning_rate": 1.737488831757997e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041306467028334737,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.67813415527344,
      "state_query_norm": 1428.3102661132812,
      "state_relation_loss": 0.00041306467028334737,
      "step": 22200,
      "student_distance_mean": 1958.6650146484376,
      "teacher_distance_mean": 1.1549121022224427,
      "weighted_auxiliary_loss": 0.00041306467028334737,
      "weighted_relation_loss": 0.00041306467028334737
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.77642656688494,
      "flow_matching_loss": 0.020005868934094905,
      "grad_norm": 0.0910479873418808,
      "learning_rate": 1.733314222723373e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041838132601697,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41929893493652,
      "state_query_norm": 1424.7035400390625,
      "state_relation_loss": 0.00041838132601697,
      "step": 22210,
      "student_distance_mean": 1950.462451171875,
      "teacher_distance_mean": 1.1512641191482544,
      "weighted_auxiliary_loss": 0.00041838132601697,
      "weighted_relation_loss": 0.00041838132601697
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.78578110383536,
      "flow_matching_loss": 0.019138025771826505,
      "grad_norm": 0.09737076610326767,
      "learning_rate": 1.7291435830199954e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004155165544943884,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.807417678833005,
      "state_query_norm": 1430.502783203125,
      "state_relation_loss": 0.0004155165544943884,
      "step": 22220,
      "student_distance_mean": 1965.6731079101562,
      "teacher_distance_mean": 1.159208858013153,
      "weighted_auxiliary_loss": 0.0004155165544943884,
      "weighted_relation_loss": 0.0004155165544943884
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.79513564078578,
      "flow_matching_loss": 0.020688233617693185,
      "grad_norm": 0.07966967672109604,
      "learning_rate": 1.7249769177155878e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004135063762078062,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.933091735839845,
      "state_query_norm": 1437.232373046875,
      "state_relation_loss": 0.0004135063762078062,
      "step": 22230,
      "student_distance_mean": 1974.1715087890625,
      "teacher_distance_mean": 1.1636799097061157,
      "weighted_auxiliary_loss": 0.0004135063762078062,
      "weighted_relation_loss": 0.0004135063762078062
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.804490177736202,
      "flow_matching_loss": 0.02178937289863825,
      "grad_norm": 0.08358219265937805,
      "learning_rate": 1.720814231873038e-05,
      "loss": 0.0222,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041175805672537534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21222114562988,
      "state_query_norm": 1408.482861328125,
      "state_relation_loss": 0.00041175805672537534,
      "step": 22240,
      "student_distance_mean": 1934.9890625,
      "teacher_distance_mean": 1.1401805996894836,
      "weighted_auxiliary_loss": 0.00041175805672537534,
      "weighted_relation_loss": 0.00041175805672537534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.813844714686624,
      "flow_matching_loss": 0.020765579491853713,
      "grad_norm": 0.10313031822443008,
      "learning_rate": 1.716655530550405e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041469247662462294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74962844848633,
      "state_query_norm": 1430.7764770507813,
      "state_relation_loss": 0.00041469247662462294,
      "step": 22250,
      "student_distance_mean": 1964.2011840820312,
      "teacher_distance_mean": 1.1563994407653808,
      "weighted_auxiliary_loss": 0.00041469247662462294,
      "weighted_relation_loss": 0.00041469247662462294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.823199251637043,
      "flow_matching_loss": 0.02063711974769831,
      "grad_norm": 0.12229204922914505,
      "learning_rate": 1.7125008188009017e-05,
      "loss": 0.021,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040690256282687186,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43910598754883,
      "state_query_norm": 1417.39287109375,
      "state_relation_loss": 0.00040690256282687186,
      "step": 22260,
      "student_distance_mean": 1948.9838500976562,
      "teacher_distance_mean": 1.148765468597412,
      "weighted_auxiliary_loss": 0.00040690256282687186,
      "weighted_relation_loss": 0.00040690256282687186
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.832553788587465,
      "flow_matching_loss": 0.01921225655823946,
      "grad_norm": 0.1054208055138588,
      "learning_rate": 1.708350101672894e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004020876425784081,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48695945739746,
      "state_query_norm": 1419.6730224609375,
      "state_relation_loss": 0.0004020876425784081,
      "step": 22270,
      "student_distance_mean": 1950.9191040039063,
      "teacher_distance_mean": 1.1489768266677856,
      "weighted_auxiliary_loss": 0.0004020876425784081,
      "weighted_relation_loss": 0.0004020876425784081
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.841908325537887,
      "flow_matching_loss": 0.020383685640990735,
      "grad_norm": 0.10504261404275894,
      "learning_rate": 1.7042033842098957e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040048017690423875,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.49023933410645,
      "state_query_norm": 1425.3687866210937,
      "state_relation_loss": 0.00040048017690423875,
      "step": 22280,
      "student_distance_mean": 1948.6936767578125,
      "teacher_distance_mean": 1.1473618388175963,
      "weighted_auxiliary_loss": 0.00040048017690423875,
      "weighted_relation_loss": 0.00040048017690423875
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.85126286248831,
      "flow_matching_loss": 0.020985289476811886,
      "grad_norm": 0.09516787528991699,
      "learning_rate": 1.7000606714505567e-05,
      "loss": 0.0214,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004005617141956463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45616188049316,
      "state_query_norm": 1421.433251953125,
      "state_relation_loss": 0.0004005617141956463,
      "step": 22290,
      "student_distance_mean": 1948.803564453125,
      "teacher_distance_mean": 1.1483714580535889,
      "weighted_auxiliary_loss": 0.0004005617141956463,
      "weighted_relation_loss": 0.0004005617141956463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.860617399438727,
      "flow_matching_loss": 0.020763250440359114,
      "grad_norm": 0.08562426269054413,
      "learning_rate": 1.695921968428668e-05,
      "loss": 0.0212,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004298083658795804,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76469955444336,
      "state_query_norm": 1425.0412109375,
      "state_relation_loss": 0.0004298083658795804,
      "step": 22300,
      "student_distance_mean": 1961.8570068359375,
      "teacher_distance_mean": 1.156326913833618,
      "weighted_auxiliary_loss": 0.0004298083658795804,
      "weighted_relation_loss": 0.0004298083658795804
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.86997193638915,
      "flow_matching_loss": 0.017719934973865748,
      "grad_norm": 0.08268671482801437,
      "learning_rate": 1.6917872801731416e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000416668935213238,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.430625915527344,
      "state_query_norm": 1417.635986328125,
      "state_relation_loss": 0.000416668935213238,
      "step": 22310,
      "student_distance_mean": 1943.3231689453125,
      "teacher_distance_mean": 1.1458574652671814,
      "weighted_auxiliary_loss": 0.000416668935213238,
      "weighted_relation_loss": 0.000416668935213238
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.87932647333957,
      "flow_matching_loss": 0.019841591827571392,
      "grad_norm": 0.07954714447259903,
      "learning_rate": 1.687656611708014e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004056056059198454,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17367744445801,
      "state_query_norm": 1412.8388793945312,
      "state_relation_loss": 0.0004056056059198454,
      "step": 22320,
      "student_distance_mean": 1934.5072021484375,
      "teacher_distance_mean": 1.1413856863975524,
      "weighted_auxiliary_loss": 0.0004056056059198454,
      "weighted_relation_loss": 0.0004056056059198454
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.88868101028999,
      "flow_matching_loss": 0.020024592522531748,
      "grad_norm": 0.12090641260147095,
      "learning_rate": 1.6835299680524368e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004037328704725951,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30565185546875,
      "state_query_norm": 1419.2784545898437,
      "state_relation_loss": 0.0004037328704725951,
      "step": 22330,
      "student_distance_mean": 1940.6107788085938,
      "teacher_distance_mean": 1.1454975128173828,
      "weighted_auxiliary_loss": 0.0004037328704725951,
      "weighted_relation_loss": 0.0004037328704725951
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.89803554724041,
      "flow_matching_loss": 0.018735687155276536,
      "grad_norm": 0.08827652037143707,
      "learning_rate": 1.679407354220669e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042363664542790504,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.443376541137695,
      "state_query_norm": 1418.178857421875,
      "state_relation_loss": 0.00042363664542790504,
      "step": 22340,
      "student_distance_mean": 1947.8173583984376,
      "teacher_distance_mean": 1.1493297576904298,
      "weighted_auxiliary_loss": 0.00042363664542790504,
      "weighted_relation_loss": 0.00042363664542790504
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.907390084190833,
      "flow_matching_loss": 0.01784876622259617,
      "grad_norm": 0.08315828442573547,
      "learning_rate": 1.6752887752220793e-05,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039241889026016,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.93211975097656,
      "state_query_norm": 1407.7674560546875,
      "state_relation_loss": 0.00039241889026016,
      "step": 22350,
      "student_distance_mean": 1922.994482421875,
      "teacher_distance_mean": 1.135176408290863,
      "weighted_auxiliary_loss": 0.00039241889026016,
      "weighted_relation_loss": 0.00039241889026016
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.916744621141255,
      "flow_matching_loss": 0.017396892327815293,
      "grad_norm": 0.07231474667787552,
      "learning_rate": 1.6711742360611277e-05,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004165746853686869,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56908493041992,
      "state_query_norm": 1419.5502197265625,
      "state_relation_loss": 0.0004165746853686869,
      "step": 22360,
      "student_distance_mean": 1952.8967407226562,
      "teacher_distance_mean": 1.1502883553504943,
      "weighted_auxiliary_loss": 0.0004165746853686869,
      "weighted_relation_loss": 0.0004165746853686869
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.926099158091674,
      "flow_matching_loss": 0.019011261500418188,
      "grad_norm": 0.07846241444349289,
      "learning_rate": 1.6670637417373652e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004374929500045255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.628288650512694,
      "state_query_norm": 1423.2855834960938,
      "state_relation_loss": 0.0004374929500045255,
      "step": 22370,
      "student_distance_mean": 1954.7951049804688,
      "teacher_distance_mean": 1.152283489704132,
      "weighted_auxiliary_loss": 0.0004374929500045255,
      "weighted_relation_loss": 0.0004374929500045255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.935453695042096,
      "flow_matching_loss": 0.02029536459594965,
      "grad_norm": 0.08796050399541855,
      "learning_rate": 1.6629572972454333e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041068850841838864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64192237854004,
      "state_query_norm": 1429.3751098632813,
      "state_relation_loss": 0.00041068850841838864,
      "step": 22380,
      "student_distance_mean": 1959.5623413085937,
      "teacher_distance_mean": 1.1559440851211549,
      "weighted_auxiliary_loss": 0.00041068850841838864,
      "weighted_relation_loss": 0.00041068850841838864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.944808231992518,
      "flow_matching_loss": 0.018099729716777802,
      "grad_norm": 0.08688458055257797,
      "learning_rate": 1.6588549075750464e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003874215442920104,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.518053436279295,
      "state_query_norm": 1428.3066162109376,
      "state_relation_loss": 0.0003874215442920104,
      "step": 22390,
      "student_distance_mean": 1955.0747192382812,
      "teacher_distance_mean": 1.1545541763305665,
      "weighted_auxiliary_loss": 0.0003874215442920104,
      "weighted_relation_loss": 0.0003874215442920104
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.954162768942936,
      "flow_matching_loss": 0.018162394780665635,
      "grad_norm": 0.10507319867610931,
      "learning_rate": 1.6547565777109976e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040543980721849946,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.054186248779295,
      "state_query_norm": 1404.390087890625,
      "state_relation_loss": 0.00040543980721849946,
      "step": 22400,
      "student_distance_mean": 1926.5231689453126,
      "teacher_distance_mean": 1.1375088691711426,
      "weighted_auxiliary_loss": 0.00040543980721849946,
      "weighted_relation_loss": 0.00040543980721849946
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.963517305893358,
      "flow_matching_loss": 0.018917169887572527,
      "grad_norm": 0.09478145092725754,
      "learning_rate": 1.650662312633143e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003929201338905841,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.79846839904785,
      "state_query_norm": 1437.9530639648438,
      "state_relation_loss": 0.0003929201338905841,
      "step": 22410,
      "student_distance_mean": 1967.3678833007812,
      "teacher_distance_mean": 1.1609055280685425,
      "weighted_auxiliary_loss": 0.0003929201338905841,
      "weighted_relation_loss": 0.0003929201338905841
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.97287184284378,
      "flow_matching_loss": 0.018952944315969944,
      "grad_norm": 0.09681154787540436,
      "learning_rate": 1.6465721173164002e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004212660831399262,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.560554885864256,
      "state_query_norm": 1423.10986328125,
      "state_relation_loss": 0.0004212660831399262,
      "step": 22420,
      "student_distance_mean": 1952.4811401367188,
      "teacher_distance_mean": 1.1513209223747254,
      "weighted_auxiliary_loss": 0.0004212660831399262,
      "weighted_relation_loss": 0.0004212660831399262
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.9822263797942,
      "flow_matching_loss": 0.019139081053435803,
      "grad_norm": 0.07854276895523071,
      "learning_rate": 1.6424859967307428e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004248120734700933,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.61894302368164,
      "state_query_norm": 1391.02802734375,
      "state_relation_loss": 0.0004248120734700933,
      "step": 22430,
      "student_distance_mean": 1905.318212890625,
      "teacher_distance_mean": 1.1241584539413452,
      "weighted_auxiliary_loss": 0.0004248120734700933,
      "weighted_relation_loss": 0.0004248120734700933
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 20.99158091674462,
      "flow_matching_loss": 0.01810386814177036,
      "grad_norm": 0.09218835830688477,
      "learning_rate": 1.6384039558411903e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003962646267609671,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21563911437988,
      "state_query_norm": 1418.7369995117188,
      "state_relation_loss": 0.0003962646267609671,
      "step": 22440,
      "student_distance_mean": 1937.9188110351563,
      "teacher_distance_mean": 1.1449050307273865,
      "weighted_auxiliary_loss": 0.0003962646267609671,
      "weighted_relation_loss": 0.0003962646267609671
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.000935453695043,
      "flow_matching_loss": 0.019533576164394616,
      "grad_norm": 0.104780413210392,
      "learning_rate": 1.634325999607811e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00038906843983568253,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.194879150390626,
      "state_query_norm": 1410.5776977539062,
      "state_relation_loss": 0.00038906843983568253,
      "step": 22450,
      "student_distance_mean": 1936.8413940429687,
      "teacher_distance_mean": 1.1428173303604126,
      "weighted_auxiliary_loss": 0.00038906843983568253,
      "weighted_relation_loss": 0.00038906843983568253
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.010289990645465,
      "flow_matching_loss": 0.019965832773596048,
      "grad_norm": 0.0924549549818039,
      "learning_rate": 1.6302521329857045e-05,
      "loss": 0.0204,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039249895489774645,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.759843063354495,
      "state_query_norm": 1433.340771484375,
      "state_relation_loss": 0.00039249895489774645,
      "step": 22460,
      "student_distance_mean": 1968.1126586914063,
      "teacher_distance_mean": 1.1611249923706055,
      "weighted_auxiliary_loss": 0.00039249895489774645,
      "weighted_relation_loss": 0.00039249895489774645
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.019644527595883,
      "flow_matching_loss": 0.02043726444244385,
      "grad_norm": 0.08714035898447037,
      "learning_rate": 1.6261823609250027e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044085177360102533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28294219970703,
      "state_query_norm": 1405.0121948242188,
      "state_relation_loss": 0.00044085177360102533,
      "step": 22470,
      "student_distance_mean": 1931.7711669921875,
      "teacher_distance_mean": 1.141118311882019,
      "weighted_auxiliary_loss": 0.00044085177360102533,
      "weighted_relation_loss": 0.00044085177360102533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.028999064546305,
      "flow_matching_loss": 0.018626300152391195,
      "grad_norm": 0.07935260236263275,
      "learning_rate": 1.6221166883708627e-05,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004581327812047675,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.28871803283691,
      "state_query_norm": 1446.8092529296875,
      "state_relation_loss": 0.0004581327812047675,
      "step": 22480,
      "student_distance_mean": 1988.6401733398438,
      "teacher_distance_mean": 1.174130129814148,
      "weighted_auxiliary_loss": 0.0004581327812047675,
      "weighted_relation_loss": 0.0004581327812047675
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.038353601496727,
      "flow_matching_loss": 0.019172961823642254,
      "grad_norm": 0.08426566421985626,
      "learning_rate": 1.6180551202634603e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043155802122782914,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.812834930419925,
      "state_query_norm": 1442.2954833984375,
      "state_relation_loss": 0.00043155802122782914,
      "step": 22490,
      "student_distance_mean": 1969.0687133789063,
      "teacher_distance_mean": 1.1611531138420106,
      "weighted_auxiliary_loss": 0.00043155802122782914,
      "weighted_relation_loss": 0.00043155802122782914
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.047708138447145,
      "flow_matching_loss": 0.01812632903456688,
      "grad_norm": 0.09398873150348663,
      "learning_rate": 1.613997661537981e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003993175661889836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56223411560059,
      "state_query_norm": 1422.1057495117188,
      "state_relation_loss": 0.0003993175661889836,
      "step": 22500,
      "student_distance_mean": 1952.0920166015626,
      "teacher_distance_mean": 1.150706398487091,
      "weighted_auxiliary_loss": 0.0003993175661889836,
      "weighted_relation_loss": 0.0003993175661889836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.057062675397567,
      "flow_matching_loss": 0.018793896399438383,
      "grad_norm": 0.08913208544254303,
      "learning_rate": 1.609944317124624e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003894446330377832,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.288272094726565,
      "state_query_norm": 1413.58642578125,
      "state_relation_loss": 0.0003894446330377832,
      "step": 22510,
      "student_distance_mean": 1939.2097290039062,
      "teacher_distance_mean": 1.1437062859535216,
      "weighted_auxiliary_loss": 0.0003894446330377832,
      "weighted_relation_loss": 0.0003894446330377832
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.06641721234799,
      "flow_matching_loss": 0.01843205960467458,
      "grad_norm": 0.08724775910377502,
      "learning_rate": 1.605895091948582e-05,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039157266146503387,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.615126037597655,
      "state_query_norm": 1427.8561401367188,
      "state_relation_loss": 0.00039157266146503387,
      "step": 22520,
      "student_distance_mean": 1958.3377197265625,
      "teacher_distance_mean": 1.1524720788002014,
      "weighted_auxiliary_loss": 0.00039157266146503387,
      "weighted_relation_loss": 0.00039157266146503387
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.07577174929841,
      "flow_matching_loss": 0.018474187050014735,
      "grad_norm": 0.08332982659339905,
      "learning_rate": 1.6018499909300477e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004130187939153984,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63363342285156,
      "state_query_norm": 1433.3372192382812,
      "state_relation_loss": 0.0004130187939153984,
      "step": 22530,
      "student_distance_mean": 1960.76025390625,
      "teacher_distance_mean": 1.1570204377174378,
      "weighted_auxiliary_loss": 0.0004130187939153984,
      "weighted_relation_loss": 0.0004130187939153984
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.08512628624883,
      "flow_matching_loss": 0.01823449060320854,
      "grad_norm": 0.08634832501411438,
      "learning_rate": 1.597809018984199e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039904802106320857,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41660385131836,
      "state_query_norm": 1417.2786376953125,
      "state_relation_loss": 0.00039904802106320857,
      "step": 22540,
      "student_distance_mean": 1946.2837524414062,
      "teacher_distance_mean": 1.1484803915023805,
      "weighted_auxiliary_loss": 0.00039904802106320857,
      "weighted_relation_loss": 0.00039904802106320857
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.09448082319925,
      "flow_matching_loss": 0.01923049371689558,
      "grad_norm": 0.09206317365169525,
      "learning_rate": 1.593772181021196e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040040850872173904,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29910011291504,
      "state_query_norm": 1410.7767578125,
      "state_relation_loss": 0.00040040850872173904,
      "step": 22550,
      "student_distance_mean": 1935.83720703125,
      "teacher_distance_mean": 1.14344881772995,
      "weighted_auxiliary_loss": 0.00040040850872173904,
      "weighted_relation_loss": 0.00040040850872173904
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.103835360149674,
      "flow_matching_loss": 0.01936287172138691,
      "grad_norm": 0.08107815682888031,
      "learning_rate": 1.5897394819461813e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003848069114610553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.253014755249026,
      "state_query_norm": 1419.8516967773437,
      "state_relation_loss": 0.0003848069114610553,
      "step": 22560,
      "student_distance_mean": 1940.3490600585938,
      "teacher_distance_mean": 1.143531286716461,
      "weighted_auxiliary_loss": 0.0003848069114610553,
      "weighted_relation_loss": 0.0003848069114610553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.113189897100092,
      "flow_matching_loss": 0.018247310165315868,
      "grad_norm": 0.10736280679702759,
      "learning_rate": 1.5857109266592625e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004258585890056565,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.8314712524414,
      "state_query_norm": 1432.0641967773438,
      "state_relation_loss": 0.0004258585890056565,
      "step": 22570,
      "student_distance_mean": 1967.8253295898437,
      "teacher_distance_mean": 1.1586123943328857,
      "weighted_auxiliary_loss": 0.0004258585890056565,
      "weighted_relation_loss": 0.0004258585890056565
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.122544434050514,
      "flow_matching_loss": 0.02049901904538274,
      "grad_norm": 0.10001643747091293,
      "learning_rate": 1.581686520055514e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040619293868076055,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71173858642578,
      "state_query_norm": 1430.4823364257813,
      "state_relation_loss": 0.00040619293868076055,
      "step": 22580,
      "student_distance_mean": 1962.4613525390625,
      "teacher_distance_mean": 1.1560216784477233,
      "weighted_auxiliary_loss": 0.00040619293868076055,
      "weighted_relation_loss": 0.00040619293868076055
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.131898971000936,
      "flow_matching_loss": 0.02021073466166854,
      "grad_norm": 0.09019912779331207,
      "learning_rate": 1.5776662670249704e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000410918949637562,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71112327575683,
      "state_query_norm": 1431.953076171875,
      "state_relation_loss": 0.000410918949637562,
      "step": 22590,
      "student_distance_mean": 1964.2605590820312,
      "teacher_distance_mean": 1.1591613650321961,
      "weighted_auxiliary_loss": 0.000410918949637562,
      "weighted_relation_loss": 0.000410918949637562
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.141253507951358,
      "flow_matching_loss": 0.019788332190364598,
      "grad_norm": 0.09074549376964569,
      "learning_rate": 1.573650172452615e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004002771514933556,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30118598937988,
      "state_query_norm": 1416.9022216796875,
      "state_relation_loss": 0.0004002771514933556,
      "step": 22600,
      "student_distance_mean": 1942.804052734375,
      "teacher_distance_mean": 1.1464101195335388,
      "weighted_auxiliary_loss": 0.0004002771514933556,
      "weighted_relation_loss": 0.0004002771514933556
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.150608044901777,
      "flow_matching_loss": 0.017956977896392345,
      "grad_norm": 0.09280087053775787,
      "learning_rate": 1.5696382412183852e-05,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038575637445319443,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.353499221801755,
      "state_query_norm": 1411.6921142578126,
      "state_relation_loss": 0.00038575637445319443,
      "step": 22610,
      "student_distance_mean": 1940.379296875,
      "teacher_distance_mean": 1.1442195892333984,
      "weighted_auxiliary_loss": 0.00038575637445319443,
      "weighted_relation_loss": 0.00038575637445319443
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.1599625818522,
      "flow_matching_loss": 0.019313657376915216,
      "grad_norm": 0.08000069111585617,
      "learning_rate": 1.565630478197155e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003940804948797449,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.179744720458984,
      "state_query_norm": 1410.643896484375,
      "state_relation_loss": 0.0003940804948797449,
      "step": 22620,
      "student_distance_mean": 1932.1296020507812,
      "teacher_distance_mean": 1.1414249300956727,
      "weighted_auxiliary_loss": 0.0003940804948797449,
      "weighted_relation_loss": 0.0003940804948797449
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.16931711880262,
      "flow_matching_loss": 0.01947270045056939,
      "grad_norm": 0.09508510679006577,
      "learning_rate": 1.5616268882587332e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003840485296677798,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.114701461791995,
      "state_query_norm": 1411.3087768554688,
      "state_relation_loss": 0.0003840485296677798,
      "step": 22630,
      "student_distance_mean": 1929.139990234375,
      "teacher_distance_mean": 1.1379701852798463,
      "weighted_auxiliary_loss": 0.0003840485296677798,
      "weighted_relation_loss": 0.0003840485296677798
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.17867165575304,
      "flow_matching_loss": 0.018606526684015988,
      "grad_norm": 0.09061796963214874,
      "learning_rate": 1.5576274762678594e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044274073734413834,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.799197387695315,
      "state_query_norm": 1433.7334228515624,
      "state_relation_loss": 0.00044274073734413834,
      "step": 22640,
      "student_distance_mean": 1964.4533325195312,
      "teacher_distance_mean": 1.1590571761131288,
      "weighted_auxiliary_loss": 0.00044274073734413834,
      "weighted_relation_loss": 0.00044274073734413834
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.18802619270346,
      "flow_matching_loss": 0.017967747431248425,
      "grad_norm": 0.0971783772110939,
      "learning_rate": 1.5536322470841952e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004053277458297089,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.446340560913086,
      "state_query_norm": 1412.492529296875,
      "state_relation_loss": 0.0004053277458297089,
      "step": 22650,
      "student_distance_mean": 1944.701904296875,
      "teacher_distance_mean": 1.1491622567176818,
      "weighted_auxiliary_loss": 0.0004053277458297089,
      "weighted_relation_loss": 0.0004053277458297089
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.197380729653883,
      "flow_matching_loss": 0.020459711365401746,
      "grad_norm": 0.07718400657176971,
      "learning_rate": 1.549641205562324e-05,
      "loss": 0.0209,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003940947965020314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44636116027832,
      "state_query_norm": 1425.887109375,
      "state_relation_loss": 0.0003940947965020314,
      "step": 22660,
      "student_distance_mean": 1953.1145263671874,
      "teacher_distance_mean": 1.152167022228241,
      "weighted_auxiliary_loss": 0.0003940947965020314,
      "weighted_relation_loss": 0.0003940947965020314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.2067352666043,
      "flow_matching_loss": 0.018924605567008257,
      "grad_norm": 0.08591064065694809,
      "learning_rate": 1.5456543565517334e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044362247863318773,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91076431274414,
      "state_query_norm": 1433.8352783203125,
      "state_relation_loss": 0.00044362247863318773,
      "step": 22670,
      "student_distance_mean": 1968.0373046875,
      "teacher_distance_mean": 1.158348596096039,
      "weighted_auxiliary_loss": 0.00044362247863318773,
      "weighted_relation_loss": 0.00044362247863318773
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.216089803554723,
      "flow_matching_loss": 0.0183131723664701,
      "grad_norm": 0.09837964922189713,
      "learning_rate": 1.541671704896825e-05,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00045674431603401897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.01859130859375,
      "state_query_norm": 1442.6156982421876,
      "state_relation_loss": 0.00045674431603401897,
      "step": 22680,
      "student_distance_mean": 1981.4882690429688,
      "teacher_distance_mean": 1.1650722861289977,
      "weighted_auxiliary_loss": 0.00045674431603401897,
      "weighted_relation_loss": 0.00045674431603401897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.225444340505145,
      "flow_matching_loss": 0.01984686069190502,
      "grad_norm": 0.07561025768518448,
      "learning_rate": 1.5376932554368938e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004211554245557636,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.652963638305664,
      "state_query_norm": 1421.718798828125,
      "state_relation_loss": 0.0004211554245557636,
      "step": 22690,
      "student_distance_mean": 1954.290087890625,
      "teacher_distance_mean": 1.1489103317260743,
      "weighted_auxiliary_loss": 0.0004211554245557636,
      "weighted_relation_loss": 0.0004211554245557636
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.234798877455567,
      "flow_matching_loss": 0.017933370545506477,
      "grad_norm": 0.09210924804210663,
      "learning_rate": 1.5337190130061318e-05,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040465256315656004,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62081794738769,
      "state_query_norm": 1431.2627319335938,
      "state_relation_loss": 0.00040465256315656004,
      "step": 22700,
      "student_distance_mean": 1959.7396240234375,
      "teacher_distance_mean": 1.1530396342277527,
      "weighted_auxiliary_loss": 0.00040465256315656004,
      "weighted_relation_loss": 0.00040465256315656004
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.244153414405986,
      "flow_matching_loss": 0.019104271568357943,
      "grad_norm": 0.10632637143135071,
      "learning_rate": 1.5297489824336165e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003889303479809314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26006965637207,
      "state_query_norm": 1416.8269287109374,
      "state_relation_loss": 0.0003889303479809314,
      "step": 22710,
      "student_distance_mean": 1940.861962890625,
      "teacher_distance_mean": 1.1443710446357727,
      "weighted_auxiliary_loss": 0.0003889303479809314,
      "weighted_relation_loss": 0.0003889303479809314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.253507951356408,
      "flow_matching_loss": 0.017437794618308543,
      "grad_norm": 0.09529036283493042,
      "learning_rate": 1.5257831685433121e-05,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043622513476293534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.1512996673584,
      "state_query_norm": 1446.6029663085938,
      "state_relation_loss": 0.00043622513476293534,
      "step": 22720,
      "student_distance_mean": 1982.1774169921875,
      "teacher_distance_mean": 1.167341947555542,
      "weighted_auxiliary_loss": 0.00043622513476293534,
      "weighted_relation_loss": 0.00043622513476293534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.26286248830683,
      "flow_matching_loss": 0.017760083824396134,
      "grad_norm": 0.10419878363609314,
      "learning_rate": 1.521821576154055e-05,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042260012996848674,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48372268676758,
      "state_query_norm": 1425.9450439453126,
      "state_relation_loss": 0.00042260012996848674,
      "step": 22730,
      "student_distance_mean": 1948.786962890625,
      "teacher_distance_mean": 1.1471741914749145,
      "weighted_auxiliary_loss": 0.00042260012996848674,
      "weighted_relation_loss": 0.00042260012996848674
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.27221702525725,
      "flow_matching_loss": 0.018232459016144276,
      "grad_norm": 0.1098431646823883,
      "learning_rate": 1.5178642100795542e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042775532929226754,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.88051872253418,
      "state_query_norm": 1428.1655639648438,
      "state_relation_loss": 0.00042775532929226754,
      "step": 22740,
      "student_distance_mean": 1967.3234741210938,
      "teacher_distance_mean": 1.1580551266670227,
      "weighted_auxiliary_loss": 0.00042775532929226754,
      "weighted_relation_loss": 0.00042775532929226754
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.28157156220767,
      "flow_matching_loss": 0.01827907608821988,
      "grad_norm": 0.10680659115314484,
      "learning_rate": 1.5139110751283819e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003951650083763525,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.8617488861084,
      "state_query_norm": 1433.2788452148438,
      "state_relation_loss": 0.0003951650083763525,
      "step": 22750,
      "student_distance_mean": 1967.8866943359376,
      "teacher_distance_mean": 1.1566694498062133,
      "weighted_auxiliary_loss": 0.0003951650083763525,
      "weighted_relation_loss": 0.0003951650083763525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.290926099158092,
      "flow_matching_loss": 0.021606320515275,
      "grad_norm": 0.0915292352437973,
      "learning_rate": 1.5099621761039683e-05,
      "loss": 0.022,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003916618821676821,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65482711791992,
      "state_query_norm": 1427.568798828125,
      "state_relation_loss": 0.0003916618821676821,
      "step": 22760,
      "student_distance_mean": 1959.8184814453125,
      "teacher_distance_mean": 1.1534447073936462,
      "weighted_auxiliary_loss": 0.0003916618821676821,
      "weighted_relation_loss": 0.0003916618821676821
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.300280636108514,
      "flow_matching_loss": 0.019947912823408843,
      "grad_norm": 0.09973379224538803,
      "learning_rate": 1.5060175178046016e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003777881385758519,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.7087963104248,
      "state_query_norm": 1433.621533203125,
      "state_relation_loss": 0.0003777881385758519,
      "step": 22770,
      "student_distance_mean": 1961.7191162109375,
      "teacher_distance_mean": 1.155271792411804,
      "weighted_auxiliary_loss": 0.0003777881385758519,
      "weighted_relation_loss": 0.0003777881385758519
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.309635173058933,
      "flow_matching_loss": 0.02069389009848237,
      "grad_norm": 0.11993826925754547,
      "learning_rate": 1.5020771050234117e-05,
      "loss": 0.0211,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040958723693620416,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4420467376709,
      "state_query_norm": 1415.4300903320313,
      "state_relation_loss": 0.00040958723693620416,
      "step": 22780,
      "student_distance_mean": 1946.1166748046876,
      "teacher_distance_mean": 1.1457237482070923,
      "weighted_auxiliary_loss": 0.00040958723693620416,
      "weighted_relation_loss": 0.00040958723693620416
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.318989710009355,
      "flow_matching_loss": 0.018810226023197173,
      "grad_norm": 0.09154392778873444,
      "learning_rate": 1.4981409425483716e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041324709018226714,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.569209289550784,
      "state_query_norm": 1422.133203125,
      "state_relation_loss": 0.00041324709018226714,
      "step": 22790,
      "student_distance_mean": 1953.328955078125,
      "teacher_distance_mean": 1.1505911946296692,
      "weighted_auxiliary_loss": 0.00041324709018226714,
      "weighted_relation_loss": 0.00041324709018226714
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.328344246959777,
      "flow_matching_loss": 0.016474564932286738,
      "grad_norm": 0.07124927639961243,
      "learning_rate": 1.4942090351622883e-05,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040602956141810864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54011421203613,
      "state_query_norm": 1419.5259887695313,
      "state_relation_loss": 0.00040602956141810864,
      "step": 22800,
      "student_distance_mean": 1949.2919555664062,
      "teacher_distance_mean": 1.1498793244361878,
      "weighted_auxiliary_loss": 0.00040602956141810864,
      "weighted_relation_loss": 0.00040602956141810864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.337698783910195,
      "flow_matching_loss": 0.018325484450906517,
      "grad_norm": 0.09368540346622467,
      "learning_rate": 1.490281387642804e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042336707992944865,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.332054901123044,
      "state_query_norm": 1420.5205078125,
      "state_relation_loss": 0.00042336707992944865,
      "step": 22810,
      "student_distance_mean": 1942.3190673828126,
      "teacher_distance_mean": 1.1462912917137147,
      "weighted_auxiliary_loss": 0.00042336707992944865,
      "weighted_relation_loss": 0.00042336707992944865
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.347053320860617,
      "flow_matching_loss": 0.018115509673953055,
      "grad_norm": 0.08818615227937698,
      "learning_rate": 1.4863580047623765e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040652632887940856,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.465385055541994,
      "state_query_norm": 1421.9859252929687,
      "state_relation_loss": 0.00040652632887940856,
      "step": 22820,
      "student_distance_mean": 1946.485986328125,
      "teacher_distance_mean": 1.1483060717582703,
      "weighted_auxiliary_loss": 0.00040652632887940856,
      "weighted_relation_loss": 0.00040652632887940856
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.35640785781104,
      "flow_matching_loss": 0.01716120485216379,
      "grad_norm": 0.0864412784576416,
      "learning_rate": 1.4824388912882897e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004035863501485437,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.707413864135745,
      "state_query_norm": 1430.6729614257813,
      "state_relation_loss": 0.0004035863501485437,
      "step": 22830,
      "student_distance_mean": 1962.0929321289063,
      "teacher_distance_mean": 1.155198884010315,
      "weighted_auxiliary_loss": 0.0004035863501485437,
      "weighted_relation_loss": 0.0004035863501485437
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.36576239476146,
      "flow_matching_loss": 0.018615574669092895,
      "grad_norm": 0.10677403211593628,
      "learning_rate": 1.4785240519826344e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004326364287408069,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62282447814941,
      "state_query_norm": 1430.192138671875,
      "state_relation_loss": 0.0004326364287408069,
      "step": 22840,
      "student_distance_mean": 1959.2795166015626,
      "teacher_distance_mean": 1.15267573595047,
      "weighted_auxiliary_loss": 0.0004326364287408069,
      "weighted_relation_loss": 0.0004326364287408069
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.37511693171188,
      "flow_matching_loss": 0.01660487512126565,
      "grad_norm": 0.07612587511539459,
      "learning_rate": 1.4746134916023097e-05,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040632194431964306,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93339920043945,
      "state_query_norm": 1434.7455932617188,
      "state_relation_loss": 0.00040632194431964306,
      "step": 22850,
      "student_distance_mean": 1972.006591796875,
      "teacher_distance_mean": 1.1595319032669067,
      "weighted_auxiliary_loss": 0.00040632194431964306,
      "weighted_relation_loss": 0.00040632194431964306
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.3844714686623,
      "flow_matching_loss": 0.018596018105745314,
      "grad_norm": 0.11565250903367996,
      "learning_rate": 1.4707072148990141e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040258138615172355,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9440731048584,
      "state_query_norm": 1434.510205078125,
      "state_relation_loss": 0.00040258138615172355,
      "step": 22860,
      "student_distance_mean": 1970.670361328125,
      "teacher_distance_mean": 1.158800756931305,
      "weighted_auxiliary_loss": 0.00040258138615172355,
      "weighted_relation_loss": 0.00040258138615172355
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.393826005612723,
      "flow_matching_loss": 0.018797215446829794,
      "grad_norm": 0.0987711101770401,
      "learning_rate": 1.4668052266192423e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037872660905122755,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.303335189819336,
      "state_query_norm": 1420.3711059570312,
      "state_relation_loss": 0.00037872660905122755,
      "step": 22870,
      "student_distance_mean": 1943.3320556640624,
      "teacher_distance_mean": 1.1450941801071166,
      "weighted_auxiliary_loss": 0.00037872660905122755,
      "weighted_relation_loss": 0.00037872660905122755
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.403180542563142,
      "flow_matching_loss": 0.019862366281449793,
      "grad_norm": 0.09318840503692627,
      "learning_rate": 1.4629075315042795e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003926362289348617,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.18498992919922,
      "state_query_norm": 1410.5912231445313,
      "state_relation_loss": 0.0003926362289348617,
      "step": 22880,
      "student_distance_mean": 1932.3521850585937,
      "teacher_distance_mean": 1.1401859879493714,
      "weighted_auxiliary_loss": 0.0003926362289348617,
      "weighted_relation_loss": 0.0003926362289348617
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.412535079513564,
      "flow_matching_loss": 0.020443871803581714,
      "grad_norm": 0.09325242787599564,
      "learning_rate": 1.4590141342901925e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004046312591526657,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.379751205444336,
      "state_query_norm": 1410.3860229492188,
      "state_relation_loss": 0.0004046312591526657,
      "step": 22890,
      "student_distance_mean": 1940.6892211914062,
      "teacher_distance_mean": 1.1471493601799012,
      "weighted_auxiliary_loss": 0.0004046312591526657,
      "weighted_relation_loss": 0.0004046312591526657
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.421889616463986,
      "flow_matching_loss": 0.019905539974570275,
      "grad_norm": 0.0973149910569191,
      "learning_rate": 1.4551250397078253e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004191572224954143,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43718376159668,
      "state_query_norm": 1423.2349731445313,
      "state_relation_loss": 0.0004191572224954143,
      "step": 22900,
      "student_distance_mean": 1950.601123046875,
      "teacher_distance_mean": 1.1509413838386535,
      "weighted_auxiliary_loss": 0.0004191572224954143,
      "weighted_relation_loss": 0.0004191572224954143
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.431244153414404,
      "flow_matching_loss": 0.018527305312454702,
      "grad_norm": 0.08665017783641815,
      "learning_rate": 1.4512402524827945e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004143363912589848,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.06306915283203,
      "state_query_norm": 1448.6419799804687,
      "state_relation_loss": 0.0004143363912589848,
      "step": 22910,
      "student_distance_mean": 1982.1593994140626,
      "teacher_distance_mean": 1.1692286133766174,
      "weighted_auxiliary_loss": 0.0004143363912589848,
      "weighted_relation_loss": 0.0004143363912589848
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.440598690364826,
      "flow_matching_loss": 0.01833038926124573,
      "grad_norm": 0.09488631039857864,
      "learning_rate": 1.447359777335482e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039013821224216373,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57079811096192,
      "state_query_norm": 1424.2919311523438,
      "state_relation_loss": 0.00039013821224216373,
      "step": 22920,
      "student_distance_mean": 1956.1679321289062,
      "teacher_distance_mean": 1.1550620317459106,
      "weighted_auxiliary_loss": 0.00039013821224216373,
      "weighted_relation_loss": 0.00039013821224216373
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.44995322731525,
      "flow_matching_loss": 0.018017451278865337,
      "grad_norm": 0.12438701093196869,
      "learning_rate": 1.4434836189810335e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004201377189019695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73964996337891,
      "state_query_norm": 1426.73115234375,
      "state_relation_loss": 0.0004201377189019695,
      "step": 22930,
      "student_distance_mean": 1959.9854614257813,
      "teacher_distance_mean": 1.1579857468605042,
      "weighted_auxiliary_loss": 0.0004201377189019695,
      "weighted_relation_loss": 0.0004201377189019695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.45930776426567,
      "flow_matching_loss": 0.01894404096528888,
      "grad_norm": 0.1123848482966423,
      "learning_rate": 1.4396117821293453e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003999307285994291,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.308105850219725,
      "state_query_norm": 1411.0818725585937,
      "state_relation_loss": 0.0003999307285994291,
      "step": 22940,
      "student_distance_mean": 1937.4861206054688,
      "teacher_distance_mean": 1.143810796737671,
      "weighted_auxiliary_loss": 0.0003999307285994291,
      "weighted_relation_loss": 0.0003999307285994291
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.46866230121609,
      "flow_matching_loss": 0.020169196650385857,
      "grad_norm": 0.08833765983581543,
      "learning_rate": 1.4357442714850633e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003746500617126003,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40720062255859,
      "state_query_norm": 1416.6028076171874,
      "state_relation_loss": 0.0003746500617126003,
      "step": 22950,
      "student_distance_mean": 1942.5453735351562,
      "teacher_distance_mean": 1.1473323941230773,
      "weighted_auxiliary_loss": 0.0003746500617126003,
      "weighted_relation_loss": 0.0003746500617126003
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.47801683816651,
      "flow_matching_loss": 0.01836325079202652,
      "grad_norm": 0.09468423575162888,
      "learning_rate": 1.43188109174758e-05,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00047118077636696396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.07597389221191,
      "state_query_norm": 1435.44072265625,
      "state_relation_loss": 0.00047118077636696396,
      "step": 22960,
      "student_distance_mean": 1973.268408203125,
      "teacher_distance_mean": 1.166421115398407,
      "weighted_auxiliary_loss": 0.00047118077636696396,
      "weighted_relation_loss": 0.00047118077636696396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.487371375116933,
      "flow_matching_loss": 0.020393156632781028,
      "grad_norm": 0.10172726213932037,
      "learning_rate": 1.4280222476110206e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004165460966760293,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5822509765625,
      "state_query_norm": 1426.2433471679688,
      "state_relation_loss": 0.0004165460966760293,
      "step": 22970,
      "student_distance_mean": 1956.5917846679688,
      "teacher_distance_mean": 1.1548826098442078,
      "weighted_auxiliary_loss": 0.0004165460966760293,
      "weighted_relation_loss": 0.0004165460966760293
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.49672591206735,
      "flow_matching_loss": 0.018608373962342738,
      "grad_norm": 0.09917981922626495,
      "learning_rate": 1.4241677437642469e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043307341111358254,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91825828552246,
      "state_query_norm": 1437.1720092773437,
      "state_relation_loss": 0.00043307341111358254,
      "step": 22980,
      "student_distance_mean": 1971.8011962890625,
      "teacher_distance_mean": 1.1622985124588012,
      "weighted_auxiliary_loss": 0.00043307341111358254,
      "weighted_relation_loss": 0.00043307341111358254
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.506080449017773,
      "flow_matching_loss": 0.018373480066657066,
      "grad_norm": 0.10159043222665787,
      "learning_rate": 1.420317584890844e-05,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004166300088400021,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45980529785156,
      "state_query_norm": 1417.26435546875,
      "state_relation_loss": 0.0004166300088400021,
      "step": 22990,
      "student_distance_mean": 1946.1855102539062,
      "teacher_distance_mean": 1.1456886768341064,
      "weighted_auxiliary_loss": 0.0004166300088400021,
      "weighted_relation_loss": 0.0004166300088400021
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.515434985968195,
      "flow_matching_loss": 0.018494314141571523,
      "grad_norm": 0.07652415335178375,
      "learning_rate": 1.4164717756691176e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004320678039221093,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58725433349609,
      "state_query_norm": 1432.8511108398438,
      "state_relation_loss": 0.0004320678039221093,
      "step": 23000,
      "student_distance_mean": 1956.8497802734375,
      "teacher_distance_mean": 1.1509159564971925,
      "weighted_auxiliary_loss": 0.0004320678039221093,
      "weighted_relation_loss": 0.0004320678039221093
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.524789522918617,
      "flow_matching_loss": 0.0202268966473639,
      "grad_norm": 0.10256031155586243,
      "learning_rate": 1.4126303207720882e-05,
      "loss": 0.0206,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037128516123630106,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.434444427490234,
      "state_query_norm": 1430.1882446289062,
      "state_relation_loss": 0.00037128516123630106,
      "step": 23010,
      "student_distance_mean": 1956.4338256835938,
      "teacher_distance_mean": 1.1492375135421753,
      "weighted_auxiliary_loss": 0.00037128516123630106,
      "weighted_relation_loss": 0.00037128516123630106
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.534144059869035,
      "flow_matching_loss": 0.02053738757967949,
      "grad_norm": 0.0918244943022728,
      "learning_rate": 1.408793224867484e-05,
      "loss": 0.021,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004247637174557894,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.05885200500488,
      "state_query_norm": 1437.7929321289062,
      "state_relation_loss": 0.0004247637174557894,
      "step": 23020,
      "student_distance_mean": 1976.307421875,
      "teacher_distance_mean": 1.1625140309333801,
      "weighted_auxiliary_loss": 0.0004247637174557894,
      "weighted_relation_loss": 0.0004247637174557894
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.543498596819457,
      "flow_matching_loss": 0.020411197375506163,
      "grad_norm": 0.09718386083841324,
      "learning_rate": 1.4049604926177423e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038799362373538316,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.208464813232425,
      "state_query_norm": 1412.5034301757812,
      "state_relation_loss": 0.00038799362373538316,
      "step": 23030,
      "student_distance_mean": 1934.265771484375,
      "teacher_distance_mean": 1.1382667541503906,
      "weighted_auxiliary_loss": 0.00038799362373538316,
      "weighted_relation_loss": 0.00038799362373538316
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.55285313376988,
      "flow_matching_loss": 0.020033251866698264,
      "grad_norm": 0.08765393495559692,
      "learning_rate": 1.4011321286799916e-05,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000430011420394294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47266387939453,
      "state_query_norm": 1424.14638671875,
      "state_relation_loss": 0.000430011420394294,
      "step": 23040,
      "student_distance_mean": 1948.8914916992187,
      "teacher_distance_mean": 1.149849569797516,
      "weighted_auxiliary_loss": 0.000430011420394294,
      "weighted_relation_loss": 0.000430011420394294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.562207670720298,
      "flow_matching_loss": 0.0179429798386991,
      "grad_norm": 0.08526160567998886,
      "learning_rate": 1.3973081377060565e-05,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003996609302703291,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02442245483398,
      "state_query_norm": 1442.9833251953125,
      "state_relation_loss": 0.0003996609302703291,
      "step": 23050,
      "student_distance_mean": 1980.5191772460937,
      "teacher_distance_mean": 1.1686772584915162,
      "weighted_auxiliary_loss": 0.0003996609302703291,
      "weighted_relation_loss": 0.0003996609302703291
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.57156220767072,
      "flow_matching_loss": 0.020282992627471685,
      "grad_norm": 0.0960473045706749,
      "learning_rate": 1.393488524342445e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004170821426669136,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.779798889160155,
      "state_query_norm": 1431.0388427734374,
      "state_relation_loss": 0.0004170821426669136,
      "step": 23060,
      "student_distance_mean": 1964.6882934570312,
      "teacher_distance_mean": 1.1584634184837341,
      "weighted_auxiliary_loss": 0.0004170821426669136,
      "weighted_relation_loss": 0.0004170821426669136
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.580916744621142,
      "flow_matching_loss": 0.020390089228749277,
      "grad_norm": 0.08238334953784943,
      "learning_rate": 1.3896732932303486e-05,
      "loss": 0.0208,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039706841635052116,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55788154602051,
      "state_query_norm": 1425.9926147460938,
      "state_relation_loss": 0.00039706841635052116,
      "step": 23070,
      "student_distance_mean": 1955.8520141601562,
      "teacher_distance_mean": 1.1549856424331666,
      "weighted_auxiliary_loss": 0.00039706841635052116,
      "weighted_relation_loss": 0.00039706841635052116
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.590271281571564,
      "flow_matching_loss": 0.017611523903906346,
      "grad_norm": 0.09953396767377853,
      "learning_rate": 1.3858624490056304e-05,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040641474479343744,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36942367553711,
      "state_query_norm": 1411.0109497070312,
      "state_relation_loss": 0.00040641474479343744,
      "step": 23080,
      "student_distance_mean": 1942.5929565429688,
      "teacher_distance_mean": 1.1468736410140992,
      "weighted_auxiliary_loss": 0.00040641474479343744,
      "weighted_relation_loss": 0.00040641474479343744
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.599625818521982,
      "flow_matching_loss": 0.01733544683083892,
      "grad_norm": 0.09169793874025345,
      "learning_rate": 1.3820559962988299e-05,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000411334348609671,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75585098266602,
      "state_query_norm": 1429.8258056640625,
      "state_relation_loss": 0.000411334348609671,
      "step": 23090,
      "student_distance_mean": 1962.6007690429688,
      "teacher_distance_mean": 1.157976996898651,
      "weighted_auxiliary_loss": 0.000411334348609671,
      "weighted_relation_loss": 0.000411334348609671
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.608980355472404,
      "flow_matching_loss": 0.019262533728033306,
      "grad_norm": 0.10380039364099503,
      "learning_rate": 1.3782539397351418e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003756152407731861,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55154647827148,
      "state_query_norm": 1421.0697387695313,
      "state_relation_loss": 0.0003756152407731861,
      "step": 23100,
      "student_distance_mean": 1952.6455444335938,
      "teacher_distance_mean": 1.1528141856193543,
      "weighted_auxiliary_loss": 0.0003756152407731861,
      "weighted_relation_loss": 0.0003756152407731861
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.618334892422826,
      "flow_matching_loss": 0.01912593301385641,
      "grad_norm": 0.09534759819507599,
      "learning_rate": 1.3744562839344266e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004201198229566216,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.68678970336914,
      "state_query_norm": 1426.0411987304688,
      "state_relation_loss": 0.0004201198229566216,
      "step": 23110,
      "student_distance_mean": 1959.0432861328125,
      "teacher_distance_mean": 1.156431293487549,
      "weighted_auxiliary_loss": 0.0004201198229566216,
      "weighted_relation_loss": 0.0004201198229566216
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.627689429373245,
      "flow_matching_loss": 0.019241308514028787,
      "grad_norm": 0.09195753186941147,
      "learning_rate": 1.370663033511193e-05,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003868296364089474,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83700180053711,
      "state_query_norm": 1432.7073486328125,
      "state_relation_loss": 0.0003868296364089474,
      "step": 23120,
      "student_distance_mean": 1969.1538330078124,
      "teacher_distance_mean": 1.1603371381759644,
      "weighted_auxiliary_loss": 0.0003868296364089474,
      "weighted_relation_loss": 0.0003868296364089474
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.637043966323667,
      "flow_matching_loss": 0.018508054688572882,
      "grad_norm": 0.06761063635349274,
      "learning_rate": 1.3668741930745964e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004309723764890805,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.491604232788085,
      "state_query_norm": 1418.13916015625,
      "state_relation_loss": 0.0004309723764890805,
      "step": 23130,
      "student_distance_mean": 1947.0080810546874,
      "teacher_distance_mean": 1.1490333795547485,
      "weighted_auxiliary_loss": 0.0004309723764890805,
      "weighted_relation_loss": 0.0004309723764890805
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.64639850327409,
      "flow_matching_loss": 0.01766735501587391,
      "grad_norm": 0.09229181706905365,
      "learning_rate": 1.3630897672284382e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004440465825609863,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83865509033203,
      "state_query_norm": 1425.9498291015625,
      "state_relation_loss": 0.0004440465825609863,
      "step": 23140,
      "student_distance_mean": 1964.1779907226562,
      "teacher_distance_mean": 1.159588062763214,
      "weighted_auxiliary_loss": 0.0004440465825609863,
      "weighted_relation_loss": 0.0004440465825609863
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.65575304022451,
      "flow_matching_loss": 0.01808273494243622,
      "grad_norm": 0.09514366090297699,
      "learning_rate": 1.3593097605711508e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003899340517818928,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37317123413086,
      "state_query_norm": 1412.7003540039063,
      "state_relation_loss": 0.0003899340517818928,
      "step": 23150,
      "student_distance_mean": 1942.7689453125,
      "teacher_distance_mean": 1.1453051209449767,
      "weighted_auxiliary_loss": 0.0003899340517818928,
      "weighted_relation_loss": 0.0003899340517818928
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.66510757717493,
      "flow_matching_loss": 0.01989467889070511,
      "grad_norm": 0.09395824372768402,
      "learning_rate": 1.3555341776957992e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040068412781693043,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47711868286133,
      "state_query_norm": 1419.84951171875,
      "state_relation_loss": 0.00040068412781693043,
      "step": 23160,
      "student_distance_mean": 1945.9736450195312,
      "teacher_distance_mean": 1.1455374836921692,
      "weighted_auxiliary_loss": 0.00040068412781693043,
      "weighted_relation_loss": 0.00040068412781693043
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.67446211412535,
      "flow_matching_loss": 0.018059827480465175,
      "grad_norm": 0.08876057714223862,
      "learning_rate": 1.3517630231900724e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004089753812877461,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.731641387939455,
      "state_query_norm": 1431.9277221679688,
      "state_relation_loss": 0.0004089753812877461,
      "step": 23170,
      "student_distance_mean": 1965.8156616210938,
      "teacher_distance_mean": 1.1583903074264525,
      "weighted_auxiliary_loss": 0.0004089753812877461,
      "weighted_relation_loss": 0.0004089753812877461
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.683816651075773,
      "flow_matching_loss": 0.017147470079362393,
      "grad_norm": 0.08915884792804718,
      "learning_rate": 1.3479963016362767e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004293820820748806,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.766340255737305,
      "state_query_norm": 1419.5300170898438,
      "state_relation_loss": 0.0004293820820748806,
      "step": 23180,
      "student_distance_mean": 1960.9015258789063,
      "teacher_distance_mean": 1.1561836123466491,
      "weighted_auxiliary_loss": 0.0004293820820748806,
      "weighted_relation_loss": 0.0004293820820748806
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.69317118802619,
      "flow_matching_loss": 0.018527308106422426,
      "grad_norm": 0.08103885501623154,
      "learning_rate": 1.3442340176113377e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003988663986092433,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.543822860717775,
      "state_query_norm": 1420.0450805664063,
      "state_relation_loss": 0.0003988663986092433,
      "step": 23190,
      "student_distance_mean": 1956.18955078125,
      "teacher_distance_mean": 1.1525065422058105,
      "weighted_auxiliary_loss": 0.0003988663986092433,
      "weighted_relation_loss": 0.0003988663986092433
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.702525724976613,
      "flow_matching_loss": 0.019645773246884345,
      "grad_norm": 0.08527937531471252,
      "learning_rate": 1.340476175686784e-05,
      "loss": 0.02,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039480863488279284,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.572989273071286,
      "state_query_norm": 1426.6864624023438,
      "state_relation_loss": 0.00039480863488279284,
      "step": 23200,
      "student_distance_mean": 1953.9098999023438,
      "teacher_distance_mean": 1.1530846118927003,
      "weighted_auxiliary_loss": 0.00039480863488279284,
      "weighted_relation_loss": 0.00039480863488279284
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.711880261927035,
      "flow_matching_loss": 0.01933477194979787,
      "grad_norm": 0.09594640135765076,
      "learning_rate": 1.3367227804287469e-05,
      "loss": 0.0197,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004020876804133877,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.532915115356445,
      "state_query_norm": 1426.1339599609375,
      "state_relation_loss": 0.0004020876804133877,
      "step": 23210,
      "student_distance_mean": 1950.4169067382813,
      "teacher_distance_mean": 1.1510636448860168,
      "weighted_auxiliary_loss": 0.0004020876804133877,
      "weighted_relation_loss": 0.0004020876804133877
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.721234798877454,
      "flow_matching_loss": 0.019514417741447687,
      "grad_norm": 0.07491262257099152,
      "learning_rate": 1.332973836397956e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003907564183464274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72737808227539,
      "state_query_norm": 1435.1344848632812,
      "state_relation_loss": 0.0003907564183464274,
      "step": 23220,
      "student_distance_mean": 1964.1354125976563,
      "teacher_distance_mean": 1.1586501479148865,
      "weighted_auxiliary_loss": 0.0003907564183464274,
      "weighted_relation_loss": 0.0003907564183464274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.730589335827876,
      "flow_matching_loss": 0.019535781256854533,
      "grad_norm": 0.07943656295537949,
      "learning_rate": 1.329229348149731e-05,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040545636147726327,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.68459014892578,
      "state_query_norm": 1433.4140747070312,
      "state_relation_loss": 0.00040545636147726327,
      "step": 23230,
      "student_distance_mean": 1959.7411376953125,
      "teacher_distance_mean": 1.1570050358772277,
      "weighted_auxiliary_loss": 0.00040545636147726327,
      "weighted_relation_loss": 0.00040545636147726327
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.739943872778298,
      "flow_matching_loss": 0.018954061344265936,
      "grad_norm": 0.08842640370130539,
      "learning_rate": 1.3254893202339796e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040506756049580873,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4353759765625,
      "state_query_norm": 1409.7332153320312,
      "state_relation_loss": 0.00040506756049580873,
      "step": 23240,
      "student_distance_mean": 1941.0119995117188,
      "teacher_distance_mean": 1.145989179611206,
      "weighted_auxiliary_loss": 0.00040506756049580873,
      "weighted_relation_loss": 0.00040506756049580873
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.74929840972872,
      "flow_matching_loss": 0.019810287281870842,
      "grad_norm": 0.08534104377031326,
      "learning_rate": 1.3217537571951872e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041233814263250677,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63620223999023,
      "state_query_norm": 1429.8585083007813,
      "state_relation_loss": 0.00041233814263250677,
      "step": 23250,
      "student_distance_mean": 1958.4988159179688,
      "teacher_distance_mean": 1.1547455191612244,
      "weighted_auxiliary_loss": 0.00041233814263250677,
      "weighted_relation_loss": 0.00041233814263250677
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.75865294667914,
      "flow_matching_loss": 0.018322011362761258,
      "grad_norm": 0.0965595468878746,
      "learning_rate": 1.3180226635724169e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038126512372400613,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75734634399414,
      "state_query_norm": 1400.9942993164063,
      "state_relation_loss": 0.00038126512372400613,
      "step": 23260,
      "student_distance_mean": 1914.8716552734375,
      "teacher_distance_mean": 1.1281619310379027,
      "weighted_auxiliary_loss": 0.00038126512372400613,
      "weighted_relation_loss": 0.00038126512372400613
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.76800748362956,
      "flow_matching_loss": 0.019067887030541898,
      "grad_norm": 0.11763026565313339,
      "learning_rate": 1.314296043899298e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041359044262208046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.34653930664062,
      "state_query_norm": 1418.0749877929688,
      "state_relation_loss": 0.00041359044262208046,
      "step": 23270,
      "student_distance_mean": 1942.7862426757813,
      "teacher_distance_mean": 1.1455548763275147,
      "weighted_auxiliary_loss": 0.00041359044262208046,
      "weighted_relation_loss": 0.00041359044262208046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.777362020579982,
      "flow_matching_loss": 0.018357479944825172,
      "grad_norm": 0.08776163309812546,
      "learning_rate": 1.3105739027040248e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039156455895863476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2131290435791,
      "state_query_norm": 1415.3788330078125,
      "state_relation_loss": 0.00039156455895863476,
      "step": 23280,
      "student_distance_mean": 1938.8316772460937,
      "teacher_distance_mean": 1.143715488910675,
      "weighted_auxiliary_loss": 0.00039156455895863476,
      "weighted_relation_loss": 0.00039156455895863476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.7867165575304,
      "flow_matching_loss": 0.018468542117625476,
      "grad_norm": 0.09453895688056946,
      "learning_rate": 1.3068562445093491e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039717557665426285,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.80436706542969,
      "state_query_norm": 1431.57744140625,
      "state_relation_loss": 0.00039717557665426285,
      "step": 23290,
      "student_distance_mean": 1966.139111328125,
      "teacher_distance_mean": 1.1606748342514037,
      "weighted_auxiliary_loss": 0.00039717557665426285,
      "weighted_relation_loss": 0.00039717557665426285
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.796071094480823,
      "flow_matching_loss": 0.019828915316611528,
      "grad_norm": 0.09201982617378235,
      "learning_rate": 1.3031430738325778e-05,
      "loss": 0.0202,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004173768014879897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62228164672852,
      "state_query_norm": 1426.1587280273438,
      "state_relation_loss": 0.0004173768014879897,
      "step": 23300,
      "student_distance_mean": 1958.1178466796875,
      "teacher_distance_mean": 1.1553236961364746,
      "weighted_auxiliary_loss": 0.0004173768014879897,
      "weighted_relation_loss": 0.0004173768014879897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.805425631431245,
      "flow_matching_loss": 0.018373100832104684,
      "grad_norm": 0.10623282194137573,
      "learning_rate": 1.299434395185563e-05,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040218921203631907,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30225105285645,
      "state_query_norm": 1415.6459838867188,
      "state_relation_loss": 0.00040218921203631907,
      "step": 23310,
      "student_distance_mean": 1939.1827880859375,
      "teacher_distance_mean": 1.1442331433296205,
      "weighted_auxiliary_loss": 0.00040218921203631907,
      "weighted_relation_loss": 0.00040218921203631907
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.814780168381667,
      "flow_matching_loss": 0.017698325216770172,
      "grad_norm": 0.07916810363531113,
      "learning_rate": 1.295730213074699e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041134652274195106,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53724479675293,
      "state_query_norm": 1425.8080810546876,
      "state_relation_loss": 0.00041134652274195106,
      "step": 23320,
      "student_distance_mean": 1952.1989013671875,
      "teacher_distance_mean": 1.1512538194656372,
      "weighted_auxiliary_loss": 0.00041134652274195106,
      "weighted_relation_loss": 0.00041134652274195106
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.824134705332085,
      "flow_matching_loss": 0.018540273141115903,
      "grad_norm": 0.07200471311807632,
      "learning_rate": 1.2920305320009152e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004169200372416526,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.918300247192384,
      "state_query_norm": 1436.280810546875,
      "state_relation_loss": 0.0004169200372416526,
      "step": 23330,
      "student_distance_mean": 1969.4732666015625,
      "teacher_distance_mean": 1.16247318983078,
      "weighted_auxiliary_loss": 0.0004169200372416526,
      "weighted_relation_loss": 0.0004169200372416526
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.833489242282507,
      "flow_matching_loss": 0.018220388516783714,
      "grad_norm": 0.0954834520816803,
      "learning_rate": 1.2883353564596729e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040131195273716,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54037971496582,
      "state_query_norm": 1421.0762451171875,
      "state_relation_loss": 0.00040131195273716,
      "step": 23340,
      "student_distance_mean": 1951.72421875,
      "teacher_distance_mean": 1.1513964533805847,
      "weighted_auxiliary_loss": 0.00040131195273716,
      "weighted_relation_loss": 0.00040131195273716
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.84284377923293,
      "flow_matching_loss": 0.020281131006777288,
      "grad_norm": 0.08869405090808868,
      "learning_rate": 1.2846446909409604e-05,
      "loss": 0.0207,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041274836112279443,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2675163269043,
      "state_query_norm": 1412.7418701171875,
      "state_relation_loss": 0.00041274836112279443,
      "step": 23350,
      "student_distance_mean": 1937.72646484375,
      "teacher_distance_mean": 1.143786358833313,
      "weighted_auxiliary_loss": 0.00041274836112279443,
      "weighted_relation_loss": 0.00041274836112279443
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.852198316183348,
      "flow_matching_loss": 0.01815860904753208,
      "grad_norm": 0.08262408524751663,
      "learning_rate": 1.280958539929284e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004177716007689014,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.07236289978027,
      "state_query_norm": 1439.8873046875,
      "state_relation_loss": 0.0004177716007689014,
      "step": 23360,
      "student_distance_mean": 1980.3608642578124,
      "teacher_distance_mean": 1.1686641931533814,
      "weighted_auxiliary_loss": 0.0004177716007689014,
      "weighted_relation_loss": 0.0004177716007689014
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.86155285313377,
      "flow_matching_loss": 0.018030404392629863,
      "grad_norm": 0.10258086025714874,
      "learning_rate": 1.277276907903664e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004179229814326391,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.95786666870117,
      "state_query_norm": 1437.6880859375,
      "state_relation_loss": 0.0004179229814326391,
      "step": 23370,
      "student_distance_mean": 1976.2977783203125,
      "teacher_distance_mean": 1.1653284192085267,
      "weighted_auxiliary_loss": 0.0004179229814326391,
      "weighted_relation_loss": 0.0004179229814326391
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.87090739008419,
      "flow_matching_loss": 0.019728250801563263,
      "grad_norm": 0.07430948317050934,
      "learning_rate": 1.2735997993376298e-05,
      "loss": 0.0201,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003941468836273998,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.253651809692386,
      "state_query_norm": 1418.8546020507813,
      "state_relation_loss": 0.0003941468836273998,
      "step": 23380,
      "student_distance_mean": 1941.9509399414062,
      "teacher_distance_mean": 1.1455213665962218,
      "weighted_auxiliary_loss": 0.0003941468836273998,
      "weighted_relation_loss": 0.0003941468836273998
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.880261927034613,
      "flow_matching_loss": 0.018116851430386306,
      "grad_norm": 0.059321969747543335,
      "learning_rate": 1.2699272186992166e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039911941275931894,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62219886779785,
      "state_query_norm": 1428.4208374023438,
      "state_relation_loss": 0.00039911941275931894,
      "step": 23390,
      "student_distance_mean": 1959.1461669921875,
      "teacher_distance_mean": 1.1550105929374694,
      "weighted_auxiliary_loss": 0.00039911941275931894,
      "weighted_relation_loss": 0.00039911941275931894
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.889616463985032,
      "flow_matching_loss": 0.01720460122451186,
      "grad_norm": 0.08415617048740387,
      "learning_rate": 1.2662591704509547e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040387748449575154,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48413238525391,
      "state_query_norm": 1421.0810668945312,
      "state_relation_loss": 0.00040387748449575154,
      "step": 23400,
      "student_distance_mean": 1951.6514770507813,
      "teacher_distance_mean": 1.1532945513725281,
      "weighted_auxiliary_loss": 0.00040387748449575154,
      "weighted_relation_loss": 0.00040387748449575154
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.898971000935454,
      "flow_matching_loss": 0.019863308407366276,
      "grad_norm": 0.07874079793691635,
      "learning_rate": 1.2625956590498711e-05,
      "loss": 0.0203,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004277550877304748,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83874282836914,
      "state_query_norm": 1433.3744995117188,
      "state_relation_loss": 0.0004277550877304748,
      "step": 23410,
      "student_distance_mean": 1968.9291870117188,
      "teacher_distance_mean": 1.164410448074341,
      "weighted_auxiliary_loss": 0.0004277550877304748,
      "weighted_relation_loss": 0.0004277550877304748
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.908325537885876,
      "flow_matching_loss": 0.017446027789264917,
      "grad_norm": 0.08592774718999863,
      "learning_rate": 1.2589366889474757e-05,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039965298783499746,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.742056655883786,
      "state_query_norm": 1425.1588500976563,
      "state_relation_loss": 0.00039965298783499746,
      "step": 23420,
      "student_distance_mean": 1962.6454833984376,
      "teacher_distance_mean": 1.1592713713645935,
      "weighted_auxiliary_loss": 0.00039965298783499746,
      "weighted_relation_loss": 0.00039965298783499746
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.917680074836294,
      "flow_matching_loss": 0.019098515808582305,
      "grad_norm": 0.07448247820138931,
      "learning_rate": 1.2552822645897621e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037742513814009726,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.069265365600586,
      "state_query_norm": 1405.2828002929687,
      "state_relation_loss": 0.00037742513814009726,
      "step": 23430,
      "student_distance_mean": 1928.1567016601562,
      "teacher_distance_mean": 1.1377313256263732,
      "weighted_auxiliary_loss": 0.00037742513814009726,
      "weighted_relation_loss": 0.00037742513814009726
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.927034611786716,
      "flow_matching_loss": 0.019016185123473407,
      "grad_norm": 0.09487727284431458,
      "learning_rate": 1.2516323904172e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040839622088242324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.19462852478027,
      "state_query_norm": 1409.154150390625,
      "state_relation_loss": 0.00040839622088242324,
      "step": 23440,
      "student_distance_mean": 1933.803076171875,
      "teacher_distance_mean": 1.1407883763313293,
      "weighted_auxiliary_loss": 0.00040839622088242324,
      "weighted_relation_loss": 0.00040839622088242324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.93638914873714,
      "flow_matching_loss": 0.018917241226881744,
      "grad_norm": 0.09079942107200623,
      "learning_rate": 1.2479870708647324e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037751720810774714,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39197845458985,
      "state_query_norm": 1414.8810546875,
      "state_relation_loss": 0.00037751720810774714,
      "step": 23450,
      "student_distance_mean": 1946.3851196289063,
      "teacher_distance_mean": 1.1483560800552368,
      "weighted_auxiliary_loss": 0.00037751720810774714,
      "weighted_relation_loss": 0.00037751720810774714
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.94574368568756,
      "flow_matching_loss": 0.01873882096260786,
      "grad_norm": 0.08996216952800751,
      "learning_rate": 1.2443463103617658e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000441300924285315,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6609504699707,
      "state_query_norm": 1425.7254028320312,
      "state_relation_loss": 0.000441300924285315,
      "step": 23460,
      "student_distance_mean": 1954.0554565429688,
      "teacher_distance_mean": 1.152008628845215,
      "weighted_auxiliary_loss": 0.000441300924285315,
      "weighted_relation_loss": 0.000441300924285315
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.95509822263798,
      "flow_matching_loss": 0.017038153763860465,
      "grad_norm": 0.08120378851890564,
      "learning_rate": 1.240710113332167e-05,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003887875034706667,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.894335174560545,
      "state_query_norm": 1446.9000244140625,
      "state_relation_loss": 0.0003887875034706667,
      "step": 23470,
      "student_distance_mean": 1976.5355102539063,
      "teacher_distance_mean": 1.1659364581108094,
      "weighted_auxiliary_loss": 0.0003887875034706667,
      "weighted_relation_loss": 0.0003887875034706667
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.9644527595884,
      "flow_matching_loss": 0.018815593514591455,
      "grad_norm": 0.0906626433134079,
      "learning_rate": 1.2370784841942584e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004373405652586371,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.294280242919925,
      "state_query_norm": 1414.6801513671876,
      "state_relation_loss": 0.0004373405652586371,
      "step": 23480,
      "student_distance_mean": 1936.6993896484375,
      "teacher_distance_mean": 1.1427284359931946,
      "weighted_auxiliary_loss": 0.0004373405652586371,
      "weighted_relation_loss": 0.0004373405652586371
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.973807296538823,
      "flow_matching_loss": 0.018298377655446528,
      "grad_norm": 0.08468882739543915,
      "learning_rate": 1.2334514273608117e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003977862710598856,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6287181854248,
      "state_query_norm": 1423.2236938476562,
      "state_relation_loss": 0.0003977862710598856,
      "step": 23490,
      "student_distance_mean": 1954.8974243164062,
      "teacher_distance_mean": 1.1521186828613281,
      "weighted_auxiliary_loss": 0.0003977862710598856,
      "weighted_relation_loss": 0.0003977862710598856
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.98316183348924,
      "flow_matching_loss": 0.019144289568066596,
      "grad_norm": 0.08667301386594772,
      "learning_rate": 1.2298289472390417e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003906826517777517,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39011459350586,
      "state_query_norm": 1419.8757568359374,
      "state_relation_loss": 0.0003906826517777517,
      "step": 23500,
      "student_distance_mean": 1945.0548706054688,
      "teacher_distance_mean": 1.1459816694259644,
      "weighted_auxiliary_loss": 0.0003906826517777517,
      "weighted_relation_loss": 0.0003906826517777517
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 21.992516370439663,
      "flow_matching_loss": 0.01824487801641226,
      "grad_norm": 0.08829186111688614,
      "learning_rate": 1.226211048230606e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003994663042249158,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.381429290771486,
      "state_query_norm": 1419.8364868164062,
      "state_relation_loss": 0.0003994663042249158,
      "step": 23510,
      "student_distance_mean": 1944.9455688476562,
      "teacher_distance_mean": 1.147159457206726,
      "weighted_auxiliary_loss": 0.0003994663042249158,
      "weighted_relation_loss": 0.0003994663042249158
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.001870907390085,
      "flow_matching_loss": 0.017259075492620467,
      "grad_norm": 0.08790100365877151,
      "learning_rate": 1.2225977347315921e-05,
      "loss": 0.0177,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0004019701882498339,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.16412239074707,
      "state_query_norm": 1422.9867919921876,
      "state_relation_loss": 0.0004019701882498339,
      "step": 23520,
      "student_distance_mean": 1939.6092651367187,
      "teacher_distance_mean": 1.1443484425544739,
      "weighted_auxiliary_loss": 0.0004019701882498339,
      "weighted_relation_loss": 0.0004019701882498339
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.011225444340504,
      "flow_matching_loss": 0.0186353019438684,
      "grad_norm": 0.08642327785491943,
      "learning_rate": 1.2189890111325148e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003886931954184547,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.19771537780762,
      "state_query_norm": 1403.7040405273438,
      "state_relation_loss": 0.0003886931954184547,
      "step": 23530,
      "student_distance_mean": 1930.8096069335938,
      "teacher_distance_mean": 1.1406014561653137,
      "weighted_auxiliary_loss": 0.0003886931954184547,
      "weighted_relation_loss": 0.0003886931954184547
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.020579981290926,
      "flow_matching_loss": 0.019055445026606322,
      "grad_norm": 0.09370727092027664,
      "learning_rate": 1.2153848818183161e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040613803721498696,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23396949768066,
      "state_query_norm": 1417.2086547851563,
      "state_relation_loss": 0.00040613803721498696,
      "step": 23540,
      "student_distance_mean": 1935.2728271484375,
      "teacher_distance_mean": 1.1433407425880433,
      "weighted_auxiliary_loss": 0.00040613803721498696,
      "weighted_relation_loss": 0.00040613803721498696
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.029934518241348,
      "flow_matching_loss": 0.01763774883002043,
      "grad_norm": 0.10850785672664642,
      "learning_rate": 1.2117853511683507e-05,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040860395529307425,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.486285018920896,
      "state_query_norm": 1418.1383911132812,
      "state_relation_loss": 0.00040860395529307425,
      "step": 23550,
      "student_distance_mean": 1947.2342895507813,
      "teacher_distance_mean": 1.1503713846206665,
      "weighted_auxiliary_loss": 0.00040860395529307425,
      "weighted_relation_loss": 0.00040860395529307425
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.03928905519177,
      "flow_matching_loss": 0.017219896987080575,
      "grad_norm": 0.09361009299755096,
      "learning_rate": 1.2081904235563906e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003814783849520609,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.73715057373047,
      "state_query_norm": 1395.5692260742187,
      "state_relation_loss": 0.0003814783849520609,
      "step": 23560,
      "student_distance_mean": 1912.6940551757812,
      "teacher_distance_mean": 1.1302446365356444,
      "weighted_auxiliary_loss": 0.0003814783849520609,
      "weighted_relation_loss": 0.0003814783849520609
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.048643592142188,
      "flow_matching_loss": 0.018042386323213578,
      "grad_norm": 0.07931903004646301,
      "learning_rate": 1.20460010335061e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004100311314687133,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.770759201049806,
      "state_query_norm": 1426.997119140625,
      "state_relation_loss": 0.0004100311314687133,
      "step": 23570,
      "student_distance_mean": 1961.3454467773438,
      "teacher_distance_mean": 1.1574193120002747,
      "weighted_auxiliary_loss": 0.0004100311314687133,
      "weighted_relation_loss": 0.0004100311314687133
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.05799812909261,
      "flow_matching_loss": 0.017654758878052235,
      "grad_norm": 0.08867126703262329,
      "learning_rate": 1.2010143949135865e-05,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039254642033483835,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74273452758789,
      "state_query_norm": 1432.927587890625,
      "state_relation_loss": 0.00039254642033483835,
      "step": 23580,
      "student_distance_mean": 1963.5813720703125,
      "teacher_distance_mean": 1.1581810355186462,
      "weighted_auxiliary_loss": 0.00039254642033483835,
      "weighted_relation_loss": 0.00039254642033483835
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.067352666043032,
      "flow_matching_loss": 0.01858250852674246,
      "grad_norm": 0.10016478598117828,
      "learning_rate": 1.1974333026022938e-05,
      "loss": 0.019,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042139757424592973,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93693733215332,
      "state_query_norm": 1431.0071899414063,
      "state_relation_loss": 0.00042139757424592973,
      "step": 23590,
      "student_distance_mean": 1972.312890625,
      "teacher_distance_mean": 1.1631363868713378,
      "weighted_auxiliary_loss": 0.00042139757424592973,
      "weighted_relation_loss": 0.00042139757424592973
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.07670720299345,
      "flow_matching_loss": 0.01915807034820318,
      "grad_norm": 0.0932859405875206,
      "learning_rate": 1.1938568307680964e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038998790842015296,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1275016784668,
      "state_query_norm": 1414.4045532226562,
      "state_relation_loss": 0.00038998790842015296,
      "step": 23600,
      "student_distance_mean": 1931.7931030273437,
      "teacher_distance_mean": 1.1393348813056945,
      "weighted_auxiliary_loss": 0.00038998790842015296,
      "weighted_relation_loss": 0.00038998790842015296
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.086061739943872,
      "flow_matching_loss": 0.017374839819967746,
      "grad_norm": 0.07533232867717743,
      "learning_rate": 1.1902849837567464e-05,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044480281649157406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72412033081055,
      "state_query_norm": 1426.7105102539062,
      "state_relation_loss": 0.00044480281649157406,
      "step": 23610,
      "student_distance_mean": 1959.1155517578125,
      "teacher_distance_mean": 1.1578118562698365,
      "weighted_auxiliary_loss": 0.00044480281649157406,
      "weighted_relation_loss": 0.00044480281649157406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.095416276894294,
      "flow_matching_loss": 0.018516538850963115,
      "grad_norm": 0.11796894669532776,
      "learning_rate": 1.1867177659083739e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004132635542191565,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65773429870605,
      "state_query_norm": 1430.9711791992188,
      "state_relation_loss": 0.0004132635542191565,
      "step": 23620,
      "student_distance_mean": 1960.01201171875,
      "teacher_distance_mean": 1.1585474252700805,
      "weighted_auxiliary_loss": 0.0004132635542191565,
      "weighted_relation_loss": 0.0004132635542191565
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.104770813844716,
      "flow_matching_loss": 0.01886802390217781,
      "grad_norm": 0.10313626378774643,
      "learning_rate": 1.1831551815574848e-05,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040160922217182817,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37733459472656,
      "state_query_norm": 1408.50068359375,
      "state_relation_loss": 0.00040160922217182817,
      "step": 23630,
      "student_distance_mean": 1940.0417358398438,
      "teacher_distance_mean": 1.1467316389083861,
      "weighted_auxiliary_loss": 0.00040160922217182817,
      "weighted_relation_loss": 0.00040160922217182817
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.114125350795135,
      "flow_matching_loss": 0.019033414497971534,
      "grad_norm": 0.10452590882778168,
      "learning_rate": 1.1795972350329554e-05,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003750826086616144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45461502075195,
      "state_query_norm": 1427.7362060546875,
      "state_relation_loss": 0.0003750826086616144,
      "step": 23640,
      "student_distance_mean": 1950.2224487304688,
      "teacher_distance_mean": 1.1524092197418212,
      "weighted_auxiliary_loss": 0.0003750826086616144,
      "weighted_relation_loss": 0.0003750826086616144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.123479887745557,
      "flow_matching_loss": 0.018662478029727935,
      "grad_norm": 0.09717706590890884,
      "learning_rate": 1.176043930658024e-05,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000393646873999387,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.076615524291995,
      "state_query_norm": 1403.7479736328125,
      "state_relation_loss": 0.000393646873999387,
      "step": 23650,
      "student_distance_mean": 1925.3860595703125,
      "teacher_distance_mean": 1.1390833497047423,
      "weighted_auxiliary_loss": 0.000393646873999387,
      "weighted_relation_loss": 0.000393646873999387
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.13283442469598,
      "flow_matching_loss": 0.018025234621018172,
      "grad_norm": 0.08191175758838654,
      "learning_rate": 1.1724952727502942e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003917683323379606,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27699584960938,
      "state_query_norm": 1415.1333862304687,
      "state_relation_loss": 0.0003917683323379606,
      "step": 23660,
      "student_distance_mean": 1937.8843872070313,
      "teacher_distance_mean": 1.1472649812698363,
      "weighted_auxiliary_loss": 0.0003917683323379606,
      "weighted_relation_loss": 0.0003917683323379606
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.142188961646397,
      "flow_matching_loss": 0.018474551476538183,
      "grad_norm": 0.09891500324010849,
      "learning_rate": 1.1689512656217178e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003801658167503774,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.78452339172363,
      "state_query_norm": 1386.6829956054687,
      "state_relation_loss": 0.0003801658167503774,
      "step": 23670,
      "student_distance_mean": 1910.0449584960938,
      "teacher_distance_mean": 1.1303858757019043,
      "weighted_auxiliary_loss": 0.0003801658167503774,
      "weighted_relation_loss": 0.0003801658167503774
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.15154349859682,
      "flow_matching_loss": 0.01847444837912917,
      "grad_norm": 0.08258240669965744,
      "learning_rate": 1.1654119135785962e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039877348754089323,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78881149291992,
      "state_query_norm": 1427.4724853515625,
      "state_relation_loss": 0.00039877348754089323,
      "step": 23680,
      "student_distance_mean": 1963.11416015625,
      "teacher_distance_mean": 1.1604889512062073,
      "weighted_auxiliary_loss": 0.00039877348754089323,
      "weighted_relation_loss": 0.00039877348754089323
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.16089803554724,
      "flow_matching_loss": 0.01854647696018219,
      "grad_norm": 0.11658723652362823,
      "learning_rate": 1.1618772209215794e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038116637442726643,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.931539916992186,
      "state_query_norm": 1408.6713989257812,
      "state_relation_loss": 0.00038116637442726643,
      "step": 23690,
      "student_distance_mean": 1923.8348876953125,
      "teacher_distance_mean": 1.1382280945777894,
      "weighted_auxiliary_loss": 0.00038116637442726643,
      "weighted_relation_loss": 0.00038116637442726643
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.170252572497663,
      "flow_matching_loss": 0.017176534701138734,
      "grad_norm": 0.08559250086545944,
      "learning_rate": 1.1583471919456506e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042646691726986317,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.325486755371095,
      "state_query_norm": 1418.6724243164062,
      "state_relation_loss": 0.00042646691726986317,
      "step": 23700,
      "student_distance_mean": 1941.5499267578125,
      "teacher_distance_mean": 1.1488443970680238,
      "weighted_auxiliary_loss": 0.00042646691726986317,
      "weighted_relation_loss": 0.00042646691726986317
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.17960710944808,
      "flow_matching_loss": 0.017436732724308967,
      "grad_norm": 0.10066430270671844,
      "learning_rate": 1.1548218309401266e-05,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043581815261859445,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02686614990235,
      "state_query_norm": 1437.927587890625,
      "state_relation_loss": 0.00043581815261859445,
      "step": 23710,
      "student_distance_mean": 1975.2614135742188,
      "teacher_distance_mean": 1.1684840679168702,
      "weighted_auxiliary_loss": 0.00043581815261859445,
      "weighted_relation_loss": 0.00043581815261859445
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.188961646398504,
      "flow_matching_loss": 0.017631145752966405,
      "grad_norm": 0.08586248755455017,
      "learning_rate": 1.1513011421886555e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004290230426704511,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.132758712768556,
      "state_query_norm": 1404.3960693359375,
      "state_relation_loss": 0.0004290230426704511,
      "step": 23720,
      "student_distance_mean": 1925.9638916015624,
      "teacher_distance_mean": 1.1386429786682128,
      "weighted_auxiliary_loss": 0.0004290230426704511,
      "weighted_relation_loss": 0.0004290230426704511
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.198316183348926,
      "flow_matching_loss": 0.01635002288967371,
      "grad_norm": 0.08869733661413193,
      "learning_rate": 1.1477851299692056e-05,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040998766198754313,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.00809288024902,
      "state_query_norm": 1433.9729614257812,
      "state_relation_loss": 0.00040998766198754313,
      "step": 23730,
      "student_distance_mean": 1973.7090576171875,
      "teacher_distance_mean": 1.1664807200431824,
      "weighted_auxiliary_loss": 0.00040998766198754313,
      "weighted_relation_loss": 0.00040998766198754313
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.207670720299344,
      "flow_matching_loss": 0.017489114310592413,
      "grad_norm": 0.09905252605676651,
      "learning_rate": 1.144273798554063e-05,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038622870051767676,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64313163757324,
      "state_query_norm": 1432.3106079101562,
      "state_relation_loss": 0.00038622870051767676,
      "step": 23740,
      "student_distance_mean": 1961.695068359375,
      "teacher_distance_mean": 1.1584913730621338,
      "weighted_auxiliary_loss": 0.00038622870051767676,
      "weighted_relation_loss": 0.00038622870051767676
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.217025257249766,
      "flow_matching_loss": 0.01760916681960225,
      "grad_norm": 0.07571858912706375,
      "learning_rate": 1.1407671522098262e-05,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040632816089782866,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.12370567321777,
      "state_query_norm": 1445.0229248046876,
      "state_relation_loss": 0.00040632816089782866,
      "step": 23750,
      "student_distance_mean": 1983.8912475585937,
      "teacher_distance_mean": 1.1723101019859314,
      "weighted_auxiliary_loss": 0.00040632816089782866,
      "weighted_relation_loss": 0.00040632816089782866
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.226379794200188,
      "flow_matching_loss": 0.018164395447820426,
      "grad_norm": 0.09685783833265305,
      "learning_rate": 1.1372651951974e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040432919340673836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.85673141479492,
      "state_query_norm": 1432.8052124023438,
      "state_relation_loss": 0.00040432919340673836,
      "step": 23760,
      "student_distance_mean": 1967.9429931640625,
      "teacher_distance_mean": 1.1637644648551941,
      "weighted_auxiliary_loss": 0.00040432919340673836,
      "weighted_relation_loss": 0.00040432919340673836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.235734331150606,
      "flow_matching_loss": 0.017199384048581123,
      "grad_norm": 0.08534202724695206,
      "learning_rate": 1.1337679317719952e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041189234470948575,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39393768310547,
      "state_query_norm": 1413.1996826171876,
      "state_relation_loss": 0.00041189234470948575,
      "step": 23770,
      "student_distance_mean": 1942.99306640625,
      "teacher_distance_mean": 1.1485602021217347,
      "weighted_auxiliary_loss": 0.00041189234470948575,
      "weighted_relation_loss": 0.00041189234470948575
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.24508886810103,
      "flow_matching_loss": 0.016565185226500034,
      "grad_norm": 0.08602714538574219,
      "learning_rate": 1.1302753661831151e-05,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004008160554803908,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59589385986328,
      "state_query_norm": 1428.5376831054687,
      "state_relation_loss": 0.0004008160554803908,
      "step": 23780,
      "student_distance_mean": 1959.522998046875,
      "teacher_distance_mean": 1.1586417078971862,
      "weighted_auxiliary_loss": 0.0004008160554803908,
      "weighted_relation_loss": 0.0004008160554803908
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.25444340505145,
      "flow_matching_loss": 0.017732287291437387,
      "grad_norm": 0.08675985783338547,
      "learning_rate": 1.1267875026745561e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038059118669480084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55880928039551,
      "state_query_norm": 1428.4134643554687,
      "state_relation_loss": 0.00038059118669480084,
      "step": 23790,
      "student_distance_mean": 1954.0329345703126,
      "teacher_distance_mean": 1.156607186794281,
      "weighted_auxiliary_loss": 0.00038059118669480084,
      "weighted_relation_loss": 0.00038059118669480084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.263797942001872,
      "flow_matching_loss": 0.018016900308430193,
      "grad_norm": 0.08949997276067734,
      "learning_rate": 1.1233043454844016e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004106082662474364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56515769958496,
      "state_query_norm": 1421.2880737304688,
      "state_relation_loss": 0.0004106082662474364,
      "step": 23800,
      "student_distance_mean": 1953.245556640625,
      "teacher_distance_mean": 1.1567546129226685,
      "weighted_auxiliary_loss": 0.0004106082662474364,
      "weighted_relation_loss": 0.0004106082662474364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.27315247895229,
      "flow_matching_loss": 0.019090131856501102,
      "grad_norm": 0.10490050166845322,
      "learning_rate": 1.1198258988450144e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004084956512087956,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.34850387573242,
      "state_query_norm": 1412.77109375,
      "state_relation_loss": 0.0004084956512087956,
      "step": 23810,
      "student_distance_mean": 1941.9626708984374,
      "teacher_distance_mean": 1.1493513345718385,
      "weighted_auxiliary_loss": 0.0004084956512087956,
      "weighted_relation_loss": 0.0004084956512087956
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.282507015902713,
      "flow_matching_loss": 0.017471860721707345,
      "grad_norm": 0.11043071746826172,
      "learning_rate": 1.1163521669830368e-05,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039706270035821944,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.464350891113284,
      "state_query_norm": 1407.3789306640624,
      "state_relation_loss": 0.00039706270035821944,
      "step": 23820,
      "student_distance_mean": 1943.0053588867188,
      "teacher_distance_mean": 1.1500035524368286,
      "weighted_auxiliary_loss": 0.00039706270035821944,
      "weighted_relation_loss": 0.00039706270035821944
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.291861552853135,
      "flow_matching_loss": 0.019428507797420023,
      "grad_norm": 0.10761751979589462,
      "learning_rate": 1.1128831541193818e-05,
      "loss": 0.0198,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004017630417365581,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36511459350586,
      "state_query_norm": 1415.2043579101562,
      "state_relation_loss": 0.0004017630417365581,
      "step": 23830,
      "student_distance_mean": 1939.8527099609375,
      "teacher_distance_mean": 1.1468455076217652,
      "weighted_auxiliary_loss": 0.0004017630417365581,
      "weighted_relation_loss": 0.0004017630417365581
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.301216089803553,
      "flow_matching_loss": 0.018495149817317723,
      "grad_norm": 0.0995529517531395,
      "learning_rate": 1.1094188644692255e-05,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004268054064596072,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71774139404297,
      "state_query_norm": 1423.3799072265624,
      "state_relation_loss": 0.0004268054064596072,
      "step": 23840,
      "student_distance_mean": 1956.6654052734375,
      "teacher_distance_mean": 1.1564500689506532,
      "weighted_auxiliary_loss": 0.0004268054064596072,
      "weighted_relation_loss": 0.0004268054064596072
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.310570626753975,
      "flow_matching_loss": 0.018016221188008785,
      "grad_norm": 0.08069012314081192,
      "learning_rate": 1.1059593022420067e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003994869388407096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.73429832458496,
      "state_query_norm": 1385.5904052734375,
      "state_relation_loss": 0.0003994869388407096,
      "step": 23850,
      "student_distance_mean": 1906.0880981445312,
      "teacher_distance_mean": 1.1270039796829223,
      "weighted_auxiliary_loss": 0.0003994869388407096,
      "weighted_relation_loss": 0.0003994869388407096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.319925163704397,
      "flow_matching_loss": 0.018282474670559167,
      "grad_norm": 0.08895934373140335,
      "learning_rate": 1.1025044716414184e-05,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003878486284520477,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44774856567383,
      "state_query_norm": 1428.6855712890624,
      "state_relation_loss": 0.0003878486284520477,
      "step": 23860,
      "student_distance_mean": 1952.2369140625,
      "teacher_distance_mean": 1.1542877435684205,
      "weighted_auxiliary_loss": 0.0003878486284520477,
      "weighted_relation_loss": 0.0003878486284520477
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.32927970065482,
      "flow_matching_loss": 0.01719933720305562,
      "grad_norm": 0.08961606025695801,
      "learning_rate": 1.0990543768654082e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003982997790444642,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16222381591797,
      "state_query_norm": 1404.935986328125,
      "state_relation_loss": 0.0003982997790444642,
      "step": 23870,
      "student_distance_mean": 1932.9182250976562,
      "teacher_distance_mean": 1.143494200706482,
      "weighted_auxiliary_loss": 0.0003982997790444642,
      "weighted_relation_loss": 0.0003982997790444642
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.338634237605238,
      "flow_matching_loss": 0.017660555616021156,
      "grad_norm": 0.09928153455257416,
      "learning_rate": 1.0956090221061655e-05,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003811609523836523,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20721702575683,
      "state_query_norm": 1406.6709106445312,
      "state_relation_loss": 0.0003811609523836523,
      "step": 23880,
      "student_distance_mean": 1933.5800659179688,
      "teacher_distance_mean": 1.1428537726402284,
      "weighted_auxiliary_loss": 0.0003811609523836523,
      "weighted_relation_loss": 0.0003811609523836523
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.34798877455566,
      "flow_matching_loss": 0.01877509066835046,
      "grad_norm": 0.08639086782932281,
      "learning_rate": 1.0921684115501208e-05,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004020666645374149,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.50234260559082,
      "state_query_norm": 1462.885693359375,
      "state_relation_loss": 0.0004020666645374149,
      "step": 23890,
      "student_distance_mean": 2005.6177978515625,
      "teacher_distance_mean": 1.1843567967414856,
      "weighted_auxiliary_loss": 0.0004020666645374149,
      "weighted_relation_loss": 0.0004020666645374149
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.35734331150608,
      "flow_matching_loss": 0.018250195030122995,
      "grad_norm": 0.10893320292234421,
      "learning_rate": 1.0887325493779405e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003850284178042784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.9668628692627,
      "state_query_norm": 1394.32041015625,
      "state_relation_loss": 0.0003850284178042784,
      "step": 23900,
      "student_distance_mean": 1917.7156616210937,
      "teacher_distance_mean": 1.1339335083961486,
      "weighted_auxiliary_loss": 0.0003850284178042784,
      "weighted_relation_loss": 0.0003850284178042784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.3666978484565,
      "flow_matching_loss": 0.016487821750342847,
      "grad_norm": 0.07675107568502426,
      "learning_rate": 1.0853014397645207e-05,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004005270136985928,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.68117408752441,
      "state_query_norm": 1434.523486328125,
      "state_relation_loss": 0.0004005270136985928,
      "step": 23910,
      "student_distance_mean": 1961.6084106445312,
      "teacher_distance_mean": 1.1592840909957887,
      "weighted_auxiliary_loss": 0.0004005270136985928,
      "weighted_relation_loss": 0.0004005270136985928
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.376052385406922,
      "flow_matching_loss": 0.01797516020014882,
      "grad_norm": 0.09254240989685059,
      "learning_rate": 1.0818750868789829e-05,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004109982488444075,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.96430435180664,
      "state_query_norm": 1434.64423828125,
      "state_relation_loss": 0.0004109982488444075,
      "step": 23920,
      "student_distance_mean": 1972.897705078125,
      "teacher_distance_mean": 1.1652435779571533,
      "weighted_auxiliary_loss": 0.0004109982488444075,
      "weighted_relation_loss": 0.0004109982488444075
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.385406922357344,
      "flow_matching_loss": 0.01818553442135453,
      "grad_norm": 0.08399961143732071,
      "learning_rate": 1.0784534948846703e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037400678847916425,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.98672180175781,
      "state_query_norm": 1396.7823852539063,
      "state_relation_loss": 0.00037400678847916425,
      "step": 23930,
      "student_distance_mean": 1921.882763671875,
      "teacher_distance_mean": 1.1340335845947265,
      "weighted_auxiliary_loss": 0.00037400678847916425,
      "weighted_relation_loss": 0.00037400678847916425
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.394761459307766,
      "flow_matching_loss": 0.018695766013115643,
      "grad_norm": 0.09881021827459335,
      "learning_rate": 1.0750366679391393e-05,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003932894789613783,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.931653594970705,
      "state_query_norm": 1439.5897094726563,
      "state_relation_loss": 0.0003932894789613783,
      "step": 23940,
      "student_distance_mean": 1975.9953979492188,
      "teacher_distance_mean": 1.1664620757102966,
      "weighted_auxiliary_loss": 0.0003932894789613783,
      "weighted_relation_loss": 0.0003932894789613783
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.404115996258184,
      "flow_matching_loss": 0.01808117190375924,
      "grad_norm": 0.10355344414710999,
      "learning_rate": 1.0716246101941557e-05,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003819767385721207,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45524101257324,
      "state_query_norm": 1420.3956665039063,
      "state_relation_loss": 0.0003819767385721207,
      "step": 23950,
      "student_distance_mean": 1945.105517578125,
      "teacher_distance_mean": 1.1501440405845642,
      "weighted_auxiliary_loss": 0.0003819767385721207,
      "weighted_relation_loss": 0.0003819767385721207
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.413470533208606,
      "flow_matching_loss": 0.017184141371399165,
      "grad_norm": 0.0941702127456665,
      "learning_rate": 1.0682173257956934e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003976275067543611,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.927988052368164,
      "state_query_norm": 1441.5125244140625,
      "state_relation_loss": 0.0003976275067543611,
      "step": 23960,
      "student_distance_mean": 1975.3434936523438,
      "teacher_distance_mean": 1.168287718296051,
      "weighted_auxiliary_loss": 0.0003976275067543611,
      "weighted_relation_loss": 0.0003976275067543611
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.42282507015903,
      "flow_matching_loss": 0.016177885700017214,
      "grad_norm": 0.10171723365783691,
      "learning_rate": 1.064814818883922e-05,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040778187685646116,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38512687683105,
      "state_query_norm": 1420.2823974609375,
      "state_relation_loss": 0.00040778187685646116,
      "step": 23970,
      "student_distance_mean": 1945.7932495117188,
      "teacher_distance_mean": 1.1509145617485046,
      "weighted_auxiliary_loss": 0.00040778187685646116,
      "weighted_relation_loss": 0.00040778187685646116
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.432179607109447,
      "flow_matching_loss": 0.017856075987219812,
      "grad_norm": 0.08406428247690201,
      "learning_rate": 1.06141709359321e-05,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004018362669739872,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55930671691895,
      "state_query_norm": 1417.9323486328126,
      "state_relation_loss": 0.0004018362669739872,
      "step": 23980,
      "student_distance_mean": 1950.4786743164063,
      "teacher_distance_mean": 1.153759813308716,
      "weighted_auxiliary_loss": 0.0004018362669739872,
      "weighted_relation_loss": 0.0004018362669739872
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.44153414405987,
      "flow_matching_loss": 0.016350468248128892,
      "grad_norm": 0.09137582778930664,
      "learning_rate": 1.058024154052114e-05,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003870570712024346,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21680946350098,
      "state_query_norm": 1412.7041015625,
      "state_relation_loss": 0.0003870570712024346,
      "step": 23990,
      "student_distance_mean": 1937.5094848632812,
      "teacher_distance_mean": 1.1445600152015687,
      "weighted_auxiliary_loss": 0.0003870570712024346,
      "weighted_relation_loss": 0.0003870570712024346
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.45088868101029,
      "flow_matching_loss": 0.01647532032802701,
      "grad_norm": 0.08590161800384521,
      "learning_rate": 1.0546360043833747e-05,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003974892111727968,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3319034576416,
      "state_query_norm": 1418.1487548828125,
      "state_relation_loss": 0.0003974892111727968,
      "step": 24000,
      "student_distance_mean": 1943.6061767578126,
      "teacher_distance_mean": 1.1469515681266784,
      "weighted_auxiliary_loss": 0.0003974892111727968,
      "weighted_relation_loss": 0.0003974892111727968
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.46024321796071,
      "flow_matching_loss": 0.019040537904947995,
      "grad_norm": 0.10325339436531067,
      "learning_rate": 1.0512526487039138e-05,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004130028915824369,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30082244873047,
      "state_query_norm": 1408.5830200195312,
      "state_relation_loss": 0.0004130028915824369,
      "step": 24010,
      "student_distance_mean": 1935.3769409179688,
      "teacher_distance_mean": 1.142453920841217,
      "weighted_auxiliary_loss": 0.0004130028915824369,
      "weighted_relation_loss": 0.0004130028915824369
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.46959775491113,
      "flow_matching_loss": 0.016476157773286104,
      "grad_norm": 0.09469626843929291,
      "learning_rate": 1.0478740911248258e-05,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040440970042254776,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.445072937011716,
      "state_query_norm": 1421.1079467773438,
      "state_relation_loss": 0.00040440970042254776,
      "step": 24020,
      "student_distance_mean": 1950.8446166992187,
      "teacher_distance_mean": 1.154152023792267,
      "weighted_auxiliary_loss": 0.00040440970042254776,
      "weighted_relation_loss": 0.00040440970042254776
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.478952291861553,
      "flow_matching_loss": 0.017506234347820282,
      "grad_norm": 0.09711170941591263,
      "learning_rate": 1.0445003357513793e-05,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039732102886773645,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.408258056640626,
      "state_query_norm": 1407.583056640625,
      "state_relation_loss": 0.00039732102886773645,
      "step": 24030,
      "student_distance_mean": 1940.6936645507812,
      "teacher_distance_mean": 1.148265528678894,
      "weighted_auxiliary_loss": 0.00039732102886773645,
      "weighted_relation_loss": 0.00039732102886773645
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.488306828811975,
      "flow_matching_loss": 0.016801051329821348,
      "grad_norm": 0.09002583473920822,
      "learning_rate": 1.0411313866830042e-05,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004001241497462615,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.7281681060791,
      "state_query_norm": 1415.5826049804687,
      "state_relation_loss": 0.0004001241497462615,
      "step": 24040,
      "student_distance_mean": 1956.0940551757812,
      "teacher_distance_mean": 1.1571417689323424,
      "weighted_auxiliary_loss": 0.0004001241497462615,
      "weighted_relation_loss": 0.0004001241497462615
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.497661365762394,
      "flow_matching_loss": 0.015561242401599885,
      "grad_norm": 0.07895790785551071,
      "learning_rate": 1.0377672480132915e-05,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003938159206882119,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64272689819336,
      "state_query_norm": 1425.3173461914062,
      "state_relation_loss": 0.0003938159206882119,
      "step": 24050,
      "student_distance_mean": 1956.8373901367188,
      "teacher_distance_mean": 1.1570772409439087,
      "weighted_auxiliary_loss": 0.0003938159206882119,
      "weighted_relation_loss": 0.0003938159206882119
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.507015902712816,
      "flow_matching_loss": 0.016909580491483212,
      "grad_norm": 0.09025347977876663,
      "learning_rate": 1.0344079238299864e-05,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004037757171317935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.99990539550781,
      "state_query_norm": 1437.3154418945312,
      "state_relation_loss": 0.0004037757171317935,
      "step": 24060,
      "student_distance_mean": 1975.2026489257812,
      "teacher_distance_mean": 1.166988468170166,
      "weighted_auxiliary_loss": 0.0004037757171317935,
      "weighted_relation_loss": 0.0004037757171317935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.516370439663238,
      "flow_matching_loss": 0.01700366148725152,
      "grad_norm": 0.062300071120262146,
      "learning_rate": 1.0310534182149833e-05,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000410547450883314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.669443130493164,
      "state_query_norm": 1426.1430053710938,
      "state_relation_loss": 0.000410547450883314,
      "step": 24070,
      "student_distance_mean": 1958.3248291015625,
      "teacher_distance_mean": 1.1569634795188903,
      "weighted_auxiliary_loss": 0.000410547450883314,
      "weighted_relation_loss": 0.000410547450883314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.525724976613656,
      "flow_matching_loss": 0.017229321412742138,
      "grad_norm": 0.08829112350940704,
      "learning_rate": 1.0277037352443258e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004096920689335093,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55520172119141,
      "state_query_norm": 1429.8040893554687,
      "state_relation_loss": 0.0004096920689335093,
      "step": 24080,
      "student_distance_mean": 1952.1579223632812,
      "teacher_distance_mean": 1.153705620765686,
      "weighted_auxiliary_loss": 0.0004096920689335093,
      "weighted_relation_loss": 0.0004096920689335093
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.535079513564078,
      "flow_matching_loss": 0.017719145864248276,
      "grad_norm": 0.08636721968650818,
      "learning_rate": 1.024358878988193e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038744953635614363,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65830268859863,
      "state_query_norm": 1428.0893432617188,
      "state_relation_loss": 0.00038744953635614363,
      "step": 24090,
      "student_distance_mean": 1959.7304443359376,
      "teacher_distance_mean": 1.1586621403694153,
      "weighted_auxiliary_loss": 0.00038744953635614363,
      "weighted_relation_loss": 0.00038744953635614363
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.5444340505145,
      "flow_matching_loss": 0.01787684066221118,
      "grad_norm": 0.07497303932905197,
      "learning_rate": 1.0210188535108995e-05,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041669008787721397,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60421714782715,
      "state_query_norm": 1419.0925903320312,
      "state_relation_loss": 0.00041669008787721397,
      "step": 24100,
      "student_distance_mean": 1951.4817016601562,
      "teacher_distance_mean": 1.1540566444396974,
      "weighted_auxiliary_loss": 0.00041669008787721397,
      "weighted_relation_loss": 0.00041669008787721397
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.553788587464922,
      "flow_matching_loss": 0.016389550734311344,
      "grad_norm": 0.0865296944975853,
      "learning_rate": 1.0176836628708935e-05,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039434045029338447,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27279891967773,
      "state_query_norm": 1417.8142456054688,
      "state_relation_loss": 0.00039434045029338447,
      "step": 24110,
      "student_distance_mean": 1940.9025024414063,
      "teacher_distance_mean": 1.1468178629875183,
      "weighted_auxiliary_loss": 0.00039434045029338447,
      "weighted_relation_loss": 0.00039434045029338447
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.56314312441534,
      "flow_matching_loss": 0.01719861989840865,
      "grad_norm": 0.10031186789274216,
      "learning_rate": 1.0143533111207454e-05,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039417038788087667,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63107414245606,
      "state_query_norm": 1425.0005126953124,
      "state_relation_loss": 0.00039417038788087667,
      "step": 24120,
      "student_distance_mean": 1955.428857421875,
      "teacher_distance_mean": 1.1543405294418334,
      "weighted_auxiliary_loss": 0.00039417038788087667,
      "weighted_relation_loss": 0.00039417038788087667
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.572497661365762,
      "flow_matching_loss": 0.018250155542045832,
      "grad_norm": 0.09630724787712097,
      "learning_rate": 1.0110278023071446e-05,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003955687570851296,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.803440475463866,
      "state_query_norm": 1439.9915283203125,
      "state_relation_loss": 0.0003955687570851296,
      "step": 24130,
      "student_distance_mean": 1968.2553955078124,
      "teacher_distance_mean": 1.1637969732284545,
      "weighted_auxiliary_loss": 0.0003955687570851296,
      "weighted_relation_loss": 0.0003955687570851296
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.581852198316184,
      "flow_matching_loss": 0.017758629471063613,
      "grad_norm": 0.08098099380731583,
      "learning_rate": 1.007707140470901e-05,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000382263827486895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.320235061645505,
      "state_query_norm": 1408.8624877929688,
      "state_relation_loss": 0.000382263827486895,
      "step": 24140,
      "student_distance_mean": 1938.6597778320313,
      "teacher_distance_mean": 1.1471860885620118,
      "weighted_auxiliary_loss": 0.000382263827486895,
      "weighted_relation_loss": 0.000382263827486895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.591206735266603,
      "flow_matching_loss": 0.016812854912132025,
      "grad_norm": 0.08512815088033676,
      "learning_rate": 1.0043913296469298e-05,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003943387797335163,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.801551818847656,
      "state_query_norm": 1393.9439819335937,
      "state_relation_loss": 0.0003943387797335163,
      "step": 24150,
      "student_distance_mean": 1912.3932861328126,
      "teacher_distance_mean": 1.132829475402832,
      "weighted_auxiliary_loss": 0.0003943387797335163,
      "weighted_relation_loss": 0.0003943387797335163
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.600561272217025,
      "flow_matching_loss": 0.01681575132533908,
      "grad_norm": 0.09178594499826431,
      "learning_rate": 1.001080373864255e-05,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004095989657798782,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.387638854980466,
      "state_query_norm": 1416.2149780273437,
      "state_relation_loss": 0.0004095989657798782,
      "step": 24160,
      "student_distance_mean": 1941.7655395507813,
      "teacher_distance_mean": 1.1497115969657898,
      "weighted_auxiliary_loss": 0.0004095989657798782,
      "weighted_relation_loss": 0.0004095989657798782
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.609915809167447,
      "flow_matching_loss": 0.017370679415762424,
      "grad_norm": 0.07550449669361115,
      "learning_rate": 9.977742771459991e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041348836093675346,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40045051574707,
      "state_query_norm": 1417.8959716796876,
      "state_relation_loss": 0.00041348836093675346,
      "step": 24170,
      "student_distance_mean": 1944.5772216796875,
      "teacher_distance_mean": 1.150784933567047,
      "weighted_auxiliary_loss": 0.00041348836093675346,
      "weighted_relation_loss": 0.00041348836093675346
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.61927034611787,
      "flow_matching_loss": 0.018203814700245857,
      "grad_norm": 0.08824098855257034,
      "learning_rate": 9.944730435093802e-06,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003768313879845664,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.102226638793944,
      "state_query_norm": 1407.1418701171874,
      "state_relation_loss": 0.0003768313879845664,
      "step": 24180,
      "student_distance_mean": 1925.9836669921874,
      "teacher_distance_mean": 1.1398410558700562,
      "weighted_auxiliary_loss": 0.0003768313879845664,
      "weighted_relation_loss": 0.0003768313879845664
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.628624883068287,
      "flow_matching_loss": 0.017658458650112153,
      "grad_norm": 0.0949486717581749,
      "learning_rate": 9.911766769657115e-06,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004087462759343907,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42288589477539,
      "state_query_norm": 1409.0595581054688,
      "state_relation_loss": 0.0004087462759343907,
      "step": 24190,
      "student_distance_mean": 1942.0420532226562,
      "teacher_distance_mean": 1.1491469502449037,
      "weighted_auxiliary_loss": 0.0004087462759343907,
      "weighted_relation_loss": 0.0004087462759343907
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.63797942001871,
      "flow_matching_loss": 0.01874350644648075,
      "grad_norm": 0.09596433490514755,
      "learning_rate": 9.878851815203882e-06,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003918475471436977,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32963256835937,
      "state_query_norm": 1409.1694091796876,
      "state_relation_loss": 0.0003918475471436977,
      "step": 24200,
      "student_distance_mean": 1939.3784790039062,
      "teacher_distance_mean": 1.1465443253517151,
      "weighted_auxiliary_loss": 0.0003918475471436977,
      "weighted_relation_loss": 0.0003918475471436977
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.64733395696913,
      "flow_matching_loss": 0.017272041738033296,
      "grad_norm": 0.09539767354726791,
      "learning_rate": 9.845985611728864e-06,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003836744232103229,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40064544677735,
      "state_query_norm": 1414.8165649414063,
      "state_relation_loss": 0.0003836744232103229,
      "step": 24210,
      "student_distance_mean": 1944.7780395507812,
      "teacher_distance_mean": 1.1490853905677796,
      "weighted_auxiliary_loss": 0.0003836744232103229,
      "weighted_relation_loss": 0.0003836744232103229
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.65668849391955,
      "flow_matching_loss": 0.019059118535369634,
      "grad_norm": 0.11920183151960373,
      "learning_rate": 9.813168199167606e-06,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038286233902908864,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.302023315429686,
      "state_query_norm": 1414.4303344726563,
      "state_relation_loss": 0.00038286233902908864,
      "step": 24220,
      "student_distance_mean": 1943.0042358398437,
      "teacher_distance_mean": 1.1485006928443908,
      "weighted_auxiliary_loss": 0.00038286233902908864,
      "weighted_relation_loss": 0.00038286233902908864
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.66604303086997,
      "flow_matching_loss": 0.02013929421082139,
      "grad_norm": 0.08800996094942093,
      "learning_rate": 9.780399617396341e-06,
      "loss": 0.0205,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039448649622499945,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25856094360351,
      "state_query_norm": 1413.5556518554688,
      "state_relation_loss": 0.00039448649622499945,
      "step": 24230,
      "student_distance_mean": 1937.0029296875,
      "teacher_distance_mean": 1.1441861510276794,
      "weighted_auxiliary_loss": 0.00039448649622499945,
      "weighted_relation_loss": 0.00039448649622499945
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.675397567820394,
      "flow_matching_loss": 0.018451230227947236,
      "grad_norm": 0.08671841025352478,
      "learning_rate": 9.747679906232016e-06,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003788605274166912,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.579491806030276,
      "state_query_norm": 1430.5943969726563,
      "state_relation_loss": 0.0003788605274166912,
      "step": 24240,
      "student_distance_mean": 1954.4771728515625,
      "teacher_distance_mean": 1.1553197383880616,
      "weighted_auxiliary_loss": 0.0003788605274166912,
      "weighted_relation_loss": 0.0003788605274166912
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.684752104770816,
      "flow_matching_loss": 0.017324971593916418,
      "grad_norm": 0.07350149005651474,
      "learning_rate": 9.715009105432138e-06,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003782523301197216,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29437484741211,
      "state_query_norm": 1415.2520629882813,
      "state_relation_loss": 0.0003782523301197216,
      "step": 24250,
      "student_distance_mean": 1940.5155395507813,
      "teacher_distance_mean": 1.1501044511795044,
      "weighted_auxiliary_loss": 0.0003782523301197216,
      "weighted_relation_loss": 0.0003782523301197216
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.694106641721234,
      "flow_matching_loss": 0.018073348701000212,
      "grad_norm": 0.0823427066206932,
      "learning_rate": 9.682387254694835e-06,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004157729825237766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65954360961914,
      "state_query_norm": 1430.32880859375,
      "state_relation_loss": 0.0004157729825237766,
      "step": 24260,
      "student_distance_mean": 1961.004833984375,
      "teacher_distance_mean": 1.1620992064476012,
      "weighted_auxiliary_loss": 0.0004157729825237766,
      "weighted_relation_loss": 0.0004157729825237766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.703461178671656,
      "flow_matching_loss": 0.018323581106960772,
      "grad_norm": 0.10710696876049042,
      "learning_rate": 9.649814393658723e-06,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004063202941324562,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38176231384277,
      "state_query_norm": 1417.596044921875,
      "state_relation_loss": 0.0004063202941324562,
      "step": 24270,
      "student_distance_mean": 1946.7224853515625,
      "teacher_distance_mean": 1.1524977803230285,
      "weighted_auxiliary_loss": 0.0004063202941324562,
      "weighted_relation_loss": 0.0004063202941324562
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.712815715622078,
      "flow_matching_loss": 0.019200692884624004,
      "grad_norm": 0.09749914705753326,
      "learning_rate": 9.617290561902881e-06,
      "loss": 0.0196,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041292035894002765,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.611941528320315,
      "state_query_norm": 1418.8048828125,
      "state_relation_loss": 0.00041292035894002765,
      "step": 24280,
      "student_distance_mean": 1952.0136352539062,
      "teacher_distance_mean": 1.1534592151641845,
      "weighted_auxiliary_loss": 0.00041292035894002765,
      "weighted_relation_loss": 0.00041292035894002765
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.722170252572496,
      "flow_matching_loss": 0.01908317757770419,
      "grad_norm": 0.09150855243206024,
      "learning_rate": 9.584815798946861e-06,
      "loss": 0.0195,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041432074212934824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28756866455078,
      "state_query_norm": 1413.2903930664063,
      "state_relation_loss": 0.00041432074212934824,
      "step": 24290,
      "student_distance_mean": 1937.0224731445312,
      "teacher_distance_mean": 1.1442163944244386,
      "weighted_auxiliary_loss": 0.00041432074212934824,
      "weighted_relation_loss": 0.00041432074212934824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.73152478952292,
      "flow_matching_loss": 0.016735466569662093,
      "grad_norm": 0.095159612596035,
      "learning_rate": 9.55239014425055e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042083511652890593,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76618270874023,
      "state_query_norm": 1425.6248779296875,
      "state_relation_loss": 0.00042083511652890593,
      "step": 24300,
      "student_distance_mean": 1961.6917724609375,
      "teacher_distance_mean": 1.1595170497894287,
      "weighted_auxiliary_loss": 0.00042083511652890593,
      "weighted_relation_loss": 0.00042083511652890593
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.74087932647334,
      "flow_matching_loss": 0.018714721780270338,
      "grad_norm": 0.10200954228639603,
      "learning_rate": 9.520013637214176e-06,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038070162700023504,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.280544662475585,
      "state_query_norm": 1415.7925048828124,
      "state_relation_loss": 0.00038070162700023504,
      "step": 24310,
      "student_distance_mean": 1940.5636474609375,
      "teacher_distance_mean": 1.1481858611106872,
      "weighted_auxiliary_loss": 0.00038070162700023504,
      "weighted_relation_loss": 0.00038070162700023504
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.75023386342376,
      "flow_matching_loss": 0.017632186878472567,
      "grad_norm": 0.08347820490598679,
      "learning_rate": 9.48768631717824e-06,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044751732202712445,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.85786895751953,
      "state_query_norm": 1429.6508178710938,
      "state_relation_loss": 0.00044751732202712445,
      "step": 24320,
      "student_distance_mean": 1963.99189453125,
      "teacher_distance_mean": 1.1605728626251222,
      "weighted_auxiliary_loss": 0.00044751732202712445,
      "weighted_relation_loss": 0.00044751732202712445
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.75958840037418,
      "flow_matching_loss": 0.016937414649873972,
      "grad_norm": 0.09171534329652786,
      "learning_rate": 9.455408223423496e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004111062327865511,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.15353889465332,
      "state_query_norm": 1442.1743408203124,
      "state_relation_loss": 0.0004111062327865511,
      "step": 24330,
      "student_distance_mean": 1985.4433227539062,
      "teacher_distance_mean": 1.1729806423187257,
      "weighted_auxiliary_loss": 0.0004111062327865511,
      "weighted_relation_loss": 0.0004111062327865511
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.768942937324603,
      "flow_matching_loss": 0.018835758231580256,
      "grad_norm": 0.0877542719244957,
      "learning_rate": 9.423179395170845e-06,
      "loss": 0.0192,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039391539467033,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44708518981933,
      "state_query_norm": 1429.5053588867188,
      "state_relation_loss": 0.00039391539467033,
      "step": 24340,
      "student_distance_mean": 1949.6497924804687,
      "teacher_distance_mean": 1.1508695006370544,
      "weighted_auxiliary_loss": 0.00039391539467033,
      "weighted_relation_loss": 0.00039391539467033
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.778297474275025,
      "flow_matching_loss": 0.01788082765415311,
      "grad_norm": 0.09941323101520538,
      "learning_rate": 9.390999871581391e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041251947986893357,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65788230895996,
      "state_query_norm": 1427.2905395507812,
      "state_relation_loss": 0.00041251947986893357,
      "step": 24350,
      "student_distance_mean": 1955.810546875,
      "teacher_distance_mean": 1.15404771566391,
      "weighted_auxiliary_loss": 0.00041251947986893357,
      "weighted_relation_loss": 0.00041251947986893357
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.787652011225443,
      "flow_matching_loss": 0.019028109684586524,
      "grad_norm": 0.09097418189048767,
      "learning_rate": 9.358869691756273e-06,
      "loss": 0.0194,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004137123556574807,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30210838317871,
      "state_query_norm": 1416.6187866210937,
      "state_relation_loss": 0.0004137123556574807,
      "step": 24360,
      "student_distance_mean": 1940.4899047851563,
      "teacher_distance_mean": 1.145678663253784,
      "weighted_auxiliary_loss": 0.0004137123556574807,
      "weighted_relation_loss": 0.0004137123556574807
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.797006548175865,
      "flow_matching_loss": 0.01749700903892517,
      "grad_norm": 0.09128636121749878,
      "learning_rate": 9.326788894736688e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039771290903445333,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.345361328125,
      "state_query_norm": 1424.389453125,
      "state_relation_loss": 0.00039771290903445333,
      "step": 24370,
      "student_distance_mean": 1944.90263671875,
      "teacher_distance_mean": 1.148313295841217,
      "weighted_auxiliary_loss": 0.00039771290903445333,
      "weighted_relation_loss": 0.00039771290903445333
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.806361085126287,
      "flow_matching_loss": 0.01744141848757863,
      "grad_norm": 0.0842709094285965,
      "learning_rate": 9.294757519503811e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004133719572564587,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.77998085021973,
      "state_query_norm": 1392.407177734375,
      "state_relation_loss": 0.0004133719572564587,
      "step": 24380,
      "student_distance_mean": 1908.2286376953125,
      "teacher_distance_mean": 1.12875155210495,
      "weighted_auxiliary_loss": 0.0004133719572564587,
      "weighted_relation_loss": 0.0004133719572564587
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.815715622076706,
      "flow_matching_loss": 0.016815301217138767,
      "grad_norm": 0.10321912914514542,
      "learning_rate": 9.262775604978819e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004092994699021801,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63253517150879,
      "state_query_norm": 1427.8837280273438,
      "state_relation_loss": 0.0004092994699021801,
      "step": 24390,
      "student_distance_mean": 1958.38349609375,
      "teacher_distance_mean": 1.1578891277313232,
      "weighted_auxiliary_loss": 0.0004092994699021801,
      "weighted_relation_loss": 0.0004092994699021801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.825070159027128,
      "flow_matching_loss": 0.01798134306445718,
      "grad_norm": 0.08582735061645508,
      "learning_rate": 9.230843190022725e-06,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038422690995503216,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17146377563476,
      "state_query_norm": 1413.2681884765625,
      "state_relation_loss": 0.00038422690995503216,
      "step": 24400,
      "student_distance_mean": 1932.5508544921875,
      "teacher_distance_mean": 1.142816996574402,
      "weighted_auxiliary_loss": 0.00038422690995503216,
      "weighted_relation_loss": 0.00038422690995503216
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.83442469597755,
      "flow_matching_loss": 0.01713395435363054,
      "grad_norm": 0.07713872939348221,
      "learning_rate": 9.198960313436445e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004147089406615123,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69429740905762,
      "state_query_norm": 1425.5574340820312,
      "state_relation_loss": 0.0004147089406615123,
      "step": 24410,
      "student_distance_mean": 1958.339111328125,
      "teacher_distance_mean": 1.1581812739372253,
      "weighted_auxiliary_loss": 0.0004147089406615123,
      "weighted_relation_loss": 0.0004147089406615123
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.84377923292797,
      "flow_matching_loss": 0.016781821567565202,
      "grad_norm": 0.10008428245782852,
      "learning_rate": 9.16712701396067e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038515618070960044,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.690408325195314,
      "state_query_norm": 1427.613720703125,
      "state_relation_loss": 0.00038515618070960044,
      "step": 24420,
      "student_distance_mean": 1962.3944702148438,
      "teacher_distance_mean": 1.160832154750824,
      "weighted_auxiliary_loss": 0.00038515618070960044,
      "weighted_relation_loss": 0.00038515618070960044
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.85313376987839,
      "flow_matching_loss": 0.016358120646327735,
      "grad_norm": 0.09857161343097687,
      "learning_rate": 9.135343330275864e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004226128658046946,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.10303001403808,
      "state_query_norm": 1435.869384765625,
      "state_relation_loss": 0.0004226128658046946,
      "step": 24430,
      "student_distance_mean": 1979.5314697265626,
      "teacher_distance_mean": 1.1710596561431885,
      "weighted_auxiliary_loss": 0.0004226128658046946,
      "weighted_relation_loss": 0.0004226128658046946
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.862488306828812,
      "flow_matching_loss": 0.015592888835817576,
      "grad_norm": 0.06417649984359741,
      "learning_rate": 9.10360930100218e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042779608629643916,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39085693359375,
      "state_query_norm": 1405.8759887695312,
      "state_relation_loss": 0.00042779608629643916,
      "step": 24440,
      "student_distance_mean": 1938.1611328125,
      "teacher_distance_mean": 1.1476914763450623,
      "weighted_auxiliary_loss": 0.00042779608629643916,
      "weighted_relation_loss": 0.00042779608629643916
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.871842843779234,
      "flow_matching_loss": 0.01708271000534296,
      "grad_norm": 0.09924524277448654,
      "learning_rate": 9.07192496469949e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003976873442297801,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.81649513244629,
      "state_query_norm": 1443.4809326171876,
      "state_relation_loss": 0.0003976873442297801,
      "step": 24450,
      "student_distance_mean": 1972.125146484375,
      "teacher_distance_mean": 1.1678561329841615,
      "weighted_auxiliary_loss": 0.0003976873442297801,
      "weighted_relation_loss": 0.0003976873442297801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.881197380729652,
      "flow_matching_loss": 0.016316520515829325,
      "grad_norm": 0.07374311983585358,
      "learning_rate": 9.040290359867231e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039879643300082533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27802200317383,
      "state_query_norm": 1404.7440673828125,
      "state_relation_loss": 0.00039879643300082533,
      "step": 24460,
      "student_distance_mean": 1935.8177978515625,
      "teacher_distance_mean": 1.146286714076996,
      "weighted_auxiliary_loss": 0.00039879643300082533,
      "weighted_relation_loss": 0.00039879643300082533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.890551917680074,
      "flow_matching_loss": 0.018109891191124917,
      "grad_norm": 0.0972588062286377,
      "learning_rate": 9.008705524944438e-06,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040581577050033955,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20807647705078,
      "state_query_norm": 1416.4648315429688,
      "state_relation_loss": 0.00040581577050033955,
      "step": 24470,
      "student_distance_mean": 1936.3948120117188,
      "teacher_distance_mean": 1.1463916301727295,
      "weighted_auxiliary_loss": 0.00040581577050033955,
      "weighted_relation_loss": 0.00040581577050033955
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.899906454630496,
      "flow_matching_loss": 0.018116412032395602,
      "grad_norm": 0.09837038815021515,
      "learning_rate": 8.977170498309651e-06,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038027118134777995,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42855682373047,
      "state_query_norm": 1418.307470703125,
      "state_relation_loss": 0.00038027118134777995,
      "step": 24480,
      "student_distance_mean": 1949.2788452148438,
      "teacher_distance_mean": 1.1521366477012633,
      "weighted_auxiliary_loss": 0.00038027118134777995,
      "weighted_relation_loss": 0.00038027118134777995
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.90926099158092,
      "flow_matching_loss": 0.01701047262176871,
      "grad_norm": 0.09416903555393219,
      "learning_rate": 8.945685318280916e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039460023981519043,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.678137588500974,
      "state_query_norm": 1430.5136596679688,
      "state_relation_loss": 0.00039460023981519043,
      "step": 24490,
      "student_distance_mean": 1960.3227416992188,
      "teacher_distance_mean": 1.159092378616333,
      "weighted_auxiliary_loss": 0.00039460023981519043,
      "weighted_relation_loss": 0.00039460023981519043
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.918615528531337,
      "flow_matching_loss": 0.018240194581449033,
      "grad_norm": 0.0984773263335228,
      "learning_rate": 8.914250023115672e-06,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003973881568526849,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43857269287109,
      "state_query_norm": 1413.5166870117187,
      "state_relation_loss": 0.0003973881568526849,
      "step": 24500,
      "student_distance_mean": 1945.287744140625,
      "teacher_distance_mean": 1.1499230146408081,
      "weighted_auxiliary_loss": 0.0003973881568526849,
      "weighted_relation_loss": 0.0003973881568526849
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.92797006548176,
      "flow_matching_loss": 0.0164539098739624,
      "grad_norm": 0.08548688143491745,
      "learning_rate": 8.882864651010798e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004140332021052018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89132995605469,
      "state_query_norm": 1442.1406005859376,
      "state_relation_loss": 0.0004140332021052018,
      "step": 24510,
      "student_distance_mean": 1976.5109375,
      "teacher_distance_mean": 1.1676937460899353,
      "weighted_auxiliary_loss": 0.0004140332021052018,
      "weighted_relation_loss": 0.0004140332021052018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.93732460243218,
      "flow_matching_loss": 0.016990842390805484,
      "grad_norm": 0.08801871538162231,
      "learning_rate": 8.851529240102462e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003875281632645056,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.170315170288085,
      "state_query_norm": 1416.7483642578125,
      "state_relation_loss": 0.0003875281632645056,
      "step": 24520,
      "student_distance_mean": 1933.9264404296875,
      "teacher_distance_mean": 1.142131757736206,
      "weighted_auxiliary_loss": 0.0003875281632645056,
      "weighted_relation_loss": 0.0003875281632645056
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.9466791393826,
      "flow_matching_loss": 0.017554950434714556,
      "grad_norm": 0.07941412925720215,
      "learning_rate": 8.820243828466134e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038287687057163565,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.581696319580075,
      "state_query_norm": 1421.5024536132812,
      "state_relation_loss": 0.00038287687057163565,
      "step": 24530,
      "student_distance_mean": 1954.26064453125,
      "teacher_distance_mean": 1.1535529732704162,
      "weighted_auxiliary_loss": 0.00038287687057163565,
      "weighted_relation_loss": 0.00038287687057163565
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.95603367633302,
      "flow_matching_loss": 0.016928731836378573,
      "grad_norm": 0.07533105462789536,
      "learning_rate": 8.789008454116566e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042849116725847126,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.8593032836914,
      "state_query_norm": 1432.11904296875,
      "state_relation_loss": 0.00042849116725847126,
      "step": 24540,
      "student_distance_mean": 1969.42607421875,
      "teacher_distance_mean": 1.1642689466476441,
      "weighted_auxiliary_loss": 0.00042849116725847126,
      "weighted_relation_loss": 0.00042849116725847126
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.965388213283443,
      "flow_matching_loss": 0.017096842359751462,
      "grad_norm": 0.10222543030977249,
      "learning_rate": 8.757823155007655e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004191446729237214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70735626220703,
      "state_query_norm": 1431.6642333984375,
      "state_relation_loss": 0.0004191446729237214,
      "step": 24550,
      "student_distance_mean": 1960.1463745117187,
      "teacher_distance_mean": 1.1583184599876404,
      "weighted_auxiliary_loss": 0.0004191446729237214,
      "weighted_relation_loss": 0.0004191446729237214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.974742750233865,
      "flow_matching_loss": 0.01588003057986498,
      "grad_norm": 0.09678830951452255,
      "learning_rate": 8.72668796903251e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040527734672650694,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.84008369445801,
      "state_query_norm": 1414.2461791992187,
      "state_relation_loss": 0.00040527734672650694,
      "step": 24560,
      "student_distance_mean": 1923.687451171875,
      "teacher_distance_mean": 1.137082278728485,
      "weighted_auxiliary_loss": 0.00040527734672650694,
      "weighted_relation_loss": 0.00040527734672650694
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.984097287184284,
      "flow_matching_loss": 0.017475319746881723,
      "grad_norm": 0.07411179691553116,
      "learning_rate": 8.695602934023301e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004060458391904831,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.513390731811526,
      "state_query_norm": 1421.9685424804688,
      "state_relation_loss": 0.0004060458391904831,
      "step": 24570,
      "student_distance_mean": 1949.9978515625,
      "teacher_distance_mean": 1.152613604068756,
      "weighted_auxiliary_loss": 0.0004060458391904831,
      "weighted_relation_loss": 0.0004060458391904831
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 22.993451824134706,
      "flow_matching_loss": 0.016124772652983666,
      "grad_norm": 0.09756921976804733,
      "learning_rate": 8.664568087751274e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004144955048104748,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.47809371948242,
      "state_query_norm": 1458.4372680664062,
      "state_relation_loss": 0.0004144955048104748,
      "step": 24580,
      "student_distance_mean": 2002.1541625976563,
      "teacher_distance_mean": 1.1841229796409607,
      "weighted_auxiliary_loss": 0.0004144955048104748,
      "weighted_relation_loss": 0.0004144955048104748
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.002806361085128,
      "flow_matching_loss": 0.01691274978220463,
      "grad_norm": 0.08456218987703323,
      "learning_rate": 8.633583467926697e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00046053029946051537,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.69220657348633,
      "state_query_norm": 1423.3477172851562,
      "state_relation_loss": 0.00046053029946051537,
      "step": 24590,
      "student_distance_mean": 1956.7903686523437,
      "teacher_distance_mean": 1.1577489852905274,
      "weighted_auxiliary_loss": 0.00046053029946051537,
      "weighted_relation_loss": 0.00046053029946051537
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.012160898035546,
      "flow_matching_loss": 0.018070631474256516,
      "grad_norm": 0.10768426209688187,
      "learning_rate": 8.602649112198796e-06,
      "loss": 0.0185,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003992086159996688,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.72719612121582,
      "state_query_norm": 1423.2933959960938,
      "state_relation_loss": 0.0003992086159996688,
      "step": 24600,
      "student_distance_mean": 1958.3470947265625,
      "teacher_distance_mean": 1.1568355798721313,
      "weighted_auxiliary_loss": 0.0003992086159996688,
      "weighted_relation_loss": 0.0003992086159996688
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.021515434985968,
      "flow_matching_loss": 0.01885799700394273,
      "grad_norm": 0.12237167358398438,
      "learning_rate": 8.571765058155745e-06,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004210844403132796,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.79037628173828,
      "state_query_norm": 1433.1904907226562,
      "state_relation_loss": 0.0004210844403132796,
      "step": 24610,
      "student_distance_mean": 1965.8551513671875,
      "teacher_distance_mean": 1.162039613723755,
      "weighted_auxiliary_loss": 0.0004210844403132796,
      "weighted_relation_loss": 0.0004210844403132796
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.03086997193639,
      "flow_matching_loss": 0.017226025741547347,
      "grad_norm": 0.10963241755962372,
      "learning_rate": 8.540931343324583e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041345267090946435,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.415863800048825,
      "state_query_norm": 1413.0089477539063,
      "state_relation_loss": 0.00041345267090946435,
      "step": 24620,
      "student_distance_mean": 1942.4819946289062,
      "teacher_distance_mean": 1.1487167954444886,
      "weighted_auxiliary_loss": 0.00041345267090946435,
      "weighted_relation_loss": 0.00041345267090946435
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.04022450888681,
      "flow_matching_loss": 0.017301152646541595,
      "grad_norm": 0.10138901323080063,
      "learning_rate": 8.51014800517117e-06,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004024823458166793,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52455024719238,
      "state_query_norm": 1425.9080444335937,
      "state_relation_loss": 0.0004024823458166793,
      "step": 24630,
      "student_distance_mean": 1951.2990966796874,
      "teacher_distance_mean": 1.1563741326332093,
      "weighted_auxiliary_loss": 0.0004024823458166793,
      "weighted_relation_loss": 0.0004024823458166793
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.04957904583723,
      "flow_matching_loss": 0.016395154409110546,
      "grad_norm": 0.07954522967338562,
      "learning_rate": 8.479415081100167e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037681986286770553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33938407897949,
      "state_query_norm": 1414.05244140625,
      "state_relation_loss": 0.00037681986286770553,
      "step": 24640,
      "student_distance_mean": 1944.6849487304687,
      "teacher_distance_mean": 1.1524423003196715,
      "weighted_auxiliary_loss": 0.00037681986286770553,
      "weighted_relation_loss": 0.00037681986286770553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.058933582787652,
      "flow_matching_loss": 0.01829411359503865,
      "grad_norm": 0.08336209505796432,
      "learning_rate": 8.448732608454967e-06,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038614838558714837,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.528097534179686,
      "state_query_norm": 1416.3687622070313,
      "state_relation_loss": 0.00038614838558714837,
      "step": 24650,
      "student_distance_mean": 1952.1270874023437,
      "teacher_distance_mean": 1.1552017331123352,
      "weighted_auxiliary_loss": 0.00038614838558714837,
      "weighted_relation_loss": 0.00038614838558714837
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.068288119738074,
      "flow_matching_loss": 0.017396206222474574,
      "grad_norm": 0.09316036105155945,
      "learning_rate": 8.418100624517688e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040742649871390314,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28018074035644,
      "state_query_norm": 1412.6444580078125,
      "state_relation_loss": 0.00040742649871390314,
      "step": 24660,
      "student_distance_mean": 1936.6276733398438,
      "teacher_distance_mean": 1.145320749282837,
      "weighted_auxiliary_loss": 0.00040742649871390314,
      "weighted_relation_loss": 0.00040742649871390314
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.077642656688493,
      "flow_matching_loss": 0.01734260991215706,
      "grad_norm": 0.0957026407122612,
      "learning_rate": 8.387519166509062e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041952186729758977,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28322868347168,
      "state_query_norm": 1407.816650390625,
      "state_relation_loss": 0.00041952186729758977,
      "step": 24670,
      "student_distance_mean": 1936.894384765625,
      "teacher_distance_mean": 1.1461585402488708,
      "weighted_auxiliary_loss": 0.00041952186729758977,
      "weighted_relation_loss": 0.00041952186729758977
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.086997193638915,
      "flow_matching_loss": 0.016811976581811904,
      "grad_norm": 0.10272247344255447,
      "learning_rate": 8.356988271588445e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003777737059863284,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14595336914063,
      "state_query_norm": 1414.0828979492187,
      "state_relation_loss": 0.0003777737059863284,
      "step": 24680,
      "student_distance_mean": 1931.107373046875,
      "teacher_distance_mean": 1.1432990789413453,
      "weighted_auxiliary_loss": 0.0003777737059863284,
      "weighted_relation_loss": 0.0003777737059863284
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.096351730589337,
      "flow_matching_loss": 0.018381374888122082,
      "grad_norm": 0.08908292651176453,
      "learning_rate": 8.326507976853765e-06,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004116218013223261,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25485763549805,
      "state_query_norm": 1410.7047607421875,
      "state_relation_loss": 0.0004116218013223261,
      "step": 24690,
      "student_distance_mean": 1936.9700561523437,
      "teacher_distance_mean": 1.1470513582229613,
      "weighted_auxiliary_loss": 0.0004116218013223261,
      "weighted_relation_loss": 0.0004116218013223261
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.105706267539755,
      "flow_matching_loss": 0.017164557240903378,
      "grad_norm": 0.09746276587247849,
      "learning_rate": 8.296078319341443e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037615846376866103,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.145013427734376,
      "state_query_norm": 1407.8368530273438,
      "state_relation_loss": 0.00037615846376866103,
      "step": 24700,
      "student_distance_mean": 1932.031787109375,
      "teacher_distance_mean": 1.1432734489440919,
      "weighted_auxiliary_loss": 0.00037615846376866103,
      "weighted_relation_loss": 0.00037615846376866103
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.115060804490177,
      "flow_matching_loss": 0.01698543140664697,
      "grad_norm": 0.09094744920730591,
      "learning_rate": 8.265699336026383e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003763722139410675,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.07377052307129,
      "state_query_norm": 1444.2007934570313,
      "state_relation_loss": 0.0003763722139410675,
      "step": 24710,
      "student_distance_mean": 1981.84931640625,
      "teacher_distance_mean": 1.1726382374763489,
      "weighted_auxiliary_loss": 0.0003763722139410675,
      "weighted_relation_loss": 0.0003763722139410675
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.1244153414406,
      "flow_matching_loss": 0.017527645360678433,
      "grad_norm": 0.08512059599161148,
      "learning_rate": 8.235371063821923e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004065090557560325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28281211853027,
      "state_query_norm": 1413.97119140625,
      "state_relation_loss": 0.0004065090557560325,
      "step": 24720,
      "student_distance_mean": 1940.5112670898438,
      "teacher_distance_mean": 1.1482786178588866,
      "weighted_auxiliary_loss": 0.0004065090557560325,
      "weighted_relation_loss": 0.0004065090557560325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.13376987839102,
      "flow_matching_loss": 0.01737778531387448,
      "grad_norm": 0.07779738306999207,
      "learning_rate": 8.205093539579767e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003945728036342189,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39452705383301,
      "state_query_norm": 1407.053564453125,
      "state_relation_loss": 0.0003945728036342189,
      "step": 24730,
      "student_distance_mean": 1941.9407470703125,
      "teacher_distance_mean": 1.1487640976905822,
      "weighted_auxiliary_loss": 0.0003945728036342189,
      "weighted_relation_loss": 0.0003945728036342189
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.14312441534144,
      "flow_matching_loss": 0.01778994584456086,
      "grad_norm": 0.08927533030509949,
      "learning_rate": 8.174866800089963e-06,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003833992232102901,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.347924041748044,
      "state_query_norm": 1409.7008178710937,
      "state_relation_loss": 0.0003833992232102901,
      "step": 24740,
      "student_distance_mean": 1937.3737548828126,
      "teacher_distance_mean": 1.1464952945709228,
      "weighted_auxiliary_loss": 0.0003833992232102901,
      "weighted_relation_loss": 0.0003833992232102901
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.15247895229186,
      "flow_matching_loss": 0.01756697064265609,
      "grad_norm": 0.07128655910491943,
      "learning_rate": 8.144690882080853e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003860534372506663,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57700958251953,
      "state_query_norm": 1424.23564453125,
      "state_relation_loss": 0.0003860534372506663,
      "step": 24750,
      "student_distance_mean": 1954.1697265625,
      "teacher_distance_mean": 1.1582021594047547,
      "weighted_auxiliary_loss": 0.0003860534372506663,
      "weighted_relation_loss": 0.0003860534372506663
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.161833489242284,
      "flow_matching_loss": 0.017462551407516003,
      "grad_norm": 0.08622793853282928,
      "learning_rate": 8.114565822219006e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039363199612125755,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64093360900879,
      "state_query_norm": 1429.6919677734375,
      "state_relation_loss": 0.00039363199612125755,
      "step": 24760,
      "student_distance_mean": 1960.2536376953126,
      "teacher_distance_mean": 1.1613902688026427,
      "weighted_auxiliary_loss": 0.00039363199612125755,
      "weighted_relation_loss": 0.00039363199612125755
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.171188026192702,
      "flow_matching_loss": 0.016560829151421784,
      "grad_norm": 0.08440318703651428,
      "learning_rate": 8.084491657109233e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004008384043117985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.03554153442383,
      "state_query_norm": 1400.268798828125,
      "state_relation_loss": 0.0004008384043117985,
      "step": 24770,
      "student_distance_mean": 1925.4259521484375,
      "teacher_distance_mean": 1.1411580204963685,
      "weighted_auxiliary_loss": 0.0004008384043117985,
      "weighted_relation_loss": 0.0004008384043117985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.180542563143124,
      "flow_matching_loss": 0.016772000305354594,
      "grad_norm": 0.10700000077486038,
      "learning_rate": 8.054468423294476e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004172740853391588,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.87039489746094,
      "state_query_norm": 1431.940576171875,
      "state_relation_loss": 0.0004172740853391588,
      "step": 24780,
      "student_distance_mean": 1965.9159545898438,
      "teacher_distance_mean": 1.1654929757118224,
      "weighted_auxiliary_loss": 0.0004172740853391588,
      "weighted_relation_loss": 0.0004172740853391588
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.189897100093546,
      "flow_matching_loss": 0.015791038610041143,
      "grad_norm": 0.08598106354475021,
      "learning_rate": 8.024496157255785e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004258982778992504,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.690949249267575,
      "state_query_norm": 1416.6571655273438,
      "state_relation_loss": 0.0004258982778992504,
      "step": 24790,
      "student_distance_mean": 1957.6254150390625,
      "teacher_distance_mean": 1.1598811864852905,
      "weighted_auxiliary_loss": 0.0004258982778992504,
      "weighted_relation_loss": 0.0004258982778992504
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.199251637043965,
      "flow_matching_loss": 0.017908709496259688,
      "grad_norm": 0.08272918313741684,
      "learning_rate": 7.994574895412293e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039240972546394914,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66920509338379,
      "state_query_norm": 1432.1143920898437,
      "state_relation_loss": 0.00039240972546394914,
      "step": 24800,
      "student_distance_mean": 1959.027490234375,
      "teacher_distance_mean": 1.1589317083358766,
      "weighted_auxiliary_loss": 0.00039240972546394914,
      "weighted_relation_loss": 0.00039240972546394914
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.208606173994387,
      "flow_matching_loss": 0.017920486442744733,
      "grad_norm": 0.10184390097856522,
      "learning_rate": 7.964704674121149e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038705682090949266,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62564964294434,
      "state_query_norm": 1426.0809326171875,
      "state_relation_loss": 0.00038705682090949266,
      "step": 24810,
      "student_distance_mean": 1957.6133544921875,
      "teacher_distance_mean": 1.1578104615211486,
      "weighted_auxiliary_loss": 0.00038705682090949266,
      "weighted_relation_loss": 0.00038705682090949266
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.21796071094481,
      "flow_matching_loss": 0.01564911799505353,
      "grad_norm": 0.07502852380275726,
      "learning_rate": 7.934885529677505e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003986946423538029,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.069932556152345,
      "state_query_norm": 1401.9548217773438,
      "state_relation_loss": 0.0003986946423538029,
      "step": 24820,
      "student_distance_mean": 1924.1221923828125,
      "teacher_distance_mean": 1.1383479475975036,
      "weighted_auxiliary_loss": 0.0003986946423538029,
      "weighted_relation_loss": 0.0003986946423538029
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.22731524789523,
      "flow_matching_loss": 0.01869007870554924,
      "grad_norm": 0.07424259185791016,
      "learning_rate": 7.905117498314412e-06,
      "loss": 0.0191,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003880176489474252,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.83812026977539,
      "state_query_norm": 1409.3509155273437,
      "state_relation_loss": 0.0003880176489474252,
      "step": 24830,
      "student_distance_mean": 1918.84990234375,
      "teacher_distance_mean": 1.134735345840454,
      "weighted_auxiliary_loss": 0.0003880176489474252,
      "weighted_relation_loss": 0.0003880176489474252
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.23666978484565,
      "flow_matching_loss": 0.016832297947257758,
      "grad_norm": 0.09169778972864151,
      "learning_rate": 7.87540061620286e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003965417738072574,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71230964660644,
      "state_query_norm": 1424.1941162109374,
      "state_relation_loss": 0.0003965417738072574,
      "step": 24840,
      "student_distance_mean": 1958.4692749023438,
      "teacher_distance_mean": 1.1589587092399598,
      "weighted_auxiliary_loss": 0.0003965417738072574,
      "weighted_relation_loss": 0.0003965417738072574
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.24602432179607,
      "flow_matching_loss": 0.01773445149883628,
      "grad_norm": 0.09177164733409882,
      "learning_rate": 7.845734919451647e-06,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004186938691418618,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75998649597168,
      "state_query_norm": 1426.3054565429688,
      "state_relation_loss": 0.0004186938691418618,
      "step": 24850,
      "student_distance_mean": 1960.8646606445313,
      "teacher_distance_mean": 1.160770046710968,
      "weighted_auxiliary_loss": 0.0004186938691418618,
      "weighted_relation_loss": 0.0004186938691418618
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.255378858746493,
      "flow_matching_loss": 0.01663607181981206,
      "grad_norm": 0.09433440119028091,
      "learning_rate": 7.816120444107383e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004193540196865797,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02407302856445,
      "state_query_norm": 1425.0305419921874,
      "state_relation_loss": 0.0004193540196865797,
      "step": 24860,
      "student_distance_mean": 1969.3084716796875,
      "teacher_distance_mean": 1.1655099987983704,
      "weighted_auxiliary_loss": 0.0004193540196865797,
      "weighted_relation_loss": 0.0004193540196865797
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.26473339569691,
      "flow_matching_loss": 0.017298410087823866,
      "grad_norm": 0.08585350215435028,
      "learning_rate": 7.786557226154473e-06,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004109395231353119,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41196784973145,
      "state_query_norm": 1417.584716796875,
      "state_relation_loss": 0.0004109395231353119,
      "step": 24870,
      "student_distance_mean": 1943.8078125,
      "teacher_distance_mean": 1.1521920323371888,
      "weighted_auxiliary_loss": 0.0004109395231353119,
      "weighted_relation_loss": 0.0004109395231353119
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.274087932647333,
      "flow_matching_loss": 0.018164414912462234,
      "grad_norm": 0.09682411700487137,
      "learning_rate": 7.757045301514998e-06,
      "loss": 0.0186,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003926878795027733,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64101295471191,
      "state_query_norm": 1429.87080078125,
      "state_relation_loss": 0.0003926878795027733,
      "step": 24880,
      "student_distance_mean": 1958.215185546875,
      "teacher_distance_mean": 1.1589542746543884,
      "weighted_auxiliary_loss": 0.0003926878795027733,
      "weighted_relation_loss": 0.0003926878795027733
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.283442469597755,
      "flow_matching_loss": 0.017532338947057725,
      "grad_norm": 0.10769114643335342,
      "learning_rate": 7.727584706048735e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041059478535316886,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59410362243652,
      "state_query_norm": 1417.2850463867187,
      "state_relation_loss": 0.00041059478535316886,
      "step": 24890,
      "student_distance_mean": 1951.1824951171875,
      "teacher_distance_mean": 1.1548359036445617,
      "weighted_auxiliary_loss": 0.00041059478535316886,
      "weighted_relation_loss": 0.00041059478535316886
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.292797006548177,
      "flow_matching_loss": 0.017506528459489346,
      "grad_norm": 0.10172916203737259,
      "learning_rate": 7.698175475553077e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003987645555753261,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21102447509766,
      "state_query_norm": 1407.9125854492188,
      "state_relation_loss": 0.0003987645555753261,
      "step": 24900,
      "student_distance_mean": 1930.426318359375,
      "teacher_distance_mean": 1.1424017429351807,
      "weighted_auxiliary_loss": 0.0003987645555753261,
      "weighted_relation_loss": 0.0003987645555753261
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.302151543498596,
      "flow_matching_loss": 0.017978669423609973,
      "grad_norm": 0.08736773580312729,
      "learning_rate": 7.668817645763021e-06,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041082642565015703,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.101494216918944,
      "state_query_norm": 1405.9405029296875,
      "state_relation_loss": 0.00041082642565015703,
      "step": 24910,
      "student_distance_mean": 1926.6026733398437,
      "teacher_distance_mean": 1.1399916291236878,
      "weighted_auxiliary_loss": 0.00041082642565015703,
      "weighted_relation_loss": 0.00041082642565015703
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.311506080449018,
      "flow_matching_loss": 0.017731083650141956,
      "grad_norm": 0.07813693583011627,
      "learning_rate": 7.639511252351088e-06,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003832509566564113,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23714179992676,
      "state_query_norm": 1411.0532592773438,
      "state_relation_loss": 0.0003832509566564113,
      "step": 24920,
      "student_distance_mean": 1937.3657958984375,
      "teacher_distance_mean": 1.1456410527229308,
      "weighted_auxiliary_loss": 0.0003832509566564113,
      "weighted_relation_loss": 0.0003832509566564113
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.32086061739944,
      "flow_matching_loss": 0.017880728095769884,
      "grad_norm": 0.08548139780759811,
      "learning_rate": 7.610256330927324e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038893600285518917,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29870872497558,
      "state_query_norm": 1408.5019409179688,
      "state_relation_loss": 0.00038893600285518917,
      "step": 24930,
      "student_distance_mean": 1935.5064453125,
      "teacher_distance_mean": 1.1455161929130555,
      "weighted_auxiliary_loss": 0.00038893600285518917,
      "weighted_relation_loss": 0.00038893600285518917
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.330215154349858,
      "flow_matching_loss": 0.016129303071647883,
      "grad_norm": 0.07424018532037735,
      "learning_rate": 7.581052917039211e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003760817460715771,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31350822448731,
      "state_query_norm": 1422.9208618164062,
      "state_relation_loss": 0.0003760817460715771,
      "step": 24940,
      "student_distance_mean": 1941.0853881835938,
      "teacher_distance_mean": 1.1493853688240052,
      "weighted_auxiliary_loss": 0.0003760817460715771,
      "weighted_relation_loss": 0.0003760817460715771
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.33956969130028,
      "flow_matching_loss": 0.017050287593156098,
      "grad_norm": 0.0969427302479744,
      "learning_rate": 7.551901046171645e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039453130739275366,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.217577362060545,
      "state_query_norm": 1407.8461547851562,
      "state_relation_loss": 0.00039453130739275366,
      "step": 24950,
      "student_distance_mean": 1934.4155517578124,
      "teacher_distance_mean": 1.146601927280426,
      "weighted_auxiliary_loss": 0.00039453130739275366,
      "weighted_relation_loss": 0.00039453130739275366
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.348924228250702,
      "flow_matching_loss": 0.01597702968865633,
      "grad_norm": 0.08526956290006638,
      "learning_rate": 7.522800753746895e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037909144593868406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.59398498535156,
      "state_query_norm": 1393.32021484375,
      "state_relation_loss": 0.00037909144593868406,
      "step": 24960,
      "student_distance_mean": 1902.7426391601562,
      "teacher_distance_mean": 1.1280629515647889,
      "weighted_auxiliary_loss": 0.00037909144593868406,
      "weighted_relation_loss": 0.00037909144593868406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.358278765201124,
      "flow_matching_loss": 0.01738256895914674,
      "grad_norm": 0.09114600718021393,
      "learning_rate": 7.493752075124577e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039870439504738895,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39970932006836,
      "state_query_norm": 1405.5551391601562,
      "state_relation_loss": 0.00039870439504738895,
      "step": 24970,
      "student_distance_mean": 1938.8412719726562,
      "teacher_distance_mean": 1.1488863825798035,
      "weighted_auxiliary_loss": 0.00039870439504738895,
      "weighted_relation_loss": 0.00039870439504738895
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.367633302151543,
      "flow_matching_loss": 0.017014584969729185,
      "grad_norm": 0.0903385654091835,
      "learning_rate": 7.464755045601557e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038123308622743933,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.45654716491699,
      "state_query_norm": 1417.1876708984375,
      "state_relation_loss": 0.00038123308622743933,
      "step": 24980,
      "student_distance_mean": 1947.8611450195312,
      "teacher_distance_mean": 1.1546525597572326,
      "weighted_auxiliary_loss": 0.00038123308622743933,
      "weighted_relation_loss": 0.00038123308622743933
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.376987839101965,
      "flow_matching_loss": 0.015705958101898433,
      "grad_norm": 0.08751743286848068,
      "learning_rate": 7.435809700411972e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003853437781799585,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.897168731689455,
      "state_query_norm": 1403.617333984375,
      "state_relation_loss": 0.0003853437781799585,
      "step": 24990,
      "student_distance_mean": 1920.76904296875,
      "teacher_distance_mean": 1.1389366388320923,
      "weighted_auxiliary_loss": 0.0003853437781799585,
      "weighted_relation_loss": 0.0003853437781799585
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.386342376052387,
      "flow_matching_loss": 0.01674388684332371,
      "grad_norm": 0.0861571878194809,
      "learning_rate": 7.40691607472715e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003974315040977672,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50163536071777,
      "state_query_norm": 1417.1478393554687,
      "state_relation_loss": 0.0003974315040977672,
      "step": 25000,
      "student_distance_mean": 1949.88857421875,
      "teacher_distance_mean": 1.1555847883224488,
      "weighted_auxiliary_loss": 0.0003974315040977672,
      "weighted_relation_loss": 0.0003974315040977672
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.395696913002805,
      "flow_matching_loss": 0.016369855124503375,
      "grad_norm": 0.08763930946588516,
      "learning_rate": 7.3780742036555606e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003866838553221896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30949020385742,
      "state_query_norm": 1424.856591796875,
      "state_relation_loss": 0.0003866838553221896,
      "step": 25010,
      "student_distance_mean": 1942.0579467773437,
      "teacher_distance_mean": 1.1525214791297913,
      "weighted_auxiliary_loss": 0.0003866838553221896,
      "weighted_relation_loss": 0.0003866838553221896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.405051449953227,
      "flow_matching_loss": 0.01670887377113104,
      "grad_norm": 0.08359253406524658,
      "learning_rate": 7.349284122242783e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037717830273322763,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1294677734375,
      "state_query_norm": 1411.4902709960938,
      "state_relation_loss": 0.00037717830273322763,
      "step": 25020,
      "student_distance_mean": 1931.759765625,
      "teacher_distance_mean": 1.1452027320861817,
      "weighted_auxiliary_loss": 0.00037717830273322763,
      "weighted_relation_loss": 0.00037717830273322763
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.41440598690365,
      "flow_matching_loss": 0.016263067722320557,
      "grad_norm": 0.07748352736234665,
      "learning_rate": 7.320545865471512e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039691076090093703,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3166805267334,
      "state_query_norm": 1408.6789184570312,
      "state_relation_loss": 0.00039691076090093703,
      "step": 25030,
      "student_distance_mean": 1936.8056518554688,
      "teacher_distance_mean": 1.14739750623703,
      "weighted_auxiliary_loss": 0.00039691076090093703,
      "weighted_relation_loss": 0.00039691076090093703
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.42376052385407,
      "flow_matching_loss": 0.01635047420859337,
      "grad_norm": 0.08743426203727722,
      "learning_rate": 7.291859468261425e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003751134441699833,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.315381622314455,
      "state_query_norm": 1417.9960083007813,
      "state_relation_loss": 0.0003751134441699833,
      "step": 25040,
      "student_distance_mean": 1941.2184814453126,
      "teacher_distance_mean": 1.1487435936927795,
      "weighted_auxiliary_loss": 0.0003751134441699833,
      "weighted_relation_loss": 0.0003751134441699833
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.43311506080449,
      "flow_matching_loss": 0.015673837810754775,
      "grad_norm": 0.09184420853853226,
      "learning_rate": 7.263224965469195e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004345045104855672,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.465769577026364,
      "state_query_norm": 1422.1258178710937,
      "state_relation_loss": 0.0004345045104855672,
      "step": 25050,
      "student_distance_mean": 1950.1244506835938,
      "teacher_distance_mean": 1.1539683222770691,
      "weighted_auxiliary_loss": 0.0004345045104855672,
      "weighted_relation_loss": 0.0004345045104855672
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.44246959775491,
      "flow_matching_loss": 0.017667463701218366,
      "grad_norm": 0.09345576912164688,
      "learning_rate": 7.234642391888446e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003728188545210287,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.49564666748047,
      "state_query_norm": 1421.4746704101562,
      "state_relation_loss": 0.0003728188545210287,
      "step": 25060,
      "student_distance_mean": 1950.5351196289062,
      "teacher_distance_mean": 1.1526113629341126,
      "weighted_auxiliary_loss": 0.0003728188545210287,
      "weighted_relation_loss": 0.0003728188545210287
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.451824134705333,
      "flow_matching_loss": 0.01750114941969514,
      "grad_norm": 0.09271756559610367,
      "learning_rate": 7.206111782249697e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00035931325692217797,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.503167724609376,
      "state_query_norm": 1427.0331665039062,
      "state_relation_loss": 0.00035931325692217797,
      "step": 25070,
      "student_distance_mean": 1953.8190673828126,
      "teacher_distance_mean": 1.1569350600242614,
      "weighted_auxiliary_loss": 0.00035931325692217797,
      "weighted_relation_loss": 0.00035931325692217797
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.46117867165575,
      "flow_matching_loss": 0.017191676422953606,
      "grad_norm": 0.0858435183763504,
      "learning_rate": 7.1776331712203385e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004084605810930952,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.856296157836915,
      "state_query_norm": 1424.1135620117188,
      "state_relation_loss": 0.0004084605810930952,
      "step": 25080,
      "student_distance_mean": 1963.56318359375,
      "teacher_distance_mean": 1.1627392292022705,
      "weighted_auxiliary_loss": 0.0004084605810930952,
      "weighted_relation_loss": 0.0004084605810930952
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.470533208606174,
      "flow_matching_loss": 0.016489362437278034,
      "grad_norm": 0.08994515240192413,
      "learning_rate": 7.149206593404561e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003907076257746667,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.81102676391602,
      "state_query_norm": 1427.655810546875,
      "state_relation_loss": 0.0003907076257746667,
      "step": 25090,
      "student_distance_mean": 1966.8260009765625,
      "teacher_distance_mean": 1.1634437561035156,
      "weighted_auxiliary_loss": 0.0003907076257746667,
      "weighted_relation_loss": 0.0003907076257746667
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.479887745556596,
      "flow_matching_loss": 0.016960876155644657,
      "grad_norm": 0.08547873049974442,
      "learning_rate": 7.120832083343337e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040565449453424665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33644561767578,
      "state_query_norm": 1406.6804931640625,
      "state_relation_loss": 0.00040565449453424665,
      "step": 25100,
      "student_distance_mean": 1937.9206665039062,
      "teacher_distance_mean": 1.1450154066085816,
      "weighted_auxiliary_loss": 0.00040565449453424665,
      "weighted_relation_loss": 0.00040565449453424665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.489242282507014,
      "flow_matching_loss": 0.014889966323971748,
      "grad_norm": 0.08855503052473068,
      "learning_rate": 7.092509675514369e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038818163448013366,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89270172119141,
      "state_query_norm": 1431.2549194335938,
      "state_relation_loss": 0.00038818163448013366,
      "step": 25110,
      "student_distance_mean": 1970.8627807617188,
      "teacher_distance_mean": 1.1655501484870912,
      "weighted_auxiliary_loss": 0.00038818163448013366,
      "weighted_relation_loss": 0.00038818163448013366
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.498596819457436,
      "flow_matching_loss": 0.016857872996479274,
      "grad_norm": 0.10042940080165863,
      "learning_rate": 7.064239404332063e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042154772381763905,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.109937286376955,
      "state_query_norm": 1405.984228515625,
      "state_relation_loss": 0.00042154772381763905,
      "step": 25120,
      "student_distance_mean": 1927.3059448242188,
      "teacher_distance_mean": 1.141261923313141,
      "weighted_auxiliary_loss": 0.00042154772381763905,
      "weighted_relation_loss": 0.00042154772381763905
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.507951356407858,
      "flow_matching_loss": 0.017000585701316594,
      "grad_norm": 0.07448410242795944,
      "learning_rate": 7.03602130414745e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038017494953237476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1825756072998,
      "state_query_norm": 1402.6547607421876,
      "state_relation_loss": 0.00038017494953237476,
      "step": 25130,
      "student_distance_mean": 1930.5031372070312,
      "teacher_distance_mean": 1.1436864495277406,
      "weighted_auxiliary_loss": 0.00038017494953237476,
      "weighted_relation_loss": 0.00038017494953237476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.51730589335828,
      "flow_matching_loss": 0.01784347128123045,
      "grad_norm": 0.11829518526792526,
      "learning_rate": 7.007855409248198e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004084286920260638,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.634194946289064,
      "state_query_norm": 1420.8935302734376,
      "state_relation_loss": 0.0004084286920260638,
      "step": 25140,
      "student_distance_mean": 1955.4072021484376,
      "teacher_distance_mean": 1.1601388692855834,
      "weighted_auxiliary_loss": 0.0004084286920260638,
      "weighted_relation_loss": 0.0004084286920260638
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.5266604303087,
      "flow_matching_loss": 0.016574359126389028,
      "grad_norm": 0.0720193088054657,
      "learning_rate": 6.979741753858521e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003790506743825972,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47352752685547,
      "state_query_norm": 1422.7957397460937,
      "state_relation_loss": 0.0003790506743825972,
      "step": 25150,
      "student_distance_mean": 1950.0727172851562,
      "teacher_distance_mean": 1.1566225647926331,
      "weighted_auxiliary_loss": 0.0003790506743825972,
      "weighted_relation_loss": 0.0003790506743825972
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.53601496725912,
      "flow_matching_loss": 0.017824449669569732,
      "grad_norm": 0.08131735026836395,
      "learning_rate": 6.951680372139158e-06,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039666332595515995,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.383332061767575,
      "state_query_norm": 1420.1232299804688,
      "state_relation_loss": 0.00039666332595515995,
      "step": 25160,
      "student_distance_mean": 1945.498291015625,
      "teacher_distance_mean": 1.1537871956825256,
      "weighted_auxiliary_loss": 0.00039666332595515995,
      "weighted_relation_loss": 0.00039666332595515995
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.545369504209543,
      "flow_matching_loss": 0.018309243582189084,
      "grad_norm": 0.08127239346504211,
      "learning_rate": 6.923671298187334e-06,
      "loss": 0.0187,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003991962963482365,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.65741539001465,
      "state_query_norm": 1432.3075805664062,
      "state_relation_loss": 0.0003991962963482365,
      "step": 25170,
      "student_distance_mean": 1961.5039306640624,
      "teacher_distance_mean": 1.163249659538269,
      "weighted_auxiliary_loss": 0.0003991962963482365,
      "weighted_relation_loss": 0.0003991962963482365
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.55472404115996,
      "flow_matching_loss": 0.01554804090410471,
      "grad_norm": 0.08151061087846756,
      "learning_rate": 6.895714566036704e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000396517789340578,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43281478881836,
      "state_query_norm": 1422.9738525390626,
      "state_relation_loss": 0.000396517789340578,
      "step": 25180,
      "student_distance_mean": 1946.919482421875,
      "teacher_distance_mean": 1.153943085670471,
      "weighted_auxiliary_loss": 0.000396517789340578,
      "weighted_relation_loss": 0.000396517789340578
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.564078578110383,
      "flow_matching_loss": 0.016154700983315708,
      "grad_norm": 0.11151725798845291,
      "learning_rate": 6.86781020965736e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038852237339597194,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26945152282715,
      "state_query_norm": 1413.9432373046875,
      "state_relation_loss": 0.00038852237339597194,
      "step": 25190,
      "student_distance_mean": 1938.8893798828126,
      "teacher_distance_mean": 1.148729956150055,
      "weighted_auxiliary_loss": 0.00038852237339597194,
      "weighted_relation_loss": 0.00038852237339597194
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.573433115060805,
      "flow_matching_loss": 0.016726154182106257,
      "grad_norm": 0.08739788085222244,
      "learning_rate": 6.839958262955709e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004102285922272131,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.199005889892575,
      "state_query_norm": 1408.2901733398437,
      "state_relation_loss": 0.0004102285922272131,
      "step": 25200,
      "student_distance_mean": 1931.7860595703125,
      "teacher_distance_mean": 1.1451400160789489,
      "weighted_auxiliary_loss": 0.0004102285922272131,
      "weighted_relation_loss": 0.0004102285922272131
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.582787652011227,
      "flow_matching_loss": 0.016866037622094154,
      "grad_norm": 0.09454939514398575,
      "learning_rate": 6.8121587597744894e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003915059962309897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06696701049805,
      "state_query_norm": 1410.1287109375,
      "state_relation_loss": 0.0003915059962309897,
      "step": 25210,
      "student_distance_mean": 1930.2580322265626,
      "teacher_distance_mean": 1.1439868330955505,
      "weighted_auxiliary_loss": 0.0003915059962309897,
      "weighted_relation_loss": 0.0003915059962309897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.592142188961645,
      "flow_matching_loss": 0.016806038655340673,
      "grad_norm": 0.07550310343503952,
      "learning_rate": 6.784411733892732e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039965932082850484,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17476463317871,
      "state_query_norm": 1409.7771240234374,
      "state_relation_loss": 0.00039965932082850484,
      "step": 25220,
      "student_distance_mean": 1932.6238891601563,
      "teacher_distance_mean": 1.1471960902214051,
      "weighted_auxiliary_loss": 0.00039965932082850484,
      "weighted_relation_loss": 0.00039965932082850484
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.601496725912067,
      "flow_matching_loss": 0.01517120748758316,
      "grad_norm": 0.09137774258852005,
      "learning_rate": 6.756717219025666e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039691657875664534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.475518798828126,
      "state_query_norm": 1420.9360107421876,
      "state_relation_loss": 0.00039691657875664534,
      "step": 25230,
      "student_distance_mean": 1952.3016235351563,
      "teacher_distance_mean": 1.1593652963638306,
      "weighted_auxiliary_loss": 0.00039691657875664534,
      "weighted_relation_loss": 0.00039691657875664534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.61085126286249,
      "flow_matching_loss": 0.016294137109071016,
      "grad_norm": 0.0933559387922287,
      "learning_rate": 6.729075248824762e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003743184235645458,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.22235641479492,
      "state_query_norm": 1406.2702392578126,
      "state_relation_loss": 0.0003743184235645458,
      "step": 25240,
      "student_distance_mean": 1932.4516723632812,
      "teacher_distance_mean": 1.1470409035682678,
      "weighted_auxiliary_loss": 0.0003743184235645458,
      "weighted_relation_loss": 0.0003743184235645458
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.620205799812908,
      "flow_matching_loss": 0.016469617001712324,
      "grad_norm": 0.07825244218111038,
      "learning_rate": 6.701485856877615e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003871289809467271,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29470634460449,
      "state_query_norm": 1417.6253051757812,
      "state_relation_loss": 0.0003871289809467271,
      "step": 25250,
      "student_distance_mean": 1941.317236328125,
      "teacher_distance_mean": 1.1516464114189149,
      "weighted_auxiliary_loss": 0.0003871289809467271,
      "weighted_relation_loss": 0.0003871289809467271
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.62956033676333,
      "flow_matching_loss": 0.01741629783064127,
      "grad_norm": 0.07662168890237808,
      "learning_rate": 6.673949076707925e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037789971684105695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38788833618164,
      "state_query_norm": 1414.0734497070312,
      "state_relation_loss": 0.00037789971684105695,
      "step": 25260,
      "student_distance_mean": 1941.0777587890625,
      "teacher_distance_mean": 1.1533538579940796,
      "weighted_auxiliary_loss": 0.00037789971684105695,
      "weighted_relation_loss": 0.00037789971684105695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.63891487371375,
      "flow_matching_loss": 0.016993409674614667,
      "grad_norm": 0.09280584752559662,
      "learning_rate": 6.646464941775499e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003968886419897899,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.277055740356445,
      "state_query_norm": 1416.5141967773438,
      "state_relation_loss": 0.0003968886419897899,
      "step": 25270,
      "student_distance_mean": 1939.8834716796875,
      "teacher_distance_mean": 1.1527075052261353,
      "weighted_auxiliary_loss": 0.0003968886419897899,
      "weighted_relation_loss": 0.0003968886419897899
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.648269410664174,
      "flow_matching_loss": 0.01485285647213459,
      "grad_norm": 0.07715414464473724,
      "learning_rate": 6.619033485476128e-06,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003925125871319324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.08047180175781,
      "state_query_norm": 1436.5778442382812,
      "state_relation_loss": 0.0003925125871319324,
      "step": 25280,
      "student_distance_mean": 1981.5347290039062,
      "teacher_distance_mean": 1.1766892552375794,
      "weighted_auxiliary_loss": 0.0003925125871319324,
      "weighted_relation_loss": 0.0003925125871319324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.657623947614592,
      "flow_matching_loss": 0.017128119058907033,
      "grad_norm": 0.07152406126260757,
      "learning_rate": 6.59165474114164e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003713980258908123,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21817283630371,
      "state_query_norm": 1409.382080078125,
      "state_relation_loss": 0.0003713980258908123,
      "step": 25290,
      "student_distance_mean": 1935.7111938476562,
      "teacher_distance_mean": 1.1495113492012023,
      "weighted_auxiliary_loss": 0.0003713980258908123,
      "weighted_relation_loss": 0.0003713980258908123
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.666978484565014,
      "flow_matching_loss": 0.016236275155097245,
      "grad_norm": 0.06931286305189133,
      "learning_rate": 6.564328742039783e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037758151011075826,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30533676147461,
      "state_query_norm": 1411.73349609375,
      "state_relation_loss": 0.00037758151011075826,
      "step": 25300,
      "student_distance_mean": 1937.948193359375,
      "teacher_distance_mean": 1.1505742192268371,
      "weighted_auxiliary_loss": 0.00037758151011075826,
      "weighted_relation_loss": 0.00037758151011075826
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.676333021515436,
      "flow_matching_loss": 0.016854822915047408,
      "grad_norm": 0.111670583486557,
      "learning_rate": 6.537055521374219e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039723094087094066,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52257804870605,
      "state_query_norm": 1419.85185546875,
      "state_relation_loss": 0.00039723094087094066,
      "step": 25310,
      "student_distance_mean": 1951.2292724609374,
      "teacher_distance_mean": 1.1592745065689087,
      "weighted_auxiliary_loss": 0.00039723094087094066,
      "weighted_relation_loss": 0.00039723094087094066
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.685687558465855,
      "flow_matching_loss": 0.017423216719180346,
      "grad_norm": 0.09928525984287262,
      "learning_rate": 6.509835112284485e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004019824496936053,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66459846496582,
      "state_query_norm": 1418.2823120117187,
      "state_relation_loss": 0.0004019824496936053,
      "step": 25320,
      "student_distance_mean": 1956.2177490234376,
      "teacher_distance_mean": 1.1624308228492737,
      "weighted_auxiliary_loss": 0.0004019824496936053,
      "weighted_relation_loss": 0.0004019824496936053
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.695042095416277,
      "flow_matching_loss": 0.01643900303170085,
      "grad_norm": 0.08299274742603302,
      "learning_rate": 6.482667547845944e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004086522210855037,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.84594955444336,
      "state_query_norm": 1427.7889526367187,
      "state_relation_loss": 0.0004086522210855037,
      "step": 25330,
      "student_distance_mean": 1966.0468872070312,
      "teacher_distance_mean": 1.1678781509399414,
      "weighted_auxiliary_loss": 0.0004086522210855037,
      "weighted_relation_loss": 0.0004086522210855037
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.7043966323667,
      "flow_matching_loss": 0.016937519516795875,
      "grad_norm": 0.09212184697389603,
      "learning_rate": 6.455552861069736e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004118968092370778,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37346572875977,
      "state_query_norm": 1412.3703491210938,
      "state_relation_loss": 0.0004118968092370778,
      "step": 25340,
      "student_distance_mean": 1940.9270629882812,
      "teacher_distance_mean": 1.1517697334289552,
      "weighted_auxiliary_loss": 0.0004118968092370778,
      "weighted_relation_loss": 0.0004118968092370778
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.71375116931712,
      "flow_matching_loss": 0.017611800506711007,
      "grad_norm": 0.08691681176424026,
      "learning_rate": 6.428491084902788e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041700474685058,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.34274520874023,
      "state_query_norm": 1413.0734252929688,
      "state_relation_loss": 0.00041700474685058,
      "step": 25350,
      "student_distance_mean": 1941.7048461914062,
      "teacher_distance_mean": 1.151496911048889,
      "weighted_auxiliary_loss": 0.00041700474685058,
      "weighted_relation_loss": 0.00041700474685058
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.72310570626754,
      "flow_matching_loss": 0.01575728375464678,
      "grad_norm": 0.07552240043878555,
      "learning_rate": 6.401482252227698e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038708194042555986,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33350105285645,
      "state_query_norm": 1418.523681640625,
      "state_relation_loss": 0.00038708194042555986,
      "step": 25360,
      "student_distance_mean": 1944.4454467773437,
      "teacher_distance_mean": 1.153493845462799,
      "weighted_auxiliary_loss": 0.00038708194042555986,
      "weighted_relation_loss": 0.00038708194042555986
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.73246024321796,
      "flow_matching_loss": 0.01489677168428898,
      "grad_norm": 0.07992721349000931,
      "learning_rate": 6.37452639586274e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039986322808545084,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41200828552246,
      "state_query_norm": 1416.2841430664062,
      "state_relation_loss": 0.00039986322808545084,
      "step": 25370,
      "student_distance_mean": 1947.15,
      "teacher_distance_mean": 1.1559306383132935,
      "weighted_auxiliary_loss": 0.00039986322808545084,
      "weighted_relation_loss": 0.00039986322808545084
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.741814780168383,
      "flow_matching_loss": 0.015454738121479749,
      "grad_norm": 0.07507173717021942,
      "learning_rate": 6.347623548561826e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003945247852243483,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31006202697754,
      "state_query_norm": 1403.1027709960938,
      "state_relation_loss": 0.0003945247852243483,
      "step": 25380,
      "student_distance_mean": 1934.7092529296874,
      "teacher_distance_mean": 1.1477420568466186,
      "weighted_auxiliary_loss": 0.0003945247852243483,
      "weighted_relation_loss": 0.0003945247852243483
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.7511693171188,
      "flow_matching_loss": 0.016545328218489884,
      "grad_norm": 0.07080033421516418,
      "learning_rate": 6.320773743014441e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040126857347786424,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.502428436279295,
      "state_query_norm": 1419.0837890625,
      "state_relation_loss": 0.00040126857347786424,
      "step": 25390,
      "student_distance_mean": 1951.5394775390625,
      "teacher_distance_mean": 1.1581907510757445,
      "weighted_auxiliary_loss": 0.00040126857347786424,
      "weighted_relation_loss": 0.00040126857347786424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.760523854069223,
      "flow_matching_loss": 0.016606110241264106,
      "grad_norm": 0.0770648643374443,
      "learning_rate": 6.293977011845648e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003847107582259923,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.74203395843506,
      "state_query_norm": 1392.62705078125,
      "state_relation_loss": 0.0003847107582259923,
      "step": 25400,
      "student_distance_mean": 1911.0856689453126,
      "teacher_distance_mean": 1.1325648069381713,
      "weighted_auxiliary_loss": 0.0003847107582259923,
      "weighted_relation_loss": 0.0003847107582259923
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.769878391019645,
      "flow_matching_loss": 0.015576518047600985,
      "grad_norm": 0.09346255660057068,
      "learning_rate": 6.2672333876159835e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004049318726174533,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53422164916992,
      "state_query_norm": 1412.165625,
      "state_relation_loss": 0.0004049318726174533,
      "step": 25410,
      "student_distance_mean": 1944.6695556640625,
      "teacher_distance_mean": 1.1518637776374816,
      "weighted_auxiliary_loss": 0.0004049318726174533,
      "weighted_relation_loss": 0.0004049318726174533
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.779232927970064,
      "flow_matching_loss": 0.01680694576352835,
      "grad_norm": 0.09926716983318329,
      "learning_rate": 6.2405429028215e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039469363691750913,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.698482513427734,
      "state_query_norm": 1428.3050659179687,
      "state_relation_loss": 0.00039469363691750913,
      "step": 25420,
      "student_distance_mean": 1959.1079956054687,
      "teacher_distance_mean": 1.160603892803192,
      "weighted_auxiliary_loss": 0.00039469363691750913,
      "weighted_relation_loss": 0.00039469363691750913
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.788587464920486,
      "flow_matching_loss": 0.017169633600860833,
      "grad_norm": 0.09972190111875534,
      "learning_rate": 6.213905589893631e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000373315264005214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15150108337402,
      "state_query_norm": 1411.751513671875,
      "state_relation_loss": 0.000373315264005214,
      "step": 25430,
      "student_distance_mean": 1934.5772827148437,
      "teacher_distance_mean": 1.1468579411506652,
      "weighted_auxiliary_loss": 0.000373315264005214,
      "weighted_relation_loss": 0.000373315264005214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.797942001870908,
      "flow_matching_loss": 0.016062835231423378,
      "grad_norm": 0.08254706114530563,
      "learning_rate": 6.187321481199221e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038407679530791936,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14036827087402,
      "state_query_norm": 1406.7195190429688,
      "state_relation_loss": 0.00038407679530791936,
      "step": 25440,
      "student_distance_mean": 1931.821728515625,
      "teacher_distance_mean": 1.1463442683219909,
      "weighted_auxiliary_loss": 0.00038407679530791936,
      "weighted_relation_loss": 0.00038407679530791936
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.80729653882133,
      "flow_matching_loss": 0.017997167352586985,
      "grad_norm": 0.08042147010564804,
      "learning_rate": 6.1607906090404895e-06,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039371414750348775,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33248596191406,
      "state_query_norm": 1422.4681640625,
      "state_relation_loss": 0.00039371414750348775,
      "step": 25450,
      "student_distance_mean": 1944.1908081054687,
      "teacher_distance_mean": 1.1526763796806336,
      "weighted_auxiliary_loss": 0.00039371414750348775,
      "weighted_relation_loss": 0.00039371414750348775
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.81665107577175,
      "flow_matching_loss": 0.017466573137789966,
      "grad_norm": 0.07759161293506622,
      "learning_rate": 6.134313005654929e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004217935202177614,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02767715454102,
      "state_query_norm": 1439.7420166015625,
      "state_relation_loss": 0.0004217935202177614,
      "step": 25460,
      "student_distance_mean": 1979.5606689453125,
      "teacher_distance_mean": 1.1735613226890564,
      "weighted_auxiliary_loss": 0.0004217935202177614,
      "weighted_relation_loss": 0.0004217935202177614
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.82600561272217,
      "flow_matching_loss": 0.017640838399529457,
      "grad_norm": 0.07738036662340164,
      "learning_rate": 6.107888703215337e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003882846707710996,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.00212783813477,
      "state_query_norm": 1398.4288818359375,
      "state_relation_loss": 0.0003882846707710996,
      "step": 25470,
      "student_distance_mean": 1920.5232177734374,
      "teacher_distance_mean": 1.1387296557426452,
      "weighted_auxiliary_loss": 0.0003882846707710996,
      "weighted_relation_loss": 0.0003882846707710996
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.835360149672592,
      "flow_matching_loss": 0.016362482123076914,
      "grad_norm": 0.0839749351143837,
      "learning_rate": 6.081517733829722e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039715044549666346,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.741630935668944,
      "state_query_norm": 1425.9425415039063,
      "state_relation_loss": 0.00039715044549666346,
      "step": 25480,
      "student_distance_mean": 1962.0109252929688,
      "teacher_distance_mean": 1.1629980564117433,
      "weighted_auxiliary_loss": 0.00039715044549666346,
      "weighted_relation_loss": 0.00039715044549666346
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.84471468662301,
      "flow_matching_loss": 0.015670929569751024,
      "grad_norm": 0.09459077566862106,
      "learning_rate": 6.055200129541294e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003977147862315178,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5846492767334,
      "state_query_norm": 1417.0217041015626,
      "state_relation_loss": 0.0003977147862315178,
      "step": 25490,
      "student_distance_mean": 1954.1962524414062,
      "teacher_distance_mean": 1.1591374278068542,
      "weighted_auxiliary_loss": 0.0003977147862315178,
      "weighted_relation_loss": 0.0003977147862315178
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.854069223573433,
      "flow_matching_loss": 0.016776561737060547,
      "grad_norm": 0.11799731850624084,
      "learning_rate": 6.0289359223284434e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037314792280085385,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.72241058349609,
      "state_query_norm": 1386.4529174804688,
      "state_relation_loss": 0.00037314792280085385,
      "step": 25500,
      "student_distance_mean": 1906.95166015625,
      "teacher_distance_mean": 1.1311758041381836,
      "weighted_auxiliary_loss": 0.00037314792280085385,
      "weighted_relation_loss": 0.00037314792280085385
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.863423760523855,
      "flow_matching_loss": 0.01688752034679055,
      "grad_norm": 0.09604865312576294,
      "learning_rate": 6.002725144104648e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039031346968840806,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.86944923400879,
      "state_query_norm": 1431.737841796875,
      "state_relation_loss": 0.00039031346968840806,
      "step": 25510,
      "student_distance_mean": 1970.8408203125,
      "teacher_distance_mean": 1.170027530193329,
      "weighted_auxiliary_loss": 0.00039031346968840806,
      "weighted_relation_loss": 0.00039031346968840806
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.872778297474277,
      "flow_matching_loss": 0.017671805806457996,
      "grad_norm": 0.09225394576787949,
      "learning_rate": 5.976567826718477e-06,
      "loss": 0.0181,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000404724245890975,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15426254272461,
      "state_query_norm": 1411.532666015625,
      "state_relation_loss": 0.000404724245890975,
      "step": 25520,
      "student_distance_mean": 1931.987939453125,
      "teacher_distance_mean": 1.1468784809112549,
      "weighted_auxiliary_loss": 0.000404724245890975,
      "weighted_relation_loss": 0.000404724245890975
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.882132834424695,
      "flow_matching_loss": 0.017902775667607784,
      "grad_norm": 0.09230566769838333,
      "learning_rate": 5.9504640019535315e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003797056328039616,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.76498260498047,
      "state_query_norm": 1398.7186889648438,
      "state_relation_loss": 0.0003797056328039616,
      "step": 25530,
      "student_distance_mean": 1916.15419921875,
      "teacher_distance_mean": 1.1357909083366393,
      "weighted_auxiliary_loss": 0.0003797056328039616,
      "weighted_relation_loss": 0.0003797056328039616
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.891487371375117,
      "flow_matching_loss": 0.016208344511687754,
      "grad_norm": 0.09502632170915604,
      "learning_rate": 5.924413701528414e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003930238919565454,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78018035888672,
      "state_query_norm": 1420.9562377929688,
      "state_relation_loss": 0.0003930238919565454,
      "step": 25540,
      "student_distance_mean": 1959.5026245117188,
      "teacher_distance_mean": 1.1614022374153137,
      "weighted_auxiliary_loss": 0.0003930238919565454,
      "weighted_relation_loss": 0.0003930238919565454
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.90084190832554,
      "flow_matching_loss": 0.01758743356913328,
      "grad_norm": 0.08598779886960983,
      "learning_rate": 5.898416957096703e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041765368368942293,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.704261779785156,
      "state_query_norm": 1434.0507568359376,
      "state_relation_loss": 0.00041765368368942293,
      "step": 25550,
      "student_distance_mean": 1962.420263671875,
      "teacher_distance_mean": 1.162111473083496,
      "weighted_auxiliary_loss": 0.00041765368368942293,
      "weighted_relation_loss": 0.00041765368368942293
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.910196445275957,
      "flow_matching_loss": 0.015684852097183465,
      "grad_norm": 0.08686058223247528,
      "learning_rate": 5.872473800246913e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003858037787722424,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.91245002746582,
      "state_query_norm": 1393.9453857421875,
      "state_relation_loss": 0.0003858037787722424,
      "step": 25560,
      "student_distance_mean": 1917.7478881835937,
      "teacher_distance_mean": 1.1354020237922668,
      "weighted_auxiliary_loss": 0.0003858037787722424,
      "weighted_relation_loss": 0.0003858037787722424
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.91955098222638,
      "flow_matching_loss": 0.014891573041677476,
      "grad_norm": 0.08473470062017441,
      "learning_rate": 5.846584262502402e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003834865026874468,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.991410446166995,
      "state_query_norm": 1433.29970703125,
      "state_relation_loss": 0.0003834865026874468,
      "step": 25570,
      "student_distance_mean": 1974.99990234375,
      "teacher_distance_mean": 1.1699272036552428,
      "weighted_auxiliary_loss": 0.0003834865026874468,
      "weighted_relation_loss": 0.0003834865026874468
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.9289055191768,
      "flow_matching_loss": 0.01571806101128459,
      "grad_norm": 0.09556949138641357,
      "learning_rate": 5.820748375321411e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003930879989638925,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31884536743164,
      "state_query_norm": 1423.98876953125,
      "state_relation_loss": 0.0003930879989638925,
      "step": 25580,
      "student_distance_mean": 1942.5996459960938,
      "teacher_distance_mean": 1.1509769797325133,
      "weighted_auxiliary_loss": 0.0003930879989638925,
      "weighted_relation_loss": 0.0003930879989638925
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.938260056127223,
      "flow_matching_loss": 0.016461122035980224,
      "grad_norm": 0.09349757432937622,
      "learning_rate": 5.794966170096977e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000384262771694921,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.40989761352539,
      "state_query_norm": 1421.7162841796876,
      "state_relation_loss": 0.000384262771694921,
      "step": 25590,
      "student_distance_mean": 1947.9206420898438,
      "teacher_distance_mean": 1.1540812969207763,
      "weighted_auxiliary_loss": 0.000384262771694921,
      "weighted_relation_loss": 0.000384262771694921
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.947614593077642,
      "flow_matching_loss": 0.015956032928079365,
      "grad_norm": 0.08120612055063248,
      "learning_rate": 5.769237678156897e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037470832758117466,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20843772888183,
      "state_query_norm": 1419.68837890625,
      "state_relation_loss": 0.00037470832758117466,
      "step": 25600,
      "student_distance_mean": 1938.5575805664062,
      "teacher_distance_mean": 1.14842791557312,
      "weighted_auxiliary_loss": 0.00037470832758117466,
      "weighted_relation_loss": 0.00037470832758117466
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.956969130028064,
      "flow_matching_loss": 0.015930190589278935,
      "grad_norm": 0.07063991576433182,
      "learning_rate": 5.743562930763735e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00035676047846209256,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.81162071228027,
      "state_query_norm": 1404.5941162109375,
      "state_relation_loss": 0.00035676047846209256,
      "step": 25610,
      "student_distance_mean": 1916.0235595703125,
      "teacher_distance_mean": 1.1358205556869507,
      "weighted_auxiliary_loss": 0.00035676047846209256,
      "weighted_relation_loss": 0.00035676047846209256
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.966323666978486,
      "flow_matching_loss": 0.016680996306240557,
      "grad_norm": 0.08287913352251053,
      "learning_rate": 5.717941959114726e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039465154986828566,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55603523254395,
      "state_query_norm": 1423.9158569335937,
      "state_relation_loss": 0.00039465154986828566,
      "step": 25620,
      "student_distance_mean": 1955.9452880859376,
      "teacher_distance_mean": 1.1588362812995912,
      "weighted_auxiliary_loss": 0.00039465154986828566,
      "weighted_relation_loss": 0.00039465154986828566
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.975678203928904,
      "flow_matching_loss": 0.01618429822847247,
      "grad_norm": 0.07455728203058243,
      "learning_rate": 5.69237479434176e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000450008109328337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71744232177734,
      "state_query_norm": 1421.5939086914063,
      "state_relation_loss": 0.000450008109328337,
      "step": 25630,
      "student_distance_mean": 1956.6995361328125,
      "teacher_distance_mean": 1.158715009689331,
      "weighted_auxiliary_loss": 0.000450008109328337,
      "weighted_relation_loss": 0.000450008109328337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.985032740879326,
      "flow_matching_loss": 0.019482151325792074,
      "grad_norm": 0.10771138966083527,
      "learning_rate": 5.666861467511353e-06,
      "loss": 0.0199,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000420543149812147,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.37468147277832,
      "state_query_norm": 1446.9787475585938,
      "state_relation_loss": 0.000420543149812147,
      "step": 25640,
      "student_distance_mean": 1995.4706665039062,
      "teacher_distance_mean": 1.1808295369148254,
      "weighted_auxiliary_loss": 0.000420543149812147,
      "weighted_relation_loss": 0.000420543149812147
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 23.99438727782975,
      "flow_matching_loss": 0.017826172057539225,
      "grad_norm": 0.08719322085380554,
      "learning_rate": 5.641402009624591e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042742048390209675,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.0282844543457,
      "state_query_norm": 1403.0638427734375,
      "state_relation_loss": 0.00042742048390209675,
      "step": 25650,
      "student_distance_mean": 1923.1811401367188,
      "teacher_distance_mean": 1.1393270134925841,
      "weighted_auxiliary_loss": 0.00042742048390209675,
      "weighted_relation_loss": 0.00042742048390209675
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.003741814780167,
      "flow_matching_loss": 0.01793204713612795,
      "grad_norm": 0.08621978014707565,
      "learning_rate": 5.615996451617145e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00040210116130765525,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.25816421508789,
      "state_query_norm": 1412.1568603515625,
      "state_relation_loss": 0.00040210116130765525,
      "step": 25660,
      "student_distance_mean": 1935.8792602539063,
      "teacher_distance_mean": 1.1477054953575134,
      "weighted_auxiliary_loss": 0.00040210116130765525,
      "weighted_relation_loss": 0.00040210116130765525
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.01309635173059,
      "flow_matching_loss": 0.017262460477650166,
      "grad_norm": 0.09785878658294678,
      "learning_rate": 5.590644824359148e-06,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004140387492952868,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57678070068359,
      "state_query_norm": 1414.5521850585938,
      "state_relation_loss": 0.0004140387492952868,
      "step": 25670,
      "student_distance_mean": 1951.1328369140624,
      "teacher_distance_mean": 1.1565730094909668,
      "weighted_auxiliary_loss": 0.0004140387492952868,
      "weighted_relation_loss": 0.0004140387492952868
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.02245088868101,
      "flow_matching_loss": 0.01640084683895111,
      "grad_norm": 0.07369852066040039,
      "learning_rate": 5.56534715865521e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037236067000776527,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.69205093383789,
      "state_query_norm": 1392.8689331054688,
      "state_relation_loss": 0.00037236067000776527,
      "step": 25680,
      "student_distance_mean": 1907.4445678710938,
      "teacher_distance_mean": 1.129972755908966,
      "weighted_auxiliary_loss": 0.00037236067000776527,
      "weighted_relation_loss": 0.00037236067000776527
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.031805425631433,
      "flow_matching_loss": 0.01667635142803192,
      "grad_norm": 0.0801120325922966,
      "learning_rate": 5.5401034852443965e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004101818893104792,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29221649169922,
      "state_query_norm": 1414.3809936523437,
      "state_relation_loss": 0.0004101818893104792,
      "step": 25690,
      "student_distance_mean": 1936.5583129882812,
      "teacher_distance_mean": 1.147730565071106,
      "weighted_auxiliary_loss": 0.0004101818893104792,
      "weighted_relation_loss": 0.0004101818893104792
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.04115996258185,
      "flow_matching_loss": 0.015722255036234856,
      "grad_norm": 0.09483359009027481,
      "learning_rate": 5.514913834800134e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004041970852995291,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62550010681152,
      "state_query_norm": 1422.6281616210938,
      "state_relation_loss": 0.0004041970852995291,
      "step": 25700,
      "student_distance_mean": 1955.9081909179688,
      "teacher_distance_mean": 1.1572587370872498,
      "weighted_auxiliary_loss": 0.0004041970852995291,
      "weighted_relation_loss": 0.0004041970852995291
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.050514499532273,
      "flow_matching_loss": 0.017852486297488212,
      "grad_norm": 0.08938150852918625,
      "learning_rate": 5.489778237930238e-06,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037952136772219093,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.277564239501956,
      "state_query_norm": 1409.6464111328125,
      "state_relation_loss": 0.00037952136772219093,
      "step": 25710,
      "student_distance_mean": 1938.8931274414062,
      "teacher_distance_mean": 1.1479936599731446,
      "weighted_auxiliary_loss": 0.00037952136772219093,
      "weighted_relation_loss": 0.00037952136772219093
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.059869036482695,
      "flow_matching_loss": 0.01623953813686967,
      "grad_norm": 0.09142784029245377,
      "learning_rate": 5.464696725176827e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041521621169522406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.360089492797854,
      "state_query_norm": 1410.4914672851562,
      "state_relation_loss": 0.00041521621169522406,
      "step": 25720,
      "student_distance_mean": 1937.6047485351562,
      "teacher_distance_mean": 1.1477172136306764,
      "weighted_auxiliary_loss": 0.00041521621169522406,
      "weighted_relation_loss": 0.00041521621169522406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.069223573433113,
      "flow_matching_loss": 0.015268525574356318,
      "grad_norm": 0.08760782331228256,
      "learning_rate": 5.439669327016294e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003823133767582476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23599281311035,
      "state_query_norm": 1402.1955078125,
      "state_relation_loss": 0.0003823133767582476,
      "step": 25730,
      "student_distance_mean": 1932.3171264648438,
      "teacher_distance_mean": 1.1444005370140076,
      "weighted_auxiliary_loss": 0.0003823133767582476,
      "weighted_relation_loss": 0.0003823133767582476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.078578110383535,
      "flow_matching_loss": 0.016115651186555623,
      "grad_norm": 0.07544038444757462,
      "learning_rate": 5.4146960738592985e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039476533129345626,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.277733612060544,
      "state_query_norm": 1408.5887939453125,
      "state_relation_loss": 0.00039476533129345626,
      "step": 25740,
      "student_distance_mean": 1933.9779541015625,
      "teacher_distance_mean": 1.1476290702819825,
      "weighted_auxiliary_loss": 0.00039476533129345626,
      "weighted_relation_loss": 0.00039476533129345626
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.087932647333957,
      "flow_matching_loss": 0.01565370038151741,
      "grad_norm": 0.07901531457901001,
      "learning_rate": 5.389776996050694e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003940684779081494,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.59419174194336,
      "state_query_norm": 1426.6791015625,
      "state_relation_loss": 0.0003940684779081494,
      "step": 25750,
      "student_distance_mean": 1955.948681640625,
      "teacher_distance_mean": 1.1605528950691224,
      "weighted_auxiliary_loss": 0.0003940684779081494,
      "weighted_relation_loss": 0.0003940684779081494
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.09728718428438,
      "flow_matching_loss": 0.01660022232681513,
      "grad_norm": 0.09866593778133392,
      "learning_rate": 5.364912123869492e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039028930477797986,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.67763748168945,
      "state_query_norm": 1421.8949951171876,
      "state_relation_loss": 0.00039028930477797986,
      "step": 25760,
      "student_distance_mean": 1959.011181640625,
      "teacher_distance_mean": 1.1619267463684082,
      "weighted_auxiliary_loss": 0.00039028930477797986,
      "weighted_relation_loss": 0.00039028930477797986
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.106641721234798,
      "flow_matching_loss": 0.014169700350612402,
      "grad_norm": 0.07923593372106552,
      "learning_rate": 5.3401014875288864e-06,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003912376152584329,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.520202255249025,
      "state_query_norm": 1417.0734008789063,
      "state_relation_loss": 0.0003912376152584329,
      "step": 25770,
      "student_distance_mean": 1945.37373046875,
      "teacher_distance_mean": 1.1539811491966248,
      "weighted_auxiliary_loss": 0.0003912376152584329,
      "weighted_relation_loss": 0.0003912376152584329
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.11599625818522,
      "flow_matching_loss": 0.018928514700382947,
      "grad_norm": 0.09754958748817444,
      "learning_rate": 5.315345117176129e-06,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039065957244019953,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73692321777344,
      "state_query_norm": 1429.0236328125,
      "state_relation_loss": 0.00039065957244019953,
      "step": 25780,
      "student_distance_mean": 1960.3491577148438,
      "teacher_distance_mean": 1.1628676414489747,
      "weighted_auxiliary_loss": 0.00039065957244019953,
      "weighted_relation_loss": 0.00039065957244019953
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.125350795135642,
      "flow_matching_loss": 0.017013722099363804,
      "grad_norm": 0.08894245326519012,
      "learning_rate": 5.2906430428925405e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041710783552844076,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25563545227051,
      "state_query_norm": 1407.6972412109376,
      "state_relation_loss": 0.00041710783552844076,
      "step": 25790,
      "student_distance_mean": 1931.5643310546875,
      "teacher_distance_mean": 1.14616916179657,
      "weighted_auxiliary_loss": 0.00041710783552844076,
      "weighted_relation_loss": 0.00041710783552844076
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.13470533208606,
      "flow_matching_loss": 0.01690022610127926,
      "grad_norm": 0.09982521086931229,
      "learning_rate": 5.265995294693471e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003986594296293333,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27855377197265,
      "state_query_norm": 1412.2021728515624,
      "state_relation_loss": 0.0003986594296293333,
      "step": 25800,
      "student_distance_mean": 1936.3207641601562,
      "teacher_distance_mean": 1.1500751972198486,
      "weighted_auxiliary_loss": 0.0003986594296293333,
      "weighted_relation_loss": 0.0003986594296293333
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.144059869036482,
      "flow_matching_loss": 0.014885498955845834,
      "grad_norm": 0.08844389766454697,
      "learning_rate": 5.241401902528254e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003983746806625277,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.468726348876956,
      "state_query_norm": 1417.5175903320312,
      "state_relation_loss": 0.0003983746806625277,
      "step": 25810,
      "student_distance_mean": 1947.2108642578125,
      "teacher_distance_mean": 1.156334900856018,
      "weighted_auxiliary_loss": 0.0003983746806625277,
      "weighted_relation_loss": 0.0003983746806625277
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.153414405986904,
      "flow_matching_loss": 0.017243652511388062,
      "grad_norm": 0.10508273541927338,
      "learning_rate": 5.216862896280189e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040579079650342463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42038192749023,
      "state_query_norm": 1412.88857421875,
      "state_relation_loss": 0.00040579079650342463,
      "step": 25820,
      "student_distance_mean": 1941.7806762695313,
      "teacher_distance_mean": 1.1529604196548462,
      "weighted_auxiliary_loss": 0.00040579079650342463,
      "weighted_relation_loss": 0.00040579079650342463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.162768942937326,
      "flow_matching_loss": 0.01606216570362449,
      "grad_norm": 0.10174749046564102,
      "learning_rate": 5.192378305766471e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040272772894240917,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41526565551758,
      "state_query_norm": 1415.7831420898438,
      "state_relation_loss": 0.00040272772894240917,
      "step": 25830,
      "student_distance_mean": 1941.2400634765625,
      "teacher_distance_mean": 1.151207649707794,
      "weighted_auxiliary_loss": 0.00040272772894240917,
      "weighted_relation_loss": 0.00040272772894240917
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.172123479887745,
      "flow_matching_loss": 0.017637693230062722,
      "grad_norm": 0.0897354930639267,
      "learning_rate": 5.1679481607382065e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003989983320934698,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32417449951172,
      "state_query_norm": 1414.62841796875,
      "state_relation_loss": 0.0003989983320934698,
      "step": 25840,
      "student_distance_mean": 1941.1683715820313,
      "teacher_distance_mean": 1.1508981585502625,
      "weighted_auxiliary_loss": 0.0003989983320934698,
      "weighted_relation_loss": 0.0003989983320934698
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.181478016838167,
      "flow_matching_loss": 0.015487254410982133,
      "grad_norm": 0.062306489795446396,
      "learning_rate": 5.1435724908803105e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003788397094467655,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43343658447266,
      "state_query_norm": 1422.73095703125,
      "state_relation_loss": 0.0003788397094467655,
      "step": 25850,
      "student_distance_mean": 1946.5104125976563,
      "teacher_distance_mean": 1.1556294798851012,
      "weighted_auxiliary_loss": 0.0003788397094467655,
      "weighted_relation_loss": 0.0003788397094467655
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.19083255378859,
      "flow_matching_loss": 0.017421242594718934,
      "grad_norm": 0.10718856006860733,
      "learning_rate": 5.119251325811514e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041027132829185575,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.748023986816406,
      "state_query_norm": 1422.311767578125,
      "state_relation_loss": 0.00041027132829185575,
      "step": 25860,
      "student_distance_mean": 1957.3996215820312,
      "teacher_distance_mean": 1.1623165726661682,
      "weighted_auxiliary_loss": 0.00041027132829185575,
      "weighted_relation_loss": 0.00041027132829185575
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.200187090739007,
      "flow_matching_loss": 0.016158920247107744,
      "grad_norm": 0.07980957627296448,
      "learning_rate": 5.094984695084348e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037274607457220553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64044876098633,
      "state_query_norm": 1422.7967895507813,
      "state_relation_loss": 0.00037274607457220553,
      "step": 25870,
      "student_distance_mean": 1958.2724487304688,
      "teacher_distance_mean": 1.1623359322547913,
      "weighted_auxiliary_loss": 0.00037274607457220553,
      "weighted_relation_loss": 0.00037274607457220553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.20954162768943,
      "flow_matching_loss": 0.01722123036161065,
      "grad_norm": 0.0991373360157013,
      "learning_rate": 5.070772628185039e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003809784073382616,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.688116455078124,
      "state_query_norm": 1432.5296630859375,
      "state_relation_loss": 0.0003809784073382616,
      "step": 25880,
      "student_distance_mean": 1963.3023803710937,
      "teacher_distance_mean": 1.1656888604164124,
      "weighted_auxiliary_loss": 0.0003809784073382616,
      "weighted_relation_loss": 0.0003809784073382616
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.21889616463985,
      "flow_matching_loss": 0.01673988839611411,
      "grad_norm": 0.07443719357252121,
      "learning_rate": 5.046615154533535e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003846881474601105,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.7123363494873,
      "state_query_norm": 1433.5216064453125,
      "state_relation_loss": 0.0003846881474601105,
      "step": 25890,
      "student_distance_mean": 1964.0219970703124,
      "teacher_distance_mean": 1.1660515308380126,
      "weighted_auxiliary_loss": 0.0003846881474601105,
      "weighted_relation_loss": 0.0003846881474601105
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.22825070159027,
      "flow_matching_loss": 0.016929799038916827,
      "grad_norm": 0.08860716968774796,
      "learning_rate": 5.022512303483451e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003824210580205545,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1416015625,
      "state_query_norm": 1416.5111083984375,
      "state_relation_loss": 0.0003824210580205545,
      "step": 25900,
      "student_distance_mean": 1934.0309936523438,
      "teacher_distance_mean": 1.1474926948547364,
      "weighted_auxiliary_loss": 0.0003824210580205545,
      "weighted_relation_loss": 0.0003824210580205545
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.23760523854069,
      "flow_matching_loss": 0.01759612327441573,
      "grad_norm": 0.08018707484006882,
      "learning_rate": 4.998464104322015e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038905256951693445,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14196891784668,
      "state_query_norm": 1401.2658325195312,
      "state_relation_loss": 0.00038905256951693445,
      "step": 25910,
      "student_distance_mean": 1932.0466674804688,
      "teacher_distance_mean": 1.1463577628135682,
      "weighted_auxiliary_loss": 0.00038905256951693445,
      "weighted_relation_loss": 0.00038905256951693445
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.246959775491113,
      "flow_matching_loss": 0.015864344872534276,
      "grad_norm": 0.09338662773370743,
      "learning_rate": 4.974470586270047e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004254225641489029,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.360714721679685,
      "state_query_norm": 1414.604443359375,
      "state_relation_loss": 0.0004254225641489029,
      "step": 25920,
      "student_distance_mean": 1941.0564819335937,
      "teacher_distance_mean": 1.1520745158195496,
      "weighted_auxiliary_loss": 0.0004254225641489029,
      "weighted_relation_loss": 0.0004254225641489029
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.256314312441535,
      "flow_matching_loss": 0.015663188043981792,
      "grad_norm": 0.08378972113132477,
      "learning_rate": 4.950531778481965e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039746844558976593,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26338539123535,
      "state_query_norm": 1411.1416748046875,
      "state_relation_loss": 0.00039746844558976593,
      "step": 25930,
      "student_distance_mean": 1937.8844604492188,
      "teacher_distance_mean": 1.150072693824768,
      "weighted_auxiliary_loss": 0.00039746844558976593,
      "weighted_relation_loss": 0.00039746844558976593
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.265668849391954,
      "flow_matching_loss": 0.016480746585875748,
      "grad_norm": 0.09442514181137085,
      "learning_rate": 4.926647710045651e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003836462536128238,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.04748954772949,
      "state_query_norm": 1396.2242553710937,
      "state_relation_loss": 0.0003836462536128238,
      "step": 25940,
      "student_distance_mean": 1921.3242065429688,
      "teacher_distance_mean": 1.1393642902374268,
      "weighted_auxiliary_loss": 0.0003836462536128238,
      "weighted_relation_loss": 0.0003836462536128238
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.275023386342376,
      "flow_matching_loss": 0.015621839463710785,
      "grad_norm": 0.07374595105648041,
      "learning_rate": 4.9028184099825125e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039969749050214884,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.98339691162109,
      "state_query_norm": 1430.3403564453124,
      "state_relation_loss": 0.00039969749050214884,
      "step": 25950,
      "student_distance_mean": 1973.34345703125,
      "teacher_distance_mean": 1.1716329336166382,
      "weighted_auxiliary_loss": 0.00039969749050214884,
      "weighted_relation_loss": 0.00039969749050214884
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.284377923292798,
      "flow_matching_loss": 0.01670446926727891,
      "grad_norm": 0.09225142002105713,
      "learning_rate": 4.879043907247382e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038598451938014476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43076133728027,
      "state_query_norm": 1414.3035034179688,
      "state_relation_loss": 0.00038598451938014476,
      "step": 25960,
      "student_distance_mean": 1942.9613403320313,
      "teacher_distance_mean": 1.1535927176475524,
      "weighted_auxiliary_loss": 0.00038598451938014476,
      "weighted_relation_loss": 0.00038598451938014476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.293732460243216,
      "flow_matching_loss": 0.017174332216382027,
      "grad_norm": 0.08884043246507645,
      "learning_rate": 4.8553242307285414e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003946820506826043,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58538131713867,
      "state_query_norm": 1428.3184936523437,
      "state_relation_loss": 0.0003946820506826043,
      "step": 25970,
      "student_distance_mean": 1956.2785400390626,
      "teacher_distance_mean": 1.1609855890274048,
      "weighted_auxiliary_loss": 0.0003946820506826043,
      "weighted_relation_loss": 0.0003946820506826043
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.30308699719364,
      "flow_matching_loss": 0.015328901447355747,
      "grad_norm": 0.07376372814178467,
      "learning_rate": 4.8316594092476185e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042695664160419255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01103630065918,
      "state_query_norm": 1398.92578125,
      "state_relation_loss": 0.00042695664160419255,
      "step": 25980,
      "student_distance_mean": 1920.2815307617188,
      "teacher_distance_mean": 1.139728558063507,
      "weighted_auxiliary_loss": 0.00042695664160419255,
      "weighted_relation_loss": 0.00042695664160419255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.31244153414406,
      "flow_matching_loss": 0.016600738186389206,
      "grad_norm": 0.10274123400449753,
      "learning_rate": 4.808049471559617e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039740637876093385,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.63758773803711,
      "state_query_norm": 1430.9533325195312,
      "state_relation_loss": 0.00039740637876093385,
      "step": 25990,
      "student_distance_mean": 1957.8380981445312,
      "teacher_distance_mean": 1.16148521900177,
      "weighted_auxiliary_loss": 0.00039740637876093385,
      "weighted_relation_loss": 0.00039740637876093385
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.321796071094482,
      "flow_matching_loss": 0.016380940563976765,
      "grad_norm": 0.09165916591882706,
      "learning_rate": 4.784494446352833e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040243563998956235,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.118370819091794,
      "state_query_norm": 1404.3143188476563,
      "state_relation_loss": 0.00040243563998956235,
      "step": 26000,
      "student_distance_mean": 1926.4783203125,
      "teacher_distance_mean": 1.1424801826477051,
      "weighted_auxiliary_loss": 0.00040243563998956235,
      "weighted_relation_loss": 0.00040243563998956235
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.3311506080449,
      "flow_matching_loss": 0.016680021677166222,
      "grad_norm": 0.07533054798841476,
      "learning_rate": 4.760994362248833e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039195455610752106,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55830078125,
      "state_query_norm": 1424.1690795898437,
      "state_relation_loss": 0.00039195455610752106,
      "step": 26010,
      "student_distance_mean": 1955.3309326171875,
      "teacher_distance_mean": 1.1597214102745057,
      "weighted_auxiliary_loss": 0.00039195455610752106,
      "weighted_relation_loss": 0.00039195455610752106
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.340505144995323,
      "flow_matching_loss": 0.01503871800377965,
      "grad_norm": 0.08568283915519714,
      "learning_rate": 4.737549247802442e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039298230549320576,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.76896018981934,
      "state_query_norm": 1387.31279296875,
      "state_relation_loss": 0.00039298230549320576,
      "step": 26020,
      "student_distance_mean": 1908.7876342773438,
      "teacher_distance_mean": 1.1316985845565797,
      "weighted_auxiliary_loss": 0.00039298230549320576,
      "weighted_relation_loss": 0.00039298230549320576
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.349859681945745,
      "flow_matching_loss": 0.01592742558568716,
      "grad_norm": 0.07927264273166656,
      "learning_rate": 4.714159131501689e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004000728949904442,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.407308959960936,
      "state_query_norm": 1414.16689453125,
      "state_relation_loss": 0.0004000728949904442,
      "step": 26030,
      "student_distance_mean": 1945.6630004882813,
      "teacher_distance_mean": 1.1534832239151,
      "weighted_auxiliary_loss": 0.0004000728949904442,
      "weighted_relation_loss": 0.0004000728949904442
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.359214218896163,
      "flow_matching_loss": 0.015411748178303241,
      "grad_norm": 0.06111431494355202,
      "learning_rate": 4.690824041767766e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037649931909982114,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31975898742676,
      "state_query_norm": 1417.3703735351562,
      "state_relation_loss": 0.00037649931909982114,
      "step": 26040,
      "student_distance_mean": 1941.980126953125,
      "teacher_distance_mean": 1.1516932249069214,
      "weighted_auxiliary_loss": 0.00037649931909982114,
      "weighted_relation_loss": 0.00037649931909982114
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.368568755846585,
      "flow_matching_loss": 0.01548704719170928,
      "grad_norm": 0.0967806875705719,
      "learning_rate": 4.6675440069550105e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003939305752282962,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29345054626465,
      "state_query_norm": 1403.7518798828125,
      "state_relation_loss": 0.0003939305752282962,
      "step": 26050,
      "student_distance_mean": 1934.3040161132812,
      "teacher_distance_mean": 1.1465286374092103,
      "weighted_auxiliary_loss": 0.0003939305752282962,
      "weighted_relation_loss": 0.0003939305752282962
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.377923292797007,
      "flow_matching_loss": 0.01662032101303339,
      "grad_norm": 0.09620696306228638,
      "learning_rate": 4.64431905535086e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004175333247985691,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50106239318848,
      "state_query_norm": 1424.1564575195312,
      "state_relation_loss": 0.0004175333247985691,
      "step": 26060,
      "student_distance_mean": 1952.5208984375,
      "teacher_distance_mean": 1.159244704246521,
      "weighted_auxiliary_loss": 0.0004175333247985691,
      "weighted_relation_loss": 0.0004175333247985691
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.38727782974743,
      "flow_matching_loss": 0.01711334865540266,
      "grad_norm": 0.10869921743869781,
      "learning_rate": 4.621149215175808e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000379916385281831,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.264976501464844,
      "state_query_norm": 1415.7776977539063,
      "state_relation_loss": 0.000379916385281831,
      "step": 26070,
      "student_distance_mean": 1940.4396240234375,
      "teacher_distance_mean": 1.1523153185844421,
      "weighted_auxiliary_loss": 0.000379916385281831,
      "weighted_relation_loss": 0.000379916385281831
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.396632366697848,
      "flow_matching_loss": 0.0156641679815948,
      "grad_norm": 0.08092953264713287,
      "learning_rate": 4.5980345145834154e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004269503871910274,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.904083251953125,
      "state_query_norm": 1395.8552734375,
      "state_relation_loss": 0.0004269503871910274,
      "step": 26080,
      "student_distance_mean": 1918.4477905273438,
      "teacher_distance_mean": 1.138677453994751,
      "weighted_auxiliary_loss": 0.0004269503871910274,
      "weighted_relation_loss": 0.0004269503871910274
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.40598690364827,
      "flow_matching_loss": 0.016364255640655755,
      "grad_norm": 0.08738332986831665,
      "learning_rate": 4.574974981660213e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004040828731376678,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70311813354492,
      "state_query_norm": 1418.341552734375,
      "state_relation_loss": 0.0004040828731376678,
      "step": 26090,
      "student_distance_mean": 1954.8665161132812,
      "teacher_distance_mean": 1.1601051092147827,
      "weighted_auxiliary_loss": 0.0004040828731376678,
      "weighted_relation_loss": 0.0004040828731376678
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.41534144059869,
      "flow_matching_loss": 0.01567801395431161,
      "grad_norm": 0.08525655418634415,
      "learning_rate": 4.551970644425707e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039112100494094194,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32603530883789,
      "state_query_norm": 1414.6092041015625,
      "state_relation_loss": 0.00039112100494094194,
      "step": 26100,
      "student_distance_mean": 1944.1559692382812,
      "teacher_distance_mean": 1.153229308128357,
      "weighted_auxiliary_loss": 0.00039112100494094194,
      "weighted_relation_loss": 0.00039112100494094194
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.42469597754911,
      "flow_matching_loss": 0.01509673185646534,
      "grad_norm": 0.09281063824892044,
      "learning_rate": 4.529021530832328e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040701023826841267,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.478345489501955,
      "state_query_norm": 1419.8531616210937,
      "state_relation_loss": 0.00040701023826841267,
      "step": 26110,
      "student_distance_mean": 1951.0740600585937,
      "teacher_distance_mean": 1.1575518727302552,
      "weighted_auxiliary_loss": 0.00040701023826841267,
      "weighted_relation_loss": 0.00040701023826841267
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.434050514499532,
      "flow_matching_loss": 0.01654435321688652,
      "grad_norm": 0.08196030557155609,
      "learning_rate": 4.5061276687654306e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041962847171816975,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41695213317871,
      "state_query_norm": 1421.2898559570312,
      "state_relation_loss": 0.00041962847171816975,
      "step": 26120,
      "student_distance_mean": 1945.992724609375,
      "teacher_distance_mean": 1.1553158640861512,
      "weighted_auxiliary_loss": 0.00041962847171816975,
      "weighted_relation_loss": 0.00041962847171816975
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.443405051449954,
      "flow_matching_loss": 0.01562602948397398,
      "grad_norm": 0.0827576220035553,
      "learning_rate": 4.4832890860431965e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003658744128188118,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.848426055908206,
      "state_query_norm": 1399.8736938476563,
      "state_relation_loss": 0.0003658744128188118,
      "step": 26130,
      "student_distance_mean": 1916.6731567382812,
      "teacher_distance_mean": 1.1379218220710754,
      "weighted_auxiliary_loss": 0.0003658744128188118,
      "weighted_relation_loss": 0.0003658744128188118
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.452759588400376,
      "flow_matching_loss": 0.015587668679654599,
      "grad_norm": 0.07861552387475967,
      "learning_rate": 4.460505810416682e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037998734333086757,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.69197044372559,
      "state_query_norm": 1430.51875,
      "state_relation_loss": 0.00037998734333086757,
      "step": 26140,
      "student_distance_mean": 1962.323095703125,
      "teacher_distance_mean": 1.1649528980255126,
      "weighted_auxiliary_loss": 0.00037998734333086757,
      "weighted_relation_loss": 0.00037998734333086757
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.462114125350794,
      "flow_matching_loss": 0.015959879383444786,
      "grad_norm": 0.0808001384139061,
      "learning_rate": 4.4377778695696984e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003857301897369325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24667320251465,
      "state_query_norm": 1419.3008544921875,
      "state_relation_loss": 0.0003857301897369325,
      "step": 26150,
      "student_distance_mean": 1941.2402587890624,
      "teacher_distance_mean": 1.1521798968315125,
      "weighted_auxiliary_loss": 0.0003857301897369325,
      "weighted_relation_loss": 0.0003857301897369325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.471468662301216,
      "flow_matching_loss": 0.01638186303898692,
      "grad_norm": 0.08282389491796494,
      "learning_rate": 4.415105291118843e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038044572283979505,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.75160865783691,
      "state_query_norm": 1395.1558227539062,
      "state_relation_loss": 0.00038044572283979505,
      "step": 26160,
      "student_distance_mean": 1911.0511474609375,
      "teacher_distance_mean": 1.134342658519745,
      "weighted_auxiliary_loss": 0.00038044572283979505,
      "weighted_relation_loss": 0.00038044572283979505
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.48082319925164,
      "flow_matching_loss": 0.015155020263046026,
      "grad_norm": 0.0859065130352974,
      "learning_rate": 4.392488102613435e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004032129887491465,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.544057846069336,
      "state_query_norm": 1417.04951171875,
      "state_relation_loss": 0.0004032129887491465,
      "step": 26170,
      "student_distance_mean": 1949.7314453125,
      "teacher_distance_mean": 1.1565356254577637,
      "weighted_auxiliary_loss": 0.0004032129887491465,
      "weighted_relation_loss": 0.0004032129887491465
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.490177736202057,
      "flow_matching_loss": 0.014922135230153798,
      "grad_norm": 0.07557253539562225,
      "learning_rate": 4.369926331535473e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003645357064669952,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.83844528198242,
      "state_query_norm": 1399.8746704101563,
      "state_relation_loss": 0.0003645357064669952,
      "step": 26180,
      "student_distance_mean": 1917.9883544921875,
      "teacher_distance_mean": 1.1377480387687684,
      "weighted_auxiliary_loss": 0.0003645357064669952,
      "weighted_relation_loss": 0.0003645357064669952
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.49953227315248,
      "flow_matching_loss": 0.01594777572900057,
      "grad_norm": 0.08395044505596161,
      "learning_rate": 4.347420005299668e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038820869231130927,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.91579093933105,
      "state_query_norm": 1395.6142700195312,
      "state_relation_loss": 0.00038820869231130927,
      "step": 26190,
      "student_distance_mean": 1916.5455688476563,
      "teacher_distance_mean": 1.1377568244934082,
      "weighted_auxiliary_loss": 0.00038820869231130927,
      "weighted_relation_loss": 0.00038820869231130927
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.5088868101029,
      "flow_matching_loss": 0.01791274268180132,
      "grad_norm": 0.11818914860486984,
      "learning_rate": 4.324969151253317e-06,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003995371953351423,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62766342163086,
      "state_query_norm": 1428.1941650390625,
      "state_relation_loss": 0.0003995371953351423,
      "step": 26200,
      "student_distance_mean": 1960.5325439453125,
      "teacher_distance_mean": 1.1631227612495423,
      "weighted_auxiliary_loss": 0.0003995371953351423,
      "weighted_relation_loss": 0.0003995371953351423
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.51824134705332,
      "flow_matching_loss": 0.016727113537490367,
      "grad_norm": 0.08036185801029205,
      "learning_rate": 4.302573796676312e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003803026338573545,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.48689346313476,
      "state_query_norm": 1386.6929077148438,
      "state_relation_loss": 0.0003803026338573545,
      "step": 26210,
      "student_distance_mean": 1899.4966552734375,
      "teacher_distance_mean": 1.1278467655181885,
      "weighted_auxiliary_loss": 0.0003803026338573545,
      "weighted_relation_loss": 0.0003803026338573545
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.52759588400374,
      "flow_matching_loss": 0.015494168270379305,
      "grad_norm": 0.09321951866149902,
      "learning_rate": 4.2802339687811385e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038867036055307836,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.34011268615723,
      "state_query_norm": 1410.0085693359374,
      "state_relation_loss": 0.00038867036055307836,
      "step": 26220,
      "student_distance_mean": 1938.7918823242187,
      "teacher_distance_mean": 1.150589394569397,
      "weighted_auxiliary_loss": 0.00038867036055307836,
      "weighted_relation_loss": 0.00038867036055307836
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.536950420954163,
      "flow_matching_loss": 0.01582525484263897,
      "grad_norm": 0.08148842304944992,
      "learning_rate": 4.257949694712771e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004257310356479138,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23695106506348,
      "state_query_norm": 1409.26337890625,
      "state_relation_loss": 0.0004257310356479138,
      "step": 26230,
      "student_distance_mean": 1935.3318603515625,
      "teacher_distance_mean": 1.1490549087524413,
      "weighted_auxiliary_loss": 0.0004257310356479138,
      "weighted_relation_loss": 0.0004257310356479138
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.546304957904585,
      "flow_matching_loss": 0.017505751457065344,
      "grad_norm": 0.08269349485635757,
      "learning_rate": 4.235721001548726e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004272433841833845,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66550521850586,
      "state_query_norm": 1421.6295532226563,
      "state_relation_loss": 0.0004272433841833845,
      "step": 26240,
      "student_distance_mean": 1956.391259765625,
      "teacher_distance_mean": 1.160883092880249,
      "weighted_auxiliary_loss": 0.0004272433841833845,
      "weighted_relation_loss": 0.0004272433841833845
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.555659494855004,
      "flow_matching_loss": 0.017004660796374083,
      "grad_norm": 0.08613984286785126,
      "learning_rate": 4.21354791629896e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039568399370182306,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.400762939453124,
      "state_query_norm": 1412.98955078125,
      "state_relation_loss": 0.00039568399370182306,
      "step": 26250,
      "student_distance_mean": 1942.3225219726562,
      "teacher_distance_mean": 1.1519389748573303,
      "weighted_auxiliary_loss": 0.00039568399370182306,
      "weighted_relation_loss": 0.00039568399370182306
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.565014031805426,
      "flow_matching_loss": 0.01521692806854844,
      "grad_norm": 0.0926278755068779,
      "learning_rate": 4.191430465905843e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003658314672065899,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.11230659484863,
      "state_query_norm": 1404.4246704101563,
      "state_relation_loss": 0.0003658314672065899,
      "step": 26260,
      "student_distance_mean": 1929.4560180664062,
      "teacher_distance_mean": 1.1445734739303588,
      "weighted_auxiliary_loss": 0.0003658314672065899,
      "weighted_relation_loss": 0.0003658314672065899
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.574368568755848,
      "flow_matching_loss": 0.018418831657618283,
      "grad_norm": 0.08677640557289124,
      "learning_rate": 4.169368677244184e-06,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040421374142169953,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.142123794555665,
      "state_query_norm": 1405.2788452148438,
      "state_relation_loss": 0.00040421374142169953,
      "step": 26270,
      "student_distance_mean": 1928.1915283203125,
      "teacher_distance_mean": 1.1435239553451537,
      "weighted_auxiliary_loss": 0.00040421374142169953,
      "weighted_relation_loss": 0.00040421374142169953
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.583723105706266,
      "flow_matching_loss": 0.015015667211264372,
      "grad_norm": 0.0785132646560669,
      "learning_rate": 4.147362577121122e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003935842396458611,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.03238029479981,
      "state_query_norm": 1437.6150512695312,
      "state_relation_loss": 0.0003935842396458611,
      "step": 26280,
      "student_distance_mean": 1977.7929321289062,
      "teacher_distance_mean": 1.1726317644119262,
      "weighted_auxiliary_loss": 0.0003935842396458611,
      "weighted_relation_loss": 0.0003935842396458611
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.593077642656688,
      "flow_matching_loss": 0.015095536503940821,
      "grad_norm": 0.09013640135526657,
      "learning_rate": 4.125412192276157e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003692981001222506,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06730766296387,
      "state_query_norm": 1403.3127197265626,
      "state_relation_loss": 0.0003692981001222506,
      "step": 26290,
      "student_distance_mean": 1926.4369506835938,
      "teacher_distance_mean": 1.1413467049598693,
      "weighted_auxiliary_loss": 0.0003692981001222506,
      "weighted_relation_loss": 0.0003692981001222506
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.60243217960711,
      "flow_matching_loss": 0.01726706949993968,
      "grad_norm": 0.09393859654664993,
      "learning_rate": 4.10351754938107e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003740493499208242,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38621101379395,
      "state_query_norm": 1416.2295776367187,
      "state_relation_loss": 0.0003740493499208242,
      "step": 26300,
      "student_distance_mean": 1945.5549438476562,
      "teacher_distance_mean": 1.1541178226470947,
      "weighted_auxiliary_loss": 0.0003740493499208242,
      "weighted_relation_loss": 0.0003740493499208242
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.611786716557532,
      "flow_matching_loss": 0.014802584517747163,
      "grad_norm": 0.0724298283457756,
      "learning_rate": 4.081678675039913e-06,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003949638194171712,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14278144836426,
      "state_query_norm": 1413.2825317382812,
      "state_relation_loss": 0.0003949638194171712,
      "step": 26310,
      "student_distance_mean": 1932.2935913085937,
      "teacher_distance_mean": 1.1458856105804442,
      "weighted_auxiliary_loss": 0.0003949638194171712,
      "weighted_relation_loss": 0.0003949638194171712
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.62114125350795,
      "flow_matching_loss": 0.018427358102053405,
      "grad_norm": 0.0797535851597786,
      "learning_rate": 4.0598955957889876e-06,
      "loss": 0.0188,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003963862225646153,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25136222839355,
      "state_query_norm": 1411.452099609375,
      "state_relation_loss": 0.0003963862225646153,
      "step": 26320,
      "student_distance_mean": 1936.44248046875,
      "teacher_distance_mean": 1.1472413420677186,
      "weighted_auxiliary_loss": 0.0003963862225646153,
      "weighted_relation_loss": 0.0003963862225646153
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.630495790458372,
      "flow_matching_loss": 0.016415292210876942,
      "grad_norm": 0.10497897863388062,
      "learning_rate": 4.038168338096776e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038100163801573216,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.99788932800293,
      "state_query_norm": 1399.0304077148437,
      "state_relation_loss": 0.00038100163801573216,
      "step": 26330,
      "student_distance_mean": 1921.9057373046876,
      "teacher_distance_mean": 1.139323091506958,
      "weighted_auxiliary_loss": 0.00038100163801573216,
      "weighted_relation_loss": 0.00038100163801573216
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.639850327408794,
      "flow_matching_loss": 0.016452469862997532,
      "grad_norm": 0.08141963928937912,
      "learning_rate": 4.016496928363944e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003804952779319137,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.138987350463864,
      "state_query_norm": 1405.6076782226562,
      "state_relation_loss": 0.0003804952779319137,
      "step": 26340,
      "student_distance_mean": 1930.88759765625,
      "teacher_distance_mean": 1.1440714955329896,
      "weighted_auxiliary_loss": 0.0003804952779319137,
      "weighted_relation_loss": 0.0003804952779319137
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.649204864359213,
      "flow_matching_loss": 0.014901230670511722,
      "grad_norm": 0.07609842717647552,
      "learning_rate": 3.994881392923317e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003722916851984337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.652741241455075,
      "state_query_norm": 1427.6189575195312,
      "state_relation_loss": 0.0003722916851984337,
      "step": 26350,
      "student_distance_mean": 1959.0014892578124,
      "teacher_distance_mean": 1.1605004668235779,
      "weighted_auxiliary_loss": 0.0003722916851984337,
      "weighted_relation_loss": 0.0003722916851984337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.658559401309635,
      "flow_matching_loss": 0.017100471537560225,
      "grad_norm": 0.08113857358694077,
      "learning_rate": 3.9733217580397945e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003525567066390067,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92230033874512,
      "state_query_norm": 1400.4357177734375,
      "state_relation_loss": 0.0003525567066390067,
      "step": 26360,
      "student_distance_mean": 1922.3859375,
      "teacher_distance_mean": 1.1385627508163452,
      "weighted_auxiliary_loss": 0.0003525567066390067,
      "weighted_relation_loss": 0.0003525567066390067
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.667913938260057,
      "flow_matching_loss": 0.016793487686663867,
      "grad_norm": 0.07755095511674881,
      "learning_rate": 3.951818049910366e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038389652909245344,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.09413795471191,
      "state_query_norm": 1413.1108154296876,
      "state_relation_loss": 0.00038389652909245344,
      "step": 26370,
      "student_distance_mean": 1930.2153076171876,
      "teacher_distance_mean": 1.144026482105255,
      "weighted_auxiliary_loss": 0.00038389652909245344,
      "weighted_relation_loss": 0.00038389652909245344
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.67726847521048,
      "flow_matching_loss": 0.015220637992024421,
      "grad_norm": 0.09281942993402481,
      "learning_rate": 3.930370294664071e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039788965368643404,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44926300048828,
      "state_query_norm": 1416.3559936523438,
      "state_relation_loss": 0.00039788965368643404,
      "step": 26380,
      "student_distance_mean": 1946.6949340820313,
      "teacher_distance_mean": 1.1540682554244994,
      "weighted_auxiliary_loss": 0.00039788965368643404,
      "weighted_relation_loss": 0.00039788965368643404
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.686623012160897,
      "flow_matching_loss": 0.015265625715255738,
      "grad_norm": 0.07224752753973007,
      "learning_rate": 3.908978518361938e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037896537396591156,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01174736022949,
      "state_query_norm": 1401.5998046875,
      "state_relation_loss": 0.00037896537396591156,
      "step": 26390,
      "student_distance_mean": 1920.7409301757812,
      "teacher_distance_mean": 1.1389710545539855,
      "weighted_auxiliary_loss": 0.00037896537396591156,
      "weighted_relation_loss": 0.00037896537396591156
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.69597754911132,
      "flow_matching_loss": 0.016240268293768168,
      "grad_norm": 0.09430975466966629,
      "learning_rate": 3.887642746997017e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038819735636934637,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3494514465332,
      "state_query_norm": 1403.4250610351562,
      "state_relation_loss": 0.00038819735636934637,
      "step": 26400,
      "student_distance_mean": 1939.1215209960938,
      "teacher_distance_mean": 1.1507857322692872,
      "weighted_auxiliary_loss": 0.00038819735636934637,
      "weighted_relation_loss": 0.00038819735636934637
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.70533208606174,
      "flow_matching_loss": 0.016143092047423126,
      "grad_norm": 0.09864863753318787,
      "learning_rate": 3.866363006494255e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004109525732928887,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.765421295166014,
      "state_query_norm": 1429.6308471679688,
      "state_relation_loss": 0.0004109525732928887,
      "step": 26410,
      "student_distance_mean": 1966.4372436523438,
      "teacher_distance_mean": 1.1654996871948242,
      "weighted_auxiliary_loss": 0.0004109525732928887,
      "weighted_relation_loss": 0.0004109525732928887
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.71468662301216,
      "flow_matching_loss": 0.0171169969253242,
      "grad_norm": 0.08807437866926193,
      "learning_rate": 3.8451393227105725e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038854962040204557,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.51375389099121,
      "state_query_norm": 1409.394189453125,
      "state_relation_loss": 0.00038854962040204557,
      "step": 26420,
      "student_distance_mean": 1947.0824829101562,
      "teacher_distance_mean": 1.1541966915130615,
      "weighted_auxiliary_loss": 0.00038854962040204557,
      "weighted_relation_loss": 0.00038854962040204557
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.72404115996258,
      "flow_matching_loss": 0.015464093163609504,
      "grad_norm": 0.1111288070678711,
      "learning_rate": 3.823971721434727e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037873446417506783,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92372627258301,
      "state_query_norm": 1397.8859008789063,
      "state_relation_loss": 0.00037873446417506783,
      "step": 26430,
      "student_distance_mean": 1919.46064453125,
      "teacher_distance_mean": 1.1384100198745728,
      "weighted_auxiliary_loss": 0.00037873446417506783,
      "weighted_relation_loss": 0.00037873446417506783
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.733395696913004,
      "flow_matching_loss": 0.015398459695279598,
      "grad_norm": 0.11554281413555145,
      "learning_rate": 3.8028602283873505e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039910584455356003,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01660575866699,
      "state_query_norm": 1402.3419189453125,
      "state_relation_loss": 0.00039910584455356003,
      "step": 26440,
      "student_distance_mean": 1923.8043090820313,
      "teacher_distance_mean": 1.140618336200714,
      "weighted_auxiliary_loss": 0.00039910584455356003,
      "weighted_relation_loss": 0.00039910584455356003
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.742750233863426,
      "flow_matching_loss": 0.016948316991329194,
      "grad_norm": 0.08540326356887817,
      "learning_rate": 3.781804869220912e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003912474639946595,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9015438079834,
      "state_query_norm": 1433.3705688476562,
      "state_relation_loss": 0.0003912474639946595,
      "step": 26450,
      "student_distance_mean": 1972.3223022460938,
      "teacher_distance_mean": 1.169921100139618,
      "weighted_auxiliary_loss": 0.0003912474639946595,
      "weighted_relation_loss": 0.0003912474639946595
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.752104770813844,
      "flow_matching_loss": 0.0157058659940958,
      "grad_norm": 0.0951894223690033,
      "learning_rate": 3.760805669519646e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004110981040867046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.824136352539064,
      "state_query_norm": 1435.791162109375,
      "state_relation_loss": 0.0004110981040867046,
      "step": 26460,
      "student_distance_mean": 1969.6202270507813,
      "teacher_distance_mean": 1.1686087489128112,
      "weighted_auxiliary_loss": 0.0004110981040867046,
      "weighted_relation_loss": 0.0004110981040867046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.761459307764266,
      "flow_matching_loss": 0.016668836865574122,
      "grad_norm": 0.08705129474401474,
      "learning_rate": 3.7398626547995587e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004027994989883155,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50725059509277,
      "state_query_norm": 1416.3383911132812,
      "state_relation_loss": 0.0004027994989883155,
      "step": 26470,
      "student_distance_mean": 1949.3879638671874,
      "teacher_distance_mean": 1.1552960634231568,
      "weighted_auxiliary_loss": 0.0004027994989883155,
      "weighted_relation_loss": 0.0004027994989883155
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.770813844714688,
      "flow_matching_loss": 0.015122141223400831,
      "grad_norm": 0.0798075944185257,
      "learning_rate": 3.7189758505083995e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037167464615777134,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16545791625977,
      "state_query_norm": 1404.5005859375,
      "state_relation_loss": 0.00037167464615777134,
      "step": 26480,
      "student_distance_mean": 1928.4498779296875,
      "teacher_distance_mean": 1.1435728311538695,
      "weighted_auxiliary_loss": 0.00037167464615777134,
      "weighted_relation_loss": 0.00037167464615777134
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.780168381665106,
      "flow_matching_loss": 0.01693543903529644,
      "grad_norm": 0.09234406799077988,
      "learning_rate": 3.6981452820255837e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000381518792710267,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.80798988342285,
      "state_query_norm": 1391.800927734375,
      "state_relation_loss": 0.000381518792710267,
      "step": 26490,
      "student_distance_mean": 1909.867822265625,
      "teacher_distance_mean": 1.1324957251548766,
      "weighted_auxiliary_loss": 0.000381518792710267,
      "weighted_relation_loss": 0.000381518792710267
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.78952291861553,
      "flow_matching_loss": 0.01640727259218693,
      "grad_norm": 0.08312578499317169,
      "learning_rate": 3.6773709746622363e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039754679310135546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28947792053223,
      "state_query_norm": 1410.256591796875,
      "state_relation_loss": 0.00039754679310135546,
      "step": 26500,
      "student_distance_mean": 1937.3390502929688,
      "teacher_distance_mean": 1.1487615346908568,
      "weighted_auxiliary_loss": 0.00039754679310135546,
      "weighted_relation_loss": 0.00039754679310135546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.79887745556595,
      "flow_matching_loss": 0.01487233592197299,
      "grad_norm": 0.09357712417840958,
      "learning_rate": 3.6566529536610995e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003840927412966266,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64744987487793,
      "state_query_norm": 1435.4390869140625,
      "state_relation_loss": 0.0003840927412966266,
      "step": 26510,
      "student_distance_mean": 1961.8481811523438,
      "teacher_distance_mean": 1.1630790829658508,
      "weighted_auxiliary_loss": 0.0003840927412966266,
      "weighted_relation_loss": 0.0003840927412966266
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.80823199251637,
      "flow_matching_loss": 0.014603008329868317,
      "grad_norm": 0.09206319600343704,
      "learning_rate": 3.635991244196513e-06,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003831209643976763,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60642929077149,
      "state_query_norm": 1426.465234375,
      "state_relation_loss": 0.0003831209643976763,
      "step": 26520,
      "student_distance_mean": 1957.5940063476562,
      "teacher_distance_mean": 1.1610547304153442,
      "weighted_auxiliary_loss": 0.0003831209643976763,
      "weighted_relation_loss": 0.0003831209643976763
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.81758652946679,
      "flow_matching_loss": 0.018879578914493322,
      "grad_norm": 0.09485337883234024,
      "learning_rate": 3.615385871374405e-06,
      "loss": 0.0193,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004000840621301904,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.063970947265624,
      "state_query_norm": 1439.0952026367188,
      "state_relation_loss": 0.0004000840621301904,
      "step": 26530,
      "student_distance_mean": 1979.97041015625,
      "teacher_distance_mean": 1.1751431345939636,
      "weighted_auxiliary_loss": 0.0004000840621301904,
      "weighted_relation_loss": 0.0004000840621301904
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.826941066417213,
      "flow_matching_loss": 0.016886132024228572,
      "grad_norm": 0.08137939125299454,
      "learning_rate": 3.5948368602322367e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004120088240597397,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73858375549317,
      "state_query_norm": 1432.9245239257812,
      "state_relation_loss": 0.0004120088240597397,
      "step": 26540,
      "student_distance_mean": 1965.1311157226562,
      "teacher_distance_mean": 1.1662106037139892,
      "weighted_auxiliary_loss": 0.0004120088240597397,
      "weighted_relation_loss": 0.0004120088240597397
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.836295603367635,
      "flow_matching_loss": 0.017123734671622516,
      "grad_norm": 0.07084795087575912,
      "learning_rate": 3.5743442357390187e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003627264202805236,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23035011291504,
      "state_query_norm": 1409.5166015625,
      "state_relation_loss": 0.0003627264202805236,
      "step": 26550,
      "student_distance_mean": 1937.9310668945313,
      "teacher_distance_mean": 1.1500937581062316,
      "weighted_auxiliary_loss": 0.0003627264202805236,
      "weighted_relation_loss": 0.0003627264202805236
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.845650140318053,
      "flow_matching_loss": 0.015911956317722797,
      "grad_norm": 0.08589042723178864,
      "learning_rate": 3.5539080227951937e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038176134694367647,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39545860290527,
      "state_query_norm": 1418.6299438476562,
      "state_relation_loss": 0.00038176134694367647,
      "step": 26560,
      "student_distance_mean": 1946.0973876953126,
      "teacher_distance_mean": 1.154049801826477,
      "weighted_auxiliary_loss": 0.00038176134694367647,
      "weighted_relation_loss": 0.00038176134694367647
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.855004677268475,
      "flow_matching_loss": 0.01591906966641545,
      "grad_norm": 0.07579053938388824,
      "learning_rate": 3.533528246232709e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038021734508220106,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75458869934082,
      "state_query_norm": 1420.8771606445312,
      "state_relation_loss": 0.00038021734508220106,
      "step": 26570,
      "student_distance_mean": 1962.0897338867187,
      "teacher_distance_mean": 1.163257646560669,
      "weighted_auxiliary_loss": 0.00038021734508220106,
      "weighted_relation_loss": 0.00038021734508220106
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.864359214218897,
      "flow_matching_loss": 0.014555768575519323,
      "grad_norm": 0.07761196792125702,
      "learning_rate": 3.5132049308149117e-06,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037480910832528026,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.67838592529297,
      "state_query_norm": 1434.0699584960937,
      "state_relation_loss": 0.00037480910832528026,
      "step": 26580,
      "student_distance_mean": 1966.6808715820312,
      "teacher_distance_mean": 1.1664002418518067,
      "weighted_auxiliary_loss": 0.00037480910832528026,
      "weighted_relation_loss": 0.00037480910832528026
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.873713751169316,
      "flow_matching_loss": 0.016163683030754328,
      "grad_norm": 0.08581140637397766,
      "learning_rate": 3.4929381012365425e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042208137456327676,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.90463981628418,
      "state_query_norm": 1432.1983520507813,
      "state_relation_loss": 0.00042208137456327676,
      "step": 26590,
      "student_distance_mean": 1971.8112426757812,
      "teacher_distance_mean": 1.1702141761779785,
      "weighted_auxiliary_loss": 0.00042208137456327676,
      "weighted_relation_loss": 0.00042208137456327676
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.883068288119738,
      "flow_matching_loss": 0.01593460738658905,
      "grad_norm": 0.08060504496097565,
      "learning_rate": 3.472727782123697e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003937382803997025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52346687316894,
      "state_query_norm": 1428.8495971679688,
      "state_relation_loss": 0.0003937382803997025,
      "step": 26600,
      "student_distance_mean": 1954.363818359375,
      "teacher_distance_mean": 1.1597514033317566,
      "weighted_auxiliary_loss": 0.0003937382803997025,
      "weighted_relation_loss": 0.0003937382803997025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.89242282507016,
      "flow_matching_loss": 0.014561046008020639,
      "grad_norm": 0.08987897634506226,
      "learning_rate": 3.4525739980338425e-06,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004524873016634956,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.496036529541016,
      "state_query_norm": 1414.1458251953125,
      "state_relation_loss": 0.0004524873016634956,
      "step": 26610,
      "student_distance_mean": 1946.3690551757813,
      "teacher_distance_mean": 1.1554986357688903,
      "weighted_auxiliary_loss": 0.0004524873016634956,
      "weighted_relation_loss": 0.0004524873016634956
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.90177736202058,
      "flow_matching_loss": 0.0159255126491189,
      "grad_norm": 0.10404610633850098,
      "learning_rate": 3.432476773455712e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003788624337175861,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.180900192260744,
      "state_query_norm": 1410.5884765625,
      "state_relation_loss": 0.0003788624337175861,
      "step": 26620,
      "student_distance_mean": 1931.162353515625,
      "teacher_distance_mean": 1.1454793691635132,
      "weighted_auxiliary_loss": 0.0003788624337175861,
      "weighted_relation_loss": 0.0003788624337175861
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.911131898971,
      "flow_matching_loss": 0.01554391896352172,
      "grad_norm": 0.08493733406066895,
      "learning_rate": 3.4124361328093378e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036686535750050096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.637339973449706,
      "state_query_norm": 1375.469140625,
      "state_relation_loss": 0.00036686535750050096,
      "step": 26630,
      "student_distance_mean": 1900.2423828125,
      "teacher_distance_mean": 1.1276514291763307,
      "weighted_auxiliary_loss": 0.00036686535750050096,
      "weighted_relation_loss": 0.00036686535750050096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.920486435921422,
      "flow_matching_loss": 0.017225743643939494,
      "grad_norm": 0.09768505394458771,
      "learning_rate": 3.392452100445975e-06,
      "loss": 0.0176,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003605225181672722,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.23606224060059,
      "state_query_norm": 1408.9146118164062,
      "state_relation_loss": 0.0003605225181672722,
      "step": 26640,
      "student_distance_mean": 1936.5542846679687,
      "teacher_distance_mean": 1.1482654452323913,
      "weighted_auxiliary_loss": 0.0003605225181672722,
      "weighted_relation_loss": 0.0003605225181672722
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.929840972871844,
      "flow_matching_loss": 0.017766299098730086,
      "grad_norm": 0.08397463709115982,
      "learning_rate": 3.3725247006481163e-06,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003853027068544179,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53238525390625,
      "state_query_norm": 1418.0944946289062,
      "state_relation_loss": 0.0003853027068544179,
      "step": 26650,
      "student_distance_mean": 1949.2447509765625,
      "teacher_distance_mean": 1.1563019037246705,
      "weighted_auxiliary_loss": 0.0003853027068544179,
      "weighted_relation_loss": 0.0003853027068544179
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.939195509822262,
      "flow_matching_loss": 0.016430338285863398,
      "grad_norm": 0.09104135632514954,
      "learning_rate": 3.3526539576294313e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003865757520543411,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29645767211914,
      "state_query_norm": 1414.8870849609375,
      "state_relation_loss": 0.0003865757520543411,
      "step": 26660,
      "student_distance_mean": 1942.09560546875,
      "teacher_distance_mean": 1.1527827262878418,
      "weighted_auxiliary_loss": 0.0003865757520543411,
      "weighted_relation_loss": 0.0003865757520543411
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.948550046772684,
      "flow_matching_loss": 0.014347053226083517,
      "grad_norm": 0.08139360696077347,
      "learning_rate": 3.332839895534745e-06,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003825636493274942,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.940382385253905,
      "state_query_norm": 1434.836181640625,
      "state_relation_loss": 0.0003825636493274942,
      "step": 26670,
      "student_distance_mean": 1972.9293090820313,
      "teacher_distance_mean": 1.1726574420928955,
      "weighted_auxiliary_loss": 0.0003825636493274942,
      "weighted_relation_loss": 0.0003825636493274942
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.957904583723106,
      "flow_matching_loss": 0.015325976721942424,
      "grad_norm": 0.09100698679685593,
      "learning_rate": 3.3130825384400155e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043228370777796953,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.249473571777344,
      "state_query_norm": 1442.7529907226562,
      "state_relation_loss": 0.00043228370777796953,
      "step": 26680,
      "student_distance_mean": 1986.4486328125,
      "teacher_distance_mean": 1.1790571689605713,
      "weighted_auxiliary_loss": 0.00043228370777796953,
      "weighted_relation_loss": 0.00043228370777796953
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.96725912067353,
      "flow_matching_loss": 0.0168120632879436,
      "grad_norm": 0.0957099050283432,
      "learning_rate": 3.293381910352278e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041466387920081617,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5668830871582,
      "state_query_norm": 1416.1008178710938,
      "state_relation_loss": 0.00041466387920081617,
      "step": 26690,
      "student_distance_mean": 1948.4570434570312,
      "teacher_distance_mean": 1.1569236397743226,
      "weighted_auxiliary_loss": 0.00041466387920081617,
      "weighted_relation_loss": 0.00041466387920081617
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.976613657623947,
      "flow_matching_loss": 0.017349015548825265,
      "grad_norm": 0.07828376442193985,
      "learning_rate": 3.273738035209678e-06,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038872642908245326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.09958000183106,
      "state_query_norm": 1407.989013671875,
      "state_relation_loss": 0.00038872642908245326,
      "step": 26700,
      "student_distance_mean": 1930.5336791992188,
      "teacher_distance_mean": 1.1449679136276245,
      "weighted_auxiliary_loss": 0.00038872642908245326,
      "weighted_relation_loss": 0.00038872642908245326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.98596819457437,
      "flow_matching_loss": 0.01633610175922513,
      "grad_norm": 0.08595903217792511,
      "learning_rate": 3.254150936881356e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043081066105514764,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.859370422363284,
      "state_query_norm": 1430.248046875,
      "state_relation_loss": 0.00043081066105514764,
      "step": 26710,
      "student_distance_mean": 1965.3938842773437,
      "teacher_distance_mean": 1.1669782638549804,
      "weighted_auxiliary_loss": 0.00043081066105514764,
      "weighted_relation_loss": 0.00043081066105514764
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 24.99532273152479,
      "flow_matching_loss": 0.014723778422921895,
      "grad_norm": 0.06896225363016129,
      "learning_rate": 3.234620639167496e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037372542137745766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53256034851074,
      "state_query_norm": 1423.047412109375,
      "state_relation_loss": 0.00037372542137745766,
      "step": 26720,
      "student_distance_mean": 1952.1450073242188,
      "teacher_distance_mean": 1.1581595301628114,
      "weighted_auxiliary_loss": 0.00037372542137745766,
      "weighted_relation_loss": 0.00037372542137745766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.00467726847521,
      "flow_matching_loss": 0.015287767350673675,
      "grad_norm": 0.12155826389789581,
      "learning_rate": 3.2151471657992482e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00038474497268907725,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 34.86899375915527,
      "state_query_norm": 1402.7749877929687,
      "state_relation_loss": 0.00038474497268907725,
      "step": 26730,
      "student_distance_mean": 1921.4553833007812,
      "teacher_distance_mean": 1.139619815349579,
      "weighted_auxiliary_loss": 0.00038474497268907725,
      "weighted_relation_loss": 0.00038474497268907725
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.01403180542563,
      "flow_matching_loss": 0.017499775066971777,
      "grad_norm": 0.08043630421161652,
      "learning_rate": 3.195730540438718e-06,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036865200963802635,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.54123382568359,
      "state_query_norm": 1425.1984008789063,
      "state_relation_loss": 0.00036865200963802635,
      "step": 26740,
      "student_distance_mean": 1955.7728393554687,
      "teacher_distance_mean": 1.1598641395568847,
      "weighted_auxiliary_loss": 0.00036865200963802635,
      "weighted_relation_loss": 0.00036865200963802635
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.023386342376053,
      "flow_matching_loss": 0.01796709969639778,
      "grad_norm": 0.08830131590366364,
      "learning_rate": 3.1763707866789336e-06,
      "loss": 0.0184,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004100223799468949,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.83763198852539,
      "state_query_norm": 1439.6456787109375,
      "state_relation_loss": 0.0004100223799468949,
      "step": 26750,
      "student_distance_mean": 1971.586572265625,
      "teacher_distance_mean": 1.1700970768928527,
      "weighted_auxiliary_loss": 0.0004100223799468949,
      "weighted_relation_loss": 0.0004100223799468949
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.03274087932647,
      "flow_matching_loss": 0.015255807433277369,
      "grad_norm": 0.09819280356168747,
      "learning_rate": 3.1570679280438165e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036806612624786796,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.89336166381836,
      "state_query_norm": 1431.9853759765624,
      "state_relation_loss": 0.00036806612624786796,
      "step": 26760,
      "student_distance_mean": 1970.0161010742188,
      "teacher_distance_mean": 1.16929692029953,
      "weighted_auxiliary_loss": 0.00036806612624786796,
      "weighted_relation_loss": 0.00036806612624786796
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.042095416276894,
      "flow_matching_loss": 0.015962069667875767,
      "grad_norm": 0.12107270210981369,
      "learning_rate": 3.1378219879881767e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004021577362436801,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.870215606689456,
      "state_query_norm": 1434.4642456054687,
      "state_relation_loss": 0.0004021577362436801,
      "step": 26770,
      "student_distance_mean": 1970.53837890625,
      "teacher_distance_mean": 1.1693170428276063,
      "weighted_auxiliary_loss": 0.0004021577362436801,
      "weighted_relation_loss": 0.0004021577362436801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.051449953227316,
      "flow_matching_loss": 0.01738790487870574,
      "grad_norm": 0.06790931522846222,
      "learning_rate": 3.11863298989763e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003829143737675622,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.95970420837402,
      "state_query_norm": 1400.545947265625,
      "state_relation_loss": 0.0003829143737675622,
      "step": 26780,
      "student_distance_mean": 1922.7516357421875,
      "teacher_distance_mean": 1.1403085470199585,
      "weighted_auxiliary_loss": 0.0003829143737675622,
      "weighted_relation_loss": 0.0003829143737675622
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.060804490177738,
      "flow_matching_loss": 0.0152598038315773,
      "grad_norm": 0.0739198550581932,
      "learning_rate": 3.0995009570886303e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003965893760323524,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.93810806274414,
      "state_query_norm": 1442.9120849609376,
      "state_relation_loss": 0.0003965893760323524,
      "step": 26790,
      "student_distance_mean": 1976.6689697265624,
      "teacher_distance_mean": 1.1730869889259339,
      "weighted_auxiliary_loss": 0.0003965893760323524,
      "weighted_relation_loss": 0.0003965893760323524
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.070159027128156,
      "flow_matching_loss": 0.014469728711992503,
      "grad_norm": 0.09699514508247375,
      "learning_rate": 3.0804259128083936e-06,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039368171710520985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66724967956543,
      "state_query_norm": 1425.694970703125,
      "state_relation_loss": 0.00039368171710520985,
      "step": 26800,
      "student_distance_mean": 1957.2864990234375,
      "teacher_distance_mean": 1.1613817453384399,
      "weighted_auxiliary_loss": 0.00039368171710520985,
      "weighted_relation_loss": 0.00039368171710520985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.079513564078578,
      "flow_matching_loss": 0.01614797245711088,
      "grad_norm": 0.08270344138145447,
      "learning_rate": 3.06140788023489e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039785551780369133,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.554391098022464,
      "state_query_norm": 1413.1540283203126,
      "state_relation_loss": 0.00039785551780369133,
      "step": 26810,
      "student_distance_mean": 1948.6388671875,
      "teacher_distance_mean": 1.1558481931686402,
      "weighted_auxiliary_loss": 0.00039785551780369133,
      "weighted_relation_loss": 0.00039785551780369133
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.088868101029,
      "flow_matching_loss": 0.015713113266974686,
      "grad_norm": 0.07207326591014862,
      "learning_rate": 3.042446882476846e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003832088754279539,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.580418014526366,
      "state_query_norm": 1422.06533203125,
      "state_relation_loss": 0.0003832088754279539,
      "step": 26820,
      "student_distance_mean": 1954.8874633789062,
      "teacher_distance_mean": 1.159861445426941,
      "weighted_auxiliary_loss": 0.0003832088754279539,
      "weighted_relation_loss": 0.0003832088754279539
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.09822263797942,
      "flow_matching_loss": 0.016192383877933026,
      "grad_norm": 0.10897057503461838,
      "learning_rate": 3.023542942573643e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003678465349366888,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.72145881652832,
      "state_query_norm": 1397.5363037109375,
      "state_relation_loss": 0.0003678465349366888,
      "step": 26830,
      "student_distance_mean": 1908.2736694335938,
      "teacher_distance_mean": 1.1325596690177917,
      "weighted_auxiliary_loss": 0.0003678465349366888,
      "weighted_relation_loss": 0.0003678465349366888
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.10757717492984,
      "flow_matching_loss": 0.014581364765763282,
      "grad_norm": 0.08505580574274063,
      "learning_rate": 3.004696083495351e-06,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039305183163378387,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71889762878418,
      "state_query_norm": 1430.1672241210938,
      "state_relation_loss": 0.00039305183163378387,
      "step": 26840,
      "student_distance_mean": 1963.5154907226563,
      "teacher_distance_mean": 1.1658529043197632,
      "weighted_auxiliary_loss": 0.00039305183163378387,
      "weighted_relation_loss": 0.00039305183163378387
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.116931711880262,
      "flow_matching_loss": 0.01703143008053303,
      "grad_norm": 0.0849170833826065,
      "learning_rate": 2.98590632814269e-06,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003725818358361721,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.139634704589845,
      "state_query_norm": 1409.2682495117188,
      "state_relation_loss": 0.0003725818358361721,
      "step": 26850,
      "student_distance_mean": 1933.5377075195313,
      "teacher_distance_mean": 1.1479463577270508,
      "weighted_auxiliary_loss": 0.0003725818358361721,
      "weighted_relation_loss": 0.0003725818358361721
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.126286248830684,
      "flow_matching_loss": 0.017380044888705017,
      "grad_norm": 0.07020648568868637,
      "learning_rate": 2.9671736993469702e-06,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003963399562053382,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.120351028442386,
      "state_query_norm": 1411.4010498046875,
      "state_relation_loss": 0.0003963399562053382,
      "step": 26860,
      "student_distance_mean": 1932.7599609375,
      "teacher_distance_mean": 1.1478965401649475,
      "weighted_auxiliary_loss": 0.0003963399562053382,
      "weighted_relation_loss": 0.0003963399562053382
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.135640785781103,
      "flow_matching_loss": 0.014931696467101574,
      "grad_norm": 0.13280999660491943,
      "learning_rate": 2.9484982198701216e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003862000710796565,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3190372467041,
      "state_query_norm": 1414.7389404296875,
      "state_relation_loss": 0.0003862000710796565,
      "step": 26870,
      "student_distance_mean": 1940.8780883789063,
      "teacher_distance_mean": 1.1528837561607361,
      "weighted_auxiliary_loss": 0.0003862000710796565,
      "weighted_relation_loss": 0.0003862000710796565
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.144995322731525,
      "flow_matching_loss": 0.015548846498131753,
      "grad_norm": 0.09169851988554001,
      "learning_rate": 2.929879912404604e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003767615125980228,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.51965141296387,
      "state_query_norm": 1416.027001953125,
      "state_relation_loss": 0.0003767615125980228,
      "step": 26880,
      "student_distance_mean": 1947.7892211914063,
      "teacher_distance_mean": 1.156218409538269,
      "weighted_auxiliary_loss": 0.0003767615125980228,
      "weighted_relation_loss": 0.0003767615125980228
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.154349859681947,
      "flow_matching_loss": 0.01676706876605749,
      "grad_norm": 0.12403638660907745,
      "learning_rate": 2.9113187995734147e-06,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041293736139778046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.492291641235354,
      "state_query_norm": 1414.7840942382813,
      "state_relation_loss": 0.00041293736139778046,
      "step": 26890,
      "student_distance_mean": 1947.612451171875,
      "teacher_distance_mean": 1.155967891216278,
      "weighted_auxiliary_loss": 0.00041293736139778046,
      "weighted_relation_loss": 0.00041293736139778046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.163704396632365,
      "flow_matching_loss": 0.014638591092079878,
      "grad_norm": 0.0726485624909401,
      "learning_rate": 2.8928149039300524e-06,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039677027089055625,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.14008903503418,
      "state_query_norm": 1396.4681884765625,
      "state_relation_loss": 0.00039677027089055625,
      "step": 26900,
      "student_distance_mean": 1925.7234008789062,
      "teacher_distance_mean": 1.1422810554504395,
      "weighted_auxiliary_loss": 0.00039677027089055625,
      "weighted_relation_loss": 0.00039677027089055625
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.173058933582787,
      "flow_matching_loss": 0.015673160925507544,
      "grad_norm": 0.09901804476976395,
      "learning_rate": 2.8743682479584975e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003997119871200994,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92543678283691,
      "state_query_norm": 1394.1858154296874,
      "state_relation_loss": 0.0003997119871200994,
      "step": 26910,
      "student_distance_mean": 1917.3334106445313,
      "teacher_distance_mean": 1.1367390990257262,
      "weighted_auxiliary_loss": 0.0003997119871200994,
      "weighted_relation_loss": 0.0003997119871200994
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.18241347053321,
      "flow_matching_loss": 0.016415551956743,
      "grad_norm": 0.06436506658792496,
      "learning_rate": 2.8559788540731824e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041603983263485136,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48942565917969,
      "state_query_norm": 1416.3008056640624,
      "state_relation_loss": 0.00041603983263485136,
      "step": 26920,
      "student_distance_mean": 1948.122265625,
      "teacher_distance_mean": 1.1559099912643434,
      "weighted_auxiliary_loss": 0.00041603983263485136,
      "weighted_relation_loss": 0.00041603983263485136
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.19176800748363,
      "flow_matching_loss": 0.01613992638885975,
      "grad_norm": 0.09645552933216095,
      "learning_rate": 2.8376467446189493e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003921406430890784,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.378958511352536,
      "state_query_norm": 1415.1183837890626,
      "state_relation_loss": 0.0003921406430890784,
      "step": 26930,
      "student_distance_mean": 1941.9826782226562,
      "teacher_distance_mean": 1.1519469380378724,
      "weighted_auxiliary_loss": 0.0003921406430890784,
      "weighted_relation_loss": 0.0003921406430890784
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.20112254443405,
      "flow_matching_loss": 0.015224253945052623,
      "grad_norm": 0.08999743312597275,
      "learning_rate": 2.8193719418710406e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041058861825149506,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10641441345215,
      "state_query_norm": 1398.084716796875,
      "state_relation_loss": 0.00041058861825149506,
      "step": 26940,
      "student_distance_mean": 1925.8865600585937,
      "teacher_distance_mean": 1.143174982070923,
      "weighted_auxiliary_loss": 0.00041058861825149506,
      "weighted_relation_loss": 0.00041058861825149506
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.21047708138447,
      "flow_matching_loss": 0.015952401608228684,
      "grad_norm": 0.07878399640321732,
      "learning_rate": 2.8011544680350667e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041428514232393353,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.90821151733398,
      "state_query_norm": 1393.805908203125,
      "state_relation_loss": 0.00041428514232393353,
      "step": 26950,
      "student_distance_mean": 1914.9776611328125,
      "teacher_distance_mean": 1.1374326586723327,
      "weighted_auxiliary_loss": 0.00041428514232393353,
      "weighted_relation_loss": 0.00041428514232393353
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.219831618334894,
      "flow_matching_loss": 0.016443101782351734,
      "grad_norm": 0.07076680660247803,
      "learning_rate": 2.7829943452469753e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003895406407536939,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42003402709961,
      "state_query_norm": 1414.05869140625,
      "state_relation_loss": 0.0003895406407536939,
      "step": 26960,
      "student_distance_mean": 1944.1529296875,
      "teacher_distance_mean": 1.1527507185935975,
      "weighted_auxiliary_loss": 0.0003895406407536939,
      "weighted_relation_loss": 0.0003895406407536939
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.229186155285312,
      "flow_matching_loss": 0.015176289714872837,
      "grad_norm": 0.0881970003247261,
      "learning_rate": 2.7648915955730213e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041153608763124796,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25015411376953,
      "state_query_norm": 1404.7665771484376,
      "state_relation_loss": 0.00041153608763124796,
      "step": 26970,
      "student_distance_mean": 1933.9530883789062,
      "teacher_distance_mean": 1.1474721431732178,
      "weighted_auxiliary_loss": 0.00041153608763124796,
      "weighted_relation_loss": 0.00041153608763124796
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.238540692235734,
      "flow_matching_loss": 0.014560995902866126,
      "grad_norm": 0.08984968066215515,
      "learning_rate": 2.746846241009765e-06,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003674739069538191,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37000846862793,
      "state_query_norm": 1418.310546875,
      "state_relation_loss": 0.0003674739069538191,
      "step": 26980,
      "student_distance_mean": 1946.3761474609375,
      "teacher_distance_mean": 1.1551862120628358,
      "weighted_auxiliary_loss": 0.0003674739069538191,
      "weighted_relation_loss": 0.0003674739069538191
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.247895229186156,
      "flow_matching_loss": 0.016641319449990987,
      "grad_norm": 0.09119805693626404,
      "learning_rate": 2.7288583034839945e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004042705550091341,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.611481475830075,
      "state_query_norm": 1429.6962768554688,
      "state_relation_loss": 0.0004042705550091341,
      "step": 26990,
      "student_distance_mean": 1958.9856079101562,
      "teacher_distance_mean": 1.1626874208450317,
      "weighted_auxiliary_loss": 0.0004042705550091341,
      "weighted_relation_loss": 0.0004042705550091341
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.257249766136574,
      "flow_matching_loss": 0.016546277236193417,
      "grad_norm": 0.07903091609477997,
      "learning_rate": 2.7109278048527752e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039058719703461976,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36591758728027,
      "state_query_norm": 1416.4113403320312,
      "state_relation_loss": 0.00039058719703461976,
      "step": 27000,
      "student_distance_mean": 1942.90966796875,
      "teacher_distance_mean": 1.1526473999023437,
      "weighted_auxiliary_loss": 0.00039058719703461976,
      "weighted_relation_loss": 0.00039058719703461976
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.266604303086996,
      "flow_matching_loss": 0.015027138311415911,
      "grad_norm": 0.07420354336500168,
      "learning_rate": 2.6930547669033413e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003948080993723124,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31905288696289,
      "state_query_norm": 1404.3554443359376,
      "state_relation_loss": 0.0003948080993723124,
      "step": 27010,
      "student_distance_mean": 1936.0257934570313,
      "teacher_distance_mean": 1.1479570388793945,
      "weighted_auxiliary_loss": 0.0003948080993723124,
      "weighted_relation_loss": 0.0003948080993723124
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.27595884003742,
      "flow_matching_loss": 0.01626001950353384,
      "grad_norm": 0.08495817333459854,
      "learning_rate": 2.675239211353109e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003843740501906723,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25807876586914,
      "state_query_norm": 1413.5861938476562,
      "state_relation_loss": 0.0003843740501906723,
      "step": 27020,
      "student_distance_mean": 1939.1924072265624,
      "teacher_distance_mean": 1.151081919670105,
      "weighted_auxiliary_loss": 0.0003843740501906723,
      "weighted_relation_loss": 0.0003843740501906723
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.28531337698784,
      "flow_matching_loss": 0.015428938250988723,
      "grad_norm": 0.08796745538711548,
      "learning_rate": 2.6574811598496786e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003763045271625742,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.489508056640624,
      "state_query_norm": 1416.9085815429687,
      "state_relation_loss": 0.0003763045271625742,
      "step": 27030,
      "student_distance_mean": 1946.507373046875,
      "teacher_distance_mean": 1.1558248281478882,
      "weighted_auxiliary_loss": 0.0003763045271625742,
      "weighted_relation_loss": 0.0003763045271625742
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.29466791393826,
      "flow_matching_loss": 0.01650685481727123,
      "grad_norm": 0.07960949093103409,
      "learning_rate": 2.6397806339707456e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003811447502812371,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07717704772949,
      "state_query_norm": 1402.4569458007813,
      "state_relation_loss": 0.0003811447502812371,
      "step": 27040,
      "student_distance_mean": 1927.9779174804687,
      "teacher_distance_mean": 1.1457426786422729,
      "weighted_auxiliary_loss": 0.0003811447502812371,
      "weighted_relation_loss": 0.0003811447502812371
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.30402245088868,
      "flow_matching_loss": 0.01534618278965354,
      "grad_norm": 0.09129869192838669,
      "learning_rate": 2.6221376552241217e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000394375654286705,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.259700012207034,
      "state_query_norm": 1403.1103637695312,
      "state_relation_loss": 0.000394375654286705,
      "step": 27050,
      "student_distance_mean": 1931.8610961914062,
      "teacher_distance_mean": 1.147810697555542,
      "weighted_auxiliary_loss": 0.000394375654286705,
      "weighted_relation_loss": 0.000394375654286705
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.313376987839103,
      "flow_matching_loss": 0.015104714967310428,
      "grad_norm": 0.10165851563215256,
      "learning_rate": 2.604552245047681e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003921522235032171,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.633954238891604,
      "state_query_norm": 1385.2388061523438,
      "state_relation_loss": 0.0003921522235032171,
      "step": 27060,
      "student_distance_mean": 1899.4661376953125,
      "teacher_distance_mean": 1.1291353702545166,
      "weighted_auxiliary_loss": 0.0003921522235032171,
      "weighted_relation_loss": 0.0003921522235032171
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.32273152478952,
      "flow_matching_loss": 0.01615718547254801,
      "grad_norm": 0.08211031556129456,
      "learning_rate": 2.587024424809359e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003812214417848736,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.713325119018556,
      "state_query_norm": 1422.8040649414063,
      "state_relation_loss": 0.0003812214417848736,
      "step": 27070,
      "student_distance_mean": 1957.8628784179687,
      "teacher_distance_mean": 1.1633562207221986,
      "weighted_auxiliary_loss": 0.0003812214417848736,
      "weighted_relation_loss": 0.0003812214417848736
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.332086061739943,
      "flow_matching_loss": 0.015371863264590502,
      "grad_norm": 0.09782157093286514,
      "learning_rate": 2.5695542158071186e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038508707657456396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32922706604004,
      "state_query_norm": 1409.8877319335938,
      "state_relation_loss": 0.00038508707657456396,
      "step": 27080,
      "student_distance_mean": 1940.9510498046875,
      "teacher_distance_mean": 1.1535693407058716,
      "weighted_auxiliary_loss": 0.00038508707657456396,
      "weighted_relation_loss": 0.00038508707657456396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.341440598690365,
      "flow_matching_loss": 0.015991474967449903,
      "grad_norm": 0.08192045241594315,
      "learning_rate": 2.5521416392689068e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000380845702602528,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.408256912231444,
      "state_query_norm": 1375.5033081054687,
      "state_relation_loss": 0.000380845702602528,
      "step": 27090,
      "student_distance_mean": 1890.6712646484375,
      "teacher_distance_mean": 1.1229800939559937,
      "weighted_auxiliary_loss": 0.000380845702602528,
      "weighted_relation_loss": 0.000380845702602528
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.350795135640787,
      "flow_matching_loss": 0.015164231974631548,
      "grad_norm": 0.06550972163677216,
      "learning_rate": 2.5347867163526384e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003869612643029541,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41507797241211,
      "state_query_norm": 1418.73701171875,
      "state_relation_loss": 0.0003869612643029541,
      "step": 27100,
      "student_distance_mean": 1943.9055908203125,
      "teacher_distance_mean": 1.1548348426818849,
      "weighted_auxiliary_loss": 0.0003869612643029541,
      "weighted_relation_loss": 0.0003869612643029541
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.360149672591206,
      "flow_matching_loss": 0.016260144766420126,
      "grad_norm": 0.08935832232236862,
      "learning_rate": 2.517489468146189e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003916702466085553,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.85959091186523,
      "state_query_norm": 1394.9623046875,
      "state_relation_loss": 0.0003916702466085553,
      "step": 27110,
      "student_distance_mean": 1914.9440795898438,
      "teacher_distance_mean": 1.1381762981414796,
      "weighted_auxiliary_loss": 0.0003916702466085553,
      "weighted_relation_loss": 0.0003916702466085553
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.369504209541628,
      "flow_matching_loss": 0.015951605513691902,
      "grad_norm": 0.08231835812330246,
      "learning_rate": 2.500249915667341e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041554978815838695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91638832092285,
      "state_query_norm": 1432.83857421875,
      "state_relation_loss": 0.00041554978815838695,
      "step": 27120,
      "student_distance_mean": 1970.6309448242187,
      "teacher_distance_mean": 1.1706868767738343,
      "weighted_auxiliary_loss": 0.00041554978815838695,
      "weighted_relation_loss": 0.00041554978815838695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.37885874649205,
      "flow_matching_loss": 0.015285462141036987,
      "grad_norm": 0.07817666977643967,
      "learning_rate": 2.4830680798637817e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003557403717422858,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15292053222656,
      "state_query_norm": 1406.3706298828124,
      "state_relation_loss": 0.0003557403717422858,
      "step": 27130,
      "student_distance_mean": 1934.3209350585937,
      "teacher_distance_mean": 1.1497519850730895,
      "weighted_auxiliary_loss": 0.0003557403717422858,
      "weighted_relation_loss": 0.0003557403717422858
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.388213283442468,
      "flow_matching_loss": 0.016077413130551575,
      "grad_norm": 0.08025416731834412,
      "learning_rate": 2.465943981613056e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004084145009983331,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.157728576660155,
      "state_query_norm": 1403.9761962890625,
      "state_relation_loss": 0.0004084145009983331,
      "step": 27140,
      "student_distance_mean": 1929.762744140625,
      "teacher_distance_mean": 1.147347366809845,
      "weighted_auxiliary_loss": 0.0004084145009983331,
      "weighted_relation_loss": 0.0004084145009983331
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.39756782039289,
      "flow_matching_loss": 0.016597354132682086,
      "grad_norm": 0.0882725864648819,
      "learning_rate": 2.448877641722569e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003599689371185377,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16357421875,
      "state_query_norm": 1409.20908203125,
      "state_relation_loss": 0.0003599689371185377,
      "step": 27150,
      "student_distance_mean": 1932.8685668945313,
      "teacher_distance_mean": 1.1489332795143128,
      "weighted_auxiliary_loss": 0.0003599689371185377,
      "weighted_relation_loss": 0.0003599689371185377
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.406922357343312,
      "flow_matching_loss": 0.01648381445556879,
      "grad_norm": 0.07126648724079132,
      "learning_rate": 2.431869080929522e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040980815247166903,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.29475326538086,
      "state_query_norm": 1440.0463745117188,
      "state_relation_loss": 0.00040980815247166903,
      "step": 27160,
      "student_distance_mean": 1988.4056396484375,
      "teacher_distance_mean": 1.182574987411499,
      "weighted_auxiliary_loss": 0.00040980815247166903,
      "weighted_relation_loss": 0.00040980815247166903
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.416276894293734,
      "flow_matching_loss": 0.015988444909453392,
      "grad_norm": 0.10086894780397415,
      "learning_rate": 2.4149183199009216e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003973371523898095,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.9664665222168,
      "state_query_norm": 1428.88828125,
      "state_relation_loss": 0.0003973371523898095,
      "step": 27170,
      "student_distance_mean": 1973.6193359375,
      "teacher_distance_mean": 1.172273874282837,
      "weighted_auxiliary_loss": 0.0003973371523898095,
      "weighted_relation_loss": 0.0003973371523898095
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.425631431244152,
      "flow_matching_loss": 0.014748042821884156,
      "grad_norm": 0.06409801542758942,
      "learning_rate": 2.3980253792335427e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036719650088343767,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.621407699584964,
      "state_query_norm": 1436.51376953125,
      "state_relation_loss": 0.00036719650088343767,
      "step": 27180,
      "student_distance_mean": 1961.6164672851562,
      "teacher_distance_mean": 1.1651742219924928,
      "weighted_auxiliary_loss": 0.00036719650088343767,
      "weighted_relation_loss": 0.00036719650088343767
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.434985968194574,
      "flow_matching_loss": 0.015635788347572087,
      "grad_norm": 0.074189193546772,
      "learning_rate": 2.381190279453899e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038120666577015074,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.39587059020996,
      "state_query_norm": 1413.3615966796874,
      "state_relation_loss": 0.00038120666577015074,
      "step": 27190,
      "student_distance_mean": 1942.6524291992187,
      "teacher_distance_mean": 1.154078185558319,
      "weighted_auxiliary_loss": 0.00038120666577015074,
      "weighted_relation_loss": 0.00038120666577015074
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.444340505144996,
      "flow_matching_loss": 0.014862744975835085,
      "grad_norm": 0.1083555743098259,
      "learning_rate": 2.364413041018232e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004122791171539575,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30853385925293,
      "state_query_norm": 1405.6523193359376,
      "state_relation_loss": 0.0004122791171539575,
      "step": 27200,
      "student_distance_mean": 1934.719287109375,
      "teacher_distance_mean": 1.149908983707428,
      "weighted_auxiliary_loss": 0.0004122791171539575,
      "weighted_relation_loss": 0.0004122791171539575
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.453695042095415,
      "flow_matching_loss": 0.016409203596413135,
      "grad_norm": 0.08192586898803711,
      "learning_rate": 2.347693684312463e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039231761766131965,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.21434135437012,
      "state_query_norm": 1409.5903198242188,
      "state_relation_loss": 0.00039231761766131965,
      "step": 27210,
      "student_distance_mean": 1937.6399291992188,
      "teacher_distance_mean": 1.1517528772354126,
      "weighted_auxiliary_loss": 0.00039231761766131965,
      "weighted_relation_loss": 0.00039231761766131965
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.463049579045837,
      "flow_matching_loss": 0.015914710890501737,
      "grad_norm": 0.09221255034208298,
      "learning_rate": 2.331032229652186e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037632090097758917,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.341033935546875,
      "state_query_norm": 1417.1826049804688,
      "state_relation_loss": 0.00037632090097758917,
      "step": 27220,
      "student_distance_mean": 1941.6882080078126,
      "teacher_distance_mean": 1.1544880270957947,
      "weighted_auxiliary_loss": 0.00037632090097758917,
      "weighted_relation_loss": 0.00037632090097758917
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.47240411599626,
      "flow_matching_loss": 0.015116368606686591,
      "grad_norm": 0.0746975988149643,
      "learning_rate": 2.314428697282628e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039983991009648887,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.60867538452148,
      "state_query_norm": 1424.4179931640624,
      "state_relation_loss": 0.00039983991009648887,
      "step": 27230,
      "student_distance_mean": 1957.5063354492188,
      "teacher_distance_mean": 1.1644562602043151,
      "weighted_auxiliary_loss": 0.00039983991009648887,
      "weighted_relation_loss": 0.00039983991009648887
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.48175865294668,
      "flow_matching_loss": 0.01588619891554117,
      "grad_norm": 0.07248031347990036,
      "learning_rate": 2.297883107378673e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036510034697130324,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24009895324707,
      "state_query_norm": 1420.8286499023438,
      "state_relation_loss": 0.00036510034697130324,
      "step": 27240,
      "student_distance_mean": 1940.2695556640624,
      "teacher_distance_mean": 1.1525403261184692,
      "weighted_auxiliary_loss": 0.00036510034697130324,
      "weighted_relation_loss": 0.00036510034697130324
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.4911131898971,
      "flow_matching_loss": 0.015445623733103275,
      "grad_norm": 0.09011862426996231,
      "learning_rate": 2.2813954800447513e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038333752891048787,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.71492729187012,
      "state_query_norm": 1384.17880859375,
      "state_relation_loss": 0.00038333752891048787,
      "step": 27250,
      "student_distance_mean": 1906.220556640625,
      "teacher_distance_mean": 1.1319282412528993,
      "weighted_auxiliary_loss": 0.00038333752891048787,
      "weighted_relation_loss": 0.00038333752891048787
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.50046772684752,
      "flow_matching_loss": 0.016131817270070313,
      "grad_norm": 0.12911218404769897,
      "learning_rate": 2.2649658353148974e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040772312204353514,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64970397949219,
      "state_query_norm": 1423.2796264648437,
      "state_relation_loss": 0.00040772312204353514,
      "step": 27260,
      "student_distance_mean": 1955.619482421875,
      "teacher_distance_mean": 1.1620108842849732,
      "weighted_auxiliary_loss": 0.00040772312204353514,
      "weighted_relation_loss": 0.00040772312204353514
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.509822263797943,
      "flow_matching_loss": 0.0152463398873806,
      "grad_norm": 0.07874052226543427,
      "learning_rate": 2.2485941931526645e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037215403572190555,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.78715438842774,
      "state_query_norm": 1390.4318115234375,
      "state_relation_loss": 0.00037215403572190555,
      "step": 27270,
      "student_distance_mean": 1910.1995849609375,
      "teacher_distance_mean": 1.1349268555641174,
      "weighted_auxiliary_loss": 0.00037215403572190555,
      "weighted_relation_loss": 0.00037215403572190555
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.51917680074836,
      "flow_matching_loss": 0.015637457091361284,
      "grad_norm": 0.07587027549743652,
      "learning_rate": 2.232280573451151e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003733530786121264,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2919319152832,
      "state_query_norm": 1410.0638549804687,
      "state_relation_loss": 0.0003733530786121264,
      "step": 27280,
      "student_distance_mean": 1937.9420166015625,
      "teacher_distance_mean": 1.150499415397644,
      "weighted_auxiliary_loss": 0.0003733530786121264,
      "weighted_relation_loss": 0.0003733530786121264
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.528531337698784,
      "flow_matching_loss": 0.016694548074156046,
      "grad_norm": 0.06458928436040878,
      "learning_rate": 2.216024996032945e-06,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003857169562252238,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.12919044494629,
      "state_query_norm": 1403.0222534179688,
      "state_relation_loss": 0.0003857169562252238,
      "step": 27290,
      "student_distance_mean": 1928.1768310546875,
      "teacher_distance_mean": 1.1449875235557556,
      "weighted_auxiliary_loss": 0.0003857169562252238,
      "weighted_relation_loss": 0.0003857169562252238
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.537885874649206,
      "flow_matching_loss": 0.015287869330495595,
      "grad_norm": 0.07428883016109467,
      "learning_rate": 2.1998274806501074e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042986632324755193,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.85818634033203,
      "state_query_norm": 1430.1423095703126,
      "state_relation_loss": 0.00042986632324755193,
      "step": 27300,
      "student_distance_mean": 1968.019140625,
      "teacher_distance_mean": 1.1686519742012025,
      "weighted_auxiliary_loss": 0.00042986632324755193,
      "weighted_relation_loss": 0.00042986632324755193
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.547240411599624,
      "flow_matching_loss": 0.016273040510714053,
      "grad_norm": 0.09493758529424667,
      "learning_rate": 2.183688046984139e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003914567205356434,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.82695541381836,
      "state_query_norm": 1392.5250610351563,
      "state_relation_loss": 0.0003914567205356434,
      "step": 27310,
      "student_distance_mean": 1911.3709838867187,
      "teacher_distance_mean": 1.1348701000213623,
      "weighted_auxiliary_loss": 0.0003914567205356434,
      "weighted_relation_loss": 0.0003914567205356434
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.556594948550046,
      "flow_matching_loss": 0.01615288946777582,
      "grad_norm": 0.09316840022802353,
      "learning_rate": 2.167606714645981e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003746329806745052,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.88071632385254,
      "state_query_norm": 1395.8109130859375,
      "state_relation_loss": 0.0003746329806745052,
      "step": 27320,
      "student_distance_mean": 1917.6337158203125,
      "teacher_distance_mean": 1.1382152676582336,
      "weighted_auxiliary_loss": 0.0003746329806745052,
      "weighted_relation_loss": 0.0003746329806745052
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.565949485500468,
      "flow_matching_loss": 0.01645187484100461,
      "grad_norm": 0.12161802500486374,
      "learning_rate": 2.151583503175958e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038528919976670294,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.761200714111325,
      "state_query_norm": 1396.323779296875,
      "state_relation_loss": 0.00038528919976670294,
      "step": 27330,
      "student_distance_mean": 1911.3237426757812,
      "teacher_distance_mean": 1.13522766828537,
      "weighted_auxiliary_loss": 0.00038528919976670294,
      "weighted_relation_loss": 0.00038528919976670294
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.57530402245089,
      "flow_matching_loss": 0.014229527581483126,
      "grad_norm": 0.08442218601703644,
      "learning_rate": 2.1356184320437957e-06,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004035243269754574,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.744515609741214,
      "state_query_norm": 1426.2556518554688,
      "state_relation_loss": 0.0004035243269754574,
      "step": 27340,
      "student_distance_mean": 1963.6979125976563,
      "teacher_distance_mean": 1.1656369686126709,
      "weighted_auxiliary_loss": 0.0004035243269754574,
      "weighted_relation_loss": 0.0004035243269754574
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.58465855940131,
      "flow_matching_loss": 0.015110879950225354,
      "grad_norm": 0.09120244532823563,
      "learning_rate": 2.119711520648554e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003865830454742536,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.122846603393555,
      "state_query_norm": 1412.5800170898438,
      "state_relation_loss": 0.0003865830454742536,
      "step": 27350,
      "student_distance_mean": 1930.6270141601562,
      "teacher_distance_mean": 1.1459423542022704,
      "weighted_auxiliary_loss": 0.0003865830454742536,
      "weighted_relation_loss": 0.0003865830454742536
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.59401309635173,
      "flow_matching_loss": 0.01563803544268012,
      "grad_norm": 0.08417536318302155,
      "learning_rate": 2.103862788318628e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037918767775408925,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50181427001953,
      "state_query_norm": 1423.0036865234374,
      "state_relation_loss": 0.00037918767775408925,
      "step": 27360,
      "student_distance_mean": 1951.990380859375,
      "teacher_distance_mean": 1.159223699569702,
      "weighted_auxiliary_loss": 0.00037918767775408925,
      "weighted_relation_loss": 0.00037918767775408925
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.603367633302152,
      "flow_matching_loss": 0.016073108557611702,
      "grad_norm": 0.09714114665985107,
      "learning_rate": 2.088072254311729e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037924476782791317,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.300269317626956,
      "state_query_norm": 1410.715771484375,
      "state_relation_loss": 0.00037924476782791317,
      "step": 27370,
      "student_distance_mean": 1940.1926147460938,
      "teacher_distance_mean": 1.1525098919868468,
      "weighted_auxiliary_loss": 0.00037924476782791317,
      "weighted_relation_loss": 0.00037924476782791317
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.61272217025257,
      "flow_matching_loss": 0.015131569281220437,
      "grad_norm": 0.09467662125825882,
      "learning_rate": 2.0723399378148435e-06,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037699023669119925,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.35966453552246,
      "state_query_norm": 1421.5692749023438,
      "state_relation_loss": 0.00037699023669119925,
      "step": 27380,
      "student_distance_mean": 1947.7659301757812,
      "teacher_distance_mean": 1.1562188744544983,
      "weighted_auxiliary_loss": 0.00037699023669119925,
      "weighted_relation_loss": 0.00037699023669119925
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.622076707202993,
      "flow_matching_loss": 0.01535831205546856,
      "grad_norm": 0.09519413113594055,
      "learning_rate": 2.0566658579442065e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003801241749897599,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.10621528625488,
      "state_query_norm": 1407.0186889648437,
      "state_relation_loss": 0.0003801241749897599,
      "step": 27390,
      "student_distance_mean": 1930.908935546875,
      "teacher_distance_mean": 1.1467458248138427,
      "weighted_auxiliary_loss": 0.0003801241749897599,
      "weighted_relation_loss": 0.0003801241749897599
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.631431244153415,
      "flow_matching_loss": 0.015851706638932227,
      "grad_norm": 0.09198204427957535,
      "learning_rate": 2.0410500337453176e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038491660088766364,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.02069244384766,
      "state_query_norm": 1407.8771362304688,
      "state_relation_loss": 0.00038491660088766364,
      "step": 27400,
      "student_distance_mean": 1925.1343872070313,
      "teacher_distance_mean": 1.1432709336280822,
      "weighted_auxiliary_loss": 0.00038491660088766364,
      "weighted_relation_loss": 0.00038491660088766364
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.640785781103837,
      "flow_matching_loss": 0.014364537876099348,
      "grad_norm": 0.07213388383388519,
      "learning_rate": 2.0254924841928645e-06,
      "loss": 0.0148,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004056024772580713,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4640625,
      "state_query_norm": 1419.6802856445313,
      "state_relation_loss": 0.0004056024772580713,
      "step": 27410,
      "student_distance_mean": 1947.253076171875,
      "teacher_distance_mean": 1.156951940059662,
      "weighted_auxiliary_loss": 0.0004056024772580713,
      "weighted_relation_loss": 0.0004056024772580713
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.650140318054255,
      "flow_matching_loss": 0.014715216588228941,
      "grad_norm": 0.08136257529258728,
      "learning_rate": 2.009993228190754e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004157224524533376,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.11994934082031,
      "state_query_norm": 1435.9341552734375,
      "state_relation_loss": 0.0004157224524533376,
      "step": 27420,
      "student_distance_mean": 1977.0534423828126,
      "teacher_distance_mean": 1.175313115119934,
      "weighted_auxiliary_loss": 0.0004157224524533376,
      "weighted_relation_loss": 0.0004157224524533376
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.659494855004677,
      "flow_matching_loss": 0.015858387015759944,
      "grad_norm": 0.08077236264944077,
      "learning_rate": 1.9945522845720323e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038245067407842724,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.90874137878418,
      "state_query_norm": 1400.563330078125,
      "state_relation_loss": 0.00038245067407842724,
      "step": 27430,
      "student_distance_mean": 1919.4011962890625,
      "teacher_distance_mean": 1.140448546409607,
      "weighted_auxiliary_loss": 0.00038245067407842724,
      "weighted_relation_loss": 0.00038245067407842724
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.6688493919551,
      "flow_matching_loss": 0.01531727248802781,
      "grad_norm": 0.09119721502065659,
      "learning_rate": 1.9791696720988963e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004014795704279095,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56099166870117,
      "state_query_norm": 1416.5362182617187,
      "state_relation_loss": 0.0004014795704279095,
      "step": 27440,
      "student_distance_mean": 1951.636669921875,
      "teacher_distance_mean": 1.1596207737922668,
      "weighted_auxiliary_loss": 0.0004014795704279095,
      "weighted_relation_loss": 0.0004014795704279095
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.678203928905518,
      "flow_matching_loss": 0.016083521023392677,
      "grad_norm": 0.07743526995182037,
      "learning_rate": 1.9638454094626833e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038452001754194496,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74713325500488,
      "state_query_norm": 1435.2566162109374,
      "state_relation_loss": 0.00038452001754194496,
      "step": 27450,
      "student_distance_mean": 1962.4964599609375,
      "teacher_distance_mean": 1.1652680158615112,
      "weighted_auxiliary_loss": 0.00038452001754194496,
      "weighted_relation_loss": 0.00038452001754194496
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.68755846585594,
      "flow_matching_loss": 0.015658685937523842,
      "grad_norm": 0.06967395544052124,
      "learning_rate": 1.9485795152838104e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040848983626347036,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.786568832397464,
      "state_query_norm": 1420.8544311523438,
      "state_relation_loss": 0.00040848983626347036,
      "step": 27460,
      "student_distance_mean": 1960.0430541992187,
      "teacher_distance_mean": 1.1641263961791992,
      "weighted_auxiliary_loss": 0.00040848983626347036,
      "weighted_relation_loss": 0.00040848983626347036
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.69691300280636,
      "flow_matching_loss": 0.014310927595943213,
      "grad_norm": 0.09089162200689316,
      "learning_rate": 1.933372008111778e-06,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036234323924873025,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.99448509216309,
      "state_query_norm": 1398.634033203125,
      "state_relation_loss": 0.00036234323924873025,
      "step": 27470,
      "student_distance_mean": 1921.9168579101563,
      "teacher_distance_mean": 1.141644036769867,
      "weighted_auxiliary_loss": 0.00036234323924873025,
      "weighted_relation_loss": 0.00036234323924873025
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.706267539756784,
      "flow_matching_loss": 0.015686372481286525,
      "grad_norm": 0.09234929084777832,
      "learning_rate": 1.918222906425143e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003877833834849298,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.6765640258789,
      "state_query_norm": 1390.2008911132812,
      "state_relation_loss": 0.0003877833834849298,
      "step": 27480,
      "student_distance_mean": 1902.477001953125,
      "teacher_distance_mean": 1.130295741558075,
      "weighted_auxiliary_loss": 0.0003877833834849298,
      "weighted_relation_loss": 0.0003877833834849298
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.715622076707202,
      "flow_matching_loss": 0.016259152069687845,
      "grad_norm": 0.07232031971216202,
      "learning_rate": 1.903132228631488e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038303416222333907,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.198643112182616,
      "state_query_norm": 1409.3211303710937,
      "state_relation_loss": 0.00038303416222333907,
      "step": 27490,
      "student_distance_mean": 1931.8677490234375,
      "teacher_distance_mean": 1.1487201809883119,
      "weighted_auxiliary_loss": 0.00038303416222333907,
      "weighted_relation_loss": 0.00038303416222333907
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.724976613657624,
      "flow_matching_loss": 0.016553661413490774,
      "grad_norm": 0.14203211665153503,
      "learning_rate": 1.8880999930674215e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003703019552631304,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.91815757751465,
      "state_query_norm": 1403.1830322265625,
      "state_relation_loss": 0.0003703019552631304,
      "step": 27500,
      "student_distance_mean": 1922.7283813476563,
      "teacher_distance_mean": 1.1426828622817993,
      "weighted_auxiliary_loss": 0.0003703019552631304,
      "weighted_relation_loss": 0.0003703019552631304
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.734331150608046,
      "flow_matching_loss": 0.015617803391069173,
      "grad_norm": 0.06674395501613617,
      "learning_rate": 1.8731262179985166e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038986651052255186,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.98790855407715,
      "state_query_norm": 1402.0353637695312,
      "state_relation_loss": 0.00038986651052255186,
      "step": 27510,
      "student_distance_mean": 1920.1239013671875,
      "teacher_distance_mean": 1.1403577327728271,
      "weighted_auxiliary_loss": 0.00038986651052255186,
      "weighted_relation_loss": 0.00038986651052255186
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.743685687558465,
      "flow_matching_loss": 0.014214446023106574,
      "grad_norm": 0.08793340623378754,
      "learning_rate": 1.8582109216193244e-06,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004059780651004985,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32275505065918,
      "state_query_norm": 1406.3199340820313,
      "state_relation_loss": 0.0004059780651004985,
      "step": 27520,
      "student_distance_mean": 1935.7170043945312,
      "teacher_distance_mean": 1.1492856979370116,
      "weighted_auxiliary_loss": 0.0004059780651004985,
      "weighted_relation_loss": 0.0004059780651004985
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.753040224508887,
      "flow_matching_loss": 0.015768442302942276,
      "grad_norm": 0.07692263275384903,
      "learning_rate": 1.8433541220533368e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004064370965352282,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.462765502929685,
      "state_query_norm": 1416.8325927734375,
      "state_relation_loss": 0.0004064370965352282,
      "step": 27530,
      "student_distance_mean": 1946.7773315429688,
      "teacher_distance_mean": 1.1562756538391112,
      "weighted_auxiliary_loss": 0.0004064370965352282,
      "weighted_relation_loss": 0.0004064370965352282
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.76239476145931,
      "flow_matching_loss": 0.014970769826322793,
      "grad_norm": 0.09253869205713272,
      "learning_rate": 1.8285558373529577e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000384693275555037,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26690788269043,
      "state_query_norm": 1412.4549438476563,
      "state_relation_loss": 0.000384693275555037,
      "step": 27540,
      "student_distance_mean": 1939.2611938476562,
      "teacher_distance_mean": 1.1522393703460694,
      "weighted_auxiliary_loss": 0.000384693275555037,
      "weighted_relation_loss": 0.000384693275555037
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.77174929840973,
      "flow_matching_loss": 0.01536353100091219,
      "grad_norm": 0.09394413977861404,
      "learning_rate": 1.8138160854995145e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038210174825508146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.87467651367187,
      "state_query_norm": 1399.61953125,
      "state_relation_loss": 0.00038210174825508146,
      "step": 27550,
      "student_distance_mean": 1917.237548828125,
      "teacher_distance_mean": 1.1391107201576234,
      "weighted_auxiliary_loss": 0.00038210174825508146,
      "weighted_relation_loss": 0.00038210174825508146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.78110383536015,
      "flow_matching_loss": 0.016869521234184504,
      "grad_norm": 0.08734478056430817,
      "learning_rate": 1.7991348844031864e-06,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004041157866595313,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.3441822052002,
      "state_query_norm": 1408.8560668945313,
      "state_relation_loss": 0.0004041157866595313,
      "step": 27560,
      "student_distance_mean": 1938.5320556640625,
      "teacher_distance_mean": 1.1506879806518555,
      "weighted_auxiliary_loss": 0.0004041157866595313,
      "weighted_relation_loss": 0.0004041157866595313
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.79045837231057,
      "flow_matching_loss": 0.014989993441849947,
      "grad_norm": 0.0884227454662323,
      "learning_rate": 1.78451225190302e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040004099428188057,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27403030395508,
      "state_query_norm": 1402.2808471679687,
      "state_relation_loss": 0.00040004099428188057,
      "step": 27570,
      "student_distance_mean": 1932.6675903320313,
      "teacher_distance_mean": 1.1485058903694152,
      "weighted_auxiliary_loss": 0.00040004099428188057,
      "weighted_relation_loss": 0.00040004099428188057
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.799812909260993,
      "flow_matching_loss": 0.015245393663644791,
      "grad_norm": 0.07225654274225235,
      "learning_rate": 1.7699482057668914e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003787265042774379,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57744789123535,
      "state_query_norm": 1421.5287841796876,
      "state_relation_loss": 0.0003787265042774379,
      "step": 27580,
      "student_distance_mean": 1953.90634765625,
      "teacher_distance_mean": 1.1609565496444703,
      "weighted_auxiliary_loss": 0.0003787265042774379,
      "weighted_relation_loss": 0.0003787265042774379
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.80916744621141,
      "flow_matching_loss": 0.015201363060623408,
      "grad_norm": 0.07753970474004745,
      "learning_rate": 1.7554427636914895e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003856976050883532,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.8226921081543,
      "state_query_norm": 1396.13896484375,
      "state_relation_loss": 0.0003856976050883532,
      "step": 27590,
      "student_distance_mean": 1914.2183227539062,
      "teacher_distance_mean": 1.1377201199531555,
      "weighted_auxiliary_loss": 0.0003856976050883532,
      "weighted_relation_loss": 0.0003856976050883532
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.818521983161833,
      "flow_matching_loss": 0.015609812177717685,
      "grad_norm": 0.072233185172081,
      "learning_rate": 1.7409959433022871e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036324794346001,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.08692665100098,
      "state_query_norm": 1403.0025024414062,
      "state_relation_loss": 0.00036324794346001,
      "step": 27600,
      "student_distance_mean": 1928.13759765625,
      "teacher_distance_mean": 1.1455289363861083,
      "weighted_auxiliary_loss": 0.00036324794346001,
      "weighted_relation_loss": 0.00036324794346001
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.827876520112255,
      "flow_matching_loss": 0.015640823263674976,
      "grad_norm": 0.06308688223361969,
      "learning_rate": 1.7266077621535481e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003806636086665094,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48294296264648,
      "state_query_norm": 1416.8606323242188,
      "state_relation_loss": 0.0003806636086665094,
      "step": 27610,
      "student_distance_mean": 1947.9856689453125,
      "teacher_distance_mean": 1.1579861760139465,
      "weighted_auxiliary_loss": 0.0003806636086665094,
      "weighted_relation_loss": 0.0003806636086665094
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.837231057062674,
      "flow_matching_loss": 0.016629234235733748,
      "grad_norm": 0.08857011049985886,
      "learning_rate": 1.7122782377282597e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003702680376591161,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.764102935791016,
      "state_query_norm": 1395.6115356445312,
      "state_relation_loss": 0.0003702680376591161,
      "step": 27620,
      "student_distance_mean": 1910.79443359375,
      "teacher_distance_mean": 1.135485005378723,
      "weighted_auxiliary_loss": 0.0003702680376591161,
      "weighted_relation_loss": 0.0003702680376591161
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.846585594013096,
      "flow_matching_loss": 0.016381200589239597,
      "grad_norm": 0.0875750482082367,
      "learning_rate": 1.6980073874381497e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004220890317810699,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.81506271362305,
      "state_query_norm": 1432.2962646484375,
      "state_relation_loss": 0.0004220890317810699,
      "step": 27630,
      "student_distance_mean": 1964.8652709960938,
      "teacher_distance_mean": 1.1675304532051087,
      "weighted_auxiliary_loss": 0.0004220890317810699,
      "weighted_relation_loss": 0.0004220890317810699
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.855940130963518,
      "flow_matching_loss": 0.015837534610182048,
      "grad_norm": 0.08022402971982956,
      "learning_rate": 1.6837952286236413e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036868589522782714,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58112411499023,
      "state_query_norm": 1413.8381713867188,
      "state_relation_loss": 0.00036868589522782714,
      "step": 27640,
      "student_distance_mean": 1951.87236328125,
      "teacher_distance_mean": 1.1599061727523803,
      "weighted_auxiliary_loss": 0.00036868589522782714,
      "weighted_relation_loss": 0.00036868589522782714
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.86529466791394,
      "flow_matching_loss": 0.014763146918267012,
      "grad_norm": 0.07516459375619888,
      "learning_rate": 1.6696417785538488e-06,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004188700608210638,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64284896850586,
      "state_query_norm": 1416.4130981445312,
      "state_relation_loss": 0.0004188700608210638,
      "step": 27650,
      "student_distance_mean": 1953.1094604492187,
      "teacher_distance_mean": 1.1606926798820496,
      "weighted_auxiliary_loss": 0.0004188700608210638,
      "weighted_relation_loss": 0.0004188700608210638
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.874649204864358,
      "flow_matching_loss": 0.018535223603248597,
      "grad_norm": 0.09777438640594482,
      "learning_rate": 1.6555470544265538e-06,
      "loss": 0.0189,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037474507989827545,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.66381988525391,
      "state_query_norm": 1431.2036987304687,
      "state_relation_loss": 0.00037474507989827545,
      "step": 27660,
      "student_distance_mean": 1961.0498901367187,
      "teacher_distance_mean": 1.1647609353065491,
      "weighted_auxiliary_loss": 0.00037474507989827545,
      "weighted_relation_loss": 0.00037474507989827545
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.88400374181478,
      "flow_matching_loss": 0.014807530026882888,
      "grad_norm": 0.07699405401945114,
      "learning_rate": 1.6415110733681738e-06,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003740989079233259,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.663913345336915,
      "state_query_norm": 1422.4114501953125,
      "state_relation_loss": 0.0003740989079233259,
      "step": 27670,
      "student_distance_mean": 1955.9752075195313,
      "teacher_distance_mean": 1.1622898936271668,
      "weighted_auxiliary_loss": 0.0003740989079233259,
      "weighted_relation_loss": 0.0003740989079233259
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.893358278765202,
      "flow_matching_loss": 0.01471178289502859,
      "grad_norm": 0.08422629535198212,
      "learning_rate": 1.6275338524337436e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038217745022848246,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53430328369141,
      "state_query_norm": 1425.2374267578125,
      "state_relation_loss": 0.00038217745022848246,
      "step": 27680,
      "student_distance_mean": 1952.88515625,
      "teacher_distance_mean": 1.159950053691864,
      "weighted_auxiliary_loss": 0.00038217745022848246,
      "weighted_relation_loss": 0.00038217745022848246
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.90271281571562,
      "flow_matching_loss": 0.015827125404030085,
      "grad_norm": 0.07792936265468597,
      "learning_rate": 1.6136154086069056e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040902472974266855,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.28802375793457,
      "state_query_norm": 1405.4859252929687,
      "state_relation_loss": 0.00040902472974266855,
      "step": 27690,
      "student_distance_mean": 1934.61142578125,
      "teacher_distance_mean": 1.1487216711044312,
      "weighted_auxiliary_loss": 0.00040902472974266855,
      "weighted_relation_loss": 0.00040902472974266855
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.912067352666043,
      "flow_matching_loss": 0.01591313201934099,
      "grad_norm": 0.0766364261507988,
      "learning_rate": 1.5997557587998868e-06,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000360768788959831,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.64866676330566,
      "state_query_norm": 1389.7995727539062,
      "state_relation_loss": 0.000360768788959831,
      "step": 27700,
      "student_distance_mean": 1906.3298950195312,
      "teacher_distance_mean": 1.1325441479682923,
      "weighted_auxiliary_loss": 0.000360768788959831,
      "weighted_relation_loss": 0.000360768788959831
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.921421889616465,
      "flow_matching_loss": 0.014710105396807194,
      "grad_norm": 0.09118419140577316,
      "learning_rate": 1.5859549198534606e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036554991675075144,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.460133743286136,
      "state_query_norm": 1422.6146118164063,
      "state_relation_loss": 0.00036554991675075144,
      "step": 27710,
      "student_distance_mean": 1951.21708984375,
      "teacher_distance_mean": 1.1589043140411377,
      "weighted_auxiliary_loss": 0.00036554991675075144,
      "weighted_relation_loss": 0.00036554991675075144
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.930776426566887,
      "flow_matching_loss": 0.017770733591169117,
      "grad_norm": 0.08475275337696075,
      "learning_rate": 1.572212908536963e-06,
      "loss": 0.0182,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039003183192107824,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55959205627441,
      "state_query_norm": 1418.6823974609374,
      "state_relation_loss": 0.00039003183192107824,
      "step": 27720,
      "student_distance_mean": 1952.5303344726562,
      "teacher_distance_mean": 1.1601454138755798,
      "weighted_auxiliary_loss": 0.00039003183192107824,
      "weighted_relation_loss": 0.00039003183192107824
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.940130963517305,
      "flow_matching_loss": 0.01714227385818958,
      "grad_norm": 0.0958927720785141,
      "learning_rate": 1.5585297415482202e-06,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004043066815938801,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41761856079101,
      "state_query_norm": 1407.6788818359375,
      "state_relation_loss": 0.0004043066815938801,
      "step": 27730,
      "student_distance_mean": 1942.5800170898438,
      "teacher_distance_mean": 1.1538634300231934,
      "weighted_auxiliary_loss": 0.0004043066815938801,
      "weighted_relation_loss": 0.0004043066815938801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.949485500467727,
      "flow_matching_loss": 0.016206872183829546,
      "grad_norm": 0.07963919639587402,
      "learning_rate": 1.5449054355135717e-06,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004048971983138472,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.858449172973636,
      "state_query_norm": 1435.3576171875,
      "state_relation_loss": 0.0004048971983138472,
      "step": 27740,
      "student_distance_mean": 1971.760791015625,
      "teacher_distance_mean": 1.1714942097663879,
      "weighted_auxiliary_loss": 0.0004048971983138472,
      "weighted_relation_loss": 0.0004048971983138472
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.95884003741815,
      "flow_matching_loss": 0.014923507813364267,
      "grad_norm": 0.06818032264709473,
      "learning_rate": 1.5313400069878415e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037680317473132163,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2878475189209,
      "state_query_norm": 1417.0165649414062,
      "state_relation_loss": 0.00037680317473132163,
      "step": 27750,
      "student_distance_mean": 1940.8742065429688,
      "teacher_distance_mean": 1.1533381938934326,
      "weighted_auxiliary_loss": 0.00037680317473132163,
      "weighted_relation_loss": 0.00037680317473132163
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.968194574368567,
      "flow_matching_loss": 0.015745235607028008,
      "grad_norm": 0.08142907917499542,
      "learning_rate": 1.5178334724542887e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003860807453747839,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.427162170410156,
      "state_query_norm": 1424.4899780273438,
      "state_relation_loss": 0.0003860807453747839,
      "step": 27760,
      "student_distance_mean": 1951.4817016601562,
      "teacher_distance_mean": 1.1595938086509705,
      "weighted_auxiliary_loss": 0.0003860807453747839,
      "weighted_relation_loss": 0.0003860807453747839
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.97754911131899,
      "flow_matching_loss": 0.01564031755551696,
      "grad_norm": 0.07079027593135834,
      "learning_rate": 1.504385848324641e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038648873451165857,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.241272354125975,
      "state_query_norm": 1419.0236206054688,
      "state_relation_loss": 0.00038648873451165857,
      "step": 27770,
      "student_distance_mean": 1938.0164306640625,
      "teacher_distance_mean": 1.152172029018402,
      "weighted_auxiliary_loss": 0.00038648873451165857,
      "weighted_relation_loss": 0.00038648873451165857
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.98690364826941,
      "flow_matching_loss": 0.01614912012591958,
      "grad_norm": 0.0790935531258583,
      "learning_rate": 1.490997150939033e-06,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039771526062395426,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.35961418151855,
      "state_query_norm": 1412.5339599609374,
      "state_relation_loss": 0.00039771526062395426,
      "step": 27780,
      "student_distance_mean": 1940.1062744140625,
      "teacher_distance_mean": 1.1533515930175782,
      "weighted_auxiliary_loss": 0.00039771526062395426,
      "weighted_relation_loss": 0.00039771526062395426
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 25.996258185219833,
      "flow_matching_loss": 0.015709912963211538,
      "grad_norm": 0.0873497873544693,
      "learning_rate": 1.4776673965659793e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004265619587386027,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.472542953491214,
      "state_query_norm": 1417.48876953125,
      "state_relation_loss": 0.0004265619587386027,
      "step": 27790,
      "student_distance_mean": 1946.9450561523438,
      "teacher_distance_mean": 1.1571722865104674,
      "weighted_auxiliary_loss": 0.0004265619587386027,
      "weighted_relation_loss": 0.0004265619587386027
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.00561272217025,
      "flow_matching_loss": 0.01551100444048643,
      "grad_norm": 0.08351609110832214,
      "learning_rate": 1.4643966014023957e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0003777150355745107,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.39130973815918,
      "state_query_norm": 1415.9244384765625,
      "state_relation_loss": 0.0003777150355745107,
      "step": 27800,
      "student_distance_mean": 1943.82431640625,
      "teacher_distance_mean": 1.1544749736785889,
      "weighted_auxiliary_loss": 0.0003777150355745107,
      "weighted_relation_loss": 0.0003777150355745107
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.014967259120674,
      "flow_matching_loss": 0.016466946620494128,
      "grad_norm": 0.08573918789625168,
      "learning_rate": 1.4511847815735503e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003809285030001774,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.22822303771973,
      "state_query_norm": 1405.6617431640625,
      "state_relation_loss": 0.0003809285030001774,
      "step": 27810,
      "student_distance_mean": 1933.3557739257812,
      "teacher_distance_mean": 1.1488323092460633,
      "weighted_auxiliary_loss": 0.0003809285030001774,
      "weighted_relation_loss": 0.0003809285030001774
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.024321796071096,
      "flow_matching_loss": 0.014686802867799997,
      "grad_norm": 0.07174663245677948,
      "learning_rate": 1.438031953133051e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003841202647890896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.98642425537109,
      "state_query_norm": 1405.2710815429687,
      "state_relation_loss": 0.0003841202647890896,
      "step": 27820,
      "student_distance_mean": 1922.557177734375,
      "teacher_distance_mean": 1.141359567642212,
      "weighted_auxiliary_loss": 0.0003841202647890896,
      "weighted_relation_loss": 0.0003841202647890896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.033676333021514,
      "flow_matching_loss": 0.014953216072171926,
      "grad_norm": 0.08538606017827988,
      "learning_rate": 1.4249381320628197e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004029700648970902,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.81107177734375,
      "state_query_norm": 1428.4403686523438,
      "state_relation_loss": 0.0004029700648970902,
      "step": 27830,
      "student_distance_mean": 1965.2534301757812,
      "teacher_distance_mean": 1.1670290946960449,
      "weighted_auxiliary_loss": 0.0004029700648970902,
      "weighted_relation_loss": 0.0004029700648970902
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.043030869971936,
      "flow_matching_loss": 0.015677935723215342,
      "grad_norm": 0.07237247377634048,
      "learning_rate": 1.4119033342730902e-06,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003701936802826822,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.238105392456056,
      "state_query_norm": 1417.004833984375,
      "state_relation_loss": 0.0003701936802826822,
      "step": 27840,
      "student_distance_mean": 1938.12353515625,
      "teacher_distance_mean": 1.1507380723953247,
      "weighted_auxiliary_loss": 0.0003701936802826822,
      "weighted_relation_loss": 0.0003701936802826822
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.052385406922358,
      "flow_matching_loss": 0.01567363878712058,
      "grad_norm": 0.06583762168884277,
      "learning_rate": 1.3989275756023656e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039242254861164837,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.732405471801755,
      "state_query_norm": 1426.5828247070312,
      "state_relation_loss": 0.00039242254861164837,
      "step": 27850,
      "student_distance_mean": 1959.1509887695313,
      "teacher_distance_mean": 1.1633116602897644,
      "weighted_auxiliary_loss": 0.00039242254861164837,
      "weighted_relation_loss": 0.00039242254861164837
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.061739943872777,
      "flow_matching_loss": 0.015436975192278624,
      "grad_norm": 0.10652577131986618,
      "learning_rate": 1.386010871817417e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003800152044277638,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.120645904541014,
      "state_query_norm": 1403.1641479492187,
      "state_relation_loss": 0.0003800152044277638,
      "step": 27860,
      "student_distance_mean": 1930.17978515625,
      "teacher_distance_mean": 1.1458528280258178,
      "weighted_auxiliary_loss": 0.0003800152044277638,
      "weighted_relation_loss": 0.0003800152044277638
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.0710944808232,
      "flow_matching_loss": 0.015185737609863281,
      "grad_norm": 0.07733969390392303,
      "learning_rate": 1.3731532386132616e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039430107863154263,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50971870422363,
      "state_query_norm": 1411.7205810546875,
      "state_relation_loss": 0.00039430107863154263,
      "step": 27870,
      "student_distance_mean": 1944.6356201171875,
      "teacher_distance_mean": 1.1549801707267762,
      "weighted_auxiliary_loss": 0.00039430107863154263,
      "weighted_relation_loss": 0.00039430107863154263
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.08044901777362,
      "flow_matching_loss": 0.017559893801808356,
      "grad_norm": 0.07730258256196976,
      "learning_rate": 1.360354691613136e-06,
      "loss": 0.018,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000405440287431702,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.79959945678711,
      "state_query_norm": 1427.1349975585938,
      "state_relation_loss": 0.000405440287431702,
      "step": 27880,
      "student_distance_mean": 1964.4496337890625,
      "teacher_distance_mean": 1.166779589653015,
      "weighted_auxiliary_loss": 0.000405440287431702,
      "weighted_relation_loss": 0.000405440287431702
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.089803554724043,
      "flow_matching_loss": 0.014795332495123149,
      "grad_norm": 0.09102602303028107,
      "learning_rate": 1.3476152463684776e-06,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039461954729631544,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.326428985595705,
      "state_query_norm": 1403.7587036132813,
      "state_relation_loss": 0.00039461954729631544,
      "step": 27890,
      "student_distance_mean": 1934.782568359375,
      "teacher_distance_mean": 1.149518871307373,
      "weighted_auxiliary_loss": 0.00039461954729631544,
      "weighted_relation_loss": 0.00039461954729631544
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.09915809167446,
      "flow_matching_loss": 0.01648561777547002,
      "grad_norm": 0.09764759987592697,
      "learning_rate": 1.3349349183589155e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040814321546349673,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16790542602539,
      "state_query_norm": 1413.29716796875,
      "state_relation_loss": 0.00040814321546349673,
      "step": 27900,
      "student_distance_mean": 1933.919873046875,
      "teacher_distance_mean": 1.14747554063797,
      "weighted_auxiliary_loss": 0.00040814321546349673,
      "weighted_relation_loss": 0.00040814321546349673
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.108512628624883,
      "flow_matching_loss": 0.01637865845113993,
      "grad_norm": 0.08650654554367065,
      "learning_rate": 1.3223137229922356e-06,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003805705811828375,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.391863632202146,
      "state_query_norm": 1424.5236083984375,
      "state_relation_loss": 0.0003805705811828375,
      "step": 27910,
      "student_distance_mean": 1948.5956176757813,
      "teacher_distance_mean": 1.1574735522270203,
      "weighted_auxiliary_loss": 0.0003805705811828375,
      "weighted_relation_loss": 0.0003805705811828375
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.117867165575305,
      "flow_matching_loss": 0.01633328991010785,
      "grad_norm": 0.0823332816362381,
      "learning_rate": 1.3097516756043981e-06,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038302358880173417,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.284535598754886,
      "state_query_norm": 1416.0359008789062,
      "state_relation_loss": 0.00038302358880173417,
      "step": 27920,
      "student_distance_mean": 1937.9685546875,
      "teacher_distance_mean": 1.1512232184410096,
      "weighted_auxiliary_loss": 0.00038302358880173417,
      "weighted_relation_loss": 0.00038302358880173417
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.127221702525723,
      "flow_matching_loss": 0.015756535343825817,
      "grad_norm": 0.08342205733060837,
      "learning_rate": 1.29724879145946e-06,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036524109018500896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.61590995788574,
      "state_query_norm": 1399.9450805664062,
      "state_relation_loss": 0.00036524109018500896,
      "step": 27930,
      "student_distance_mean": 1910.4354858398438,
      "teacher_distance_mean": 1.1346131324768067,
      "weighted_auxiliary_loss": 0.00036524109018500896,
      "weighted_relation_loss": 0.00036524109018500896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.136576239476145,
      "flow_matching_loss": 0.014874236937612296,
      "grad_norm": 0.09164093434810638,
      "learning_rate": 1.284805085749613e-06,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003832847054582089,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44420318603515,
      "state_query_norm": 1419.11552734375,
      "state_relation_loss": 0.0003832847054582089,
      "step": 27940,
      "student_distance_mean": 1947.83212890625,
      "teacher_distance_mean": 1.1569565534591675,
      "weighted_auxiliary_loss": 0.0003832847054582089,
      "weighted_relation_loss": 0.0003832847054582089
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.145930776426567,
      "flow_matching_loss": 0.015811140555888413,
      "grad_norm": 0.11803512275218964,
      "learning_rate": 1.2724205735951288e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003628002043114975,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.93249702453613,
      "state_query_norm": 1398.4099243164062,
      "state_relation_loss": 0.0003628002043114975,
      "step": 27950,
      "student_distance_mean": 1920.2985595703126,
      "teacher_distance_mean": 1.1405303359031678,
      "weighted_auxiliary_loss": 0.0003628002043114975,
      "weighted_relation_loss": 0.0003628002043114975
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.15528531337699,
      "flow_matching_loss": 0.015360258612781763,
      "grad_norm": 0.07613860815763474,
      "learning_rate": 1.2600952700443591e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003672431339509785,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4958438873291,
      "state_query_norm": 1427.8061767578124,
      "state_relation_loss": 0.0003672431339509785,
      "step": 27960,
      "student_distance_mean": 1954.2315551757813,
      "teacher_distance_mean": 1.1606438040733338,
      "weighted_auxiliary_loss": 0.0003672431339509785,
      "weighted_relation_loss": 0.0003672431339509785
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.164639850327408,
      "flow_matching_loss": 0.014496096968650818,
      "grad_norm": 0.08197852969169617,
      "learning_rate": 1.247829190073707e-06,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036724153615068643,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.346594619750974,
      "state_query_norm": 1416.5307373046876,
      "state_relation_loss": 0.00036724153615068643,
      "step": 27970,
      "student_distance_mean": 1942.6911254882812,
      "teacher_distance_mean": 1.1541640996932983,
      "weighted_auxiliary_loss": 0.00036724153615068643,
      "weighted_relation_loss": 0.00036724153615068643
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.17399438727783,
      "flow_matching_loss": 0.01455309521406889,
      "grad_norm": 0.0935596153140068,
      "learning_rate": 1.2356223485876173e-06,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00034819027641788127,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.187461853027344,
      "state_query_norm": 1405.207275390625,
      "state_relation_loss": 0.00034819027641788127,
      "step": 27980,
      "student_distance_mean": 1934.3029907226562,
      "teacher_distance_mean": 1.1490979671478272,
      "weighted_auxiliary_loss": 0.00034819027641788127,
      "weighted_relation_loss": 0.00034819027641788127
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.18334892422825,
      "flow_matching_loss": 0.014176632277667522,
      "grad_norm": 0.06919434666633606,
      "learning_rate": 1.2234747604185526e-06,
      "loss": 0.0145,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037131886056158693,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.248254013061526,
      "state_query_norm": 1410.9217895507813,
      "state_relation_loss": 0.00037131886056158693,
      "step": 27990,
      "student_distance_mean": 1935.8875854492187,
      "teacher_distance_mean": 1.149317240715027,
      "weighted_auxiliary_loss": 0.00037131886056158693,
      "weighted_relation_loss": 0.00037131886056158693
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.19270346117867,
      "flow_matching_loss": 0.015823454782366752,
      "grad_norm": 0.08478929102420807,
      "learning_rate": 1.2113864403269836e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003790132323047146,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73702392578125,
      "state_query_norm": 1425.1272338867188,
      "state_relation_loss": 0.0003790132323047146,
      "step": 28000,
      "student_distance_mean": 1962.0913696289062,
      "teacher_distance_mean": 1.1654396176338195,
      "weighted_auxiliary_loss": 0.0003790132323047146,
      "weighted_relation_loss": 0.0003790132323047146
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.202057998129092,
      "flow_matching_loss": 0.016551144886761904,
      "grad_norm": 0.0894058346748352,
      "learning_rate": 1.1993574030013555e-06,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038624289154540745,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.2860725402832,
      "state_query_norm": 1413.543017578125,
      "state_relation_loss": 0.00038624289154540745,
      "step": 28010,
      "student_distance_mean": 1939.897998046875,
      "teacher_distance_mean": 1.1531041741371155,
      "weighted_auxiliary_loss": 0.00038624289154540745,
      "weighted_relation_loss": 0.00038624289154540745
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.211412535079514,
      "flow_matching_loss": 0.01579991923645139,
      "grad_norm": 0.08067157864570618,
      "learning_rate": 1.1873876630580816e-06,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003569566208170727,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.88181838989258,
      "state_query_norm": 1401.3317016601563,
      "state_relation_loss": 0.0003569566208170727,
      "step": 28020,
      "student_distance_mean": 1919.0635864257813,
      "teacher_distance_mean": 1.1405728578567504,
      "weighted_auxiliary_loss": 0.0003569566208170727,
      "weighted_relation_loss": 0.0003569566208170727
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.220767072029936,
      "flow_matching_loss": 0.016552161797881125,
      "grad_norm": 0.10551141947507858,
      "learning_rate": 1.1754772350415278e-06,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004129366978304461,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.50288314819336,
      "state_query_norm": 1373.4045166015626,
      "state_relation_loss": 0.0004129366978304461,
      "step": 28030,
      "student_distance_mean": 1891.6910034179687,
      "teacher_distance_mean": 1.1242812514305114,
      "weighted_auxiliary_loss": 0.0004129366978304461,
      "weighted_relation_loss": 0.0004129366978304461
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.230121608980355,
      "flow_matching_loss": 0.016030238755047322,
      "grad_norm": 0.089871346950531,
      "learning_rate": 1.163626133423984e-06,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040737150702625514,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.77048873901367,
      "state_query_norm": 1426.8205200195312,
      "state_relation_loss": 0.00040737150702625514,
      "step": 28040,
      "student_distance_mean": 1962.559912109375,
      "teacher_distance_mean": 1.166312003135681,
      "weighted_auxiliary_loss": 0.00040737150702625514,
      "weighted_relation_loss": 0.00040737150702625514
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.239476145930777,
      "flow_matching_loss": 0.015416475012898445,
      "grad_norm": 0.09035075455904007,
      "learning_rate": 1.1518343726056591e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039669265679549427,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16808738708496,
      "state_query_norm": 1403.3486938476562,
      "state_relation_loss": 0.00039669265679549427,
      "step": 28050,
      "student_distance_mean": 1929.29853515625,
      "teacher_distance_mean": 1.1475545048713685,
      "weighted_auxiliary_loss": 0.00039669265679549427,
      "weighted_relation_loss": 0.00039669265679549427
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.2488306828812,
      "flow_matching_loss": 0.015351691003888845,
      "grad_norm": 0.06448126584291458,
      "learning_rate": 1.1401019669146474e-06,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003919171285815537,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.70081253051758,
      "state_query_norm": 1422.965869140625,
      "state_relation_loss": 0.0003919171285815537,
      "step": 28060,
      "student_distance_mean": 1957.9478271484375,
      "teacher_distance_mean": 1.1643533110618591,
      "weighted_auxiliary_loss": 0.0003919171285815537,
      "weighted_relation_loss": 0.0003919171285815537
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.258185219831617,
      "flow_matching_loss": 0.015351286996155978,
      "grad_norm": 0.08626557886600494,
      "learning_rate": 1.128428930606934e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042606214701663704,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.5603084564209,
      "state_query_norm": 1426.9520141601563,
      "state_relation_loss": 0.00042606214701663704,
      "step": 28070,
      "student_distance_mean": 1957.7587524414062,
      "teacher_distance_mean": 1.1631937503814698,
      "weighted_auxiliary_loss": 0.00042606214701663704,
      "weighted_relation_loss": 0.00042606214701663704
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.26753975678204,
      "flow_matching_loss": 0.015380952600389719,
      "grad_norm": 0.07300214469432831,
      "learning_rate": 1.1168152778663621e-06,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000412558694370091,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.186126708984375,
      "state_query_norm": 1404.6512573242187,
      "state_relation_loss": 0.000412558694370091,
      "step": 28080,
      "student_distance_mean": 1933.7168334960938,
      "teacher_distance_mean": 1.1494624257087707,
      "weighted_auxiliary_loss": 0.000412558694370091,
      "weighted_relation_loss": 0.000412558694370091
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.27689429373246,
      "flow_matching_loss": 0.015181452874094248,
      "grad_norm": 0.0779203325510025,
      "learning_rate": 1.10526102280461e-06,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038728004146832974,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58899192810058,
      "state_query_norm": 1414.4371826171875,
      "state_relation_loss": 0.00038728004146832974,
      "step": 28090,
      "student_distance_mean": 1951.8236328125,
      "teacher_distance_mean": 1.1596441507339477,
      "weighted_auxiliary_loss": 0.00038728004146832974,
      "weighted_relation_loss": 0.00038728004146832974
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.28624883068288,
      "flow_matching_loss": 0.015051455330103635,
      "grad_norm": 0.06508681923151016,
      "learning_rate": 1.0937661794611864e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038643532316200434,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.9972957611084,
      "state_query_norm": 1399.6857177734375,
      "state_relation_loss": 0.00038643532316200434,
      "step": 28100,
      "student_distance_mean": 1921.7916259765625,
      "teacher_distance_mean": 1.1423489689826964,
      "weighted_auxiliary_loss": 0.00038643532316200434,
      "weighted_relation_loss": 0.00038643532316200434
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.2956033676333,
      "flow_matching_loss": 0.014099639374762774,
      "grad_norm": 0.08418241888284683,
      "learning_rate": 1.082330761803413e-06,
      "loss": 0.0145,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00044251073559280484,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.281606674194336,
      "state_query_norm": 1403.039794921875,
      "state_relation_loss": 0.00044251073559280484,
      "step": 28110,
      "student_distance_mean": 1931.5568359375,
      "teacher_distance_mean": 1.1488555192947387,
      "weighted_auxiliary_loss": 0.00044251073559280484,
      "weighted_relation_loss": 0.00044251073559280484
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.304957904583723,
      "flow_matching_loss": 0.01406130837276578,
      "grad_norm": 0.08258867263793945,
      "learning_rate": 1.0709547837263966e-06,
      "loss": 0.0144,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037005593185313046,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.320251846313475,
      "state_query_norm": 1413.149609375,
      "state_relation_loss": 0.00037005593185313046,
      "step": 28120,
      "student_distance_mean": 1940.5473754882812,
      "teacher_distance_mean": 1.1539824247360229,
      "weighted_auxiliary_loss": 0.00037005593185313046,
      "weighted_relation_loss": 0.00037005593185313046
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.314312441534145,
      "flow_matching_loss": 0.015009378269314765,
      "grad_norm": 0.06679973751306534,
      "learning_rate": 1.059638259053025e-06,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003927646670490503,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.86374092102051,
      "state_query_norm": 1430.0422973632812,
      "state_relation_loss": 0.0003927646670490503,
      "step": 28130,
      "student_distance_mean": 1968.8363647460938,
      "teacher_distance_mean": 1.1701130509376525,
      "weighted_auxiliary_loss": 0.0003927646670490503,
      "weighted_relation_loss": 0.0003927646670490503
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.323666978484564,
      "flow_matching_loss": 0.014741685520857573,
      "grad_norm": 0.09374424070119858,
      "learning_rate": 1.0483812015339434e-06,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003736958635272458,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.107659912109376,
      "state_query_norm": 1406.5882080078125,
      "state_relation_loss": 0.0003736958635272458,
      "step": 28140,
      "student_distance_mean": 1930.3138549804687,
      "teacher_distance_mean": 1.1480968832969665,
      "weighted_auxiliary_loss": 0.0003736958635272458,
      "weighted_relation_loss": 0.0003736958635272458
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.333021515434986,
      "flow_matching_loss": 0.01431772941723466,
      "grad_norm": 0.08148568123579025,
      "learning_rate": 1.037183624847543e-06,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003956554777687415,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36038932800293,
      "state_query_norm": 1424.4764526367187,
      "state_relation_loss": 0.0003956554777687415,
      "step": 28150,
      "student_distance_mean": 1946.0815551757812,
      "teacher_distance_mean": 1.1566439986228942,
      "weighted_auxiliary_loss": 0.0003956554777687415,
      "weighted_relation_loss": 0.0003956554777687415
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.342376052385408,
      "flow_matching_loss": 0.015512045100331306,
      "grad_norm": 0.08781109005212784,
      "learning_rate": 1.0260455425999238e-06,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004207480378681794,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.548024368286136,
      "state_query_norm": 1416.141064453125,
      "state_relation_loss": 0.0004207480378681794,
      "step": 28160,
      "student_distance_mean": 1948.2037841796875,
      "teacher_distance_mean": 1.157337474822998,
      "weighted_auxiliary_loss": 0.0004207480378681794,
      "weighted_relation_loss": 0.0004207480378681794
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.351730589335826,
      "flow_matching_loss": 0.014632855821400882,
      "grad_norm": 0.08955862373113632,
      "learning_rate": 1.0149669683249208e-06,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037616093759424986,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15133857727051,
      "state_query_norm": 1407.38583984375,
      "state_relation_loss": 0.00037616093759424986,
      "step": 28170,
      "student_distance_mean": 1931.932666015625,
      "teacher_distance_mean": 1.1478393435478211,
      "weighted_auxiliary_loss": 0.00037616093759424986,
      "weighted_relation_loss": 0.00037616093759424986
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.36108512628625,
      "flow_matching_loss": 0.013553228974342347,
      "grad_norm": 0.07682018727064133,
      "learning_rate": 1.003947915484027e-06,
      "loss": 0.014,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003993632039055228,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.13399047851563,
      "state_query_norm": 1399.5020751953125,
      "state_relation_loss": 0.0003993632039055228,
      "step": 28180,
      "student_distance_mean": 1925.7516723632812,
      "teacher_distance_mean": 1.1436203360557555,
      "weighted_auxiliary_loss": 0.0003993632039055228,
      "weighted_relation_loss": 0.0003993632039055228
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.37043966323667,
      "flow_matching_loss": 0.015059453062713146,
      "grad_norm": 0.09749984741210938,
      "learning_rate": 9.92988397466449e-07,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038345360953826455,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.670658111572266,
      "state_query_norm": 1393.7535034179687,
      "state_relation_loss": 0.00038345360953826455,
      "step": 28190,
      "student_distance_mean": 1908.3300170898438,
      "teacher_distance_mean": 1.13369277715683,
      "weighted_auxiliary_loss": 0.00038345360953826455,
      "weighted_relation_loss": 0.00038345360953826455
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.379794200187092,
      "flow_matching_loss": 0.015418847836554051,
      "grad_norm": 0.07983316481113434,
      "learning_rate": 9.820884275890286e-07,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003786556655541062,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37963790893555,
      "state_query_norm": 1419.3918090820312,
      "state_relation_loss": 0.0003786556655541062,
      "step": 28200,
      "student_distance_mean": 1943.4723999023438,
      "teacher_distance_mean": 1.15331974029541,
      "weighted_auxiliary_loss": 0.0003786556655541062,
      "weighted_relation_loss": 0.0003786556655541062
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.38914873713751,
      "flow_matching_loss": 0.017464390583336355,
      "grad_norm": 0.07779340445995331,
      "learning_rate": 9.712480190962546e-07,
      "loss": 0.0178,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003757495956961066,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.72716102600098,
      "state_query_norm": 1400.603173828125,
      "state_relation_loss": 0.0003757495956961066,
      "step": 28210,
      "student_distance_mean": 1913.5957885742187,
      "teacher_distance_mean": 1.1365682244300843,
      "weighted_auxiliary_loss": 0.0003757495956961066,
      "weighted_relation_loss": 0.0003757495956961066
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.398503274087933,
      "flow_matching_loss": 0.01519369799643755,
      "grad_norm": 0.09030016511678696,
      "learning_rate": 9.604671851602464e-07,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003868948668241501,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.099235153198244,
      "state_query_norm": 1398.1177978515625,
      "state_relation_loss": 0.0003868948668241501,
      "step": 28220,
      "student_distance_mean": 1926.539306640625,
      "teacher_distance_mean": 1.144059407711029,
      "weighted_auxiliary_loss": 0.0003868948668241501,
      "weighted_relation_loss": 0.0003868948668241501
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.407857811038355,
      "flow_matching_loss": 0.013890737388283014,
      "grad_norm": 0.07242070138454437,
      "learning_rate": 9.497459388807306e-07,
      "loss": 0.0143,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040447090577799827,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.658747863769534,
      "state_query_norm": 1425.0937133789062,
      "state_relation_loss": 0.00040447090577799827,
      "step": 28230,
      "student_distance_mean": 1957.8376098632812,
      "teacher_distance_mean": 1.163430142402649,
      "weighted_auxiliary_loss": 0.00040447090577799827,
      "weighted_relation_loss": 0.00040447090577799827
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.417212347988773,
      "flow_matching_loss": 0.015740729961544277,
      "grad_norm": 0.07491487264633179,
      "learning_rate": 9.390842932850364e-07,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004141860292293131,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.85664367675781,
      "state_query_norm": 1422.8061889648438,
      "state_relation_loss": 0.0004141860292293131,
      "step": 28240,
      "student_distance_mean": 1964.741259765625,
      "teacher_distance_mean": 1.1671704649925232,
      "weighted_auxiliary_loss": 0.0004141860292293131,
      "weighted_relation_loss": 0.0004141860292293131
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.426566884939195,
      "flow_matching_loss": 0.014258935768157243,
      "grad_norm": 0.08358324319124222,
      "learning_rate": 9.28482261328073e-07,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000359137007035315,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36996345520019,
      "state_query_norm": 1407.22919921875,
      "state_relation_loss": 0.000359137007035315,
      "step": 28250,
      "student_distance_mean": 1939.8977661132812,
      "teacher_distance_mean": 1.1523578763008118,
      "weighted_auxiliary_loss": 0.000359137007035315,
      "weighted_relation_loss": 0.000359137007035315
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.435921421889617,
      "flow_matching_loss": 0.015696664340794086,
      "grad_norm": 0.07058031857013702,
      "learning_rate": 9.179398558923025e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040534175059292465,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27976417541504,
      "state_query_norm": 1415.7301635742188,
      "state_relation_loss": 0.00040534175059292465,
      "step": 28260,
      "student_distance_mean": 1941.6726806640625,
      "teacher_distance_mean": 1.1535322546958924,
      "weighted_auxiliary_loss": 0.00040534175059292465,
      "weighted_relation_loss": 0.00040534175059292465
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.44527595884004,
      "flow_matching_loss": 0.015574270207434893,
      "grad_norm": 0.07543524354696274,
      "learning_rate": 9.074570897877388e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00043914363195654007,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.360968017578124,
      "state_query_norm": 1408.544091796875,
      "state_relation_loss": 0.00043914363195654007,
      "step": 28270,
      "student_distance_mean": 1937.0314086914063,
      "teacher_distance_mean": 1.1508558988571167,
      "weighted_auxiliary_loss": 0.00043914363195654007,
      "weighted_relation_loss": 0.00043914363195654007
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.454630495790457,
      "flow_matching_loss": 0.015027664229273796,
      "grad_norm": 0.081053726375103,
      "learning_rate": 8.970339757519375e-07,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038488671998493373,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47432861328125,
      "state_query_norm": 1410.2960327148437,
      "state_relation_loss": 0.00038488671998493373,
      "step": 28280,
      "student_distance_mean": 1946.6941650390625,
      "teacher_distance_mean": 1.1566782236099242,
      "weighted_auxiliary_loss": 0.00038488671998493373,
      "weighted_relation_loss": 0.00038488671998493373
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.46398503274088,
      "flow_matching_loss": 0.014408988133072852,
      "grad_norm": 0.07624007016420364,
      "learning_rate": 8.866705264499619e-07,
      "loss": 0.0148,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038192120555322615,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.978498458862305,
      "state_query_norm": 1401.5615844726562,
      "state_relation_loss": 0.00038192120555322615,
      "step": 28290,
      "student_distance_mean": 1923.09091796875,
      "teacher_distance_mean": 1.1428630590438842,
      "weighted_auxiliary_loss": 0.00038192120555322615,
      "weighted_relation_loss": 0.00038192120555322615
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.4733395696913,
      "flow_matching_loss": 0.015492142457515002,
      "grad_norm": 0.09275736659765244,
      "learning_rate": 8.763667544743836e-07,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003857708710711449,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50996856689453,
      "state_query_norm": 1417.4793212890625,
      "state_relation_loss": 0.0003857708710711449,
      "step": 28300,
      "student_distance_mean": 1946.4838745117188,
      "teacher_distance_mean": 1.157181167602539,
      "weighted_auxiliary_loss": 0.0003857708710711449,
      "weighted_relation_loss": 0.0003857708710711449
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.48269410664172,
      "flow_matching_loss": 0.017328830156475305,
      "grad_norm": 0.08935817331075668,
      "learning_rate": 8.661226723452543e-07,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003960253554396331,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.31071243286133,
      "state_query_norm": 1417.7383544921875,
      "state_relation_loss": 0.0003960253554396331,
      "step": 28310,
      "student_distance_mean": 1939.9332153320313,
      "teacher_distance_mean": 1.1532329320907593,
      "weighted_auxiliary_loss": 0.0003960253554396331,
      "weighted_relation_loss": 0.0003960253554396331
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.492048643592142,
      "flow_matching_loss": 0.015106599871069193,
      "grad_norm": 0.09091980755329132,
      "learning_rate": 8.559382925101001e-07,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039891908527351917,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71867218017578,
      "state_query_norm": 1436.3514038085937,
      "state_relation_loss": 0.00039891908527351917,
      "step": 28320,
      "student_distance_mean": 1966.9081420898438,
      "teacher_distance_mean": 1.1681571125984191,
      "weighted_auxiliary_loss": 0.00039891908527351917,
      "weighted_relation_loss": 0.00039891908527351917
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.501403180542564,
      "flow_matching_loss": 0.014796716533601284,
      "grad_norm": 0.08522870391607285,
      "learning_rate": 8.458136273438943e-07,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037632263265550134,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.9396728515625,
      "state_query_norm": 1393.834033203125,
      "state_relation_loss": 0.00037632263265550134,
      "step": 28330,
      "student_distance_mean": 1920.0657104492188,
      "teacher_distance_mean": 1.1410432934761048,
      "weighted_auxiliary_loss": 0.00037632263265550134,
      "weighted_relation_loss": 0.00037632263265550134
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.510757717492986,
      "flow_matching_loss": 0.01541390735656023,
      "grad_norm": 0.07561823725700378,
      "learning_rate": 8.357486891490795e-07,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003858657815726474,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75407752990723,
      "state_query_norm": 1424.7958374023438,
      "state_relation_loss": 0.0003858657815726474,
      "step": 28340,
      "student_distance_mean": 1961.8350952148437,
      "teacher_distance_mean": 1.1658100366592408,
      "weighted_auxiliary_loss": 0.0003858657815726474,
      "weighted_relation_loss": 0.0003858657815726474
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.520112254443404,
      "flow_matching_loss": 0.01456092894077301,
      "grad_norm": 0.07245747745037079,
      "learning_rate": 8.257434901554895e-07,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004003319190815091,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36634902954101,
      "state_query_norm": 1411.1809448242188,
      "state_relation_loss": 0.0004003319190815091,
      "step": 28350,
      "student_distance_mean": 1940.25927734375,
      "teacher_distance_mean": 1.151935076713562,
      "weighted_auxiliary_loss": 0.0004003319190815091,
      "weighted_relation_loss": 0.0004003319190815091
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.529466791393826,
      "flow_matching_loss": 0.016025519650429487,
      "grad_norm": 0.09678418189287186,
      "learning_rate": 8.157980425203937e-07,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00035105108399875463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26442947387695,
      "state_query_norm": 1412.4810546875,
      "state_relation_loss": 0.00035105108399875463,
      "step": 28360,
      "student_distance_mean": 1941.8573852539062,
      "teacher_distance_mean": 1.15316903591156,
      "weighted_auxiliary_loss": 0.00035105108399875463,
      "weighted_relation_loss": 0.00035105108399875463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.53882132834425,
      "flow_matching_loss": 0.015597242396324873,
      "grad_norm": 0.07393420487642288,
      "learning_rate": 8.059123583284367e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004060671548359096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.48475570678711,
      "state_query_norm": 1409.546142578125,
      "state_relation_loss": 0.0004060671548359096,
      "step": 28370,
      "student_distance_mean": 1942.0131225585938,
      "teacher_distance_mean": 1.153141975402832,
      "weighted_auxiliary_loss": 0.0004060671548359096,
      "weighted_relation_loss": 0.0004060671548359096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.548175865294667,
      "flow_matching_loss": 0.017882782220840453,
      "grad_norm": 0.07915687561035156,
      "learning_rate": 7.960864495916654e-07,
      "loss": 0.0183,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004067758389282972,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.73526954650879,
      "state_query_norm": 1422.8833374023438,
      "state_relation_loss": 0.0004067758389282972,
      "step": 28380,
      "student_distance_mean": 1960.3499267578125,
      "teacher_distance_mean": 1.164377999305725,
      "weighted_auxiliary_loss": 0.0004067758389282972,
      "weighted_relation_loss": 0.0004067758389282972
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.55753040224509,
      "flow_matching_loss": 0.01372800413519144,
      "grad_norm": 0.09783967584371567,
      "learning_rate": 7.863203282494847e-07,
      "loss": 0.0141,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040281791298184543,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.84455947875976,
      "state_query_norm": 1426.6220947265624,
      "state_relation_loss": 0.00040281791298184543,
      "step": 28390,
      "student_distance_mean": 1965.5477905273438,
      "teacher_distance_mean": 1.168105137348175,
      "weighted_auxiliary_loss": 0.00040281791298184543,
      "weighted_relation_loss": 0.00040281791298184543
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.56688493919551,
      "flow_matching_loss": 0.016916204802691938,
      "grad_norm": 0.08327732980251312,
      "learning_rate": 7.766140061686522e-07,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041597581293899565,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.373603439331056,
      "state_query_norm": 1415.6935302734375,
      "state_relation_loss": 0.00041597581293899565,
      "step": 28400,
      "student_distance_mean": 1942.0359252929688,
      "teacher_distance_mean": 1.15260990858078,
      "weighted_auxiliary_loss": 0.00041597581293899565,
      "weighted_relation_loss": 0.00041597581293899565
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.57623947614593,
      "flow_matching_loss": 0.01681313281878829,
      "grad_norm": 0.09999007731676102,
      "learning_rate": 7.669674951432615e-07,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039197679434437304,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.56513671875,
      "state_query_norm": 1424.0432373046874,
      "state_relation_loss": 0.00039197679434437304,
      "step": 28410,
      "student_distance_mean": 1950.5643188476563,
      "teacher_distance_mean": 1.1583441138267516,
      "weighted_auxiliary_loss": 0.00039197679434437304,
      "weighted_relation_loss": 0.00039197679434437304
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.58559401309635,
      "flow_matching_loss": 0.014864231366664172,
      "grad_norm": 0.08137664943933487,
      "learning_rate": 7.573808068947363e-07,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003819255070993677,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.13538818359375,
      "state_query_norm": 1396.9963989257812,
      "state_relation_loss": 0.0003819255070993677,
      "step": 28420,
      "student_distance_mean": 1926.1128295898438,
      "teacher_distance_mean": 1.1437556266784668,
      "weighted_auxiliary_loss": 0.0003819255070993677,
      "weighted_relation_loss": 0.0003819255070993677
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.594948550046773,
      "flow_matching_loss": 0.01675354540348053,
      "grad_norm": 0.08094313740730286,
      "learning_rate": 7.478539530718087e-07,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003737760125659406,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.093099975585936,
      "state_query_norm": 1408.6225341796876,
      "state_relation_loss": 0.0003737760125659406,
      "step": 28430,
      "student_distance_mean": 1927.9130981445312,
      "teacher_distance_mean": 1.1461357355117798,
      "weighted_auxiliary_loss": 0.0003737760125659406,
      "weighted_relation_loss": 0.0003737760125659406
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.604303086997195,
      "flow_matching_loss": 0.01659687515348196,
      "grad_norm": 0.08270999044179916,
      "learning_rate": 7.383869452504965e-07,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039425802242476493,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.18554878234863,
      "state_query_norm": 1409.952587890625,
      "state_relation_loss": 0.00039425802242476493,
      "step": 28440,
      "student_distance_mean": 1932.6502563476563,
      "teacher_distance_mean": 1.1487747788429261,
      "weighted_auxiliary_loss": 0.00039425802242476493,
      "weighted_relation_loss": 0.00039425802242476493
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.613657623947613,
      "flow_matching_loss": 0.015248160157352687,
      "grad_norm": 0.07964707911014557,
      "learning_rate": 7.289797949341204e-07,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004015923972474411,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.571828079223636,
      "state_query_norm": 1417.57548828125,
      "state_relation_loss": 0.0004015923972474411,
      "step": 28450,
      "student_distance_mean": 1952.9793701171875,
      "teacher_distance_mean": 1.161604356765747,
      "weighted_auxiliary_loss": 0.0004015923972474411,
      "weighted_relation_loss": 0.0004015923972474411
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.623012160898035,
      "flow_matching_loss": 0.0142813959158957,
      "grad_norm": 0.09136270731687546,
      "learning_rate": 7.196325135532423e-07,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003907943260855973,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.476562118530275,
      "state_query_norm": 1419.6246215820313,
      "state_relation_loss": 0.0003907943260855973,
      "step": 28460,
      "student_distance_mean": 1946.8880004882812,
      "teacher_distance_mean": 1.1573665857315063,
      "weighted_auxiliary_loss": 0.0003907943260855973,
      "weighted_relation_loss": 0.0003907943260855973
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.632366697848457,
      "flow_matching_loss": 0.016738584823906423,
      "grad_norm": 0.08756737411022186,
      "learning_rate": 7.10345112465699e-07,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038258032873272896,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.369414520263675,
      "state_query_norm": 1415.9899169921875,
      "state_relation_loss": 0.00038258032873272896,
      "step": 28470,
      "student_distance_mean": 1940.5200317382812,
      "teacher_distance_mean": 1.1526268362998962,
      "weighted_auxiliary_loss": 0.00038258032873272896,
      "weighted_relation_loss": 0.00038258032873272896
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.641721234798876,
      "flow_matching_loss": 0.016293063946068286,
      "grad_norm": 0.08029946684837341,
      "learning_rate": 7.011176029565525e-07,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003812787588685751,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.13731155395508,
      "state_query_norm": 1403.8823608398438,
      "state_relation_loss": 0.0003812787588685751,
      "step": 28480,
      "student_distance_mean": 1929.8194580078125,
      "teacher_distance_mean": 1.1459205150604248,
      "weighted_auxiliary_loss": 0.0003812787588685751,
      "weighted_relation_loss": 0.0003812787588685751
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.651075771749298,
      "flow_matching_loss": 0.013637306541204453,
      "grad_norm": 0.08661315590143204,
      "learning_rate": 6.919499962381004e-07,
      "loss": 0.014,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039141787274274973,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.87191276550293,
      "state_query_norm": 1391.58701171875,
      "state_relation_loss": 0.00039141787274274973,
      "step": 28490,
      "student_distance_mean": 1911.9028198242188,
      "teacher_distance_mean": 1.1354985237121582,
      "weighted_auxiliary_loss": 0.00039141787274274973,
      "weighted_relation_loss": 0.00039141787274274973
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.66043030869972,
      "flow_matching_loss": 0.01625008387491107,
      "grad_norm": 0.09279558062553406,
      "learning_rate": 6.828423034498488e-07,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040429531945846976,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.353651428222655,
      "state_query_norm": 1409.7442260742187,
      "state_relation_loss": 0.00040429531945846976,
      "step": 28500,
      "student_distance_mean": 1936.5755981445313,
      "teacher_distance_mean": 1.1502776145935059,
      "weighted_auxiliary_loss": 0.00040429531945846976,
      "weighted_relation_loss": 0.00040429531945846976
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.669784845650142,
      "flow_matching_loss": 0.015411995258182288,
      "grad_norm": 0.07310859858989716,
      "learning_rate": 6.737945356585007e-07,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004372053488623351,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.07122116088867,
      "state_query_norm": 1432.8695190429687,
      "state_relation_loss": 0.0004372053488623351,
      "step": 28510,
      "student_distance_mean": 1973.700439453125,
      "teacher_distance_mean": 1.1734979629516602,
      "weighted_auxiliary_loss": 0.0004372053488623351,
      "weighted_relation_loss": 0.0004372053488623351
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.67913938260056,
      "flow_matching_loss": 0.015946599841117858,
      "grad_norm": 0.08796274662017822,
      "learning_rate": 6.648067038579508e-07,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037955912994220854,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.151606750488284,
      "state_query_norm": 1408.9255859375,
      "state_relation_loss": 0.00037955912994220854,
      "step": 28520,
      "student_distance_mean": 1932.4390747070313,
      "teacher_distance_mean": 1.148102045059204,
      "weighted_auxiliary_loss": 0.00037955912994220854,
      "weighted_relation_loss": 0.00037955912994220854
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.688493919550982,
      "flow_matching_loss": 0.015141044743359089,
      "grad_norm": 0.0744711235165596,
      "learning_rate": 6.558788189692578e-07,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038052008021622897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27902526855469,
      "state_query_norm": 1407.8248046875,
      "state_relation_loss": 0.00038052008021622897,
      "step": 28530,
      "student_distance_mean": 1936.2291870117188,
      "teacher_distance_mean": 1.1504671454429627,
      "weighted_auxiliary_loss": 0.00038052008021622897,
      "weighted_relation_loss": 0.00038052008021622897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.697848456501404,
      "flow_matching_loss": 0.016850371193140746,
      "grad_norm": 0.07093743979930878,
      "learning_rate": 6.470108918406492e-07,
      "loss": 0.0172,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003870200162054971,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.05831642150879,
      "state_query_norm": 1404.4397827148437,
      "state_relation_loss": 0.0003870200162054971,
      "step": 28540,
      "student_distance_mean": 1927.6616455078124,
      "teacher_distance_mean": 1.145806908607483,
      "weighted_auxiliary_loss": 0.0003870200162054971,
      "weighted_relation_loss": 0.0003870200162054971
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.707202993451823,
      "flow_matching_loss": 0.01569658387452364,
      "grad_norm": 0.10528483986854553,
      "learning_rate": 6.382029332474893e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038902817177586255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37168922424316,
      "state_query_norm": 1408.7508056640625,
      "state_relation_loss": 0.00038902817177586255,
      "step": 28550,
      "student_distance_mean": 1940.68447265625,
      "teacher_distance_mean": 1.1530395984649657,
      "weighted_auxiliary_loss": 0.00038902817177586255,
      "weighted_relation_loss": 0.00038902817177586255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.716557530402245,
      "flow_matching_loss": 0.016233462654054166,
      "grad_norm": 0.0809355080127716,
      "learning_rate": 6.294549538922778e-07,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003824673447525129,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.04044380187988,
      "state_query_norm": 1406.3129638671876,
      "state_relation_loss": 0.0003824673447525129,
      "step": 28560,
      "student_distance_mean": 1923.9694580078126,
      "teacher_distance_mean": 1.143404769897461,
      "weighted_auxiliary_loss": 0.0003824673447525129,
      "weighted_relation_loss": 0.0003824673447525129
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.725912067352667,
      "flow_matching_loss": 0.01569576719775796,
      "grad_norm": 0.08429328352212906,
      "learning_rate": 6.207669644046344e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042379388760309665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.80720863342285,
      "state_query_norm": 1395.1697387695312,
      "state_relation_loss": 0.00042379388760309665,
      "step": 28570,
      "student_distance_mean": 1910.1064331054688,
      "teacher_distance_mean": 1.1350555062294005,
      "weighted_auxiliary_loss": 0.00042379388760309665,
      "weighted_relation_loss": 0.00042379388760309665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.73526660430309,
      "flow_matching_loss": 0.015517752710729837,
      "grad_norm": 0.0713493749499321,
      "learning_rate": 6.121389753412865e-07,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003941879491321743,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.01315536499023,
      "state_query_norm": 1396.2926513671875,
      "state_relation_loss": 0.0003941879491321743,
      "step": 28580,
      "student_distance_mean": 1922.2300903320313,
      "teacher_distance_mean": 1.1424030900001525,
      "weighted_auxiliary_loss": 0.0003941879491321743,
      "weighted_relation_loss": 0.0003941879491321743
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.744621141253507,
      "flow_matching_loss": 0.0161115238443017,
      "grad_norm": 0.07973598688840866,
      "learning_rate": 6.035709971860592e-07,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036934686359018086,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.914450073242186,
      "state_query_norm": 1397.73115234375,
      "state_relation_loss": 0.00036934686359018086,
      "step": 28590,
      "student_distance_mean": 1920.76708984375,
      "teacher_distance_mean": 1.1409679889678954,
      "weighted_auxiliary_loss": 0.00036934686359018086,
      "weighted_relation_loss": 0.00036934686359018086
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.75397567820393,
      "flow_matching_loss": 0.01558753103017807,
      "grad_norm": 0.08785796910524368,
      "learning_rate": 5.950630403498469e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004202647105557844,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07208023071289,
      "state_query_norm": 1399.4448120117188,
      "state_relation_loss": 0.0004202647105557844,
      "step": 28600,
      "student_distance_mean": 1926.17685546875,
      "teacher_distance_mean": 1.1454961776733399,
      "weighted_auxiliary_loss": 0.0004202647105557844,
      "weighted_relation_loss": 0.0004202647105557844
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.76333021515435,
      "flow_matching_loss": 0.014590066950768232,
      "grad_norm": 0.07903730124235153,
      "learning_rate": 5.866151151706189e-07,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003848676715278998,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.62420845031738,
      "state_query_norm": 1419.976171875,
      "state_relation_loss": 0.0003848676715278998,
      "step": 28610,
      "student_distance_mean": 1954.8355102539062,
      "teacher_distance_mean": 1.160878801345825,
      "weighted_auxiliary_loss": 0.0003848676715278998,
      "weighted_relation_loss": 0.0003848676715278998
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.77268475210477,
      "flow_matching_loss": 0.01571591179817915,
      "grad_norm": 0.07457143813371658,
      "learning_rate": 5.782272319134086e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038901258667465297,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47765846252442,
      "state_query_norm": 1421.8903442382812,
      "state_relation_loss": 0.00038901258667465297,
      "step": 28620,
      "student_distance_mean": 1949.2516845703126,
      "teacher_distance_mean": 1.1585597038269042,
      "weighted_auxiliary_loss": 0.00038901258667465297,
      "weighted_relation_loss": 0.00038901258667465297
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.78203928905519,
      "flow_matching_loss": 0.016942930594086646,
      "grad_norm": 0.09988360106945038,
      "learning_rate": 5.698994007702796e-07,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004077517980476841,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6444034576416,
      "state_query_norm": 1416.5573364257812,
      "state_relation_loss": 0.0004077517980476841,
      "step": 28630,
      "student_distance_mean": 1956.362451171875,
      "teacher_distance_mean": 1.161939299106598,
      "weighted_auxiliary_loss": 0.0004077517980476841,
      "weighted_relation_loss": 0.0004077517980476841
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.791393826005613,
      "flow_matching_loss": 0.017032862827181815,
      "grad_norm": 0.08181865513324738,
      "learning_rate": 5.616316318603321e-07,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003835614479612559,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.122873306274414,
      "state_query_norm": 1405.5768798828126,
      "state_relation_loss": 0.0003835614479612559,
      "step": 28640,
      "student_distance_mean": 1927.2553344726562,
      "teacher_distance_mean": 1.1453044652938842,
      "weighted_auxiliary_loss": 0.0003835614479612559,
      "weighted_relation_loss": 0.0003835614479612559
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.800748362956035,
      "flow_matching_loss": 0.016032267827540636,
      "grad_norm": 0.08141129463911057,
      "learning_rate": 5.534239352296799e-07,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003925601311493665,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58827590942383,
      "state_query_norm": 1426.0019897460938,
      "state_relation_loss": 0.0003925601311493665,
      "step": 28650,
      "student_distance_mean": 1956.369775390625,
      "teacher_distance_mean": 1.1628029346466064,
      "weighted_auxiliary_loss": 0.0003925601311493665,
      "weighted_relation_loss": 0.0003925601311493665
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.810102899906454,
      "flow_matching_loss": 0.01567639298737049,
      "grad_norm": 0.08765114098787308,
      "learning_rate": 5.452763208514621e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003556883777491748,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.02808952331543,
      "state_query_norm": 1408.0047973632813,
      "state_relation_loss": 0.0003556883777491748,
      "step": 28660,
      "student_distance_mean": 1927.38916015625,
      "teacher_distance_mean": 1.1460144162178039,
      "weighted_auxiliary_loss": 0.0003556883777491748,
      "weighted_relation_loss": 0.0003556883777491748
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.819457436856876,
      "flow_matching_loss": 0.015258228313177824,
      "grad_norm": 0.10184884816408157,
      "learning_rate": 5.371887986257873e-07,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042460253462195396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.57360420227051,
      "state_query_norm": 1431.0756225585938,
      "state_relation_loss": 0.00042460253462195396,
      "step": 28670,
      "student_distance_mean": 1955.847216796875,
      "teacher_distance_mean": 1.1625343084335327,
      "weighted_auxiliary_loss": 0.00042460253462195396,
      "weighted_relation_loss": 0.00042460253462195396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.828811973807298,
      "flow_matching_loss": 0.01538267293944955,
      "grad_norm": 0.09596333652734756,
      "learning_rate": 5.291613783797611e-07,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003929564671125263,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25196800231934,
      "state_query_norm": 1401.8600708007812,
      "state_relation_loss": 0.0003929564671125263,
      "step": 28680,
      "student_distance_mean": 1934.1164672851562,
      "teacher_distance_mean": 1.1497724294662475,
      "weighted_auxiliary_loss": 0.0003929564671125263,
      "weighted_relation_loss": 0.0003929564671125263
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.838166510757716,
      "flow_matching_loss": 0.015899961907416583,
      "grad_norm": 0.10461805760860443,
      "learning_rate": 5.211940698674534e-07,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039059109694790096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.17248001098633,
      "state_query_norm": 1404.6608276367188,
      "state_relation_loss": 0.00039059109694790096,
      "step": 28690,
      "student_distance_mean": 1929.644287109375,
      "teacher_distance_mean": 1.1470002293586732,
      "weighted_auxiliary_loss": 0.00039059109694790096,
      "weighted_relation_loss": 0.00039059109694790096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.84752104770814,
      "flow_matching_loss": 0.01584630962461233,
      "grad_norm": 0.07710260897874832,
      "learning_rate": 5.132868827698978e-07,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003980926761869341,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.87334213256836,
      "state_query_norm": 1437.3223754882813,
      "state_relation_loss": 0.0003980926761869341,
      "step": 28700,
      "student_distance_mean": 1969.3040161132812,
      "teacher_distance_mean": 1.1706629872322083,
      "weighted_auxiliary_loss": 0.0003980926761869341,
      "weighted_relation_loss": 0.0003980926761869341
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.85687558465856,
      "flow_matching_loss": 0.016383982542902232,
      "grad_norm": 0.08283084630966187,
      "learning_rate": 5.054398266950755e-07,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038409885601140557,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.922163772583005,
      "state_query_norm": 1399.5568115234375,
      "state_relation_loss": 0.00038409885601140557,
      "step": 28710,
      "student_distance_mean": 1919.3209594726563,
      "teacher_distance_mean": 1.1403758645057678,
      "weighted_auxiliary_loss": 0.00038409885601140557,
      "weighted_relation_loss": 0.00038409885601140557
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.86623012160898,
      "flow_matching_loss": 0.017315188888460398,
      "grad_norm": 0.08477824181318283,
      "learning_rate": 4.976529111778872e-07,
      "loss": 0.0177,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003832194022834301,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.140428924560545,
      "state_query_norm": 1404.5346923828124,
      "state_relation_loss": 0.0003832194022834301,
      "step": 28720,
      "student_distance_mean": 1929.6483154296875,
      "teacher_distance_mean": 1.1465584635734558,
      "weighted_auxiliary_loss": 0.0003832194022834301,
      "weighted_relation_loss": 0.0003832194022834301
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.8755846585594,
      "flow_matching_loss": 0.015358358155936003,
      "grad_norm": 0.0901925191283226,
      "learning_rate": 4.899261456801862e-07,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036814592021983117,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.988613510131835,
      "state_query_norm": 1404.999072265625,
      "state_relation_loss": 0.00036814592021983117,
      "step": 28730,
      "student_distance_mean": 1925.6696166992188,
      "teacher_distance_mean": 1.1441932916641235,
      "weighted_auxiliary_loss": 0.00036814592021983117,
      "weighted_relation_loss": 0.00036814592021983117
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.884939195509823,
      "flow_matching_loss": 0.015744952019304037,
      "grad_norm": 0.08053003996610641,
      "learning_rate": 4.822595395907126e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000385352858575061,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.554941940307614,
      "state_query_norm": 1422.2919067382813,
      "state_relation_loss": 0.000385352858575061,
      "step": 28740,
      "student_distance_mean": 1954.8464965820312,
      "teacher_distance_mean": 1.1620085120201111,
      "weighted_auxiliary_loss": 0.000385352858575061,
      "weighted_relation_loss": 0.000385352858575061
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.894293732460245,
      "flow_matching_loss": 0.014994054939597845,
      "grad_norm": 0.0835825577378273,
      "learning_rate": 4.7465310222510886e-07,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003755169891519472,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15992698669434,
      "state_query_norm": 1400.2402954101562,
      "state_relation_loss": 0.0003755169891519472,
      "step": 28750,
      "student_distance_mean": 1933.1906127929688,
      "teacher_distance_mean": 1.1486234307289123,
      "weighted_auxiliary_loss": 0.0003755169891519472,
      "weighted_relation_loss": 0.0003755169891519472
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.903648269410663,
      "flow_matching_loss": 0.016000102367252113,
      "grad_norm": 0.08485066145658493,
      "learning_rate": 4.6710684282593175e-07,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039086781034711746,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.68427848815918,
      "state_query_norm": 1430.6237670898438,
      "state_relation_loss": 0.00039086781034711746,
      "step": 28760,
      "student_distance_mean": 1961.9851806640625,
      "teacher_distance_mean": 1.1652729392051697,
      "weighted_auxiliary_loss": 0.00039086781034711746,
      "weighted_relation_loss": 0.00039086781034711746
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.913002806361085,
      "flow_matching_loss": 0.015256279055029154,
      "grad_norm": 0.09838742017745972,
      "learning_rate": 4.5962077056257993e-07,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003873886045766994,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06264991760254,
      "state_query_norm": 1396.5555541992187,
      "state_relation_loss": 0.0003873886045766994,
      "step": 28770,
      "student_distance_mean": 1924.0064697265625,
      "teacher_distance_mean": 1.1440036296844482,
      "weighted_auxiliary_loss": 0.0003873886045766994,
      "weighted_relation_loss": 0.0003873886045766994
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.922357343311507,
      "flow_matching_loss": 0.014197308477014304,
      "grad_norm": 0.0817698985338211,
      "learning_rate": 4.521948945313492e-07,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000383378891274333,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.41914253234863,
      "state_query_norm": 1417.3457397460938,
      "state_relation_loss": 0.000383378891274333,
      "step": 28780,
      "student_distance_mean": 1947.297607421875,
      "teacher_distance_mean": 1.157459020614624,
      "weighted_auxiliary_loss": 0.000383378891274333,
      "weighted_relation_loss": 0.000383378891274333
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.931711880261926,
      "flow_matching_loss": 0.01619885051622987,
      "grad_norm": 0.08310261368751526,
      "learning_rate": 4.4482922375537195e-07,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038038138882257044,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58911285400391,
      "state_query_norm": 1422.61552734375,
      "state_relation_loss": 0.00038038138882257044,
      "step": 28790,
      "student_distance_mean": 1953.06416015625,
      "teacher_distance_mean": 1.159779417514801,
      "weighted_auxiliary_loss": 0.00038038138882257044,
      "weighted_relation_loss": 0.00038038138882257044
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.941066417212348,
      "flow_matching_loss": 0.015606733690947294,
      "grad_norm": 0.06561138480901718,
      "learning_rate": 4.375237671846333e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003859695163555443,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.22084083557129,
      "state_query_norm": 1410.9063720703125,
      "state_relation_loss": 0.0003859695163555443,
      "step": 28800,
      "student_distance_mean": 1934.900048828125,
      "teacher_distance_mean": 1.1500012159347535,
      "weighted_auxiliary_loss": 0.0003859695163555443,
      "weighted_relation_loss": 0.0003859695163555443
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.95042095416277,
      "flow_matching_loss": 0.014645727165043355,
      "grad_norm": 0.0821617916226387,
      "learning_rate": 4.302785336959547e-07,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003844686696538702,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.47020034790039,
      "state_query_norm": 1421.3292724609375,
      "state_relation_loss": 0.0003844686696538702,
      "step": 28810,
      "student_distance_mean": 1948.9431518554688,
      "teacher_distance_mean": 1.1571364164352418,
      "weighted_auxiliary_loss": 0.0003844686696538702,
      "weighted_relation_loss": 0.0003844686696538702
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.95977549111319,
      "flow_matching_loss": 0.015802676882594825,
      "grad_norm": 0.09845304489135742,
      "learning_rate": 4.2309353209297744e-07,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003811121394392103,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.18711967468262,
      "state_query_norm": 1411.62939453125,
      "state_relation_loss": 0.0003811121394392103,
      "step": 28820,
      "student_distance_mean": 1934.0697998046876,
      "teacher_distance_mean": 1.1488817930221558,
      "weighted_auxiliary_loss": 0.0003811121394392103,
      "weighted_relation_loss": 0.0003811121394392103
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.96913002806361,
      "flow_matching_loss": 0.014586922153830529,
      "grad_norm": 0.07909108698368073,
      "learning_rate": 4.159687711061566e-07,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039618907612748444,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.7042739868164,
      "state_query_norm": 1425.897802734375,
      "state_relation_loss": 0.00039618907612748444,
      "step": 28830,
      "student_distance_mean": 1959.68740234375,
      "teacher_distance_mean": 1.1634756684303285,
      "weighted_auxiliary_loss": 0.00039618907612748444,
      "weighted_relation_loss": 0.00039618907612748444
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.978484565014032,
      "flow_matching_loss": 0.01696497416123748,
      "grad_norm": 0.09418327361345291,
      "learning_rate": 4.0890425939275055e-07,
      "loss": 0.0173,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036069937923457476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.252392578125,
      "state_query_norm": 1413.065283203125,
      "state_relation_loss": 0.00036069937923457476,
      "step": 28840,
      "student_distance_mean": 1939.522265625,
      "teacher_distance_mean": 1.1522245407104492,
      "weighted_auxiliary_loss": 0.00036069937923457476,
      "weighted_relation_loss": 0.00036069937923457476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.987839101964454,
      "flow_matching_loss": 0.014639915060251951,
      "grad_norm": 0.097206711769104,
      "learning_rate": 4.0190000553679827e-07,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038541512622032316,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.09789543151855,
      "state_query_norm": 1402.3271362304688,
      "state_relation_loss": 0.00038541512622032316,
      "step": 28850,
      "student_distance_mean": 1927.6012573242188,
      "teacher_distance_mean": 1.1452414870262146,
      "weighted_auxiliary_loss": 0.00038541512622032316,
      "weighted_relation_loss": 0.00038541512622032316
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 26.997193638914872,
      "flow_matching_loss": 0.015749713126569986,
      "grad_norm": 0.08340485394001007,
      "learning_rate": 3.9495601804913627e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039771623269189147,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.76523857116699,
      "state_query_norm": 1422.6079833984375,
      "state_relation_loss": 0.00039771623269189147,
      "step": 28860,
      "student_distance_mean": 1960.5390380859376,
      "teacher_distance_mean": 1.164830219745636,
      "weighted_auxiliary_loss": 0.00039771623269189147,
      "weighted_relation_loss": 0.00039771623269189147
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.006548175865294,
      "flow_matching_loss": 0.015317971259355545,
      "grad_norm": 0.08144579827785492,
      "learning_rate": 3.880723053673652e-07,
      "loss": 0.0157,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.0003773542703129351,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 35.748012924194335,
      "state_query_norm": 1417.418505859375,
      "state_relation_loss": 0.0003773542703129351,
      "step": 28870,
      "student_distance_mean": 1960.3385498046875,
      "teacher_distance_mean": 1.164900577068329,
      "weighted_auxiliary_loss": 0.0003773542703129351,
      "weighted_relation_loss": 0.0003773542703129351
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.015902712815716,
      "flow_matching_loss": 0.015862756874412298,
      "grad_norm": 0.07787173241376877,
      "learning_rate": 3.812488758558386e-07,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004265617928467691,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4997371673584,
      "state_query_norm": 1408.5234619140624,
      "state_relation_loss": 0.0004265617928467691,
      "step": 28880,
      "student_distance_mean": 1941.62900390625,
      "teacher_distance_mean": 1.1535611391067504,
      "weighted_auxiliary_loss": 0.0004265617928467691,
      "weighted_relation_loss": 0.0004265617928467691
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.02525724976614,
      "flow_matching_loss": 0.015814115293323994,
      "grad_norm": 0.07863350957632065,
      "learning_rate": 3.744857378056743e-07,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004019038053229451,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 36.02875328063965,
      "state_query_norm": 1439.1567626953124,
      "state_relation_loss": 0.0004019038053229451,
      "step": 28890,
      "student_distance_mean": 1974.6495971679688,
      "teacher_distance_mean": 1.173201358318329,
      "weighted_auxiliary_loss": 0.0004019038053229451,
      "weighted_relation_loss": 0.0004019038053229451
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.034611786716557,
      "flow_matching_loss": 0.015694870986044406,
      "grad_norm": 0.08393540233373642,
      "learning_rate": 3.677828994347154e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00035785895597655325,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24945220947266,
      "state_query_norm": 1415.1135986328125,
      "state_relation_loss": 0.00035785895597655325,
      "step": 28900,
      "student_distance_mean": 1940.1289428710938,
      "teacher_distance_mean": 1.153021466732025,
      "weighted_auxiliary_loss": 0.00035785895597655325,
      "weighted_relation_loss": 0.00035785895597655325
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.04396632366698,
      "flow_matching_loss": 0.0159080246463418,
      "grad_norm": 0.08599148690700531,
      "learning_rate": 3.61140368887547e-07,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003895901376381516,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.365305709838864,
      "state_query_norm": 1408.481005859375,
      "state_relation_loss": 0.0003895901376381516,
      "step": 28910,
      "student_distance_mean": 1937.0467163085937,
      "teacher_distance_mean": 1.1506300687789917,
      "weighted_auxiliary_loss": 0.0003895901376381516,
      "weighted_relation_loss": 0.0003895901376381516
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.0533208606174,
      "flow_matching_loss": 0.015176508296281099,
      "grad_norm": 0.07447730749845505,
      "learning_rate": 3.545581542354681e-07,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038713410904165355,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.09123306274414,
      "state_query_norm": 1401.8107055664063,
      "state_relation_loss": 0.00038713410904165355,
      "step": 28920,
      "student_distance_mean": 1924.871240234375,
      "teacher_distance_mean": 1.1439804911613465,
      "weighted_auxiliary_loss": 0.00038713410904165355,
      "weighted_relation_loss": 0.00038713410904165355
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.06267539756782,
      "flow_matching_loss": 0.015560239274054765,
      "grad_norm": 0.09475462883710861,
      "learning_rate": 3.480362634764922e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003954724525101483,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.94711532592773,
      "state_query_norm": 1430.5022827148437,
      "state_relation_loss": 0.0003954724525101483,
      "step": 28930,
      "student_distance_mean": 1973.4843017578125,
      "teacher_distance_mean": 1.1729514718055725,
      "weighted_auxiliary_loss": 0.0003954724525101483,
      "weighted_relation_loss": 0.0003954724525101483
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.07202993451824,
      "flow_matching_loss": 0.014811593387275935,
      "grad_norm": 0.10058114677667618,
      "learning_rate": 3.4157470453533014e-07,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036692833236884326,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.947016906738284,
      "state_query_norm": 1402.1436645507813,
      "state_relation_loss": 0.00036692833236884326,
      "step": 28940,
      "student_distance_mean": 1924.2576782226563,
      "teacher_distance_mean": 1.143227744102478,
      "weighted_auxiliary_loss": 0.00036692833236884326,
      "weighted_relation_loss": 0.00036692833236884326
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.081384471468663,
      "flow_matching_loss": 0.016227549593895673,
      "grad_norm": 0.08131242543458939,
      "learning_rate": 3.3517348526339034e-07,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003741894179256633,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.436589050292966,
      "state_query_norm": 1425.5242309570312,
      "state_relation_loss": 0.0003741894179256633,
      "step": 28950,
      "student_distance_mean": 1948.3562866210937,
      "teacher_distance_mean": 1.1584801912307738,
      "weighted_auxiliary_loss": 0.0003741894179256633,
      "weighted_relation_loss": 0.0003741894179256633
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.09073900841908,
      "flow_matching_loss": 0.016575527470558883,
      "grad_norm": 0.09791603684425354,
      "learning_rate": 3.288326134387454e-07,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003653318708529696,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.88298377990723,
      "state_query_norm": 1391.9299926757812,
      "state_relation_loss": 0.0003653318708529696,
      "step": 28960,
      "student_distance_mean": 1915.6684326171876,
      "teacher_distance_mean": 1.1381760478019713,
      "weighted_auxiliary_loss": 0.0003653318708529696,
      "weighted_relation_loss": 0.0003653318708529696
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.100093545369504,
      "flow_matching_loss": 0.01655788263306022,
      "grad_norm": 0.09779474139213562,
      "learning_rate": 3.225520967661655e-07,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039673847495578227,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07965354919433,
      "state_query_norm": 1404.848095703125,
      "state_relation_loss": 0.00039673847495578227,
      "step": 28970,
      "student_distance_mean": 1925.6824951171875,
      "teacher_distance_mean": 1.1449839115142821,
      "weighted_auxiliary_loss": 0.00039673847495578227,
      "weighted_relation_loss": 0.00039673847495578227
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.109448082319926,
      "flow_matching_loss": 0.014565544482320548,
      "grad_norm": 0.08064503222703934,
      "learning_rate": 3.163319428770628e-07,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003665195661596954,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37012825012207,
      "state_query_norm": 1417.2169067382813,
      "state_relation_loss": 0.0003665195661596954,
      "step": 28980,
      "student_distance_mean": 1943.9035034179688,
      "teacher_distance_mean": 1.1549969553947448,
      "weighted_auxiliary_loss": 0.0003665195661596954,
      "weighted_relation_loss": 0.0003665195661596954
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.118802619270348,
      "flow_matching_loss": 0.016604498121887447,
      "grad_norm": 0.09859494864940643,
      "learning_rate": 3.1017215932951374e-07,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037948024983052163,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36889381408692,
      "state_query_norm": 1413.2686279296875,
      "state_relation_loss": 0.00037948024983052163,
      "step": 28990,
      "student_distance_mean": 1942.6713012695313,
      "teacher_distance_mean": 1.1547728061676026,
      "weighted_auxiliary_loss": 0.00037948024983052163,
      "weighted_relation_loss": 0.00037948024983052163
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.128157156220766,
      "flow_matching_loss": 0.016305378545075654,
      "grad_norm": 0.08750079572200775,
      "learning_rate": 3.040727536082366e-07,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040967433305922897,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20934944152832,
      "state_query_norm": 1402.3764038085938,
      "state_relation_loss": 0.00040967433305922897,
      "step": 29000,
      "student_distance_mean": 1931.399658203125,
      "teacher_distance_mean": 1.148435604572296,
      "weighted_auxiliary_loss": 0.00040967433305922897,
      "weighted_relation_loss": 0.00040967433305922897
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.137511693171188,
      "flow_matching_loss": 0.01396202864125371,
      "grad_norm": 0.09085171669721603,
      "learning_rate": 2.980337331245864e-07,
      "loss": 0.0143,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003650539671070874,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.168646240234374,
      "state_query_norm": 1401.8554443359376,
      "state_relation_loss": 0.0003650539671070874,
      "step": 29010,
      "student_distance_mean": 1932.8241821289062,
      "teacher_distance_mean": 1.1493211030960082,
      "weighted_auxiliary_loss": 0.0003650539671070874,
      "weighted_relation_loss": 0.0003650539671070874
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.14686623012161,
      "flow_matching_loss": 0.01556712929159403,
      "grad_norm": 0.10565997660160065,
      "learning_rate": 2.9205510521653213e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004050977557199076,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.36201057434082,
      "state_query_norm": 1419.5301635742187,
      "state_relation_loss": 0.0004050977557199076,
      "step": 29020,
      "student_distance_mean": 1943.5028442382813,
      "teacher_distance_mean": 1.155109965801239,
      "weighted_auxiliary_loss": 0.0004050977557199076,
      "weighted_relation_loss": 0.0004050977557199076
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.15622076707203,
      "flow_matching_loss": 0.015668288711458443,
      "grad_norm": 0.07329157739877701,
      "learning_rate": 2.86136877148685e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037619606882799417,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.358965682983396,
      "state_query_norm": 1419.9341430664062,
      "state_relation_loss": 0.00037619606882799417,
      "step": 29030,
      "student_distance_mean": 1943.473046875,
      "teacher_distance_mean": 1.155000627040863,
      "weighted_auxiliary_loss": 0.00037619606882799417,
      "weighted_relation_loss": 0.00037619606882799417
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.16557530402245,
      "flow_matching_loss": 0.015025309845805168,
      "grad_norm": 0.06383353471755981,
      "learning_rate": 2.8027905611223704e-07,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003886763850459829,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.35455322265625,
      "state_query_norm": 1423.273486328125,
      "state_relation_loss": 0.0003886763850459829,
      "step": 29040,
      "student_distance_mean": 1945.8796752929688,
      "teacher_distance_mean": 1.156461477279663,
      "weighted_auxiliary_loss": 0.0003886763850459829,
      "weighted_relation_loss": 0.0003886763850459829
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.174929840972872,
      "flow_matching_loss": 0.015297714527696371,
      "grad_norm": 0.08527637273073196,
      "learning_rate": 2.7448164922500573e-07,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037496915902011094,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.358788299560544,
      "state_query_norm": 1410.4552734375,
      "state_relation_loss": 0.00037496915902011094,
      "step": 29050,
      "student_distance_mean": 1942.04541015625,
      "teacher_distance_mean": 1.1542376279830933,
      "weighted_auxiliary_loss": 0.00037496915902011094,
      "weighted_relation_loss": 0.00037496915902011094
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.184284377923294,
      "flow_matching_loss": 0.01655405927449465,
      "grad_norm": 0.08627606928348541,
      "learning_rate": 2.687446635313784e-07,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003640780778368935,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.856530380249026,
      "state_query_norm": 1393.2116821289062,
      "state_relation_loss": 0.0003640780778368935,
      "step": 29060,
      "student_distance_mean": 1916.8419799804688,
      "teacher_distance_mean": 1.1391499280929565,
      "weighted_auxiliary_loss": 0.0003640780778368935,
      "weighted_relation_loss": 0.0003640780778368935
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.193638914873713,
      "flow_matching_loss": 0.01636608811095357,
      "grad_norm": 0.0814288854598999,
      "learning_rate": 2.630681060023343e-07,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038016719045117495,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.84255523681641,
      "state_query_norm": 1390.06826171875,
      "state_relation_loss": 0.00038016719045117495,
      "step": 29070,
      "student_distance_mean": 1912.5907470703125,
      "teacher_distance_mean": 1.1363131403923035,
      "weighted_auxiliary_loss": 0.00038016719045117495,
      "weighted_relation_loss": 0.00038016719045117495
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.202993451824135,
      "flow_matching_loss": 0.01627672677859664,
      "grad_norm": 0.07054807990789413,
      "learning_rate": 2.5745198353542833e-07,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004174610337940976,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.04682960510254,
      "state_query_norm": 1398.2603149414062,
      "state_relation_loss": 0.0004174610337940976,
      "step": 29080,
      "student_distance_mean": 1923.4217651367187,
      "teacher_distance_mean": 1.1439104795455932,
      "weighted_auxiliary_loss": 0.0004174610337940976,
      "weighted_relation_loss": 0.0004174610337940976
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.212347988774557,
      "flow_matching_loss": 0.013051029108464719,
      "grad_norm": 0.07203348726034164,
      "learning_rate": 2.518963029547794e-07,
      "loss": 0.0134,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036979505093768237,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.99363059997559,
      "state_query_norm": 1404.0790893554688,
      "state_relation_loss": 0.00036979505093768237,
      "step": 29090,
      "student_distance_mean": 1922.115283203125,
      "teacher_distance_mean": 1.1423454761505127,
      "weighted_auxiliary_loss": 0.00036979505093768237,
      "weighted_relation_loss": 0.00036979505093768237
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.221702525724975,
      "flow_matching_loss": 0.014040231890976428,
      "grad_norm": 0.06801053136587143,
      "learning_rate": 2.464010710110598e-07,
      "loss": 0.0145,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004140358563745394,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.698720169067386,
      "state_query_norm": 1422.5691528320312,
      "state_relation_loss": 0.0004140358563745394,
      "step": 29100,
      "student_distance_mean": 1956.2042724609375,
      "teacher_distance_mean": 1.1622146248817444,
      "weighted_auxiliary_loss": 0.0004140358563745394,
      "weighted_relation_loss": 0.0004140358563745394
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.231057062675397,
      "flow_matching_loss": 0.015006784163415433,
      "grad_norm": 0.10015171021223068,
      "learning_rate": 2.4096629438150054e-07,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003962478367611766,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.19308319091797,
      "state_query_norm": 1419.9421020507812,
      "state_relation_loss": 0.0003962478367611766,
      "step": 29110,
      "student_distance_mean": 1935.9342041015625,
      "teacher_distance_mean": 1.150728976726532,
      "weighted_auxiliary_loss": 0.0003962478367611766,
      "weighted_relation_loss": 0.0003962478367611766
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.24041159962582,
      "flow_matching_loss": 0.016477001830935478,
      "grad_norm": 0.07346776127815247,
      "learning_rate": 2.3559197966985802e-07,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003836536983726546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.44857177734375,
      "state_query_norm": 1418.468310546875,
      "state_relation_loss": 0.0003836536983726546,
      "step": 29120,
      "student_distance_mean": 1948.5753662109375,
      "teacher_distance_mean": 1.1579856634140016,
      "weighted_auxiliary_loss": 0.0003836536983726546,
      "weighted_relation_loss": 0.0003836536983726546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.24976613657624,
      "flow_matching_loss": 0.014763833396136761,
      "grad_norm": 0.10940803587436676,
      "learning_rate": 2.3027813340644188e-07,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039806413988117126,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20532379150391,
      "state_query_norm": 1417.308349609375,
      "state_relation_loss": 0.00039806413988117126,
      "step": 29130,
      "student_distance_mean": 1937.80947265625,
      "teacher_distance_mean": 1.1512670516967773,
      "weighted_auxiliary_loss": 0.00039806413988117126,
      "weighted_relation_loss": 0.00039806413988117126
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.25912067352666,
      "flow_matching_loss": 0.01646987712010741,
      "grad_norm": 0.09012328833341599,
      "learning_rate": 2.2502476204807055e-07,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003865612088702619,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.4805419921875,
      "state_query_norm": 1414.3535522460938,
      "state_relation_loss": 0.0003865612088702619,
      "step": 29140,
      "student_distance_mean": 1948.2284545898438,
      "teacher_distance_mean": 1.1574927806854247,
      "weighted_auxiliary_loss": 0.0003865612088702619,
      "weighted_relation_loss": 0.0003865612088702619
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.26847521047708,
      "flow_matching_loss": 0.014992549270391464,
      "grad_norm": 0.09062456339597702,
      "learning_rate": 2.1983187197808786e-07,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039205507491715255,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.25361518859863,
      "state_query_norm": 1414.5496215820312,
      "state_relation_loss": 0.00039205507491715255,
      "step": 29150,
      "student_distance_mean": 1935.82587890625,
      "teacher_distance_mean": 1.1507283568382263,
      "weighted_auxiliary_loss": 0.00039205507491715255,
      "weighted_relation_loss": 0.00039205507491715255
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.277829747427504,
      "flow_matching_loss": 0.014255167543888092,
      "grad_norm": 0.07546941190958023,
      "learning_rate": 2.1469946950634644e-07,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004117223201319575,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.50733489990235,
      "state_query_norm": 1411.5995727539062,
      "state_relation_loss": 0.0004117223201319575,
      "step": 29160,
      "student_distance_mean": 1944.22548828125,
      "teacher_distance_mean": 1.1549129128456115,
      "weighted_auxiliary_loss": 0.0004117223201319575,
      "weighted_relation_loss": 0.0004117223201319575
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.287184284377922,
      "flow_matching_loss": 0.016720253229141235,
      "grad_norm": 0.09301359951496124,
      "learning_rate": 2.096275608691911e-07,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003977796499384567,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29503898620605,
      "state_query_norm": 1418.4390502929687,
      "state_relation_loss": 0.0003977796499384567,
      "step": 29170,
      "student_distance_mean": 1941.699951171875,
      "teacher_distance_mean": 1.154080832004547,
      "weighted_auxiliary_loss": 0.0003977796499384567,
      "weighted_relation_loss": 0.0003977796499384567
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.296538821328344,
      "flow_matching_loss": 0.014119393937289716,
      "grad_norm": 0.07662629336118698,
      "learning_rate": 2.046161522294754e-07,
      "loss": 0.0145,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038738481234759094,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.840890884399414,
      "state_query_norm": 1396.0358642578126,
      "state_relation_loss": 0.00038738481234759094,
      "step": 29180,
      "student_distance_mean": 1916.4965209960938,
      "teacher_distance_mean": 1.1390376448631288,
      "weighted_auxiliary_loss": 0.00038738481234759094,
      "weighted_relation_loss": 0.00038738481234759094
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.305893358278766,
      "flow_matching_loss": 0.014335647970438004,
      "grad_norm": 0.10506031662225723,
      "learning_rate": 1.9966524967653944e-07,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004105216939933598,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.27756805419922,
      "state_query_norm": 1406.5977172851562,
      "state_relation_loss": 0.0004105216939933598,
      "step": 29190,
      "student_distance_mean": 1934.5449829101562,
      "teacher_distance_mean": 1.150023138523102,
      "weighted_auxiliary_loss": 0.0004105216939933598,
      "weighted_relation_loss": 0.0004105216939933598
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.315247895229184,
      "flow_matching_loss": 0.01606588698923588,
      "grad_norm": 0.08056038618087769,
      "learning_rate": 1.9477485922618222e-07,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003785332984989509,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.06511344909668,
      "state_query_norm": 1406.3478393554688,
      "state_relation_loss": 0.0003785332984989509,
      "step": 29200,
      "student_distance_mean": 1931.2878051757812,
      "teacher_distance_mean": 1.1479801774024962,
      "weighted_auxiliary_loss": 0.0003785332984989509,
      "weighted_relation_loss": 0.0003785332984989509
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.324602432179606,
      "flow_matching_loss": 0.015603089425712823,
      "grad_norm": 0.07506099343299866,
      "learning_rate": 1.899449868207004e-07,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039118029235396534,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37931823730469,
      "state_query_norm": 1409.3444213867188,
      "state_relation_loss": 0.00039118029235396534,
      "step": 29210,
      "student_distance_mean": 1940.56337890625,
      "teacher_distance_mean": 1.1526376843452453,
      "weighted_auxiliary_loss": 0.00039118029235396534,
      "weighted_relation_loss": 0.00039118029235396534
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.33395696913003,
      "flow_matching_loss": 0.014704994391649961,
      "grad_norm": 0.07178153097629547,
      "learning_rate": 1.851756383288439e-07,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003673320898087695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.886421585083006,
      "state_query_norm": 1392.8503662109374,
      "state_relation_loss": 0.0003673320898087695,
      "step": 29220,
      "student_distance_mean": 1914.3392333984375,
      "teacher_distance_mean": 1.1378850936889648,
      "weighted_auxiliary_loss": 0.0003673320898087695,
      "weighted_relation_loss": 0.0003673320898087695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.34331150608045,
      "flow_matching_loss": 0.014901145081967115,
      "grad_norm": 0.07369785010814667,
      "learning_rate": 1.8046681954581035e-07,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003769510396523401,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.383404922485354,
      "state_query_norm": 1415.704150390625,
      "state_relation_loss": 0.0003769510396523401,
      "step": 29230,
      "student_distance_mean": 1945.644775390625,
      "teacher_distance_mean": 1.155687153339386,
      "weighted_auxiliary_loss": 0.0003769510396523401,
      "weighted_relation_loss": 0.0003769510396523401
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.35266604303087,
      "flow_matching_loss": 0.014700848888605834,
      "grad_norm": 0.08848576992750168,
      "learning_rate": 1.7581853619327294e-07,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003812467533862218,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1187931060791,
      "state_query_norm": 1406.2850708007813,
      "state_relation_loss": 0.0003812467533862218,
      "step": 29240,
      "student_distance_mean": 1931.1763061523438,
      "teacher_distance_mean": 1.1487002849578858,
      "weighted_auxiliary_loss": 0.0003812467533862218,
      "weighted_relation_loss": 0.0003812467533862218
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.36202057998129,
      "flow_matching_loss": 0.014446322247385979,
      "grad_norm": 0.07647345215082169,
      "learning_rate": 1.7123079391932472e-07,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004058642138261348,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.456242370605466,
      "state_query_norm": 1417.1063110351563,
      "state_relation_loss": 0.0004058642138261348,
      "step": 29250,
      "student_distance_mean": 1946.5991821289062,
      "teacher_distance_mean": 1.1568435072898864,
      "weighted_auxiliary_loss": 0.0004058642138261348,
      "weighted_relation_loss": 0.0004058642138261348
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.371375116931713,
      "flow_matching_loss": 0.014733001496642828,
      "grad_norm": 0.07255487143993378,
      "learning_rate": 1.6670359829850657e-07,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038796023873146626,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.74276542663574,
      "state_query_norm": 1420.450634765625,
      "state_relation_loss": 0.00038796023873146626,
      "step": 29260,
      "student_distance_mean": 1958.9972290039063,
      "teacher_distance_mean": 1.1646906733512878,
      "weighted_auxiliary_loss": 0.00038796023873146626,
      "weighted_relation_loss": 0.00038796023873146626
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.38072965388213,
      "flow_matching_loss": 0.01449509123340249,
      "grad_norm": 0.08680326491594315,
      "learning_rate": 1.6223695483179037e-07,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003779026214033365,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.079284286499025,
      "state_query_norm": 1403.6929321289062,
      "state_relation_loss": 0.0003779026214033365,
      "step": 29270,
      "student_distance_mean": 1924.5080322265626,
      "teacher_distance_mean": 1.1435180187225342,
      "weighted_auxiliary_loss": 0.0003779026214033365,
      "weighted_relation_loss": 0.0003779026214033365
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.390084190832553,
      "flow_matching_loss": 0.016351488418877125,
      "grad_norm": 0.07763632386922836,
      "learning_rate": 1.5783086894656795e-07,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000403206457849592,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.71602783203125,
      "state_query_norm": 1417.8966430664063,
      "state_relation_loss": 0.000403206457849592,
      "step": 29280,
      "student_distance_mean": 1956.7160278320312,
      "teacher_distance_mean": 1.1628939509391785,
      "weighted_auxiliary_loss": 0.000403206457849592,
      "weighted_relation_loss": 0.000403206457849592
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.399438727782975,
      "flow_matching_loss": 0.015801511332392694,
      "grad_norm": 0.09664175659418106,
      "learning_rate": 1.5348534599665121e-07,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00042353475582785905,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64521446228027,
      "state_query_norm": 1420.6027099609375,
      "state_relation_loss": 0.00042353475582785905,
      "step": 29290,
      "student_distance_mean": 1955.9112060546875,
      "teacher_distance_mean": 1.162707018852234,
      "weighted_auxiliary_loss": 0.00042353475582785905,
      "weighted_relation_loss": 0.00042353475582785905
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.408793264733397,
      "flow_matching_loss": 0.014330290351063013,
      "grad_norm": 0.09131162613630295,
      "learning_rate": 1.4920039126225527e-07,
      "loss": 0.0147,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038162257696967573,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.410797119140625,
      "state_query_norm": 1412.6991088867187,
      "state_relation_loss": 0.00038162257696967573,
      "step": 29300,
      "student_distance_mean": 1943.7549194335938,
      "teacher_distance_mean": 1.155170977115631,
      "weighted_auxiliary_loss": 0.00038162257696967573,
      "weighted_relation_loss": 0.00038162257696967573
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.418147801683816,
      "flow_matching_loss": 0.015697160735726357,
      "grad_norm": 0.06745237112045288,
      "learning_rate": 1.449760099500097e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004326217487687245,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20510482788086,
      "state_query_norm": 1407.6287353515625,
      "state_relation_loss": 0.0004326217487687245,
      "step": 29310,
      "student_distance_mean": 1932.4762329101563,
      "teacher_distance_mean": 1.1491796851158143,
      "weighted_auxiliary_loss": 0.0004326217487687245,
      "weighted_relation_loss": 0.0004326217487687245
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.427502338634238,
      "flow_matching_loss": 0.01573757138103247,
      "grad_norm": 0.06623747199773788,
      "learning_rate": 1.4081220719293075e-07,
      "loss": 0.0161,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003897379152476788,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.20051536560059,
      "state_query_norm": 1407.737939453125,
      "state_relation_loss": 0.0003897379152476788,
      "step": 29320,
      "student_distance_mean": 1934.0556518554688,
      "teacher_distance_mean": 1.1500903129577638,
      "weighted_auxiliary_loss": 0.0003897379152476788,
      "weighted_relation_loss": 0.0003897379152476788
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.43685687558466,
      "flow_matching_loss": 0.013849343918263913,
      "grad_norm": 0.06938495486974716,
      "learning_rate": 1.3670898805043797e-07,
      "loss": 0.0142,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003835173003608361,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.24748611450195,
      "state_query_norm": 1409.3615844726562,
      "state_relation_loss": 0.0003835173003608361,
      "step": 29330,
      "student_distance_mean": 1937.484130859375,
      "teacher_distance_mean": 1.1522485375404359,
      "weighted_auxiliary_loss": 0.0003835173003608361,
      "weighted_relation_loss": 0.0003835173003608361
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.446211412535078,
      "flow_matching_loss": 0.015205207094550133,
      "grad_norm": 0.06401821225881577,
      "learning_rate": 1.326663575083209e-07,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039029517502058296,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.675062561035155,
      "state_query_norm": 1423.91572265625,
      "state_relation_loss": 0.00039029517502058296,
      "step": 29340,
      "student_distance_mean": 1957.20341796875,
      "teacher_distance_mean": 1.1627551198005677,
      "weighted_auxiliary_loss": 0.00039029517502058296,
      "weighted_relation_loss": 0.00039029517502058296
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.4555659494855,
      "flow_matching_loss": 0.01695034056901932,
      "grad_norm": 0.06634949892759323,
      "learning_rate": 1.2868432047876688e-07,
      "loss": 0.0174,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004026401700684801,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.26945037841797,
      "state_query_norm": 1410.988427734375,
      "state_relation_loss": 0.0004026401700684801,
      "step": 29350,
      "student_distance_mean": 1937.9390747070313,
      "teacher_distance_mean": 1.151087236404419,
      "weighted_auxiliary_loss": 0.0004026401700684801,
      "weighted_relation_loss": 0.0004026401700684801
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.464920486435922,
      "flow_matching_loss": 0.01623888881877065,
      "grad_norm": 0.08894316852092743,
      "learning_rate": 1.2476288180032213e-07,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041510613809805366,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.488501739501956,
      "state_query_norm": 1408.763427734375,
      "state_relation_loss": 0.00041510613809805366,
      "step": 29360,
      "student_distance_mean": 1943.761181640625,
      "teacher_distance_mean": 1.1555665016174317,
      "weighted_auxiliary_loss": 0.00041510613809805366,
      "weighted_relation_loss": 0.00041510613809805366
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.474275023386344,
      "flow_matching_loss": 0.015833403170108794,
      "grad_norm": 0.09042549878358841,
      "learning_rate": 1.209020462379029e-07,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004171493405010551,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33241081237793,
      "state_query_norm": 1406.9225708007812,
      "state_relation_loss": 0.0004171493405010551,
      "step": 29370,
      "student_distance_mean": 1937.8718994140625,
      "teacher_distance_mean": 1.1522343635559082,
      "weighted_auxiliary_loss": 0.0004171493405010551,
      "weighted_relation_loss": 0.0004171493405010551
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.483629560336762,
      "flow_matching_loss": 0.015554903913289309,
      "grad_norm": 0.09338303655385971,
      "learning_rate": 1.1710181848278435e-07,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003821924328804016,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.70346221923828,
      "state_query_norm": 1390.8605590820312,
      "state_relation_loss": 0.0003821924328804016,
      "step": 29380,
      "student_distance_mean": 1906.6404907226563,
      "teacher_distance_mean": 1.1335509538650512,
      "weighted_auxiliary_loss": 0.0003821924328804016,
      "weighted_relation_loss": 0.0003821924328804016
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.492984097287184,
      "flow_matching_loss": 0.014637629501521587,
      "grad_norm": 0.09809450805187225,
      "learning_rate": 1.133622031526116e-07,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041743548645172267,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38406105041504,
      "state_query_norm": 1412.1923461914062,
      "state_relation_loss": 0.00041743548645172267,
      "step": 29390,
      "student_distance_mean": 1937.951220703125,
      "teacher_distance_mean": 1.1518790960311889,
      "weighted_auxiliary_loss": 0.00041743548645172267,
      "weighted_relation_loss": 0.00041743548645172267
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.502338634237606,
      "flow_matching_loss": 0.014267647918313742,
      "grad_norm": 0.11201643198728561,
      "learning_rate": 1.0968320479136095e-07,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003808111738180742,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29662551879883,
      "state_query_norm": 1404.0051879882812,
      "state_relation_loss": 0.0003808111738180742,
      "step": 29400,
      "student_distance_mean": 1937.9082275390624,
      "teacher_distance_mean": 1.1521227717399598,
      "weighted_auxiliary_loss": 0.0003808111738180742,
      "weighted_relation_loss": 0.0003808111738180742
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.511693171188025,
      "flow_matching_loss": 0.014903498440980911,
      "grad_norm": 0.07999534159898758,
      "learning_rate": 1.0606482786936767e-07,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003956253553042188,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.75832824707031,
      "state_query_norm": 1426.252099609375,
      "state_relation_loss": 0.0003956253553042188,
      "step": 29410,
      "student_distance_mean": 1961.04375,
      "teacher_distance_mean": 1.1651583194732666,
      "weighted_auxiliary_loss": 0.0003956253553042188,
      "weighted_relation_loss": 0.0003956253553042188
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.521047708138447,
      "flow_matching_loss": 0.01658365884795785,
      "grad_norm": 0.06904720515012741,
      "learning_rate": 1.0250707678329808e-07,
      "loss": 0.017,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038388636021409185,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.213107681274415,
      "state_query_norm": 1407.6837036132813,
      "state_relation_loss": 0.00038388636021409185,
      "step": 29420,
      "student_distance_mean": 1933.4060791015625,
      "teacher_distance_mean": 1.1486190915107728,
      "weighted_auxiliary_loss": 0.00038388636021409185,
      "weighted_relation_loss": 0.00038388636021409185
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.53040224508887,
      "flow_matching_loss": 0.01600148342549801,
      "grad_norm": 0.1149967759847641,
      "learning_rate": 9.900995585615525e-08,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004049189039506018,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.61773147583008,
      "state_query_norm": 1422.5850219726562,
      "state_relation_loss": 0.0004049189039506018,
      "step": 29430,
      "student_distance_mean": 1954.8789428710938,
      "teacher_distance_mean": 1.1624136567115784,
      "weighted_auxiliary_loss": 0.0004049189039506018,
      "weighted_relation_loss": 0.0004049189039506018
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.53975678203929,
      "flow_matching_loss": 0.01608111774548888,
      "grad_norm": 0.08809304237365723,
      "learning_rate": 9.55734693372734e-08,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003720512555446476,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.772835159301756,
      "state_query_norm": 1396.51103515625,
      "state_relation_loss": 0.0003720512555446476,
      "step": 29440,
      "student_distance_mean": 1914.8719970703125,
      "teacher_distance_mean": 1.1382292747497558,
      "weighted_auxiliary_loss": 0.0003720512555446476,
      "weighted_relation_loss": 0.0003720512555446476
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.54911131898971,
      "flow_matching_loss": 0.014921459183096886,
      "grad_norm": 0.09484155476093292,
      "learning_rate": 9.219762140231236e-08,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039175138517748563,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38318290710449,
      "state_query_norm": 1413.4846435546874,
      "state_relation_loss": 0.00039175138517748563,
      "step": 29450,
      "student_distance_mean": 1942.8211181640625,
      "teacher_distance_mean": 1.154581618309021,
      "weighted_auxiliary_loss": 0.00039175138517748563,
      "weighted_relation_loss": 0.00039175138517748563
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.55846585594013,
      "flow_matching_loss": 0.015072388853877784,
      "grad_norm": 0.07785835862159729,
      "learning_rate": 8.888241615322978e-08,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038439863128587605,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.1704288482666,
      "state_query_norm": 1408.3475219726563,
      "state_relation_loss": 0.00038439863128587605,
      "step": 29460,
      "student_distance_mean": 1931.630712890625,
      "teacher_distance_mean": 1.1482651114463807,
      "weighted_auxiliary_loss": 0.00038439863128587605,
      "weighted_relation_loss": 0.00038439863128587605
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.567820392890553,
      "flow_matching_loss": 0.015950234606862068,
      "grad_norm": 0.07536455988883972,
      "learning_rate": 8.562785761833114e-08,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040754820220172404,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.18483352661133,
      "state_query_norm": 1405.6544677734375,
      "state_relation_loss": 0.00040754820220172404,
      "step": 29470,
      "student_distance_mean": 1930.0090454101562,
      "teacher_distance_mean": 1.1475746989250184,
      "weighted_auxiliary_loss": 0.00040754820220172404,
      "weighted_relation_loss": 0.00040754820220172404
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.57717492984097,
      "flow_matching_loss": 0.01362198470160365,
      "grad_norm": 0.09223993122577667,
      "learning_rate": 8.243394975219753e-08,
      "loss": 0.014,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000392863096203655,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.29964179992676,
      "state_query_norm": 1416.6240234375,
      "state_relation_loss": 0.000392863096203655,
      "step": 29480,
      "student_distance_mean": 1941.4481811523438,
      "teacher_distance_mean": 1.154717171192169,
      "weighted_auxiliary_loss": 0.000392863096203655,
      "weighted_relation_loss": 0.000392863096203655
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.586529466791394,
      "flow_matching_loss": 0.015801230911165476,
      "grad_norm": 0.09079215675592422,
      "learning_rate": 7.930069643573568e-08,
      "loss": 0.0162,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004014718986582011,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.86663932800293,
      "state_query_norm": 1428.4342041015625,
      "state_relation_loss": 0.0004014718986582011,
      "step": 29490,
      "student_distance_mean": 1966.1940063476563,
      "teacher_distance_mean": 1.1689536452293396,
      "weighted_auxiliary_loss": 0.0004014718986582011,
      "weighted_relation_loss": 0.0004014718986582011
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.595884003741816,
      "flow_matching_loss": 0.015265331603586674,
      "grad_norm": 0.08141608536243439,
      "learning_rate": 7.622810147614456e-08,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036959124845452606,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16652946472168,
      "state_query_norm": 1408.5831420898437,
      "state_relation_loss": 0.00036959124845452606,
      "step": 29500,
      "student_distance_mean": 1932.9589233398438,
      "teacher_distance_mean": 1.1487419366836549,
      "weighted_auxiliary_loss": 0.00036959124845452606,
      "weighted_relation_loss": 0.00036959124845452606
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.605238540692234,
      "flow_matching_loss": 0.014749036729335785,
      "grad_norm": 0.08300336450338364,
      "learning_rate": 7.321616860690995e-08,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003753881435841322,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.07066764831543,
      "state_query_norm": 1408.5997680664063,
      "state_relation_loss": 0.0003753881435841322,
      "step": 29510,
      "student_distance_mean": 1930.9066650390625,
      "teacher_distance_mean": 1.147382378578186,
      "weighted_auxiliary_loss": 0.0003753881435841322,
      "weighted_relation_loss": 0.0003753881435841322
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.614593077642656,
      "flow_matching_loss": 0.017102673649787903,
      "grad_norm": 0.07858273386955261,
      "learning_rate": 7.026490148782095e-08,
      "loss": 0.0175,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036951578804291787,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.92450103759766,
      "state_query_norm": 1404.9475708007812,
      "state_relation_loss": 0.00036951578804291787,
      "step": 29520,
      "student_distance_mean": 1920.0018432617187,
      "teacher_distance_mean": 1.1408459186553954,
      "weighted_auxiliary_loss": 0.00036951578804291787,
      "weighted_relation_loss": 0.00036951578804291787
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.623947614593078,
      "flow_matching_loss": 0.015508861187845469,
      "grad_norm": 0.0819854661822319,
      "learning_rate": 6.737430370494236e-08,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003618164191721007,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.56288185119629,
      "state_query_norm": 1391.0191040039062,
      "state_relation_loss": 0.0003618164191721007,
      "step": 29530,
      "student_distance_mean": 1903.791796875,
      "teacher_distance_mean": 1.131865096092224,
      "weighted_auxiliary_loss": 0.0003618164191721007,
      "weighted_relation_loss": 0.0003618164191721007
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.6333021515435,
      "flow_matching_loss": 0.01554682170972228,
      "grad_norm": 0.09305180609226227,
      "learning_rate": 6.454437877062569e-08,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003857333038467914,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42317771911621,
      "state_query_norm": 1419.934521484375,
      "state_relation_loss": 0.0003857333038467914,
      "step": 29540,
      "student_distance_mean": 1948.4004272460938,
      "teacher_distance_mean": 1.159082818031311,
      "weighted_auxiliary_loss": 0.0003857333038467914,
      "weighted_relation_loss": 0.0003857333038467914
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.64265668849392,
      "flow_matching_loss": 0.015205033216625452,
      "grad_norm": 0.10216566920280457,
      "learning_rate": 6.177513012349256e-08,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039848097076173873,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.98038101196289,
      "state_query_norm": 1435.95322265625,
      "state_relation_loss": 0.00039848097076173873,
      "step": 29550,
      "student_distance_mean": 1973.3789794921875,
      "teacher_distance_mean": 1.1732938408851623,
      "weighted_auxiliary_loss": 0.00039848097076173873,
      "weighted_relation_loss": 0.00039848097076173873
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.65201122544434,
      "flow_matching_loss": 0.014194167871028184,
      "grad_norm": 0.06544127315282822,
      "learning_rate": 5.9066561128445775e-08,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037289976316969844,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.843448638916016,
      "state_query_norm": 1390.0512573242188,
      "state_relation_loss": 0.00037289976316969844,
      "step": 29560,
      "student_distance_mean": 1910.5886108398438,
      "teacher_distance_mean": 1.1354588627815247,
      "weighted_auxiliary_loss": 0.00037289976316969844,
      "weighted_relation_loss": 0.00037289976316969844
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.661365762394762,
      "flow_matching_loss": 0.015011003892868757,
      "grad_norm": 0.0728025734424591,
      "learning_rate": 5.6418675076641556e-08,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004144320788327605,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.38462448120117,
      "state_query_norm": 1418.306689453125,
      "state_relation_loss": 0.0004144320788327605,
      "step": 29570,
      "student_distance_mean": 1944.5811645507813,
      "teacher_distance_mean": 1.155993914604187,
      "weighted_auxiliary_loss": 0.0004144320788327605,
      "weighted_relation_loss": 0.0004144320788327605
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.67072029934518,
      "flow_matching_loss": 0.015155072975903749,
      "grad_norm": 0.07845994830131531,
      "learning_rate": 5.383147518552845e-08,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003716472536325455,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6782039642334,
      "state_query_norm": 1426.3397827148438,
      "state_relation_loss": 0.0003716472536325455,
      "step": 29580,
      "student_distance_mean": 1958.6006103515624,
      "teacher_distance_mean": 1.1634895920753479,
      "weighted_auxiliary_loss": 0.0003716472536325455,
      "weighted_relation_loss": 0.0003716472536325455
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.680074836295603,
      "flow_matching_loss": 0.015944964718073606,
      "grad_norm": 0.0935092568397522,
      "learning_rate": 5.1304964598786196e-08,
      "loss": 0.0164,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00041231533396057785,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.6784065246582,
      "state_query_norm": 1422.64560546875,
      "state_relation_loss": 0.00041231533396057785,
      "step": 29590,
      "student_distance_mean": 1959.011669921875,
      "teacher_distance_mean": 1.1645301699638366,
      "weighted_auxiliary_loss": 0.00041231533396057785,
      "weighted_relation_loss": 0.00041231533396057785
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.689429373246025,
      "flow_matching_loss": 0.014741263538599014,
      "grad_norm": 0.08460190147161484,
      "learning_rate": 4.883914638636467e-08,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004008356627309695,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.115976333618164,
      "state_query_norm": 1402.2919921875,
      "state_relation_loss": 0.0004008356627309695,
      "step": 29600,
      "student_distance_mean": 1925.5594604492187,
      "teacher_distance_mean": 1.144008719921112,
      "weighted_auxiliary_loss": 0.0004008356627309695,
      "weighted_relation_loss": 0.0004008356627309695
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.698783910196447,
      "flow_matching_loss": 0.014469897840172052,
      "grad_norm": 0.06735725700855255,
      "learning_rate": 4.643402354446713e-08,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003952069935621694,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.13807487487793,
      "state_query_norm": 1398.8808837890624,
      "state_relation_loss": 0.0003952069935621694,
      "step": 29610,
      "student_distance_mean": 1929.3568603515625,
      "teacher_distance_mean": 1.146523928642273,
      "weighted_auxiliary_loss": 0.0003952069935621694,
      "weighted_relation_loss": 0.0003952069935621694
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.708138447146865,
      "flow_matching_loss": 0.015932380221784115,
      "grad_norm": 0.08776932209730148,
      "learning_rate": 4.4089598995544766e-08,
      "loss": 0.0163,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003834722243482247,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.51970291137695,
      "state_query_norm": 1418.4925170898437,
      "state_relation_loss": 0.0003834722243482247,
      "step": 29620,
      "student_distance_mean": 1951.0938720703125,
      "teacher_distance_mean": 1.1599701642990112,
      "weighted_auxiliary_loss": 0.0003834722243482247,
      "weighted_relation_loss": 0.0003834722243482247
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.717492984097287,
      "flow_matching_loss": 0.013868057448416949,
      "grad_norm": 0.06346490979194641,
      "learning_rate": 4.180587558829663e-08,
      "loss": 0.0143,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003956655884394422,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.37881851196289,
      "state_query_norm": 1409.4358154296874,
      "state_relation_loss": 0.0003956655884394422,
      "step": 29630,
      "student_distance_mean": 1939.5224365234376,
      "teacher_distance_mean": 1.1524327158927918,
      "weighted_auxiliary_loss": 0.0003956655884394422,
      "weighted_relation_loss": 0.0003956655884394422
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.72684752104771,
      "flow_matching_loss": 0.016730557847768068,
      "grad_norm": 0.08351118117570877,
      "learning_rate": 3.958285609765855e-08,
      "loss": 0.0171,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036749645660165696,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.04779243469238,
      "state_query_norm": 1407.5960083007812,
      "state_relation_loss": 0.00036749645660165696,
      "step": 29640,
      "student_distance_mean": 1930.3761840820312,
      "teacher_distance_mean": 1.14699889421463,
      "weighted_auxiliary_loss": 0.00036749645660165696,
      "weighted_relation_loss": 0.00036749645660165696
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.736202057998128,
      "flow_matching_loss": 0.014268230553716421,
      "grad_norm": 0.09918300807476044,
      "learning_rate": 3.74205432248087e-08,
      "loss": 0.0146,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038069788715802135,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.16053924560547,
      "state_query_norm": 1407.6376586914062,
      "state_relation_loss": 0.00038069788715802135,
      "step": 29650,
      "student_distance_mean": 1932.9330810546876,
      "teacher_distance_mean": 1.1485795021057128,
      "weighted_auxiliary_loss": 0.00038069788715802135,
      "weighted_relation_loss": 0.00038069788715802135
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.74555659494855,
      "flow_matching_loss": 0.014783100597560406,
      "grad_norm": 0.136185422539711,
      "learning_rate": 3.531893959716204e-08,
      "loss": 0.0152,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000414554329472594,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.434461975097655,
      "state_query_norm": 1419.3394287109375,
      "state_relation_loss": 0.000414554329472594,
      "step": 29660,
      "student_distance_mean": 1944.3233764648437,
      "teacher_distance_mean": 1.1566421151161195,
      "weighted_auxiliary_loss": 0.000414554329472594,
      "weighted_relation_loss": 0.000414554329472594
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.75491113189897,
      "flow_matching_loss": 0.016489108838140965,
      "grad_norm": 0.09302158653736115,
      "learning_rate": 3.327804776837029e-08,
      "loss": 0.0169,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003766268724575639,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.88450126647949,
      "state_query_norm": 1405.4335083007813,
      "state_relation_loss": 0.0003766268724575639,
      "step": 29670,
      "student_distance_mean": 1922.9264892578126,
      "teacher_distance_mean": 1.1433433532714843,
      "weighted_auxiliary_loss": 0.0003766268724575639,
      "weighted_relation_loss": 0.0003766268724575639
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.764265668849394,
      "flow_matching_loss": 0.014472746849060058,
      "grad_norm": 0.0747046247124672,
      "learning_rate": 3.129787021829977e-08,
      "loss": 0.0148,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036872976343147454,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.03132705688476,
      "state_query_norm": 1403.8115844726562,
      "state_relation_loss": 0.00036872976343147454,
      "step": 29680,
      "student_distance_mean": 1924.316552734375,
      "teacher_distance_mean": 1.143888568878174,
      "weighted_auxiliary_loss": 0.00036872976343147454,
      "weighted_relation_loss": 0.00036872976343147454
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.773620205799812,
      "flow_matching_loss": 0.014942662790417671,
      "grad_norm": 0.07658109068870544,
      "learning_rate": 2.9378409353059133e-08,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003827021166216582,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.93986701965332,
      "state_query_norm": 1399.4016967773437,
      "state_relation_loss": 0.0003827021166216582,
      "step": 29690,
      "student_distance_mean": 1918.519921875,
      "teacher_distance_mean": 1.1402872681617737,
      "weighted_auxiliary_loss": 0.0003827021166216582,
      "weighted_relation_loss": 0.0003827021166216582
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.782974742750234,
      "flow_matching_loss": 0.016328945476561784,
      "grad_norm": 0.07795587182044983,
      "learning_rate": 2.7519667504971593e-08,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003987451986176893,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30633316040039,
      "state_query_norm": 1406.1907592773437,
      "state_relation_loss": 0.0003987451986176893,
      "step": 29700,
      "student_distance_mean": 1933.7864624023437,
      "teacher_distance_mean": 1.1498364686965943,
      "weighted_auxiliary_loss": 0.0003987451986176893,
      "weighted_relation_loss": 0.0003987451986176893
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.792329279700656,
      "flow_matching_loss": 0.015247558709234,
      "grad_norm": 0.07524777948856354,
      "learning_rate": 2.572164693258605e-08,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003951348247937858,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.304130172729494,
      "state_query_norm": 1410.7224975585937,
      "state_relation_loss": 0.0003951348247937858,
      "step": 29710,
      "student_distance_mean": 1939.8450317382812,
      "teacher_distance_mean": 1.1533969402313233,
      "weighted_auxiliary_loss": 0.0003951348247937858,
      "weighted_relation_loss": 0.0003951348247937858
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.801683816651074,
      "flow_matching_loss": 0.01461256192997098,
      "grad_norm": 0.07482123374938965,
      "learning_rate": 2.3984349820665996e-08,
      "loss": 0.015,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003815305535681546,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.0809986114502,
      "state_query_norm": 1405.0379760742187,
      "state_relation_loss": 0.0003815305535681546,
      "step": 29720,
      "student_distance_mean": 1926.58642578125,
      "teacher_distance_mean": 1.1454339146614074,
      "weighted_auxiliary_loss": 0.0003815305535681546,
      "weighted_relation_loss": 0.0003815305535681546
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.811038353601496,
      "flow_matching_loss": 0.01642491314560175,
      "grad_norm": 0.08525459468364716,
      "learning_rate": 2.2307778280189485e-08,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003817224904196337,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.43870735168457,
      "state_query_norm": 1419.835791015625,
      "state_relation_loss": 0.0003817224904196337,
      "step": 29730,
      "student_distance_mean": 1948.7291259765625,
      "teacher_distance_mean": 1.1582888841629029,
      "weighted_auxiliary_loss": 0.0003817224904196337,
      "weighted_relation_loss": 0.0003817224904196337
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.82039289055192,
      "flow_matching_loss": 0.015026091691106557,
      "grad_norm": 0.08891444653272629,
      "learning_rate": 2.0691934348354704e-08,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004095642361789942,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.58596000671387,
      "state_query_norm": 1426.2711669921875,
      "state_relation_loss": 0.0004095642361789942,
      "step": 29740,
      "student_distance_mean": 1951.7157348632813,
      "teacher_distance_mean": 1.1590904116630554,
      "weighted_auxiliary_loss": 0.0004095642361789942,
      "weighted_relation_loss": 0.0004095642361789942
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.82974742750234,
      "flow_matching_loss": 0.014925557002425194,
      "grad_norm": 0.06667371094226837,
      "learning_rate": 1.9136819988557763e-08,
      "loss": 0.0153,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004009297204902396,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.53827743530273,
      "state_query_norm": 1414.8988525390625,
      "state_relation_loss": 0.0004009297204902396,
      "step": 29750,
      "student_distance_mean": 1946.9177368164062,
      "teacher_distance_mean": 1.1571775555610657,
      "weighted_auxiliary_loss": 0.0004009297204902396,
      "weighted_relation_loss": 0.0004009297204902396
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.83910196445276,
      "flow_matching_loss": 0.015403343085199594,
      "grad_norm": 0.08340305835008621,
      "learning_rate": 1.7642437090414908e-08,
      "loss": 0.0158,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003944536583730951,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.30822601318359,
      "state_query_norm": 1415.3743530273437,
      "state_relation_loss": 0.0003944536583730951,
      "step": 29760,
      "student_distance_mean": 1939.414697265625,
      "teacher_distance_mean": 1.1529806137084961,
      "weighted_auxiliary_loss": 0.0003944536583730951,
      "weighted_relation_loss": 0.0003944536583730951
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.84845650140318,
      "flow_matching_loss": 0.015088405460119247,
      "grad_norm": 0.08723856508731842,
      "learning_rate": 1.6208787469734755e-08,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038474550528917463,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.42367286682129,
      "state_query_norm": 1409.8779052734376,
      "state_relation_loss": 0.00038474550528917463,
      "step": 29770,
      "student_distance_mean": 1942.7642822265625,
      "teacher_distance_mean": 1.1550674915313721,
      "weighted_auxiliary_loss": 0.00038474550528917463,
      "weighted_relation_loss": 0.00038474550528917463
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.857811038353603,
      "flow_matching_loss": 0.014730075653642417,
      "grad_norm": 0.0735628604888916,
      "learning_rate": 1.483587286854604e-08,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040160354983527214,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.52976913452149,
      "state_query_norm": 1424.4368896484375,
      "state_relation_loss": 0.00040160354983527214,
      "step": 29780,
      "student_distance_mean": 1954.727392578125,
      "teacher_distance_mean": 1.1621949911117553,
      "weighted_auxiliary_loss": 0.00040160354983527214,
      "weighted_relation_loss": 0.00040160354983527214
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.86716557530402,
      "flow_matching_loss": 0.015191586408764123,
      "grad_norm": 0.0838298425078392,
      "learning_rate": 1.3523694955064336e-08,
      "loss": 0.0156,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00039598156872671096,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.80260581970215,
      "state_query_norm": 1433.7657104492187,
      "state_relation_loss": 0.00039598156872671096,
      "step": 29790,
      "student_distance_mean": 1966.4337036132813,
      "teacher_distance_mean": 1.1686363697052002,
      "weighted_auxiliary_loss": 0.00039598156872671096,
      "weighted_relation_loss": 0.00039598156872671096
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.876520112254443,
      "flow_matching_loss": 0.015320839826017618,
      "grad_norm": 0.0645233616232872,
      "learning_rate": 1.2272255323708681e-08,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003551549511030316,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.15575523376465,
      "state_query_norm": 1402.8176635742188,
      "state_relation_loss": 0.0003551549511030316,
      "step": 29800,
      "student_distance_mean": 1929.639892578125,
      "teacher_distance_mean": 1.1471195816993713,
      "weighted_auxiliary_loss": 0.0003551549511030316,
      "weighted_relation_loss": 0.0003551549511030316
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.885874649204865,
      "flow_matching_loss": 0.01633699079975486,
      "grad_norm": 0.07058993726968765,
      "learning_rate": 1.1081555495096042e-08,
      "loss": 0.0167,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003941055154427886,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 34.924869537353516,
      "state_query_norm": 1390.7286010742187,
      "state_relation_loss": 0.0003941055154427886,
      "step": 29810,
      "student_distance_mean": 1916.1402465820313,
      "teacher_distance_mean": 1.1389636993408203,
      "weighted_auxiliary_loss": 0.0003941055154427886,
      "weighted_relation_loss": 0.0003941055154427886
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.895229186155284,
      "flow_matching_loss": 0.015060238726437091,
      "grad_norm": 0.09507483243942261,
      "learning_rate": 9.951596916041306e-09,
      "loss": 0.0154,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037160647043492644,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.421372985839845,
      "state_query_norm": 1419.3802124023437,
      "state_relation_loss": 0.00037160647043492644,
      "step": 29820,
      "student_distance_mean": 1948.7281860351563,
      "teacher_distance_mean": 1.1583100080490112,
      "weighted_auxiliary_loss": 0.00037160647043492644,
      "weighted_relation_loss": 0.00037160647043492644
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.904583723105706,
      "flow_matching_loss": 0.015050525777041912,
      "grad_norm": 0.10330428183078766,
      "learning_rate": 8.882380959551739e-09,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00040607426199130715,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.8564868927002,
      "state_query_norm": 1429.1587158203124,
      "state_relation_loss": 0.00040607426199130715,
      "step": 29830,
      "student_distance_mean": 1967.277734375,
      "teacher_distance_mean": 1.169283926486969,
      "weighted_auxiliary_loss": 0.00040607426199130715,
      "weighted_relation_loss": 0.00040607426199130715
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.913938260056128,
      "flow_matching_loss": 0.014376670215278864,
      "grad_norm": 0.08815333992242813,
      "learning_rate": 7.873908924821428e-09,
      "loss": 0.0148,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003852870169794187,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.33863830566406,
      "state_query_norm": 1410.7337036132812,
      "state_relation_loss": 0.0003852870169794187,
      "step": 29840,
      "student_distance_mean": 1939.990869140625,
      "teacher_distance_mean": 1.153715693950653,
      "weighted_auxiliary_loss": 0.0003852870169794187,
      "weighted_relation_loss": 0.0003852870169794187
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.92329279700655,
      "flow_matching_loss": 0.016352342814207076,
      "grad_norm": 0.0886288657784462,
      "learning_rate": 6.926182037242379e-09,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004155915783485398,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.78081359863281,
      "state_query_norm": 1429.41640625,
      "state_relation_loss": 0.0004155915783485398,
      "step": 29850,
      "student_distance_mean": 1961.5890625,
      "teacher_distance_mean": 1.165397548675537,
      "weighted_auxiliary_loss": 0.0004155915783485398,
      "weighted_relation_loss": 0.0004155915783485398
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.932647333956968,
      "flow_matching_loss": 0.01451787119731307,
      "grad_norm": 0.08529394865036011,
      "learning_rate": 6.039201448387876e-09,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003794563497649506,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.55335235595703,
      "state_query_norm": 1419.2641235351562,
      "state_relation_loss": 0.0003794563497649506,
      "step": 29860,
      "student_distance_mean": 1950.5523559570313,
      "teacher_distance_mean": 1.1590242266654969,
      "weighted_auxiliary_loss": 0.0003794563497649506,
      "weighted_relation_loss": 0.0003794563497649506
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.94200187090739,
      "flow_matching_loss": 0.017522887885570528,
      "grad_norm": 0.09868411719799042,
      "learning_rate": 5.212968236029125e-09,
      "loss": 0.0179,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037210386944934725,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.32043533325195,
      "state_query_norm": 1411.4236083984374,
      "state_relation_loss": 0.00037210386944934725,
      "step": 29870,
      "student_distance_mean": 1938.4820190429687,
      "teacher_distance_mean": 1.152447807788849,
      "weighted_auxiliary_loss": 0.00037210386944934725,
      "weighted_relation_loss": 0.00037210386944934725
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.951356407857812,
      "flow_matching_loss": 0.015535815246403217,
      "grad_norm": 0.06931373476982117,
      "learning_rate": 4.447483404118602e-09,
      "loss": 0.0159,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00036585620255209507,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.364976501464845,
      "state_query_norm": 1415.8240478515625,
      "state_relation_loss": 0.00036585620255209507,
      "step": 29880,
      "student_distance_mean": 1943.7323608398438,
      "teacher_distance_mean": 1.1552301406860352,
      "weighted_auxiliary_loss": 0.00036585620255209507,
      "weighted_relation_loss": 0.00036585620255209507
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.96071094480823,
      "flow_matching_loss": 0.01640744209289551,
      "grad_norm": 0.09973742067813873,
      "learning_rate": 3.742747882784503e-09,
      "loss": 0.0168,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003668051213026047,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.216814804077146,
      "state_query_norm": 1408.62607421875,
      "state_relation_loss": 0.0003668051213026047,
      "step": 29890,
      "student_distance_mean": 1934.8529907226562,
      "teacher_distance_mean": 1.1500423789024352,
      "weighted_auxiliary_loss": 0.0003668051213026047,
      "weighted_relation_loss": 0.0003668051213026047
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.970065481758652,
      "flow_matching_loss": 0.014734569750726224,
      "grad_norm": 0.07972979545593262,
      "learning_rate": 3.0987625283473987e-09,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00037864497862756253,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.411305618286136,
      "state_query_norm": 1413.138818359375,
      "state_relation_loss": 0.00037864497862756253,
      "step": 29900,
      "student_distance_mean": 1943.0017700195312,
      "teacher_distance_mean": 1.1544561505317688,
      "weighted_auxiliary_loss": 0.00037864497862756253,
      "weighted_relation_loss": 0.00037864497862756253
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.979420018709074,
      "flow_matching_loss": 0.015622962825000286,
      "grad_norm": 0.09161408245563507,
      "learning_rate": 2.515528123320232e-09,
      "loss": 0.016,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038325478381011634,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.46360893249512,
      "state_query_norm": 1416.668994140625,
      "state_relation_loss": 0.00038325478381011634,
      "step": 29910,
      "student_distance_mean": 1951.1586547851562,
      "teacher_distance_mean": 1.1592378497123719,
      "weighted_auxiliary_loss": 0.00038325478381011634,
      "weighted_relation_loss": 0.00038325478381011634
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.988774555659496,
      "flow_matching_loss": 0.014542175363749266,
      "grad_norm": 0.08520129323005676,
      "learning_rate": 1.9930453763750133e-09,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003656622197013348,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.378853988647464,
      "state_query_norm": 1417.7164794921875,
      "state_relation_loss": 0.0003656622197013348,
      "step": 29920,
      "student_distance_mean": 1945.974951171875,
      "teacher_distance_mean": 1.1568496584892274,
      "weighted_auxiliary_loss": 0.0003656622197013348,
      "weighted_relation_loss": 0.0003656622197013348
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 27.998129092609915,
      "flow_matching_loss": 0.01617075065150857,
      "grad_norm": 0.07170943915843964,
      "learning_rate": 1.5313149223872281e-09,
      "loss": 0.0165,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.000368700354010798,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.05563125610352,
      "state_query_norm": 1408.7427368164062,
      "state_relation_loss": 0.000368700354010798,
      "step": 29930,
      "student_distance_mean": 1928.7062744140626,
      "teacher_distance_mean": 1.1460488200187684,
      "weighted_auxiliary_loss": 0.000368700354010798,
      "weighted_relation_loss": 0.000368700354010798
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.007483629560337,
      "flow_matching_loss": 0.015378132835030556,
      "grad_norm": 0.09241583198308945,
      "learning_rate": 1.1303373224025305e-09,
      "loss": 0.0158,
      "positive_teacher_pairs": 29535.6,
      "relation_loss": 0.00038153828936629,
      "selected_pairs": 29535.6,
      "state_query_batch_std": 34.76767101287842,
      "state_query_norm": 1394.5896240234374,
      "state_relation_loss": 0.00038153828936629,
      "step": 29940,
      "student_distance_mean": 1912.891650390625,
      "teacher_distance_mean": 1.1372549057006835,
      "weighted_auxiliary_loss": 0.00038153828936629,
      "weighted_relation_loss": 0.00038153828936629
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.01683816651076,
      "flow_matching_loss": 0.014713512174785138,
      "grad_norm": 0.07349007576704025,
      "learning_rate": 7.901130636367438e-10,
      "loss": 0.0151,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0003544928476912901,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.204398727416994,
      "state_query_norm": 1414.2843994140626,
      "state_relation_loss": 0.0003544928476912901,
      "step": 29950,
      "student_distance_mean": 1940.0712280273438,
      "teacher_distance_mean": 1.152607500553131,
      "weighted_auxiliary_loss": 0.0003544928476912901,
      "weighted_relation_loss": 0.0003544928476912901
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.026192703461177,
      "flow_matching_loss": 0.016251351125538348,
      "grad_norm": 0.07209387421607971,
      "learning_rate": 5.106425595036158e-10,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038943090185057373,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.91185455322265,
      "state_query_norm": 1441.3403442382812,
      "state_relation_loss": 0.00038943090185057373,
      "step": 29960,
      "student_distance_mean": 1974.0808471679688,
      "teacher_distance_mean": 1.173105001449585,
      "weighted_auxiliary_loss": 0.00038943090185057373,
      "weighted_relation_loss": 0.00038943090185057373
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.0355472404116,
      "flow_matching_loss": 0.015067714173346758,
      "grad_norm": 0.10419059544801712,
      "learning_rate": 2.9192614958706286e-10,
      "loss": 0.0155,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004184536403045058,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.51689071655274,
      "state_query_norm": 1428.4105834960938,
      "state_relation_loss": 0.0004184536403045058,
      "step": 29970,
      "student_distance_mean": 1953.1077758789063,
      "teacher_distance_mean": 1.161269974708557,
      "weighted_auxiliary_loss": 0.0004184536403045058,
      "weighted_relation_loss": 0.0004184536403045058
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.04490177736202,
      "flow_matching_loss": 0.015262381546199322,
      "grad_norm": 0.08078903704881668,
      "learning_rate": 1.3396409964117064e-10,
      "loss": 0.0157,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004155737900873646,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.773583221435544,
      "state_query_norm": 1429.8831420898437,
      "state_relation_loss": 0.0004155737900873646,
      "step": 29980,
      "student_distance_mean": 1961.8140869140625,
      "teacher_distance_mean": 1.1656406998634339,
      "weighted_auxiliary_loss": 0.0004155737900873646,
      "weighted_relation_loss": 0.0004155737900873646
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.054256314312443,
      "flow_matching_loss": 0.014547235053032636,
      "grad_norm": 0.0697249323129654,
      "learning_rate": 3.6756601606846574e-11,
      "loss": 0.0149,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.00038389794644899666,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.158634948730466,
      "state_query_norm": 1399.047607421875,
      "state_relation_loss": 0.00038389794644899666,
      "step": 29990,
      "student_distance_mean": 1927.5641723632812,
      "teacher_distance_mean": 1.1453713774681091,
      "weighted_auxiliary_loss": 0.00038389794644899666,
      "weighted_relation_loss": 0.00038389794644899666
    },
    {
      "auxiliary_loss_weight": 1.0,
      "epoch": 28.06361085126286,
      "flow_matching_loss": 0.016143257077783347,
      "grad_norm": 0.0950896367430687,
      "learning_rate": 3.0377360626943076e-13,
      "loss": 0.0166,
      "positive_teacher_pairs": 32640.0,
      "relation_loss": 0.0004107696586288512,
      "selected_pairs": 32640.0,
      "state_query_batch_std": 35.64100799560547,
      "state_query_norm": 1423.4804443359376,
      "state_relation_loss": 0.0004107696586288512,
      "step": 30000,
      "student_distance_mean": 1953.9875366210938,
      "teacher_distance_mean": 1.1611921548843385,
      "weighted_auxiliary_loss": 0.0004107696586288512,
      "weighted_relation_loss": 0.0004107696586288512
    },
    {
      "epoch": 28.06361085126286,
      "step": 30000,
      "total_flos": 0.0,
      "train_loss": 0.03990069806029399,
      "train_runtime": 61767.4208,
      "train_samples_per_second": 124.337,
      "train_steps_per_second": 0.486
    }
  ],
  "logging_steps": 10,
  "max_steps": 30000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 29,
  "save_steps": 10000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 256,
  "trial_name": null,
  "trial_params": null
}