{
  "best_global_step": 1100,
  "best_metric": 13.589922904968262,
  "best_model_checkpoint": "../mms-fusion/results/mms-1b-all-bemgen-m50f50-ft-sd-dat-gdro-fusion-42/checkpoint-600",
  "epoch": 7.106598984771574,
  "eval_steps": 100,
  "global_step": 1400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0,
      "gdro_loss": 2308.5205078125,
      "loss_g0": 2283.408447265625,
      "loss_g1": 2333.62451171875,
      "q_g0": 0.499921053647995,
      "q_g1": 0.5000789761543274,
      "step": 0
    },
    {
      "epoch": 0,
      "gdro_loss": 1986.8875732421875,
      "loss_g0": 2052.758544921875,
      "loss_g1": 1920.9827880859375,
      "q_g0": 0.5001282691955566,
      "q_g1": 0.49987173080444336,
      "step": 0
    },
    {
      "epoch": 0.5076142131979695,
      "grad_norm": 2811.123779296875,
      "learning_rate": 0.00040480000000000003,
      "loss": 591.6625,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 809.1090698242188,
      "loss_g0": 984.8280029296875,
      "loss_g1": 806.1497802734375,
      "q_g0": 0.01656188815832138,
      "q_g1": 0.9834380745887756,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 669.0565185546875,
      "loss_g0": 0.0,
      "loss_g1": 669.0565185546875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 614.7470092773438,
      "loss_g0": 612.6997680664062,
      "loss_g1": 614.7813720703125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 508.60400390625,
      "loss_g0": 624.4375,
      "loss_g1": 506.65655517578125,
      "q_g0": 0.016534464433789253,
      "q_g1": 0.9834655523300171,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 553.3844604492188,
      "loss_g0": 467.7896423339844,
      "loss_g1": 554.82275390625,
      "q_g0": 0.016525564715266228,
      "q_g1": 0.9834744334220886,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 536.8174438476562,
      "loss_g0": 585.6150512695312,
      "loss_g1": 535.9971923828125,
      "q_g0": 0.016530640423297882,
      "q_g1": 0.9834693670272827,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 516.3961791992188,
      "loss_g0": 495.951171875,
      "loss_g1": 516.73974609375,
      "q_g0": 0.016528518870472908,
      "q_g1": 0.9834715127944946,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 518.115478515625,
      "loss_g0": 495.0645446777344,
      "loss_g1": 518.5028076171875,
      "q_g0": 0.016526123508810997,
      "q_g1": 0.9834738373756409,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 524.3229370117188,
      "loss_g0": 498.9658203125,
      "loss_g1": 524.7489624023438,
      "q_g0": 0.016523487865924835,
      "q_g1": 0.9834765195846558,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 450.3695068359375,
      "loss_g0": 407.7215270996094,
      "loss_g1": 451.08599853515625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 470.34722900390625,
      "loss_g0": 497.9361572265625,
      "loss_g1": 469.8836669921875,
      "q_g0": 0.016525287181138992,
      "q_g1": 0.9834747910499573,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 409.0244140625,
      "loss_g0": 437.42413330078125,
      "loss_g1": 408.547119140625,
      "q_g0": 0.016528239473700523,
      "q_g1": 0.9834717512130737,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 400.5539855957031,
      "loss_g0": 400.5539855957031,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 403.2846374511719,
      "loss_g0": 404.5768737792969,
      "loss_g1": 326.3651123046875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 411.9927062988281,
      "loss_g0": 411.67449951171875,
      "loss_g1": 430.93212890625,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524387523531914,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 360.9324035644531,
      "loss_g0": 361.4365234375,
      "loss_g1": 330.9248962402344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 297.8448486328125,
      "loss_g0": 295.29083251953125,
      "loss_g1": 449.7212829589844,
      "q_g0": 0.9834617376327515,
      "q_g1": 0.016538208350539207,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 397.2027893066406,
      "loss_g0": 397.5465087890625,
      "loss_g1": 376.7589111328125,
      "q_g0": 0.9834639430046082,
      "q_g1": 0.016536086797714233,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 349.1286315917969,
      "loss_g0": 349.47955322265625,
      "loss_g1": 328.25439453125,
      "q_g0": 0.9834660887718201,
      "q_g1": 0.016533922404050827,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 346.0028991699219,
      "loss_g0": 345.8759460449219,
      "loss_g1": 353.553955078125,
      "q_g0": 0.983465313911438,
      "q_g1": 0.016534706577658653,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 294.2676696777344,
      "loss_g0": 293.72540283203125,
      "loss_g1": 326.5135498046875,
      "q_g0": 0.9834619760513306,
      "q_g1": 0.016538063064217567,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 335.34521484375,
      "loss_g0": 335.30767822265625,
      "loss_g1": 337.576416015625,
      "q_g0": 0.9834617376327515,
      "q_g1": 0.01653829589486122,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 336.8099060058594,
      "loss_g0": 336.66485595703125,
      "loss_g1": 345.43560791015625,
      "q_g0": 0.9834607839584351,
      "q_g1": 0.0165391955524683,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 389.69708251953125,
      "loss_g0": 0.0,
      "loss_g1": 389.69708251953125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 349.2480773925781,
      "loss_g0": 378.2163391113281,
      "loss_g1": 348.76141357421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 340.5412902832031,
      "loss_g0": 0.0,
      "loss_g1": 340.5412902832031,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 327.8504638671875,
      "loss_g0": 275.9081726074219,
      "loss_g1": 328.72308349609375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 347.3083801269531,
      "loss_g0": 292.2791442871094,
      "loss_g1": 348.23284912109375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 281.52532958984375,
      "loss_g0": 297.36431884765625,
      "loss_g1": 281.25921630859375,
      "q_g0": 0.016524069011211395,
      "q_g1": 0.983475923538208,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 268.35662841796875,
      "loss_g0": 268.35662841796875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 248.96563720703125,
      "loss_g0": 248.85067749023438,
      "loss_g1": 255.80868530273438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 180.73236083984375,
      "loss_g0": 178.80703735351562,
      "loss_g1": 295.2510986328125,
      "q_g0": 0.9834656715393066,
      "q_g1": 0.016534332185983658,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 228.23129272460938,
      "loss_g0": 0.0,
      "loss_g1": 228.23129272460938,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 308.0117492675781,
      "loss_g0": 308.0117492675781,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 242.60714721679688,
      "loss_g0": 239.384521484375,
      "loss_g1": 434.4306945800781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 252.10989379882812,
      "loss_g0": 252.087158203125,
      "loss_g1": 253.46347045898438,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652255840599537,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 181.86512756347656,
      "loss_g0": 180.87686157226562,
      "loss_g1": 240.66799926757812,
      "q_g0": 0.9834713339805603,
      "q_g1": 0.016528671607375145,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 227.67218017578125,
      "loss_g0": 227.63217163085938,
      "loss_g1": 230.05233764648438,
      "q_g0": 0.9834710955619812,
      "q_g1": 0.016528921201825142,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 229.84909057617188,
      "loss_g0": 229.84909057617188,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 173.84259033203125,
      "loss_g0": 172.37979125976562,
      "loss_g1": 260.9136962890625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 187.88458251953125,
      "loss_g0": 187.88458251953125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 267.6958312988281,
      "loss_g0": 269.8993225097656,
      "loss_g1": 136.53543090820312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 357.05291748046875,
      "loss_g0": 358.8711242675781,
      "loss_g1": 248.82577514648438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 160.28460693359375,
      "loss_g0": 160.28460693359375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 138.66407775878906,
      "loss_g0": 137.55599975585938,
      "loss_g1": 204.62039184570312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 138.9376220703125,
      "loss_g0": 138.32395935058594,
      "loss_g1": 175.45611572265625,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.01652621664106846,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 169.10464477539062,
      "loss_g0": 169.14324951171875,
      "loss_g1": 166.8066864013672,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525983810424805,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 134.33572387695312,
      "loss_g0": 132.73757934570312,
      "loss_g1": 229.385009765625,
      "q_g0": 0.9834641814231873,
      "q_g1": 0.016535868868231773,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 134.81382751464844,
      "loss_g0": 134.27542114257812,
      "loss_g1": 166.8284454345703,
      "q_g0": 0.9834607839584351,
      "q_g1": 0.016539203003048897,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 143.7320098876953,
      "loss_g0": 143.61972045898438,
      "loss_g1": 150.4091339111328,
      "q_g0": 0.9834601283073425,
      "q_g1": 0.01653989963233471,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 824.3179931640625,
      "loss_g0": 824.61376953125,
      "loss_g1": 806.7274780273438,
      "q_g0": 0.9834619760513306,
      "q_g1": 0.016538070514798164,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 717.607177734375,
      "loss_g0": 720.3955078125,
      "loss_g1": 551.6328125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 574.8270263671875,
      "loss_g0": 575.23828125,
      "loss_g1": 550.3472290039062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 623.5372924804688,
      "loss_g0": 624.4019165039062,
      "loss_g1": 572.0711669921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 540.1595458984375,
      "loss_g0": 540.3223876953125,
      "loss_g1": 530.4656372070312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 494.3065490722656,
      "loss_g0": 494.81396484375,
      "loss_g1": 464.102783203125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 511.4664611816406,
      "loss_g0": 511.658203125,
      "loss_g1": 500.052490234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 486.30181884765625,
      "loss_g0": 485.86083984375,
      "loss_g1": 512.5467529296875,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525154933333397,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 459.4168701171875,
      "loss_g0": 459.781494140625,
      "loss_g1": 437.7139587402344,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522899270057678,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 432.74029541015625,
      "loss_g0": 431.5677490234375,
      "loss_g1": 502.5013427734375,
      "q_g0": 0.9834697842597961,
      "q_g1": 0.016530152410268784,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 402.9981689453125,
      "loss_g0": 401.5556945800781,
      "loss_g1": 488.77142333984375,
      "q_g0": 0.9834609031677246,
      "q_g1": 0.0165390707552433,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 367.46539306640625,
      "loss_g0": 365.9322509765625,
      "loss_g1": 458.5762634277344,
      "q_g0": 0.9834514856338501,
      "q_g1": 0.016548551619052887,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 364.12432861328125,
      "loss_g0": 362.8626403808594,
      "loss_g1": 439.06781005859375,
      "q_g0": 0.9834436178207397,
      "q_g1": 0.016556359827518463,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 362.1487121582031,
      "loss_g0": 362.9393615722656,
      "loss_g1": 315.1707763671875,
      "q_g0": 0.9834485650062561,
      "q_g1": 0.016551461070775986,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 467.0845642089844,
      "loss_g0": 468.51544189453125,
      "loss_g1": 382.01910400390625,
      "q_g0": 0.9834573268890381,
      "q_g1": 0.016542606055736542,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 259.2344055175781,
      "loss_g0": 258.2372741699219,
      "loss_g1": 318.4908752441406,
      "q_g0": 0.983451247215271,
      "q_g1": 0.016548769548535347,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 428.8324279785156,
      "loss_g0": 431.1620788574219,
      "loss_g1": 290.263427734375,
      "q_g0": 0.9834656715393066,
      "q_g1": 0.01653434708714485,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 279.26715087890625,
      "loss_g0": 277.64788818359375,
      "loss_g1": 375.52252197265625,
      "q_g0": 0.9834555983543396,
      "q_g1": 0.016544360667467117,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 344.5010681152344,
      "loss_g0": 343.639404296875,
      "loss_g1": 395.7047119140625,
      "q_g0": 0.9834502935409546,
      "q_g1": 0.016549693420529366,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 303.8619689941406,
      "loss_g0": 302.64398193359375,
      "loss_g1": 376.2067565917969,
      "q_g0": 0.9834428429603577,
      "q_g1": 0.016557229682803154,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 303.8081359863281,
      "loss_g0": 302.98345947265625,
      "loss_g1": 352.77716064453125,
      "q_g0": 0.9834376573562622,
      "q_g1": 0.016562331467866898,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 361.171142578125,
      "loss_g0": 361.7413635253906,
      "loss_g1": 327.3040466308594,
      "q_g0": 0.9834412336349487,
      "q_g1": 0.016558798030018806,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 276.5008544921875,
      "loss_g0": 276.35302734375,
      "loss_g1": 285.28009033203125,
      "q_g0": 0.9834402799606323,
      "q_g1": 0.016559714451432228,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 268.3480529785156,
      "loss_g0": 266.8402099609375,
      "loss_g1": 357.84490966796875,
      "q_g0": 0.9834309816360474,
      "q_g1": 0.01656903512775898,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 374.8141784667969,
      "loss_g0": 375.9325866699219,
      "loss_g1": 308.4033203125,
      "q_g0": 0.9834378957748413,
      "q_g1": 0.016562113538384438,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 264.44549560546875,
      "loss_g0": 263.2358093261719,
      "loss_g1": 336.2427978515625,
      "q_g0": 0.9834303855895996,
      "q_g1": 0.01656959392130375,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 304.1418151855469,
      "loss_g0": 303.89715576171875,
      "loss_g1": 318.6616516113281,
      "q_g0": 0.9834288954734802,
      "q_g1": 0.016571100801229477,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 340.6588439941406,
      "loss_g0": 341.20379638671875,
      "loss_g1": 308.310302734375,
      "q_g0": 0.9834322333335876,
      "q_g1": 0.01656772755086422,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 207.95179748535156,
      "loss_g0": 207.2183074951172,
      "loss_g1": 251.4789276123047,
      "q_g0": 0.9834277033805847,
      "q_g1": 0.016572266817092896,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 283.394775390625,
      "loss_g0": 283.69549560546875,
      "loss_g1": 265.54803466796875,
      "q_g0": 0.9834295511245728,
      "q_g1": 0.016570409759879112,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 250.62306213378906,
      "loss_g0": 250.420166015625,
      "loss_g1": 262.6634521484375,
      "q_g0": 0.9834282994270325,
      "q_g1": 0.016571668907999992,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 261.682373046875,
      "loss_g0": 260.1217041015625,
      "loss_g1": 354.2427978515625,
      "q_g0": 0.9834187030792236,
      "q_g1": 0.01658131740987301,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 338.826904296875,
      "loss_g0": 0.0,
      "loss_g1": 338.826904296875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 274.9100036621094,
      "loss_g0": 256.93719482421875,
      "loss_g1": 275.2119140625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 323.7916259765625,
      "loss_g0": 233.890869140625,
      "loss_g1": 325.3019714355469,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 246.27621459960938,
      "loss_g0": 222.9558563232422,
      "loss_g1": 246.66798400878906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 247.4562530517578,
      "loss_g0": 358.943115234375,
      "loss_g1": 245.58192443847656,
      "q_g0": 0.01653401367366314,
      "q_g1": 0.9834660291671753,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 250.17201232910156,
      "loss_g0": 251.6294403076172,
      "loss_g1": 250.14752197265625,
      "q_g0": 0.016534162685275078,
      "q_g1": 0.9834659099578857,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 255.45729064941406,
      "loss_g0": 249.83143615722656,
      "loss_g1": 255.55184936523438,
      "q_g0": 0.016533581539988518,
      "q_g1": 0.9834664463996887,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 165.2566375732422,
      "loss_g0": 195.27462768554688,
      "loss_g1": 164.7518768310547,
      "q_g0": 0.016536705195903778,
      "q_g1": 0.9834633469581604,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 212.48074340820312,
      "loss_g0": 153.83656311035156,
      "loss_g1": 213.46646118164062,
      "q_g0": 0.016530603170394897,
      "q_g1": 0.9834693670272827,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 223.19830322265625,
      "loss_g0": 0.0,
      "loss_g1": 223.19830322265625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 178.59873962402344,
      "loss_g0": 211.36923217773438,
      "loss_g1": 178.04820251464844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 318.6204833984375,
      "loss_g0": 165.04351806640625,
      "loss_g1": 321.2005615234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 142.91419982910156,
      "loss_g0": 123.46942138671875,
      "loss_g1": 143.24087524414062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 176.17889404296875,
      "loss_g0": 176.17889404296875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 220.53280639648438,
      "loss_g0": 220.53280639648438,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 121.88101959228516,
      "loss_g0": 121.88101959228516,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 134.9547576904297,
      "loss_g0": 134.96853637695312,
      "loss_g1": 134.1346435546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 267.5787048339844,
      "loss_g0": 268.7238464355469,
      "loss_g1": 199.4153289794922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 727.3700561523438,
      "loss_g0": 726.5535888671875,
      "loss_g1": 775.9568481445312,
      "q_g0": 0.9834725856781006,
      "q_g1": 0.016527481377124786,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 701.6986694335938,
      "loss_g0": 702.923095703125,
      "loss_g1": 628.8143920898438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 642.2921142578125,
      "loss_g0": 644.130615234375,
      "loss_g1": 532.854736328125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 602.6213989257812,
      "loss_g0": 602.380615234375,
      "loss_g1": 616.9511108398438,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.01652391441166401,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 633.6068115234375,
      "loss_g0": 634.6959228515625,
      "loss_g1": 568.7787475585938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 579.5007934570312,
      "loss_g0": 579.4663696289062,
      "loss_g1": 581.5471801757812,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 575.64990234375,
      "loss_g0": 575.9910888671875,
      "loss_g1": 555.34228515625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 513.305419921875,
      "loss_g0": 512.9674072265625,
      "loss_g1": 533.4237060546875,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524510458111763,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 398.9029235839844,
      "loss_g0": 396.59698486328125,
      "loss_g1": 536.0220947265625,
      "q_g0": 0.9834612607955933,
      "q_g1": 0.016538767144083977,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 373.5898742675781,
      "loss_g0": 372.0798645019531,
      "loss_g1": 463.3304138183594,
      "q_g0": 0.9834519624710083,
      "q_g1": 0.01654810830950737,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 485.6070251464844,
      "loss_g0": 484.4641418457031,
      "loss_g1": 553.4987182617188,
      "q_g0": 0.9834448099136353,
      "q_g1": 0.016555169597268105,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 407.368408203125,
      "loss_g0": 408.05975341796875,
      "loss_g1": 366.2880859375,
      "q_g0": 0.9834490418434143,
      "q_g1": 0.01655089296400547,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 514.5484008789062,
      "loss_g0": 516.5440063476562,
      "loss_g1": 395.87896728515625,
      "q_g0": 0.9834614992141724,
      "q_g1": 0.016538551077246666,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 438.766357421875,
      "loss_g0": 439.0086364746094,
      "loss_g1": 424.3582763671875,
      "q_g0": 0.9834628701210022,
      "q_g1": 0.016537051647901535,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 370.38323974609375,
      "loss_g0": 370.76434326171875,
      "loss_g1": 347.7149353027344,
      "q_g0": 0.983465313911438,
      "q_g1": 0.016534697264432907,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 411.1316223144531,
      "loss_g0": 411.1316223144531,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 386.6917419433594,
      "loss_g0": 385.69427490234375,
      "loss_g1": 446.0638427734375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 421.462158203125,
      "loss_g0": 422.2831726074219,
      "loss_g1": 372.59173583984375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 383.49713134765625,
      "loss_g0": 382.4112548828125,
      "loss_g1": 448.1061096191406,
      "q_g0": 0.9834708571434021,
      "q_g1": 0.016529137268662453,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 350.0870666503906,
      "loss_g0": 350.48016357421875,
      "loss_g1": 326.6936950683594,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.016526702791452408,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 327.0751037597656,
      "loss_g0": 326.0186767578125,
      "loss_g1": 389.91619873046875,
      "q_g0": 0.9834668040275574,
      "q_g1": 0.016533231362700462,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 471.90106201171875,
      "loss_g0": 473.6016845703125,
      "loss_g1": 370.6759033203125,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522705554962158,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 304.11212158203125,
      "loss_g0": 303.3712463378906,
      "loss_g1": 348.20037841796875,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527283936738968,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 369.95977783203125,
      "loss_g0": 0.0,
      "loss_g1": 369.95977783203125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 413.82183837890625,
      "loss_g0": 300.3157653808594,
      "loss_g1": 415.7287292480469,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 386.06231689453125,
      "loss_g0": 360.0744323730469,
      "loss_g1": 386.4989013671875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 336.558837890625,
      "loss_g0": 355.5538330078125,
      "loss_g1": 336.23968505859375,
      "q_g0": 0.01652439311146736,
      "q_g1": 0.9834755659103394,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 355.76470947265625,
      "loss_g0": 366.53302001953125,
      "loss_g1": 355.5837707519531,
      "q_g0": 0.0165255106985569,
      "q_g1": 0.9834745526313782,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 345.14263916015625,
      "loss_g0": 262.3846435546875,
      "loss_g1": 346.532958984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 264.4140319824219,
      "loss_g0": 264.4140319824219,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 285.7432861328125,
      "loss_g0": 285.25634765625,
      "loss_g1": 314.7285461425781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 291.8882141113281,
      "loss_g0": 291.867919921875,
      "loss_g1": 293.0950927734375,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 301.53826904296875,
      "loss_g0": 302.55487060546875,
      "loss_g1": 241.02500915527344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 247.29034423828125,
      "loss_g0": 245.9439697265625,
      "loss_g1": 327.39105224609375,
      "q_g0": 0.9834692478179932,
      "q_g1": 0.016530748456716537,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 229.29042053222656,
      "loss_g0": 229.23167419433594,
      "loss_g1": 232.78561401367188,
      "q_g0": 0.9834688901901245,
      "q_g1": 0.01653110608458519,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 300.35748291015625,
      "loss_g0": 0.0,
      "loss_g1": 300.35748291015625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 222.68348693847656,
      "loss_g0": 162.990966796875,
      "loss_g1": 223.68630981445312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 270.028076171875,
      "loss_g0": 410.95135498046875,
      "loss_g1": 267.658447265625,
      "q_g0": 0.016537075862288475,
      "q_g1": 0.9834628701210022,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 199.41543579101562,
      "loss_g0": 200.27633666992188,
      "loss_g1": 199.4009552001953,
      "q_g0": 0.016537168994545937,
      "q_g1": 0.9834627509117126,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 264.25726318359375,
      "loss_g0": 264.07086181640625,
      "loss_g1": 264.2604064941406,
      "q_g0": 0.016537148505449295,
      "q_g1": 0.9834628701210022,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 225.36106872558594,
      "loss_g0": 213.32357788085938,
      "loss_g1": 225.5634765625,
      "q_g0": 0.016535891219973564,
      "q_g1": 0.9834641814231873,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 308.3072814941406,
      "loss_g0": 182.22471618652344,
      "loss_g1": 310.4255065917969,
      "q_g0": 0.016522781923413277,
      "q_g1": 0.9834771752357483,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 195.96707153320312,
      "loss_g0": 186.7777862548828,
      "loss_g1": 196.12144470214844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 185.1635284423828,
      "loss_g0": 298.08038330078125,
      "loss_g1": 183.26516723632812,
      "q_g0": 0.016534162685275078,
      "q_g1": 0.9834659099578857,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 214.4371795654297,
      "loss_g0": 142.44961547851562,
      "loss_g1": 215.6468963623047,
      "q_g0": 0.016526680439710617,
      "q_g1": 0.9834732413291931,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 176.56597900390625,
      "loss_g0": 132.82443237304688,
      "loss_g1": 177.30084228515625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 157.27540588378906,
      "loss_g0": 157.27540588378906,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 153.67221069335938,
      "loss_g0": 153.74575805664062,
      "loss_g1": 149.2946014404297,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 118.03339385986328,
      "loss_g0": 117.28962707519531,
      "loss_g1": 162.29299926757812,
      "q_g0": 0.9834730625152588,
      "q_g1": 0.016527023166418076,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 103.42955017089844,
      "loss_g0": 102.80155181884766,
      "loss_g1": 140.79092407226562,
      "q_g0": 0.9834691286087036,
      "q_g1": 0.016530906781554222,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 769.5625610351562,
      "loss_g0": 769.1915283203125,
      "loss_g1": 791.6339111328125,
      "q_g0": 0.9834668040275574,
      "q_g1": 0.01653320901095867,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 724.6134643554688,
      "loss_g0": 724.46142578125,
      "loss_g1": 733.6586303710938,
      "q_g0": 0.9834659099578857,
      "q_g1": 0.01653415523469448,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 613.4983520507812,
      "loss_g0": 614.7271728515625,
      "loss_g1": 540.3723754882812,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.01652655005455017,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 566.5775146484375,
      "loss_g0": 564.2474365234375,
      "loss_g1": 705.1135864257812,
      "q_g0": 0.9834590554237366,
      "q_g1": 0.016540955752134323,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 494.02557373046875,
      "loss_g0": 493.3194274902344,
      "loss_g1": 535.9972534179688,
      "q_g0": 0.983454704284668,
      "q_g1": 0.01654532551765442,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 555.985107421875,
      "loss_g0": 555.985107421875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 497.7013244628906,
      "loss_g0": 497.79168701171875,
      "loss_g1": 492.3235778808594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 512.3469848632812,
      "loss_g0": 512.4411010742188,
      "loss_g1": 506.7432556152344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 424.7759704589844,
      "loss_g0": 422.8282775878906,
      "loss_g1": 540.6246337890625,
      "q_g0": 0.9834655523300171,
      "q_g1": 0.016534464433789253,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 558.6041870117188,
      "loss_g0": 559.0486450195312,
      "loss_g1": 532.1605224609375,
      "q_g0": 0.9834682941436768,
      "q_g1": 0.016531718894839287,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 446.50244140625,
      "loss_g0": 446.0747985839844,
      "loss_g1": 471.937744140625,
      "q_g0": 0.9834656715393066,
      "q_g1": 0.016534363850951195,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 539.1517333984375,
      "loss_g0": 541.043212890625,
      "loss_g1": 426.5641174316406,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522657126188278,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 453.9500732421875,
      "loss_g0": 0.0,
      "loss_g1": 453.9500732421875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 407.5539245605469,
      "loss_g0": 0.0,
      "loss_g1": 407.5539245605469,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 514.2056884765625,
      "loss_g0": 0.0,
      "loss_g1": 514.2056884765625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 412.9114074707031,
      "loss_g0": 475.81463623046875,
      "loss_g1": 411.8546447753906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 474.1258239746094,
      "loss_g0": 470.50823974609375,
      "loss_g1": 474.1866149902344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 472.8420715332031,
      "loss_g0": 339.7091369628906,
      "loss_g1": 475.0787048339844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 347.80078125,
      "loss_g0": 332.5692138671875,
      "loss_g1": 348.0566711425781,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 298.7611083984375,
      "loss_g0": 402.38348388671875,
      "loss_g1": 297.01910400390625,
      "q_g0": 0.016533199697732925,
      "q_g1": 0.9834668040275574,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 379.2774963378906,
      "loss_g0": 408.9423828125,
      "loss_g1": 378.7787170410156,
      "q_g0": 0.016536278650164604,
      "q_g1": 0.983463704586029,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 359.52459716796875,
      "loss_g0": 352.0433349609375,
      "loss_g1": 359.650390625,
      "q_g0": 0.016535503789782524,
      "q_g1": 0.9834644794464111,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 413.54522705078125,
      "loss_g0": 352.020263671875,
      "loss_g1": 414.57928466796875,
      "q_g0": 0.016529114916920662,
      "q_g1": 0.9834708571434021,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 334.2294006347656,
      "loss_g0": 255.81094360351562,
      "loss_g1": 335.54681396484375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 365.2358703613281,
      "loss_g0": 365.2358703613281,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 286.9700622558594,
      "loss_g0": 286.0857849121094,
      "loss_g1": 339.60565185546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 262.67547607421875,
      "loss_g0": 260.41748046875,
      "loss_g1": 396.9646911621094,
      "q_g0": 0.9834635853767395,
      "q_g1": 0.016536379233002663,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 367.0858459472656,
      "loss_g0": 368.3448486328125,
      "loss_g1": 292.1744384765625,
      "q_g0": 0.9834713935852051,
      "q_g1": 0.01652858778834343,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 331.98486328125,
      "loss_g0": 330.0692138671875,
      "loss_g1": 445.8847351074219,
      "q_g0": 0.9834595322608948,
      "q_g1": 0.01654043421149254,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 246.59707641601562,
      "loss_g0": 246.59707641601562,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 270.1338195800781,
      "loss_g0": 269.06121826171875,
      "loss_g1": 333.9787292480469,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 240.1836700439453,
      "loss_g0": 238.59262084960938,
      "loss_g1": 334.83172607421875,
      "q_g0": 0.9834677577018738,
      "q_g1": 0.016532262787222862,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 280.08294677734375,
      "loss_g0": 278.3835144042969,
      "loss_g1": 381.113037109375,
      "q_g0": 0.9834572672843933,
      "q_g1": 0.016542769968509674,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 256.1728210449219,
      "loss_g0": 255.0197296142578,
      "loss_g1": 324.6936340332031,
      "q_g0": 0.9834501147270203,
      "q_g1": 0.01654990203678608,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 206.66177368164062,
      "loss_g0": 206.66177368164062,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 225.84912109375,
      "loss_g0": 226.18235778808594,
      "loss_g1": 206.01364135742188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 266.1886901855469,
      "loss_g0": 267.59295654296875,
      "loss_g1": 182.59988403320312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 182.82229614257812,
      "loss_g0": 181.53182983398438,
      "loss_g1": 259.5977478027344,
      "q_g0": 0.9834696054458618,
      "q_g1": 0.01653040200471878,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 217.76905822753906,
      "loss_g0": 217.51426696777344,
      "loss_g1": 232.92626953125,
      "q_g0": 0.9834679365158081,
      "q_g1": 0.016531983390450478,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 211.14610290527344,
      "loss_g0": 207.29937744140625,
      "loss_g1": 439.64959716796875,
      "q_g0": 0.9834442138671875,
      "q_g1": 0.01655576191842556,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 298.917236328125,
      "loss_g0": 298.839111328125,
      "loss_g1": 303.5586242675781,
      "q_g0": 0.9834437370300293,
      "q_g1": 0.01655624248087406,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 225.15049743652344,
      "loss_g0": 224.88624572753906,
      "loss_g1": 240.84498596191406,
      "q_g0": 0.9834421277046204,
      "q_g1": 0.016557874158024788,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 280.5535888671875,
      "loss_g0": 280.7325439453125,
      "loss_g1": 269.9244079589844,
      "q_g0": 0.9834432005882263,
      "q_g1": 0.016556769609451294,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 290.01171875,
      "loss_g0": 291.1016845703125,
      "loss_g1": 225.24322509765625,
      "q_g0": 0.9834499955177307,
      "q_g1": 0.016550026834011078,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 158.16567993164062,
      "loss_g0": 158.008056640625,
      "loss_g1": 167.53163146972656,
      "q_g0": 0.9834490418434143,
      "q_g1": 0.016551004722714424,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 119.39924621582031,
      "loss_g0": 117.83759307861328,
      "loss_g1": 212.13619995117188,
      "q_g0": 0.9834393858909607,
      "q_g1": 0.016560662537813187,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 172.33065795898438,
      "loss_g0": 172.07449340820312,
      "loss_g1": 187.54058837890625,
      "q_g0": 0.9834377765655518,
      "q_g1": 0.016562245786190033,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 152.50804138183594,
      "loss_g0": 150.91122436523438,
      "loss_g1": 247.26620483398438,
      "q_g0": 0.9834278225898743,
      "q_g1": 0.016572127118706703,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 106.14502716064453,
      "loss_g0": 105.05241394042969,
      "loss_g1": 170.95591735839844,
      "q_g0": 0.9834211468696594,
      "q_g1": 0.016578882932662964,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 117.79900360107422,
      "loss_g0": 116.70451354980469,
      "loss_g1": 182.69508361816406,
      "q_g0": 0.983414351940155,
      "q_g1": 0.01658564805984497,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 831.1178588867188,
      "loss_g0": 827.5265502929688,
      "loss_g1": 1043.7664794921875,
      "q_g0": 0.9833921194076538,
      "q_g1": 0.016607843339443207,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 795.6911010742188,
      "loss_g0": 0.0,
      "loss_g1": 795.6911010742188,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 739.7791137695312,
      "loss_g0": 833.577880859375,
      "loss_g1": 738.2033081054688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 600.85791015625,
      "loss_g0": 708.49609375,
      "loss_g1": 599.04833984375,
      "q_g0": 0.016533611342310905,
      "q_g1": 0.9834663271903992,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 627.2825317382812,
      "loss_g0": 636.2421875,
      "loss_g1": 627.1319580078125,
      "q_g0": 0.01653454825282097,
      "q_g1": 0.9834654331207275,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 691.3954467773438,
      "loss_g0": 637.2118530273438,
      "loss_g1": 692.3060913085938,
      "q_g0": 0.016528913751244545,
      "q_g1": 0.9834710955619812,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 550.6683349609375,
      "loss_g0": 559.8716430664062,
      "loss_g1": 550.513671875,
      "q_g0": 0.01652986742556095,
      "q_g1": 0.9834701418876648,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 553.3907470703125,
      "loss_g0": 589.958251953125,
      "loss_g1": 552.7760009765625,
      "q_g0": 0.01653367467224598,
      "q_g1": 0.9834663271903992,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 478.2202453613281,
      "loss_g0": 0.0,
      "loss_g1": 478.2202453613281,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 491.773681640625,
      "loss_g0": 535.3741455078125,
      "loss_g1": 491.04119873046875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 413.7413330078125,
      "loss_g0": 401.523193359375,
      "loss_g1": 413.94659423828125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 459.65777587890625,
      "loss_g0": 0.0,
      "loss_g1": 459.65777587890625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 422.63690185546875,
      "loss_g0": 456.65728759765625,
      "loss_g1": 422.06536865234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 442.7475280761719,
      "loss_g0": 370.9709167480469,
      "loss_g1": 443.953369140625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 455.31512451171875,
      "loss_g0": 375.8495788574219,
      "loss_g1": 456.6501159667969,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 364.9946594238281,
      "loss_g0": 343.793701171875,
      "loss_g1": 365.350830078125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 356.62493896484375,
      "loss_g0": 407.1626892089844,
      "loss_g1": 355.7756042480469,
      "q_g0": 0.016527680680155754,
      "q_g1": 0.9834723472595215,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 454.71112060546875,
      "loss_g0": 355.71337890625,
      "loss_g1": 456.374267578125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 431.6068420410156,
      "loss_g0": 315.3074951171875,
      "loss_g1": 433.5606384277344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 363.6481018066406,
      "loss_g0": 389.46356201171875,
      "loss_g1": 363.2143249511719,
      "q_g0": 0.01652509905397892,
      "q_g1": 0.983474850654602,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 369.74951171875,
      "loss_g0": 0.0,
      "loss_g1": 369.74951171875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 324.85833740234375,
      "loss_g0": 311.26873779296875,
      "loss_g1": 325.0866394042969,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 327.17266845703125,
      "loss_g0": 251.56080627441406,
      "loss_g1": 328.44293212890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 363.2355041503906,
      "loss_g0": 294.29962158203125,
      "loss_g1": 364.39361572265625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 349.7669372558594,
      "loss_g0": 0.0,
      "loss_g1": 349.7669372558594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 339.1083984375,
      "loss_g0": 269.83514404296875,
      "loss_g1": 340.27215576171875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 248.92190551757812,
      "loss_g0": 330.7685241699219,
      "loss_g1": 247.546142578125,
      "q_g0": 0.016530929133296013,
      "q_g1": 0.9834691286087036,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 331.6563415527344,
      "loss_g0": 315.0745544433594,
      "loss_g1": 331.9350280761719,
      "q_g0": 0.016529200598597527,
      "q_g1": 0.9834708571434021,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 329.78521728515625,
      "loss_g0": 324.333251953125,
      "loss_g1": 329.8768310546875,
      "q_g0": 0.016528626903891563,
      "q_g1": 0.9834713935852051,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 354.3306884765625,
      "loss_g0": 0.0,
      "loss_g1": 354.3306884765625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 284.4513854980469,
      "loss_g0": 355.19781494140625,
      "loss_g1": 283.2628173828125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 272.82470703125,
      "loss_g0": 320.6712951660156,
      "loss_g1": 272.0206298828125,
      "q_g0": 0.016527393832802773,
      "q_g1": 0.9834725856781006,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 283.0792236328125,
      "loss_g0": 226.8118133544922,
      "loss_g1": 284.0245361328125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 205.39744567871094,
      "loss_g0": 205.39744567871094,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 186.07498168945312,
      "loss_g0": 183.61492919921875,
      "loss_g1": 332.506591796875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 175.70787048339844,
      "loss_g0": 173.614990234375,
      "loss_g1": 300.18536376953125,
      "q_g0": 0.9834645986557007,
      "q_g1": 0.016535362228751183,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 201.02142333984375,
      "loss_g0": 200.9957275390625,
      "loss_g1": 202.54953002929688,
      "q_g0": 0.9834644794464111,
      "q_g1": 0.016535526141524315,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 224.75653076171875,
      "loss_g0": 223.3950653076172,
      "loss_g1": 305.68829345703125,
      "q_g0": 0.9834560751914978,
      "q_g1": 0.016543950885534286,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 183.2860565185547,
      "loss_g0": 182.5641326904297,
      "loss_g1": 226.1888427734375,
      "q_g0": 0.9834516048431396,
      "q_g1": 0.01654841937124729,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 152.59349060058594,
      "loss_g0": 151.86483764648438,
      "loss_g1": 195.8848114013672,
      "q_g0": 0.9834470748901367,
      "q_g1": 0.01655292883515358,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 173.7306671142578,
      "loss_g0": 173.42822265625,
      "loss_g1": 191.69668579101562,
      "q_g0": 0.9834452271461487,
      "q_g1": 0.016554800793528557,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 183.05638122558594,
      "loss_g0": 181.21759033203125,
      "loss_g1": 292.21343994140625,
      "q_g0": 0.9834338426589966,
      "q_g1": 0.01656617596745491,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 146.17831420898438,
      "loss_g0": 142.69122314453125,
      "loss_g1": 352.9123229980469,
      "q_g0": 0.9834122657775879,
      "q_g1": 0.016587724909186363,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 210.96798706054688,
      "loss_g0": 0.0,
      "loss_g1": 210.96798706054688,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 237.69290161132812,
      "loss_g0": 198.8861541748047,
      "loss_g1": 238.3448486328125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 188.56109619140625,
      "loss_g0": 162.2191162109375,
      "loss_g1": 189.00363159179688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 151.61383056640625,
      "loss_g0": 203.66217041015625,
      "loss_g1": 150.7391357421875,
      "q_g0": 0.01652783527970314,
      "q_g1": 0.9834721088409424,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 218.38604736328125,
      "loss_g0": 113.91862487792969,
      "loss_g1": 220.14109802246094,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 195.068603515625,
      "loss_g0": 146.62400817871094,
      "loss_g1": 195.88247680664062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 179.82923889160156,
      "loss_g0": 182.11216735839844,
      "loss_g1": 179.7908935546875,
      "q_g0": 0.016522657126188278,
      "q_g1": 0.9834772944450378,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 930.4751586914062,
      "loss_g0": 844.2576904296875,
      "loss_g1": 931.9236450195312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 776.5352172851562,
      "loss_g0": 666.1983032226562,
      "loss_g1": 778.3888549804688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 722.4658203125,
      "loss_g0": 0.0,
      "loss_g1": 722.4658203125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 650.033935546875,
      "loss_g0": 841.086181640625,
      "loss_g1": 646.82421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 603.5912475585938,
      "loss_g0": 603.5912475585938,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 568.304443359375,
      "loss_g0": 568.1476440429688,
      "loss_g1": 577.63720703125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 515.1842041015625,
      "loss_g0": 513.3736572265625,
      "loss_g1": 622.8815307617188,
      "q_g0": 0.9834663271903992,
      "q_g1": 0.016533618792891502,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 546.9828491210938,
      "loss_g0": 547.9940185546875,
      "loss_g1": 486.81146240234375,
      "q_g0": 0.9834725856781006,
      "q_g1": 0.016527369618415833,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 493.4223327636719,
      "loss_g0": 490.9376220703125,
      "loss_g1": 641.1377563476562,
      "q_g0": 0.9834572672843933,
      "q_g1": 0.01654273085296154,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 495.02435302734375,
      "loss_g0": 495.02435302734375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 512.3856811523438,
      "loss_g0": 512.7218017578125,
      "loss_g1": 492.37811279296875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 543.5917358398438,
      "loss_g0": 545.037109375,
      "loss_g1": 457.5552978515625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 519.4313354492188,
      "loss_g0": 520.32421875,
      "loss_g1": 466.28369140625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 484.92803955078125,
      "loss_g0": 485.01971435546875,
      "loss_g1": 479.4716796875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 480.1915588378906,
      "loss_g0": 481.3652038574219,
      "loss_g1": 410.3301696777344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 420.2264099121094,
      "loss_g0": 419.8045654296875,
      "loss_g1": 445.33294677734375,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016525031998753548,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 417.2153625488281,
      "loss_g0": 417.73004150390625,
      "loss_g1": 386.5800476074219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 421.95849609375,
      "loss_g0": 0.0,
      "loss_g1": 421.95849609375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 376.6658935546875,
      "loss_g0": 333.97894287109375,
      "loss_g1": 377.383056640625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 361.6206970214844,
      "loss_g0": 361.6206970214844,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 316.64654541015625,
      "loss_g0": 316.05279541015625,
      "loss_g1": 351.9880065917969,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 365.1795349121094,
      "loss_g0": 365.6458435058594,
      "loss_g1": 337.423583984375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 396.82220458984375,
      "loss_g0": 395.5205078125,
      "loss_g1": 474.26641845703125,
      "q_g0": 0.9834696054458618,
      "q_g1": 0.01653047278523445,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 330.9158020019531,
      "loss_g0": 330.9158020019531,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 285.682373046875,
      "loss_g0": 284.20806884765625,
      "loss_g1": 373.43743896484375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 327.91876220703125,
      "loss_g0": 325.80059814453125,
      "loss_g1": 453.8984680175781,
      "q_g0": 0.9834644794464111,
      "q_g1": 0.016535520553588867,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 303.97344970703125,
      "loss_g0": 303.3401184082031,
      "loss_g1": 341.6321105957031,
      "q_g0": 0.9834606051445007,
      "q_g1": 0.01653943583369255,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 376.60711669921875,
      "loss_g0": 377.60797119140625,
      "loss_g1": 317.0718078613281,
      "q_g0": 0.9834668040275574,
      "q_g1": 0.016533248126506805,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 266.1001281738281,
      "loss_g0": 264.8598327636719,
      "loss_g1": 339.8443298339844,
      "q_g0": 0.9834590554237366,
      "q_g1": 0.016540924087166786,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 284.274658203125,
      "loss_g0": 283.63238525390625,
      "loss_g1": 322.45245361328125,
      "q_g0": 0.9834550619125366,
      "q_g1": 0.016544895246624947,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 259.10205078125,
      "loss_g0": 257.9297180175781,
      "loss_g1": 328.7547912597656,
      "q_g0": 0.9834479093551636,
      "q_g1": 0.016552144661545753,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 299.22625732421875,
      "loss_g0": 299.85723876953125,
      "loss_g1": 261.7294921875,
      "q_g0": 0.9834518432617188,
      "q_g1": 0.016548234969377518,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 248.2241973876953,
      "loss_g0": 247.0432891845703,
      "loss_g1": 318.3729248046875,
      "q_g0": 0.9834444522857666,
      "q_g1": 0.0165555439889431,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 228.4042205810547,
      "loss_g0": 227.26766967773438,
      "loss_g1": 295.8896179199219,
      "q_g0": 0.9834374189376831,
      "q_g1": 0.01656257174909115,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 297.5876770019531,
      "loss_g0": 297.1602478027344,
      "loss_g1": 322.9617919921875,
      "q_g0": 0.9834347367286682,
      "q_g1": 0.016565212979912758,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 252.7330322265625,
      "loss_g0": 251.79190063476562,
      "loss_g1": 308.58502197265625,
      "q_g0": 0.9834290146827698,
      "q_g1": 0.016571033746004105,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 241.98873901367188,
      "loss_g0": 241.1976318359375,
      "loss_g1": 288.9240417480469,
      "q_g0": 0.9834240078926086,
      "q_g1": 0.016575919464230537,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 186.42991638183594,
      "loss_g0": 184.86318969726562,
      "loss_g1": 279.3264465332031,
      "q_g0": 0.983414351940155,
      "q_g1": 0.016585607081651688,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 190.9445343017578,
      "loss_g0": 191.0756072998047,
      "loss_g1": 183.17344665527344,
      "q_g0": 0.9834151864051819,
      "q_g1": 0.01658480055630207,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 160.32846069335938,
      "loss_g0": 158.3782196044922,
      "loss_g1": 275.885498046875,
      "q_g0": 0.9834031462669373,
      "q_g1": 0.016596859320998192,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 190.22552490234375,
      "loss_g0": 188.78282165527344,
      "loss_g1": 275.66290283203125,
      "q_g0": 0.9833941459655762,
      "q_g1": 0.016605781391263008,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 145.73886108398438,
      "loss_g0": 144.7105712890625,
      "loss_g1": 206.61126708984375,
      "q_g0": 0.98338782787323,
      "q_g1": 0.016612142324447632,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 157.25274658203125,
      "loss_g0": 156.55372619628906,
      "loss_g1": 198.62188720703125,
      "q_g0": 0.9833835959434509,
      "q_g1": 0.016616467386484146,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 179.72512817382812,
      "loss_g0": 178.4757080078125,
      "loss_g1": 253.63182067871094,
      "q_g0": 0.9833758473396301,
      "q_g1": 0.016624197363853455,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 123.1568374633789,
      "loss_g0": 121.14618682861328,
      "loss_g1": 242.0032958984375,
      "q_g0": 0.9833633899688721,
      "q_g1": 0.016636621206998825,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 140.24575805664062,
      "loss_g0": 139.60653686523438,
      "loss_g1": 178.02059936523438,
      "q_g0": 0.9833594560623169,
      "q_g1": 0.016640570014715195,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 153.1612091064453,
      "loss_g0": 153.1612091064453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 160.2812957763672,
      "loss_g0": 158.74172973632812,
      "loss_g1": 251.9226837158203,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 131.5394287109375,
      "loss_g0": 131.34893798828125,
      "loss_g1": 142.87815856933594,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523605212569237,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 107.64259338378906,
      "loss_g0": 107.74305725097656,
      "loss_g1": 101.66249084472656,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522984951734543,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 665.0842895507812,
      "loss_g0": 661.385986328125,
      "loss_g1": 884.90234375,
      "q_g0": 0.9834541082382202,
      "q_g1": 0.016545843333005905,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 664.1000366210938,
      "loss_g0": 663.844970703125,
      "loss_g1": 679.259765625,
      "q_g0": 0.9834526181221008,
      "q_g1": 0.01654742658138275,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 674.4884033203125,
      "loss_g0": 675.5182495117188,
      "loss_g1": 613.257080078125,
      "q_g0": 0.9834589958190918,
      "q_g1": 0.01654105633497238,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 601.9395141601562,
      "loss_g0": 601.9283447265625,
      "loss_g1": 602.602783203125,
      "q_g0": 0.9834588766098022,
      "q_g1": 0.016541127115488052,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 475.6618957519531,
      "loss_g0": 474.98590087890625,
      "loss_g1": 515.8433227539062,
      "q_g0": 0.983454704284668,
      "q_g1": 0.016545316204428673,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 515.4909057617188,
      "loss_g0": 514.236572265625,
      "loss_g1": 590.0130615234375,
      "q_g0": 0.9834469556808472,
      "q_g1": 0.016553077846765518,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 545.5691528320312,
      "loss_g0": 545.5691528320312,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 562.2252197265625,
      "loss_g0": 563.3914794921875,
      "loss_g1": 492.80633544921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 506.7923278808594,
      "loss_g0": 506.5794982910156,
      "loss_g1": 519.4598999023438,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523735597729683,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 455.61383056640625,
      "loss_g0": 455.4303894042969,
      "loss_g1": 466.53009033203125,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524873673915863,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 507.0436096191406,
      "loss_g0": 509.17529296875,
      "loss_g1": 380.1579895019531,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 429.3552551269531,
      "loss_g0": 429.378662109375,
      "loss_g1": 427.9618225097656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 425.4788513183594,
      "loss_g0": 424.7139892578125,
      "loss_g1": 470.99334716796875,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.01652715355157852,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 400.95343017578125,
      "loss_g0": 399.50054931640625,
      "loss_g1": 487.3617248535156,
      "q_g0": 0.9834638237953186,
      "q_g1": 0.016536137089133263,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 464.3393859863281,
      "loss_g0": 463.89447021484375,
      "loss_g1": 490.794189453125,
      "q_g0": 0.9834611415863037,
      "q_g1": 0.016538893803954124,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 408.4835205078125,
      "loss_g0": 408.26312255859375,
      "loss_g1": 421.58856201171875,
      "q_g0": 0.9834597706794739,
      "q_g1": 0.01654025726020336,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 349.18621826171875,
      "loss_g0": 348.40081787109375,
      "loss_g1": 395.87054443359375,
      "q_g0": 0.9834549427032471,
      "q_g1": 0.016545115038752556,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 357.6671142578125,
      "loss_g0": 358.68133544921875,
      "loss_g1": 297.3594970703125,
      "q_g0": 0.9834611415863037,
      "q_g1": 0.01653883047401905,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 385.961181640625,
      "loss_g0": 0.0,
      "loss_g1": 385.961181640625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 295.1399230957031,
      "loss_g0": 340.3651123046875,
      "loss_g1": 294.380126953125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 390.88055419921875,
      "loss_g0": 316.69378662109375,
      "loss_g1": 392.12689208984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 352.7027282714844,
      "loss_g0": 257.3788146972656,
      "loss_g1": 354.30413818359375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 358.2464599609375,
      "loss_g0": 358.2464599609375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 328.78558349609375,
      "loss_g0": 326.9742431640625,
      "loss_g1": 436.6028747558594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 238.8531951904297,
      "loss_g0": 237.59365844726562,
      "loss_g1": 313.7895202636719,
      "q_g0": 0.9834697842597961,
      "q_g1": 0.01653021015226841,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 369.88092041015625,
      "loss_g0": 371.26605224609375,
      "loss_g1": 287.43310546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 350.4095458984375,
      "loss_g0": 351.69879150390625,
      "loss_g1": 273.66766357421875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 294.50927734375,
      "loss_g0": 293.236572265625,
      "loss_g1": 370.22857666015625,
      "q_g0": 0.9834696650505066,
      "q_g1": 0.016530293971300125,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 274.710693359375,
      "loss_g0": 273.754150390625,
      "loss_g1": 331.5989990234375,
      "q_g0": 0.9834638237953186,
      "q_g1": 0.016536207869648933,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 264.09033203125,
      "loss_g0": 263.62420654296875,
      "loss_g1": 291.8075866699219,
      "q_g0": 0.9834609031677246,
      "q_g1": 0.016539085656404495,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 279.41644287109375,
      "loss_g0": 279.865234375,
      "loss_g1": 252.72451782226562,
      "q_g0": 0.983463704586029,
      "q_g1": 0.016536308452486992,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 230.53305053710938,
      "loss_g0": 229.4210662841797,
      "loss_g1": 296.6390075683594,
      "q_g0": 0.9834567904472351,
      "q_g1": 0.01654318906366825,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 283.1271667480469,
      "loss_g0": 283.55181884765625,
      "loss_g1": 257.8797607421875,
      "q_g0": 0.9834594130516052,
      "q_g1": 0.016540560871362686,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 230.18751525878906,
      "loss_g0": 229.46847534179688,
      "loss_g1": 272.9291687011719,
      "q_g0": 0.9834550619125366,
      "q_g1": 0.0165450070053339,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 215.54698181152344,
      "loss_g0": 214.25457763671875,
      "loss_g1": 292.3313293457031,
      "q_g0": 0.9834469556808472,
      "q_g1": 0.0165529977530241,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 316.80950927734375,
      "loss_g0": 317.69647216796875,
      "loss_g1": 264.09454345703125,
      "q_g0": 0.9834524989128113,
      "q_g1": 0.016547512263059616,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 248.83193969726562,
      "loss_g0": 248.50613403320312,
      "loss_g1": 268.1928405761719,
      "q_g0": 0.9834504127502441,
      "q_g1": 0.016549527645111084,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 206.6115264892578,
      "loss_g0": 206.79107666015625,
      "loss_g1": 195.94039916992188,
      "q_g0": 0.9834516048431396,
      "q_g1": 0.01654842682182789,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 174.12652587890625,
      "loss_g0": 173.61767578125,
      "loss_g1": 204.36160278320312,
      "q_g0": 0.9834485054016113,
      "q_g1": 0.016551578417420387,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 284.17779541015625,
      "loss_g0": 285.1640625,
      "loss_g1": 225.55343627929688,
      "q_g0": 0.9834545850753784,
      "q_g1": 0.01654547080397606,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 232.2783966064453,
      "loss_g0": 233.1143798828125,
      "loss_g1": 182.57150268554688,
      "q_g0": 0.9834596514701843,
      "q_g1": 0.0165402889251709,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 200.83177185058594,
      "loss_g0": 200.83177185058594,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 157.05490112304688,
      "loss_g0": 155.87564086914062,
      "loss_g1": 227.24851989746094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 118.57107543945312,
      "loss_g0": 117.15214538574219,
      "loss_g1": 202.98574829101562,
      "q_g0": 0.983468770980835,
      "q_g1": 0.01653119921684265,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 179.7622833251953,
      "loss_g0": 180.5211639404297,
      "loss_g1": 134.6018524169922,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.016526510939002037,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 109.0697021484375,
      "loss_g0": 108.4856948852539,
      "loss_g1": 143.81546020507812,
      "q_g0": 0.9834699034690857,
      "q_g1": 0.016530122607946396,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 197.74261474609375,
      "loss_g0": 198.76596069335938,
      "loss_g1": 136.8345947265625,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652379147708416,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 127.61620330810547,
      "loss_g0": 126.00473022460938,
      "loss_g1": 223.47030639648438,
      "q_g0": 0.9834662079811096,
      "q_g1": 0.016533760353922844,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 128.13072204589844,
      "loss_g0": 126.53746032714844,
      "loss_g1": 222.84371948242188,
      "q_g0": 0.9834564328193665,
      "q_g1": 0.01654362492263317,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 156.646728515625,
      "loss_g0": 155.6932830810547,
      "loss_g1": 213.30459594726562,
      "q_g0": 0.9834504127502441,
      "q_g1": 0.016549520194530487,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 910.986572265625,
      "loss_g0": 913.2677001953125,
      "loss_g1": 775.3125610351562,
      "q_g0": 0.9834645986557007,
      "q_g1": 0.016535403206944466,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 655.3109130859375,
      "loss_g0": 655.5814208984375,
      "loss_g1": 639.222900390625,
      "q_g0": 0.9834662079811096,
      "q_g1": 0.016533726826310158,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 660.4705200195312,
      "loss_g0": 660.0816650390625,
      "loss_g1": 683.5955810546875,
      "q_g0": 0.9834638237953186,
      "q_g1": 0.016536129638552666,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 538.861572265625,
      "loss_g0": 0.0,
      "loss_g1": 538.861572265625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 572.6830444335938,
      "loss_g0": 589.021728515625,
      "loss_g1": 572.4085693359375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 546.7052612304688,
      "loss_g0": 692.9346313476562,
      "loss_g1": 544.246337890625,
      "q_g0": 0.016537629067897797,
      "q_g1": 0.9834624528884888,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 468.7213134765625,
      "loss_g0": 630.2340087890625,
      "loss_g1": 466.0025329589844,
      "q_g0": 0.016554435715079308,
      "q_g1": 0.9834455847740173,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 528.4486694335938,
      "loss_g0": 537.37451171875,
      "loss_g1": 528.2984008789062,
      "q_g0": 0.016555367037653923,
      "q_g1": 0.9834446907043457,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 387.2369689941406,
      "loss_g0": 465.74847412109375,
      "loss_g1": 385.9146423339844,
      "q_g0": 0.016563542187213898,
      "q_g1": 0.9834364652633667,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 451.56256103515625,
      "loss_g0": 470.71124267578125,
      "loss_g1": 451.2400207519531,
      "q_g0": 0.016565537080168724,
      "q_g1": 0.9834344387054443,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 466.34014892578125,
      "loss_g0": 439.61199951171875,
      "loss_g1": 466.7902526855469,
      "q_g0": 0.016562750563025475,
      "q_g1": 0.9834372997283936,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 489.0672912597656,
      "loss_g0": 419.18963623046875,
      "loss_g1": 490.24359130859375,
      "q_g0": 0.016555476933717728,
      "q_g1": 0.9834445714950562,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 533.9444580078125,
      "loss_g0": 493.88787841796875,
      "loss_g1": 534.6185913085938,
      "q_g0": 0.01655130460858345,
      "q_g1": 0.9834486842155457,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 446.64495849609375,
      "loss_g0": 411.372802734375,
      "loss_g1": 447.2384338378906,
      "q_g0": 0.016547635197639465,
      "q_g1": 0.9834523797035217,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 410.9867248535156,
      "loss_g0": 398.67047119140625,
      "loss_g1": 411.19390869140625,
      "q_g0": 0.016546355560421944,
      "q_g1": 0.9834536910057068,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 377.9122619628906,
      "loss_g0": 377.9122619628906,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 403.5576477050781,
      "loss_g0": 403.5576477050781,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 396.480224609375,
      "loss_g0": 396.8304138183594,
      "loss_g1": 375.63458251953125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 348.678466796875,
      "loss_g0": 347.82666015625,
      "loss_g1": 399.3634033203125,
      "q_g0": 0.9834723472595215,
      "q_g1": 0.0165276899933815,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 358.3023986816406,
      "loss_g0": 358.7206115722656,
      "loss_g1": 333.4122314453125,
      "q_g0": 0.983474850654602,
      "q_g1": 0.01652509905397892,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 378.1705017089844,
      "loss_g0": 0.0,
      "loss_g1": 378.1705017089844,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 391.0063171386719,
      "loss_g0": 295.85198974609375,
      "loss_g1": 392.60491943359375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 360.9023132324219,
      "loss_g0": 309.5520935058594,
      "loss_g1": 361.7649841308594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 325.8876953125,
      "loss_g0": 342.3872985839844,
      "loss_g1": 325.6104431152344,
      "q_g0": 0.01652413234114647,
      "q_g1": 0.983475923538208,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 310.6081237792969,
      "loss_g0": 344.13885498046875,
      "loss_g1": 310.04461669921875,
      "q_g0": 0.01652761735022068,
      "q_g1": 0.9834723472595215,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 292.2498779296875,
      "loss_g0": 318.9427490234375,
      "loss_g1": 291.80120849609375,
      "q_g0": 0.016530385240912437,
      "q_g1": 0.9834696054458618,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 327.9919738769531,
      "loss_g0": 292.453125,
      "loss_g1": 328.58917236328125,
      "q_g0": 0.016526687890291214,
      "q_g1": 0.9834732413291931,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 255.6027069091797,
      "loss_g0": 342.0876159667969,
      "loss_g1": 254.14859008789062,
      "q_g0": 0.0165356807410717,
      "q_g1": 0.9834643006324768,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 332.91668701171875,
      "loss_g0": 339.5120544433594,
      "loss_g1": 332.8057861328125,
      "q_g0": 0.016536369919776917,
      "q_g1": 0.9834635853767395,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 278.3798522949219,
      "loss_g0": 278.3798522949219,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 254.20704650878906,
      "loss_g0": 253.40692138671875,
      "loss_g1": 301.8330993652344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 330.427001953125,
      "loss_g0": 0.0,
      "loss_g1": 330.427001953125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 304.91387939453125,
      "loss_g0": 274.02593994140625,
      "loss_g1": 305.43280029296875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 294.7821044921875,
      "loss_g0": 173.38282775878906,
      "loss_g1": 296.82159423828125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 260.3771057128906,
      "loss_g0": 179.0311737060547,
      "loss_g1": 261.74371337890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 255.07254028320312,
      "loss_g0": 254.84857177734375,
      "loss_g1": 255.0762939453125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 267.7226257324219,
      "loss_g0": 214.97703552246094,
      "loss_g1": 268.6087341308594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 244.29635620117188,
      "loss_g0": 203.3719940185547,
      "loss_g1": 244.98388671875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 301.2017517089844,
      "loss_g0": 167.41143798828125,
      "loss_g1": 303.44940185546875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 228.185791015625,
      "loss_g0": 304.7732849121094,
      "loss_g1": 226.8984832763672,
      "q_g0": 0.016530385240912437,
      "q_g1": 0.9834696054458618,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 239.67820739746094,
      "loss_g0": 0.0,
      "loss_g1": 239.67820739746094,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 251.58370971679688,
      "loss_g0": 0.0,
      "loss_g1": 251.58370971679688,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 176.83413696289062,
      "loss_g0": 181.6644287109375,
      "loss_g1": 176.7529754638672,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 185.5489044189453,
      "loss_g0": 185.5489044189453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 217.73944091796875,
      "loss_g0": 218.1814422607422,
      "loss_g1": 191.4296112060547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 161.50897216796875,
      "loss_g0": 161.06578063964844,
      "loss_g1": 187.8845977783203,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525164246559143,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 167.5791015625,
      "loss_g0": 167.83551025390625,
      "loss_g1": 152.31802368164062,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.0165235735476017,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 236.01170349121094,
      "loss_g0": 237.24969482421875,
      "loss_g1": 162.32167053222656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 111.3128890991211,
      "loss_g0": 109.90682983398438,
      "loss_g1": 194.9617919921875,
      "q_g0": 0.9834688901901245,
      "q_g1": 0.016531122848391533,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 143.6080322265625,
      "loss_g0": 143.0946502685547,
      "loss_g1": 174.14382934570312,
      "q_g0": 0.9834656715393066,
      "q_g1": 0.016534293070435524,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 868.0515747070312,
      "loss_g0": 869.41943359375,
      "loss_g1": 786.6478271484375,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.01652582734823227,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 680.9321899414062,
      "loss_g0": 681.50634765625,
      "loss_g1": 646.7564697265625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 658.4449462890625,
      "loss_g0": 659.7140502929688,
      "loss_g1": 582.9047241210938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 535.2298583984375,
      "loss_g0": 536.7022094726562,
      "loss_g1": 447.5891418457031,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 433.86590576171875,
      "loss_g0": 434.0791931152344,
      "loss_g1": 421.1694641113281,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 423.8189392089844,
      "loss_g0": 423.6012878417969,
      "loss_g1": 436.77197265625,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652376726269722,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 335.7167663574219,
      "loss_g0": 334.295654296875,
      "loss_g1": 420.25311279296875,
      "q_g0": 0.9834674000740051,
      "q_g1": 0.016532564535737038,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 299.52752685546875,
      "loss_g0": 298.4677734375,
      "loss_g1": 362.54443359375,
      "q_g0": 0.9834609031677246,
      "q_g1": 0.016539117321372032,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 334.8163757324219,
      "loss_g0": 333.51983642578125,
      "loss_g1": 411.87420654296875,
      "q_g0": 0.9834528565406799,
      "q_g1": 0.01654713787138462,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 330.80987548828125,
      "loss_g0": 329.69146728515625,
      "loss_g1": 397.25244140625,
      "q_g0": 0.983445942401886,
      "q_g1": 0.016554053872823715,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 233.1768341064453,
      "loss_g0": 231.46829223632812,
      "loss_g1": 334.61224365234375,
      "q_g0": 0.983435332775116,
      "q_g1": 0.01656462252140045,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 345.0101013183594,
      "loss_g0": 0.0,
      "loss_g1": 345.0101013183594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 312.9724426269531,
      "loss_g0": 224.25071716308594,
      "loss_g1": 314.46295166015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 259.5816650390625,
      "loss_g0": 198.2857666015625,
      "loss_g1": 260.6114501953125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 234.6254119873047,
      "loss_g0": 177.9066162109375,
      "loss_g1": 235.57827758789062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 189.69410705566406,
      "loss_g0": 168.52474975585938,
      "loss_g1": 190.04974365234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 184.63926696777344,
      "loss_g0": 118.2716064453125,
      "loss_g1": 185.75424194335938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 112.48138427734375,
      "loss_g0": 112.48138427734375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 146.69732666015625,
      "loss_g0": 146.0112762451172,
      "loss_g1": 187.5337371826172,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "eval_cer": 0.8478282060279955,
      "eval_loss": 350.9228210449219,
      "eval_runtime": 86.9805,
      "eval_samples_per_second": 19.269,
      "eval_steps_per_second": 4.817,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 696.7216796875,
      "loss_g0": 696.876708984375,
      "loss_g1": 687.4938354492188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 0.5076142131979695,
      "gdro_loss": 752.7330932617188,
      "loss_g0": 753.269287109375,
      "loss_g1": 720.8133544921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 100
    },
    {
      "epoch": 1.015228426395939,
      "grad_norm": 1893.3450927734375,
      "learning_rate": 0.00043835294117647056,
      "loss": 329.8372,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 667.1250610351562,
      "loss_g0": 661.83544921875,
      "loss_g1": 667.2144775390625,
      "q_g0": 0.016619326546788216,
      "q_g1": 0.9833806753158569,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 490.06329345703125,
      "loss_g0": 501.96270751953125,
      "loss_g1": 489.8621826171875,
      "q_g0": 0.016620565205812454,
      "q_g1": 0.9833794236183167,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 489.4463806152344,
      "loss_g0": 520.3771362304688,
      "loss_g1": 488.92352294921875,
      "q_g0": 0.01662379316985607,
      "q_g1": 0.9833762049674988,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 459.6966857910156,
      "loss_g0": 380.0965881347656,
      "loss_g1": 461.0415954589844,
      "q_g0": 0.01661546900868416,
      "q_g1": 0.9833844900131226,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 462.72314453125,
      "loss_g0": 408.15277099609375,
      "loss_g1": 463.64483642578125,
      "q_g0": 0.016609767451882362,
      "q_g1": 0.9833902716636658,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 345.6661682128906,
      "loss_g0": 656.507568359375,
      "loss_g1": 340.405517578125,
      "q_g0": 0.016642270609736443,
      "q_g1": 0.9833577275276184,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 352.56890869140625,
      "loss_g0": 385.1308898925781,
      "loss_g1": 352.0177001953125,
      "q_g0": 0.016645679250359535,
      "q_g1": 0.9833542704582214,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 865.5009765625,
      "loss_g0": 388.46563720703125,
      "loss_g1": 873.5513305664062,
      "q_g0": 0.016595808789134026,
      "q_g1": 0.9834042191505432,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 311.26165771484375,
      "loss_g0": 321.5866394042969,
      "loss_g1": 311.08740234375,
      "q_g0": 0.016596883535385132,
      "q_g1": 0.9834031462669373,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 344.33111572265625,
      "loss_g0": 289.0765686035156,
      "loss_g1": 345.2633056640625,
      "q_g0": 0.016591116786003113,
      "q_g1": 0.9834089279174805,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 357.1861572265625,
      "loss_g0": 322.4949645996094,
      "loss_g1": 357.77130126953125,
      "q_g0": 0.01658749207854271,
      "q_g1": 0.983412504196167,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 324.1682434082031,
      "loss_g0": 333.6742858886719,
      "loss_g1": 324.0079040527344,
      "q_g0": 0.0165884867310524,
      "q_g1": 0.9834114909172058,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 335.07366943359375,
      "loss_g0": 372.74603271484375,
      "loss_g1": 334.43804931640625,
      "q_g0": 0.016592422500252724,
      "q_g1": 0.9834075570106506,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 337.0206298828125,
      "loss_g0": 281.3038024902344,
      "loss_g1": 337.96038818359375,
      "q_g0": 0.016586612910032272,
      "q_g1": 0.9834134578704834,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 278.71002197265625,
      "loss_g0": 278.71002197265625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 275.9187927246094,
      "loss_g0": 276.0162048339844,
      "loss_g1": 270.12005615234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 287.866943359375,
      "loss_g0": 288.1083679199219,
      "loss_g1": 273.4975891113281,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 252.98374938964844,
      "loss_g0": 252.31591796875,
      "loss_g1": 292.725341796875,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.01652655005455017,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 222.28883361816406,
      "loss_g0": 221.05941772460938,
      "loss_g1": 295.416748046875,
      "q_g0": 0.9834659099578857,
      "q_g1": 0.01653415523469448,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 274.3663330078125,
      "loss_g0": 0.0,
      "loss_g1": 274.3663330078125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 236.3671112060547,
      "loss_g0": 250.93194580078125,
      "loss_g1": 236.12240600585938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 252.2732391357422,
      "loss_g0": 237.535888671875,
      "loss_g1": 252.5208282470703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 260.4516906738281,
      "loss_g0": 211.59344482421875,
      "loss_g1": 261.27252197265625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 266.4856872558594,
      "loss_g0": 203.96929931640625,
      "loss_g1": 267.5359802246094,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 261.1211242675781,
      "loss_g0": 333.4659729003906,
      "loss_g1": 259.9051818847656,
      "q_g0": 0.01652994379401207,
      "q_g1": 0.9834700226783752,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 262.1022033691406,
      "loss_g0": 0.0,
      "loss_g1": 262.1022033691406,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 243.93557739257812,
      "loss_g0": 209.94525146484375,
      "loss_g1": 244.50660705566406,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 239.68963623046875,
      "loss_g0": 239.3590087890625,
      "loss_g1": 239.6951904296875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 201.924072265625,
      "loss_g0": 227.2102508544922,
      "loss_g1": 201.49917602539062,
      "q_g0": 0.01652505435049534,
      "q_g1": 0.9834749698638916,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 230.74313354492188,
      "loss_g0": 191.6842498779297,
      "loss_g1": 231.39932250976562,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 199.97879028320312,
      "loss_g0": 230.84788513183594,
      "loss_g1": 199.46009826660156,
      "q_g0": 0.0165256280452013,
      "q_g1": 0.9834744334220886,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 325.5876770019531,
      "loss_g0": 208.27804565429688,
      "loss_g1": 327.5584716796875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 214.53111267089844,
      "loss_g0": 176.3232879638672,
      "loss_g1": 215.17300415039062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 241.8583221435547,
      "loss_g0": 174.59817504882812,
      "loss_g1": 242.98829650878906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 207.63003540039062,
      "loss_g0": 0.0,
      "loss_g1": 207.63003540039062,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 379.0102233886719,
      "loss_g0": 0.0,
      "loss_g1": 379.0102233886719,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 159.0234375,
      "loss_g0": 171.12786865234375,
      "loss_g1": 158.820068359375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 153.28280639648438,
      "loss_g0": 153.28280639648438,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 419.5208435058594,
      "loss_g0": 422.26873779296875,
      "loss_g1": 255.95584106445312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 147.4369354248047,
      "loss_g0": 147.34295654296875,
      "loss_g1": 153.03082275390625,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523005440831184,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 381.3362121582031,
      "loss_g0": 385.0910339355469,
      "loss_g1": 157.83425903320312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 238.52499389648438,
      "loss_g0": 238.52499389648438,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 108.67803192138672,
      "loss_g0": 106.94454193115234,
      "loss_g1": 211.862060546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 133.612060546875,
      "loss_g0": 133.612060546875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 212.03961181640625,
      "loss_g0": 213.1772918701172,
      "loss_g1": 144.32000732421875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 137.38055419921875,
      "loss_g0": 137.48837280273438,
      "loss_g1": 130.96231079101562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 103.52295684814453,
      "loss_g0": 103.31454467773438,
      "loss_g1": 115.92729187011719,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523711383342743,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 240.63302612304688,
      "loss_g0": 240.63302612304688,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 176.34454345703125,
      "loss_g0": 177.17965698242188,
      "loss_g1": 126.63467407226562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 142.17752075195312,
      "loss_g0": 142.99168395996094,
      "loss_g1": 93.7151107788086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 530.2919311523438,
      "loss_g0": 528.5516967773438,
      "loss_g1": 633.8099365234375,
      "q_g0": 0.9834668040275574,
      "q_g1": 0.016533182933926582,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 643.0842895507812,
      "loss_g0": 644.4681396484375,
      "loss_g1": 560.7251586914062,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524620354175568,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 449.26947021484375,
      "loss_g0": 448.70220947265625,
      "loss_g1": 483.02337646484375,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.01652812957763672,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 515.34228515625,
      "loss_g0": 515.34228515625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 466.38818359375,
      "loss_g0": 0.0,
      "loss_g1": 466.38818359375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 411.3288879394531,
      "loss_g0": 357.94024658203125,
      "loss_g1": 412.225830078125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 714.6281127929688,
      "loss_g0": 381.020751953125,
      "loss_g1": 720.232666015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 439.4106750488281,
      "loss_g0": 414.9486083984375,
      "loss_g1": 439.8216247558594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 448.04974365234375,
      "loss_g0": 361.1172790527344,
      "loss_g1": 449.5102233886719,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 454.87982177734375,
      "loss_g0": 382.685791015625,
      "loss_g1": 456.0926513671875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 350.1756896972656,
      "loss_g0": 420.2051086425781,
      "loss_g1": 348.9986572265625,
      "q_g0": 0.01652970165014267,
      "q_g1": 0.9834702610969543,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 355.84588623046875,
      "loss_g0": 402.267822265625,
      "loss_g1": 355.0653991699219,
      "q_g0": 0.01653452403843403,
      "q_g1": 0.9834654331207275,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 324.066162109375,
      "loss_g0": 388.09173583984375,
      "loss_g1": 322.9892883300781,
      "q_g0": 0.01654118113219738,
      "q_g1": 0.9834588766098022,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 361.6521911621094,
      "loss_g0": 308.8765563964844,
      "loss_g1": 362.53955078125,
      "q_g0": 0.016535688191652298,
      "q_g1": 0.9834643006324768,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 346.08270263671875,
      "loss_g0": 0.0,
      "loss_g1": 346.08270263671875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 309.59722900390625,
      "loss_g0": 309.59722900390625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 370.98565673828125,
      "loss_g0": 372.08123779296875,
      "loss_g1": 305.7722473144531,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 268.7102355957031,
      "loss_g0": 268.48297119140625,
      "loss_g1": 282.23736572265625,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523828729987144,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 458.13726806640625,
      "loss_g0": 460.6187438964844,
      "loss_g1": 310.43109130859375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 301.8494873046875,
      "loss_g0": 303.19903564453125,
      "loss_g1": 221.5197296142578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 251.95013427734375,
      "loss_g0": 251.69113159179688,
      "loss_g1": 267.36688232421875,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524024307727814,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 231.56378173828125,
      "loss_g0": 227.28237915039062,
      "loss_g1": 485.96978759765625,
      "q_g0": 0.9834495186805725,
      "q_g1": 0.01655048318207264,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 237.6742706298828,
      "loss_g0": 238.0439910888672,
      "loss_g1": 215.7015838623047,
      "q_g0": 0.9834518432617188,
      "q_g1": 0.016548193991184235,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 267.54290771484375,
      "loss_g0": 0.0,
      "loss_g1": 267.54290771484375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 246.19007873535156,
      "loss_g0": 279.2155456542969,
      "loss_g1": 245.63525390625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 245.40951538085938,
      "loss_g0": 368.6183776855469,
      "loss_g1": 243.33798217773438,
      "q_g0": 0.016535231843590736,
      "q_g1": 0.9834647178649902,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 254.4031219482422,
      "loss_g0": 235.86764526367188,
      "loss_g1": 254.7147216796875,
      "q_g0": 0.016533300280570984,
      "q_g1": 0.9834666848182678,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 250.40565490722656,
      "loss_g0": 0.0,
      "loss_g1": 250.40565490722656,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 201.40628051757812,
      "loss_g0": 200.4066619873047,
      "loss_g1": 201.42306518554688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 243.5652618408203,
      "loss_g0": 243.90037536621094,
      "loss_g1": 243.55963134765625,
      "q_g0": 0.01652245782315731,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 185.9023895263672,
      "loss_g0": 185.9023895263672,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 227.259765625,
      "loss_g0": 221.9259490966797,
      "loss_g1": 544.7486572265625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 183.48330688476562,
      "loss_g0": 182.88101196289062,
      "loss_g1": 219.326416015625,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.016526145860552788,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 166.23379516601562,
      "loss_g0": 163.7824249267578,
      "loss_g1": 311.9794006347656,
      "q_g0": 0.9834587574005127,
      "q_g1": 0.01654130592942238,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 332.0980529785156,
      "loss_g0": 334.02783203125,
      "loss_g1": 217.2800750732422,
      "q_g0": 0.983470618724823,
      "q_g1": 0.016529355198144913,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 273.98480224609375,
      "loss_g0": 274.88690185546875,
      "loss_g1": 220.29208374023438,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523776575922966,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 169.13739013671875,
      "loss_g0": 169.0498046875,
      "loss_g1": 174.34963989257812,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524318605661392,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 164.057861328125,
      "loss_g0": 164.02786254882812,
      "loss_g1": 165.84303283691406,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524510458111763,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 144.58753967285156,
      "loss_g0": 144.28834533691406,
      "loss_g1": 162.39187622070312,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.0165263619273901,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 177.7455291748047,
      "loss_g0": 178.09402465820312,
      "loss_g1": 157.00338745117188,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524210572242737,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 202.2797088623047,
      "loss_g0": 202.2797088623047,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 146.5333709716797,
      "loss_g0": 146.376953125,
      "loss_g1": 155.8443145751953,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 275.2269287109375,
      "loss_g0": 0.0,
      "loss_g1": 275.2269287109375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 286.10687255859375,
      "loss_g0": 179.66522216796875,
      "loss_g1": 287.89508056640625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 114.38921356201172,
      "loss_g0": 199.7949676513672,
      "loss_g1": 112.95361328125,
      "q_g0": 0.016531307250261307,
      "q_g1": 0.9834686517715454,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 97.39762878417969,
      "loss_g0": 97.39762878417969,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 112.45812225341797,
      "loss_g0": 112.84574890136719,
      "loss_g1": 89.38514709472656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 107.42560577392578,
      "loss_g0": 105.04019927978516,
      "loss_g1": 249.28543090820312,
      "q_g0": 0.9834627509117126,
      "q_g1": 0.016537168994545937,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 172.79295349121094,
      "loss_g0": 171.96316528320312,
      "loss_g1": 222.1244659423828,
      "q_g0": 0.9834576845169067,
      "q_g1": 0.016542304307222366,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 91.7367935180664,
      "loss_g0": 90.10538482666016,
      "loss_g1": 188.66583251953125,
      "q_g0": 0.9834475517272949,
      "q_g1": 0.0165523923933506,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 479.13751220703125,
      "loss_g0": 476.4566955566406,
      "loss_g1": 638.253662109375,
      "q_g0": 0.9834310412406921,
      "q_g1": 0.016568975523114204,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 470.058349609375,
      "loss_g0": 468.78607177734375,
      "loss_g1": 545.5355834960938,
      "q_g0": 0.983423113822937,
      "q_g1": 0.016576845198869705,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 520.0906372070312,
      "loss_g0": 522.0197143554688,
      "loss_g1": 405.5652160644531,
      "q_g0": 0.9834350943565369,
      "q_g1": 0.016564909368753433,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 364.8732604980469,
      "loss_g0": 363.80108642578125,
      "loss_g1": 428.5005798339844,
      "q_g0": 0.983428418636322,
      "q_g1": 0.016571536660194397,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 498.6400146484375,
      "loss_g0": 500.0606384277344,
      "loss_g1": 414.28662109375,
      "q_g0": 0.9834372997283936,
      "q_g1": 0.016562750563025475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 383.7319030761719,
      "loss_g0": 382.9720153808594,
      "loss_g1": 428.8362121582031,
      "q_g0": 0.9834325909614563,
      "q_g1": 0.016567450016736984,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 460.48486328125,
      "loss_g0": 462.3427734375,
      "loss_g1": 350.1217956542969,
      "q_g0": 0.983444094657898,
      "q_g1": 0.016555948182940483,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 313.5554504394531,
      "loss_g0": 312.52362060546875,
      "loss_g1": 374.8226318359375,
      "q_g0": 0.9834376573562622,
      "q_g1": 0.01656232215464115,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 356.69183349609375,
      "loss_g0": 357.041748046875,
      "loss_g1": 335.9110107421875,
      "q_g0": 0.9834398031234741,
      "q_g1": 0.016560157760977745,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 689.1140747070312,
      "loss_g0": 694.058837890625,
      "loss_g1": 394.9100646972656,
      "q_g0": 0.9834704995155334,
      "q_g1": 0.016529541462659836,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 355.2889404296875,
      "loss_g0": 0.0,
      "loss_g1": 355.2889404296875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 341.6182861328125,
      "loss_g0": 284.9537353515625,
      "loss_g1": 342.57025146484375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 295.9512634277344,
      "loss_g0": 308.76092529296875,
      "loss_g1": 295.73602294921875,
      "q_g0": 0.016523750498890877,
      "q_g1": 0.9834762811660767,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 282.93719482421875,
      "loss_g0": 313.37432861328125,
      "loss_g1": 282.42572021484375,
      "q_g0": 0.016526920720934868,
      "q_g1": 0.9834730625152588,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 332.85882568359375,
      "loss_g0": 301.1273193359375,
      "loss_g1": 333.3919677734375,
      "q_g0": 0.01652361825108528,
      "q_g1": 0.9834764003753662,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 323.39068603515625,
      "loss_g0": 289.8395690917969,
      "loss_g1": 323.954345703125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 332.09283447265625,
      "loss_g0": 426.356201171875,
      "loss_g1": 330.5082702636719,
      "q_g0": 0.01653222180902958,
      "q_g1": 0.9834677577018738,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 241.52928161621094,
      "loss_g0": 268.379150390625,
      "loss_g1": 241.07785034179688,
      "q_g0": 0.01653500646352768,
      "q_g1": 0.9834649562835693,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 218.72279357910156,
      "loss_g0": 255.18157958984375,
      "loss_g1": 218.1096649169922,
      "q_g0": 0.016538798809051514,
      "q_g1": 0.9834611415863037,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 291.5994873046875,
      "loss_g0": 281.1300048828125,
      "loss_g1": 291.7755126953125,
      "q_g0": 0.01653771474957466,
      "q_g1": 0.9834623336791992,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 316.9633483886719,
      "loss_g0": 443.58868408203125,
      "loss_g1": 314.832275390625,
      "q_g0": 0.016550887376070023,
      "q_g1": 0.9834491610527039,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 297.1263122558594,
      "loss_g0": 207.8971405029297,
      "loss_g1": 298.6271057128906,
      "q_g0": 0.016541598364710808,
      "q_g1": 0.983458399772644,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 253.51846313476562,
      "loss_g0": 210.140869140625,
      "loss_g1": 254.24786376953125,
      "q_g0": 0.016537092626094818,
      "q_g1": 0.9834628701210022,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 244.0254669189453,
      "loss_g0": 263.177734375,
      "loss_g1": 243.70338439941406,
      "q_g0": 0.016539085656404495,
      "q_g1": 0.9834609031677246,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 261.0409240722656,
      "loss_g0": 250.233154296875,
      "loss_g1": 261.22265625,
      "q_g0": 0.01653796248137951,
      "q_g1": 0.9834620952606201,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 252.91993713378906,
      "loss_g0": 0.0,
      "loss_g1": 252.91993713378906,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 263.66796875,
      "loss_g0": 196.1332550048828,
      "loss_g1": 264.80255126953125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 253.2084197998047,
      "loss_g0": 186.27297973632812,
      "loss_g1": 254.3329315185547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 248.99868774414062,
      "loss_g0": 227.322265625,
      "loss_g1": 249.3628387451172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 242.3783416748047,
      "loss_g0": 217.0807342529297,
      "loss_g1": 242.80332946777344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 252.1677703857422,
      "loss_g0": 292.8577880859375,
      "loss_g1": 251.48399353027344,
      "q_g0": 0.01652665250003338,
      "q_g1": 0.9834733605384827,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 195.25576782226562,
      "loss_g0": 334.6121826171875,
      "loss_g1": 192.91188049316406,
      "q_g0": 0.016541149467229843,
      "q_g1": 0.9834588766098022,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 196.60971069335938,
      "loss_g0": 198.591796875,
      "loss_g1": 196.57638549804688,
      "q_g0": 0.016541358083486557,
      "q_g1": 0.9834586381912231,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 212.2382354736328,
      "loss_g0": 342.29931640625,
      "loss_g1": 210.04884338378906,
      "q_g0": 0.01655489206314087,
      "q_g1": 0.9834451079368591,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 383.3359375,
      "loss_g0": 359.18878173828125,
      "loss_g1": 383.7423400878906,
      "q_g0": 0.016552377492189407,
      "q_g1": 0.9834476709365845,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 176.83236694335938,
      "loss_g0": 198.8397674560547,
      "loss_g1": 176.46189880371094,
      "q_g0": 0.01655467599630356,
      "q_g1": 0.9834453463554382,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 205.73492431640625,
      "loss_g0": 173.63438415527344,
      "loss_g1": 206.27516174316406,
      "q_g0": 0.01655132882297039,
      "q_g1": 0.9834486842155457,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 165.6968994140625,
      "loss_g0": 153.40541076660156,
      "loss_g1": 165.90374755859375,
      "q_g0": 0.016550056636333466,
      "q_g1": 0.9834499955177307,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 229.25807189941406,
      "loss_g0": 140.06484985351562,
      "loss_g1": 230.75820922851562,
      "q_g0": 0.016540776938199997,
      "q_g1": 0.9834591746330261,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 192.47496032714844,
      "loss_g0": 0.0,
      "loss_g1": 192.47496032714844,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 496.0992431640625,
      "loss_g0": 152.59149169921875,
      "loss_g1": 501.87017822265625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 253.63922119140625,
      "loss_g0": 137.06544494628906,
      "loss_g1": 255.59765625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 186.35276794433594,
      "loss_g0": 0.0,
      "loss_g1": 186.35276794433594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 150.05142211914062,
      "loss_g0": 132.09408569335938,
      "loss_g1": 150.3531036376953,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 212.49691772460938,
      "loss_g0": 137.58786010742188,
      "loss_g1": 213.75538635253906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 132.8787384033203,
      "loss_g0": 117.15534973144531,
      "loss_g1": 133.1428985595703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 119.46063232421875,
      "loss_g0": 271.5554504394531,
      "loss_g1": 116.9029541015625,
      "q_g0": 0.016538232564926147,
      "q_g1": 0.9834617376327515,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 328.0624084472656,
      "loss_g0": 122.10527801513672,
      "loss_g1": 331.5224609375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 192.22314453125,
      "loss_g0": 183.12075805664062,
      "loss_g1": 192.3760528564453,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 229.67532348632812,
      "loss_g0": 200.13406372070312,
      "loss_g1": 230.17161560058594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 627.830322265625,
      "loss_g0": 548.6474609375,
      "loss_g1": 629.1605834960938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 518.4080810546875,
      "loss_g0": 605.227294921875,
      "loss_g1": 516.94873046875,
      "q_g0": 0.01653144881129265,
      "q_g1": 0.9834685325622559,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 529.6498413085938,
      "loss_g0": 494.6237487792969,
      "loss_g1": 530.2384643554688,
      "q_g0": 0.016527807340025902,
      "q_g1": 0.9834722280502319,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 468.2603454589844,
      "loss_g0": 423.9232482910156,
      "loss_g1": 469.0052185058594,
      "q_g0": 0.016523193567991257,
      "q_g1": 0.9834768176078796,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 407.5467834472656,
      "loss_g0": 397.24468994140625,
      "loss_g1": 407.7198486328125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 447.92974853515625,
      "loss_g0": 0.0,
      "loss_g1": 447.92974853515625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 394.2167053222656,
      "loss_g0": 467.41375732421875,
      "loss_g1": 392.98699951171875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 400.5577392578125,
      "loss_g0": 288.2004699707031,
      "loss_g1": 402.4453125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 403.6441345214844,
      "loss_g0": 0.0,
      "loss_g1": 403.6441345214844,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 366.20098876953125,
      "loss_g0": 348.20477294921875,
      "loss_g1": 366.5033264160156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 313.4718322753906,
      "loss_g0": 423.003173828125,
      "loss_g1": 311.6304016113281,
      "q_g0": 0.016533812507987022,
      "q_g1": 0.9834662079811096,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 303.992431640625,
      "loss_g0": 346.41632080078125,
      "loss_g1": 303.27899169921875,
      "q_g0": 0.01653822511434555,
      "q_g1": 0.9834617376327515,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 352.89306640625,
      "loss_g0": 390.2449951171875,
      "loss_g1": 352.2647705078125,
      "q_g0": 0.016542118042707443,
      "q_g1": 0.9834579229354858,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 396.53973388671875,
      "loss_g0": 296.95355224609375,
      "loss_g1": 398.2137451171875,
      "q_g0": 0.01653175801038742,
      "q_g1": 0.9834681749343872,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 339.4922790527344,
      "loss_g0": 292.6336669921875,
      "loss_g1": 340.2796936035156,
      "q_g0": 0.01652689091861248,
      "q_g1": 0.9834731221199036,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 321.9188537597656,
      "loss_g0": 291.8027648925781,
      "loss_g1": 322.4248352050781,
      "q_g0": 0.016523759812116623,
      "q_g1": 0.9834762811660767,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 283.67431640625,
      "loss_g0": 283.67431640625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 266.2029724121094,
      "loss_g0": 265.956298828125,
      "loss_g1": 280.88433837890625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 236.5669708251953,
      "loss_g0": 236.10304260253906,
      "loss_g1": 264.17724609375,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.01652529276907444,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 270.8486328125,
      "loss_g0": 0.0,
      "loss_g1": 270.8486328125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 292.0394592285156,
      "loss_g0": 256.0581359863281,
      "loss_g1": 292.6439208984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 294.8072509765625,
      "loss_g0": 0.0,
      "loss_g1": 294.8072509765625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 289.4385986328125,
      "loss_g0": 221.0285186767578,
      "loss_g1": 290.587890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 298.8621520996094,
      "loss_g0": 240.73123168945312,
      "loss_g1": 299.8387451171875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 302.2331848144531,
      "loss_g0": 237.48135375976562,
      "loss_g1": 303.3210144042969,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 421.41082763671875,
      "loss_g0": 235.70950317382812,
      "loss_g1": 424.5306091308594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 354.7445068359375,
      "loss_g0": 237.90151977539062,
      "loss_g1": 356.7074890136719,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 272.27020263671875,
      "loss_g0": 236.9158935546875,
      "loss_g1": 272.8641357421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 274.9490966796875,
      "loss_g0": 356.1864929199219,
      "loss_g1": 273.5836181640625,
      "q_g0": 0.01653086580336094,
      "q_g1": 0.9834691286087036,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 442.2732849121094,
      "loss_g0": 181.14523315429688,
      "loss_g1": 446.6602478027344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 226.519287109375,
      "loss_g0": 243.19976806640625,
      "loss_g1": 226.239013671875,
      "q_g0": 0.01652415469288826,
      "q_g1": 0.9834758043289185,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 233.76010131835938,
      "loss_g0": 167.73776245117188,
      "loss_g1": 234.8692626953125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 236.77850341796875,
      "loss_g0": 182.294677734375,
      "loss_g1": 237.69383239746094,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 268.1590576171875,
      "loss_g0": 178.61074829101562,
      "loss_g1": 269.6634521484375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 195.6990509033203,
      "loss_g0": 215.5435333251953,
      "loss_g1": 195.3656005859375,
      "q_g0": 0.016524480655789375,
      "q_g1": 0.9834755659103394,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 281.607421875,
      "loss_g0": 154.088623046875,
      "loss_g1": 283.7497253417969,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 271.2242736816406,
      "loss_g0": 0.0,
      "loss_g1": 271.2242736816406,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 246.89376831054688,
      "loss_g0": 170.06935119628906,
      "loss_g1": 248.18441772460938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 178.59466552734375,
      "loss_g0": 151.613037109375,
      "loss_g1": 179.04794311523438,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 272.833251953125,
      "loss_g0": 137.34022521972656,
      "loss_g1": 275.1095275878906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 458.4612731933594,
      "loss_g0": 187.32699584960938,
      "loss_g1": 463.0163269042969,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 207.56431579589844,
      "loss_g0": 139.36184692382812,
      "loss_g1": 208.71011352539062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 180.8096466064453,
      "loss_g0": 221.31109619140625,
      "loss_g1": 180.12904357910156,
      "q_g0": 0.01652662642300129,
      "q_g1": 0.9834733605384827,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 170.63336181640625,
      "loss_g0": 135.80213928222656,
      "loss_g1": 171.21853637695312,
      "q_g0": 0.016523005440831184,
      "q_g1": 0.9834769368171692,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 144.00970458984375,
      "loss_g0": 0.0,
      "loss_g1": 144.00970458984375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 237.76548767089844,
      "loss_g0": 248.59103393554688,
      "loss_g1": 237.5836181640625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 331.70220947265625,
      "loss_g0": 122.47320556640625,
      "loss_g1": 335.2172546386719,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 109.61167907714844,
      "loss_g0": 123.14259338378906,
      "loss_g1": 109.38433837890625,
      "q_g0": 0.016523828729987144,
      "q_g1": 0.9834761619567871,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 295.3731384277344,
      "loss_g0": 98.4200439453125,
      "loss_g1": 298.68194580078125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 260.1298522949219,
      "loss_g0": 89.2362289428711,
      "loss_g1": 263.0008544921875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 562.8651123046875,
      "loss_g0": 553.0325927734375,
      "loss_g1": 563.0302734375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 451.2828369140625,
      "loss_g0": 465.2005310058594,
      "loss_g1": 451.0489807128906,
      "q_g0": 0.01652386784553528,
      "q_g1": 0.9834761619567871,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 455.031005859375,
      "loss_g0": 455.031005859375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 303.5655517578125,
      "loss_g0": 300.7393798828125,
      "loss_g1": 471.7899169921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 483.525634765625,
      "loss_g0": 485.37823486328125,
      "loss_g1": 373.25115966796875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 356.0936584472656,
      "loss_g0": 356.35858154296875,
      "loss_g1": 340.3243408203125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 354.67486572265625,
      "loss_g0": 348.62939453125,
      "loss_g1": 713.6987915039062,
      "q_g0": 0.9834401607513428,
      "q_g1": 0.016559774056077003,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 388.525390625,
      "loss_g0": 389.0463562011719,
      "loss_g1": 357.5811462402344,
      "q_g0": 0.9834433794021606,
      "q_g1": 0.016556551679968834,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 459.60772705078125,
      "loss_g0": 460.82073974609375,
      "loss_g1": 387.5240478515625,
      "q_g0": 0.9834508895874023,
      "q_g1": 0.016549047082662582,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 394.23321533203125,
      "loss_g0": 395.45697021484375,
      "loss_g1": 321.4761962890625,
      "q_g0": 0.9834585189819336,
      "q_g1": 0.016541482880711555,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 287.22125244140625,
      "loss_g0": 286.1580505371094,
      "loss_g1": 350.407470703125,
      "q_g0": 0.9834519624710083,
      "q_g1": 0.016548052430152893,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 308.1690368652344,
      "loss_g0": 305.9583740234375,
      "loss_g1": 439.43780517578125,
      "q_g0": 0.9834383130073547,
      "q_g1": 0.016561726108193398,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 269.164794921875,
      "loss_g0": 268.71710205078125,
      "loss_g1": 295.74517822265625,
      "q_g0": 0.9834354519844055,
      "q_g1": 0.016564497724175453,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 278.3553466796875,
      "loss_g0": 277.9849853515625,
      "loss_g1": 300.34002685546875,
      "q_g0": 0.9834332466125488,
      "q_g1": 0.016566790640354156,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 298.6062927246094,
      "loss_g0": 298.8048400878906,
      "loss_g1": 286.81884765625,
      "q_g0": 0.9834344387054443,
      "q_g1": 0.016565553843975067,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 401.8496398925781,
      "loss_g0": 403.9628601074219,
      "loss_g1": 276.2955627441406,
      "q_g0": 0.9834475517272949,
      "q_g1": 0.016552478075027466,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 260.13629150390625,
      "loss_g0": 258.9781188964844,
      "loss_g1": 328.9167175292969,
      "q_g0": 0.9834403991699219,
      "q_g1": 0.01655963808298111,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 228.72787475585938,
      "loss_g0": 228.0044708251953,
      "loss_g1": 271.67706298828125,
      "q_g0": 0.9834359288215637,
      "q_g1": 0.016564110293984413,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 235.5135955810547,
      "loss_g0": 235.025146484375,
      "loss_g1": 264.50811767578125,
      "q_g0": 0.9834328889846802,
      "q_g1": 0.016567131504416466,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 265.12176513671875,
      "loss_g0": 264.9398193359375,
      "loss_g1": 275.9207458496094,
      "q_g0": 0.9834317564964294,
      "q_g1": 0.01656825840473175,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 256.884765625,
      "loss_g0": 256.3221435546875,
      "loss_g1": 290.2732849121094,
      "q_g0": 0.9834282994270325,
      "q_g1": 0.016571739688515663,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 253.18666076660156,
      "loss_g0": 252.98690795898438,
      "loss_g1": 265.03955078125,
      "q_g0": 0.9834270477294922,
      "q_g1": 0.016572967171669006,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 200.91128540039062,
      "loss_g0": 200.80992126464844,
      "loss_g1": 206.92623901367188,
      "q_g0": 0.9834264516830444,
      "q_g1": 0.016573596745729446,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 218.05735778808594,
      "loss_g0": 217.61773681640625,
      "loss_g1": 244.13848876953125,
      "q_g0": 0.9834237098693848,
      "q_g1": 0.016576308757066727,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 577.0382080078125,
      "loss_g0": 582.2113037109375,
      "loss_g1": 269.52813720703125,
      "q_g0": 0.9834557175636292,
      "q_g1": 0.016544269397854805,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 647.1705322265625,
      "loss_g0": 654.04443359375,
      "loss_g1": 238.00863647460938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 212.5711669921875,
      "loss_g0": 211.3095703125,
      "loss_g1": 287.63037109375,
      "q_g0": 0.9834697842597961,
      "q_g1": 0.016530223190784454,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 198.68026733398438,
      "loss_g0": 197.45736694335938,
      "loss_g1": 271.40313720703125,
      "q_g0": 0.9834622144699097,
      "q_g1": 0.016537781804800034,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 361.5113830566406,
      "loss_g0": 363.01934814453125,
      "loss_g1": 271.7845764160156,
      "q_g0": 0.9834715127944946,
      "q_g1": 0.016528455540537834,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 215.04171752929688,
      "loss_g0": 215.01321411132812,
      "loss_g1": 216.7369842529297,
      "q_g0": 0.9834713935852051,
      "q_g1": 0.01652863621711731,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 303.4454040527344,
      "loss_g0": 0.0,
      "loss_g1": 303.4454040527344,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 225.07421875,
      "loss_g0": 193.05111694335938,
      "loss_g1": 225.61221313476562,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 522.5598754882812,
      "loss_g0": 297.7275390625,
      "loss_g1": 526.3370361328125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 177.9173126220703,
      "loss_g0": 171.36517333984375,
      "loss_g1": 178.02737426757812,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 203.93435668945312,
      "loss_g0": 170.59939575195312,
      "loss_g1": 204.494384765625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 172.37001037597656,
      "loss_g0": 197.01657104492188,
      "loss_g1": 171.9558868408203,
      "q_g0": 0.016524983569979668,
      "q_g1": 0.9834749698638916,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 205.8854217529297,
      "loss_g0": 178.47366333007812,
      "loss_g1": 206.34593200683594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 141.87730407714844,
      "loss_g0": 232.33319091796875,
      "loss_g1": 140.3567657470703,
      "q_g0": 0.01653182879090309,
      "q_g1": 0.9834681749343872,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 293.8524169921875,
      "loss_g0": 144.333251953125,
      "loss_g1": 296.36431884765625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 145.53221130371094,
      "loss_g0": 136.9039306640625,
      "loss_g1": 145.6771697998047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 195.7832489013672,
      "loss_g0": 242.45809936523438,
      "loss_g1": 194.99888610839844,
      "q_g0": 0.016527269035577774,
      "q_g1": 0.9834727048873901,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 373.023681640625,
      "loss_g0": 179.49520874023438,
      "loss_g1": 376.27496337890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 134.5319366455078,
      "loss_g0": 273.0401611328125,
      "loss_g1": 132.20294189453125,
      "q_g0": 0.016536829993128777,
      "q_g1": 0.9834631085395813,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 221.35853576660156,
      "loss_g0": 221.35853576660156,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 263.83209228515625,
      "loss_g0": 265.3221435546875,
      "loss_g1": 175.13778686523438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 131.43356323242188,
      "loss_g0": 131.43356323242188,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 109.3606185913086,
      "loss_g0": 108.8792724609375,
      "loss_g1": 138.01202392578125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 125.26176452636719,
      "loss_g0": 125.81261444091797,
      "loss_g1": 92.47285461425781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 194.24314880371094,
      "loss_g0": 194.39541625976562,
      "loss_g1": 185.17941284179688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 125.90184020996094,
      "loss_g0": 124.05841064453125,
      "loss_g1": 235.55264282226562,
      "q_g0": 0.9834662079811096,
      "q_g1": 0.01653382182121277,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 553.029541015625,
      "loss_g0": 551.0477294921875,
      "loss_g1": 670.8211669921875,
      "q_g0": 0.9834538698196411,
      "q_g1": 0.01654607430100441,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 420.3746337890625,
      "loss_g0": 418.1770935058594,
      "loss_g1": 550.881103515625,
      "q_g0": 0.9834402799606323,
      "q_g1": 0.016559666022658348,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 434.1938171386719,
      "loss_g0": 433.102783203125,
      "loss_g1": 498.96002197265625,
      "q_g0": 0.9834336042404175,
      "q_g1": 0.01656641811132431,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 456.6075134277344,
      "loss_g0": 457.097412109375,
      "loss_g1": 427.5224304199219,
      "q_g0": 0.9834365844726562,
      "q_g1": 0.01656338758766651,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 458.4644775390625,
      "loss_g0": 0.0,
      "loss_g1": 458.4644775390625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 350.66778564453125,
      "loss_g0": 419.8651123046875,
      "loss_g1": 349.5052490234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 371.7806396484375,
      "loss_g0": 402.536376953125,
      "loss_g1": 371.2638244628906,
      "q_g0": 0.016525618731975555,
      "q_g1": 0.9834744334220886,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 412.9530944824219,
      "loss_g0": 412.9530944824219,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 349.67645263671875,
      "loss_g0": 348.8874816894531,
      "loss_g1": 396.6397705078125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 386.49365234375,
      "loss_g0": 386.470458984375,
      "loss_g1": 387.8757019042969,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 341.9383850097656,
      "loss_g0": 342.4269714355469,
      "loss_g1": 312.8545837402344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 476.27825927734375,
      "loss_g0": 478.33392333984375,
      "loss_g1": 353.91693115234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 338.6418151855469,
      "loss_g0": 338.6541748046875,
      "loss_g1": 337.90545654296875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 296.119873046875,
      "loss_g0": 294.61273193359375,
      "loss_g1": 385.7784423828125,
      "q_g0": 0.9834682941436768,
      "q_g1": 0.016531743109226227,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 293.94793701171875,
      "loss_g0": 291.37847900390625,
      "loss_g1": 446.655517578125,
      "q_g0": 0.9834523797035217,
      "q_g1": 0.016547635197639465,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 350.86566162109375,
      "loss_g0": 350.2135009765625,
      "loss_g1": 389.61297607421875,
      "q_g0": 0.9834483861923218,
      "q_g1": 0.01655167154967785,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 288.2068176269531,
      "loss_g0": 286.9169616699219,
      "loss_g1": 364.80682373046875,
      "q_g0": 0.9834403991699219,
      "q_g1": 0.016559645533561707,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 268.73626708984375,
      "loss_g0": 264.38232421875,
      "loss_g1": 526.881591796875,
      "q_g0": 0.9834134578704834,
      "q_g1": 0.0165865495800972,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 338.2387390136719,
      "loss_g0": 338.74530029296875,
      "loss_g1": 308.1985778808594,
      "q_g0": 0.9834165573120117,
      "q_g1": 0.016583414748311043,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 230.59202575683594,
      "loss_g0": 228.72113037109375,
      "loss_g1": 341.45965576171875,
      "q_g0": 0.9834049940109253,
      "q_g1": 0.016594983637332916,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 282.97186279296875,
      "loss_g0": 283.04766845703125,
      "loss_g1": 278.4798583984375,
      "q_g0": 0.9834054708480835,
      "q_g1": 0.01659451797604561,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 268.4300231933594,
      "loss_g0": 267.01507568359375,
      "loss_g1": 352.2353515625,
      "q_g0": 0.9833967089653015,
      "q_g1": 0.016603264957666397,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 238.58753967285156,
      "loss_g0": 237.6204833984375,
      "loss_g1": 295.8448486328125,
      "q_g0": 0.9833906888961792,
      "q_g1": 0.01660924404859543,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 234.38722229003906,
      "loss_g0": 234.6199951171875,
      "loss_g1": 220.6033172607422,
      "q_g0": 0.9833922386169434,
      "q_g1": 0.016607804223895073,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 240.13565063476562,
      "loss_g0": 237.11355590820312,
      "loss_g1": 418.87725830078125,
      "q_g0": 0.9833735227584839,
      "q_g1": 0.01662648282945156,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 649.1266479492188,
      "loss_g0": 656.2545166015625,
      "loss_g1": 226.4052734375,
      "q_g0": 0.9834176898002625,
      "q_g1": 0.016582341864705086,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 260.86834716796875,
      "loss_g0": 261.4119567871094,
      "loss_g1": 228.62327575683594,
      "q_g0": 0.9834210276603699,
      "q_g1": 0.016578983515501022,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 299.61505126953125,
      "loss_g0": 300.365966796875,
      "loss_g1": 255.06039428710938,
      "q_g0": 0.9834256768226624,
      "q_g1": 0.01657434180378914,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 297.00555419921875,
      "loss_g0": 0.0,
      "loss_g1": 297.00555419921875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 274.1037902832031,
      "loss_g0": 214.43199157714844,
      "loss_g1": 275.10626220703125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 222.90757751464844,
      "loss_g0": 183.52699279785156,
      "loss_g1": 223.56915283203125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 210.77455139160156,
      "loss_g0": 241.6309051513672,
      "loss_g1": 210.25607299804688,
      "q_g0": 0.0165256280452013,
      "q_g1": 0.9834744334220886,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 195.270263671875,
      "loss_g0": 162.094970703125,
      "loss_g1": 195.82760620117188,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 322.3286437988281,
      "loss_g0": 175.9624481201172,
      "loss_g1": 324.78759765625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 180.8153533935547,
      "loss_g0": 180.8153533935547,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 171.5311279296875,
      "loss_g0": 170.82778930664062,
      "loss_g1": 213.39605712890625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 161.86343383789062,
      "loss_g0": 161.86343383789062,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 167.90670776367188,
      "loss_g0": 168.0172576904297,
      "loss_g1": 161.32566833496094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 166.66722106933594,
      "loss_g0": 166.14541625976562,
      "loss_g1": 197.72056579589844,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.016525648534297943,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 149.36793518066406,
      "loss_g0": 149.02581787109375,
      "loss_g1": 169.72463989257812,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.01652776636183262,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 160.29481506347656,
      "loss_g0": 160.37490844726562,
      "loss_g1": 155.52828979492188,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527269035577774,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 314.3952331542969,
      "loss_g0": 0.0,
      "loss_g1": 314.3952331542969,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 162.07940673828125,
      "loss_g0": 280.06585693359375,
      "loss_g1": 160.09722900390625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 226.62559509277344,
      "loss_g0": 0.0,
      "loss_g1": 226.62559509277344,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 243.82090759277344,
      "loss_g0": 340.12213134765625,
      "loss_g1": 242.20303344726562,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 161.31646728515625,
      "loss_g0": 225.64739990234375,
      "loss_g1": 160.23526000976562,
      "q_g0": 0.016529114916920662,
      "q_g1": 0.9834708571434021,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 234.65003967285156,
      "loss_g0": 108.99192810058594,
      "loss_g1": 236.76109313964844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 198.82289123535156,
      "loss_g0": 176.54183959960938,
      "loss_g1": 199.19720458984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 239.00253295898438,
      "loss_g0": 154.87918090820312,
      "loss_g1": 240.41580200195312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 195.93931579589844,
      "loss_g0": 144.4184112548828,
      "loss_g1": 196.80487060546875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 560.2391357421875,
      "loss_g0": 513.9151611328125,
      "loss_g1": 561.017333984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 527.4212646484375,
      "loss_g0": 524.20068359375,
      "loss_g1": 527.475341796875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 515.1564331054688,
      "loss_g0": 446.11846923828125,
      "loss_g1": 516.3162841796875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 413.8653564453125,
      "loss_g0": 487.73797607421875,
      "loss_g1": 412.62371826171875,
      "q_g0": 0.016530107706785202,
      "q_g1": 0.9834699034690857,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 450.6275634765625,
      "loss_g0": 393.604736328125,
      "loss_g1": 451.5856628417969,
      "q_g0": 0.01652417704463005,
      "q_g1": 0.9834758043289185,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 382.1007385253906,
      "loss_g0": 410.9461669921875,
      "loss_g1": 381.6159973144531,
      "q_g0": 0.016527175903320312,
      "q_g1": 0.9834728240966797,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 418.6485290527344,
      "loss_g0": 429.6649169921875,
      "loss_g1": 418.46337890625,
      "q_g0": 0.016528325155377388,
      "q_g1": 0.9834716320037842,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 396.6488037109375,
      "loss_g0": 373.2825927734375,
      "loss_g1": 397.04144287109375,
      "q_g0": 0.01652589812874794,
      "q_g1": 0.98347407579422,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 373.83349609375,
      "loss_g0": 386.88275146484375,
      "loss_g1": 373.61419677734375,
      "q_g0": 0.01652725413441658,
      "q_g1": 0.9834727048873901,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 330.75091552734375,
      "loss_g0": 330.75091552734375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 403.4836730957031,
      "loss_g0": 404.65338134765625,
      "loss_g1": 333.85760498046875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 324.2505187988281,
      "loss_g0": 323.4128112792969,
      "loss_g1": 374.09844970703125,
      "q_g0": 0.9834723472595215,
      "q_g1": 0.016527602449059486,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 299.25592041015625,
      "loss_g0": 298.7650146484375,
      "loss_g1": 328.46282958984375,
      "q_g0": 0.9834693670272827,
      "q_g1": 0.016530640423297882,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 288.6600341796875,
      "loss_g0": 288.6600341796875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 386.1077880859375,
      "loss_g0": 388.00048828125,
      "loss_g1": 273.44769287109375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 289.0691223144531,
      "loss_g0": 288.7699890136719,
      "loss_g1": 306.8721008300781,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524270176887512,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 286.57940673828125,
      "loss_g0": 285.94012451171875,
      "loss_g1": 324.6187744140625,
      "q_g0": 0.9834717512130737,
      "q_g1": 0.01652822457253933,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 289.3149719238281,
      "loss_g0": 289.3149719238281,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 244.68881225585938,
      "loss_g0": 244.10565185546875,
      "loss_g1": 279.4010009765625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 254.83538818359375,
      "loss_g0": 0.0,
      "loss_g1": 254.83538818359375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 293.3451843261719,
      "loss_g0": 290.3388366699219,
      "loss_g1": 293.39569091796875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 311.0392761230469,
      "loss_g0": 281.4209289550781,
      "loss_g1": 311.536865234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 269.55859375,
      "loss_g0": 255.32742309570312,
      "loss_g1": 269.79766845703125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 261.1785583496094,
      "loss_g0": 261.1785583496094,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 212.27908325195312,
      "loss_g0": 211.87313842773438,
      "loss_g1": 236.4427490234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 323.62841796875,
      "loss_g0": 0.0,
      "loss_g1": 323.62841796875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 216.03536987304688,
      "loss_g0": 216.17623901367188,
      "loss_g1": 216.03298950195312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 309.7587585449219,
      "loss_g0": 214.50228881835938,
      "loss_g1": 311.35906982421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 206.34449768066406,
      "loss_g0": 206.34449768066406,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 182.39242553710938,
      "loss_g0": 180.88894653320312,
      "loss_g1": 271.8853454589844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 347.525390625,
      "loss_g0": 349.783447265625,
      "loss_g1": 213.11758422851562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 209.29513549804688,
      "loss_g0": 0.0,
      "loss_g1": 209.29513549804688,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 284.2814636230469,
      "loss_g0": 195.94180297851562,
      "loss_g1": 285.76556396484375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 206.55157470703125,
      "loss_g0": 208.15597534179688,
      "loss_g1": 206.52462768554688,
      "q_g0": 0.016522590070962906,
      "q_g1": 0.9834774136543274,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 475.5397644042969,
      "loss_g0": 166.1666259765625,
      "loss_g1": 480.7372131347656,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 192.79554748535156,
      "loss_g0": 139.30384826660156,
      "loss_g1": 193.69419860839844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 333.6396484375,
      "loss_g0": 333.6396484375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 270.1572265625,
      "loss_g0": 272.3223571777344,
      "loss_g1": 141.28062438964844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 291.9117736816406,
      "loss_g0": 292.2686462402344,
      "loss_g1": 270.6690368652344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 143.9155731201172,
      "loss_g0": 143.27911376953125,
      "loss_g1": 181.7901611328125,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.0165263619273901,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 131.2843780517578,
      "loss_g0": 129.5493927001953,
      "loss_g1": 234.46353149414062,
      "q_g0": 0.9834628701210022,
      "q_g1": 0.016537092626094818,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 198.94735717773438,
      "loss_g0": 200.0216522216797,
      "loss_g1": 135.0326385498047,
      "q_g0": 0.9834696054458618,
      "q_g1": 0.016530441120266914,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 198.5165252685547,
      "loss_g0": 199.22714233398438,
      "loss_g1": 156.22760009765625,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652604527771473,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 136.1856231689453,
      "loss_g0": 134.06326293945312,
      "loss_g1": 262.3871154785156,
      "q_g0": 0.9834607839584351,
      "q_g1": 0.01653917133808136,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 109.37775421142578,
      "loss_g0": 108.58065795898438,
      "loss_g1": 156.76071166992188,
      "q_g0": 0.9834558367729187,
      "q_g1": 0.016544103622436523,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 245.4683380126953,
      "loss_g0": 247.72134399414062,
      "loss_g1": 111.42456817626953,
      "q_g0": 0.9834697842597961,
      "q_g1": 0.01653016172349453,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 126.51080322265625,
      "loss_g0": 126.28160095214844,
      "loss_g1": 140.14625549316406,
      "q_g0": 0.9834684133529663,
      "q_g1": 0.016531581059098244,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 201.99440002441406,
      "loss_g0": 201.89724731445312,
      "loss_g1": 207.77403259277344,
      "q_g0": 0.9834678173065186,
      "q_g1": 0.016532186418771744,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 149.56991577148438,
      "loss_g0": 148.3066864013672,
      "loss_g1": 224.680908203125,
      "q_g0": 0.983460009098053,
      "q_g1": 0.016540000215172768,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 142.10321044921875,
      "loss_g0": 139.9017333984375,
      "loss_g1": 272.89190673828125,
      "q_g0": 0.9834464192390442,
      "q_g1": 0.016553610563278198,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 623.7937622070312,
      "loss_g0": 622.3043823242188,
      "loss_g1": 712.2283935546875,
      "q_g0": 0.983437180519104,
      "q_g1": 0.016562819480895996,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 466.73876953125,
      "loss_g0": 465.1693115234375,
      "loss_g1": 559.8723754882812,
      "q_g0": 0.9834274649620056,
      "q_g1": 0.01657252572476864,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 472.6129150390625,
      "loss_g0": 471.533447265625,
      "loss_g1": 536.6430053710938,
      "q_g0": 0.9834207892417908,
      "q_g1": 0.01657920330762863,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 491.69989013671875,
      "loss_g0": 491.82586669921875,
      "loss_g1": 484.2265625,
      "q_g0": 0.9834216237068176,
      "q_g1": 0.016578424721956253,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 431.70263671875,
      "loss_g0": 430.645263671875,
      "loss_g1": 494.39959716796875,
      "q_g0": 0.9834150671958923,
      "q_g1": 0.016584962606430054,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 440.1304931640625,
      "loss_g0": 441.0216064453125,
      "loss_g1": 387.2733154296875,
      "q_g0": 0.9834205508232117,
      "q_g1": 0.01657945103943348,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 401.1348571777344,
      "loss_g0": 401.43365478515625,
      "loss_g1": 383.4091796875,
      "q_g0": 0.9834223985671997,
      "q_g1": 0.01657760888338089,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 425.9447021484375,
      "loss_g0": 425.9447021484375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 360.1567687988281,
      "loss_g0": 358.7815856933594,
      "loss_g1": 442.01385498046875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 366.5521545410156,
      "loss_g0": 366.36480712890625,
      "loss_g1": 377.704833984375,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523580998182297,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 352.8870849609375,
      "loss_g0": 352.45068359375,
      "loss_g1": 378.85797119140625,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.01652628742158413,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 323.11029052734375,
      "loss_g0": 0.0,
      "loss_g1": 323.11029052734375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 409.76666259765625,
      "loss_g0": 310.976318359375,
      "loss_g1": 411.42633056640625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 346.92437744140625,
      "loss_g0": 314.61492919921875,
      "loss_g1": 347.4671630859375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 328.9643249511719,
      "loss_g0": 268.49066162109375,
      "loss_g1": 329.9802551269531,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 310.83734130859375,
      "loss_g0": 310.83734130859375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 301.8022155761719,
      "loss_g0": 301.1599426269531,
      "loss_g1": 340.0335693359375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 569.5203247070312,
      "loss_g0": 572.5748901367188,
      "loss_g1": 387.70172119140625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 301.08990478515625,
      "loss_g0": 301.0516357421875,
      "loss_g1": 303.36822509765625,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522657126188278,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 279.1285705566406,
      "loss_g0": 279.1285705566406,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 278.903564453125,
      "loss_g0": 278.19293212890625,
      "loss_g1": 321.2021179199219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 287.99322509765625,
      "loss_g0": 287.9900817871094,
      "loss_g1": 288.17999267578125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 706.1597900390625,
      "loss_g0": 713.417236328125,
      "loss_g1": 274.16748046875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 404.8316345214844,
      "loss_g0": 408.0501403808594,
      "loss_g1": 213.25408935546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 278.5605163574219,
      "loss_g0": 278.6167907714844,
      "loss_g1": 275.2104797363281,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 259.8708190917969,
      "loss_g0": 259.5904235839844,
      "loss_g1": 276.55859375,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652415469288826,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 254.09727478027344,
      "loss_g0": 253.79135131835938,
      "loss_g1": 272.3025817871094,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652604527771473,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 224.8341827392578,
      "loss_g0": 225.2177734375,
      "loss_g1": 202.0037078857422,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523675993084908,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 369.08880615234375,
      "loss_g0": 367.1155700683594,
      "loss_g1": 486.444091796875,
      "q_g0": 0.9834641814231873,
      "q_g1": 0.01653587631881237,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 242.54420471191406,
      "loss_g0": 242.68724060058594,
      "loss_g1": 234.03675842285156,
      "q_g0": 0.9834649562835693,
      "q_g1": 0.016534989699721336,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 227.1243896484375,
      "loss_g0": 227.1243896484375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 198.34620666503906,
      "loss_g0": 197.74522399902344,
      "loss_g1": 234.11865234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 181.46630859375,
      "loss_g0": 181.5968017578125,
      "loss_g1": 173.69869995117188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 199.01097106933594,
      "loss_g0": 196.63047790527344,
      "loss_g1": 340.5791015625,
      "q_g0": 0.9834628701210022,
      "q_g1": 0.016537148505449295,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 144.62350463867188,
      "loss_g0": 143.3460235595703,
      "loss_g1": 220.55906677246094,
      "q_g0": 0.9834549427032471,
      "q_g1": 0.016545042395591736,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 176.28770446777344,
      "loss_g0": 176.21804809570312,
      "loss_g1": 180.42758178710938,
      "q_g0": 0.9834545850753784,
      "q_g1": 0.01654547080397606,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 177.0690460205078,
      "loss_g0": 176.68597412109375,
      "loss_g1": 199.8349151611328,
      "q_g0": 0.9834521412849426,
      "q_g1": 0.01654783822596073,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 149.95724487304688,
      "loss_g0": 149.15188598632812,
      "loss_g1": 197.8057403564453,
      "q_g0": 0.9834471940994263,
      "q_g1": 0.016552813351154327,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 148.72137451171875,
      "loss_g0": 148.59426879882812,
      "loss_g1": 156.27236938476562,
      "q_g0": 0.9834464192390442,
      "q_g1": 0.016553595662117004,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 212.38516235351562,
      "loss_g0": 212.89865112304688,
      "loss_g1": 181.8720245361328,
      "q_g0": 0.9834496378898621,
      "q_g1": 0.016550421714782715,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 175.5321807861328,
      "loss_g0": 171.12734985351562,
      "loss_g1": 436.83685302734375,
      "q_g0": 0.9834222793579102,
      "q_g1": 0.016577646136283875,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 127.75566101074219,
      "loss_g0": 127.55369567871094,
      "loss_g1": 139.73516845703125,
      "q_g0": 0.9834211468696594,
      "q_g1": 0.016578897833824158,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 307.8074951171875,
      "loss_g0": 310.46453857421875,
      "loss_g1": 150.03843688964844,
      "q_g0": 0.9834375381469727,
      "q_g1": 0.01656244695186615,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 196.314208984375,
      "loss_g0": 197.0510711669922,
      "loss_g1": 152.5481719970703,
      "q_g0": 0.9834421277046204,
      "q_g1": 0.01655789092183113,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 158.751953125,
      "loss_g0": 157.7161865234375,
      "loss_g1": 220.2458038330078,
      "q_g0": 0.9834356904029846,
      "q_g1": 0.016564296558499336,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 106.9393310546875,
      "loss_g0": 105.34382629394531,
      "loss_g1": 201.6084442138672,
      "q_g0": 0.9834258556365967,
      "q_g1": 0.016574157401919365,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 106.25755310058594,
      "loss_g0": 106.0647201538086,
      "loss_g1": 117.6982192993164,
      "q_g0": 0.9834246039390564,
      "q_g1": 0.016575351357460022,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 198.29693603515625,
      "loss_g0": 196.92515563964844,
      "loss_g1": 279.64251708984375,
      "q_g0": 0.9834161996841431,
      "q_g1": 0.01658383570611477,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 119.984130859375,
      "loss_g0": 120.31295776367188,
      "loss_g1": 100.482421875,
      "q_g0": 0.9834181666374207,
      "q_g1": 0.01658179797232151,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 187.20858764648438,
      "loss_g0": 187.95179748535156,
      "loss_g1": 143.11819458007812,
      "q_g0": 0.9834227561950684,
      "q_g1": 0.016577204689383507,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 611.6004638671875,
      "loss_g0": 611.718994140625,
      "loss_g1": 604.5714111328125,
      "q_g0": 0.9834235906600952,
      "q_g1": 0.016576474532485008,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 407.0406494140625,
      "loss_g0": 407.0406494140625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 392.31298828125,
      "loss_g0": 391.90765380859375,
      "loss_g1": 416.4393005371094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 400.1142883300781,
      "loss_g0": 400.93426513671875,
      "loss_g1": 351.3049621582031,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 337.00360107421875,
      "loss_g0": 0.0,
      "loss_g1": 337.00360107421875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 338.45172119140625,
      "loss_g0": 283.3431396484375,
      "loss_g1": 339.3775329589844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 412.1845397949219,
      "loss_g0": 249.75244140625,
      "loss_g1": 414.91339111328125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 260.8196105957031,
      "loss_g0": 391.7021789550781,
      "loss_g1": 258.61895751953125,
      "q_g0": 0.016536029055714607,
      "q_g1": 0.9834639430046082,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 257.1150817871094,
      "loss_g0": 256.80548095703125,
      "loss_g1": 257.12030029296875,
      "q_g0": 0.01653599925339222,
      "q_g1": 0.9834640622138977,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 226.6103057861328,
      "loss_g0": 227.95858764648438,
      "loss_g1": 226.58761596679688,
      "q_g0": 0.016536137089133263,
      "q_g1": 0.9834638237953186,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 239.48101806640625,
      "loss_g0": 0.0,
      "loss_g1": 239.48101806640625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 247.030029296875,
      "loss_g0": 0.0,
      "loss_g1": 247.030029296875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 240.81260681152344,
      "loss_g0": 146.83282470703125,
      "loss_g1": 242.39144897460938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 301.0968017578125,
      "loss_g0": 316.56634521484375,
      "loss_g1": 300.8368835449219,
      "q_g0": 0.016524028033018112,
      "q_g1": 0.983475923538208,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 151.2652130126953,
      "loss_g0": 151.2652130126953,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 254.3467254638672,
      "loss_g0": 256.1726989746094,
      "loss_g1": 145.6576690673828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 184.78204345703125,
      "loss_g0": 183.2234344482422,
      "loss_g1": 277.50201416015625,
      "q_g0": 0.9834679365158081,
      "q_g1": 0.016532069072127342,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 109.18803405761719,
      "loss_g0": 109.18803405761719,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 128.99346923828125,
      "loss_g0": 0.0,
      "loss_g1": 128.99346923828125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "eval_cer": 0.8289955940265918,
      "eval_loss": 294.0757141113281,
      "eval_runtime": 83.4062,
      "eval_samples_per_second": 20.094,
      "eval_steps_per_second": 5.024,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 560.63671875,
      "loss_g0": 468.7535400390625,
      "loss_g1": 562.1803588867188,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 200
    },
    {
      "epoch": 1.015228426395939,
      "gdro_loss": 483.33935546875,
      "loss_g0": 487.4143981933594,
      "loss_g1": 483.2708740234375,
      "q_g0": 0.016522854566574097,
      "q_g1": 0.9834771752357483,
      "step": 200
    },
    {
      "epoch": 1.5228426395939088,
      "grad_norm": 1187.6170654296875,
      "learning_rate": 0.00041375401069518715,
      "loss": 292.9459,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 402.150634765625,
      "loss_g0": 402.99652099609375,
      "loss_g1": 351.90423583984375,
      "q_g0": 0.983444333076477,
      "q_g1": 0.01655563712120056,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 351.8898620605469,
      "loss_g0": 352.64990234375,
      "loss_g1": 306.72967529296875,
      "q_g0": 0.9834490418434143,
      "q_g1": 0.01655094139277935,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 315.3815002441406,
      "loss_g0": 314.50994873046875,
      "loss_g1": 367.15118408203125,
      "q_g0": 0.9834436178207397,
      "q_g1": 0.016556335613131523,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 295.6739196777344,
      "loss_g0": 296.87750244140625,
      "loss_g1": 224.14889526367188,
      "q_g0": 0.9834511280059814,
      "q_g1": 0.016548892483115196,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 284.7319030761719,
      "loss_g0": 285.00311279296875,
      "loss_g1": 268.61260986328125,
      "q_g0": 0.9834527373313904,
      "q_g1": 0.016547206789255142,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 285.03643798828125,
      "loss_g0": 284.6787109375,
      "loss_g1": 306.2951354980469,
      "q_g0": 0.9834505319595337,
      "q_g1": 0.016549410298466682,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 266.73284912109375,
      "loss_g0": 267.45416259765625,
      "loss_g1": 223.85775756835938,
      "q_g0": 0.9834550619125366,
      "q_g1": 0.016544951125979424,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 273.2735900878906,
      "loss_g0": 273.53955078125,
      "loss_g1": 257.4649963378906,
      "q_g0": 0.9834566712379456,
      "q_g1": 0.016543304547667503,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 222.57212829589844,
      "loss_g0": 222.02325439453125,
      "loss_g1": 255.1947479248047,
      "q_g0": 0.9834533333778381,
      "q_g1": 0.016546696424484253,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 269.52374267578125,
      "loss_g0": 269.2978210449219,
      "loss_g1": 282.94903564453125,
      "q_g0": 0.9834519624710083,
      "q_g1": 0.016548093408346176,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 248.40643310546875,
      "loss_g0": 249.20669555664062,
      "loss_g1": 200.83151245117188,
      "q_g0": 0.9834569096565247,
      "q_g1": 0.01654314249753952,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 228.13400268554688,
      "loss_g0": 228.27391052246094,
      "loss_g1": 219.8164520263672,
      "q_g0": 0.9834576845169067,
      "q_g1": 0.01654227450489998,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 237.7344512939453,
      "loss_g0": 238.3851776123047,
      "loss_g1": 199.03807067871094,
      "q_g0": 0.9834617376327515,
      "q_g1": 0.01653824932873249,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 214.71188354492188,
      "loss_g0": 214.3752899169922,
      "loss_g1": 234.7249755859375,
      "q_g0": 0.9834596514701843,
      "q_g1": 0.016540326178073883,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 236.21055603027344,
      "loss_g0": 236.9525146484375,
      "loss_g1": 192.08236694335938,
      "q_g0": 0.9834643006324768,
      "q_g1": 0.016535736620426178,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 216.82534790039062,
      "loss_g0": 217.0988006591797,
      "loss_g1": 200.56100463867188,
      "q_g0": 0.9834660291671753,
      "q_g1": 0.016534045338630676,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 223.67774963378906,
      "loss_g0": 223.7903594970703,
      "loss_g1": 216.97866821289062,
      "q_g0": 0.9834666848182678,
      "q_g1": 0.016533348709344864,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 206.8870391845703,
      "loss_g0": 206.44699096679688,
      "loss_g1": 233.0584259033203,
      "q_g0": 0.9834639430046082,
      "q_g1": 0.016536060720682144,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 201.25489807128906,
      "loss_g0": 201.28443908691406,
      "loss_g1": 199.49676513671875,
      "q_g0": 0.9834641814231873,
      "q_g1": 0.01653587631881237,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 202.05899047851562,
      "loss_g0": 0.0,
      "loss_g1": 202.05899047851562,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 212.66360473632812,
      "loss_g0": 197.25198364257812,
      "loss_g1": 212.92251586914062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 194.9060821533203,
      "loss_g0": 253.83009338378906,
      "loss_g1": 193.915771484375,
      "q_g0": 0.016528548672795296,
      "q_g1": 0.9834713935852051,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 200.65560913085938,
      "loss_g0": 0.0,
      "loss_g1": 200.65560913085938,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 187.28753662109375,
      "loss_g0": 169.4910888671875,
      "loss_g1": 187.58651733398438,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 192.37545776367188,
      "loss_g0": 150.45379638671875,
      "loss_g1": 193.07974243164062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 181.6822509765625,
      "loss_g0": 133.9823760986328,
      "loss_g1": 182.48361206054688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 188.9638214111328,
      "loss_g0": 148.04742431640625,
      "loss_g1": 189.65121459960938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 201.58486938476562,
      "loss_g0": 182.504638671875,
      "loss_g1": 201.90541076660156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 146.7854766845703,
      "loss_g0": 183.809814453125,
      "loss_g1": 146.163330078125,
      "q_g0": 0.016526278108358383,
      "q_g1": 0.9834737181663513,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 173.2490234375,
      "loss_g0": 164.84429931640625,
      "loss_g1": 173.3902587890625,
      "q_g0": 0.016525395214557648,
      "q_g1": 0.9834746718406677,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 141.7167205810547,
      "loss_g0": 108.54739379882812,
      "loss_g1": 142.27395629882812,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 150.77423095703125,
      "loss_g0": 179.15802001953125,
      "loss_g1": 150.29730224609375,
      "q_g0": 0.016525372862815857,
      "q_g1": 0.9834746718406677,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 122.87061309814453,
      "loss_g0": 129.1862030029297,
      "loss_g1": 122.76448822021484,
      "q_g0": 0.01652602106332779,
      "q_g1": 0.9834739565849304,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 103.22283172607422,
      "loss_g0": 148.37953186035156,
      "loss_g1": 102.46380615234375,
      "q_g0": 0.01653071865439415,
      "q_g1": 0.9834692478179932,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 135.81503295898438,
      "loss_g0": 137.55026245117188,
      "loss_g1": 135.78585815429688,
      "q_g0": 0.016530897468328476,
      "q_g1": 0.9834691286087036,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 148.46597290039062,
      "loss_g0": 136.54795837402344,
      "loss_g1": 148.66627502441406,
      "q_g0": 0.01652964949607849,
      "q_g1": 0.9834703803062439,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 131.0395050048828,
      "loss_g0": 124.43327331542969,
      "loss_g1": 131.15052795410156,
      "q_g0": 0.016528960317373276,
      "q_g1": 0.9834710955619812,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 118.94525146484375,
      "loss_g0": 158.39170837402344,
      "loss_g1": 118.28211975097656,
      "q_g0": 0.016533059999346733,
      "q_g1": 0.9834669232368469,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 115.74974822998047,
      "loss_g0": 89.44969177246094,
      "loss_g1": 116.19181060791016,
      "q_g0": 0.01653033122420311,
      "q_g1": 0.9834696650505066,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 145.71083068847656,
      "loss_g0": 129.25779724121094,
      "loss_g1": 145.98733520507812,
      "q_g0": 0.016528626903891563,
      "q_g1": 0.9834713935852051,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 130.25064086914062,
      "loss_g0": 130.25064086914062,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 115.065673828125,
      "loss_g0": 114.963134765625,
      "loss_g1": 121.16889190673828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 77.94771575927734,
      "loss_g0": 77.11522674560547,
      "loss_g1": 127.48432922363281,
      "q_g0": 0.983472466468811,
      "q_g1": 0.0165275726467371,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 94.59320068359375,
      "loss_g0": 94.56941223144531,
      "loss_g1": 96.0090103149414,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.01652771793305874,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 94.80307006835938,
      "loss_g0": 95.14269256591797,
      "loss_g1": 74.59114074707031,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.016525618731975555,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 76.70138549804688,
      "loss_g0": 76.3037338256836,
      "loss_g1": 100.36293029785156,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.016528083011507988,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 85.45660400390625,
      "loss_g0": 85.17283630371094,
      "loss_g1": 102.34011840820312,
      "q_g0": 0.9834701418876648,
      "q_g1": 0.016529835760593414,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 80.10023498535156,
      "loss_g0": 80.20394134521484,
      "loss_g1": 73.92939758300781,
      "q_g0": 0.9834708571434021,
      "q_g1": 0.01652919314801693,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 84.09757995605469,
      "loss_g0": 0.0,
      "loss_g1": 84.09757995605469,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 85.19010925292969,
      "loss_g0": 65.02141571044922,
      "loss_g1": 85.52894592285156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 454.4266662597656,
      "loss_g0": 370.0656433105469,
      "loss_g1": 455.84393310546875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 362.391357421875,
      "loss_g0": 362.391357421875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 318.72869873046875,
      "loss_g0": 318.87396240234375,
      "loss_g1": 310.081298828125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 313.43499755859375,
      "loss_g0": 0.0,
      "loss_g1": 313.43499755859375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 299.1062316894531,
      "loss_g0": 296.49603271484375,
      "loss_g1": 299.15008544921875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 262.6250915527344,
      "loss_g0": 287.61474609375,
      "loss_g1": 262.2052001953125,
      "q_g0": 0.0165250226855278,
      "q_g1": 0.9834749698638916,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 251.27403259277344,
      "loss_g0": 280.2896423339844,
      "loss_g1": 250.78639221191406,
      "q_g0": 0.016528038308024406,
      "q_g1": 0.9834719896316528,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 277.0328674316406,
      "loss_g0": 235.3437957763672,
      "loss_g1": 277.7332763671875,
      "q_g0": 0.016523705795407295,
      "q_g1": 0.9834762811660767,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 218.9482879638672,
      "loss_g0": 237.62416076660156,
      "loss_g1": 218.63446044921875,
      "q_g0": 0.016525648534297943,
      "q_g1": 0.9834743142127991,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 258.24652099609375,
      "loss_g0": 257.3614807128906,
      "loss_g1": 258.2613830566406,
      "q_g0": 0.01652555726468563,
      "q_g1": 0.9834744334220886,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 244.41525268554688,
      "loss_g0": 272.5512390136719,
      "loss_g1": 243.94239807128906,
      "q_g0": 0.016528479754924774,
      "q_g1": 0.9834715127944946,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 199.0032501220703,
      "loss_g0": 222.3065185546875,
      "loss_g1": 198.61154174804688,
      "q_g0": 0.016530906781554222,
      "q_g1": 0.9834691286087036,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 244.77374267578125,
      "loss_g0": 0.0,
      "loss_g1": 244.77374267578125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 247.5575714111328,
      "loss_g0": 231.5243377685547,
      "loss_g1": 247.82691955566406,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 185.82455444335938,
      "loss_g0": 185.82455444335938,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 181.24168395996094,
      "loss_g0": 180.98968505859375,
      "loss_g1": 196.2412109375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 210.4136962890625,
      "loss_g0": 0.0,
      "loss_g1": 210.4136962890625,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 186.65966796875,
      "loss_g0": 0.0,
      "loss_g1": 186.65966796875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 210.9446563720703,
      "loss_g0": 174.48947143554688,
      "loss_g1": 211.55709838867188,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 202.9586639404297,
      "loss_g0": 172.77053833007812,
      "loss_g1": 203.4658203125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 209.22193908691406,
      "loss_g0": 152.69439697265625,
      "loss_g1": 210.1715850830078,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 216.198974609375,
      "loss_g0": 163.77532958984375,
      "loss_g1": 217.07969665527344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 135.88150024414062,
      "loss_g0": 183.05455017089844,
      "loss_g1": 135.0887451171875,
      "q_g0": 0.016527332365512848,
      "q_g1": 0.9834727048873901,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 157.12362670898438,
      "loss_g0": 153.4152069091797,
      "loss_g1": 157.18594360351562,
      "q_g0": 0.016526944935321808,
      "q_g1": 0.9834730625152588,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 142.32212829589844,
      "loss_g0": 145.03253173828125,
      "loss_g1": 142.2765655517578,
      "q_g0": 0.016527224332094193,
      "q_g1": 0.9834728240966797,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 192.0074462890625,
      "loss_g0": 155.8519287109375,
      "loss_g1": 192.61489868164062,
      "q_g0": 0.016523465514183044,
      "q_g1": 0.9834765195846558,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 148.373046875,
      "loss_g0": 148.54132080078125,
      "loss_g1": 148.37022399902344,
      "q_g0": 0.01652348041534424,
      "q_g1": 0.9834765195846558,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 182.9847412109375,
      "loss_g0": 137.20957946777344,
      "loss_g1": 183.75376892089844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 149.55722045898438,
      "loss_g0": 190.01779174804688,
      "loss_g1": 148.8773193359375,
      "q_g0": 0.016526635736227036,
      "q_g1": 0.9834733605384827,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 132.37850952148438,
      "loss_g0": 176.3566131591797,
      "loss_g1": 131.6392822265625,
      "q_g0": 0.01653120666742325,
      "q_g1": 0.983468770980835,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 138.42640686035156,
      "loss_g0": 112.0230484008789,
      "loss_g1": 138.8701629638672,
      "q_g0": 0.01652846299111843,
      "q_g1": 0.9834715127944946,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 147.8395538330078,
      "loss_g0": 125.3426742553711,
      "loss_g1": 148.21759033203125,
      "q_g0": 0.016526123508810997,
      "q_g1": 0.9834738373756409,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 157.70639038085938,
      "loss_g0": 155.9089813232422,
      "loss_g1": 157.73660278320312,
      "q_g0": 0.01652594469487667,
      "q_g1": 0.98347407579422,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 122.05325317382812,
      "loss_g0": 114.7829818725586,
      "loss_g1": 122.1754150390625,
      "q_g0": 0.016525182873010635,
      "q_g1": 0.9834747910499573,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 142.2957000732422,
      "loss_g0": 125.79959869384766,
      "loss_g1": 142.57286071777344,
      "q_g0": 0.016523465514183044,
      "q_g1": 0.9834765195846558,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 140.5316162109375,
      "loss_g0": 117.9286117553711,
      "loss_g1": 140.91134643554688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 117.7083740234375,
      "loss_g0": 113.10155487060547,
      "loss_g1": 117.7857666015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 132.7000732421875,
      "loss_g0": 103.71299743652344,
      "loss_g1": 133.1870574951172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 145.6149444580078,
      "loss_g0": 112.21115112304688,
      "loss_g1": 146.17611694335938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 124.71269226074219,
      "loss_g0": 114.60391235351562,
      "loss_g1": 124.88252258300781,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 153.9689178466797,
      "loss_g0": 107.89002990722656,
      "loss_g1": 154.7430419921875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 105.98776245117188,
      "loss_g0": 90.29940795898438,
      "loss_g1": 106.25132751464844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 90.23934173583984,
      "loss_g0": 96.17920684814453,
      "loss_g1": 90.1395492553711,
      "q_g0": 0.016523046419024467,
      "q_g1": 0.9834769368171692,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 98.55781555175781,
      "loss_g0": 60.671058654785156,
      "loss_g1": 99.1943130493164,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 87.2412109375,
      "loss_g0": 100.50309753417969,
      "loss_g1": 87.01839447021484,
      "q_g0": 0.016523806378245354,
      "q_g1": 0.9834761619567871,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 85.45960998535156,
      "loss_g0": 57.03482437133789,
      "loss_g1": 85.93714904785156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 87.26375579833984,
      "loss_g0": 87.26375579833984,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 107.87255859375,
      "loss_g0": 0.0,
      "loss_g1": 107.87255859375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 83.22695922851562,
      "loss_g0": 70.79007720947266,
      "loss_g1": 83.43589782714844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 54.80500030517578,
      "loss_g0": 46.67091369628906,
      "loss_g1": 54.941650390625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 427.6156921386719,
      "loss_g0": 363.0391845703125,
      "loss_g1": 428.7005615234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 367.0525207519531,
      "loss_g0": 318.8278503417969,
      "loss_g1": 367.8626708984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 310.0548095703125,
      "loss_g0": 310.0548095703125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 306.3904724121094,
      "loss_g0": 305.8318786621094,
      "loss_g1": 339.6390380859375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 279.4601745605469,
      "loss_g0": 278.7676086425781,
      "loss_g1": 320.6747741699219,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.016526702791452408,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 289.065673828125,
      "loss_g0": 289.9884948730469,
      "loss_g1": 234.1360626220703,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 312.37109375,
      "loss_g0": 313.4814453125,
      "loss_g1": 246.27867126464844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 280.2182312011719,
      "loss_g0": 280.1354675292969,
      "loss_g1": 285.1448669433594,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522930935025215,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 259.2483825683594,
      "loss_g0": 258.8448181152344,
      "loss_g1": 283.2666320800781,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525432467460632,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 239.06591796875,
      "loss_g0": 238.9673309326172,
      "loss_g1": 244.9333038330078,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652604527771473,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 241.4402313232422,
      "loss_g0": 241.45301818847656,
      "loss_g1": 240.6785888671875,
      "q_g0": 0.98347407579422,
      "q_g1": 0.01652596890926361,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 253.47604370117188,
      "loss_g0": 253.23074340820312,
      "loss_g1": 268.0720520019531,
      "q_g0": 0.983472466468811,
      "q_g1": 0.016527486965060234,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 215.6484375,
      "loss_g0": 215.68637084960938,
      "loss_g1": 213.39071655273438,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.01652725413441658,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 233.64065551757812,
      "loss_g0": 233.64065551757812,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 254.20413208007812,
      "loss_g0": 253.82522583007812,
      "loss_g1": 276.7574462890625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 187.37799072265625,
      "loss_g0": 186.51084899902344,
      "loss_g1": 238.97669982910156,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.016527781262993813,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 236.48248291015625,
      "loss_g0": 237.0540008544922,
      "loss_g1": 202.4666748046875,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652424782514572,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 219.942626953125,
      "loss_g0": 219.942626953125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 196.27061462402344,
      "loss_g0": 195.9582977294922,
      "loss_g1": 214.86090087890625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 189.9998016357422,
      "loss_g0": 190.16600036621094,
      "loss_g1": 180.10711669921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 186.8645782470703,
      "loss_g0": 185.95838928222656,
      "loss_g1": 240.7852783203125,
      "q_g0": 0.9834719896316528,
      "q_g1": 0.01652802899479866,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 190.8350067138672,
      "loss_g0": 190.49879455566406,
      "loss_g1": 210.83798217773438,
      "q_g0": 0.9834699034690857,
      "q_g1": 0.016530107706785202,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 208.12322998046875,
      "loss_g0": 208.65809631347656,
      "loss_g1": 176.29519653320312,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.01652679592370987,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 186.49734497070312,
      "loss_g0": 186.47509765625,
      "loss_g1": 187.8209228515625,
      "q_g0": 0.9834730625152588,
      "q_g1": 0.01652693562209606,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 176.9654998779297,
      "loss_g0": 177.0729217529297,
      "loss_g1": 170.57334899902344,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526270657777786,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 157.97235107421875,
      "loss_g0": 157.35581970214844,
      "loss_g1": 194.6534881591797,
      "q_g0": 0.9834699034690857,
      "q_g1": 0.016530083492398262,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 200.76422119140625,
      "loss_g0": 200.88330078125,
      "loss_g1": 193.67918395996094,
      "q_g0": 0.983470618724823,
      "q_g1": 0.016529347747564316,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 174.4368438720703,
      "loss_g0": 174.95883178710938,
      "loss_g1": 143.3731689453125,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.0165261160582304,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 154.80262756347656,
      "loss_g0": 154.80262756347656,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 152.73719787597656,
      "loss_g0": 153.23583984375,
      "loss_g1": 123.05596923828125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 122.85673522949219,
      "loss_g0": 122.07622528076172,
      "loss_g1": 169.30223083496094,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.016527248546481133,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 153.52239990234375,
      "loss_g0": 153.28282165527344,
      "loss_g1": 167.7784423828125,
      "q_g0": 0.9834713339805603,
      "q_g1": 0.016528727486729622,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 127.13094329833984,
      "loss_g0": 126.90238189697266,
      "loss_g1": 140.729736328125,
      "q_g0": 0.9834697842597961,
      "q_g1": 0.01653013750910759,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 144.1905975341797,
      "loss_g0": 144.2291259765625,
      "loss_g1": 141.89842224121094,
      "q_g0": 0.9834700226783752,
      "q_g1": 0.016529902815818787,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 131.26089477539062,
      "loss_g0": 131.0312957763672,
      "loss_g1": 144.91998291015625,
      "q_g0": 0.9834686517715454,
      "q_g1": 0.01653132401406765,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 129.11741638183594,
      "loss_g0": 128.86061096191406,
      "loss_g1": 144.3931121826172,
      "q_g0": 0.9834670424461365,
      "q_g1": 0.01653292030096054,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 135.05213928222656,
      "loss_g0": 134.85655212402344,
      "loss_g1": 146.68603515625,
      "q_g0": 0.9834659099578857,
      "q_g1": 0.01653413102030754,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 97.94105529785156,
      "loss_g0": 97.1906967163086,
      "loss_g1": 142.56005859375,
      "q_g0": 0.9834612607955933,
      "q_g1": 0.016538776457309723,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 184.00521850585938,
      "loss_g0": 184.84518432617188,
      "loss_g1": 134.04190063476562,
      "q_g0": 0.9834664463996887,
      "q_g1": 0.016533581539988518,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 155.12289428710938,
      "loss_g0": 155.43927001953125,
      "loss_g1": 136.301025390625,
      "q_g0": 0.9834684133529663,
      "q_g1": 0.016531627625226974,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 106.4235610961914,
      "loss_g0": 106.28984069824219,
      "loss_g1": 114.37832641601562,
      "q_g0": 0.9834675192832947,
      "q_g1": 0.016532454639673233,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 116.51957702636719,
      "loss_g0": 0.0,
      "loss_g1": 116.51957702636719,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 60.89330291748047,
      "loss_g0": 93.93791198730469,
      "loss_g1": 60.33815383911133,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 100.4786148071289,
      "loss_g0": 107.69120788574219,
      "loss_g1": 100.35743713378906,
      "q_g0": 0.016523178666830063,
      "q_g1": 0.9834768176078796,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 113.08123016357422,
      "loss_g0": 76.0480728149414,
      "loss_g1": 113.70338439941406,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 84.57875061035156,
      "loss_g0": 84.57875061035156,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 91.2393798828125,
      "loss_g0": 91.2393798828125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 99.99828338623047,
      "loss_g0": 100.40999603271484,
      "loss_g1": 75.49107360839844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 87.2086410522461,
      "loss_g0": 87.1842269897461,
      "loss_g1": 88.66159057617188,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652258075773716,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 86.82357025146484,
      "loss_g0": 0.0,
      "loss_g1": 86.82357025146484,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 401.89013671875,
      "loss_g0": 418.96209716796875,
      "loss_g1": 401.60333251953125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 352.4595642089844,
      "loss_g0": 0.0,
      "loss_g1": 352.4595642089844,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 334.73858642578125,
      "loss_g0": 299.5201110839844,
      "loss_g1": 335.33026123046875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 315.6595764160156,
      "loss_g0": 320.9737243652344,
      "loss_g1": 315.5702819824219,
      "q_g0": 0.016522977501153946,
      "q_g1": 0.9834770560264587,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 279.6299743652344,
      "loss_g0": 273.0989074707031,
      "loss_g1": 279.73968505859375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 291.0751037597656,
      "loss_g0": 250.4845733642578,
      "loss_g1": 291.75701904296875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 290.44757080078125,
      "loss_g0": 290.44757080078125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 267.10821533203125,
      "loss_g0": 268.2845153808594,
      "loss_g1": 197.08963012695312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 194.7831268310547,
      "loss_g0": 193.91436767578125,
      "loss_g1": 246.47752380371094,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.016527798026800156,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 266.5652160644531,
      "loss_g0": 266.7867126464844,
      "loss_g1": 253.3839111328125,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.016526425257325172,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 258.0390625,
      "loss_g0": 259.1910095214844,
      "loss_g1": 189.47071838378906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 267.8633728027344,
      "loss_g0": 268.2629089355469,
      "loss_g1": 244.0813446044922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 230.080078125,
      "loss_g0": 229.91946411132812,
      "loss_g1": 239.638916015625,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523417085409164,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 217.65101623535156,
      "loss_g0": 217.25265502929688,
      "loss_g1": 241.3572998046875,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525881364941597,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 220.19212341308594,
      "loss_g0": 220.83834838867188,
      "loss_g1": 181.72633361816406,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 191.8031463623047,
      "loss_g0": 191.7117919921875,
      "loss_g1": 197.24066162109375,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652299053966999,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 200.1549530029297,
      "loss_g0": 200.33749389648438,
      "loss_g1": 189.2890167236328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 166.48104858398438,
      "loss_g0": 165.85194396972656,
      "loss_g1": 203.91873168945312,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526319086551666,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 238.29238891601562,
      "loss_g0": 239.85028076171875,
      "loss_g1": 145.5607147216797,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 199.58135986328125,
      "loss_g0": 198.32296752929688,
      "loss_g1": 274.4501647949219,
      "q_g0": 0.9834697842597961,
      "q_g1": 0.016530200839042664,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 167.31443786621094,
      "loss_g0": 166.78875732421875,
      "loss_g1": 198.5828857421875,
      "q_g0": 0.9834665656089783,
      "q_g1": 0.01653345674276352,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 163.10462951660156,
      "loss_g0": 162.79893493652344,
      "loss_g1": 181.28692626953125,
      "q_g0": 0.9834645986557007,
      "q_g1": 0.016535339877009392,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 186.1232452392578,
      "loss_g0": 186.45030212402344,
      "loss_g1": 166.66827392578125,
      "q_g0": 0.9834666848182678,
      "q_g1": 0.016533324494957924,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 127.8675308227539,
      "loss_g0": 127.21809387207031,
      "loss_g1": 166.48907470703125,
      "q_g0": 0.9834626317024231,
      "q_g1": 0.016537342220544815,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 161.97506713867188,
      "loss_g0": 161.56809997558594,
      "loss_g1": 186.17349243164062,
      "q_g0": 0.9834601283073425,
      "q_g1": 0.016539862379431725,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 163.77784729003906,
      "loss_g0": 163.52359008789062,
      "loss_g1": 178.89418029785156,
      "q_g0": 0.9834585189819336,
      "q_g1": 0.016541436314582825,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 174.39303588867188,
      "loss_g0": 174.33901977539062,
      "loss_g1": 177.60426330566406,
      "q_g0": 0.9834582805633545,
      "q_g1": 0.016541769728064537,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 136.67042541503906,
      "loss_g0": 136.55624389648438,
      "loss_g1": 143.45855712890625,
      "q_g0": 0.9834575653076172,
      "q_g1": 0.0165424682199955,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 152.65774536132812,
      "loss_g0": 152.33493041992188,
      "loss_g1": 171.8468017578125,
      "q_g0": 0.9834555387496948,
      "q_g1": 0.016544470563530922,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 133.21665954589844,
      "loss_g0": 132.92611694335938,
      "loss_g1": 150.48541259765625,
      "q_g0": 0.9834538102149963,
      "q_g1": 0.01654626987874508,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 143.23214721679688,
      "loss_g0": 143.3336181640625,
      "loss_g1": 137.20111083984375,
      "q_g0": 0.9834543466567993,
      "q_g1": 0.01654564030468464,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 140.89202880859375,
      "loss_g0": 140.66297912597656,
      "loss_g1": 154.50576782226562,
      "q_g0": 0.9834529757499695,
      "q_g1": 0.016547061502933502,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 127.5229263305664,
      "loss_g0": 127.07938385009766,
      "loss_g1": 153.8793182373047,
      "q_g0": 0.9834502339363098,
      "q_g1": 0.016549808904528618,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 140.42466735839844,
      "loss_g0": 140.31854248046875,
      "loss_g1": 146.7305908203125,
      "q_g0": 0.9834495186805725,
      "q_g1": 0.016550466418266296,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 138.4359893798828,
      "loss_g0": 138.3082275390625,
      "loss_g1": 146.0267333984375,
      "q_g0": 0.9834486842155457,
      "q_g1": 0.01655125804245472,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 121.20405578613281,
      "loss_g0": 120.55809020996094,
      "loss_g1": 159.57754516601562,
      "q_g0": 0.9834448099136353,
      "q_g1": 0.016555259004235268,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 124.45024108886719,
      "loss_g0": 124.26124572753906,
      "loss_g1": 135.67686462402344,
      "q_g0": 0.9834434986114502,
      "q_g1": 0.016556428745388985,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 116.55746459960938,
      "loss_g0": 116.55746459960938,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 103.58253479003906,
      "loss_g0": 103.3933334350586,
      "loss_g1": 114.84452819824219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 97.90311431884766,
      "loss_g0": 97.55760192871094,
      "loss_g1": 118.46673583984375,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.01652456447482109,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 117.46882629394531,
      "loss_g0": 117.22797393798828,
      "loss_g1": 131.8019256591797,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.016526060178875923,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 114.20413208007812,
      "loss_g0": 114.50531768798828,
      "loss_g1": 96.27806854248047,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652420125901699,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 74.67965698242188,
      "loss_g0": 73.79812622070312,
      "loss_g1": 127.1281509399414,
      "q_g0": 0.9834703803062439,
      "q_g1": 0.016529658809304237,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 95.64238739013672,
      "loss_g0": 95.31536865234375,
      "loss_g1": 115.09690856933594,
      "q_g0": 0.9834682941436768,
      "q_g1": 0.016531681641936302,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 84.4861831665039,
      "loss_g0": 84.13311767578125,
      "loss_g1": 105.487060546875,
      "q_g0": 0.9834660887718201,
      "q_g1": 0.016533859074115753,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 92.41568756103516,
      "loss_g0": 92.57470703125,
      "loss_g1": 82.95606231689453,
      "q_g0": 0.983467161655426,
      "q_g1": 0.01653287559747696,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 86.17507934570312,
      "loss_g0": 86.39400482177734,
      "loss_g1": 73.15108489990234,
      "q_g0": 0.9834684133529663,
      "q_g1": 0.01653151772916317,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 86.80953979492188,
      "loss_g0": 86.9007797241211,
      "loss_g1": 81.38128662109375,
      "q_g0": 0.9834690093994141,
      "q_g1": 0.01653095707297325,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 86.71360778808594,
      "loss_g0": 86.62411499023438,
      "loss_g1": 92.03736114501953,
      "q_g0": 0.9834685325622559,
      "q_g1": 0.016531510278582573,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 69.03538513183594,
      "loss_g0": 69.34842681884766,
      "loss_g1": 50.41005325317383,
      "q_g0": 0.9834703803062439,
      "q_g1": 0.016529571264982224,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 280.2710266113281,
      "loss_g0": 278.4482116699219,
      "loss_g1": 388.6491394042969,
      "q_g0": 0.9834591746330261,
      "q_g1": 0.016540847718715668,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 367.913818359375,
      "loss_g0": 367.4896240234375,
      "loss_g1": 393.13165283203125,
      "q_g0": 0.983456552028656,
      "q_g1": 0.016543468460440636,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 350.33648681640625,
      "loss_g0": 350.1234130859375,
      "loss_g1": 363.002685546875,
      "q_g0": 0.9834551811218262,
      "q_g1": 0.01654478907585144,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 344.81005859375,
      "loss_g0": 345.3973083496094,
      "loss_g1": 309.8948974609375,
      "q_g0": 0.9834588766098022,
      "q_g1": 0.01654115691781044,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 264.93218994140625,
      "loss_g0": 264.204345703125,
      "loss_g1": 308.1936950683594,
      "q_g0": 0.9834543466567993,
      "q_g1": 0.016545657068490982,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 302.3821105957031,
      "loss_g0": 302.3572082519531,
      "loss_g1": 303.86236572265625,
      "q_g0": 0.9834542274475098,
      "q_g1": 0.016545811668038368,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 208.9618377685547,
      "loss_g0": 207.77886962890625,
      "loss_g1": 279.2440185546875,
      "q_g0": 0.9834468364715576,
      "q_g1": 0.016553130000829697,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 280.0539855957031,
      "loss_g0": 280.2958984375,
      "loss_g1": 265.6787109375,
      "q_g0": 0.9834483861923218,
      "q_g1": 0.016551639884710312,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 292.9195556640625,
      "loss_g0": 293.4546813964844,
      "loss_g1": 261.11590576171875,
      "q_g0": 0.9834517240524292,
      "q_g1": 0.01654832623898983,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 278.10980224609375,
      "loss_g0": 278.5671081542969,
      "loss_g1": 250.9284210205078,
      "q_g0": 0.9834544658660889,
      "q_g1": 0.01654549315571785,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 260.0344543457031,
      "loss_g0": 260.7236328125,
      "loss_g1": 219.058349609375,
      "q_g0": 0.9834587574005127,
      "q_g1": 0.016541235148906708,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 246.20980834960938,
      "loss_g0": 0.0,
      "loss_g1": 246.20980834960938,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 231.8926544189453,
      "loss_g0": 215.5196990966797,
      "loss_g1": 232.16770935058594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 224.5858612060547,
      "loss_g0": 240.52215576171875,
      "loss_g1": 224.31809997558594,
      "q_g0": 0.01652408577501774,
      "q_g1": 0.983475923538208,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 162.42230224609375,
      "loss_g0": 210.20138549804688,
      "loss_g1": 161.61927795410156,
      "q_g0": 0.016529053449630737,
      "q_g1": 0.9834709763526917,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 226.885986328125,
      "loss_g0": 199.2014923095703,
      "loss_g1": 227.3511962890625,
      "q_g0": 0.016526177525520325,
      "q_g1": 0.9834738373756409,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 201.7717742919922,
      "loss_g0": 237.32763671875,
      "loss_g1": 201.1741485595703,
      "q_g0": 0.016529882326722145,
      "q_g1": 0.9834701418876648,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 203.31057739257812,
      "loss_g0": 0.0,
      "loss_g1": 203.31057739257812,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 215.15647888183594,
      "loss_g0": 191.10891723632812,
      "loss_g1": 215.56048583984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 150.5511932373047,
      "loss_g0": 193.49160766601562,
      "loss_g1": 149.82958984375,
      "q_g0": 0.016526883468031883,
      "q_g1": 0.9834731221199036,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 182.11166381835938,
      "loss_g0": 161.81741333007812,
      "loss_g1": 182.45266723632812,
      "q_g0": 0.016524773091077805,
      "q_g1": 0.9834752082824707,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 190.2835693359375,
      "loss_g0": 204.56752014160156,
      "loss_g1": 190.04354858398438,
      "q_g0": 0.016526253893971443,
      "q_g1": 0.9834737181663513,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 182.51255798339844,
      "loss_g0": 205.94041442871094,
      "loss_g1": 182.11880493164062,
      "q_g0": 0.016528688371181488,
      "q_g1": 0.9834713339805603,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 177.43548583984375,
      "loss_g0": 177.43548583984375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 169.40745544433594,
      "loss_g0": 169.1002960205078,
      "loss_g1": 187.69065856933594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 199.51519775390625,
      "loss_g0": 199.8824920654297,
      "loss_g1": 177.6523895263672,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 126.08958435058594,
      "loss_g0": 125.05241394042969,
      "loss_g1": 187.80096435546875,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.016528837382793427,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 214.80458068847656,
      "loss_g0": 215.07919311523438,
      "loss_g1": 198.4629364013672,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.01652713678777218,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 191.75582885742188,
      "loss_g0": 192.07528686523438,
      "loss_g1": 172.74386596679688,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525164246559143,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 147.10037231445312,
      "loss_g0": 146.98617553710938,
      "loss_g1": 153.89657592773438,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.016525868326425552,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 160.75430297851562,
      "loss_g0": 160.67062377929688,
      "loss_g1": 165.7342071533203,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.01652637869119644,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 153.205810546875,
      "loss_g0": 153.20965576171875,
      "loss_g1": 152.97715759277344,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.0165263544768095,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 160.06863403320312,
      "loss_g0": 160.24986267089844,
      "loss_g1": 149.28260803222656,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.016525231301784515,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 171.99630737304688,
      "loss_g0": 0.0,
      "loss_g1": 171.99630737304688,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 147.738525390625,
      "loss_g0": 147.738525390625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 118.03624725341797,
      "loss_g0": 117.49598693847656,
      "loss_g1": 150.19439697265625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 138.89093017578125,
      "loss_g0": 138.88400268554688,
      "loss_g1": 139.30360412597656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522472724318504,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 106.06074523925781,
      "loss_g0": 105.55615234375,
      "loss_g1": 136.09024047851562,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.016525588929653168,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 120.26691436767578,
      "loss_g0": 120.79339599609375,
      "loss_g1": 88.92839050292969,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 104.08998107910156,
      "loss_g0": 104.08998107910156,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 107.19436645507812,
      "loss_g0": 106.83087158203125,
      "loss_g1": 128.8310546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 100.4368896484375,
      "loss_g0": 100.32750701904297,
      "loss_g1": 106.94729614257812,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523102298378944,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 104.200927734375,
      "loss_g0": 104.200927734375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 122.73503112792969,
      "loss_g0": 123.26339721679688,
      "loss_g1": 91.28459167480469,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 81.54654693603516,
      "loss_g0": 81.54654693603516,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 86.57559967041016,
      "loss_g0": 86.83583068847656,
      "loss_g1": 71.08577728271484,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 79.7540283203125,
      "loss_g0": 79.78932189941406,
      "loss_g1": 77.65302276611328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 73.48826599121094,
      "loss_g0": 73.79766845703125,
      "loss_g1": 55.071163177490234,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 86.99971771240234,
      "loss_g0": 87.48873901367188,
      "loss_g1": 57.891170501708984,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 67.82674407958984,
      "loss_g0": 67.86042785644531,
      "loss_g1": 65.82176208496094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 383.0563049316406,
      "loss_g0": 383.9338684082031,
      "loss_g1": 330.8204345703125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 289.9089660644531,
      "loss_g0": 288.9457702636719,
      "loss_g1": 347.2198181152344,
      "q_g0": 0.9834716320037842,
      "q_g1": 0.016528379172086716,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 342.27606201171875,
      "loss_g0": 342.27606201171875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 322.2868957519531,
      "loss_g0": 322.2326965332031,
      "loss_g1": 325.51251220703125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 272.38836669921875,
      "loss_g0": 272.6989440917969,
      "loss_g1": 253.90127563476562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 300.760009765625,
      "loss_g0": 301.956298828125,
      "loss_g1": 229.55194091796875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 320.435791015625,
      "loss_g0": 321.26513671875,
      "loss_g1": 271.07025146484375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 326.5602722167969,
      "loss_g0": 327.7556457519531,
      "loss_g1": 255.40670776367188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 279.15338134765625,
      "loss_g0": 279.0701599121094,
      "loss_g1": 284.1065979003906,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522938385605812,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 309.3675537109375,
      "loss_g0": 310.7737731933594,
      "loss_g1": 225.66412353515625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 256.75677490234375,
      "loss_g0": 256.777099609375,
      "loss_g1": 255.54598999023438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 245.51905822753906,
      "loss_g0": 0.0,
      "loss_g1": 245.51905822753906,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 258.5574035644531,
      "loss_g0": 214.58700561523438,
      "loss_g1": 259.2961120605469,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 214.0097198486328,
      "loss_g0": 245.63719177246094,
      "loss_g1": 213.478271484375,
      "q_g0": 0.016525711864233017,
      "q_g1": 0.9834743142127991,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 211.2529754638672,
      "loss_g0": 233.28399658203125,
      "loss_g1": 210.8827362060547,
      "q_g0": 0.016527999192476273,
      "q_g1": 0.9834719896316528,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 311.0502624511719,
      "loss_g0": 239.80931091308594,
      "loss_g1": 312.24713134765625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 204.4747772216797,
      "loss_g0": 192.52203369140625,
      "loss_g1": 204.6755828857422,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 244.98460388183594,
      "loss_g0": 244.98460388183594,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 198.84666442871094,
      "loss_g0": 198.67758178710938,
      "loss_g1": 208.910888671875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 191.96482849121094,
      "loss_g0": 191.96482849121094,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 196.6217498779297,
      "loss_g0": 196.95379638671875,
      "loss_g1": 176.8567352294922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 192.19252014160156,
      "loss_g0": 192.50579833984375,
      "loss_g1": 173.5448760986328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 189.4510955810547,
      "loss_g0": 189.409423828125,
      "loss_g1": 191.932373046875,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 188.53297424316406,
      "loss_g0": 188.74874877929688,
      "loss_g1": 175.68995666503906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 202.45870971679688,
      "loss_g0": 202.46945190429688,
      "loss_g1": 201.81883239746094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 199.13401794433594,
      "loss_g0": 0.0,
      "loss_g1": 199.13401794433594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 177.33282470703125,
      "loss_g0": 165.42666625976562,
      "loss_g1": 177.53285217285156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 212.5496826171875,
      "loss_g0": 153.9141845703125,
      "loss_g1": 213.53475952148438,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 187.33380126953125,
      "loss_g0": 151.70602416992188,
      "loss_g1": 187.93234252929688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 164.8559112548828,
      "loss_g0": 157.91000366210938,
      "loss_g1": 164.97259521484375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 169.8064422607422,
      "loss_g0": 176.279052734375,
      "loss_g1": 169.69769287109375,
      "q_g0": 0.016523092985153198,
      "q_g1": 0.9834769368171692,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 146.01040649414062,
      "loss_g0": 168.41488647460938,
      "loss_g1": 145.63394165039062,
      "q_g0": 0.016525425016880035,
      "q_g1": 0.9834745526313782,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 153.81378173828125,
      "loss_g0": 148.85772705078125,
      "loss_g1": 153.89706420898438,
      "q_g0": 0.016524914652109146,
      "q_g1": 0.9834750890731812,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 114.9584732055664,
      "loss_g0": 107.00318908691406,
      "loss_g1": 115.0921401977539,
      "q_g0": 0.01652408577501774,
      "q_g1": 0.983475923538208,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 136.05210876464844,
      "loss_g0": 153.4798583984375,
      "loss_g1": 135.75926208496094,
      "q_g0": 0.01652589812874794,
      "q_g1": 0.98347407579422,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 135.11524963378906,
      "loss_g0": 106.46095275878906,
      "loss_g1": 135.59666442871094,
      "q_g0": 0.016522914171218872,
      "q_g1": 0.9834770560264587,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 133.93333435058594,
      "loss_g0": 112.5372543334961,
      "loss_g1": 134.29278564453125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 98.99627685546875,
      "loss_g0": 105.35531616210938,
      "loss_g1": 98.88944244384766,
      "q_g0": 0.0165230855345726,
      "q_g1": 0.9834769368171692,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 126.19581604003906,
      "loss_g0": 120.2108154296875,
      "loss_g1": 126.29637145996094,
      "q_g0": 0.016522465273737907,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 104.91619110107422,
      "loss_g0": 101.59138488769531,
      "loss_g1": 104.9720458984375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 102.5481185913086,
      "loss_g0": 101.81998443603516,
      "loss_g1": 102.56034851074219,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 114.04532623291016,
      "loss_g0": 114.04532623291016,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 111.6742935180664,
      "loss_g0": 111.85505676269531,
      "loss_g1": 100.9144515991211,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 83.75012969970703,
      "loss_g0": 83.16860961914062,
      "loss_g1": 118.3570556640625,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652601547539234,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 97.4315414428711,
      "loss_g0": 97.51848602294922,
      "loss_g1": 92.2568588256836,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525480896234512,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 91.01341247558594,
      "loss_g0": 90.984375,
      "loss_g1": 92.7417221069336,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.01652565784752369,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 81.58732604980469,
      "loss_g0": 81.5890121459961,
      "loss_g1": 81.48698425292969,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.016525648534297943,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 69.26049041748047,
      "loss_g0": 68.86553955078125,
      "loss_g1": 92.76106262207031,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.016528092324733734,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 54.13591003417969,
      "loss_g0": 53.23420333862305,
      "loss_g1": 107.77188873291016,
      "q_g0": 0.9834663271903992,
      "q_g1": 0.01653367467224598,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 61.54535675048828,
      "loss_g0": 61.3235969543457,
      "loss_g1": 74.73507690429688,
      "q_g0": 0.9834649562835693,
      "q_g1": 0.01653505302965641,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 373.59686279296875,
      "loss_g0": 373.0154724121094,
      "loss_g1": 408.1672668457031,
      "q_g0": 0.9834613800048828,
      "q_g1": 0.016538651660084724,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 350.7883605957031,
      "loss_g0": 350.02191162109375,
      "loss_g1": 396.3529357910156,
      "q_g0": 0.9834566712379456,
      "q_g1": 0.016543392091989517,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 351.1783447265625,
      "loss_g0": 351.6671142578125,
      "loss_g1": 322.11810302734375,
      "q_g0": 0.9834596514701843,
      "q_g1": 0.016540367156267166,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 293.5262145996094,
      "loss_g0": 292.7701416015625,
      "loss_g1": 338.4684753417969,
      "q_g0": 0.9834549427032471,
      "q_g1": 0.016545042395591736,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 311.97149658203125,
      "loss_g0": 312.6546630859375,
      "loss_g1": 271.3537292480469,
      "q_g0": 0.9834591746330261,
      "q_g1": 0.016540808603167534,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 331.16729736328125,
      "loss_g0": 331.12762451171875,
      "loss_g1": 333.52508544921875,
      "q_g0": 0.9834589958190918,
      "q_g1": 0.01654105633497238,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 269.4065246582031,
      "loss_g0": 268.8056335449219,
      "loss_g1": 305.12530517578125,
      "q_g0": 0.9834553003311157,
      "q_g1": 0.016544774174690247,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 242.4285888671875,
      "loss_g0": 241.73658752441406,
      "loss_g1": 283.55120849609375,
      "q_g0": 0.9834508895874023,
      "q_g1": 0.016549047082662582,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 267.8173828125,
      "loss_g0": 267.8173828125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 261.0963134765625,
      "loss_g0": 261.3581237792969,
      "loss_g1": 245.5128631591797,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 279.8467712402344,
      "loss_g0": 280.6060485839844,
      "loss_g1": 234.65133666992188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 282.8738708496094,
      "loss_g0": 283.4237976074219,
      "loss_g1": 250.13821411132812,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 229.81739807128906,
      "loss_g0": 229.6130828857422,
      "loss_g1": 241.97914123535156,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523689031600952,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 302.6246643066406,
      "loss_g0": 303.7462158203125,
      "loss_g1": 235.864990234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 188.46737670898438,
      "loss_g0": 188.24200439453125,
      "loss_g1": 201.88095092773438,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652381382882595,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 232.6381378173828,
      "loss_g0": 232.30245971679688,
      "loss_g1": 252.61441040039062,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525890678167343,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 197.77879333496094,
      "loss_g0": 197.28607177734375,
      "loss_g1": 227.09490966796875,
      "q_g0": 0.9834710955619812,
      "q_g1": 0.016528945416212082,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 212.48240661621094,
      "loss_g0": 212.23532104492188,
      "loss_g1": 227.18316650390625,
      "q_g0": 0.9834696054458618,
      "q_g1": 0.0165304783731699,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 244.6273651123047,
      "loss_g0": 245.1354522705078,
      "loss_g1": 214.3928680419922,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527332365512848,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 177.48171997070312,
      "loss_g0": 177.48007202148438,
      "loss_g1": 177.57847595214844,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527339816093445,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 184.789306640625,
      "loss_g0": 184.789306640625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 176.699462890625,
      "loss_g0": 176.37442016601562,
      "loss_g1": 196.04708862304688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 182.79248046875,
      "loss_g0": 183.11257934570312,
      "loss_g1": 163.73883056640625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 198.56509399414062,
      "loss_g0": 0.0,
      "loss_g1": 198.56509399414062,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 172.9186248779297,
      "loss_g0": 175.53375244140625,
      "loss_g1": 172.87469482421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 168.76806640625,
      "loss_g0": 145.3790283203125,
      "loss_g1": 169.1610107421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 194.8961181640625,
      "loss_g0": 133.12318420410156,
      "loss_g1": 195.93389892578125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 166.3695831298828,
      "loss_g0": 182.24508666992188,
      "loss_g1": 166.10284423828125,
      "q_g0": 0.016524069011211395,
      "q_g1": 0.983475923538208,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 173.69467163085938,
      "loss_g0": 145.21917724609375,
      "loss_g1": 174.17306518554688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 174.97012329101562,
      "loss_g0": 132.59182739257812,
      "loss_g1": 175.6820831298828,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 180.32879638671875,
      "loss_g0": 157.27684020996094,
      "loss_g1": 180.716064453125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 164.634033203125,
      "loss_g0": 118.24786376953125,
      "loss_g1": 165.413330078125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 162.71209716796875,
      "loss_g0": 148.50625610351562,
      "loss_g1": 162.95074462890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 135.57508850097656,
      "loss_g0": 135.57508850097656,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 163.23194885253906,
      "loss_g0": 163.00393676757812,
      "loss_g1": 176.80389404296875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 141.88015747070312,
      "loss_g0": 142.0795135498047,
      "loss_g1": 130.01336669921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 121.0252914428711,
      "loss_g0": 120.24134826660156,
      "loss_g1": 167.67495727539062,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527269035577774,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 156.4921417236328,
      "loss_g0": 156.9747314453125,
      "loss_g1": 127.76962280273438,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652427762746811,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 120.6919174194336,
      "loss_g0": 120.11334991455078,
      "loss_g1": 155.11883544921875,
      "q_g0": 0.9834721088409424,
      "q_g1": 0.016527850180864334,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 129.1767578125,
      "loss_g0": 129.1767578125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 134.0228271484375,
      "loss_g0": 134.0228271484375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 120.76506042480469,
      "loss_g0": 120.15778350830078,
      "loss_g1": 156.9126434326172,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 87.88341522216797,
      "loss_g0": 87.37635040283203,
      "loss_g1": 118.06035614013672,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.01652555726468563,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 105.02780151367188,
      "loss_g0": 105.02495574951172,
      "loss_g1": 105.19705200195312,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.01652558147907257,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 105.22894287109375,
      "loss_g0": 105.22894287109375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 107.97842407226562,
      "loss_g0": 108.0340576171875,
      "loss_g1": 104.66673278808594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 148.34213256835938,
      "loss_g0": 149.28311157226562,
      "loss_g1": 92.33123016357422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 93.6549301147461,
      "loss_g0": 93.28343200683594,
      "loss_g1": 115.76471710205078,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524719074368477,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 51.936187744140625,
      "loss_g0": 51.936187744140625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 72.2411880493164,
      "loss_g0": 72.21743774414062,
      "loss_g1": 73.65487670898438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 436.8741455078125,
      "loss_g0": 0.0,
      "loss_g1": 436.8741455078125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 348.0650634765625,
      "loss_g0": 387.62567138671875,
      "loss_g1": 347.40045166015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 366.5714111328125,
      "loss_g0": 441.33673095703125,
      "loss_g1": 365.31475830078125,
      "q_g0": 0.016530200839042664,
      "q_g1": 0.9834697842597961,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 355.24261474609375,
      "loss_g0": 355.24261474609375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 300.3114013671875,
      "loss_g0": 300.96551513671875,
      "loss_g1": 261.3763122558594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 303.4150390625,
      "loss_g0": 302.60784912109375,
      "loss_g1": 351.4466247558594,
      "q_g0": 0.9834725856781006,
      "q_g1": 0.016527416184544563,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 321.0884094238281,
      "loss_g0": 0.0,
      "loss_g1": 321.0884094238281,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 288.96417236328125,
      "loss_g0": 302.43267822265625,
      "loss_g1": 288.7378845214844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 267.6329345703125,
      "loss_g0": 267.6329345703125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 284.8564758300781,
      "loss_g0": 284.96661376953125,
      "loss_g1": 278.3011169433594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 285.4738464355469,
      "loss_g0": 285.45269775390625,
      "loss_g1": 286.7327880859375,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 235.1081085205078,
      "loss_g0": 234.6887969970703,
      "loss_g1": 260.063232421875,
      "q_g0": 0.983474850654602,
      "q_g1": 0.0165251474827528,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 252.6747283935547,
      "loss_g0": 252.79339599609375,
      "loss_g1": 245.61172485351562,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652441918849945,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 224.89004516601562,
      "loss_g0": 224.60763549804688,
      "loss_g1": 241.6964569091797,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.01652616076171398,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 214.886962890625,
      "loss_g0": 214.4520721435547,
      "loss_g1": 240.76370239257812,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.01652885414659977,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 230.06130981445312,
      "loss_g0": 0.0,
      "loss_g1": 230.06130981445312,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 205.27830505371094,
      "loss_g0": 181.30812072753906,
      "loss_g1": 205.68099975585938,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 250.2097625732422,
      "loss_g0": 236.20118713378906,
      "loss_g1": 250.44509887695312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 214.0222930908203,
      "loss_g0": 161.19418334960938,
      "loss_g1": 214.90980529785156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 234.93780517578125,
      "loss_g0": 224.81236267089844,
      "loss_g1": 235.10791015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 207.4936065673828,
      "loss_g0": 218.8467254638672,
      "loss_g1": 207.3028564453125,
      "q_g0": 0.016523605212569237,
      "q_g1": 0.9834764003753662,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 239.12551879882812,
      "loss_g0": 187.29168701171875,
      "loss_g1": 239.99632263183594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 201.35269165039062,
      "loss_g0": 0.0,
      "loss_g1": 201.35269165039062,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 196.49192810058594,
      "loss_g0": 201.64657592773438,
      "loss_g1": 196.4053192138672,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 210.06005859375,
      "loss_g0": 0.0,
      "loss_g1": 210.06005859375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 185.66200256347656,
      "loss_g0": 230.11619567871094,
      "loss_g1": 184.9151611328125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 222.52194213867188,
      "loss_g0": 195.48597717285156,
      "loss_g1": 222.97613525390625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 181.24978637695312,
      "loss_g0": 173.11151123046875,
      "loss_g1": 181.38650512695312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 162.0019989013672,
      "loss_g0": 168.83966064453125,
      "loss_g1": 161.88711547851562,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 151.5568084716797,
      "loss_g0": 162.91383361816406,
      "loss_g1": 151.36599731445312,
      "q_g0": 0.016524309292435646,
      "q_g1": 0.9834756851196289,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 166.55975341796875,
      "loss_g0": 130.00791931152344,
      "loss_g1": 167.173828125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 147.64822387695312,
      "loss_g0": 126.47559356689453,
      "loss_g1": 148.00390625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 158.6526336669922,
      "loss_g0": 153.51309204101562,
      "loss_g1": 158.73898315429688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 149.9093017578125,
      "loss_g0": 134.88754272460938,
      "loss_g1": 150.1616668701172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 135.1591796875,
      "loss_g0": 135.53004455566406,
      "loss_g1": 135.1529541015625,
      "q_g0": 0.016522465273737907,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 141.19386291503906,
      "loss_g0": 124.5723648071289,
      "loss_g1": 141.4730987548828,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 115.86712646484375,
      "loss_g0": 133.65655517578125,
      "loss_g1": 115.56822204589844,
      "q_g0": 0.016524270176887512,
      "q_g1": 0.9834756851196289,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 149.16119384765625,
      "loss_g0": 126.18681335449219,
      "loss_g1": 149.5471649169922,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 133.65780639648438,
      "loss_g0": 122.42666625976562,
      "loss_g1": 133.84649658203125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 137.63192749023438,
      "loss_g0": 132.59716796875,
      "loss_g1": 137.7165069580078,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 113.61006164550781,
      "loss_g0": 134.89578247070312,
      "loss_g1": 113.25241088867188,
      "q_g0": 0.01652463525533676,
      "q_g1": 0.9834753274917603,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 97.63573455810547,
      "loss_g0": 121.34866333007812,
      "loss_g1": 97.23724365234375,
      "q_g0": 0.01652710698544979,
      "q_g1": 0.9834729433059692,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 111.00477600097656,
      "loss_g0": 99.70539855957031,
      "loss_g1": 111.19464111328125,
      "q_g0": 0.016525927931070328,
      "q_g1": 0.98347407579422,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 143.26248168945312,
      "loss_g0": 105.64317321777344,
      "loss_g1": 143.8944854736328,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 114.05895233154297,
      "loss_g0": 118.45670318603516,
      "loss_g1": 113.98506927490234,
      "q_g0": 0.016522882506251335,
      "q_g1": 0.9834770560264587,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 84.33912658691406,
      "loss_g0": 76.21880340576172,
      "loss_g1": 84.47554016113281,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 78.98297119140625,
      "loss_g0": 41.17829132080078,
      "loss_g1": 79.61808776855469,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 99.1841049194336,
      "loss_g0": 102.65478515625,
      "loss_g1": 99.12579345703125,
      "q_g0": 0.016522781923413277,
      "q_g1": 0.9834771752357483,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 112.85969543457031,
      "loss_g0": 112.85969543457031,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 69.07621765136719,
      "loss_g0": 69.14613342285156,
      "loss_g1": 64.91474151611328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 406.7964172363281,
      "loss_g0": 406.639892578125,
      "loss_g1": 416.1122741699219,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523394733667374,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 307.6073913574219,
      "loss_g0": 307.8539733886719,
      "loss_g1": 292.9290771484375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 283.56378173828125,
      "loss_g0": 282.9544982910156,
      "loss_g1": 319.822265625,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.01652619242668152,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 279.4520568847656,
      "loss_g0": 279.3991394042969,
      "loss_g1": 282.60040283203125,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.016526520252227783,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 266.39208984375,
      "loss_g0": 266.572998046875,
      "loss_g1": 255.627197265625,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.016525402665138245,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 249.93533325195312,
      "loss_g0": 249.93533325195312,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 241.28689575195312,
      "loss_g0": 241.46197509765625,
      "loss_g1": 230.8656005859375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 246.4231719970703,
      "loss_g0": 246.1710968017578,
      "loss_g1": 261.42559814453125,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523977741599083,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 256.9483642578125,
      "loss_g0": 257.74102783203125,
      "loss_g1": 209.76644897460938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 173.6517791748047,
      "loss_g0": 173.2083282470703,
      "loss_g1": 200.04248046875,
      "q_g0": 0.983474850654602,
      "q_g1": 0.01652517169713974,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 173.1404266357422,
      "loss_g0": 172.5603790283203,
      "loss_g1": 207.65438842773438,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.01652875542640686,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 191.6190948486328,
      "loss_g0": 192.4234161376953,
      "loss_g1": 143.74755859375,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523784026503563,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 180.8237762451172,
      "loss_g0": 180.8981475830078,
      "loss_g1": 176.39669799804688,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523325815796852,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 118.3781967163086,
      "loss_g0": 117.89384460449219,
      "loss_g1": 147.20172119140625,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526324674487114,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 106.28251647949219,
      "loss_g0": 106.24246215820312,
      "loss_g1": 108.66619110107422,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526563093066216,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 117.14057922363281,
      "loss_g0": 117.14057922363281,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 118.04763793945312,
      "loss_g0": 0.0,
      "loss_g1": 118.04763793945312,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 74.67428588867188,
      "loss_g0": 74.67428588867188,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 84.68389129638672,
      "loss_g0": 84.65359497070312,
      "loss_g1": 86.48741149902344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "eval_cer": 0.9626467033181269,
      "eval_loss": 188.8120880126953,
      "eval_runtime": 79.2657,
      "eval_samples_per_second": 21.144,
      "eval_steps_per_second": 5.286,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 335.9715881347656,
      "loss_g0": 336.5156555175781,
      "loss_g1": 303.5860595703125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 300
    },
    {
      "epoch": 1.5228426395939088,
      "gdro_loss": 351.8994445800781,
      "loss_g0": 351.4517822265625,
      "loss_g1": 378.541748046875,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.01652519218623638,
      "step": 300
    },
    {
      "epoch": 2.030456852791878,
      "grad_norm": 99.83296966552734,
      "learning_rate": 0.00038915508021390374,
      "loss": 54.7296,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.139883041381836,
      "loss_g0": 15.139883041381836,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 23.831594467163086,
      "loss_g0": 23.831594467163086,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 35.63885498046875,
      "loss_g0": 0.0,
      "loss_g1": 35.63885498046875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 26.842147827148438,
      "loss_g0": 26.93232536315918,
      "loss_g1": 26.840633392333984,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.894240379333496,
      "loss_g0": 15.894240379333496,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 40.12489700317383,
      "loss_g0": 40.622615814208984,
      "loss_g1": 10.498785972595215,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.580260276794434,
      "loss_g0": 9.621950149536133,
      "loss_g1": 7.098707675933838,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.839277267456055,
      "loss_g0": 11.453337669372559,
      "loss_g1": 34.80854415893555,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652480475604534,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.295807361602783,
      "loss_g0": 6.998719215393066,
      "loss_g1": 24.975051879882812,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526635736227036,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.900171279907227,
      "loss_g0": 18.824932098388672,
      "loss_g1": 23.377378463745117,
      "q_g0": 0.9834729433059692,
      "q_g1": 0.016527099534869194,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.32154369354248,
      "loss_g0": 15.313014030456543,
      "loss_g1": 15.8291654586792,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.01652715355157852,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.401540756225586,
      "loss_g0": 0.0,
      "loss_g1": 14.401540756225586,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.284820556640625,
      "loss_g0": 15.837091445922852,
      "loss_g1": 19.342741012573242,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.949347972869873,
      "loss_g0": 13.203718185424805,
      "loss_g1": 3.793865203857422,
      "q_g0": 0.016523387283086777,
      "q_g1": 0.9834766387939453,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.03752613067627,
      "loss_g0": 12.076109886169434,
      "loss_g1": 12.036877632141113,
      "q_g0": 0.016523387283086777,
      "q_g1": 0.9834766387939453,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.837363243103027,
      "loss_g0": 10.517802238464355,
      "loss_g1": 11.859533309936523,
      "q_g0": 0.016523247584700584,
      "q_g1": 0.9834766983985901,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.363722085952759,
      "loss_g0": 43.8115348815918,
      "loss_g1": 1.667178750038147,
      "q_g0": 0.016527557745575905,
      "q_g1": 0.983472466468811,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 23.10894012451172,
      "loss_g0": 23.10894012451172,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 27.182405471801758,
      "loss_g0": 27.448036193847656,
      "loss_g1": 11.371097564697266,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.883161544799805,
      "loss_g0": 8.883161544799805,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.57050132751465,
      "loss_g0": 21.583330154418945,
      "loss_g1": 20.806936264038086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.781822204589844,
      "loss_g0": 7.7608184814453125,
      "loss_g1": 9.032008171081543,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.32973051071167,
      "loss_g0": 6.314701557159424,
      "loss_g1": 7.224299430847168,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 27.9144287109375,
      "loss_g0": 28.186904907226562,
      "loss_g1": 11.695587158203125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.8125975131988525,
      "loss_g0": 3.7207839488983154,
      "loss_g1": 9.277495384216309,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522984951734543,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.89598274230957,
      "loss_g0": 6.736473083496094,
      "loss_g1": 16.389705657958984,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523968428373337,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.259051322937012,
      "loss_g0": 9.339554786682129,
      "loss_g1": 4.4674577713012695,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652347296476364,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.907931327819824,
      "loss_g0": 14.120269775390625,
      "loss_g1": 1.2687240839004517,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.311134338378906,
      "loss_g0": 15.426769256591797,
      "loss_g1": 8.428117752075195,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.499556064605713,
      "loss_g0": 3.306753158569336,
      "loss_g1": 14.975083351135254,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652361825108528,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.115954399108887,
      "loss_g0": 7.140207290649414,
      "loss_g1": 5.672420024871826,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652347296476364,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.39639949798584,
      "loss_g0": 13.294573783874512,
      "loss_g1": 19.456819534301758,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524100676178932,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.64156723022461,
      "loss_g0": 13.771782875061035,
      "loss_g1": 5.8910675048828125,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523294150829315,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.08233642578125,
      "loss_g0": 10.882410049438477,
      "loss_g1": 22.981182098388672,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524534672498703,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.013749122619629,
      "loss_g0": 9.074007987976074,
      "loss_g1": 5.427279472351074,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524160280823708,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.5791492462158203,
      "loss_g0": 3.485976457595825,
      "loss_g1": 9.124361038208008,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652473583817482,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.55489158630371,
      "loss_g0": 18.785337448120117,
      "loss_g1": 4.838580131530762,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523318365216255,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.623064041137695,
      "loss_g0": 26.008394241333008,
      "loss_g1": 2.6867830753326416,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.847182273864746,
      "loss_g0": 10.969100952148438,
      "loss_g1": 3.5901455879211426,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.933902740478516,
      "loss_g0": 16.933902740478516,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 26.90757942199707,
      "loss_g0": 27.348289489746094,
      "loss_g1": 0.6748387217521667,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.76309871673584,
      "loss_g0": 10.76309871673584,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.501943588256836,
      "loss_g0": 16.59315299987793,
      "loss_g1": 11.072769165039062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.2290167808532715,
      "loss_g0": 7.286137104034424,
      "loss_g1": 3.828993320465088,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.150821208953857,
      "loss_g0": 7.19979190826416,
      "loss_g1": 4.235884666442871,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.938864707946777,
      "loss_g0": 10.045154571533203,
      "loss_g1": 3.6120481491088867,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 27.434101104736328,
      "loss_g0": 27.861751556396484,
      "loss_g1": 1.9786957502365112,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.314169883728027,
      "loss_g0": 10.419010162353516,
      "loss_g1": 4.073653221130371,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.8454723358154297,
      "loss_g0": 3.758687973022461,
      "loss_g1": 9.011038780212402,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522962599992752,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.8753204345703125,
      "loss_g0": 5.8753204345703125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.286882400512695,
      "loss_g0": 22.409866333007812,
      "loss_g1": 14.966278076171875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.761228561401367,
      "loss_g0": 0.0,
      "loss_g1": 28.761228561401367,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 80.40673828125,
      "loss_g0": 16.514429092407227,
      "loss_g1": 81.4801254272461,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 33.248016357421875,
      "loss_g0": 33.248016357421875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.643815994262695,
      "loss_g0": 28.903764724731445,
      "loss_g1": 13.170608520507812,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 26.571626663208008,
      "loss_g0": 26.599109649658203,
      "loss_g1": 24.935747146606445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 30.710037231445312,
      "loss_g0": 30.529693603515625,
      "loss_g1": 41.4442024230957,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523543745279312,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 53.20542526245117,
      "loss_g0": 53.655433654785156,
      "loss_g1": 26.419099807739258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.5502290725708,
      "loss_g0": 12.383500099182129,
      "loss_g1": 22.473888397216797,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.0165234487503767,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.112455368041992,
      "loss_g0": 16.115299224853516,
      "loss_g1": 15.94329833984375,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652342639863491,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.455497741699219,
      "loss_g0": 9.352766036987305,
      "loss_g1": 15.56986141204834,
      "q_g0": 0.983475923538208,
      "q_g1": 0.0165240541100502,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.589302062988281,
      "loss_g0": 6.464169502258301,
      "loss_g1": 14.036555290222168,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652483083307743,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.579050540924072,
      "loss_g0": 6.37388801574707,
      "loss_g1": 18.788314819335938,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.016526101157069206,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.410099029541016,
      "loss_g0": 19.410099029541016,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 56.234012603759766,
      "loss_g0": 56.995296478271484,
      "loss_g1": 10.919431686401367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.885579109191895,
      "loss_g0": 0.0,
      "loss_g1": 9.885579109191895,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.024415016174316,
      "loss_g0": 9.164260864257812,
      "loss_g1": 3.938065528869629,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 23.00576400756836,
      "loss_g0": 12.863356590270996,
      "loss_g1": 23.17615509033203,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.126495361328125,
      "loss_g0": 0.0,
      "loss_g1": 17.126495361328125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.9950647354125977,
      "loss_g0": 11.059316635131836,
      "loss_g1": 3.87638521194458,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.769179821014404,
      "loss_g0": 8.071125984191895,
      "loss_g1": 7.7641072273254395,
      "q_g0": 0.016522448509931564,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.070821762084961,
      "loss_g0": 20.122880935668945,
      "loss_g1": 12.95234203338623,
      "q_g0": 0.016523178666830063,
      "q_g1": 0.9834768176078796,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.543669700622559,
      "loss_g0": 17.723615646362305,
      "loss_g1": 11.439837455749512,
      "q_g0": 0.0165238194167614,
      "q_g1": 0.9834761619567871,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.7024574279785156,
      "loss_g0": 14.8806791305542,
      "loss_g1": 3.514633893966675,
      "q_g0": 0.01652497425675392,
      "q_g1": 0.9834749698638916,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.966472625732422,
      "loss_g0": 2.9947423934936523,
      "loss_g1": 10.083610534667969,
      "q_g0": 0.01652425527572632,
      "q_g1": 0.9834756851196289,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.266027450561523,
      "loss_g0": 12.266027450561523,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.438484191894531,
      "loss_g0": 10.438484191894531,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.917724609375,
      "loss_g0": 26.14082145690918,
      "loss_g1": 12.638166427612305,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.804024696350098,
      "loss_g0": 0.0,
      "loss_g1": 13.804024696350098,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.295710563659668,
      "loss_g0": 4.295710563659668,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 27.76673698425293,
      "loss_g0": 27.995561599731445,
      "loss_g1": 14.146209716796875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.867273330688477,
      "loss_g0": 11.920433044433594,
      "loss_g1": 8.703027725219727,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 23.079875946044922,
      "loss_g0": 23.301834106445312,
      "loss_g1": 9.868030548095703,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.486144065856934,
      "loss_g0": 14.637826919555664,
      "loss_g1": 5.457394123077393,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.896775245666504,
      "loss_g0": 3.840216875076294,
      "loss_g1": 7.263269424438477,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652277447283268,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.40873908996582,
      "loss_g0": 9.963497161865234,
      "loss_g1": 36.9061164855957,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.01652553305029869,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.09469223022461,
      "loss_g0": 16.242185592651367,
      "loss_g1": 7.31647253036499,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524627804756165,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.8918657302856445,
      "loss_g0": 7.8918657302856445,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.038628578186035,
      "loss_g0": 7.971011161804199,
      "loss_g1": 12.06346321105957,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.810329437255859,
      "loss_g0": 5.895101547241211,
      "loss_g1": 0.7643535733222961,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.606082916259766,
      "loss_g0": 11.748785972595215,
      "loss_g1": 3.1118521690368652,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.653413772583008,
      "loss_g0": 4.564973831176758,
      "loss_g1": 9.91750717163086,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522971913218498,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.148305892944336,
      "loss_g0": 9.200819969177246,
      "loss_g1": 6.022555828094482,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.322683811187744,
      "loss_g0": 6.403464317321777,
      "loss_g1": 1.514327049255371,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.8964204788208,
      "loss_g0": 11.8964204788208,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.971780776977539,
      "loss_g0": 2.9862546920776367,
      "loss_g1": 2.1102330684661865,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.551309585571289,
      "loss_g0": 9.56971549987793,
      "loss_g1": 8.455718994140625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.482862949371338,
      "loss_g0": 6.503198623657227,
      "loss_g1": 5.272408485412598,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.554393291473389,
      "loss_g0": 4.578144073486328,
      "loss_g1": 3.1406562328338623,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.9615134000778198,
      "loss_g0": 1.8553210496902466,
      "loss_g1": 8.28222942352295,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523078083992004,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 42.92364501953125,
      "loss_g0": 42.95330810546875,
      "loss_g1": 41.15812301635742,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522889956831932,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.349576950073242,
      "loss_g0": 28.37603187561035,
      "loss_g1": 26.774940490722656,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.0165227223187685,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.993729591369629,
      "loss_g0": 15.48499870300293,
      "loss_g1": 46.26897048950195,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.016525868326425552,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 57.7304573059082,
      "loss_g0": 58.547996520996094,
      "loss_g1": 9.06729793548584,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 38.50077438354492,
      "loss_g0": 38.60204315185547,
      "loss_g1": 32.472923278808594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.64128875732422,
      "loss_g0": 25.611572265625,
      "loss_g1": 27.41006088256836,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522612422704697,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.391483306884766,
      "loss_g0": 19.012062072753906,
      "loss_g1": 41.972625732421875,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016524961218237877,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 36.04169464111328,
      "loss_g0": 36.20818328857422,
      "loss_g1": 26.132537841796875,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523927450180054,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.49009132385254,
      "loss_g0": 17.521921157836914,
      "loss_g1": 15.59559440612793,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523735597729683,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 57.189762115478516,
      "loss_g0": 57.929664611816406,
      "loss_g1": 13.147899627685547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.843660354614258,
      "loss_g0": 14.926168441772461,
      "loss_g1": 9.932472229003906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.644161224365234,
      "loss_g0": 18.636829376220703,
      "loss_g1": 19.08055877685547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522472724318504,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.033004760742188,
      "loss_g0": 28.843257904052734,
      "loss_g1": 40.326622009277344,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652364432811737,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.750515937805176,
      "loss_g0": 11.803175926208496,
      "loss_g1": 8.616121292114258,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523318365216255,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.974224090576172,
      "loss_g0": 26.157812118530273,
      "loss_g1": 15.046370506286621,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.72974967956543,
      "loss_g0": 25.685409545898438,
      "loss_g1": 28.369047164916992,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652269810438156,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.135009765625,
      "loss_g0": 18.161603927612305,
      "loss_g1": 16.55200958251953,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.789600372314453,
      "loss_g0": 10.668935775756836,
      "loss_g1": 17.971649169921875,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652327924966812,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.433305740356445,
      "loss_g0": 17.573360443115234,
      "loss_g1": 9.096659660339355,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.230358123779297,
      "loss_g0": 25.521461486816406,
      "loss_g1": 7.9027910232543945,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.05062198638916,
      "loss_g0": 9.953085899353027,
      "loss_g1": 15.85609245300293,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523022204637527,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.230905532836914,
      "loss_g0": 16.175107955932617,
      "loss_g1": 19.551979064941406,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523364931344986,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.948169708251953,
      "loss_g0": 21.948169708251953,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.876748085021973,
      "loss_g0": 15.876748085021973,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.411109924316406,
      "loss_g0": 28.77349281311035,
      "loss_g1": 6.8406805992126465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.713479995727539,
      "loss_g0": 0.0,
      "loss_g1": 8.713479995727539,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.785088539123535,
      "loss_g0": 8.069005966186523,
      "loss_g1": 10.830718040466309,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.646927833557129,
      "loss_g0": 8.489201545715332,
      "loss_g1": 12.716776847839355,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.71784210205078,
      "loss_g0": 28.71784210205078,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.270926475524902,
      "loss_g0": 10.326674461364746,
      "loss_g1": 6.95261812210083,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.346907138824463,
      "loss_g0": 6.252865791320801,
      "loss_g1": 11.944411277770996,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523005440831184,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.0040974617004395,
      "loss_g0": 3.9677536487579346,
      "loss_g1": 6.16730260848999,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652323268353939,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.837573051452637,
      "loss_g0": 13.874513626098633,
      "loss_g1": 11.638799667358398,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523005440831184,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.333492279052734,
      "loss_g0": 11.404827117919922,
      "loss_g1": 7.087426662445068,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652255840599537,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.5318074226379395,
      "loss_g0": 4.440868854522705,
      "loss_g1": 9.94459342956543,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652311533689499,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.326648712158203,
      "loss_g0": 9.24643611907959,
      "loss_g1": 14.10086441040039,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523612663149834,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 20.188966751098633,
      "loss_g0": 20.48524284362793,
      "loss_g1": 2.5534110069274902,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.604458332061768,
      "loss_g0": 5.532559394836426,
      "loss_g1": 9.884023666381836,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652287133038044,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.31812286376953,
      "loss_g0": 19.433921813964844,
      "loss_g1": 12.425272941589355,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.608429431915283,
      "loss_g0": 7.677001953125,
      "loss_g1": 3.5267131328582764,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.60487937927246,
      "loss_g0": 21.60487937927246,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.102839469909668,
      "loss_g0": 9.099592208862305,
      "loss_g1": 9.29611587524414,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.0543341636657715,
      "loss_g0": 3.9897539615631104,
      "loss_g1": 7.8983001708984375,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.035578727722168,
      "loss_g0": 9.171630859375,
      "loss_g1": 0.9372288584709167,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.536060333251953,
      "loss_g0": 9.626410484313965,
      "loss_g1": 4.158115386962891,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.389093279838562,
      "loss_g0": 1.2401282787322998,
      "loss_g1": 10.255562782287598,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523340716958046,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.906649112701416,
      "loss_g0": 6.980447292327881,
      "loss_g1": 2.514019250869751,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.44819450378418,
      "loss_g0": 8.473200798034668,
      "loss_g1": 6.959743976593018,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522729769349098,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.8354921340942383,
      "loss_g0": 3.8048038482666016,
      "loss_g1": 5.662112236022949,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652292162179947,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.103869676589966,
      "loss_g0": 3.1477036476135254,
      "loss_g1": 0.49473413825035095,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 23.41042137145996,
      "loss_g0": 0.0,
      "loss_g1": 23.41042137145996,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 33.72087097167969,
      "loss_g0": 36.584022521972656,
      "loss_g1": 33.672767639160156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.248809814453125,
      "loss_g0": 0.0,
      "loss_g1": 21.248809814453125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.516290664672852,
      "loss_g0": 19.858802795410156,
      "loss_g1": 7.308936595916748,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.994598388671875,
      "loss_g0": 15.786903381347656,
      "loss_g1": 26.166088104248047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 34.72135543823242,
      "loss_g0": 27.271682739257812,
      "loss_g1": 34.84650802612305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 26.04594612121582,
      "loss_g0": 22.491973876953125,
      "loss_g1": 26.10565185546875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.023021697998047,
      "loss_g0": 10.023021697998047,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.401978492736816,
      "loss_g0": 15.444847106933594,
      "loss_g1": 12.850275039672852,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.295753479003906,
      "loss_g0": 0.0,
      "loss_g1": 15.295753479003906,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.39471435546875,
      "loss_g0": 21.479734420776367,
      "loss_g1": 29.527687072753906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 24.052261352539062,
      "loss_g0": 24.052261352539062,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.751788139343262,
      "loss_g0": 0.0,
      "loss_g1": 12.751788139343262,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.444183349609375,
      "loss_g0": 10.657346725463867,
      "loss_g1": 16.54140281677246,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.264066696166992,
      "loss_g0": 33.41172409057617,
      "loss_g1": 19.026365280151367,
      "q_g0": 0.016523899510502815,
      "q_g1": 0.9834761619567871,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.047143936157227,
      "loss_g0": 28.624053955078125,
      "loss_g1": 21.93663787841797,
      "q_g0": 0.016524581238627434,
      "q_g1": 0.9834754467010498,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.83804702758789,
      "loss_g0": 20.210359573364258,
      "loss_g1": 16.781383514404297,
      "q_g0": 0.016524938866496086,
      "q_g1": 0.9834750890731812,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.826316833496094,
      "loss_g0": 26.434268951416016,
      "loss_g1": 25.81610107421875,
      "q_g0": 0.01652500033378601,
      "q_g1": 0.9834749698638916,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.073118686676025,
      "loss_g0": 17.941987991333008,
      "loss_g1": 5.8736748695373535,
      "q_g0": 0.01652623899281025,
      "q_g1": 0.9834737181663513,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 32.31940460205078,
      "loss_g0": 32.31940460205078,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 30.148035049438477,
      "loss_g0": 30.49471092224121,
      "loss_g1": 9.512453079223633,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.521129846572876,
      "loss_g0": 3.4363198280334473,
      "loss_g1": 8.56918716430664,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652294583618641,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.136573791503906,
      "loss_g0": 12.182815551757812,
      "loss_g1": 9.384079933166504,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522657126188278,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 41.49416732788086,
      "loss_g0": 41.73043441772461,
      "loss_g1": 27.430557250976562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.031024932861328,
      "loss_g0": 5.935247421264648,
      "loss_g1": 11.73187255859375,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652301475405693,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.474391460418701,
      "loss_g0": 5.421457767486572,
      "loss_g1": 8.625032424926758,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523340716958046,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.315572738647461,
      "loss_g0": 12.409347534179688,
      "loss_g1": 6.733860969543457,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522767022252083,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.135567665100098,
      "loss_g0": 11.173940658569336,
      "loss_g1": 8.851503372192383,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.40919303894043,
      "loss_g0": 13.564472198486328,
      "loss_g1": 4.166338920593262,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.71841812133789,
      "loss_g0": 0.0,
      "loss_g1": 13.71841812133789,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.232248783111572,
      "loss_g0": 12.315784454345703,
      "loss_g1": 6.130045413970947,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.22909164428711,
      "loss_g0": 3.2586252689361572,
      "loss_g1": 11.362995147705078,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.7134370803833,
      "loss_g0": 8.983489990234375,
      "loss_g1": 8.708900451660156,
      "q_g0": 0.01652245782315731,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.97350025177002,
      "loss_g0": 9.97350025177002,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.0500309467315674,
      "loss_g0": 1.9650603532791138,
      "loss_g1": 7.107793807983398,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 32.50261688232422,
      "loss_g0": 32.99252700805664,
      "loss_g1": 3.341282606124878,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.770368576049805,
      "loss_g0": 12.770368576049805,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.170393466949463,
      "loss_g0": 6.071371078491211,
      "loss_g1": 12.064567565917969,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 31.107513427734375,
      "loss_g0": 31.5815372467041,
      "loss_g1": 2.891727924346924,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.690217018127441,
      "loss_g0": 8.781240463256836,
      "loss_g1": 3.2721498012542725,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.218819618225098,
      "loss_g0": 12.355063438415527,
      "loss_g1": 4.109015464782715,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.3383402824401855,
      "loss_g0": 2.2289206981658936,
      "loss_g1": 8.851129531860352,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523102298378944,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.342525482177734,
      "loss_g0": 8.317758560180664,
      "loss_g1": 9.816651344299316,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523247584700584,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.792701721191406,
      "loss_g0": 15.906585693359375,
      "loss_g1": 9.013917922973633,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.785326957702637,
      "loss_g0": 9.856019020080566,
      "loss_g1": 5.577489852905273,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.527639389038086,
      "loss_g0": 8.663477897644043,
      "loss_g1": 0.44203564524650574,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.79940128326416,
      "loss_g0": 9.911459922790527,
      "loss_g1": 3.129227638244629,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 57.4539680480957,
      "loss_g0": 58.33998489379883,
      "loss_g1": 4.71478796005249,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.629033088684082,
      "loss_g0": 4.629033088684082,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 0.9505838751792908,
      "loss_g0": 0.9305859208106995,
      "loss_g1": 2.1409366130828857,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 57.77680587768555,
      "loss_g0": 58.332218170166016,
      "loss_g1": 24.71640396118164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 56.79113006591797,
      "loss_g0": 57.38925552368164,
      "loss_g1": 21.188392639160156,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.052631378173828,
      "loss_g0": 29.02872085571289,
      "loss_g1": 30.475936889648438,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.850048065185547,
      "loss_g0": 28.850048065185547,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 20.17229652404785,
      "loss_g0": 20.131511688232422,
      "loss_g1": 22.599895477294922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 31.597370147705078,
      "loss_g0": 31.796058654785156,
      "loss_g1": 19.77065658569336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.594873428344727,
      "loss_g0": 14.393470764160156,
      "loss_g1": 26.582286834716797,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523675993084908,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 24.64349937438965,
      "loss_g0": 24.552227020263672,
      "loss_g1": 30.075698852539062,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524240374565125,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.60982322692871,
      "loss_g0": 28.85720443725586,
      "loss_g1": 13.88502025604248,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522713005542755,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.323298454284668,
      "loss_g0": 13.308283805847168,
      "loss_g1": 14.217041015625,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522813588380814,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.495342254638672,
      "loss_g0": 19.419647216796875,
      "loss_g1": 24.000835418701172,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523286700248718,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.693363189697266,
      "loss_g0": 16.56645965576172,
      "loss_g1": 24.246356964111328,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524076461791992,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.1471428871154785,
      "loss_g0": 5.059807777404785,
      "loss_g1": 10.344982147216797,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.01652461104094982,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.04639434814453,
      "loss_g0": 16.914901733398438,
      "loss_g1": 24.871936798095703,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525425016880035,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.1013822555542,
      "loss_g0": 11.1013822555542,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.194951057434082,
      "loss_g0": 12.130912780761719,
      "loss_g1": 16.006710052490234,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.980335235595703,
      "loss_g0": 29.825275421142578,
      "loss_g1": 39.209434509277344,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652337983250618,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.493249893188477,
      "loss_g0": 9.324563980102539,
      "loss_g1": 19.5328426361084,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652441918849945,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.206592559814453,
      "loss_g0": 17.206592559814453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.447504997253418,
      "loss_g0": 13.408567428588867,
      "loss_g1": 15.765202522277832,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.71744155883789,
      "loss_g0": 11.631040573120117,
      "loss_g1": 16.860139846801758,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522955149412155,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.726608276367188,
      "loss_g0": 18.935272216796875,
      "loss_g1": 6.3061323165893555,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.353387832641602,
      "loss_g0": 9.353387832641602,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.051769256591797,
      "loss_g0": 13.060251235961914,
      "loss_g1": 12.546862602233887,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.053794860839844,
      "loss_g0": 6.04947566986084,
      "loss_g1": 6.310903072357178,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522448509931564,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.382833480834961,
      "loss_g0": 0.0,
      "loss_g1": 8.382833480834961,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.593814373016357,
      "loss_g0": 13.717090606689453,
      "loss_g1": 5.457343101501465,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 35.255062103271484,
      "loss_g0": 35.255062103271484,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 31.469451904296875,
      "loss_g0": 31.926122665405273,
      "loss_g1": 4.286627292633057,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.159585952758789,
      "loss_g0": 9.905515670776367,
      "loss_g1": 25.281335830688477,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523992642760277,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.630056381225586,
      "loss_g0": 21.81085968017578,
      "loss_g1": 10.86813735961914,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652287133038044,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.737799644470215,
      "loss_g0": 6.6854658126831055,
      "loss_g1": 9.852743148803711,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523193567991257,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.175848007202148,
      "loss_g0": 9.123151779174805,
      "loss_g1": 12.312317848205566,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523517668247223,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.5177083015441895,
      "loss_g0": 7.436509132385254,
      "loss_g1": 12.350546836853027,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524024307727814,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.2118706703186035,
      "loss_g0": 7.121332168579102,
      "loss_g1": 12.600326538085938,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524581238627434,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.839885711669922,
      "loss_g0": 10.900883674621582,
      "loss_g1": 7.209429740905762,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652420125901699,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.057659149169922,
      "loss_g0": 15.186274528503418,
      "loss_g1": 7.402378082275391,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523411497473717,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.472597122192383,
      "loss_g0": 4.398221492767334,
      "loss_g1": 8.899328231811523,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652386784553528,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.785526752471924,
      "loss_g0": 5.829931259155273,
      "loss_g1": 3.142601728439331,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523588448762894,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 42.29154968261719,
      "loss_g0": 42.88550567626953,
      "loss_g1": 6.936960220336914,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.905115127563477,
      "loss_g0": 0.0,
      "loss_g1": 8.905115127563477,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.128846168518066,
      "loss_g0": 10.682555198669434,
      "loss_g1": 5.035543918609619,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.862027645111084,
      "loss_g0": 18.538585662841797,
      "loss_g1": 2.598635196685791,
      "q_g0": 0.0165240541100502,
      "q_g1": 0.983475923538208,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.621813774108887,
      "loss_g0": 8.034614562988281,
      "loss_g1": 6.598075866699219,
      "q_g0": 0.01652420125901699,
      "q_g1": 0.9834758043289185,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.322340250015259,
      "loss_g0": 4.585196018218994,
      "loss_g1": 3.301121711730957,
      "q_g0": 0.016524333506822586,
      "q_g1": 0.9834756851196289,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.1499154567718506,
      "loss_g0": 17.384368896484375,
      "loss_g1": 0.877116322517395,
      "q_g0": 0.01652602106332779,
      "q_g1": 0.9834739565849304,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.8927323818206787,
      "loss_g0": 4.817140579223633,
      "loss_g1": 1.843590497970581,
      "q_g0": 0.01652633212506771,
      "q_g1": 0.9834735989570618,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.8142757415771484,
      "loss_g0": 1.6845167875289917,
      "loss_g1": 3.8500638008117676,
      "q_g0": 0.016526106745004654,
      "q_g1": 0.9834738373756409,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.8167619705200195,
      "loss_g0": 12.035162925720215,
      "loss_g1": 4.695459842681885,
      "q_g0": 0.01652686670422554,
      "q_g1": 0.9834731221199036,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 0.5229355692863464,
      "loss_g0": 2.3204658031463623,
      "loss_g1": 0.49272847175598145,
      "q_g0": 0.016527052968740463,
      "q_g1": 0.9834729433059692,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 49.4326057434082,
      "loss_g0": 0.0,
      "loss_g1": 49.4326057434082,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 27.616710662841797,
      "loss_g0": 22.210174560546875,
      "loss_g1": 27.70754051208496,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.529777526855469,
      "loss_g0": 24.3970947265625,
      "loss_g1": 14.363995552062988,
      "q_g0": 0.0165234487503767,
      "q_g1": 0.9834765195846558,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.716266632080078,
      "loss_g0": 34.74380874633789,
      "loss_g1": 25.564586639404297,
      "q_g0": 0.016524387523531914,
      "q_g1": 0.9834755659103394,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 43.37718963623047,
      "loss_g0": 32.57173538208008,
      "loss_g1": 43.5587272644043,
      "q_g0": 0.016523264348506927,
      "q_g1": 0.9834766983985901,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 31.38835334777832,
      "loss_g0": 26.920440673828125,
      "loss_g1": 31.463417053222656,
      "q_g0": 0.01652279868721962,
      "q_g1": 0.9834771752357483,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 32.529052734375,
      "loss_g0": 35.64892578125,
      "loss_g1": 32.47663497924805,
      "q_g0": 0.016523122787475586,
      "q_g1": 0.9834768176078796,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.798470497131348,
      "loss_g0": 27.04987144470215,
      "loss_g1": 11.542210578918457,
      "q_g0": 0.01652471348643303,
      "q_g1": 0.9834753274917603,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.677406311035156,
      "loss_g0": 10.205556869506836,
      "loss_g1": 15.769342422485352,
      "q_g0": 0.016524145379662514,
      "q_g1": 0.9834758043289185,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.119354248046875,
      "loss_g0": 0.0,
      "loss_g1": 17.119354248046875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.91276741027832,
      "loss_g0": 23.788740158081055,
      "loss_g1": 18.83085060119629,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.08562660217285,
      "loss_g0": 11.633476257324219,
      "loss_g1": 18.19402313232422,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.864476203918457,
      "loss_g0": 13.99374008178711,
      "loss_g1": 15.895904541015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 55.884559631347656,
      "loss_g0": 28.103416442871094,
      "loss_g1": 56.351280212402344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.13332748413086,
      "loss_g0": 0.0,
      "loss_g1": 14.13332748413086,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.767596244812012,
      "loss_g0": 14.28762435913086,
      "loss_g1": 9.691659927368164,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.26556396484375,
      "loss_g0": 25.000638961791992,
      "loss_g1": 15.102005004882812,
      "q_g0": 0.016523433849215508,
      "q_g1": 0.9834766387939453,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.957195281982422,
      "loss_g0": 27.524450302124023,
      "loss_g1": 9.66201400756836,
      "q_g0": 0.016525255516171455,
      "q_g1": 0.9834747910499573,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.402785301208496,
      "loss_g0": 21.82930564880371,
      "loss_g1": 12.244382858276367,
      "q_g0": 0.01652623899281025,
      "q_g1": 0.9834737181663513,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 37.506370544433594,
      "loss_g0": 11.484611511230469,
      "loss_g1": 37.943565368652344,
      "q_g0": 0.01652352698147297,
      "q_g1": 0.9834765195846558,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.83980941772461,
      "loss_g0": 42.38990783691406,
      "loss_g1": 21.49449920654297,
      "q_g0": 0.01652565784752369,
      "q_g1": 0.9834743142127991,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.529449462890625,
      "loss_g0": 9.2824125289917,
      "loss_g1": 25.802425384521484,
      "q_g0": 0.016523968428373337,
      "q_g1": 0.9834760427474976,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.207775115966797,
      "loss_g0": 12.543145179748535,
      "loss_g1": 10.168536186218262,
      "q_g0": 0.016524210572242737,
      "q_g1": 0.9834758043289185,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.288532733917236,
      "loss_g0": 11.396490097045898,
      "loss_g1": 6.202706813812256,
      "q_g0": 0.01652473583817482,
      "q_g1": 0.9834752082824707,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.605820655822754,
      "loss_g0": 13.178659439086914,
      "loss_g1": 11.579392433166504,
      "q_g0": 0.0165249053388834,
      "q_g1": 0.9834750890731812,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.679403305053711,
      "loss_g0": 7.7291951179504395,
      "loss_g1": 12.762577056884766,
      "q_g0": 0.016524387523531914,
      "q_g1": 0.9834755659103394,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.689004898071289,
      "loss_g0": 3.746669292449951,
      "loss_g1": 5.721639633178711,
      "q_g0": 0.01652417704463005,
      "q_g1": 0.9834758043289185,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.151088714599609,
      "loss_g0": 24.095001220703125,
      "loss_g1": 5.8495635986328125,
      "q_g0": 0.01652604527771473,
      "q_g1": 0.9834739565849304,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.055485725402832,
      "loss_g0": 15.70594596862793,
      "loss_g1": 11.994142532348633,
      "q_g0": 0.016526425257325172,
      "q_g1": 0.9834735989570618,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.479090690612793,
      "loss_g0": 8.316234588623047,
      "loss_g1": 11.532238006591797,
      "q_g0": 0.016526101157069206,
      "q_g1": 0.9834739565849304,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.66176176071167,
      "loss_g0": 5.369739055633545,
      "loss_g1": 4.649864673614502,
      "q_g0": 0.016526177525520325,
      "q_g1": 0.9834738373756409,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.903592109680176,
      "loss_g0": 27.760112762451172,
      "loss_g1": 9.603498458862305,
      "q_g0": 0.016528038308024406,
      "q_g1": 0.9834719896316528,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.815910816192627,
      "loss_g0": 21.303152084350586,
      "loss_g1": 4.53879976272583,
      "q_g0": 0.016529759392142296,
      "q_g1": 0.9834702610969543,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.495882511138916,
      "loss_g0": 34.139678955078125,
      "loss_g1": 3.9975476264953613,
      "q_g0": 0.016532836481928825,
      "q_g1": 0.983467161655426,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.778131484985352,
      "loss_g0": 13.217830657958984,
      "loss_g1": 8.703495025634766,
      "q_g0": 0.016533292829990387,
      "q_g1": 0.9834666848182678,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.6025390625,
      "loss_g0": 11.944443702697754,
      "loss_g1": 18.71446418762207,
      "q_g0": 0.01653260365128517,
      "q_g1": 0.9834674000740051,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.843619346618652,
      "loss_g0": 15.144840240478516,
      "loss_g1": 9.754499435424805,
      "q_g0": 0.01653316058218479,
      "q_g1": 0.9834668040275574,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.533656597137451,
      "loss_g0": 4.533656597137451,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.682226181030273,
      "loss_g0": 26.011234283447266,
      "loss_g1": 6.098315715789795,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.8030362129211426,
      "loss_g0": 3.822490692138672,
      "loss_g1": 2.6450369358062744,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.759667158126831,
      "loss_g0": 2.691652774810791,
      "loss_g1": 6.808035850524902,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652284525334835,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.7549397945404053,
      "loss_g0": 3.7915821075439453,
      "loss_g1": 1.573861002922058,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522621735930443,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.984364986419678,
      "loss_g0": 4.984364986419678,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.200045108795166,
      "loss_g0": 6.281064033508301,
      "loss_g1": 1.377500057220459,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.749774694442749,
      "loss_g0": 1.6024620532989502,
      "loss_g1": 10.517894744873047,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652333326637745,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.655792713165283,
      "loss_g0": 5.681695461273193,
      "loss_g1": 4.114043712615967,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523178666830063,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.243685722351074,
      "loss_g0": 7.323744773864746,
      "loss_g1": 2.478330373764038,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.7333946228027344,
      "loss_g0": 1.5476173162460327,
      "loss_g1": 12.790603637695312,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652383618056774,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.17299747467041,
      "loss_g0": 0.0,
      "loss_g1": 4.17299747467041,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.53327178955078,
      "loss_g0": 3.7623019218444824,
      "loss_g1": 16.74782371520996,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 24.630245208740234,
      "loss_g0": 58.98784637451172,
      "loss_g1": 24.05290985107422,
      "q_g0": 0.0165259912610054,
      "q_g1": 0.9834739565849304,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 33.817359924316406,
      "loss_g0": 29.412052154541016,
      "loss_g1": 33.89138412475586,
      "q_g0": 0.01652553305029869,
      "q_g1": 0.9834744334220886,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.457067489624023,
      "loss_g0": 30.902700424194336,
      "loss_g1": 17.231117248535156,
      "q_g0": 0.01652693562209606,
      "q_g1": 0.9834730625152588,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 23.320068359375,
      "loss_g0": 24.571044921875,
      "loss_g1": 23.29904556274414,
      "q_g0": 0.016527069732546806,
      "q_g1": 0.9834729433059692,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.754085540771484,
      "loss_g0": 0.0,
      "loss_g1": 18.754085540771484,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.546810150146484,
      "loss_g0": 16.43695831298828,
      "loss_g1": 29.76705551147461,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.396854400634766,
      "loss_g0": 13.899255752563477,
      "loss_g1": 22.539613723754883,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.938389778137207,
      "loss_g0": 23.67315673828125,
      "loss_g1": 15.808439254760742,
      "q_g0": 0.016523223370313644,
      "q_g1": 0.9834766983985901,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 38.121150970458984,
      "loss_g0": 38.247798919677734,
      "loss_g1": 38.119022369384766,
      "q_g0": 0.016523247584700584,
      "q_g1": 0.9834766983985901,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 34.575164794921875,
      "loss_g0": 34.575164794921875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.16626739501953,
      "loss_g0": 28.338735580444336,
      "loss_g1": 17.90020751953125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.637714385986328,
      "loss_g0": 0.0,
      "loss_g1": 13.637714385986328,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.249942779541016,
      "loss_g0": 20.901453018188477,
      "loss_g1": 13.121397018432617,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.514592170715332,
      "loss_g0": 16.786266326904297,
      "loss_g1": 13.459627151489258,
      "q_g0": 0.016522759571671486,
      "q_g1": 0.9834771752357483,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.715568542480469,
      "loss_g0": 15.534429550170898,
      "loss_g1": 4.533795356750488,
      "q_g0": 0.01652389205992222,
      "q_g1": 0.9834761619567871,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.14378547668457,
      "loss_g0": 40.669193267822266,
      "loss_g1": 16.748462677001953,
      "q_g0": 0.01652633212506771,
      "q_g1": 0.9834735989570618,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 84.5836181640625,
      "loss_g0": 25.938329696655273,
      "loss_g1": 85.56885528564453,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 13.856077194213867,
      "loss_g0": 12.611587524414062,
      "loss_g1": 13.876984596252441,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.853212356567383,
      "loss_g0": 11.356964111328125,
      "loss_g1": 14.911949157714844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.66951560974121,
      "loss_g0": 53.05126190185547,
      "loss_g1": 21.142196655273438,
      "q_g0": 0.016525689512491226,
      "q_g1": 0.9834743142127991,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 26.756746292114258,
      "loss_g0": 2.504223585128784,
      "loss_g1": 27.164207458496094,
      "q_g0": 0.016523171216249466,
      "q_g1": 0.9834768176078796,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.80948257446289,
      "loss_g0": 28.850557327270508,
      "loss_g1": 9.48953914642334,
      "q_g0": 0.016525154933333397,
      "q_g1": 0.983474850654602,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.42983341217041,
      "loss_g0": 3.439491033554077,
      "loss_g1": 12.580886840820312,
      "q_g0": 0.016524218022823334,
      "q_g1": 0.9834758043289185,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.659860610961914,
      "loss_g0": 0.0,
      "loss_g1": 12.659860610961914,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.837593078613281,
      "loss_g0": 19.622461318969727,
      "loss_g1": 11.706806182861328,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.918527603149414,
      "loss_g0": 22.59859848022461,
      "loss_g1": 6.655076503753662,
      "q_g0": 0.0165240541100502,
      "q_g1": 0.983475923538208,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.708919525146484,
      "loss_g0": 18.520540237426758,
      "loss_g1": 8.544057846069336,
      "q_g0": 0.016525069251656532,
      "q_g1": 0.983474850654602,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.498011350631714,
      "loss_g0": 4.252591133117676,
      "loss_g1": 3.485332489013672,
      "q_g0": 0.0165251474827528,
      "q_g1": 0.983474850654602,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.3089599609375,
      "loss_g0": 10.872072219848633,
      "loss_g1": 8.265891075134277,
      "q_g0": 0.016525408253073692,
      "q_g1": 0.9834745526313782,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.038902282714844,
      "loss_g0": 8.463062286376953,
      "loss_g1": 10.065380096435547,
      "q_g0": 0.01652524806559086,
      "q_g1": 0.9834747910499573,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 40.582454681396484,
      "loss_g0": 13.623409271240234,
      "loss_g1": 41.03536605834961,
      "q_g0": 0.016522448509931564,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.498651027679443,
      "loss_g0": 12.109529495239258,
      "loss_g1": 4.370781898498535,
      "q_g0": 0.016523240134119987,
      "q_g1": 0.9834766983985901,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.3138251304626465,
      "loss_g0": 16.133974075317383,
      "loss_g1": 7.165630340576172,
      "q_g0": 0.016524160280823708,
      "q_g1": 0.9834758043289185,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.390456199645996,
      "loss_g0": 3.7396328449249268,
      "loss_g1": 14.56939697265625,
      "q_g0": 0.016523055732250214,
      "q_g1": 0.9834769368171692,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.97076416015625,
      "loss_g0": 6.575095176696777,
      "loss_g1": 11.044611930847168,
      "q_g0": 0.016522597521543503,
      "q_g1": 0.9834774136543274,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 10.037599563598633,
      "loss_g0": 0.0,
      "loss_g1": 10.037599563598633,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.7084131240844727,
      "loss_g0": 15.452054977416992,
      "loss_g1": 3.511119842529297,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.573759078979492,
      "loss_g0": 15.547599792480469,
      "loss_g1": 5.406188011169434,
      "q_g0": 0.0165234562009573,
      "q_g1": 0.9834765195846558,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.762510299682617,
      "loss_g0": 8.113580703735352,
      "loss_g1": 3.689405918121338,
      "q_g0": 0.01652390882372856,
      "q_g1": 0.9834761619567871,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.695528030395508,
      "loss_g0": 4.354716777801514,
      "loss_g1": 6.734857082366943,
      "q_g0": 0.01652366667985916,
      "q_g1": 0.9834764003753662,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.0469870567321777,
      "loss_g0": 1.9550522565841675,
      "loss_g1": 2.0485315322875977,
      "q_g0": 0.016523659229278564,
      "q_g1": 0.9834764003753662,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.1265456676483154,
      "loss_g0": 22.540603637695312,
      "loss_g1": 1.7835187911987305,
      "q_g0": 0.01652577519416809,
      "q_g1": 0.9834741950035095,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.2815439701080322,
      "loss_g0": 9.662235260009766,
      "loss_g1": 2.1575169563293457,
      "q_g0": 0.016526535153388977,
      "q_g1": 0.9834734797477722,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.922494411468506,
      "loss_g0": 4.169669151306152,
      "loss_g1": 4.935145378112793,
      "q_g0": 0.01652645505964756,
      "q_g1": 0.9834734797477722,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.938620567321777,
      "loss_g0": 9.938620567321777,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.2937681674957275,
      "loss_g0": 3.3380181789398193,
      "loss_g1": 0.65984708070755,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.630946159362793,
      "loss_g0": 4.689520359039307,
      "loss_g1": 1.144376516342163,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.764358639717102,
      "loss_g0": 1.7652409076690674,
      "loss_g1": 1.7118347883224487,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.9806060791015625,
      "loss_g0": 8.100543022155762,
      "loss_g1": 0.8415014743804932,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.906479358673096,
      "loss_g0": 7.775554180145264,
      "loss_g1": 15.699272155761719,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652323268353939,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.496505737304688,
      "loss_g0": 16.496505737304688,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.96179962158203,
      "loss_g0": 28.784320831298828,
      "loss_g1": 39.52593994140625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.246536254882812,
      "loss_g0": 22.329570770263672,
      "loss_g1": 17.304004669189453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 24.737953186035156,
      "loss_g0": 24.751346588134766,
      "loss_g1": 23.940797805786133,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.398845672607422,
      "loss_g0": 28.20305633544922,
      "loss_g1": 100.54430389404297,
      "q_g0": 0.9834701418876648,
      "q_g1": 0.01652981899678707,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.187530517578125,
      "loss_g0": 29.303316116333008,
      "loss_g1": 22.298370361328125,
      "q_g0": 0.9834708571434021,
      "q_g1": 0.016529105603694916,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.370698928833008,
      "loss_g0": 7.245682716369629,
      "loss_g1": 14.808719635009766,
      "q_g0": 0.9834701418876648,
      "q_g1": 0.016529874876141548,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.15133285522461,
      "loss_g0": 19.15133285522461,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 29.61724090576172,
      "loss_g0": 29.61724090576172,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.58664321899414,
      "loss_g0": 17.77576446533203,
      "loss_g1": 6.329510688781738,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 35.014312744140625,
      "loss_g0": 35.42599868774414,
      "loss_g1": 10.509244918823242,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 33.341529846191406,
      "loss_g0": 33.505252838134766,
      "loss_g1": 23.596240997314453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 21.612319946289062,
      "loss_g0": 21.80086898803711,
      "loss_g1": 10.389169692993164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.794143676757812,
      "loss_g0": 0.0,
      "loss_g1": 22.794143676757812,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.591564178466797,
      "loss_g0": 17.880903244018555,
      "loss_g1": 22.670703887939453,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.715753555297852,
      "loss_g0": 27.929624557495117,
      "loss_g1": 8.39292049407959,
      "q_g0": 0.01652441918849945,
      "q_g1": 0.9834755659103394,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 34.683353424072266,
      "loss_g0": 20.334686279296875,
      "loss_g1": 34.92441940307617,
      "q_g0": 0.01652292162179947,
      "q_g1": 0.9834770560264587,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 18.157955169677734,
      "loss_g0": 19.082014083862305,
      "loss_g1": 18.14242935180664,
      "q_g0": 0.01652301475405693,
      "q_g1": 0.9834769368171692,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 20.521007537841797,
      "loss_g0": 0.0,
      "loss_g1": 20.521007537841797,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 16.1195011138916,
      "loss_g0": 7.081690311431885,
      "loss_g1": 16.27133560180664,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.732292175292969,
      "loss_g0": 48.710689544677734,
      "loss_g1": 11.110908508300781,
      "q_g0": 0.016526270657777786,
      "q_g1": 0.9834737181663513,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.843352794647217,
      "loss_g0": 48.163719177246094,
      "loss_g1": 6.14882230758667,
      "q_g0": 0.016530564054846764,
      "q_g1": 0.9834694862365723,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 19.82447052001953,
      "loss_g0": 29.10845947265625,
      "loss_g1": 19.668413162231445,
      "q_g0": 0.016531525179743767,
      "q_g1": 0.9834684133529663,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.675945281982422,
      "loss_g0": 0.0,
      "loss_g1": 12.675945281982422,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.561583042144775,
      "loss_g0": 14.461309432983398,
      "loss_g1": 4.395267486572266,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 14.723759651184082,
      "loss_g0": 11.3374662399292,
      "loss_g1": 14.780649185180664,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.55443572998047,
      "loss_g0": 13.20898723602295,
      "loss_g1": 17.627439498901367,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.334819793701172,
      "loss_g0": 7.2058634757995605,
      "loss_g1": 17.504985809326172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.31893539428711,
      "loss_g0": 16.53787612915039,
      "loss_g1": 17.33205795288086,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.669511795043945,
      "loss_g0": 21.1015682220459,
      "loss_g1": 17.611852645874023,
      "q_g0": 0.01652277447283268,
      "q_g1": 0.9834771752357483,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.563799858093262,
      "loss_g0": 10.82044506072998,
      "loss_g1": 6.492284774780273,
      "q_g0": 0.016523223370313644,
      "q_g1": 0.9834766983985901,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.107988357543945,
      "loss_g0": 0.0,
      "loss_g1": 8.107988357543945,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.447874069213867,
      "loss_g0": 25.447874069213867,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.778539657592773,
      "loss_g0": 11.86009407043457,
      "loss_g1": 6.924106597900391,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 15.10180377960205,
      "loss_g0": 15.169441223144531,
      "loss_g1": 11.075758934020996,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.210468292236328,
      "loss_g0": 25.46538543701172,
      "loss_g1": 10.036798477172852,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 26.827756881713867,
      "loss_g0": 27.222196578979492,
      "loss_g1": 3.3491759300231934,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.209818840026855,
      "loss_g0": 8.287650108337402,
      "loss_g1": 3.5770103931427,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 12.912210464477539,
      "loss_g0": 12.974791526794434,
      "loss_g1": 9.187093734741211,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.9789204597473145,
      "loss_g0": 2.863615036010742,
      "loss_g1": 9.842028617858887,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652313955128193,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.771821022033691,
      "loss_g0": 5.812867164611816,
      "loss_g1": 3.3286690711975098,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.036099433898926,
      "loss_g0": 0.0,
      "loss_g1": 8.036099433898926,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.463440895080566,
      "loss_g0": 7.937896251678467,
      "loss_g1": 6.4386701583862305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.5431342124938965,
      "loss_g0": 0.8839645385742188,
      "loss_g1": 3.587808132171631,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.359071731567383,
      "loss_g0": 9.01385498046875,
      "loss_g1": 7.331270694732666,
      "q_g0": 0.016522597521543503,
      "q_g1": 0.9834774136543274,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 4.255946159362793,
      "loss_g0": 5.663856506347656,
      "loss_g1": 4.232292652130127,
      "q_g0": 0.016522744670510292,
      "q_g1": 0.9834772944450378,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.3835690021514893,
      "loss_g0": 7.045633316040039,
      "loss_g1": 1.2884409427642822,
      "q_g0": 0.01652333326637745,
      "q_g1": 0.9834766387939453,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 7.221598148345947,
      "loss_g0": 12.749900817871094,
      "loss_g1": 7.128714561462402,
      "q_g0": 0.01652390882372856,
      "q_g1": 0.9834761619567871,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.599849224090576,
      "loss_g0": 12.656451225280762,
      "loss_g1": 6.498085021972656,
      "q_g0": 0.0165245421230793,
      "q_g1": 0.9834754467010498,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.20197296142578,
      "loss_g0": 22.20197296142578,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 31.01544189453125,
      "loss_g0": 31.03734016418457,
      "loss_g1": 29.71194839477539,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 34.640830993652344,
      "loss_g0": 34.74244689941406,
      "loss_g1": 28.592227935791016,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 22.8519287109375,
      "loss_g0": 22.914047241210938,
      "loss_g1": 19.15436553955078,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 32.19086837768555,
      "loss_g0": 32.31878662109375,
      "loss_g1": 24.576597213745117,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.618231773376465,
      "loss_g0": 7.836420059204102,
      "loss_g1": 55.14080047607422,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.01652725413441658,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.401962757110596,
      "loss_g0": 6.203001022338867,
      "loss_g1": 18.240510940551758,
      "q_g0": 0.9834715127944946,
      "q_g1": 0.01652848720550537,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.710205078125,
      "loss_g0": 29.028961181640625,
      "loss_g1": 9.741446495056152,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.016526520252227783,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.817497253417969,
      "loss_g0": 11.806044578552246,
      "loss_g1": 12.498971939086914,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526587307453156,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 25.023456573486328,
      "loss_g0": 0.0,
      "loss_g1": 25.023456573486328,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 17.782556533813477,
      "loss_g0": 0.0,
      "loss_g1": 17.782556533813477,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 1.7428524494171143,
      "loss_g0": 9.20669174194336,
      "loss_g1": 1.6174598932266235,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.398176193237305,
      "loss_g0": 9.516109466552734,
      "loss_g1": 5.32899284362793,
      "q_g0": 0.016522854566574097,
      "q_g1": 0.9834771752357483,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 8.60075855255127,
      "loss_g0": 1.4506243467330933,
      "loss_g1": 8.720880508422852,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 3.1127262115478516,
      "loss_g0": 7.137569427490234,
      "loss_g1": 3.045106887817383,
      "q_g0": 0.016522839665412903,
      "q_g1": 0.9834771752357483,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 11.552567481994629,
      "loss_g0": 31.911575317382812,
      "loss_g1": 11.21048355102539,
      "q_g0": 0.01652495190501213,
      "q_g1": 0.9834749698638916,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 9.883913040161133,
      "loss_g0": 0.0,
      "loss_g1": 9.883913040161133,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 5.738917827606201,
      "loss_g0": 14.9725341796875,
      "loss_g1": 5.583792686462402,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 2.8254106044769287,
      "loss_g0": 4.828089237213135,
      "loss_g1": 2.791764974594116,
      "q_g0": 0.016522636637091637,
      "q_g1": 0.9834774136543274,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 6.5923871994018555,
      "loss_g0": 19.890377044677734,
      "loss_g1": 6.368958473205566,
      "q_g0": 0.016524016857147217,
      "q_g1": 0.9834760427474976,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "eval_cer": 0.06647951027410613,
      "eval_loss": 15.722418785095215,
      "eval_runtime": 80.3792,
      "eval_samples_per_second": 20.851,
      "eval_steps_per_second": 5.213,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 28.187427520751953,
      "loss_g0": 46.2003288269043,
      "loss_g1": 27.88474464416504,
      "q_g0": 0.016525881364941597,
      "q_g1": 0.98347407579422,
      "step": 400
    },
    {
      "epoch": 2.030456852791878,
      "gdro_loss": 58.26785659790039,
      "loss_g0": 31.857513427734375,
      "loss_g1": 58.71156692504883,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 400
    },
    {
      "epoch": 2.5380710659898478,
      "grad_norm": 62.7445182800293,
      "learning_rate": 0.00036455614973262033,
      "loss": 38.1779,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.337066650390625,
      "loss_g0": 19.92679786682129,
      "loss_g1": 18.31035804748535,
      "q_g0": 0.016523689031600952,
      "q_g1": 0.9834762811660767,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 28.208024978637695,
      "loss_g0": 17.172008514404297,
      "loss_g1": 28.393430709838867,
      "q_g0": 0.016522550955414772,
      "q_g1": 0.9834774136543274,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.198524475097656,
      "loss_g0": 14.771013259887695,
      "loss_g1": 19.272907257080078,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.949878692626953,
      "loss_g0": 23.325332641601562,
      "loss_g1": 18.876367568969727,
      "q_g0": 0.016522882506251335,
      "q_g1": 0.9834770560264587,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 35.896636962890625,
      "loss_g0": 22.14375877380371,
      "loss_g1": 36.127685546875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.17408561706543,
      "loss_g0": 19.17408561706543,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.28564453125,
      "loss_g0": 9.90831470489502,
      "loss_g1": 32.74574279785156,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 29.47649574279785,
      "loss_g0": 29.53028678894043,
      "loss_g1": 26.274587631225586,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.44174575805664,
      "loss_g0": 19.692218780517578,
      "loss_g1": 4.532597541809082,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.333048343658447,
      "loss_g0": 6.082542896270752,
      "loss_g1": 21.24266815185547,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523968428373337,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.06808853149414,
      "loss_g0": 27.39239501953125,
      "loss_g1": 7.7641706466674805,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.625701904296875,
      "loss_g0": 22.540075302124023,
      "loss_g1": 27.722381591796875,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522955149412155,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.3385009765625,
      "loss_g0": 21.572479248046875,
      "loss_g1": 7.4112653732299805,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.599008560180664,
      "loss_g0": 20.44305992126465,
      "loss_g1": 29.88100242614746,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523387283086777,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 106.9784927368164,
      "loss_g0": 108.2977294921875,
      "loss_g1": 28.45242691040039,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.647380828857422,
      "loss_g0": 27.647380828857422,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 25.170942306518555,
      "loss_g0": 25.425243377685547,
      "loss_g1": 10.033960342407227,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.199159622192383,
      "loss_g0": 0.0,
      "loss_g1": 10.199159622192383,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.690942764282227,
      "loss_g0": 1.6470896005630493,
      "loss_g1": 16.943679809570312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.9977338314056396,
      "loss_g0": 20.547502517700195,
      "loss_g1": 1.6860606670379639,
      "q_g0": 0.01652434840798378,
      "q_g1": 0.9834756851196289,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.752250671386719,
      "loss_g0": 5.3192548751831055,
      "loss_g1": 15.927533149719238,
      "q_g0": 0.01652325503528118,
      "q_g1": 0.9834766983985901,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.147329330444336,
      "loss_g0": 14.747200012207031,
      "loss_g1": 12.103647232055664,
      "q_g0": 0.016523534432053566,
      "q_g1": 0.9834765195846558,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.252166748046875,
      "loss_g0": 21.212215423583984,
      "loss_g1": 6.000796318054199,
      "q_g0": 0.016525091603398323,
      "q_g1": 0.983474850654602,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.529735565185547,
      "loss_g0": 0.0,
      "loss_g1": 8.529735565185547,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.394837379455566,
      "loss_g0": 33.49982833862305,
      "loss_g1": 12.040273666381836,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.406208038330078,
      "loss_g0": 3.2041358947753906,
      "loss_g1": 10.527202606201172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.574466705322266,
      "loss_g0": 4.942942142486572,
      "loss_g1": 7.61867618560791,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.314165115356445,
      "loss_g0": 14.14822769165039,
      "loss_g1": 10.249751091003418,
      "q_g0": 0.01652282290160656,
      "q_g1": 0.9834771752357483,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.762459754943848,
      "loss_g0": 12.458269119262695,
      "loss_g1": 10.733969688415527,
      "q_g0": 0.016522999852895737,
      "q_g1": 0.9834769368171692,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.79296588897705,
      "loss_g0": 3.044520854949951,
      "loss_g1": 11.939939498901367,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.458988189697266,
      "loss_g0": 0.0,
      "loss_g1": 14.458988189697266,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.847978591918945,
      "loss_g0": 15.990707397460938,
      "loss_g1": 18.895978927612305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.464146137237549,
      "loss_g0": 19.05602264404297,
      "loss_g1": 6.252585411071777,
      "q_g0": 0.016523728147149086,
      "q_g1": 0.9834762811660767,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.130068778991699,
      "loss_g0": 19.35267448425293,
      "loss_g1": 3.874283790588379,
      "q_g0": 0.016525307670235634,
      "q_g1": 0.9834746718406677,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.711710453033447,
      "loss_g0": 17.188274383544922,
      "loss_g1": 5.518856048583984,
      "q_g0": 0.01652650348842144,
      "q_g1": 0.9834734797477722,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.505589485168457,
      "loss_g0": 7.8219804763793945,
      "loss_g1": 6.483468532562256,
      "q_g0": 0.016526643186807632,
      "q_g1": 0.9834733605384827,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.980848789215088,
      "loss_g0": 20.6949462890625,
      "loss_g1": 7.767179012298584,
      "q_g0": 0.016527967527508736,
      "q_g1": 0.9834719896316528,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.1455719470977783,
      "loss_g0": 2.6560232639312744,
      "loss_g1": 2.136993408203125,
      "q_g0": 0.016528021544218063,
      "q_g1": 0.9834719896316528,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.2475059032440186,
      "loss_g0": 49.89396286010742,
      "loss_g1": 2.463341236114502,
      "q_g0": 0.01653287559747696,
      "q_g1": 0.983467161655426,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.344005584716797,
      "loss_g0": 13.070147514343262,
      "loss_g1": 5.21411657333374,
      "q_g0": 0.01653367467224598,
      "q_g1": 0.9834663271903992,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.92401123046875,
      "loss_g0": 10.664576530456543,
      "loss_g1": 2.793872833251953,
      "q_g0": 0.016534488648176193,
      "q_g1": 0.9834655523300171,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.1404242515563965,
      "loss_g0": 8.746241569519043,
      "loss_g1": 7.1134257316589355,
      "q_g0": 0.016534648835659027,
      "q_g1": 0.983465313911438,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.54222297668457,
      "loss_g0": 3.2193241119384766,
      "loss_g1": 14.732576370239258,
      "q_g0": 0.016533464193344116,
      "q_g1": 0.9834665656089783,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.1431870460510254,
      "loss_g0": 12.421099662780762,
      "loss_g1": 1.9703892469406128,
      "q_g0": 0.016534533351659775,
      "q_g1": 0.9834654331207275,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.78458571434021,
      "loss_g0": 0.9341654181480408,
      "loss_g1": 2.8156957626342773,
      "q_g0": 0.016534339636564255,
      "q_g1": 0.9834656715393066,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.4923698902130127,
      "loss_g0": 3.473792552947998,
      "loss_g1": 2.47586989402771,
      "q_g0": 0.016534440219402313,
      "q_g1": 0.9834655523300171,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.169485330581665,
      "loss_g0": 16.187280654907227,
      "loss_g1": 1.933790922164917,
      "q_g0": 0.01653590425848961,
      "q_g1": 0.9834640622138977,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.3961893320083618,
      "loss_g0": 9.312318801879883,
      "loss_g1": 1.2630813121795654,
      "q_g0": 0.01653672754764557,
      "q_g1": 0.9834632277488708,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.9665729999542236,
      "loss_g0": 2.9665729999542236,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.11720609664917,
      "loss_g0": 6.213493347167969,
      "loss_g1": 0.38582974672317505,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 99.78489685058594,
      "loss_g0": 100.80162811279297,
      "loss_g1": 39.26531219482422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 25.279979705810547,
      "loss_g0": 25.27089500427246,
      "loss_g1": 25.820755004882812,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.0165224801748991,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 33.82019805908203,
      "loss_g0": 0.0,
      "loss_g1": 33.82019805908203,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 71.8717269897461,
      "loss_g0": 22.953941345214844,
      "loss_g1": 72.69354248046875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 35.221683502197266,
      "loss_g0": 30.137149810791016,
      "loss_g1": 35.30710220336914,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.005756378173828,
      "loss_g0": 19.632055282592773,
      "loss_g1": 12.894430160522461,
      "q_g0": 0.016523106023669243,
      "q_g1": 0.9834768176078796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.09123992919922,
      "loss_g0": 7.157749176025391,
      "loss_g1": 19.291723251342773,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.943045616149902,
      "loss_g0": 12.308910369873047,
      "loss_g1": 7.869697093963623,
      "q_g0": 0.01652287319302559,
      "q_g1": 0.9834770560264587,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.457170486450195,
      "loss_g0": 23.925350189208984,
      "loss_g1": 16.331695556640625,
      "q_g0": 0.01652364991605282,
      "q_g1": 0.9834764003753662,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 38.40223693847656,
      "loss_g0": 55.39943313598633,
      "loss_g1": 38.11662673950195,
      "q_g0": 0.01652541570365429,
      "q_g1": 0.9834745526313782,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.190969467163086,
      "loss_g0": 17.190969467163086,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.028168678283691,
      "loss_g0": 0.0,
      "loss_g1": 13.028168678283691,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.542381286621094,
      "loss_g0": 39.351280212402344,
      "loss_g1": 14.125591278076172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.34817123413086,
      "loss_g0": 16.622516632080078,
      "loss_g1": 9.225955963134766,
      "q_g0": 0.016523178666830063,
      "q_g1": 0.9834768176078796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.01650619506836,
      "loss_g0": 13.852737426757812,
      "loss_g1": 21.136856079101562,
      "q_g0": 0.01652243360877037,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 35.403076171875,
      "loss_g0": 8.263035774230957,
      "loss_g1": 35.85902786254883,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.170772552490234,
      "loss_g0": 31.240646362304688,
      "loss_g1": 4.732725620269775,
      "q_g0": 0.016525132581591606,
      "q_g1": 0.983474850654602,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.89846420288086,
      "loss_g0": 25.921791076660156,
      "loss_g1": 10.646005630493164,
      "q_g0": 0.01652669534087181,
      "q_g1": 0.9834732413291931,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.480952262878418,
      "loss_g0": 23.232112884521484,
      "loss_g1": 15.350691795349121,
      "q_g0": 0.01652749441564083,
      "q_g1": 0.983472466468811,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.711992263793945,
      "loss_g0": 42.16876220703125,
      "loss_g1": 16.284114837646484,
      "q_g0": 0.016530144959688187,
      "q_g1": 0.9834697842597961,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.557943344116211,
      "loss_g0": 8.365458488464355,
      "loss_g1": 14.662022590637207,
      "q_g0": 0.016529500484466553,
      "q_g1": 0.9834704995155334,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.963886737823486,
      "loss_g0": 8.57811450958252,
      "loss_g1": 6.936755657196045,
      "q_g0": 0.016529666259884834,
      "q_g1": 0.9834703803062439,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.415452003479004,
      "loss_g0": 13.415452003479004,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.954423904418945,
      "loss_g0": 14.954423904418945,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.947534561157227,
      "loss_g0": 16.827974319458008,
      "loss_g1": 24.06418228149414,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.749274253845215,
      "loss_g0": 12.851042747497559,
      "loss_g1": 6.691618919372559,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.715155601501465,
      "loss_g0": 10.837286949157715,
      "loss_g1": 3.4454071521759033,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.123178482055664,
      "loss_g0": 23.320581436157227,
      "loss_g1": 11.373039245605469,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 24.185871124267578,
      "loss_g0": 24.334613800048828,
      "loss_g1": 15.332110404968262,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 43.0509033203125,
      "loss_g0": 43.39632797241211,
      "loss_g1": 22.489824295043945,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.122526168823242,
      "loss_g0": 5.143856525421143,
      "loss_g1": 3.8528685569763184,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.576848030090332,
      "loss_g0": 11.723413467407227,
      "loss_g1": 2.8526721000671387,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.220736503601074,
      "loss_g0": 9.155525207519531,
      "loss_g1": 13.102229118347168,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522830352187157,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.952890396118164,
      "loss_g0": 16.00052833557129,
      "loss_g1": 13.117303848266602,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.5918426513671875,
      "loss_g0": 1.4717141389846802,
      "loss_g1": 8.741985321044922,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523271799087524,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.504696369171143,
      "loss_g0": 7.516762733459473,
      "loss_g1": 6.786470890045166,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523201018571854,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.818511009216309,
      "loss_g0": 8.8176851272583,
      "loss_g1": 8.867685317993164,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.0165232103317976,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.354751110076904,
      "loss_g0": 5.412448883056641,
      "loss_g1": 1.9204514026641846,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522854566574097,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.618864059448242,
      "loss_g0": 10.759340286254883,
      "loss_g1": 2.257183074951172,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.587512969970703,
      "loss_g0": 6.587512969970703,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.94141960144043,
      "loss_g0": 9.94141960144043,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.239226341247559,
      "loss_g0": 15.433695793151855,
      "loss_g1": 3.6636385917663574,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.3645174503326416,
      "loss_g0": 2.3113324642181396,
      "loss_g1": 5.530211925506592,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522753983736038,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.670786380767822,
      "loss_g0": 4.698988914489746,
      "loss_g1": 2.9921014308929443,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.866796016693115,
      "loss_g0": 5.904078960418701,
      "loss_g1": 3.6475625038146973,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.201964378356934,
      "loss_g0": 10.318655967712402,
      "loss_g1": 3.256063461303711,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.68613052368164,
      "loss_g0": 13.901105880737305,
      "loss_g1": 0.8899987936019897,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 28.75313377380371,
      "loss_g0": 29.121047973632812,
      "loss_g1": 6.853423595428467,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.7000346183776855,
      "loss_g0": 0.0,
      "loss_g1": 3.7000346183776855,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.161674976348877,
      "loss_g0": 4.2945942878723145,
      "loss_g1": 5.176241874694824,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 30.644254684448242,
      "loss_g0": 69.17727661132812,
      "loss_g1": 29.996740341186523,
      "q_g0": 0.01652643457055092,
      "q_g1": 0.9834735989570618,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.205806732177734,
      "loss_g0": 39.082942962646484,
      "loss_g1": 16.8381290435791,
      "q_g0": 0.016528712585568428,
      "q_g1": 0.9834713339805603,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.540950775146484,
      "loss_g0": 19.39265251159668,
      "loss_g1": 9.375367164611816,
      "q_g0": 0.016529735177755356,
      "q_g1": 0.9834702610969543,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 50.58699417114258,
      "loss_g0": 14.242284774780273,
      "loss_g1": 51.197715759277344,
      "q_g0": 0.016525959596037865,
      "q_g1": 0.98347407579422,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.937769889831543,
      "loss_g0": 15.236994743347168,
      "loss_g1": 14.93274211883545,
      "q_g0": 0.0165259912610054,
      "q_g1": 0.9834739565849304,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.452964782714844,
      "loss_g0": 19.124059677124023,
      "loss_g1": 23.52570343017578,
      "q_g0": 0.016525542363524437,
      "q_g1": 0.9834744334220886,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.731035232543945,
      "loss_g0": 16.731035232543945,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.654353141784668,
      "loss_g0": 0.0,
      "loss_g1": 15.654353141784668,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.419564723968506,
      "loss_g0": 18.082229614257812,
      "loss_g1": 4.190032005310059,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.756051063537598,
      "loss_g0": 11.442412376403809,
      "loss_g1": 12.778120040893555,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.716240406036377,
      "loss_g0": 18.934125900268555,
      "loss_g1": 7.527766704559326,
      "q_g0": 0.016523588448762894,
      "q_g1": 0.9834764003753662,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 39.404945373535156,
      "loss_g0": 49.23295974731445,
      "loss_g1": 39.239810943603516,
      "q_g0": 0.016524603590369225,
      "q_g1": 0.9834753274917603,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.813331604003906,
      "loss_g0": 5.629913806915283,
      "loss_g1": 20.051624298095703,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.08286476135254,
      "loss_g0": 0.0,
      "loss_g1": 17.08286476135254,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.932157516479492,
      "loss_g0": 25.088438034057617,
      "loss_g1": 9.677531242370605,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.275836944580078,
      "loss_g0": 21.275836944580078,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.547104835510254,
      "loss_g0": 11.417521476745605,
      "loss_g1": 19.26041603088379,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.195625305175781,
      "loss_g0": 9.154329299926758,
      "loss_g1": 11.653704643249512,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 64.72859954833984,
      "loss_g0": 65.4520034790039,
      "loss_g1": 21.66859245300293,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.894186496734619,
      "loss_g0": 0.0,
      "loss_g1": 6.894186496734619,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.813929557800293,
      "loss_g0": 18.065589904785156,
      "loss_g1": 9.675301551818848,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.797689437866211,
      "loss_g0": 0.0,
      "loss_g1": 10.797689437866211,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.887284278869629,
      "loss_g0": 1.1167469024658203,
      "loss_g1": 10.034628868103027,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.119085311889648,
      "loss_g0": 14.24846076965332,
      "loss_g1": 12.083311080932617,
      "q_g0": 0.016522642225027084,
      "q_g1": 0.9834772944450378,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.440841674804688,
      "loss_g0": 8.11400318145752,
      "loss_g1": 12.513532638549805,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.532447814941406,
      "loss_g0": 6.571664333343506,
      "loss_g1": 21.783788681030273,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.198442459106445,
      "loss_g0": 10.758193969726562,
      "loss_g1": 15.273037910461426,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.093975067138672,
      "loss_g0": 9.295961380004883,
      "loss_g1": 9.090581893920898,
      "q_g0": 0.016522448509931564,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.377139568328857,
      "loss_g0": 6.762639999389648,
      "loss_g1": 4.337062835693359,
      "q_g0": 0.016522690653800964,
      "q_g1": 0.9834772944450378,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.437967300415039,
      "loss_g0": 0.0,
      "loss_g1": 6.437967300415039,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.53904914855957,
      "loss_g0": 8.404190063476562,
      "loss_g1": 5.490914821624756,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.2799465656280518,
      "loss_g0": 7.054537296295166,
      "loss_g1": 2.19973087310791,
      "q_g0": 0.01652292162179947,
      "q_g1": 0.9834770560264587,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.028444766998291,
      "loss_g0": 6.921807289123535,
      "loss_g1": 7.03023624420166,
      "q_g0": 0.016522906720638275,
      "q_g1": 0.9834770560264587,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.090926647186279,
      "loss_g0": 4.199592113494873,
      "loss_g1": 7.139501094818115,
      "q_g0": 0.016522597521543503,
      "q_g1": 0.9834774136543274,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 24.22968101501465,
      "loss_g0": 9.300703048706055,
      "loss_g1": 24.480487823486328,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.902679443359375,
      "loss_g0": 5.704968452453613,
      "loss_g1": 8.956400871276855,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.701608657836914,
      "loss_g0": 9.289413452148438,
      "loss_g1": 9.70853328704834,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.178226470947266,
      "loss_g0": 9.50291633605957,
      "loss_g1": 8.15597152709961,
      "q_g0": 0.01652255840599537,
      "q_g1": 0.9834774136543274,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.757374286651611,
      "loss_g0": 3.1343696117401123,
      "loss_g1": 4.784640789031982,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.958626747131348,
      "loss_g0": 3.3021206855773926,
      "loss_g1": 11.08725643157959,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 0.3012249171733856,
      "loss_g0": 2.114469051361084,
      "loss_g1": 0.27076205611228943,
      "q_g0": 0.016522612422704697,
      "q_g1": 0.9834774136543274,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.3752946853637695,
      "loss_g0": 13.800115585327148,
      "loss_g1": 4.216946601867676,
      "q_g0": 0.016523588448762894,
      "q_g1": 0.9834764003753662,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.213117599487305,
      "loss_g0": 17.065444946289062,
      "loss_g1": 12.131589889526367,
      "q_g0": 0.016524093225598335,
      "q_g1": 0.983475923538208,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.9585790634155273,
      "loss_g0": 8.570216178894043,
      "loss_g1": 1.847487449645996,
      "q_g0": 0.01652478240430355,
      "q_g1": 0.9834752082824707,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.690962791442871,
      "loss_g0": 8.363311767578125,
      "loss_g1": 13.780477523803711,
      "q_g0": 0.01652422547340393,
      "q_g1": 0.9834758043289185,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.009273052215576,
      "loss_g0": 10.344399452209473,
      "loss_g1": 6.953235626220703,
      "q_g0": 0.016524571925401688,
      "q_g1": 0.9834754467010498,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.123319387435913,
      "loss_g0": 2.1510682106018066,
      "loss_g1": 3.139655113220215,
      "q_g0": 0.01652447320520878,
      "q_g1": 0.9834755659103394,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.8368767499923706,
      "loss_g0": 1.8368767499923706,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.658504486083984,
      "loss_g0": 18.918975830078125,
      "loss_g1": 3.1542956829071045,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.880934715270996,
      "loss_g0": 1.7857764959335327,
      "loss_g1": 7.5449066162109375,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652301475405693,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.685675621032715,
      "loss_g0": 6.184260368347168,
      "loss_g1": 36.52505874633789,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.0165261160582304,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 36.285953521728516,
      "loss_g0": 36.638099670410156,
      "loss_g1": 15.326985359191895,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.0165239367634058,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.068815231323242,
      "loss_g0": 22.022220611572266,
      "loss_g1": 24.841960906982422,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652422547340393,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.008118629455566,
      "loss_g0": 12.895257949829102,
      "loss_g1": 19.7249755859375,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524922102689743,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 31.405160903930664,
      "loss_g0": 31.395355224609375,
      "loss_g1": 31.9886474609375,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016524983569979668,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.19022560119629,
      "loss_g0": 27.281644821166992,
      "loss_g1": 21.749263763427734,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652441918849945,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.57411003112793,
      "loss_g0": 20.53292465209961,
      "loss_g1": 23.025279998779297,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524672508239746,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 29.200475692749023,
      "loss_g0": 29.416616439819336,
      "loss_g1": 16.335758209228516,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652333326637745,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.957849502563477,
      "loss_g0": 24.26626205444336,
      "loss_g1": 5.599900722503662,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.6632466316223145,
      "loss_g0": 6.606901168823242,
      "loss_g1": 10.017072677612305,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522767022252083,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.982430458068848,
      "loss_g0": 15.807958602905273,
      "loss_g1": 26.36673355102539,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523851081728935,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.943801879882812,
      "loss_g0": 28.183746337890625,
      "loss_g1": 13.661375045776367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.99553680419922,
      "loss_g0": 21.25181770324707,
      "loss_g1": 5.74068546295166,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.166672706604004,
      "loss_g0": 12.006013870239258,
      "loss_g1": 21.72903823852539,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523417085409164,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.630034446716309,
      "loss_g0": 8.662063598632812,
      "loss_g1": 6.723652362823486,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523215919733047,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.552542686462402,
      "loss_g0": 14.688604354858398,
      "loss_g1": 6.453639507293701,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.249787330627441,
      "loss_g0": 9.035070419311523,
      "loss_g1": 22.02947998046875,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523750498890877,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.5247802734375,
      "loss_g0": 17.5247802734375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.845909833908081,
      "loss_g0": 3.757296085357666,
      "loss_g1": 9.120536804199219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 51.982975006103516,
      "loss_g0": 52.677391052246094,
      "loss_g1": 10.64857006072998,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.82358455657959,
      "loss_g0": 9.892257690429688,
      "loss_g1": 5.735888957977295,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.937178611755371,
      "loss_g0": 9.919977188110352,
      "loss_g1": 10.961057662963867,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.387287139892578,
      "loss_g0": 23.566204071044922,
      "loss_g1": 12.737431526184082,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.384259223937988,
      "loss_g0": 10.38807487487793,
      "loss_g1": 10.157099723815918,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.48726749420166,
      "loss_g0": 11.529226303100586,
      "loss_g1": 8.989700317382812,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.197962760925293,
      "loss_g0": 0.0,
      "loss_g1": 9.197962760925293,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.057249069213867,
      "loss_g0": 8.23123550415039,
      "loss_g1": 11.10472583770752,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.309171676635742,
      "loss_g0": 6.818981170654297,
      "loss_g1": 11.38460636138916,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.47125816345215,
      "loss_g0": 22.47125816345215,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.427319526672363,
      "loss_g0": 9.435110092163086,
      "loss_g1": 8.963587760925293,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.096006393432617,
      "loss_g0": 7.096006393432617,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.251453399658203,
      "loss_g0": 10.238122940063477,
      "loss_g1": 11.044883728027344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.141674041748047,
      "loss_g0": 13.222505569458008,
      "loss_g1": 8.330254554748535,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.744311332702637,
      "loss_g0": 11.89704704284668,
      "loss_g1": 2.652855634689331,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.856477737426758,
      "loss_g0": 10.911582946777344,
      "loss_g1": 7.576404571533203,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.04560661315918,
      "loss_g0": 11.128290176391602,
      "loss_g1": 6.12395715713501,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.932697057723999,
      "loss_g0": 3.8691000938415527,
      "loss_g1": 7.718128204345703,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 32.03215789794922,
      "loss_g0": 32.53605651855469,
      "loss_g1": 2.038151264190674,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.065491676330566,
      "loss_g0": 10.027704238891602,
      "loss_g1": 12.314685821533203,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.995166778564453,
      "loss_g0": 0.0,
      "loss_g1": 9.995166778564453,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.398743152618408,
      "loss_g0": 42.214569091796875,
      "loss_g1": 6.813837051391602,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.813186645507812,
      "loss_g0": 0.0,
      "loss_g1": 10.813186645507812,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.438036918640137,
      "loss_g0": 8.285400390625,
      "loss_g1": 5.390201091766357,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.56319808959961,
      "loss_g0": 7.342790603637695,
      "loss_g1": 12.650900840759277,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.792740821838379,
      "loss_g0": 5.581080436706543,
      "loss_g1": 11.897096633911133,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.5465303659439087,
      "loss_g0": 3.931133270263672,
      "loss_g1": 1.506468415260315,
      "q_g0": 0.01652267388999462,
      "q_g1": 0.9834772944450378,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.5490771532058716,
      "loss_g0": 5.900479793548584,
      "loss_g1": 1.4759703874588013,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.0763280391693115,
      "loss_g0": 4.973720550537109,
      "loss_g1": 2.02764892578125,
      "q_g0": 0.016523433849215508,
      "q_g1": 0.9834766387939453,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.986740112304688,
      "loss_g0": 16.333492279052734,
      "loss_g1": 16.99771499633789,
      "q_g0": 0.01652335561811924,
      "q_g1": 0.9834766387939453,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 0.48208460211753845,
      "loss_g0": 10.840502738952637,
      "loss_g1": 0.3080417215824127,
      "q_g0": 0.016524426639080048,
      "q_g1": 0.9834755659103394,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 36.18851089477539,
      "loss_g0": 83.71261596679688,
      "loss_g1": 35.389766693115234,
      "q_g0": 0.01652936451137066,
      "q_g1": 0.983470618724823,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 24.706573486328125,
      "loss_g0": 20.39239501953125,
      "loss_g1": 24.77907943725586,
      "q_g0": 0.016528913751244545,
      "q_g1": 0.9834710955619812,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.667734146118164,
      "loss_g0": 0.0,
      "loss_g1": 20.667734146118164,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 24.143352508544922,
      "loss_g0": 36.892982482910156,
      "loss_g1": 23.929157257080078,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.83359146118164,
      "loss_g0": 25.612525939941406,
      "loss_g1": 21.770103454589844,
      "q_g0": 0.016522813588380814,
      "q_g1": 0.9834771752357483,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 26.84828758239746,
      "loss_g0": 28.758813858032227,
      "loss_g1": 26.816190719604492,
      "q_g0": 0.01652301475405693,
      "q_g1": 0.9834769368171692,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.383027076721191,
      "loss_g0": 20.38037872314453,
      "loss_g1": 10.215055465698242,
      "q_g0": 0.0165240541100502,
      "q_g1": 0.983475923538208,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.814266204833984,
      "loss_g0": 24.985042572021484,
      "loss_g1": 12.609761238098145,
      "q_g0": 0.016525324434041977,
      "q_g1": 0.9834746718406677,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.082160949707031,
      "loss_g0": 46.84300994873047,
      "loss_g1": 13.531566619873047,
      "q_g0": 0.016528727486729622,
      "q_g1": 0.9834713339805603,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.83071517944336,
      "loss_g0": 21.05609893798828,
      "loss_g1": 13.70927619934082,
      "q_g0": 0.01652947999536991,
      "q_g1": 0.9834704995155334,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.066577911376953,
      "loss_g0": 0.0,
      "loss_g1": 13.066577911376953,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.771787643432617,
      "loss_g0": 0.0,
      "loss_g1": 14.771787643432617,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 36.49275207519531,
      "loss_g0": 18.9864559173584,
      "loss_g1": 36.78685760498047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.746971130371094,
      "loss_g0": 15.17040729522705,
      "loss_g1": 8.639053344726562,
      "q_g0": 0.016523092985153198,
      "q_g1": 0.9834769368171692,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.068517684936523,
      "loss_g0": 11.14334774017334,
      "loss_g1": 19.20166015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.913549423217773,
      "loss_g0": 29.434612274169922,
      "loss_g1": 17.719980239868164,
      "q_g0": 0.01652361825108528,
      "q_g1": 0.9834764003753662,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.242249488830566,
      "loss_g0": 8.067707061767578,
      "loss_g1": 15.362784385681152,
      "q_g0": 0.01652287133038044,
      "q_g1": 0.9834771752357483,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.568034172058105,
      "loss_g0": 7.737618923187256,
      "loss_g1": 15.6995849609375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.399983406066895,
      "loss_g0": 10.222311019897461,
      "loss_g1": 9.386167526245117,
      "q_g0": 0.016522511839866638,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.58487319946289,
      "loss_g0": 12.785615921020508,
      "loss_g1": 27.833499908447266,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.591816902160645,
      "loss_g0": 18.403085708618164,
      "loss_g1": 10.460580825805664,
      "q_g0": 0.016523240134119987,
      "q_g1": 0.9834766983985901,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.88874626159668,
      "loss_g0": 22.667856216430664,
      "loss_g1": 19.842052459716797,
      "q_g0": 0.016523534432053566,
      "q_g1": 0.9834765195846558,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.800915718078613,
      "loss_g0": 2.0951614379882812,
      "loss_g1": 15.014371871948242,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.020981788635254,
      "loss_g0": 11.57292366027832,
      "loss_g1": 8.978109359741211,
      "q_g0": 0.016522690653800964,
      "q_g1": 0.9834772944450378,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.972888946533203,
      "loss_g0": 16.58818817138672,
      "loss_g1": 14.945751190185547,
      "q_g0": 0.016522862017154694,
      "q_g1": 0.9834771752357483,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.367895603179932,
      "loss_g0": 23.5479679107666,
      "loss_g1": 6.079230785369873,
      "q_g0": 0.016524652019143105,
      "q_g1": 0.9834753274917603,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.835168838500977,
      "loss_g0": 8.07238483428955,
      "loss_g1": 11.898390769958496,
      "q_g0": 0.01652425527572632,
      "q_g1": 0.9834756851196289,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.199307441711426,
      "loss_g0": 6.820552825927734,
      "loss_g1": 8.22247314453125,
      "q_g0": 0.016524117439985275,
      "q_g1": 0.983475923538208,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.090354919433594,
      "loss_g0": 6.166874885559082,
      "loss_g1": 23.37466812133789,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.6107239723205566,
      "loss_g0": 21.154508590698242,
      "loss_g1": 3.31595516204834,
      "q_g0": 0.01652424782514572,
      "q_g1": 0.9834758043289185,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.807828903198242,
      "loss_g0": 36.4249267578125,
      "loss_g1": 16.47818374633789,
      "q_g0": 0.01652628742158413,
      "q_g1": 0.9834737181663513,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.208709239959717,
      "loss_g0": 8.46601676940918,
      "loss_g1": 4.137167453765869,
      "q_g0": 0.016526727005839348,
      "q_g1": 0.9834732413291931,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.001023769378662,
      "loss_g0": 25.08637237548828,
      "loss_g1": 6.6970744132995605,
      "q_g0": 0.016528604552149773,
      "q_g1": 0.9834713935852051,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.154695510864258,
      "loss_g0": 9.071260452270508,
      "loss_g1": 6.105677604675293,
      "q_g0": 0.0165289044380188,
      "q_g1": 0.9834710955619812,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.590098857879639,
      "loss_g0": 1.227889060974121,
      "loss_g1": 7.697022438049316,
      "q_g0": 0.01652824692428112,
      "q_g1": 0.9834717512130737,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.602654457092285,
      "loss_g0": 14.602654457092285,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.719637870788574,
      "loss_g0": 15.731776237487793,
      "loss_g1": 14.997074127197266,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.469147682189941,
      "loss_g0": 6.47308874130249,
      "loss_g1": 6.234557151794434,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.505280017852783,
      "loss_g0": 5.472845077514648,
      "loss_g1": 7.435882568359375,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522621735930443,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.08416748046875,
      "loss_g0": 27.434558868408203,
      "loss_g1": 6.227551460266113,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.114860534667969,
      "loss_g0": 9.211620330810547,
      "loss_g1": 3.3553454875946045,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.435626983642578,
      "loss_g0": 4.3541364669799805,
      "loss_g1": 9.286111831665039,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652292162179947,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.683061599731445,
      "loss_g0": 16.900957107543945,
      "loss_g1": 3.71311354637146,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.140573501586914,
      "loss_g0": 14.356319427490234,
      "loss_g1": 1.2985255718231201,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.626338958740234,
      "loss_g0": 10.719215393066406,
      "loss_g1": 5.097936153411865,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.3920135498046875,
      "loss_g0": 6.4627227783203125,
      "loss_g1": 2.183107852935791,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.9797481298446655,
      "loss_g0": 2.0022857189178467,
      "loss_g1": 0.6382272243499756,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.939530372619629,
      "loss_g0": 12.056074142456055,
      "loss_g1": 5.002372741699219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.920276165008545,
      "loss_g0": 2.778108596801758,
      "loss_g1": 11.38216781616211,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523301601409912,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.433866500854492,
      "loss_g0": 2.459916830062866,
      "loss_g1": 0.8833233118057251,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523147001862526,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 35.84471130371094,
      "loss_g0": 35.805450439453125,
      "loss_g1": 38.18147277832031,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523394733667374,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.9378662109375,
      "loss_g0": 28.005626678466797,
      "loss_g1": 23.904613494873047,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522971913218498,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 29.46559715270996,
      "loss_g0": 29.64089584350586,
      "loss_g1": 19.031129837036133,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 59.36271286010742,
      "loss_g0": 59.36271286010742,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.581859588623047,
      "loss_g0": 18.581859588623047,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.714559555053711,
      "loss_g0": 5.827065467834473,
      "loss_g1": 59.541587829589844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.26248550415039,
      "loss_g0": 21.335519790649414,
      "loss_g1": 16.915224075317383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.094331741333008,
      "loss_g0": 6.811764717102051,
      "loss_g1": 23.912006378173828,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524169594049454,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.45627498626709,
      "loss_g0": 15.210627555847168,
      "loss_g1": 30.07517433166504,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.016525696963071823,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 64.30079650878906,
      "loss_g0": 65.03388214111328,
      "loss_g1": 20.66486358642578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.753007888793945,
      "loss_g0": 12.753007888793945,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.385231018066406,
      "loss_g0": 19.5333251953125,
      "loss_g1": 10.570009231567383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.391788482666016,
      "loss_g0": 0.0,
      "loss_g1": 14.391788482666016,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.887163162231445,
      "loss_g0": 0.0,
      "loss_g1": 18.887163162231445,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.360944747924805,
      "loss_g0": 34.713836669921875,
      "loss_g1": 19.103015899658203,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.787942886352539,
      "loss_g0": 10.906739234924316,
      "loss_g1": 15.86994743347168,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.933847427368164,
      "loss_g0": 17.303009033203125,
      "loss_g1": 12.860444068908691,
      "q_g0": 0.01652287319302559,
      "q_g1": 0.9834770560264587,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.50619888305664,
      "loss_g0": 16.7830867767334,
      "loss_g1": 14.46794605255127,
      "q_g0": 0.016523106023669243,
      "q_g1": 0.9834768176078796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.45982837677002,
      "loss_g0": 71.7808837890625,
      "loss_g1": 8.412372589111328,
      "q_g0": 0.016529588028788567,
      "q_g1": 0.9834703803062439,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.8625617027282715,
      "loss_g0": 12.734460830688477,
      "loss_g1": 2.6966304779052734,
      "q_g0": 0.016530610620975494,
      "q_g1": 0.9834693670272827,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.54524803161621,
      "loss_g0": 22.591825485229492,
      "loss_g1": 20.510848999023438,
      "q_g0": 0.016530821099877357,
      "q_g1": 0.9834692478179932,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 24.612934112548828,
      "loss_g0": 10.367456436157227,
      "loss_g1": 24.852359771728516,
      "q_g0": 0.01652933843433857,
      "q_g1": 0.983470618724823,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.1940152645111084,
      "loss_g0": 7.7594146728515625,
      "loss_g1": 1.0836650133132935,
      "q_g0": 0.016530020162463188,
      "q_g1": 0.9834699034690857,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.960092067718506,
      "loss_g0": 20.325632095336914,
      "loss_g1": 7.752236843109131,
      "q_g0": 0.016531307250261307,
      "q_g1": 0.9834686517715454,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.072710990905762,
      "loss_g0": 11.132551193237305,
      "loss_g1": 8.021276473999023,
      "q_g0": 0.016531627625226974,
      "q_g1": 0.9834684133529663,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.402814865112305,
      "loss_g0": 8.074732780456543,
      "loss_g1": 22.64364242553711,
      "q_g0": 0.01653013750910759,
      "q_g1": 0.9834697842597961,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 29.74512481689453,
      "loss_g0": 29.74512481689453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.908554077148438,
      "loss_g0": 12.988754272460938,
      "loss_g1": 8.134733200073242,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.643723964691162,
      "loss_g0": 7.478976249694824,
      "loss_g1": 17.449533462524414,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523443162441254,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.705058097839355,
      "loss_g0": 9.734103202819824,
      "loss_g1": 7.976267337799072,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523264348506927,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.455049514770508,
      "loss_g0": 12.527915000915527,
      "loss_g1": 8.11795425415039,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522813588380814,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.160502910614014,
      "loss_g0": 5.132297515869141,
      "loss_g1": 6.839354991912842,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652299053966999,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.9091732501983643,
      "loss_g0": 2.8314368724823,
      "loss_g1": 7.536046504974365,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523465514183044,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.85973358154297,
      "loss_g0": 28.011367797851562,
      "loss_g1": 18.833826065063477,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522521153092384,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.6469446420669556,
      "loss_g0": 1.6030726432800293,
      "loss_g1": 4.258315086364746,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.060894966125488,
      "loss_g0": 11.136149406433105,
      "loss_g1": 6.581486701965332,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.827499389648438,
      "loss_g0": 21.05309295654297,
      "loss_g1": 7.399212837219238,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.392626762390137,
      "loss_g0": 11.515528678894043,
      "loss_g1": 4.0770416259765625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.027228355407715,
      "loss_g0": 11.081059455871582,
      "loss_g1": 7.82297945022583,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.760955810546875,
      "loss_g0": 23.93269920349121,
      "loss_g1": 13.538153648376465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.532573699951172,
      "loss_g0": 9.555377960205078,
      "loss_g1": 8.175145149230957,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.510221004486084,
      "loss_g0": 5.546009063720703,
      "loss_g1": 3.3799538612365723,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 0.856674075126648,
      "loss_g0": 0.837990939617157,
      "loss_g1": 1.9687556028366089,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522543504834175,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.00503158569336,
      "loss_g0": 15.12087345123291,
      "loss_g1": 8.109649658203125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.78063440322876,
      "loss_g0": 4.787723541259766,
      "loss_g1": 4.358682155609131,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.328043937683105,
      "loss_g0": 11.513877868652344,
      "loss_g1": 0.2665187418460846,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.10483455657959,
      "loss_g0": 9.131875038146973,
      "loss_g1": 7.495308876037598,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.29072380065918,
      "loss_g0": 12.29072380065918,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.121753692626953,
      "loss_g0": 0.0,
      "loss_g1": 4.121753692626953,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 0.5527008771896362,
      "loss_g0": 3.28670597076416,
      "loss_g1": 0.506769597530365,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.157421112060547,
      "loss_g0": 19.157421112060547,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.88577938079834,
      "loss_g0": 15.985118865966797,
      "loss_g1": 9.972716331481934,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 33.60390853881836,
      "loss_g0": 33.60390853881836,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.00278091430664,
      "loss_g0": 27.01971435546875,
      "loss_g1": 25.994796752929688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 20.760848999023438,
      "loss_g0": 20.760848999023438,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 23.341707229614258,
      "loss_g0": 23.52523422241211,
      "loss_g1": 12.417367935180664,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.645015716552734,
      "loss_g0": 22.726356506347656,
      "loss_g1": 17.803298950195312,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.093684196472168,
      "loss_g0": 14.062536239624023,
      "loss_g1": 15.947661399841309,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522612422704697,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 38.21878433227539,
      "loss_g0": 38.63923645019531,
      "loss_g1": 13.191776275634766,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 35.49675750732422,
      "loss_g0": 35.691139221191406,
      "loss_g1": 23.92629623413086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.98896312713623,
      "loss_g0": 0.0,
      "loss_g1": 12.98896312713623,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 30.58222007751465,
      "loss_g0": 6.46209192276001,
      "loss_g1": 30.987438201904297,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.545711517333984,
      "loss_g0": 21.634441375732422,
      "loss_g1": 14.426615715026855,
      "q_g0": 0.016523154452443123,
      "q_g1": 0.9834768176078796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.471519470214844,
      "loss_g0": 10.955653190612793,
      "loss_g1": 17.58098602294922,
      "q_g0": 0.0165224801748991,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.030082702636719,
      "loss_g0": 11.996025085449219,
      "loss_g1": 8.980253219604492,
      "q_g0": 0.016522781923413277,
      "q_g1": 0.9834771752357483,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.376567840576172,
      "loss_g0": 12.343390464782715,
      "loss_g1": 16.444324493408203,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.886468887329102,
      "loss_g0": 20.07826805114746,
      "loss_g1": 9.715234756469727,
      "q_g0": 0.01652348041534424,
      "q_g1": 0.9834765195846558,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 11.518877983093262,
      "loss_g0": 20.89809226989746,
      "loss_g1": 11.361287117004395,
      "q_g0": 0.016524456441402435,
      "q_g1": 0.9834755659103394,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.934822082519531,
      "loss_g0": 18.331871032714844,
      "loss_g1": 8.776922225952148,
      "q_g0": 0.016525432467460632,
      "q_g1": 0.9834745526313782,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.10958480834961,
      "loss_g0": 23.819210052490234,
      "loss_g1": 14.963228225708008,
      "q_g0": 0.01652633771300316,
      "q_g1": 0.9834735989570618,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.1834285259246826,
      "loss_g0": 19.909523010253906,
      "loss_g1": 2.902332305908203,
      "q_g0": 0.016528083011507988,
      "q_g1": 0.9834718704223633,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.60828685760498,
      "loss_g0": 16.61370086669922,
      "loss_g1": 8.47374153137207,
      "q_g0": 0.016528913751244545,
      "q_g1": 0.9834710955619812,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.287166118621826,
      "loss_g0": 52.42286682128906,
      "loss_g1": 3.4779138565063477,
      "q_g0": 0.016533922404050827,
      "q_g1": 0.9834660887718201,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.070261001586914,
      "loss_g0": 20.98272705078125,
      "loss_g1": 11.920415878295898,
      "q_g0": 0.016534851863980293,
      "q_g1": 0.9834651947021484,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.89731502532959,
      "loss_g0": 26.072154998779297,
      "loss_g1": 3.524439811706543,
      "q_g0": 0.016537154093384743,
      "q_g1": 0.9834628701210022,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.380859851837158,
      "loss_g0": 14.576987266540527,
      "loss_g1": 4.20939826965332,
      "q_g0": 0.016538208350539207,
      "q_g1": 0.9834617376327515,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.380578994750977,
      "loss_g0": 14.33803939819336,
      "loss_g1": 16.414928436279297,
      "q_g0": 0.016537999734282494,
      "q_g1": 0.9834619760513306,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.3478622436523438,
      "loss_g0": 13.004730224609375,
      "loss_g1": 1.1518244743347168,
      "q_g0": 0.01653921790421009,
      "q_g1": 0.9834607839584351,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 36.386959075927734,
      "loss_g0": 13.114401817321777,
      "loss_g1": 36.778282165527344,
      "q_g0": 0.01653680019080639,
      "q_g1": 0.9834632277488708,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 10.242812156677246,
      "loss_g0": 10.242812156677246,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.73513650894165,
      "loss_g0": 4.779516220092773,
      "loss_g1": 2.0934696197509766,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.716827392578125,
      "loss_g0": 22.716827392578125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 39.7883186340332,
      "loss_g0": 40.28220748901367,
      "loss_g1": 10.390064239501953,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.42503547668457,
      "loss_g0": 9.53787612915039,
      "loss_g1": 2.708347797393799,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.675739288330078,
      "loss_g0": 18.869626998901367,
      "loss_g1": 7.13477897644043,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.493703842163086,
      "loss_g0": 0.0,
      "loss_g1": 8.493703842163086,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.579648494720459,
      "loss_g0": 13.60400390625,
      "loss_g1": 4.42803955078125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.16626501083374,
      "loss_g0": 12.54653263092041,
      "loss_g1": 6.059072017669678,
      "q_g0": 0.016523092985153198,
      "q_g1": 0.9834769368171692,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.6944797039031982,
      "loss_g0": 10.409526824951172,
      "loss_g1": 2.5648550987243652,
      "q_g0": 0.01652389205992222,
      "q_g1": 0.9834761619567871,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.381960868835449,
      "loss_g0": 12.802417755126953,
      "loss_g1": 5.257280349731445,
      "q_g0": 0.016524657607078552,
      "q_g1": 0.9834753274917603,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.746052265167236,
      "loss_g0": 10.464618682861328,
      "loss_g1": 4.64996337890625,
      "q_g0": 0.016525255516171455,
      "q_g1": 0.9834747910499573,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.013057708740234,
      "loss_g0": 11.87051010131836,
      "loss_g1": 6.931435585021973,
      "q_g0": 0.016525765880942345,
      "q_g1": 0.9834741950035095,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.3581345081329346,
      "loss_g0": 9.463911056518555,
      "loss_g1": 1.2219223976135254,
      "q_g0": 0.016526611521840096,
      "q_g1": 0.9834733605384827,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.145442962646484,
      "loss_g0": 2.098828077316284,
      "loss_g1": 4.179834365844727,
      "q_g0": 0.01652640290558338,
      "q_g1": 0.9834735989570618,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.50398850440979,
      "loss_g0": 5.915653228759766,
      "loss_g1": 3.4634618759155273,
      "q_g0": 0.01652665250003338,
      "q_g1": 0.9834733605384827,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.3407864570617676,
      "loss_g0": 5.171083927154541,
      "loss_g1": 1.2764190435409546,
      "q_g0": 0.016527052968740463,
      "q_g1": 0.9834729433059692,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.2237725257873535,
      "loss_g0": 23.668182373046875,
      "loss_g1": 3.896972417831421,
      "q_g0": 0.016529075801372528,
      "q_g1": 0.9834709763526917,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 0.8933871984481812,
      "loss_g0": 10.372336387634277,
      "loss_g1": 0.7340659499168396,
      "q_g0": 0.01653006114065647,
      "q_g1": 0.9834699034690857,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.859018325805664,
      "loss_g0": 17.859018325805664,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.9151171445846558,
      "loss_g0": 1.8524773120880127,
      "loss_g1": 5.643673896789551,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.175085067749023,
      "loss_g0": 16.020889282226562,
      "loss_g1": 25.352828979492188,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523372381925583,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 22.517728805541992,
      "loss_g0": 22.464696884155273,
      "loss_g1": 25.6742000579834,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523705795407295,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 42.804725646972656,
      "loss_g0": 42.6032829284668,
      "loss_g1": 54.79359436035156,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.01652495190501213,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.496192932128906,
      "loss_g0": 21.429962158203125,
      "loss_g1": 25.437705993652344,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.01652536541223526,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 19.44108009338379,
      "loss_g0": 19.130727767944336,
      "loss_g1": 37.908912658691406,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.01652727834880352,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 53.54555892944336,
      "loss_g0": 53.905792236328125,
      "loss_g1": 32.10663604736328,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.01652505435049534,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 36.14284896850586,
      "loss_g0": 36.26876449584961,
      "loss_g1": 28.64883041381836,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652427762746811,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 40.001678466796875,
      "loss_g0": 39.58624267578125,
      "loss_g1": 64.7233657836914,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.01652684435248375,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.870627403259277,
      "loss_g0": 13.60759162902832,
      "loss_g1": 29.5217227935791,
      "q_g0": 0.9834715127944946,
      "q_g1": 0.01652846299111843,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.308147430419922,
      "loss_g0": 16.951522827148438,
      "loss_g1": 38.525028228759766,
      "q_g0": 0.9834693670272827,
      "q_g1": 0.016530673950910568,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.810625076293945,
      "loss_g0": 14.743844032287598,
      "loss_g1": 18.783634185791016,
      "q_g0": 0.9834688901901245,
      "q_g1": 0.0165310837328434,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.982272148132324,
      "loss_g0": 16.015344619750977,
      "loss_g1": 14.01465892791748,
      "q_g0": 0.9834691286087036,
      "q_g1": 0.016530882567167282,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 29.512907028198242,
      "loss_g0": 29.60993194580078,
      "loss_g1": 23.740325927734375,
      "q_g0": 0.9834696650505066,
      "q_g1": 0.01653028465807438,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 27.73748207092285,
      "loss_g0": 27.8785457611084,
      "loss_g1": 19.344379425048828,
      "q_g0": 0.983470618724823,
      "q_g1": 0.016529416665434837,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 21.871456146240234,
      "loss_g0": 21.896581649780273,
      "loss_g1": 20.37653350830078,
      "q_g0": 0.9834707379341125,
      "q_g1": 0.01652926206588745,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 12.92184066772461,
      "loss_g0": 12.939528465270996,
      "loss_g1": 11.869457244873047,
      "q_g0": 0.9834708571434021,
      "q_g1": 0.016529154032468796,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.13545036315918,
      "loss_g0": 17.141950607299805,
      "loss_g1": 16.748706817626953,
      "q_g0": 0.9834708571434021,
      "q_g1": 0.016529114916920662,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.060147285461426,
      "loss_g0": 14.06519603729248,
      "loss_g1": 13.759801864624023,
      "q_g0": 0.9834709763526917,
      "q_g1": 0.016529085114598274,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 36.83759307861328,
      "loss_g0": 36.73466110229492,
      "loss_g1": 42.961769104003906,
      "q_g0": 0.9834702610969543,
      "q_g1": 0.016529718413949013,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.908714771270752,
      "loss_g0": 7.956380844116211,
      "loss_g1": 5.072643280029297,
      "q_g0": 0.983470618724823,
      "q_g1": 0.016529425978660583,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.999641418457031,
      "loss_g0": 14.819628715515137,
      "loss_g1": 25.709333419799805,
      "q_g0": 0.9834694862365723,
      "q_g1": 0.016530539840459824,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.662334442138672,
      "loss_g0": 13.681325912475586,
      "loss_g1": 12.532466888427734,
      "q_g0": 0.9834696054458618,
      "q_g1": 0.016530416905879974,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.526447296142578,
      "loss_g0": 15.39428424835205,
      "loss_g1": 23.389022827148438,
      "q_g0": 0.983468770980835,
      "q_g1": 0.01653123088181019,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 8.293089866638184,
      "loss_g0": 8.37613296508789,
      "loss_g1": 3.352562189102173,
      "q_g0": 0.9834692478179932,
      "q_g1": 0.01653071865439415,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.460967063903809,
      "loss_g0": 13.580888748168945,
      "loss_g1": 6.3261027336120605,
      "q_g0": 0.9834700226783752,
      "q_g1": 0.016529975458979607,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.210868835449219,
      "loss_g0": 15.449665069580078,
      "loss_g1": 1.00211501121521,
      "q_g0": 0.9834715127944946,
      "q_g1": 0.016528494656085968,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.517689228057861,
      "loss_g0": 4.366480827331543,
      "loss_g1": 13.51426887512207,
      "q_g0": 0.983470618724823,
      "q_g1": 0.01652943342924118,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.38943862915039,
      "loss_g0": 17.581388473510742,
      "loss_g1": 5.968048095703125,
      "q_g0": 0.9834717512130737,
      "q_g1": 0.01652824692428112,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.105836868286133,
      "loss_g0": 9.02177619934082,
      "loss_g1": 14.107542991638184,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.016528764739632607,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.20544147491455,
      "loss_g0": 14.380104064941406,
      "loss_g1": 3.812155246734619,
      "q_g0": 0.9834723472595215,
      "q_g1": 0.016527680680155754,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.81487774848938,
      "loss_g0": 2.7177929878234863,
      "loss_g1": 8.59165096282959,
      "q_g0": 0.9834717512130737,
      "q_g1": 0.016528278589248657,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.698361396789551,
      "loss_g0": 5.5731611251831055,
      "loss_g1": 13.147697448730469,
      "q_g0": 0.9834709763526917,
      "q_g1": 0.01652904599905014,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.672119140625,
      "loss_g0": 14.828572273254395,
      "loss_g1": 5.362651348114014,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.016528083011507988,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.3073208332061768,
      "loss_g0": 3.2828941345214844,
      "loss_g1": 4.760782241821289,
      "q_g0": 0.9834717512130737,
      "q_g1": 0.016528230160474777,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.152939796447754,
      "loss_g0": 9.202980041503906,
      "loss_g1": 6.175384044647217,
      "q_g0": 0.9834721088409424,
      "q_g1": 0.016527913510799408,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.191821098327637,
      "loss_g0": 9.193071365356445,
      "loss_g1": 9.117408752441406,
      "q_g0": 0.9834721088409424,
      "q_g1": 0.01652790792286396,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.026304244995117,
      "loss_g0": 17.14156723022461,
      "loss_g1": 10.167459487915039,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.016527192667126656,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.113088607788086,
      "loss_g0": 15.202506065368652,
      "loss_g1": 9.791997909545898,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526635736227036,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.9817476272583,
      "loss_g0": 15.893619537353516,
      "loss_g1": 21.225910186767578,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.01652718335390091,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.735755443572998,
      "loss_g0": 2.735682487487793,
      "loss_g1": 2.7401061058044434,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.01652718335390091,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.363208293914795,
      "loss_g0": 4.377590179443359,
      "loss_g1": 3.507383346557617,
      "q_g0": 0.9834729433059692,
      "q_g1": 0.016527092084288597,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.395500183105469,
      "loss_g0": 6.373849391937256,
      "loss_g1": 7.6838603019714355,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.016527224332094193,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 28.789409637451172,
      "loss_g0": 29.170753479003906,
      "loss_g1": 6.093719482421875,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.01652486063539982,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 1.8178702592849731,
      "loss_g0": 1.782462239265442,
      "loss_g1": 3.9251461029052734,
      "q_g0": 0.983474850654602,
      "q_g1": 0.01652507483959198,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 13.82100772857666,
      "loss_g0": 14.0131196975708,
      "loss_g1": 2.386749744415283,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652388460934162,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.243404865264893,
      "loss_g0": 7.359737396240234,
      "loss_g1": 0.31917351484298706,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652316376566887,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 7.380957126617432,
      "loss_g0": 7.309745788574219,
      "loss_g1": 11.61942195892334,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523605212569237,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.305802345275879,
      "loss_g0": 4.330864906311035,
      "loss_g1": 2.814068555831909,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.0165234487503767,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.3566765785217285,
      "loss_g0": 3.34277606010437,
      "loss_g1": 4.184016227722168,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523534432053566,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 4.176124572753906,
      "loss_g0": 4.221438884735107,
      "loss_g1": 1.478973150253296,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652325503528118,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 29.89394760131836,
      "loss_g0": 0.0,
      "loss_g1": 29.89394760131836,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 24.58864974975586,
      "loss_g0": 29.972597122192383,
      "loss_g1": 24.498199462890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 15.756733894348145,
      "loss_g0": 47.2419548034668,
      "loss_g1": 15.227675437927246,
      "q_g0": 0.016525696963071823,
      "q_g1": 0.9834743142127991,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 28.581411361694336,
      "loss_g0": 32.02040100097656,
      "loss_g1": 28.523622512817383,
      "q_g0": 0.016526052728295326,
      "q_g1": 0.9834739565849304,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 14.405667304992676,
      "loss_g0": 5.446713924407959,
      "loss_g1": 14.55620288848877,
      "q_g0": 0.01652512326836586,
      "q_g1": 0.983474850654602,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 18.251203536987305,
      "loss_g0": 25.1136531829834,
      "loss_g1": 18.135889053344727,
      "q_g0": 0.016525836661458015,
      "q_g1": 0.9834741950035095,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 32.37386703491211,
      "loss_g0": 46.767555236816406,
      "loss_g1": 32.131980895996094,
      "q_g0": 0.016527332365512848,
      "q_g1": 0.9834727048873901,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.239213466644287,
      "loss_g0": 4.637740612030029,
      "loss_g1": 6.2661261558532715,
      "q_g0": 0.016527162864804268,
      "q_g1": 0.9834728240966797,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 17.64863395690918,
      "loss_g0": 0.0,
      "loss_g1": 17.64863395690918,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 9.405420303344727,
      "loss_g0": 7.5852484703063965,
      "loss_g1": 9.435998916625977,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.6944150924682617,
      "loss_g0": 11.545351028442383,
      "loss_g1": 3.5625128746032715,
      "q_g0": 0.016523240134119987,
      "q_g1": 0.9834766983985901,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 3.7133307456970215,
      "loss_g0": 16.888622283935547,
      "loss_g1": 3.4919559955596924,
      "q_g0": 0.01652461104094982,
      "q_g1": 0.9834753274917603,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 5.6432695388793945,
      "loss_g0": 13.054839134216309,
      "loss_g1": 5.518732070922852,
      "q_g0": 0.016525380313396454,
      "q_g1": 0.9834746718406677,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.0404434204101562,
      "loss_g0": 9.556561470031738,
      "loss_g1": 1.9141438007354736,
      "q_g0": 0.016526153311133385,
      "q_g1": 0.9834738373756409,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.121533393859863,
      "loss_g0": 7.190752983093262,
      "loss_g1": 6.1035661697387695,
      "q_g0": 0.01652626134455204,
      "q_g1": 0.9834737181663513,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.308200836181641,
      "loss_g0": 10.14738655090332,
      "loss_g1": 6.243685722351074,
      "q_g0": 0.016526658087968826,
      "q_g1": 0.9834733605384827,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 2.9754161834716797,
      "loss_g0": 2.9754161834716797,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.971625328063965,
      "loss_g0": 7.0541090965271,
      "loss_g1": 2.0618762969970703,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 6.5014238357543945,
      "loss_g0": 6.55919885635376,
      "loss_g1": 3.0624172687530518,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "eval_cer": 0.06486138729675986,
      "eval_loss": 14.666666030883789,
      "eval_runtime": 81.192,
      "eval_samples_per_second": 20.642,
      "eval_steps_per_second": 5.161,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 16.876937866210938,
      "loss_g0": 16.2907657623291,
      "loss_g1": 51.760318756103516,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.016526052728295326,
      "step": 500
    },
    {
      "epoch": 2.5380710659898478,
      "gdro_loss": 31.09197425842285,
      "loss_g0": 30.35103988647461,
      "loss_g1": 75.1729507446289,
      "q_g0": 0.9834693670272827,
      "q_g1": 0.016530640423297882,
      "step": 500
    },
    {
      "epoch": 3.045685279187817,
      "grad_norm": 96.63529968261719,
      "learning_rate": 0.0003399572192513369,
      "loss": 25.4896,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.912988662719727,
      "loss_g0": 19.958908081054688,
      "loss_g1": 17.19029998779297,
      "q_g0": 0.9834138751029968,
      "q_g1": 0.01658605970442295,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.859804153442383,
      "loss_g0": 19.954364776611328,
      "loss_g1": 14.252876281738281,
      "q_g0": 0.9834144711494446,
      "q_g1": 0.016585474833846092,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 23.78574562072754,
      "loss_g0": 23.76508140563965,
      "loss_g1": 25.010944366455078,
      "q_g0": 0.983414351940155,
      "q_g1": 0.016585607081651688,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.616106986999512,
      "loss_g0": 11.460838317871094,
      "loss_g1": 20.821910858154297,
      "q_g0": 0.9834134578704834,
      "q_g1": 0.01658657379448414,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.258782386779785,
      "loss_g0": 12.055496215820312,
      "loss_g1": 24.310653686523438,
      "q_g0": 0.9834121465682983,
      "q_g1": 0.01658783294260502,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.639178276062012,
      "loss_g0": 15.619487762451172,
      "loss_g1": 16.806533813476562,
      "q_g0": 0.9834120273590088,
      "q_g1": 0.016587959602475166,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.473980903625488,
      "loss_g0": 13.53177261352539,
      "loss_g1": 10.047760009765625,
      "q_g0": 0.9834123849868774,
      "q_g1": 0.01658760756254196,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.222042083740234,
      "loss_g0": 11.194822311401367,
      "loss_g1": 12.835739135742188,
      "q_g0": 0.9834122657775879,
      "q_g1": 0.016587771475315094,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 25.149253845214844,
      "loss_g0": 25.44104766845703,
      "loss_g1": 7.848247528076172,
      "q_g0": 0.9834139943122864,
      "q_g1": 0.01658596657216549,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.506771087646484,
      "loss_g0": 0.0,
      "loss_g1": 16.506771087646484,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 24.307926177978516,
      "loss_g0": 20.9424991607666,
      "loss_g1": 24.364463806152344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.926909446716309,
      "loss_g0": 33.45008087158203,
      "loss_g1": 14.615682601928711,
      "q_g0": 0.01652434840798378,
      "q_g1": 0.9834756851196289,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.39899444580078,
      "loss_g0": 11.279593467712402,
      "loss_g1": 18.51860809326172,
      "q_g0": 0.016523605212569237,
      "q_g1": 0.9834764003753662,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.745759010314941,
      "loss_g0": 26.763362884521484,
      "loss_g1": 10.476615905761719,
      "q_g0": 0.0165252722799778,
      "q_g1": 0.9834747910499573,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.078306198120117,
      "loss_g0": 19.102493286132812,
      "loss_g1": 21.11150550842285,
      "q_g0": 0.016525069251656532,
      "q_g1": 0.983474850654602,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.991037368774414,
      "loss_g0": 5.44220495223999,
      "loss_g1": 18.201873779296875,
      "q_g0": 0.01652376726269722,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.72547721862793,
      "loss_g0": 7.907643795013428,
      "loss_g1": 8.739216804504395,
      "q_g0": 0.016523681581020355,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.230852127075195,
      "loss_g0": 20.264202117919922,
      "loss_g1": 20.23029136657715,
      "q_g0": 0.016523681581020355,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.854703903198242,
      "loss_g0": 13.699477195739746,
      "loss_g1": 5.722894668579102,
      "q_g0": 0.016524503007531166,
      "q_g1": 0.9834754467010498,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.029624938964844,
      "loss_g0": 12.713655471801758,
      "loss_g1": 7.950920104980469,
      "q_g0": 0.016524991020560265,
      "q_g1": 0.9834749698638916,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.607161521911621,
      "loss_g0": 5.794318675994873,
      "loss_g1": 7.637621879577637,
      "q_g0": 0.016524799168109894,
      "q_g1": 0.9834752082824707,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.8731689453125,
      "loss_g0": 17.8731689453125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.214249610900879,
      "loss_g0": 9.273048400878906,
      "loss_g1": 5.714349746704102,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.874638557434082,
      "loss_g0": 14.033642768859863,
      "loss_g1": 4.410113334655762,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.914416313171387,
      "loss_g0": 8.812891006469727,
      "loss_g1": 14.957355499267578,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523046419024467,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.066764831542969,
      "loss_g0": 7.946876525878906,
      "loss_g1": 15.202367782592773,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652379147708416,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.50073528289795,
      "loss_g0": 12.611262321472168,
      "loss_g1": 5.922065734863281,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523106023669243,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.884082794189453,
      "loss_g0": 6.800113201141357,
      "loss_g1": 11.881901741027832,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523627564311028,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.330450534820557,
      "loss_g0": 6.332394123077393,
      "loss_g1": 6.214748382568359,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523612663149834,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.204771995544434,
      "loss_g0": 13.263559341430664,
      "loss_g1": 9.705730438232422,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523247584700584,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.0786075592041,
      "loss_g0": 17.313533782958984,
      "loss_g1": 3.094832181930542,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.564774990081787,
      "loss_g0": 7.524626731872559,
      "loss_g1": 9.954503059387207,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522666439414024,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.187137603759766,
      "loss_g0": 13.350641250610352,
      "loss_g1": 3.454796552658081,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.687503814697266,
      "loss_g0": 6.649072170257568,
      "loss_g1": 8.975082397460938,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522657126188278,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.557510375976562,
      "loss_g0": 16.36073875427246,
      "loss_g1": 28.26904296875,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523877158761024,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.552616596221924,
      "loss_g0": 4.552616596221924,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.045011520385742,
      "loss_g0": 16.213977813720703,
      "loss_g1": 5.9874749183654785,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.7440366744995117,
      "loss_g0": 3.6923680305480957,
      "loss_g1": 6.819484233856201,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522744670510292,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.39447021484375,
      "loss_g0": 11.379240036010742,
      "loss_g1": 12.300990104675293,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522839665412903,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.63776683807373,
      "loss_g0": 10.758171081542969,
      "loss_g1": 3.4708268642425537,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 36.16434097290039,
      "loss_g0": 36.70168685913086,
      "loss_g1": 4.179388046264648,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.288388252258301,
      "loss_g0": 6.134705066680908,
      "loss_g1": 15.435650825500488,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652337983250618,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.737351655960083,
      "loss_g0": 3.762660026550293,
      "loss_g1": 2.230987548828125,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523215919733047,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.25625228881836,
      "loss_g0": 8.316816329956055,
      "loss_g1": 4.651291847229004,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652284525334835,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.696289539337158,
      "loss_g0": 5.761882781982422,
      "loss_g1": 1.7919350862503052,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.989555358886719,
      "loss_g0": 10.989555358886719,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.683488845825195,
      "loss_g0": 13.776802062988281,
      "loss_g1": 8.12911605834961,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.2824153900146484,
      "loss_g0": 3.323601722717285,
      "loss_g1": 0.8308506608009338,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.8233323097229,
      "loss_g0": 7.765443801879883,
      "loss_g1": 11.268978118896484,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.386462926864624,
      "loss_g0": 2.376880645751953,
      "loss_g1": 2.9568192958831787,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652284525334835,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 49.01200866699219,
      "loss_g0": 49.52556228637695,
      "loss_g1": 18.44332504272461,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.594467163085938,
      "loss_g0": 16.322444915771484,
      "loss_g1": 32.784488677978516,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524100676178932,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.960717678070068,
      "loss_g0": 4.317864894866943,
      "loss_g1": 43.21245574951172,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.01652807556092739,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.170372009277344,
      "loss_g0": 19.170372009277344,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 22.98398208618164,
      "loss_g0": 23.00098419189453,
      "loss_g1": 21.971954345703125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.192670822143555,
      "loss_g0": 19.406108856201172,
      "loss_g1": 6.488000392913818,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.209891319274902,
      "loss_g0": 13.176300048828125,
      "loss_g1": 15.209291458129883,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.1042423248291,
      "loss_g0": 17.946529388427734,
      "loss_g1": 27.491209030151367,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523612663149834,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.689714431762695,
      "loss_g0": 18.662551879882812,
      "loss_g1": 20.306427001953125,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523784026503563,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.163114547729492,
      "loss_g0": 17.263168334960938,
      "loss_g1": 11.207828521728516,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523171216249466,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.64625358581543,
      "loss_g0": 8.468583106994629,
      "loss_g1": 19.22071075439453,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524270176887512,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 24.17009925842285,
      "loss_g0": 24.4635066986084,
      "loss_g1": 6.705386638641357,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522448509931564,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.563060760498047,
      "loss_g0": 16.750354766845703,
      "loss_g1": 5.4146270751953125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.830318450927734,
      "loss_g0": 4.7366251945495605,
      "loss_g1": 10.40711498260498,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016522999852895737,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.440837860107422,
      "loss_g0": 0.0,
      "loss_g1": 17.440837860107422,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.247888565063477,
      "loss_g0": 17.878643035888672,
      "loss_g1": 11.136491775512695,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.8737568855285645,
      "loss_g0": 36.0816650390625,
      "loss_g1": 6.382972717285156,
      "q_g0": 0.016525456681847572,
      "q_g1": 0.9834745526313782,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.37289810180664,
      "loss_g0": 22.121593475341797,
      "loss_g1": 14.24268913269043,
      "q_g0": 0.01652626134455204,
      "q_g1": 0.9834737181663513,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.51516056060791,
      "loss_g0": 40.17058181762695,
      "loss_g1": 11.033548355102539,
      "q_g0": 0.01652923785150051,
      "q_g1": 0.9834707379341125,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.83759117126465,
      "loss_g0": 8.540971755981445,
      "loss_g1": 20.02743911743164,
      "q_g0": 0.016528068110346794,
      "q_g1": 0.9834718704223633,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 24.321622848510742,
      "loss_g0": 15.658526420593262,
      "loss_g1": 24.467205047607422,
      "q_g0": 0.016527168452739716,
      "q_g1": 0.9834728240966797,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.456876754760742,
      "loss_g0": 15.321582794189453,
      "loss_g1": 19.526369094848633,
      "q_g0": 0.016526736319065094,
      "q_g1": 0.9834732413291931,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.988515853881836,
      "loss_g0": 8.875215530395508,
      "loss_g1": 18.141651153564453,
      "q_g0": 0.01652578078210354,
      "q_g1": 0.9834741950035095,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.573808670043945,
      "loss_g0": 28.996917724609375,
      "loss_g1": 12.29781436920166,
      "q_g0": 0.016527486965060234,
      "q_g1": 0.983472466468811,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.531431674957275,
      "loss_g0": 21.415996551513672,
      "loss_g1": 5.264460563659668,
      "q_g0": 0.0165291465818882,
      "q_g1": 0.9834708571434021,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.497014999389648,
      "loss_g0": 5.5774431228637695,
      "loss_g1": 6.51246976852417,
      "q_g0": 0.016529053449630737,
      "q_g1": 0.9834709763526917,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.495288848876953,
      "loss_g0": 17.95853042602539,
      "loss_g1": 9.35304069519043,
      "q_g0": 0.016529928892850876,
      "q_g1": 0.9834700226783752,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.644648551940918,
      "loss_g0": 9.82818603515625,
      "loss_g1": 15.742405891418457,
      "q_g0": 0.016529330983757973,
      "q_g1": 0.983470618724823,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.219154357910156,
      "loss_g0": 16.30371856689453,
      "loss_g1": 19.268152236938477,
      "q_g0": 0.016529029235243797,
      "q_g1": 0.9834709763526917,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.564004898071289,
      "loss_g0": 19.665672302246094,
      "loss_g1": 10.411026000976562,
      "q_g0": 0.01652996800839901,
      "q_g1": 0.9834700226783752,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.480496883392334,
      "loss_g0": 10.307004928588867,
      "loss_g1": 6.416179656982422,
      "q_g0": 0.016530362889170647,
      "q_g1": 0.9834696650505066,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.448914051055908,
      "loss_g0": 4.676194667816162,
      "loss_g1": 4.445094108581543,
      "q_g0": 0.016530394554138184,
      "q_g1": 0.9834696054458618,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.760632514953613,
      "loss_g0": 12.41092300415039,
      "loss_g1": 7.682466983795166,
      "q_g0": 0.016530882567167282,
      "q_g1": 0.9834691286087036,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.567995071411133,
      "loss_g0": 17.567995071411133,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.341392517089844,
      "loss_g0": 0.0,
      "loss_g1": 8.341392517089844,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.017971038818359,
      "loss_g0": 2.0986123085021973,
      "loss_g1": 4.050216197967529,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.681674480438232,
      "loss_g0": 15.475069046020508,
      "loss_g1": 6.533937454223633,
      "q_g0": 0.01652333326637745,
      "q_g1": 0.9834766387939453,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.777318000793457,
      "loss_g0": 10.827707290649414,
      "loss_g1": 6.70926570892334,
      "q_g0": 0.016523759812116623,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.9326286315917969,
      "loss_g0": 11.527654647827148,
      "loss_g1": 1.7714089155197144,
      "q_g0": 0.01652476005256176,
      "q_g1": 0.9834752082824707,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.798154354095459,
      "loss_g0": 27.18306541442871,
      "loss_g1": 2.3883650302886963,
      "q_g0": 0.016527298837900162,
      "q_g1": 0.9834727048873901,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.166605472564697,
      "loss_g0": 1.0895522832870483,
      "loss_g1": 4.2183146476745605,
      "q_g0": 0.016526974737644196,
      "q_g1": 0.9834730625152588,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.059434413909912,
      "loss_g0": 6.867542266845703,
      "loss_g1": 4.01224422454834,
      "q_g0": 0.016527269035577774,
      "q_g1": 0.9834727048873901,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.170597076416016,
      "loss_g0": 0.5254770517349243,
      "loss_g1": 4.231851577758789,
      "q_g0": 0.01652689091861248,
      "q_g1": 0.9834731221199036,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.3964660167694092,
      "loss_g0": 10.351293563842773,
      "loss_g1": 1.2459747791290283,
      "q_g0": 0.016527827829122543,
      "q_g1": 0.9834722280502319,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.690121650695801,
      "loss_g0": 3.419189453125,
      "loss_g1": 3.6946747303009033,
      "q_g0": 0.016527798026800156,
      "q_g1": 0.9834722280502319,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.197146892547607,
      "loss_g0": 0.0,
      "loss_g1": 4.197146892547607,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.616218566894531,
      "loss_g0": 12.982948303222656,
      "loss_g1": 5.492457389831543,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.0771431922912598,
      "loss_g0": 7.385009765625,
      "loss_g1": 3.0047688484191895,
      "q_g0": 0.01652287133038044,
      "q_g1": 0.9834771752357483,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.869961261749268,
      "loss_g0": 4.869961261749268,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.014589309692383,
      "loss_g0": 19.224903106689453,
      "loss_g1": 6.495936870574951,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 60.9935302734375,
      "loss_g0": 61.44397735595703,
      "loss_g1": 34.181095123291016,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.713115692138672,
      "loss_g0": 18.4284725189209,
      "loss_g1": 35.654361724853516,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652417704463005,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 28.44798469543457,
      "loss_g0": 28.13560676574707,
      "loss_g1": 47.03770065307617,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.016526106745004654,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 34.65373992919922,
      "loss_g0": 34.95372772216797,
      "loss_g1": 16.799213409423828,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652425527572632,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 28.003070831298828,
      "loss_g0": 28.078596115112305,
      "loss_g1": 23.508014678955078,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652379147708416,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 36.33892059326172,
      "loss_g0": 36.33892059326172,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.631821632385254,
      "loss_g0": 13.344032287597656,
      "loss_g1": 30.76209259033203,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.691208839416504,
      "loss_g0": 9.557024955749512,
      "loss_g1": 17.677947998046875,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652325503528118,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 76.45559692382812,
      "loss_g0": 77.400390625,
      "loss_g1": 20.217575073242188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 31.722373962402344,
      "loss_g0": 31.722373962402344,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.669282913208008,
      "loss_g0": 21.6025390625,
      "loss_g1": 25.642114639282227,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.71140193939209,
      "loss_g0": 13.745084762573242,
      "loss_g1": 11.706486701965332,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.1209492683410645,
      "loss_g0": 5.840334892272949,
      "loss_g1": 22.822399139404297,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524160280823708,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.586130142211914,
      "loss_g0": 13.72372817993164,
      "loss_g1": 5.396185398101807,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523310914635658,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.22769546508789,
      "loss_g0": 15.047342300415039,
      "loss_g1": 25.96167755126953,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524426639080048,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.114920616149902,
      "loss_g0": 6.95844030380249,
      "loss_g1": 16.42755126953125,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.016525395214557648,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 22.931991577148438,
      "loss_g0": 23.165313720703125,
      "loss_g1": 9.044995307922363,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523944213986397,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.059125900268555,
      "loss_g0": 15.199627876281738,
      "loss_g1": 6.696247100830078,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523070633411407,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.743268013000488,
      "loss_g0": 9.835140228271484,
      "loss_g1": 4.274682521820068,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 25.88678741455078,
      "loss_g0": 26.2240047454834,
      "loss_g1": 5.81419563293457,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.501742839813232,
      "loss_g0": 6.467260360717773,
      "loss_g1": 8.554238319396973,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.928779602050781,
      "loss_g0": 5.896448135375977,
      "loss_g1": 7.853219509124756,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522839665412903,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.629911422729492,
      "loss_g0": 20.350006103515625,
      "loss_g1": 37.28873825073242,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524571925401688,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.350651741027832,
      "loss_g0": 5.320531845092773,
      "loss_g1": 7.14321231842041,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652476005256176,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.369447708129883,
      "loss_g0": 11.481681823730469,
      "loss_g1": 4.689511775970459,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524069011211395,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.627887725830078,
      "loss_g0": 10.713348388671875,
      "loss_g1": 5.5413007736206055,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523543745279312,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.946330070495605,
      "loss_g0": 8.906804084777832,
      "loss_g1": 11.298921585083008,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523784026503563,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.974833965301514,
      "loss_g0": 7.044084548950195,
      "loss_g1": 2.853018283843994,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652335561811924,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.470159530639648,
      "loss_g0": 13.667655944824219,
      "loss_g1": 1.7144019603729248,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.092642784118652,
      "loss_g0": 14.657984733581543,
      "loss_g1": 40.96086120605469,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525106504559517,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 30.76544952392578,
      "loss_g0": 31.204410552978516,
      "loss_g1": 4.636770725250244,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.079691410064697,
      "loss_g0": 6.09420108795166,
      "loss_g1": 5.216008186340332,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.3571486473083496,
      "loss_g0": 2.309041738510132,
      "loss_g1": 5.22059440612793,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.0165227223187685,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.77804708480835,
      "loss_g0": 0.0,
      "loss_g1": 7.77804708480835,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.381749153137207,
      "loss_g0": 14.651778221130371,
      "loss_g1": 11.326812744140625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.720459938049316,
      "loss_g0": 30.37053680419922,
      "loss_g1": 12.423904418945312,
      "q_g0": 0.016524260863661766,
      "q_g1": 0.9834756851196289,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.855518817901611,
      "loss_g0": 12.513544082641602,
      "loss_g1": 6.7604498863220215,
      "q_g0": 0.016524843871593475,
      "q_g1": 0.9834750890731812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.843216896057129,
      "loss_g0": 9.65378189086914,
      "loss_g1": 9.846399307250977,
      "q_g0": 0.016524821519851685,
      "q_g1": 0.9834752082824707,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.442102909088135,
      "loss_g0": 27.373193740844727,
      "loss_g1": 6.090361595153809,
      "q_g0": 0.01652699150145054,
      "q_g1": 0.9834730625152588,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.1487438678741455,
      "loss_g0": 12.643824577331543,
      "loss_g1": 2.9891719818115234,
      "q_g0": 0.01652798242866993,
      "q_g1": 0.9834719896316528,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 39.4816780090332,
      "loss_g0": 9.129932403564453,
      "loss_g1": 39.991661071777344,
      "q_g0": 0.01652483083307743,
      "q_g1": 0.9834752082824707,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.6766650676727295,
      "loss_g0": 11.144781112670898,
      "loss_g1": 2.534371852874756,
      "q_g0": 0.016525711864233017,
      "q_g1": 0.9834743142127991,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.059348106384277,
      "loss_g0": 8.059348106384277,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.487186431884766,
      "loss_g0": 11.508419036865234,
      "loss_g1": 10.223298072814941,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.618892669677734,
      "loss_g0": 4.674264907836914,
      "loss_g1": 1.3229241371154785,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.802867889404297,
      "loss_g0": 5.802867889404297,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.4914093017578125,
      "loss_g0": 4.494381427764893,
      "loss_g1": 4.314483642578125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 27.587379455566406,
      "loss_g0": 27.996463775634766,
      "loss_g1": 3.237001657485962,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.7806552052497864,
      "loss_g0": 0.7320554256439209,
      "loss_g1": 3.673445224761963,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.0165227223187685,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.4019851684570312,
      "loss_g0": 2.2044005393981934,
      "loss_g1": 14.16187572479248,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523944213986397,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 34.574649810791016,
      "loss_g0": 0.0,
      "loss_g1": 34.574649810791016,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.973285675048828,
      "loss_g0": 20.649259567260742,
      "loss_g1": 16.911529541015625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 25.198644638061523,
      "loss_g0": 25.978057861328125,
      "loss_g1": 25.185550689697266,
      "q_g0": 0.01652250438928604,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.719892501831055,
      "loss_g0": 27.66523551940918,
      "loss_g1": 12.468786239624023,
      "q_g0": 0.0165240541100502,
      "q_g1": 0.983475923538208,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 55.08478546142578,
      "loss_g0": 34.2446174621582,
      "loss_g1": 55.434898376464844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 37.4616813659668,
      "loss_g0": 16.498065948486328,
      "loss_g1": 37.81386947631836,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 25.812637329101562,
      "loss_g0": 31.650585174560547,
      "loss_g1": 25.714553833007812,
      "q_g0": 0.016523031517863274,
      "q_g1": 0.9834769368171692,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.140933990478516,
      "loss_g0": 11.140933990478516,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.407363891601562,
      "loss_g0": 16.45346450805664,
      "loss_g1": 13.663219451904297,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.168383598327637,
      "loss_g0": 0.0,
      "loss_g1": 15.168383598327637,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.938181400299072,
      "loss_g0": 31.462669372558594,
      "loss_g1": 7.542969703674316,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 37.89182662963867,
      "loss_g0": 17.231300354003906,
      "loss_g1": 38.238922119140625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.3665285110473633,
      "loss_g0": 19.675739288330078,
      "loss_g1": 3.0925047397613525,
      "q_g0": 0.016524124890565872,
      "q_g1": 0.983475923538208,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.604336738586426,
      "loss_g0": 9.314057350158691,
      "loss_g1": 11.642816543579102,
      "q_g0": 0.01652388460934162,
      "q_g1": 0.9834761619567871,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.861431121826172,
      "loss_g0": 22.00813865661621,
      "loss_g1": 11.690939903259277,
      "q_g0": 0.016524938866496086,
      "q_g1": 0.9834750890731812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.0197300910949707,
      "loss_g0": 15.7617826461792,
      "loss_g1": 1.7888076305389404,
      "q_g0": 0.0165263619273901,
      "q_g1": 0.9834735989570618,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.13548755645752,
      "loss_g0": 15.719134330749512,
      "loss_g1": 13.092070579528809,
      "q_g0": 0.01652662642300129,
      "q_g1": 0.9834733605384827,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.038594245910645,
      "loss_g0": 21.207353591918945,
      "loss_g1": 13.918122291564941,
      "q_g0": 0.01652737893164158,
      "q_g1": 0.9834725856781006,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.986693382263184,
      "loss_g0": 13.195541381835938,
      "loss_g1": 8.915960311889648,
      "q_g0": 0.016527820378541946,
      "q_g1": 0.9834722280502319,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 23.267898559570312,
      "loss_g0": 58.47312545776367,
      "loss_g1": 22.67612075805664,
      "q_g0": 0.016531480476260185,
      "q_g1": 0.9834685325622559,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.607126235961914,
      "loss_g0": 16.488426208496094,
      "loss_g1": 21.693164825439453,
      "q_g0": 0.016530953347682953,
      "q_g1": 0.9834690093994141,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.932668209075928,
      "loss_g0": 16.549837112426758,
      "loss_g1": 4.737382888793945,
      "q_g0": 0.016532152891159058,
      "q_g1": 0.9834678173065186,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.310964584350586,
      "loss_g0": 8.86257266998291,
      "loss_g1": 14.402549743652344,
      "q_g0": 0.01653158664703369,
      "q_g1": 0.9834684133529663,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.075906276702881,
      "loss_g0": 4.4410858154296875,
      "loss_g1": 7.120194911956787,
      "q_g0": 0.016531316563487053,
      "q_g1": 0.9834686517715454,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.711898803710938,
      "loss_g0": 26.271800994873047,
      "loss_g1": 10.450323104858398,
      "q_g0": 0.01653294637799263,
      "q_g1": 0.9834670424461365,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.7107715606689453,
      "loss_g0": 7.788122177124023,
      "loss_g1": 2.6254141330718994,
      "q_g0": 0.016533473506569862,
      "q_g1": 0.9834665656089783,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.251914978027344,
      "loss_g0": 0.0,
      "loss_g1": 7.251914978027344,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.989524841308594,
      "loss_g0": 38.518314361572266,
      "loss_g1": 8.493441581726074,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.442196369171143,
      "loss_g0": 6.054708480834961,
      "loss_g1": 7.465506076812744,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.930348873138428,
      "loss_g0": 9.725130081176758,
      "loss_g1": 7.900196075439453,
      "q_g0": 0.0165226049721241,
      "q_g1": 0.9834774136543274,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.972991943359375,
      "loss_g0": 10.617874145507812,
      "loss_g1": 17.079757690429688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.246209144592285,
      "loss_g0": 20.693395614624023,
      "loss_g1": 4.98667049407959,
      "q_g0": 0.016524028033018112,
      "q_g1": 0.983475923538208,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.488580703735352,
      "loss_g0": 3.7739615440368652,
      "loss_g1": 8.567792892456055,
      "q_g0": 0.016523543745279312,
      "q_g1": 0.9834765195846558,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.813268184661865,
      "loss_g0": 5.410809516906738,
      "loss_g1": 6.8368306159973145,
      "q_g0": 0.016523394733667374,
      "q_g1": 0.9834766387939453,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.992874145507812,
      "loss_g0": 9.951847076416016,
      "loss_g1": 12.027164459228516,
      "q_g0": 0.016523178666830063,
      "q_g1": 0.9834768176078796,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.49130630493164,
      "loss_g0": 0.0,
      "loss_g1": 10.49130630493164,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.318264961242676,
      "loss_g0": 6.973600387573242,
      "loss_g1": 10.374454498291016,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.54897928237915,
      "loss_g0": 7.301480293273926,
      "loss_g1": 6.536336898803711,
      "q_g0": 0.016522496938705444,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.033575057983398,
      "loss_g0": 11.413568496704102,
      "loss_g1": 12.043991088867188,
      "q_g0": 0.01652243360877037,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.477473258972168,
      "loss_g0": 4.900318622589111,
      "loss_g1": 2.4367687702178955,
      "q_g0": 0.016522681340575218,
      "q_g1": 0.9834772944450378,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.803483963012695,
      "loss_g0": 0.0,
      "loss_g1": 5.803483963012695,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.9407315254211426,
      "loss_g0": 33.68124771118164,
      "loss_g1": 0.39069080352783203,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.030265808105469,
      "loss_g0": 0.0,
      "loss_g1": 6.030265808105469,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.4799370765686035,
      "loss_g0": 9.63294506072998,
      "loss_g1": 4.39336633682251,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 23.025665283203125,
      "loss_g0": 5.019229412078857,
      "loss_g1": 23.32817268371582,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.506181716918945,
      "loss_g0": 8.506181716918945,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.833319187164307,
      "loss_g0": 6.9241766929626465,
      "loss_g1": 1.4251316785812378,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.829728841781616,
      "loss_g0": 3.8844268321990967,
      "loss_g1": 0.5738969445228577,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.516160488128662,
      "loss_g0": 3.5471720695495605,
      "loss_g1": 1.6702260971069336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.456927299499512,
      "loss_g0": 7.53241491317749,
      "loss_g1": 2.9636073112487793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 56.49298858642578,
      "loss_g0": 57.036930084228516,
      "loss_g1": 24.115463256835938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 22.517826080322266,
      "loss_g0": 22.6795654296875,
      "loss_g1": 12.890544891357422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 44.24797058105469,
      "loss_g0": 44.24797058105469,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 26.894926071166992,
      "loss_g0": 26.79561996459961,
      "loss_g1": 32.80600357055664,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 45.02902603149414,
      "loss_g0": 45.25658416748047,
      "loss_g1": 31.483985900878906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 35.98219299316406,
      "loss_g0": 0.0,
      "loss_g1": 35.98219299316406,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.098663330078125,
      "loss_g0": 16.03177261352539,
      "loss_g1": 21.183788299560547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.10303497314453,
      "loss_g0": 16.10303497314453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 33.998130798339844,
      "loss_g0": 34.25288391113281,
      "loss_g1": 18.834232330322266,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 38.83997344970703,
      "loss_g0": 38.99974822998047,
      "loss_g1": 29.329631805419922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 35.0931510925293,
      "loss_g0": 35.46104431152344,
      "loss_g1": 13.194796562194824,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 69.88701629638672,
      "loss_g0": 70.69041442871094,
      "loss_g1": 22.065471649169922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 32.508331298828125,
      "loss_g0": 32.8404426574707,
      "loss_g1": 12.739672660827637,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.6878662109375,
      "loss_g0": 19.744117736816406,
      "loss_g1": 16.339452743530273,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 29.39791488647461,
      "loss_g0": 29.707965850830078,
      "loss_g1": 10.942495346069336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 29.516605377197266,
      "loss_g0": 29.32898712158203,
      "loss_g1": 40.68357467651367,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523588448762894,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.492178440093994,
      "loss_g0": 7.288524627685547,
      "loss_g1": 19.612613677978516,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.01652485318481922,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.219566345214844,
      "loss_g0": 18.202224731445312,
      "loss_g1": 19.25160026550293,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016524961218237877,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.160539627075195,
      "loss_g0": 21.276710510253906,
      "loss_g1": 14.246258735656738,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524240374565125,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 41.37006759643555,
      "loss_g0": 41.92848587036133,
      "loss_g1": 8.130874633789062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.678027153015137,
      "loss_g0": 0.0,
      "loss_g1": 11.678027153015137,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.446881294250488,
      "loss_g0": 22.543458938598633,
      "loss_g1": 10.243658065795898,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.088340759277344,
      "loss_g0": 0.0,
      "loss_g1": 20.088340759277344,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.54752254486084,
      "loss_g0": 0.0,
      "loss_g1": 6.54752254486084,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.990791320800781,
      "loss_g0": 7.990791320800781,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.736095428466797,
      "loss_g0": 16.880901336669922,
      "loss_g1": 8.116707801818848,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.235053062438965,
      "loss_g0": 10.369951248168945,
      "loss_g1": 2.2053475379943848,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.74671745300293,
      "loss_g0": 16.79056167602539,
      "loss_g1": 14.136919021606445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 39.384090423583984,
      "loss_g0": 39.5540771484375,
      "loss_g1": 29.26593589782715,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 29.04002571105957,
      "loss_g0": 29.376461029052734,
      "loss_g1": 9.014110565185547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.067633628845215,
      "loss_g0": 13.111589431762695,
      "loss_g1": 10.451180458068848,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.471429824829102,
      "loss_g0": 11.593742370605469,
      "loss_g1": 4.190944671630859,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.6773974895477295,
      "loss_g0": 2.628357410430908,
      "loss_g1": 5.596382141113281,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522729769349098,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.075233459472656,
      "loss_g0": 6.7979254722595215,
      "loss_g1": 23.579620361328125,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652444154024124,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.4615461826324463,
      "loss_g0": 1.3270493745803833,
      "loss_g1": 9.465909957885742,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.0165252722799778,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.599862098693848,
      "loss_g0": 0.0,
      "loss_g1": 6.599862098693848,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.96055793762207,
      "loss_g0": 7.896390914916992,
      "loss_g1": 15.079236030578613,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.44083309173584,
      "loss_g0": 16.16223907470703,
      "loss_g1": 9.327908515930176,
      "q_g0": 0.016523122787475586,
      "q_g1": 0.9834768176078796,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.663263320922852,
      "loss_g0": 6.250373363494873,
      "loss_g1": 7.687000274658203,
      "q_g0": 0.016522977501153946,
      "q_g1": 0.9834770560264587,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.378140449523926,
      "loss_g0": 11.571033477783203,
      "loss_g1": 5.274091720581055,
      "q_g0": 0.01652361825108528,
      "q_g1": 0.9834764003753662,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.0148427486419678,
      "loss_g0": 7.226709365844727,
      "loss_g1": 1.9272739887237549,
      "q_g0": 0.016524160280823708,
      "q_g1": 0.9834758043289185,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.12706184387207,
      "loss_g0": 8.995209693908691,
      "loss_g1": 6.07887077331543,
      "q_g0": 0.016524463891983032,
      "q_g1": 0.9834755659103394,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.330540180206299,
      "loss_g0": 6.639147758483887,
      "loss_g1": 4.291750431060791,
      "q_g0": 0.016524706035852432,
      "q_g1": 0.9834753274917603,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.554842472076416,
      "loss_g0": 13.45102596282959,
      "loss_g1": 6.43896484375,
      "q_g0": 0.01652541570365429,
      "q_g1": 0.9834745526313782,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.0661885738372803,
      "loss_g0": 1.878038763999939,
      "loss_g1": 2.069350004196167,
      "q_g0": 0.016525402665138245,
      "q_g1": 0.9834746718406677,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.322299003601074,
      "loss_g0": 2.3664777278900146,
      "loss_g1": 11.47277545928955,
      "q_g0": 0.016524463891983032,
      "q_g1": 0.9834755659103394,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.359169840812683,
      "loss_g0": 4.772253513336182,
      "loss_g1": 1.3018215894699097,
      "q_g0": 0.016524821519851685,
      "q_g1": 0.9834752082824707,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.4383806586265564,
      "loss_g0": 4.759541988372803,
      "loss_g1": 0.36577245593070984,
      "q_g0": 0.0165252722799778,
      "q_g1": 0.9834747910499573,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.420599937438965,
      "loss_g0": 13.532831192016602,
      "loss_g1": 2.23386812210083,
      "q_g0": 0.01652643457055092,
      "q_g1": 0.9834735989570618,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.13381290435791,
      "loss_g0": 2.13381290435791,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.394940376281738,
      "loss_g0": 14.349709510803223,
      "loss_g1": 17.087249755859375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 24.711746215820312,
      "loss_g0": 24.70990753173828,
      "loss_g1": 24.82118034362793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 56.87710189819336,
      "loss_g0": 57.55655288696289,
      "loss_g1": 16.433534622192383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 36.30373764038086,
      "loss_g0": 36.66862487792969,
      "loss_g1": 14.584209442138672,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.847421646118164,
      "loss_g0": 16.808488845825195,
      "loss_g1": 19.16480255126953,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522666439414024,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.433967590332031,
      "loss_g0": 14.324888229370117,
      "loss_g1": 20.92642593383789,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523340716958046,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.482917785644531,
      "loss_g0": 13.482917785644531,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.183164596557617,
      "loss_g0": 19.342422485351562,
      "loss_g1": 9.703580856323242,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 24.16071128845215,
      "loss_g0": 24.10890007019043,
      "loss_g1": 27.24462890625,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522744670510292,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 28.73560333251953,
      "loss_g0": 29.02413558959961,
      "loss_g1": 11.561094284057617,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 22.958694458007812,
      "loss_g0": 23.121326446533203,
      "loss_g1": 13.278122901916504,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.53327751159668,
      "loss_g0": 18.635074615478516,
      "loss_g1": 12.473872184753418,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.102794647216797,
      "loss_g0": 0.0,
      "loss_g1": 21.102794647216797,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.989563941955566,
      "loss_g0": 29.71623992919922,
      "loss_g1": 15.758955955505371,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 69.70504760742188,
      "loss_g0": 14.142016410827637,
      "loss_g1": 70.63851165771484,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.17704963684082,
      "loss_g0": 35.98830032348633,
      "loss_g1": 18.894590377807617,
      "q_g0": 0.016524169594049454,
      "q_g1": 0.9834758043289185,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.149853706359863,
      "loss_g0": 19.11492919921875,
      "loss_g1": 14.066428184509277,
      "q_g0": 0.01652468927204609,
      "q_g1": 0.9834753274917603,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.80196762084961,
      "loss_g0": 16.16567039489746,
      "loss_g1": 16.812658309936523,
      "q_g0": 0.016524620354175568,
      "q_g1": 0.9834753274917603,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.114776611328125,
      "loss_g0": 0.0,
      "loss_g1": 14.114776611328125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.300877571105957,
      "loss_g0": 10.592061996459961,
      "loss_g1": 6.228785514831543,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.541906356811523,
      "loss_g0": 10.758844375610352,
      "loss_g1": 17.655860900878906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.93083381652832,
      "loss_g0": 12.410247802734375,
      "loss_g1": 8.872377395629883,
      "q_g0": 0.016522789373993874,
      "q_g1": 0.9834771752357483,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.031856536865234,
      "loss_g0": 18.877702713012695,
      "loss_g1": 15.984044075012207,
      "q_g0": 0.0165230855345726,
      "q_g1": 0.9834769368171692,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.994319915771484,
      "loss_g0": 0.0,
      "loss_g1": 14.994319915771484,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 47.24055480957031,
      "loss_g0": 6.166938781738281,
      "loss_g1": 47.93058776855469,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.551725387573242,
      "loss_g0": 9.958440780639648,
      "loss_g1": 14.62889289855957,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.399892807006836,
      "loss_g0": 17.569562911987305,
      "loss_g1": 9.262635231018066,
      "q_g0": 0.016523271799087524,
      "q_g1": 0.9834766983985901,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 28.710878372192383,
      "loss_g0": 10.069316864013672,
      "loss_g1": 29.02405548095703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.7274343967437744,
      "loss_g0": 8.053016662597656,
      "loss_g1": 3.6547622680664062,
      "q_g0": 0.01652287133038044,
      "q_g1": 0.9834771752357483,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.3690290451049805,
      "loss_g0": 8.799095153808594,
      "loss_g1": 2.2609965801239014,
      "q_g0": 0.016523543745279312,
      "q_g1": 0.9834765195846558,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.13435173034668,
      "loss_g0": 6.437922477722168,
      "loss_g1": 16.297252655029297,
      "q_g0": 0.01652253419160843,
      "q_g1": 0.9834774136543274,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.866496562957764,
      "loss_g0": 12.016289710998535,
      "loss_g1": 4.746374130249023,
      "q_g0": 0.01652327924966812,
      "q_g1": 0.9834766983985901,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.461993217468262,
      "loss_g0": 22.105680465698242,
      "loss_g1": 12.29996109008789,
      "q_g0": 0.016524286940693855,
      "q_g1": 0.9834756851196289,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.756525993347168,
      "loss_g0": 10.097721099853516,
      "loss_g1": 4.666780471801758,
      "q_g0": 0.016524843871593475,
      "q_g1": 0.9834750890731812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.278233051300049,
      "loss_g0": 6.604146957397461,
      "loss_g1": 4.2391510009765625,
      "q_g0": 0.016525082290172577,
      "q_g1": 0.983474850654602,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.667637348175049,
      "loss_g0": 12.30332088470459,
      "loss_g1": 6.572939395904541,
      "q_g0": 0.016525674611330032,
      "q_g1": 0.9834743142127991,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.927651882171631,
      "loss_g0": 36.2335090637207,
      "loss_g1": 4.40150260925293,
      "q_g0": 0.01652892865240574,
      "q_g1": 0.9834710955619812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.4853296279907227,
      "loss_g0": 33.58969497680664,
      "loss_g1": 2.979276180267334,
      "q_g0": 0.016532061621546745,
      "q_g1": 0.9834679365158081,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.95266342163086,
      "loss_g0": 18.888181686401367,
      "loss_g1": 13.869695663452148,
      "q_g0": 0.016532571986317635,
      "q_g1": 0.9834674000740051,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.205142498016357,
      "loss_g0": 2.8873748779296875,
      "loss_g1": 7.277724266052246,
      "q_g0": 0.01653212122619152,
      "q_g1": 0.9834678173065186,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.656538724899292,
      "loss_g0": 5.309462547302246,
      "loss_g1": 2.6119422912597656,
      "q_g0": 0.016532402485609055,
      "q_g1": 0.9834676384925842,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.574925184249878,
      "loss_g0": 15.650386810302734,
      "loss_g1": 2.3551042079925537,
      "q_g0": 0.016533760353922844,
      "q_g1": 0.9834662079811096,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.318058967590332,
      "loss_g0": 4.015352249145508,
      "loss_g1": 6.356770992279053,
      "q_g0": 0.01653352566063404,
      "q_g1": 0.9834664463996887,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.59399938583374,
      "loss_g0": 4.59399938583374,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.18876838684082,
      "loss_g0": 21.52206802368164,
      "loss_g1": 1.3494994640350342,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.747056007385254,
      "loss_g0": 5.821248531341553,
      "loss_g1": 1.3308309316635132,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 26.057519912719727,
      "loss_g0": 26.057519912719727,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.13372230529785,
      "loss_g0": 20.466089248657227,
      "loss_g1": 0.34989893436431885,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.353548049926758,
      "loss_g0": 0.0,
      "loss_g1": 3.353548049926758,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.34249022603034973,
      "loss_g0": 6.514859199523926,
      "loss_g1": 0.23879441618919373,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 31.86273956298828,
      "loss_g0": 0.0,
      "loss_g1": 31.86273956298828,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 42.13089370727539,
      "loss_g0": 0.0,
      "loss_g1": 42.13089370727539,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.413591384887695,
      "loss_g0": 9.333288192749023,
      "loss_g1": 20.59973907470703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.3602352142334,
      "loss_g0": 29.73495101928711,
      "loss_g1": 19.185928344726562,
      "q_g0": 0.01652349717915058,
      "q_g1": 0.9834765195846558,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 38.78638458251953,
      "loss_g0": 18.68790054321289,
      "loss_g1": 39.12403869628906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.45832633972168,
      "loss_g0": 26.347286224365234,
      "loss_g1": 17.308984756469727,
      "q_g0": 0.016523348167538643,
      "q_g1": 0.9834766387939453,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.370863914489746,
      "loss_g0": 7.015928745269775,
      "loss_g1": 11.444028854370117,
      "q_g0": 0.016522899270057678,
      "q_g1": 0.9834770560264587,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.075737953186035,
      "loss_g0": 14.498417854309082,
      "loss_g1": 7.967829704284668,
      "q_g0": 0.016523558646440506,
      "q_g1": 0.9834765195846558,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 23.36817741394043,
      "loss_g0": 7.0799174308776855,
      "loss_g1": 23.64181900024414,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.036113739013672,
      "loss_g0": 20.097797393798828,
      "loss_g1": 15.967874526977539,
      "q_g0": 0.01652284525334835,
      "q_g1": 0.9834771752357483,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.20043659210205,
      "loss_g0": 34.730682373046875,
      "loss_g1": 11.82186508178711,
      "q_g0": 0.016525182873010635,
      "q_g1": 0.9834747910499573,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 36.858909606933594,
      "loss_g0": 11.83271598815918,
      "loss_g1": 37.279354095458984,
      "q_g0": 0.01652258075773716,
      "q_g1": 0.9834774136543274,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.06287956237793,
      "loss_g0": 26.54288673400879,
      "loss_g1": 19.954008102416992,
      "q_g0": 0.01652325503528118,
      "q_g1": 0.9834766983985901,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 22.628219604492188,
      "loss_g0": 0.0,
      "loss_g1": 22.628219604492188,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 36.175689697265625,
      "loss_g0": 26.14805030822754,
      "loss_g1": 36.344154357910156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.683581352233887,
      "loss_g0": 16.726882934570312,
      "loss_g1": 6.514842987060547,
      "q_g0": 0.01652347296476364,
      "q_g1": 0.9834765195846558,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.767092227935791,
      "loss_g0": 20.29936981201172,
      "loss_g1": 6.539715766906738,
      "q_g0": 0.016524873673915863,
      "q_g1": 0.9834750890731812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.074295043945312,
      "loss_g0": 7.479099273681641,
      "loss_g1": 20.28590965270996,
      "q_g0": 0.016523564234375954,
      "q_g1": 0.9834764003753662,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.43603229522705,
      "loss_g0": 0.0,
      "loss_g1": 13.43603229522705,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.684861421585083,
      "loss_g0": 12.792095184326172,
      "loss_g1": 3.531859874725342,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.166871547698975,
      "loss_g0": 16.75125503540039,
      "loss_g1": 7.0058441162109375,
      "q_g0": 0.016523417085409164,
      "q_g1": 0.9834766387939453,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.194398880004883,
      "loss_g0": 11.082477569580078,
      "loss_g1": 16.28028106689453,
      "q_g0": 0.016522882506251335,
      "q_g1": 0.9834770560264587,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.046144008636475,
      "loss_g0": 14.109319686889648,
      "loss_g1": 5.910671710968018,
      "q_g0": 0.01652371883392334,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.06706428527832,
      "loss_g0": 6.672441005706787,
      "loss_g1": 7.073694229125977,
      "q_g0": 0.016523681581020355,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.712776184082031,
      "loss_g0": 18.792388916015625,
      "loss_g1": 6.509807109832764,
      "q_g0": 0.016524938866496086,
      "q_g1": 0.9834750890731812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.526295185089111,
      "loss_g0": 5.470396995544434,
      "loss_g1": 7.5608391761779785,
      "q_g0": 0.016524719074368477,
      "q_g1": 0.9834752082824707,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.754985809326172,
      "loss_g0": 6.186801433563232,
      "loss_g1": 16.932544708251953,
      "q_g0": 0.01652361825108528,
      "q_g1": 0.9834764003753662,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.305548667907715,
      "loss_g0": 11.502527236938477,
      "loss_g1": 12.319039344787598,
      "q_g0": 0.016523534432053566,
      "q_g1": 0.9834765195846558,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.55240821838379,
      "loss_g0": 13.7584228515625,
      "loss_g1": 19.649751663208008,
      "q_g0": 0.016522930935025215,
      "q_g1": 0.9834770560264587,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.8418474197387695,
      "loss_g0": 0.0,
      "loss_g1": 6.8418474197387695,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.151981353759766,
      "loss_g0": 2.4471778869628906,
      "loss_g1": 8.247822761535645,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.446362495422363,
      "loss_g0": 15.982356071472168,
      "loss_g1": 15.437356948852539,
      "q_g0": 0.0165224801748991,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.000133514404297,
      "loss_g0": 14.533732414245605,
      "loss_g1": 17.041568756103516,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 10.34657096862793,
      "loss_g0": 24.2792911529541,
      "loss_g1": 10.112480163574219,
      "q_g0": 0.01652386784553528,
      "q_g1": 0.9834761619567871,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.6958417892456055,
      "loss_g0": 7.199110984802246,
      "loss_g1": 4.653782367706299,
      "q_g0": 0.016524124890565872,
      "q_g1": 0.983475923538208,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.196009635925293,
      "loss_g0": 26.991540908813477,
      "loss_g1": 6.863368034362793,
      "q_g0": 0.016526177525520325,
      "q_g1": 0.9834738373756409,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.931746482849121,
      "loss_g0": 19.573843002319336,
      "loss_g1": 3.6688718795776367,
      "q_g0": 0.016527807340025902,
      "q_g1": 0.9834722280502319,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.740696907043457,
      "loss_g0": 3.3705244064331055,
      "loss_g1": 5.7805280685424805,
      "q_g0": 0.016527557745575905,
      "q_g1": 0.983472466468811,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.450458526611328,
      "loss_g0": 7.450458526611328,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.5714099407196045,
      "loss_g0": 3.479940176010132,
      "loss_g1": 9.016036987304688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.333285808563232,
      "loss_g0": 5.311121463775635,
      "loss_g1": 6.652578353881836,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522565856575966,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.123404502868652,
      "loss_g0": 13.21324634552002,
      "loss_g1": 7.775615215301514,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.119674682617188,
      "loss_g0": 15.208662986755371,
      "loss_g1": 9.822729110717773,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.355198860168457,
      "loss_g0": 11.512838363647461,
      "loss_g1": 1.971914529800415,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.634570121765137,
      "loss_g0": 0.0,
      "loss_g1": 4.634570121765137,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.13287925720215,
      "loss_g0": 5.230068206787109,
      "loss_g1": 17.33284568786621,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.315225601196289,
      "loss_g0": 3.9494857788085938,
      "loss_g1": 5.338170051574707,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.490157127380371,
      "loss_g0": 8.490157127380371,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.54762864112854,
      "loss_g0": 3.5498805046081543,
      "loss_g1": 3.4135916233062744,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.25404167175293,
      "loss_g0": 0.0,
      "loss_g1": 4.25404167175293,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 20.6800479888916,
      "loss_g0": 32.1641960144043,
      "loss_g1": 20.48711395263672,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 25.872846603393555,
      "loss_g0": 17.244279861450195,
      "loss_g1": 26.017805099487305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.985671043395996,
      "loss_g0": 35.04494094848633,
      "loss_g1": 15.665435791015625,
      "q_g0": 0.016524402424693108,
      "q_g1": 0.9834755659103394,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 24.999895095825195,
      "loss_g0": 30.95149803161621,
      "loss_g1": 24.899892807006836,
      "q_g0": 0.016525015234947205,
      "q_g1": 0.9834749698638916,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 19.03690528869629,
      "loss_g0": 13.630514144897461,
      "loss_g1": 19.127742767333984,
      "q_g0": 0.016524456441402435,
      "q_g1": 0.9834755659103394,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 28.368417739868164,
      "loss_g0": 0.0,
      "loss_g1": 28.368417739868164,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.739856243133545,
      "loss_g0": 19.297000885009766,
      "loss_g1": 7.545695781707764,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 27.160215377807617,
      "loss_g0": 31.3071231842041,
      "loss_g1": 27.090545654296875,
      "q_g0": 0.016522854566574097,
      "q_g1": 0.9834771752357483,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 44.642921447753906,
      "loss_g0": 44.8114013671875,
      "loss_g1": 44.64008712768555,
      "q_g0": 0.01652287133038044,
      "q_g1": 0.9834771752357483,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 28.185733795166016,
      "loss_g0": 17.635406494140625,
      "loss_g1": 28.362977981567383,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.992265701293945,
      "loss_g0": 13.317598342895508,
      "loss_g1": 15.02039909362793,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.26761245727539,
      "loss_g0": 25.316688537597656,
      "loss_g1": 14.081974983215332,
      "q_g0": 0.0165235735476017,
      "q_g1": 0.9834764003753662,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.019708633422852,
      "loss_g0": 12.019708633422852,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.091253280639648,
      "loss_g0": 15.091253280639648,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.2900333404541,
      "loss_g0": 18.338882446289062,
      "loss_g1": 15.382343292236328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 46.484886169433594,
      "loss_g0": 46.98180389404297,
      "loss_g1": 16.906408309936523,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.801080703735352,
      "loss_g0": 11.684480667114258,
      "loss_g1": 18.741226196289062,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652313955128193,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.730000972747803,
      "loss_g0": 6.1620001792907715,
      "loss_g1": 40.53074645996094,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526658087968826,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 2.7990000247955322,
      "loss_g0": 2.477282762527466,
      "loss_g1": 21.941513061523438,
      "q_g0": 0.9834713339805603,
      "q_g1": 0.016528649255633354,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 115.09699249267578,
      "loss_g0": 116.69877624511719,
      "loss_g1": 19.752704620361328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 13.668776512145996,
      "loss_g0": 0.0,
      "loss_g1": 13.668776512145996,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 17.164897918701172,
      "loss_g0": 11.453226089477539,
      "loss_g1": 17.260852813720703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.915994167327881,
      "loss_g0": 39.45427322387695,
      "loss_g1": 5.35243034362793,
      "q_g0": 0.016525907441973686,
      "q_g1": 0.98347407579422,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.280966758728027,
      "loss_g0": 11.606715202331543,
      "loss_g1": 14.32590389251709,
      "q_g0": 0.0165256280452013,
      "q_g1": 0.9834744334220886,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.458940505981445,
      "loss_g0": 13.460390090942383,
      "loss_g1": 9.391700744628906,
      "q_g0": 0.01652604527771473,
      "q_g1": 0.9834739565849304,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.06467342376709,
      "loss_g0": 10.032758712768555,
      "loss_g1": 9.048405647277832,
      "q_g0": 0.016526145860552788,
      "q_g1": 0.9834738373756409,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.403948783874512,
      "loss_g0": 14.172443389892578,
      "loss_g1": 7.290207386016846,
      "q_g0": 0.016526851803064346,
      "q_g1": 0.9834731221199036,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.033615112304688,
      "loss_g0": 9.163263320922852,
      "loss_g1": 8.014631271362305,
      "q_g0": 0.0165269672870636,
      "q_g1": 0.9834730625152588,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 15.040923118591309,
      "loss_g0": 15.040923118591309,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.807494163513184,
      "loss_g0": 4.849156856536865,
      "loss_g1": 2.3275656700134277,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 18.545368194580078,
      "loss_g0": 18.73345184326172,
      "loss_g1": 7.350006103515625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 8.037123680114746,
      "loss_g0": 0.0,
      "loss_g1": 8.037123680114746,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.10596227645874,
      "loss_g0": 3.6396584510803223,
      "loss_g1": 6.147396087646484,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.008506774902344,
      "loss_g0": 10.927008628845215,
      "loss_g1": 8.976275444030762,
      "q_g0": 0.016522621735930443,
      "q_g1": 0.9834774136543274,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.8723843097686768,
      "loss_g0": 15.033004760742188,
      "loss_g1": 3.6848700046539307,
      "q_g0": 0.016523776575922966,
      "q_g1": 0.9834762811660767,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.2859978675842285,
      "loss_g0": 2.6665828227996826,
      "loss_g1": 6.346807479858398,
      "q_g0": 0.01652340218424797,
      "q_g1": 0.9834766387939453,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.8129265308380127,
      "loss_g0": 8.774008750915527,
      "loss_g1": 1.6959680318832397,
      "q_g0": 0.016524124890565872,
      "q_g1": 0.983475923538208,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.553812026977539,
      "loss_g0": 0.0,
      "loss_g1": 7.553812026977539,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.126565933227539,
      "loss_g0": 2.9972944259643555,
      "loss_g1": 7.195937633514404,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.9318084716796875,
      "loss_g0": 2.5969836711883545,
      "loss_g1": 5.987833023071289,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.41255319118499756,
      "loss_g0": 13.743476867675781,
      "loss_g1": 0.18857458233833313,
      "q_g0": 0.016523806378245354,
      "q_g1": 0.9834761619567871,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.716367244720459,
      "loss_g0": 16.312158584594727,
      "loss_g1": 5.538330554962158,
      "q_g0": 0.0165249053388834,
      "q_g1": 0.9834750890731812,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.4467318058013916,
      "loss_g0": 8.38518238067627,
      "loss_g1": 3.363750457763672,
      "q_g0": 0.01652541570365429,
      "q_g1": 0.9834745526313782,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.8911057710647583,
      "loss_g0": 7.5048346519470215,
      "loss_g1": 0.7799699306488037,
      "q_g0": 0.016526106745004654,
      "q_g1": 0.9834738373756409,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.349767684936523,
      "loss_g0": 28.478862762451172,
      "loss_g1": 3.944244146347046,
      "q_g0": 0.016528621315956116,
      "q_g1": 0.9834713935852051,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.45560884475708,
      "loss_g0": 5.938737392425537,
      "loss_g1": 4.430682182312012,
      "q_g0": 0.016528772190213203,
      "q_g1": 0.9834712147712708,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.1597936153411865,
      "loss_g0": 1.1535842418670654,
      "loss_g1": 3.1935107707977295,
      "q_g0": 0.01652856357395649,
      "q_g1": 0.9834713935852051,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 5.124877452850342,
      "loss_g0": 3.677701473236084,
      "loss_g1": 5.14919900894165,
      "q_g0": 0.0165284164249897,
      "q_g1": 0.9834715127944946,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 0.9996551275253296,
      "loss_g0": 2.493161678314209,
      "loss_g1": 0.974554717540741,
      "q_g0": 0.016528580337762833,
      "q_g1": 0.9834713935852051,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 7.205880641937256,
      "loss_g0": 6.044781684875488,
      "loss_g1": 7.225394248962402,
      "q_g0": 0.016528455540537834,
      "q_g1": 0.9834715127944946,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 30.541948318481445,
      "loss_g0": 16.281049728393555,
      "loss_g1": 30.781597137451172,
      "q_g0": 0.016526974737644196,
      "q_g1": 0.9834730625152588,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 22.90976333618164,
      "loss_g0": 22.90976333618164,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 14.009549140930176,
      "loss_g0": 13.895487785339355,
      "loss_g1": 20.79886245727539,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 27.525569915771484,
      "loss_g0": 27.814754486083984,
      "loss_g1": 10.31218433380127,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.01532745361328,
      "loss_g0": 21.028926849365234,
      "loss_g1": 20.205821990966797,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 23.133846282958984,
      "loss_g0": 23.292776107788086,
      "loss_g1": 13.673712730407715,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.843034744262695,
      "loss_g0": 11.74363899230957,
      "loss_g1": 17.75925064086914,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652303896844387,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 21.386341094970703,
      "loss_g0": 21.386341094970703,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 9.520455360412598,
      "loss_g0": 9.525514602661133,
      "loss_g1": 9.219331741333008,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.764857769012451,
      "loss_g0": 3.6472554206848145,
      "loss_g1": 10.764678955078125,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523147001862526,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.117719650268555,
      "loss_g0": 12.187522888183594,
      "loss_g1": 7.962855339050293,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522713005542755,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 6.360980033874512,
      "loss_g0": 6.291719913482666,
      "loss_g1": 10.483420372009277,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652313955128193,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 11.375062942504883,
      "loss_g0": 11.375062942504883,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 3.641981840133667,
      "loss_g0": 3.5622129440307617,
      "loss_g1": 8.390127182006836,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 12.285197257995605,
      "loss_g0": 12.41643238067627,
      "loss_g1": 4.473572254180908,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.9363268613815308,
      "loss_g0": 1.9484822750091553,
      "loss_g1": 1.2127883434295654,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 16.392290115356445,
      "loss_g0": 16.643714904785156,
      "loss_g1": 1.4264940023422241,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 4.7613606452941895,
      "loss_g0": 4.823165416717529,
      "loss_g1": 1.0825164318084717,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 1.6812082529067993,
      "loss_g0": 1.6553400754928589,
      "loss_g1": 3.2209725379943848,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652258075773716,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "eval_cer": 0.06331149842086793,
      "eval_loss": 14.459641456604004,
      "eval_runtime": 80.7806,
      "eval_samples_per_second": 20.748,
      "eval_steps_per_second": 5.187,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 38.1093864440918,
      "loss_g0": 37.83769226074219,
      "loss_g1": 54.27978515625,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524260863661766,
      "step": 600
    },
    {
      "epoch": 3.045685279187817,
      "gdro_loss": 25.92783546447754,
      "loss_g0": 25.500471115112305,
      "loss_g1": 51.35918045043945,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.016526898369193077,
      "step": 600
    },
    {
      "epoch": 3.553299492385787,
      "grad_norm": 105.00165557861328,
      "learning_rate": 0.00031535828877005345,
      "loss": 29.7732,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.283145904541016,
      "loss_g0": 26.66552734375,
      "loss_g1": 16.10871124267578,
      "q_g0": 0.01652350276708603,
      "q_g1": 0.9834765195846558,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.714534759521484,
      "loss_g0": 32.854854583740234,
      "loss_g1": 25.594562530517578,
      "q_g0": 0.016524240374565125,
      "q_g1": 0.9834758043289185,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.559776306152344,
      "loss_g0": 15.486857414245605,
      "loss_g1": 21.661808013916016,
      "q_g0": 0.016523612663149834,
      "q_g1": 0.9834764003753662,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.909050941467285,
      "loss_g0": 17.540891647338867,
      "loss_g1": 9.7808198928833,
      "q_g0": 0.016524402424693108,
      "q_g1": 0.9834755659103394,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 29.754383087158203,
      "loss_g0": 23.181053161621094,
      "loss_g1": 29.864824295043945,
      "q_g0": 0.01652371883392334,
      "q_g1": 0.9834762811660767,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.843096733093262,
      "loss_g0": 23.183706283569336,
      "loss_g1": 14.70295524597168,
      "q_g0": 0.01652458868920803,
      "q_g1": 0.9834754467010498,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.021759033203125,
      "loss_g0": 40.12635040283203,
      "loss_g1": 9.515836715698242,
      "q_g0": 0.01652771234512329,
      "q_g1": 0.9834723472595215,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.379841804504395,
      "loss_g0": 66.23954010009766,
      "loss_g1": 11.474392890930176,
      "q_g0": 0.01653330959379673,
      "q_g1": 0.9834666848182678,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.161354064941406,
      "loss_g0": 26.42388916015625,
      "loss_g1": 12.938375473022461,
      "q_g0": 0.01653468981385231,
      "q_g1": 0.983465313911438,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.325143814086914,
      "loss_g0": 15.835780143737793,
      "loss_g1": 16.333370208740234,
      "q_g0": 0.016534633934497833,
      "q_g1": 0.983465313911438,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.96002769470215,
      "loss_g0": 30.478618621826172,
      "loss_g1": 22.833614349365234,
      "q_g0": 0.016535410657525063,
      "q_g1": 0.9834645986557007,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 54.77720642089844,
      "loss_g0": 14.232145309448242,
      "loss_g1": 55.45873260498047,
      "q_g0": 0.01653119921684265,
      "q_g1": 0.983468770980835,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.612776517868042,
      "loss_g0": 25.081783294677734,
      "loss_g1": 3.2518529891967773,
      "q_g0": 0.01653343252837658,
      "q_g1": 0.9834665656089783,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.275422096252441,
      "loss_g0": 15.05356502532959,
      "loss_g1": 13.245528221130371,
      "q_g0": 0.016533618792891502,
      "q_g1": 0.9834663271903992,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.82963752746582,
      "loss_g0": 20.847301483154297,
      "loss_g1": 11.678028106689453,
      "q_g0": 0.016534557566046715,
      "q_g1": 0.9834654331207275,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 17.521038055419922,
      "loss_g0": 17.1986026763916,
      "loss_g1": 17.526458740234375,
      "q_g0": 0.01653452403843403,
      "q_g1": 0.9834654331207275,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.510971069335938,
      "loss_g0": 26.510971069335938,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.041964054107666,
      "loss_g0": 5.941444396972656,
      "loss_g1": 12.025275230407715,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 23.623258590698242,
      "loss_g0": 23.814313888549805,
      "loss_g1": 12.250886917114258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 27.42784309387207,
      "loss_g0": 27.604248046875,
      "loss_g1": 16.92750358581543,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.760150909423828,
      "loss_g0": 10.714774131774902,
      "loss_g1": 13.461118698120117,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522705554962158,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.61916446685791,
      "loss_g0": 14.46502685546875,
      "loss_g1": 23.793277740478516,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523659229278564,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 28.624805450439453,
      "loss_g0": 29.025957107543945,
      "loss_g1": 4.74671745300293,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.306899070739746,
      "loss_g0": 7.239904403686523,
      "loss_g1": 11.294561386108398,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522839665412903,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.399341583251953,
      "loss_g0": 24.73681640625,
      "loss_g1": 4.311556816101074,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.601350784301758,
      "loss_g0": 0.0,
      "loss_g1": 5.601350784301758,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.468303680419922,
      "loss_g0": 0.0,
      "loss_g1": 7.468303680419922,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.796443939208984,
      "loss_g0": 8.920425415039062,
      "loss_g1": 26.0799617767334,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.465373992919922,
      "loss_g0": 11.070684432983398,
      "loss_g1": 8.42160415649414,
      "q_g0": 0.016522690653800964,
      "q_g1": 0.9834772944450378,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.7026915550231934,
      "loss_g0": 21.851234436035156,
      "loss_g1": 2.380951404571533,
      "q_g0": 0.016524681821465492,
      "q_g1": 0.9834753274917603,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.8098461627960205,
      "loss_g0": 7.42735481262207,
      "loss_g1": 2.7322587966918945,
      "q_g0": 0.016525164246559143,
      "q_g1": 0.983474850654602,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.272464752197266,
      "loss_g0": 6.651849269866943,
      "loss_g1": 11.350101470947266,
      "q_g0": 0.01652468927204609,
      "q_g1": 0.9834753274917603,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.340999126434326,
      "loss_g0": 5.56980562210083,
      "loss_g1": 4.320352077484131,
      "q_g0": 0.016524814069271088,
      "q_g1": 0.9834752082824707,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.4728214740753174,
      "loss_g0": 10.009333610534668,
      "loss_g1": 3.3629870414733887,
      "q_g0": 0.016525497660040855,
      "q_g1": 0.9834745526313782,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.7852144241333,
      "loss_g0": 9.7852144241333,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.727928161621094,
      "loss_g0": 11.871902465820312,
      "loss_g1": 3.1580352783203125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.771665573120117,
      "loss_g0": 18.902034759521484,
      "loss_g1": 11.01162338256836,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.4846014976501465,
      "loss_g0": 0.0,
      "loss_g1": 6.4846014976501465,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.084344863891602,
      "loss_g0": 21.814189910888672,
      "loss_g1": 7.853682994842529,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.381505250930786,
      "loss_g0": 4.97845458984375,
      "loss_g1": 3.3546762466430664,
      "q_g0": 0.016522590070962906,
      "q_g1": 0.9834774136543274,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.6828413009643555,
      "loss_g0": 9.094772338867188,
      "loss_g1": 3.591916799545288,
      "q_g0": 0.016523147001862526,
      "q_g1": 0.9834768176078796,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.7100882530212402,
      "loss_g0": 7.9878435134887695,
      "loss_g1": 1.6046130657196045,
      "q_g0": 0.016523797065019608,
      "q_g1": 0.9834761619567871,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.239362716674805,
      "loss_g0": 17.12093734741211,
      "loss_g1": 8.090130805969238,
      "q_g0": 0.016524719074368477,
      "q_g1": 0.9834752082824707,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.981281280517578,
      "loss_g0": 70.57423400878906,
      "loss_g1": 3.878701686859131,
      "q_g0": 0.016531532630324364,
      "q_g1": 0.9834684133529663,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.600293159484863,
      "loss_g0": 9.077140808105469,
      "loss_g1": 9.609086990356445,
      "q_g0": 0.016531480476260185,
      "q_g1": 0.9834685325622559,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.308990478515625,
      "loss_g0": 7.61503791809082,
      "loss_g1": 21.539155960083008,
      "q_g0": 0.016530051827430725,
      "q_g1": 0.9834699034690857,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.01895809173584,
      "loss_g0": 17.192745208740234,
      "loss_g1": 3.797515630722046,
      "q_g0": 0.01653142459690571,
      "q_g1": 0.9834685325622559,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.1702232360839844,
      "loss_g0": 4.408855438232422,
      "loss_g1": 3.149402379989624,
      "q_g0": 0.016531549394130707,
      "q_g1": 0.9834684133529663,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.8441061973571777,
      "loss_g0": 19.787065505981445,
      "loss_g1": 3.5760867595672607,
      "q_g0": 0.01653320901095867,
      "q_g1": 0.9834668040275574,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 0.6220934391021729,
      "loss_g0": 0.2904559373855591,
      "loss_g1": 0.6276686191558838,
      "q_g0": 0.016533177345991135,
      "q_g1": 0.9834668040275574,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.74930763244629,
      "loss_g0": 19.758811950683594,
      "loss_g1": 24.833202362060547,
      "q_g0": 0.016532665118575096,
      "q_g1": 0.9834674000740051,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 29.73310089111328,
      "loss_g0": 29.73310089111328,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 33.21611785888672,
      "loss_g0": 33.585853576660156,
      "loss_g1": 11.207984924316406,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.347824096679688,
      "loss_g0": 12.189250946044922,
      "loss_g1": 21.78609848022461,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652340218424797,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 40.155025482177734,
      "loss_g0": 40.334205627441406,
      "loss_g1": 29.489608764648438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.869832992553711,
      "loss_g0": 7.71236515045166,
      "loss_g1": 17.242361068725586,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523394733667374,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.007490158081055,
      "loss_g0": 17.80036163330078,
      "loss_g1": 30.33489990234375,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524672508239746,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.127124786376953,
      "loss_g0": 14.187758445739746,
      "loss_g1": 10.518372535705566,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652430184185505,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.225852966308594,
      "loss_g0": 0.0,
      "loss_g1": 22.225852966308594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 43.741004943847656,
      "loss_g0": 17.071514129638672,
      "loss_g1": 44.18905258178711,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.429258346557617,
      "loss_g0": 19.2240047454834,
      "loss_g1": 4.180683135986328,
      "q_g0": 0.01652396097779274,
      "q_g1": 0.9834760427474976,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.024210929870605,
      "loss_g0": 4.965451240539551,
      "loss_g1": 14.176403045654297,
      "q_g0": 0.01652301475405693,
      "q_g1": 0.9834769368171692,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.883912086486816,
      "loss_g0": 19.451873779296875,
      "loss_g1": 11.756759643554688,
      "q_g0": 0.016523806378245354,
      "q_g1": 0.9834761619567871,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.7916841506958,
      "loss_g0": 23.458173751831055,
      "loss_g1": 10.578851699829102,
      "q_g0": 0.01652512326836586,
      "q_g1": 0.983474850654602,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.674165725708008,
      "loss_g0": 16.488168716430664,
      "loss_g1": 26.845308303833008,
      "q_g0": 0.0165240615606308,
      "q_g1": 0.983475923538208,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 50.22768020629883,
      "loss_g0": 15.159685134887695,
      "loss_g1": 50.81682205200195,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.83615493774414,
      "loss_g0": 14.688064575195312,
      "loss_g1": 16.872243881225586,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.0838723182678223,
      "loss_g0": 15.357673645019531,
      "loss_g1": 2.8776562213897705,
      "q_g0": 0.016523705795407295,
      "q_g1": 0.9834762811660767,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.189114570617676,
      "loss_g0": 60.70318603515625,
      "loss_g1": 10.356949806213379,
      "q_g0": 0.016528859734535217,
      "q_g1": 0.9834712147712708,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.301602363586426,
      "loss_g0": 5.191460609436035,
      "loss_g1": 13.437899589538574,
      "q_g0": 0.016528014093637466,
      "q_g1": 0.9834719896316528,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.751360893249512,
      "loss_g0": 34.91779708862305,
      "loss_g1": 8.311539649963379,
      "q_g0": 0.01653074286878109,
      "q_g1": 0.9834692478179932,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.533476829528809,
      "loss_g0": 4.547708988189697,
      "loss_g1": 11.650893211364746,
      "q_g0": 0.016530020162463188,
      "q_g1": 0.9834699034690857,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.667551040649414,
      "loss_g0": 13.817495346069336,
      "loss_g1": 13.665030479431152,
      "q_g0": 0.01653003692626953,
      "q_g1": 0.9834699034690857,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.2527360916137695,
      "loss_g0": 16.863513946533203,
      "loss_g1": 4.0407586097717285,
      "q_g0": 0.01653134822845459,
      "q_g1": 0.9834686517715454,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.531500816345215,
      "loss_g0": 2.1924242973327637,
      "loss_g1": 6.604434967041016,
      "q_g0": 0.016530897468328476,
      "q_g1": 0.9834691286087036,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.58932876586914,
      "loss_g0": 4.725974082946777,
      "loss_g1": 19.83913803100586,
      "q_g0": 0.016529347747564316,
      "q_g1": 0.983470618724823,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.25800895690918,
      "loss_g0": 14.369504928588867,
      "loss_g1": 16.28974723815918,
      "q_g0": 0.016529154032468796,
      "q_g1": 0.9834708571434021,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.77486801147461,
      "loss_g0": 24.555747985839844,
      "loss_g1": 11.560043334960938,
      "q_g0": 0.0165304783731699,
      "q_g1": 0.9834696054458618,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.932578086853027,
      "loss_g0": 8.355978012084961,
      "loss_g1": 4.875035285949707,
      "q_g0": 0.016530826687812805,
      "q_g1": 0.9834691286087036,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.828776359558105,
      "loss_g0": 24.704421997070312,
      "loss_g1": 10.595523834228516,
      "q_g0": 0.01653227023780346,
      "q_g1": 0.9834677577018738,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.4553780555725098,
      "loss_g0": 13.98111343383789,
      "loss_g1": 2.2616138458251953,
      "q_g0": 0.016533473506569862,
      "q_g1": 0.9834665656089783,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.6110098361968994,
      "loss_g0": 22.116769790649414,
      "loss_g1": 3.2998647689819336,
      "q_g0": 0.01653539389371872,
      "q_g1": 0.9834645986557007,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.25135040283203,
      "loss_g0": 12.23580551147461,
      "loss_g1": 21.402923583984375,
      "q_g0": 0.016534456983208656,
      "q_g1": 0.9834655523300171,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.24326229095459,
      "loss_g0": 3.9773027896881104,
      "loss_g1": 6.281358242034912,
      "q_g0": 0.016534214839339256,
      "q_g1": 0.9834657907485962,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.1864652633667,
      "loss_g0": 13.1864652633667,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.445956707000732,
      "loss_g0": 0.0,
      "loss_g1": 4.445956707000732,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.7817487716674805,
      "loss_g0": 17.567487716674805,
      "loss_g1": 7.6173481941223145,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.665889739990234,
      "loss_g0": 15.447772979736328,
      "loss_g1": 5.501543998718262,
      "q_g0": 0.016523443162441254,
      "q_g1": 0.9834766387939453,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 0.7246532440185547,
      "loss_g0": 7.103377819061279,
      "loss_g1": 0.6174795627593994,
      "q_g0": 0.01652410998940468,
      "q_g1": 0.983475923538208,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.867567539215088,
      "loss_g0": 0.0,
      "loss_g1": 6.867567539215088,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.926425933837891,
      "loss_g0": 5.926425933837891,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.107160568237305,
      "loss_g0": 15.322830200195312,
      "loss_g1": 2.2696597576141357,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.327978134155273,
      "loss_g0": 6.387893199920654,
      "loss_g1": 2.761589288711548,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.46120548248291,
      "loss_g0": 10.58973217010498,
      "loss_g1": 2.8108134269714355,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.731419324874878,
      "loss_g0": 1.7529624700546265,
      "loss_g1": 0.44909170269966125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.354679107666016,
      "loss_g0": 22.354679107666016,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.135547637939453,
      "loss_g0": 8.263275146484375,
      "loss_g1": 0.5327126979827881,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.4406647682189941,
      "loss_g0": 1.4521405696868896,
      "loss_g1": 0.7575798034667969,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.0229638814926147,
      "loss_g0": 1.0229638814926147,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.8461389541625977,
      "loss_g0": 2.8143510818481445,
      "loss_g1": 4.738279342651367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 64.46514129638672,
      "loss_g0": 64.84066009521484,
      "loss_g1": 42.112857818603516,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 23.754375457763672,
      "loss_g0": 23.754375457763672,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.533735275268555,
      "loss_g0": 18.4003849029541,
      "loss_g1": 26.471166610717773,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.345645904541016,
      "loss_g0": 25.987842559814453,
      "loss_g1": 47.640499114990234,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.01652463525533676,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.1874942779541,
      "loss_g0": 16.056514739990234,
      "loss_g1": 23.98248863220215,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.01652543991804123,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 37.995277404785156,
      "loss_g0": 38.18242645263672,
      "loss_g1": 26.856719970703125,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652427762746811,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.673507690429688,
      "loss_g0": 22.525020599365234,
      "loss_g1": 31.510583877563477,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.01652519218623638,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.74784278869629,
      "loss_g0": 22.775806427001953,
      "loss_g1": 21.08366584777832,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016525015234947205,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 27.78072738647461,
      "loss_g0": 28.04741096496582,
      "loss_g1": 11.907693862915039,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523364931344986,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.665167808532715,
      "loss_g0": 12.499606132507324,
      "loss_g1": 22.51881217956543,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652439311146736,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 30.808921813964844,
      "loss_g0": 30.959463119506836,
      "loss_g1": 21.848827362060547,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.0165234562009573,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 28.98493194580078,
      "loss_g0": 29.180442810058594,
      "loss_g1": 17.347373962402344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 28.547395706176758,
      "loss_g0": 28.436954498291016,
      "loss_g1": 35.12097930908203,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523102298378944,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.018465042114258,
      "loss_g0": 19.03163719177246,
      "loss_g1": 18.234500885009766,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523022204637527,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.144596099853516,
      "loss_g0": 26.3753604888916,
      "loss_g1": 12.408610343933105,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.9345784187316895,
      "loss_g0": 7.839321136474609,
      "loss_g1": 13.604446411132812,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523005440831184,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.722657203674316,
      "loss_g0": 10.568230628967285,
      "loss_g1": 19.913833618164062,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.01652396097779274,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.092915534973145,
      "loss_g0": 14.002596855163574,
      "loss_g1": 19.468305587768555,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.01652451977133751,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 38.99660873413086,
      "loss_g0": 39.099037170410156,
      "loss_g1": 32.89997100830078,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652388460934162,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.44242286682129,
      "loss_g0": 18.602279663085938,
      "loss_g1": 8.927356719970703,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522899270057678,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 66.66629028320312,
      "loss_g0": 67.453369140625,
      "loss_g1": 19.81637954711914,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.47728729248047,
      "loss_g0": 22.72998809814453,
      "loss_g1": 7.435633659362793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.677245140075684,
      "loss_g0": 10.8038911819458,
      "loss_g1": 3.1387603282928467,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.411494255065918,
      "loss_g0": 14.578146934509277,
      "loss_g1": 4.491644382476807,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.0283918380737305,
      "loss_g0": 4.910170078277588,
      "loss_g1": 12.06507396697998,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523154452443123,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.317626953125,
      "loss_g0": 7.3775858879089355,
      "loss_g1": 3.748727798461914,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 17.76239013671875,
      "loss_g0": 17.921709060668945,
      "loss_g1": 8.279102325439453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.664322853088379,
      "loss_g0": 8.61031723022461,
      "loss_g1": 11.878859519958496,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522759571671486,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.36624813079834,
      "loss_g0": 9.43712043762207,
      "loss_g1": 5.147642135620117,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.386873245239258,
      "loss_g0": 25.697513580322266,
      "loss_g1": 6.896349906921387,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.09243392944336,
      "loss_g0": 9.089523315429688,
      "loss_g1": 9.265731811523438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.93204116821289,
      "loss_g0": 11.019796371459961,
      "loss_g1": 5.708512783050537,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.263223171234131,
      "loss_g0": 3.19035005569458,
      "loss_g1": 7.600774765014648,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652287133038044,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.189661026000977,
      "loss_g0": 11.242158889770508,
      "loss_g1": 8.06479263305664,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522543504834175,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.056039810180664,
      "loss_g0": 10.100282669067383,
      "loss_g1": 7.422517776489258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.992758750915527,
      "loss_g0": 9.061691284179688,
      "loss_g1": 4.889631748199463,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.175793647766113,
      "loss_g0": 4.109404563903809,
      "loss_g1": 8.12740421295166,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522830352187157,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.519293785095215,
      "loss_g0": 12.684635162353516,
      "loss_g1": 2.677492618560791,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.7012271881103516,
      "loss_g0": 1.625176191329956,
      "loss_g1": 6.227944374084473,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522889956831932,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.994115829467773,
      "loss_g0": 22.251903533935547,
      "loss_g1": 6.649593353271484,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.4675750732421875,
      "loss_g0": 7.377333641052246,
      "loss_g1": 12.838902473449707,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522977501153946,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.672964096069336,
      "loss_g0": 12.643289566040039,
      "loss_g1": 14.43922233581543,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523154452443123,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.502045631408691,
      "loss_g0": 8.602094650268555,
      "loss_g1": 2.5467336177825928,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522543504834175,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.9920430183410645,
      "loss_g0": 7.897994041442871,
      "loss_g1": 13.589948654174805,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523122787475586,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.309671401977539,
      "loss_g0": 3.309671401977539,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.83852481842041,
      "loss_g0": 6.910638809204102,
      "loss_g1": 2.546024799346924,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.138529777526855,
      "loss_g0": 12.242094993591309,
      "loss_g1": 5.97392463684082,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.459927558898926,
      "loss_g0": 12.621851921081543,
      "loss_g1": 2.8216075897216797,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.082263946533203,
      "loss_g0": 6.082263946533203,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.9830193519592285,
      "loss_g0": 6.064006805419922,
      "loss_g1": 1.1623245477676392,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 55.14441680908203,
      "loss_g0": 55.383392333984375,
      "loss_g1": 40.91949462890625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.76653480529785,
      "loss_g0": 0.0,
      "loss_g1": 22.76653480529785,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 30.326692581176758,
      "loss_g0": 30.326692581176758,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.38996124267578,
      "loss_g0": 16.131107330322266,
      "loss_g1": 31.797861099243164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 37.405216217041016,
      "loss_g0": 36.56620407104492,
      "loss_g1": 87.33036804199219,
      "q_g0": 0.9834723472595215,
      "q_g1": 0.016527609899640083,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.3870849609375,
      "loss_g0": 16.3870849609375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.418285369873047,
      "loss_g0": 20.418285369873047,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.981359481811523,
      "loss_g0": 20.713165283203125,
      "loss_g1": 36.94525146484375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 32.97862243652344,
      "loss_g0": 33.213951110839844,
      "loss_g1": 18.97083854675293,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.908529281616211,
      "loss_g0": 11.72553539276123,
      "loss_g1": 22.800329208374023,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523558646440506,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 41.14459991455078,
      "loss_g0": 41.70183181762695,
      "loss_g1": 7.976008415222168,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 30.635292053222656,
      "loss_g0": 30.88625144958496,
      "loss_g1": 15.697148323059082,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.51783561706543,
      "loss_g0": 21.602218627929688,
      "loss_g1": 16.495040893554688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.852001190185547,
      "loss_g0": 19.012802124023438,
      "loss_g1": 9.280410766601562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.284629821777344,
      "loss_g0": 24.579374313354492,
      "loss_g1": 6.740305423736572,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.549620628356934,
      "loss_g0": 13.55668830871582,
      "loss_g1": 13.128952980041504,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.484457015991211,
      "loss_g0": 13.410661697387695,
      "loss_g1": 17.876873016357422,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.476107597351074,
      "loss_g0": 10.570882797241211,
      "loss_g1": 4.834754467010498,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.10083770751953,
      "loss_g0": 0.0,
      "loss_g1": 24.10083770751953,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.84421443939209,
      "loss_g0": 0.0,
      "loss_g1": 4.84421443939209,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.338581085205078,
      "loss_g0": 20.183786392211914,
      "loss_g1": 8.139581680297852,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.161843299865723,
      "loss_g0": 25.99665069580078,
      "loss_g1": 11.929397583007812,
      "q_g0": 0.01652386039495468,
      "q_g1": 0.9834761619567871,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.141247272491455,
      "loss_g0": 12.159087181091309,
      "loss_g1": 6.040134429931641,
      "q_g0": 0.016524480655789375,
      "q_g1": 0.9834755659103394,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.027883529663086,
      "loss_g0": 28.773311614990234,
      "loss_g1": 24.964950561523438,
      "q_g0": 0.016524866223335266,
      "q_g1": 0.9834750890731812,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.973623275756836,
      "loss_g0": 8.973623275756836,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.269851684570312,
      "loss_g0": 24.269851684570312,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.711153984069824,
      "loss_g0": 4.737190246582031,
      "loss_g1": 3.161388635635376,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.9196057319641113,
      "loss_g0": 2.854739189147949,
      "loss_g1": 6.780608654022217,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.9370832443237305,
      "loss_g0": 5.9435343742370605,
      "loss_g1": 5.553102493286133,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.11188793182373,
      "loss_g0": 9.190722465515137,
      "loss_g1": 4.419303894042969,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.264164924621582,
      "loss_g0": 7.22970724105835,
      "loss_g1": 9.315153121948242,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.73031234741211,
      "loss_g0": 16.815906524658203,
      "loss_g1": 11.635394096374512,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.509676933288574,
      "loss_g0": 10.509676933288574,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.735794067382812,
      "loss_g0": 26.066308975219727,
      "loss_g1": 6.062204360961914,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.841287612915039,
      "loss_g0": 13.030882835388184,
      "loss_g1": 1.5558446645736694,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.547034740447998,
      "loss_g0": 7.547034740447998,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.86880874633789,
      "loss_g0": 9.983733177185059,
      "loss_g1": 3.028036117553711,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.43897533416748,
      "loss_g0": 11.522037506103516,
      "loss_g1": 6.494748115539551,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.807178020477295,
      "loss_g0": 4.847305774688721,
      "loss_g1": 2.418623924255371,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.419978618621826,
      "loss_g0": 7.4054951667785645,
      "loss_g1": 8.28205680847168,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522511839866638,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.87483787536621,
      "loss_g0": 17.012758255004883,
      "loss_g1": 8.665289878845215,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.7514609098434448,
      "loss_g0": 1.772369623184204,
      "loss_g1": 0.506891131401062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.1093969345092773,
      "loss_g0": 2.1131978034973145,
      "loss_g1": 1.8831512928009033,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.14748764038086,
      "loss_g0": 11.211593627929688,
      "loss_g1": 7.331653594970703,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.519975662231445,
      "loss_g0": 8.643472671508789,
      "loss_g1": 1.1689575910568237,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.965425491333008,
      "loss_g0": 12.158885955810547,
      "loss_g1": 0.4499492943286896,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.261289596557617,
      "loss_g0": 4.288295269012451,
      "loss_g1": 2.6537797451019287,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.231747627258301,
      "loss_g0": 5.24073600769043,
      "loss_g1": 4.696747779846191,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.3037785291671753,
      "loss_g0": 1.1604406833648682,
      "loss_g1": 9.835330963134766,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523310914635658,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.7027177810668945,
      "loss_g0": 0.0,
      "loss_g1": 7.7027177810668945,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 31.37925148010254,
      "loss_g0": 81.09190368652344,
      "loss_g1": 30.544078826904297,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 27.520784378051758,
      "loss_g0": 40.970218658447266,
      "loss_g1": 27.294815063476562,
      "q_g0": 0.0165238194167614,
      "q_g1": 0.9834761619567871,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 46.885108947753906,
      "loss_g0": 13.309748649597168,
      "loss_g1": 47.44917297363281,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 38.80171203613281,
      "loss_g0": 30.901941299438477,
      "loss_g1": 38.93442916870117,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.37825584411621,
      "loss_g0": 18.27151870727539,
      "loss_g1": 20.41364860534668,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.09410572052002,
      "loss_g0": 14.165664672851562,
      "loss_g1": 15.10970401763916,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.156196594238281,
      "loss_g0": 21.169113159179688,
      "loss_g1": 15.05517578125,
      "q_g0": 0.016523046419024467,
      "q_g1": 0.9834769368171692,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 28.993234634399414,
      "loss_g0": 12.775154113769531,
      "loss_g1": 29.265697479248047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.619868278503418,
      "loss_g0": 18.519392013549805,
      "loss_g1": 8.453545570373535,
      "q_g0": 0.0165234487503767,
      "q_g1": 0.9834765195846558,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.40960693359375,
      "loss_g0": 12.742666244506836,
      "loss_g1": 26.639211654663086,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.802206039428711,
      "loss_g0": 25.21059799194336,
      "loss_g1": 15.644134521484375,
      "q_g0": 0.01652340218424797,
      "q_g1": 0.9834766387939453,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.295787811279297,
      "loss_g0": 30.504913330078125,
      "loss_g1": 21.141056060791016,
      "q_g0": 0.016524355858564377,
      "q_g1": 0.9834756851196289,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.97218132019043,
      "loss_g0": 18.76474380493164,
      "loss_g1": 8.807635307312012,
      "q_g0": 0.016525372862815857,
      "q_g1": 0.9834746718406677,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.893102645874023,
      "loss_g0": 6.628836631774902,
      "loss_g1": 22.14956283569336,
      "q_g0": 0.016523776575922966,
      "q_g1": 0.9834762811660767,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.007100105285645,
      "loss_g0": 21.07704734802246,
      "loss_g1": 11.854703903198242,
      "q_g0": 0.016524719074368477,
      "q_g1": 0.9834752082824707,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 28.594694137573242,
      "loss_g0": 11.634328842163086,
      "loss_g1": 28.879636764526367,
      "q_g0": 0.016522955149412155,
      "q_g1": 0.9834770560264587,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 34.56232833862305,
      "loss_g0": 16.32537078857422,
      "loss_g1": 34.868709564208984,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.717307090759277,
      "loss_g0": 12.60612964630127,
      "loss_g1": 15.769575119018555,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.691421508789062,
      "loss_g0": 28.031892776489258,
      "loss_g1": 11.416872024536133,
      "q_g0": 0.016524124890565872,
      "q_g1": 0.983475923538208,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.833279609680176,
      "loss_g0": 13.354369163513184,
      "loss_g1": 12.824524879455566,
      "q_g0": 0.01652417704463005,
      "q_g1": 0.9834758043289185,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.720050811767578,
      "loss_g0": 22.690244674682617,
      "loss_g1": 16.619735717773438,
      "q_g0": 0.016524799168109894,
      "q_g1": 0.9834752082824707,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.29870891571045,
      "loss_g0": 49.212547302246094,
      "loss_g1": 7.611075401306152,
      "q_g0": 0.016529053449630737,
      "q_g1": 0.9834709763526917,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 38.6605339050293,
      "loss_g0": 21.765188217163086,
      "loss_g1": 38.944461822509766,
      "q_g0": 0.016527293249964714,
      "q_g1": 0.9834727048873901,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.587589263916016,
      "loss_g0": 11.674264907836914,
      "loss_g1": 9.552522659301758,
      "q_g0": 0.016527509316802025,
      "q_g1": 0.983472466468811,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.685309410095215,
      "loss_g0": 11.565752983093262,
      "loss_g1": 5.586483478546143,
      "q_g0": 0.01652812957763672,
      "q_g1": 0.9834718704223633,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.206604957580566,
      "loss_g0": 13.314373016357422,
      "loss_g1": 15.238404273986816,
      "q_g0": 0.016527937725186348,
      "q_g1": 0.9834721088409424,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.470453262329102,
      "loss_g0": 16.258630752563477,
      "loss_g1": 12.406787872314453,
      "q_g0": 0.016528325155377388,
      "q_g1": 0.9834716320037842,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.8774056434631348,
      "loss_g0": 14.359901428222656,
      "loss_g1": 2.684415578842163,
      "q_g0": 0.016529517248272896,
      "q_g1": 0.9834704995155334,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.02227783203125,
      "loss_g0": 0.0,
      "loss_g1": 8.02227783203125,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.401520729064941,
      "loss_g0": 37.58201217651367,
      "loss_g1": 7.911288261413574,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.402459740638733,
      "loss_g0": 16.740318298339844,
      "loss_g1": 1.1447583436965942,
      "q_g0": 0.016524016857147217,
      "q_g1": 0.9834760427474976,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.297744750976562,
      "loss_g0": 7.546816349029541,
      "loss_g1": 10.343963623046875,
      "q_g0": 0.016523728147149086,
      "q_g1": 0.9834762811660767,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.273985385894775,
      "loss_g0": 8.98159122467041,
      "loss_g1": 7.245295524597168,
      "q_g0": 0.01652390882372856,
      "q_g1": 0.9834761619567871,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.136204719543457,
      "loss_g0": 17.00511932373047,
      "loss_g1": 2.903164863586426,
      "q_g0": 0.016525348648428917,
      "q_g1": 0.9834746718406677,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.390181064605713,
      "loss_g0": 3.6516668796539307,
      "loss_g1": 2.3689842224121094,
      "q_g0": 0.016525473445653915,
      "q_g1": 0.9834745526313782,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.744596481323242,
      "loss_g0": 4.421564102172852,
      "loss_g1": 8.817234992980957,
      "q_g0": 0.0165250226855278,
      "q_g1": 0.9834749698638916,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.194012641906738,
      "loss_g0": 44.90680694580078,
      "loss_g1": 10.627425193786621,
      "q_g0": 0.016528524458408356,
      "q_g1": 0.9834713935852051,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.638507843017578,
      "loss_g0": 23.036590576171875,
      "loss_g1": 3.312457323074341,
      "q_g0": 0.016530539840459824,
      "q_g1": 0.9834694862365723,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.214855194091797,
      "loss_g0": 7.3219194412231445,
      "loss_g1": 6.196247100830078,
      "q_g0": 0.016530657187104225,
      "q_g1": 0.9834693670272827,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.439665794372559,
      "loss_g0": 25.29645347595215,
      "loss_g1": 5.105859756469727,
      "q_g0": 0.01653272844851017,
      "q_g1": 0.9834672808647156,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 40.98733901977539,
      "loss_g0": 6.920210838317871,
      "loss_g1": 41.559906005859375,
      "q_g0": 0.016529183834791183,
      "q_g1": 0.9834708571434021,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.296814441680908,
      "loss_g0": 7.687559127807617,
      "loss_g1": 4.2398247718811035,
      "q_g0": 0.01652953401207924,
      "q_g1": 0.9834704995155334,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.126530647277832,
      "loss_g0": 10.280462265014648,
      "loss_g1": 4.02309513092041,
      "q_g0": 0.016530169174075127,
      "q_g1": 0.9834697842597961,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.636030912399292,
      "loss_g0": 10.062294006347656,
      "loss_g1": 3.5280139446258545,
      "q_g0": 0.016530843451619148,
      "q_g1": 0.9834691286087036,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.8522310256958,
      "loss_g0": 10.728742599487305,
      "loss_g1": 15.938347816467285,
      "q_g0": 0.016530316323041916,
      "q_g1": 0.9834696650505066,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.239286184310913,
      "loss_g0": 13.386370658874512,
      "loss_g1": 3.068721055984497,
      "q_g0": 0.016531378030776978,
      "q_g1": 0.9834686517715454,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.37376594543457,
      "loss_g0": 4.3555521965026855,
      "loss_g1": 4.374072074890137,
      "q_g0": 0.01653137244284153,
      "q_g1": 0.9834686517715454,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.625941276550293,
      "loss_g0": 6.323342323303223,
      "loss_g1": 10.698262214660645,
      "q_g0": 0.016530921682715416,
      "q_g1": 0.9834691286087036,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.2860493659973145,
      "loss_g0": 4.2860493659973145,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.0993311405181885,
      "loss_g0": 3.1316773891448975,
      "loss_g1": 1.1739486455917358,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.229969024658203,
      "loss_g0": 12.00260066986084,
      "loss_g1": 25.762615203857422,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523828729987144,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.3419132232666,
      "loss_g0": 22.300594329833984,
      "loss_g1": 24.80105972290039,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524076461791992,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.321857452392578,
      "loss_g0": 16.292835235595703,
      "loss_g1": 18.049118041992188,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652425527572632,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.549758911132812,
      "loss_g0": 26.58000946044922,
      "loss_g1": 24.749343872070312,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524069011211395,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 28.658498764038086,
      "loss_g0": 28.316246032714844,
      "loss_g1": 49.025856018066406,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.01652618497610092,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.360076904296875,
      "loss_g0": 12.236658096313477,
      "loss_g1": 19.704376220703125,
      "q_g0": 0.9834730625152588,
      "q_g1": 0.016526944935321808,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.76381492614746,
      "loss_g0": 20.803497314453125,
      "loss_g1": 18.402420043945312,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.016526702791452408,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.260499000549316,
      "loss_g0": 13.196332931518555,
      "loss_g1": 17.07880973815918,
      "q_g0": 0.9834729433059692,
      "q_g1": 0.016527099534869194,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.160078048706055,
      "loss_g0": 11.12088680267334,
      "loss_g1": 13.492115020751953,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.01652734912931919,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.174846649169922,
      "loss_g0": 20.385740280151367,
      "loss_g1": 7.624434947967529,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652604527771473,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.646204948425293,
      "loss_g0": 15.805025100708008,
      "loss_g1": 6.194194316864014,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525069251656532,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.992406845092773,
      "loss_g0": 23.024627685546875,
      "loss_g1": 21.074756622314453,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524873673915863,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.973886489868164,
      "loss_g0": 16.118497848510742,
      "loss_g1": 7.366878032684326,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523977741599083,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.2586088180542,
      "loss_g0": 9.18713665008545,
      "loss_g1": 13.512412071228027,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652441918849945,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.222124099731445,
      "loss_g0": 7.968729019165039,
      "loss_g1": 23.301830291748047,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525983810424805,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.2302141189575195,
      "loss_g0": 2.9613237380981445,
      "loss_g1": 19.230443954467773,
      "q_g0": 0.9834723472595215,
      "q_g1": 0.016527649015188217,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.365927219390869,
      "loss_g0": 6.246151924133301,
      "loss_g1": 13.492774963378906,
      "q_g0": 0.9834716320037842,
      "q_g1": 0.016528386622667313,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.696857452392578,
      "loss_g0": 22.696857452392578,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.655464172363281,
      "loss_g0": 13.759417533874512,
      "loss_g1": 7.46776819229126,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.45103931427002,
      "loss_g0": 15.474287033081055,
      "loss_g1": 14.06721305847168,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.78142261505127,
      "loss_g0": 8.820128440856934,
      "loss_g1": 6.477508544921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.561039924621582,
      "loss_g0": 11.479732513427734,
      "loss_g1": 16.400619506835938,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652292162179947,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.130850315093994,
      "loss_g0": 0.0,
      "loss_g1": 6.130850315093994,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.76525354385376,
      "loss_g0": 15.513035774230957,
      "loss_g1": 6.618290901184082,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.946954727172852,
      "loss_g0": 13.217021942138672,
      "loss_g1": 8.875215530395508,
      "q_g0": 0.016522862017154694,
      "q_g1": 0.9834771752357483,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.446269989013672,
      "loss_g0": 9.278468132019043,
      "loss_g1": 11.482688903808594,
      "q_g0": 0.016522636637091637,
      "q_g1": 0.9834774136543274,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.062509536743164,
      "loss_g0": 9.727766036987305,
      "loss_g1": 6.0009307861328125,
      "q_g0": 0.01652301475405693,
      "q_g1": 0.9834769368171692,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.115373611450195,
      "loss_g0": 0.0,
      "loss_g1": 4.115373611450195,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 17.039867401123047,
      "loss_g0": 7.815547943115234,
      "loss_g1": 17.194835662841797,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.6732661724090576,
      "loss_g0": 9.43936824798584,
      "loss_g1": 2.5595905780792236,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.904562473297119,
      "loss_g0": 6.904562473297119,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.58651065826416,
      "loss_g0": 4.353292465209961,
      "loss_g1": 18.468549728393555,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.140049934387207,
      "loss_g0": 10.144580841064453,
      "loss_g1": 9.870351791381836,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.143851280212402,
      "loss_g0": 12.085238456726074,
      "loss_g1": 15.632606506347656,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.421313285827637,
      "loss_g0": 4.445411682128906,
      "loss_g1": 2.986888885498047,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.7688400745391846,
      "loss_g0": 2.727573871612549,
      "loss_g1": 5.225089073181152,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522889956831932,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.857909679412842,
      "loss_g0": 6.832795143127441,
      "loss_g1": 8.352774620056152,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652303896844387,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.564186096191406,
      "loss_g0": 18.788564682006836,
      "loss_g1": 5.208322525024414,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.8700578212738037,
      "loss_g0": 0.0,
      "loss_g1": 3.8700578212738037,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.644935607910156,
      "loss_g0": 11.651506423950195,
      "loss_g1": 7.577625274658203,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.8345961570739746,
      "loss_g0": 3.281005859375,
      "loss_g1": 2.827096462249756,
      "q_g0": 0.016522465273737907,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.495474815368652,
      "loss_g0": 3.1794893741607666,
      "loss_g1": 6.551183223724365,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.019057273864746,
      "loss_g0": 36.44607162475586,
      "loss_g1": 8.558202743530273,
      "q_g0": 0.016525279730558395,
      "q_g1": 0.9834747910499573,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.268829345703125,
      "loss_g0": 3.802070140838623,
      "loss_g1": 11.394287109375,
      "q_g0": 0.016524503007531166,
      "q_g1": 0.9834754467010498,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.179780960083008,
      "loss_g0": 8.179780960083008,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.0963211059570312,
      "loss_g0": 3.0963211059570312,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.841108798980713,
      "loss_g0": 6.936102867126465,
      "loss_g1": 1.1866910457611084,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.348467469215393,
      "loss_g0": 1.345432162284851,
      "loss_g1": 1.5291434526443481,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.1113831996917725,
      "loss_g0": 1.8755820989608765,
      "loss_g1": 16.145883560180664,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652390882372856,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.0348968505859375,
      "loss_g0": 7.103904724121094,
      "loss_g1": 2.927539825439453,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523487865924835,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 23.624582290649414,
      "loss_g0": 23.669889450073242,
      "loss_g1": 20.92780876159668,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523201018571854,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.209840774536133,
      "loss_g0": 0.0,
      "loss_g1": 20.209840774536133,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 31.56574249267578,
      "loss_g0": 18.985904693603516,
      "loss_g1": 31.777082443237305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.220867156982422,
      "loss_g0": 25.164283752441406,
      "loss_g1": 18.10421371459961,
      "q_g0": 0.016523147001862526,
      "q_g1": 0.9834768176078796,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.74681854248047,
      "loss_g0": 61.96282958984375,
      "loss_g1": 22.087793350219727,
      "q_g0": 0.016527224332094193,
      "q_g1": 0.9834728240966797,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 45.34731674194336,
      "loss_g0": 26.964920043945312,
      "loss_g1": 45.65619659423828,
      "q_g0": 0.016525307670235634,
      "q_g1": 0.9834746718406677,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.833768844604492,
      "loss_g0": 20.419801712036133,
      "loss_g1": 25.92473602294922,
      "q_g0": 0.016524743288755417,
      "q_g1": 0.9834752082824707,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.361690044403076,
      "loss_g0": 20.060211181640625,
      "loss_g1": 7.148306846618652,
      "q_g0": 0.01652606949210167,
      "q_g1": 0.9834739565849304,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.834823608398438,
      "loss_g0": 0.0,
      "loss_g1": 21.834823608398438,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 43.52156448364258,
      "loss_g0": 18.12120246887207,
      "loss_g1": 43.94828796386719,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 18.18248748779297,
      "loss_g0": 58.01043701171875,
      "loss_g1": 17.513206481933594,
      "q_g0": 0.016526563093066216,
      "q_g1": 0.9834733605384827,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.15694236755371,
      "loss_g0": 26.225805282592773,
      "loss_g1": 25.138978958129883,
      "q_g0": 0.01652667298913002,
      "q_g1": 0.9834733605384827,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.425323486328125,
      "loss_g0": 14.621097564697266,
      "loss_g1": 11.371620178222656,
      "q_g0": 0.016526998952031136,
      "q_g1": 0.9834730625152588,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.065641403198242,
      "loss_g0": 0.0,
      "loss_g1": 14.065641403198242,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.767720222473145,
      "loss_g0": 14.767720222473145,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.65009880065918,
      "loss_g0": 19.66883087158203,
      "loss_g1": 18.534996032714844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.191425323486328,
      "loss_g0": 19.250286102294922,
      "loss_g1": 15.68785285949707,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.208362579345703,
      "loss_g0": 24.439687728881836,
      "loss_g1": 10.43901538848877,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.342118263244629,
      "loss_g0": 14.320067405700684,
      "loss_g1": 15.654670715332031,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522565856575966,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.689047813415527,
      "loss_g0": 7.632121562957764,
      "loss_g1": 11.077444076538086,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522914171218872,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.437809944152832,
      "loss_g0": 14.459956169128418,
      "loss_g1": 13.119608879089355,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652277447283268,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.285317420959473,
      "loss_g0": 0.0,
      "loss_g1": 15.285317420959473,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.002630233764648,
      "loss_g0": 29.801982879638672,
      "loss_g1": 11.703600883483887,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.528249740600586,
      "loss_g0": 12.871237754821777,
      "loss_g1": 8.45528507232666,
      "q_g0": 0.01652287319302559,
      "q_g1": 0.9834770560264587,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.991368293762207,
      "loss_g0": 28.244308471679688,
      "loss_g1": 11.718282699584961,
      "q_g0": 0.01652456447482109,
      "q_g1": 0.9834754467010498,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.7129241228103638,
      "loss_g0": 12.306219100952148,
      "loss_g1": 1.5349212884902954,
      "q_g0": 0.016525665298104286,
      "q_g1": 0.9834743142127991,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.570100784301758,
      "loss_g0": 12.057812690734863,
      "loss_g1": 9.528298377990723,
      "q_g0": 0.016525927931070328,
      "q_g1": 0.98347407579422,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.031147480010986,
      "loss_g0": 8.552281379699707,
      "loss_g1": 5.9887824058532715,
      "q_g0": 0.01652619242668152,
      "q_g1": 0.9834738373756409,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.347447395324707,
      "loss_g0": 13.616169929504395,
      "loss_g1": 8.258909225463867,
      "q_g0": 0.01652674376964569,
      "q_g1": 0.9834732413291931,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.003228187561035,
      "loss_g0": 14.915596008300781,
      "loss_g1": 11.954286575317383,
      "q_g0": 0.016527045518159866,
      "q_g1": 0.9834729433059692,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.948254585266113,
      "loss_g0": 17.547060012817383,
      "loss_g1": 8.80374526977539,
      "q_g0": 0.016527937725186348,
      "q_g1": 0.9834721088409424,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.092445373535156,
      "loss_g0": 28.43598747253418,
      "loss_g1": 13.85136890411377,
      "q_g0": 0.016529425978660583,
      "q_g1": 0.983470618724823,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.1925687789917,
      "loss_g0": 3.7718682289123535,
      "loss_g1": 10.3004789352417,
      "q_g0": 0.01652875542640686,
      "q_g1": 0.9834712147712708,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.14262580871582,
      "loss_g0": 20.77895736694336,
      "loss_g1": 7.930236339569092,
      "q_g0": 0.016530068591237068,
      "q_g1": 0.9834699034690857,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.588783264160156,
      "loss_g0": 5.588783264160156,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 27.33222198486328,
      "loss_g0": 27.705631256103516,
      "loss_g1": 5.105477809906006,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.667332649230957,
      "loss_g0": 13.667332649230957,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.146401405334473,
      "loss_g0": 14.210592269897461,
      "loss_g1": 10.325540542602539,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.9060158729553223,
      "loss_g0": 3.8677759170532227,
      "loss_g1": 6.182172775268555,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522666439414024,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 26.427331924438477,
      "loss_g0": 26.767236709594727,
      "loss_g1": 6.194911479949951,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.5191501379013062,
      "loss_g0": 1.4206830263137817,
      "loss_g1": 7.380067348480225,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523031517863274,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.274495124816895,
      "loss_g0": 9.35191822052002,
      "loss_g1": 4.6659674644470215,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.477914810180664,
      "loss_g0": 6.515311241149902,
      "loss_g1": 4.251931667327881,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 12.52248477935791,
      "loss_g0": 12.686169624328613,
      "loss_g1": 2.7793474197387695,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.2777791023254395,
      "loss_g0": 6.3044633865356445,
      "loss_g1": 4.689431190490723,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.453317642211914,
      "loss_g0": 4.451440811157227,
      "loss_g1": 4.5650177001953125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.102201461791992,
      "loss_g0": 13.293120384216309,
      "loss_g1": 1.7379436492919922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.863561153411865,
      "loss_g0": 6.9004926681518555,
      "loss_g1": 4.665252685546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.74479866027832,
      "loss_g0": 22.052101135253906,
      "loss_g1": 3.452943801879883,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.242051601409912,
      "loss_g0": 5.178516387939453,
      "loss_g1": 9.023831367492676,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522813588380814,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.84465980529785,
      "loss_g0": 22.81603240966797,
      "loss_g1": 24.548582077026367,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652299053966999,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 24.218841552734375,
      "loss_g0": 24.218841552734375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.01251220703125,
      "loss_g0": 15.943120956420898,
      "loss_g1": 20.14295768737793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 31.27674102783203,
      "loss_g0": 31.592926025390625,
      "loss_g1": 12.45616626739502,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.40060806274414,
      "loss_g0": 0.0,
      "loss_g1": 16.40060806274414,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.2373046875,
      "loss_g0": 0.0,
      "loss_g1": 16.2373046875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.856929302215576,
      "loss_g0": 18.876638412475586,
      "loss_g1": 6.654998302459717,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.092506408691406,
      "loss_g0": 25.311237335205078,
      "loss_g1": 8.820003509521484,
      "q_g0": 0.01652410998940468,
      "q_g1": 0.983475923538208,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.003124237060547,
      "loss_g0": 19.003124237060547,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.469492435455322,
      "loss_g0": 6.318056106567383,
      "loss_g1": 15.483572006225586,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.484973907470703,
      "loss_g0": 25.66644287109375,
      "loss_g1": 14.683233261108398,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.976989507675171,
      "loss_g0": 2.8182525634765625,
      "loss_g1": 12.425039291381836,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652340218424797,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 23.399402618408203,
      "loss_g0": 23.553726196289062,
      "loss_g1": 14.213504791259766,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 36.29656982421875,
      "loss_g0": 36.420108795166016,
      "loss_g1": 28.94318199157715,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.893696784973145,
      "loss_g0": 14.065410614013672,
      "loss_g1": 3.672593832015991,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.814926147460938,
      "loss_g0": 22.762920379638672,
      "loss_g1": 25.910341262817383,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522744670510292,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.915711402893066,
      "loss_g0": 11.991264343261719,
      "loss_g1": 7.418519496917725,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.230491638183594,
      "loss_g0": 4.829001426696777,
      "loss_g1": 29.12505340576172,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.0165249053388834,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.002387046813965,
      "loss_g0": 13.093006134033203,
      "loss_g1": 7.609014987945557,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524342820048332,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.208367347717285,
      "loss_g0": 14.898574829101562,
      "loss_g1": 33.64405822753906,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.01652626134455204,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.408735752105713,
      "loss_g0": 5.253111362457275,
      "loss_g1": 14.669355392456055,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.016527224332094193,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.89113998413086,
      "loss_g0": 13.96634292602539,
      "loss_g1": 9.415976524353027,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.016526753082871437,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.14520835876465,
      "loss_g0": 19.2819766998291,
      "loss_g1": 11.006063461303711,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525907441973686,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.243873119354248,
      "loss_g0": 7.282993316650391,
      "loss_g1": 4.915777206420898,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.016525665298104286,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.29532241821289,
      "loss_g0": 10.31155776977539,
      "loss_g1": 9.329154014587402,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.016525564715266228,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 16.440698623657227,
      "loss_g0": 16.528974533081055,
      "loss_g1": 11.18702220916748,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.0165250226855278,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.849592208862305,
      "loss_g0": 14.922074317932129,
      "loss_g1": 10.53577995300293,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524571925401688,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.879212856292725,
      "loss_g0": 6.557904243469238,
      "loss_g1": 25.99987030029297,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526563093066216,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 13.85627269744873,
      "loss_g0": 13.990673065185547,
      "loss_g1": 5.857893943786621,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.01652572862803936,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.194612503051758,
      "loss_g0": 3.157672882080078,
      "loss_g1": 5.392913818359375,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525959596037865,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.952786445617676,
      "loss_g0": 7.796294212341309,
      "loss_g1": 17.265243530273438,
      "q_g0": 0.9834730625152588,
      "q_g1": 0.016526928171515465,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.599479675292969,
      "loss_g0": 9.456542015075684,
      "loss_g1": 18.10487174987793,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.0165278110653162,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 9.217549324035645,
      "loss_g0": 9.217549324035645,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.133946418762207,
      "loss_g0": 0.0,
      "loss_g1": 10.133946418762207,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.022705554962158,
      "loss_g0": 11.80186939239502,
      "loss_g1": 4.908815383911133,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 7.691764831542969,
      "loss_g0": 46.67730712890625,
      "loss_g1": 7.036643981933594,
      "q_g0": 0.0165264792740345,
      "q_g1": 0.9834734797477722,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 10.849719047546387,
      "loss_g0": 6.723748683929443,
      "loss_g1": 10.919050216674805,
      "q_g0": 0.016526052728295326,
      "q_g1": 0.9834739565849304,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.1329023838043213,
      "loss_g0": 2.5343639850616455,
      "loss_g1": 3.1429598331451416,
      "q_g0": 0.0165259912610054,
      "q_g1": 0.9834739565849304,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.955744504928589,
      "loss_g0": 8.700626373291016,
      "loss_g1": 2.859205722808838,
      "q_g0": 0.016526587307453156,
      "q_g1": 0.9834733605384827,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.3831018209457397,
      "loss_g0": 2.8920230865478516,
      "loss_g1": 1.3577451705932617,
      "q_g0": 0.016526753082871437,
      "q_g1": 0.9834732413291931,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.4082722663879395,
      "loss_g0": 13.335943222045898,
      "loss_g1": 5.275044918060303,
      "q_g0": 0.016527581959962845,
      "q_g1": 0.983472466468811,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 3.921100378036499,
      "loss_g0": 1.5811747312545776,
      "loss_g1": 3.960422992706299,
      "q_g0": 0.016527339816093445,
      "q_g1": 0.9834727048873901,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.824779033660889,
      "loss_g0": 6.5312676429748535,
      "loss_g1": 6.829710960388184,
      "q_g0": 0.016527308151125908,
      "q_g1": 0.9834727048873901,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 1.6151435375213623,
      "loss_g0": 1.5872461795806885,
      "loss_g1": 1.615612268447876,
      "q_g0": 0.016527308151125908,
      "q_g1": 0.9834727048873901,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.048675537109375,
      "loss_g0": 11.048675537109375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.99495792388916,
      "loss_g0": 9.109496116638184,
      "loss_g1": 2.1771953105926514,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 32.978363037109375,
      "loss_g0": 33.396087646484375,
      "loss_g1": 8.113794326782227,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 0.9046645164489746,
      "loss_g0": 0.8357358574867249,
      "loss_g1": 5.007452964782715,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522854566574097,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 14.992152214050293,
      "loss_g0": 15.210205078125,
      "loss_g1": 2.012784481048584,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.272329330444336,
      "loss_g0": 6.272329330444336,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 20.439090728759766,
      "loss_g0": 20.476104736328125,
      "loss_g1": 18.235815048217773,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.290167808532715,
      "loss_g0": 10.984286308288574,
      "loss_g1": 29.495311737060547,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524318605661392,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 25.32452964782715,
      "loss_g0": 25.474210739135742,
      "loss_g1": 16.415422439575195,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523394733667374,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 17.925880432128906,
      "loss_g0": 18.00756072998047,
      "loss_g1": 13.064130783081055,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 30.111209869384766,
      "loss_g0": 30.46485710144043,
      "loss_g1": 9.060779571533203,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 15.399173736572266,
      "loss_g0": 15.366945266723633,
      "loss_g1": 17.317481994628906,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522621735930443,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.123842239379883,
      "loss_g0": 21.34853172302246,
      "loss_g1": 7.749532699584961,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 23.776134490966797,
      "loss_g0": 23.99498748779297,
      "loss_g1": 10.749175071716309,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 11.790287971496582,
      "loss_g0": 11.787003517150879,
      "loss_g1": 11.985818862915039,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.231181144714355,
      "loss_g0": 8.240124702453613,
      "loss_g1": 7.698810577392578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 34.991432189941406,
      "loss_g0": 35.33158874511719,
      "loss_g1": 14.743988037109375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.739433288574219,
      "loss_g0": 8.7305326461792,
      "loss_g1": 9.269213676452637,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522472724318504,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 19.275665283203125,
      "loss_g0": 19.44449806213379,
      "loss_g1": 9.226102828979492,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 6.043360710144043,
      "loss_g0": 6.047173976898193,
      "loss_g1": 5.8163957595825195,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 8.267596244812012,
      "loss_g0": 0.0,
      "loss_g1": 8.267596244812012,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 4.802558898925781,
      "loss_g0": 8.186182022094727,
      "loss_g1": 4.74571418762207,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 2.9758472442626953,
      "loss_g0": 2.9758472442626953,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 5.882727146148682,
      "loss_g0": 5.947596073150635,
      "loss_g1": 2.0214791297912598,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.16274642944336,
      "loss_g0": 21.48111915588379,
      "loss_g1": 2.2119107246398926,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "eval_cer": 0.06244395055951963,
      "eval_loss": 13.980732917785645,
      "eval_runtime": 80.8124,
      "eval_samples_per_second": 20.739,
      "eval_steps_per_second": 5.185,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 21.666641235351562,
      "loss_g0": 21.598247528076172,
      "loss_g1": 25.737585067749023,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522854566574097,
      "step": 700
    },
    {
      "epoch": 3.553299492385787,
      "gdro_loss": 22.13494873046875,
      "loss_g0": 21.67767333984375,
      "loss_g1": 49.34828567504883,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.01652568019926548,
      "step": 700
    },
    {
      "epoch": 4.060913705583756,
      "grad_norm": 57.51197814941406,
      "learning_rate": 0.00029075935828877004,
      "loss": 23.3068,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 26.93155288696289,
      "loss_g0": 26.989225387573242,
      "loss_g1": 23.521480560302734,
      "q_g0": 0.9833695292472839,
      "q_g1": 0.016630545258522034,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 33.462589263916016,
      "loss_g0": 33.44959259033203,
      "loss_g1": 34.23097610473633,
      "q_g0": 0.9833694100379944,
      "q_g1": 0.0166306309401989,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 24.185068130493164,
      "loss_g0": 24.154836654663086,
      "loss_g1": 25.972644805908203,
      "q_g0": 0.9833691716194153,
      "q_g1": 0.01663082465529442,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 41.86326599121094,
      "loss_g0": 42.415462493896484,
      "loss_g1": 9.205294609069824,
      "q_g0": 0.9833726286888123,
      "q_g1": 0.01662740297615528,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 22.759132385253906,
      "loss_g0": 22.705215454101562,
      "loss_g1": 25.94780921936035,
      "q_g0": 0.9833722710609436,
      "q_g1": 0.016627738252282143,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 41.02040100097656,
      "loss_g0": 41.376617431640625,
      "loss_g1": 19.95075225830078,
      "q_g0": 0.9833744764328003,
      "q_g1": 0.016625531017780304,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.207788467407227,
      "loss_g0": 18.1980037689209,
      "loss_g1": 18.78655433654785,
      "q_g0": 0.9833744764328003,
      "q_g1": 0.016625594347715378,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.479707717895508,
      "loss_g0": 23.673526763916016,
      "loss_g1": 12.01479721069336,
      "q_g0": 0.983375608921051,
      "q_g1": 0.016624391078948975,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 79.2611083984375,
      "loss_g0": 80.20759582519531,
      "loss_g1": 23.253952026367188,
      "q_g0": 0.9833815097808838,
      "q_g1": 0.01661854051053524,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 40.38020706176758,
      "loss_g0": 40.530479431152344,
      "loss_g1": 31.487560272216797,
      "q_g0": 0.9833824038505554,
      "q_g1": 0.016617612913250923,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 26.303707122802734,
      "loss_g0": 26.29262924194336,
      "loss_g1": 26.959335327148438,
      "q_g0": 0.9833822846412659,
      "q_g1": 0.01661768928170204,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.1382174491882324,
      "loss_g0": 1.9242740869522095,
      "loss_g1": 14.797682762145996,
      "q_g0": 0.9833810329437256,
      "q_g1": 0.016619008034467697,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 31.767168045043945,
      "loss_g0": 32.157413482666016,
      "loss_g1": 8.672201156616211,
      "q_g0": 0.9833834767341614,
      "q_g1": 0.016616594046354294,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.06216049194336,
      "loss_g0": 0.0,
      "loss_g1": 21.06216049194336,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.007933616638184,
      "loss_g0": 5.976823329925537,
      "loss_g1": 13.126055717468262,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.039022445678711,
      "loss_g0": 19.187469482421875,
      "loss_g1": 14.96932601928711,
      "q_g0": 0.016522854566574097,
      "q_g1": 0.9834771752357483,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.336225509643555,
      "loss_g0": 14.313189506530762,
      "loss_g1": 8.235806465148926,
      "q_g0": 0.01652348041534424,
      "q_g1": 0.9834765195846558,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.303062438964844,
      "loss_g0": 22.739086151123047,
      "loss_g1": 11.110910415649414,
      "q_g0": 0.01652466505765915,
      "q_g1": 0.9834753274917603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.538740158081055,
      "loss_g0": 31.631393432617188,
      "loss_g1": 15.268319129943848,
      "q_g0": 0.01652633771300316,
      "q_g1": 0.9834735989570618,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.720316886901855,
      "loss_g0": 0.0,
      "loss_g1": 8.720316886901855,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.247221946716309,
      "loss_g0": 42.776885986328125,
      "loss_g1": 12.751123428344727,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.40649700164795,
      "loss_g0": 8.640798568725586,
      "loss_g1": 11.452960968017578,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.25825834274292,
      "loss_g0": 6.425841331481934,
      "loss_g1": 4.221842288970947,
      "q_g0": 0.01652265153825283,
      "q_g1": 0.9834772944450378,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.678317070007324,
      "loss_g0": 34.64707946777344,
      "loss_g1": 6.208351135253906,
      "q_g0": 0.01652555726468563,
      "q_g1": 0.9834744334220886,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.823740005493164,
      "loss_g0": 12.108710289001465,
      "loss_g1": 12.835753440856934,
      "q_g0": 0.01652548834681511,
      "q_g1": 0.9834745526313782,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.741267204284668,
      "loss_g0": 17.56785011291504,
      "loss_g1": 6.559333801269531,
      "q_g0": 0.016526611521840096,
      "q_g1": 0.9834733605384827,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.902769088745117,
      "loss_g0": 11.137049674987793,
      "loss_g1": 16.99965476989746,
      "q_g0": 0.01652601547539234,
      "q_g1": 0.9834739565849304,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.836772918701172,
      "loss_g0": 17.2576904296875,
      "loss_g1": 6.661650657653809,
      "q_g0": 0.016527099534869194,
      "q_g1": 0.9834729433059692,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.911773681640625,
      "loss_g0": 27.65782356262207,
      "loss_g1": 2.495854377746582,
      "q_g0": 0.01652967371046543,
      "q_g1": 0.9834703803062439,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 27.0734920501709,
      "loss_g0": 6.025113105773926,
      "loss_g1": 27.427215576171875,
      "q_g0": 0.016527481377124786,
      "q_g1": 0.9834725856781006,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.4757280349731445,
      "loss_g0": 0.0,
      "loss_g1": 7.4757280349731445,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.49550724029541,
      "loss_g0": 18.329103469848633,
      "loss_g1": 14.431102752685547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.988535165786743,
      "loss_g0": 6.393085479736328,
      "loss_g1": 2.931337356567383,
      "q_g0": 0.01652277447283268,
      "q_g1": 0.9834771752357483,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.248020648956299,
      "loss_g0": 6.140954494476318,
      "loss_g1": 4.2162184715271,
      "q_g0": 0.016522977501153946,
      "q_g1": 0.9834770560264587,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.172731399536133,
      "loss_g0": 11.3103666305542,
      "loss_g1": 16.254419326782227,
      "q_g0": 0.016522465273737907,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.477058410644531,
      "loss_g0": 17.82564353942871,
      "loss_g1": 7.303189754486084,
      "q_g0": 0.016523534432053566,
      "q_g1": 0.9834765195846558,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.611992835998535,
      "loss_g0": 23.355388641357422,
      "loss_g1": 11.414674758911133,
      "q_g0": 0.016524752601981163,
      "q_g1": 0.9834752082824707,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.195327758789062,
      "loss_g0": 3.780691623687744,
      "loss_g1": 8.269502639770508,
      "q_g0": 0.016524292528629303,
      "q_g1": 0.9834756851196289,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.1305603981018066,
      "loss_g0": 8.737756729125977,
      "loss_g1": 1.00273859500885,
      "q_g0": 0.01652507483959198,
      "q_g1": 0.983474850654602,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.2067760229110718,
      "loss_g0": 22.899723052978516,
      "loss_g1": 0.8422244191169739,
      "q_g0": 0.016527332365512848,
      "q_g1": 0.9834727048873901,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.467659950256348,
      "loss_g0": 13.235902786254883,
      "loss_g1": 8.387526512145996,
      "q_g0": 0.016527827829122543,
      "q_g1": 0.9834722280502319,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.658318519592285,
      "loss_g0": 14.971603393554688,
      "loss_g1": 3.4681785106658936,
      "q_g0": 0.01652899757027626,
      "q_g1": 0.9834709763526917,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.93947982788086,
      "loss_g0": 15.405593872070312,
      "loss_g1": 10.864416122436523,
      "q_g0": 0.01652945578098297,
      "q_g1": 0.9834704995155334,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.920301675796509,
      "loss_g0": 1.9116102457046509,
      "loss_g1": 2.9372549057006836,
      "q_g0": 0.016529347747564316,
      "q_g1": 0.983470618724823,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.084044933319092,
      "loss_g0": 0.0,
      "loss_g1": 6.084044933319092,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.0752756595611572,
      "loss_g0": 4.854315757751465,
      "loss_g1": 2.028587818145752,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.391902446746826,
      "loss_g0": 14.474418640136719,
      "loss_g1": 5.239306926727295,
      "q_g0": 0.016523364931344986,
      "q_g1": 0.9834766387939453,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.092621803283691,
      "loss_g0": 5.558088302612305,
      "loss_g1": 9.15200424194336,
      "q_g0": 0.016522999852895737,
      "q_g1": 0.9834769368171692,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 0.5804879069328308,
      "loss_g0": 2.2206482887268066,
      "loss_g1": 0.5529319643974304,
      "q_g0": 0.016523171216249466,
      "q_g1": 0.9834768176078796,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.673006057739258,
      "loss_g0": 0.4615500271320343,
      "loss_g1": 3.7269599437713623,
      "q_g0": 0.016522839665412903,
      "q_g1": 0.9834771752357483,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.469825744628906,
      "loss_g0": 51.84441375732422,
      "loss_g1": 12.824957847595215,
      "q_g0": 0.016526827588677406,
      "q_g1": 0.9834731221199036,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 22.041885375976562,
      "loss_g0": 22.23468017578125,
      "loss_g1": 22.038644790649414,
      "q_g0": 0.01652684435248375,
      "q_g1": 0.9834731221199036,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 41.13517379760742,
      "loss_g0": 18.834070205688477,
      "loss_g1": 41.5098762512207,
      "q_g0": 0.01652451977133751,
      "q_g1": 0.9834754467010498,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.44502830505371,
      "loss_g0": 29.39192771911621,
      "loss_g1": 20.294692993164062,
      "q_g0": 0.016525447368621826,
      "q_g1": 0.9834745526313782,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.284902572631836,
      "loss_g0": 25.144729614257812,
      "loss_g1": 23.253650665283203,
      "q_g0": 0.016525642946362495,
      "q_g1": 0.9834743142127991,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 28.189952850341797,
      "loss_g0": 13.865209579467773,
      "loss_g1": 28.430633544921875,
      "q_g0": 0.01652415469288826,
      "q_g1": 0.9834758043289185,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 36.21457290649414,
      "loss_g0": 15.58871841430664,
      "loss_g1": 36.56108474731445,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.370853424072266,
      "loss_g0": 19.34642219543457,
      "loss_g1": 7.16964864730835,
      "q_g0": 0.01652366667985916,
      "q_g1": 0.9834764003753662,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.666013717651367,
      "loss_g0": 12.82777214050293,
      "loss_g1": 14.696898460388184,
      "q_g0": 0.01652347296476364,
      "q_g1": 0.9834765195846558,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 27.594310760498047,
      "loss_g0": 6.269388675689697,
      "loss_g1": 27.95256805419922,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.01882266998291,
      "loss_g0": 23.39927101135254,
      "loss_g1": 12.844419479370117,
      "q_g0": 0.01652350276708603,
      "q_g1": 0.9834765195846558,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.32482147216797,
      "loss_g0": 25.034809112548828,
      "loss_g1": 20.245685577392578,
      "q_g0": 0.016523992642760277,
      "q_g1": 0.9834760427474976,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.347942352294922,
      "loss_g0": 28.570920944213867,
      "loss_g1": 12.075342178344727,
      "q_g0": 0.016525674611330032,
      "q_g1": 0.9834743142127991,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.747867107391357,
      "loss_g0": 23.048599243164062,
      "loss_g1": 6.473931312561035,
      "q_g0": 0.016527369618415833,
      "q_g1": 0.9834725856781006,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.170836448669434,
      "loss_g0": 13.26753044128418,
      "loss_g1": 11.135600090026855,
      "q_g0": 0.016527589410543442,
      "q_g1": 0.983472466468811,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 32.961387634277344,
      "loss_g0": 7.619489669799805,
      "loss_g1": 33.38719940185547,
      "q_g0": 0.016524961218237877,
      "q_g1": 0.9834749698638916,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.784152030944824,
      "loss_g0": 6.888633728027344,
      "loss_g1": 10.849605560302734,
      "q_g0": 0.016524557024240494,
      "q_g1": 0.9834754467010498,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.007262229919434,
      "loss_g0": 0.0,
      "loss_g1": 10.007262229919434,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.370498657226562,
      "loss_g0": 17.581274032592773,
      "loss_g1": 21.43415641784668,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.42519474029541,
      "loss_g0": 7.39846658706665,
      "loss_g1": 14.543243408203125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.19482421875,
      "loss_g0": 7.558570861816406,
      "loss_g1": 17.356714248657227,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.083307266235352,
      "loss_g0": 9.083307266235352,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.1966564655303955,
      "loss_g0": 1.0836819410324097,
      "loss_g1": 7.921329021453857,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.130079746246338,
      "loss_g0": 5.108778953552246,
      "loss_g1": 6.39794921875,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.415347099304199,
      "loss_g0": 0.0,
      "loss_g1": 5.415347099304199,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.666057586669922,
      "loss_g0": 31.7315731048584,
      "loss_g1": 11.328956604003906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 27.19499397277832,
      "loss_g0": 5.0342559814453125,
      "loss_g1": 27.56729507446289,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.290945053100586,
      "loss_g0": 10.385058403015137,
      "loss_g1": 9.272562980651855,
      "q_g0": 0.01652253419160843,
      "q_g1": 0.9834774136543274,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.8088486194610596,
      "loss_g0": 2.455876350402832,
      "loss_g1": 3.831578493118286,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.744976043701172,
      "loss_g0": 10.572372436523438,
      "loss_g1": 9.731075286865234,
      "q_g0": 0.01652250438928604,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.22275733947754,
      "loss_g0": 15.356722831726074,
      "loss_g1": 23.35490608215332,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.937472820281982,
      "loss_g0": 5.186655044555664,
      "loss_g1": 7.983686447143555,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.576292037963867,
      "loss_g0": 3.7198948860168457,
      "loss_g1": 18.825878143310547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.247053146362305,
      "loss_g0": 14.556320190429688,
      "loss_g1": 7.124251842498779,
      "q_g0": 0.01652318611741066,
      "q_g1": 0.9834768176078796,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.604619026184082,
      "loss_g0": 13.49951457977295,
      "loss_g1": 15.639986038208008,
      "q_g0": 0.016522971913218498,
      "q_g1": 0.9834770560264587,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.542070388793945,
      "loss_g0": 18.0024356842041,
      "loss_g1": 8.383121490478516,
      "q_g0": 0.016523951664566994,
      "q_g1": 0.9834760427474976,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.731956481933594,
      "loss_g0": 12.74095344543457,
      "loss_g1": 7.647794723510742,
      "q_g0": 0.01652447320520878,
      "q_g1": 0.9834755659103394,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.74858570098877,
      "loss_g0": 6.435829162597656,
      "loss_g1": 10.82104778289795,
      "q_g0": 0.016524024307727814,
      "q_g1": 0.9834760427474976,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.76801586151123,
      "loss_g0": 6.011841773986816,
      "loss_g1": 8.814323425292969,
      "q_g0": 0.016523728147149086,
      "q_g1": 0.9834762811660767,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.093213081359863,
      "loss_g0": 8.566651344299316,
      "loss_g1": 14.186063766479492,
      "q_g0": 0.016523154452443123,
      "q_g1": 0.9834768176078796,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.057221412658691,
      "loss_g0": 4.601560592651367,
      "loss_g1": 12.182476043701172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.576745510101318,
      "loss_g0": 7.576745510101318,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.034919261932373,
      "loss_g0": 3.9202933311462402,
      "loss_g1": 10.857892036437988,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.551803588867188,
      "loss_g0": 10.609732627868652,
      "loss_g1": 7.103682041168213,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.990165710449219,
      "loss_g0": 5.049563884735107,
      "loss_g1": 1.4545477628707886,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.45755672454834,
      "loss_g0": 8.570502281188965,
      "loss_g1": 1.7345781326293945,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.533611297607422,
      "loss_g0": 21.879505157470703,
      "loss_g1": 0.9446567296981812,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 24.6639404296875,
      "loss_g0": 25.026281356811523,
      "loss_g1": 3.0959596633911133,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.16618013381958,
      "loss_g0": 5.238439559936523,
      "loss_g1": 0.8650340437889099,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.7955427169799805,
      "loss_g0": 5.887454509735107,
      "loss_g1": 0.32458242774009705,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 86.134033203125,
      "loss_g0": 86.76477813720703,
      "loss_g1": 48.58982467651367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 30.191055297851562,
      "loss_g0": 30.132322311401367,
      "loss_g1": 33.686920166015625,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.439602375030518,
      "loss_g0": 5.1079487800598145,
      "loss_g1": 25.177974700927734,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.01652483642101288,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 19.741443634033203,
      "loss_g0": 19.922216415405273,
      "loss_g1": 8.982003211975098,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523711383342743,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 36.02816390991211,
      "loss_g0": 36.217041015625,
      "loss_g1": 24.785411834716797,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522543504834175,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 43.2104606628418,
      "loss_g0": 43.6107063293457,
      "loss_g1": 19.386192321777344,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.839953422546387,
      "loss_g0": 15.734763145446777,
      "loss_g1": 22.101043701171875,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523078083992004,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 29.49146270751953,
      "loss_g0": 29.4531307220459,
      "loss_g1": 31.773029327392578,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523310914635658,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.148031234741211,
      "loss_g0": 15.343230247497559,
      "loss_g1": 3.5289769172668457,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.814143180847168,
      "loss_g0": 14.793521881103516,
      "loss_g1": 16.04161834716797,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.751483917236328,
      "loss_g0": 23.671857833862305,
      "loss_g1": 28.49103355407715,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523046419024467,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.045792579650879,
      "loss_g0": 15.161097526550293,
      "loss_g1": 8.182360649108887,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.33192253112793,
      "loss_g0": 18.307369232177734,
      "loss_g1": 19.793413162231445,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.06080150604248,
      "loss_g0": 7.871100425720215,
      "loss_g1": 19.351587295532227,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652374491095543,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.46439552307129,
      "loss_g0": 20.702632904052734,
      "loss_g1": 6.283616542816162,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.819658279418945,
      "loss_g0": 20.81694793701172,
      "loss_g1": 20.98096466064453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.197848320007324,
      "loss_g0": 8.273344039916992,
      "loss_g1": 3.704075813293457,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.276420593261719,
      "loss_g0": 7.092026710510254,
      "loss_g1": 18.25147247314453,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523564234375954,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.42896842956543,
      "loss_g0": 0.0,
      "loss_g1": 8.42896842956543,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.539356231689453,
      "loss_g0": 18.539356231689453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.06623649597168,
      "loss_g0": 16.06623649597168,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.994774580001831,
      "loss_g0": 3.917961359024048,
      "loss_g1": 8.566991806030273,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.698786735534668,
      "loss_g0": 14.786001205444336,
      "loss_g1": 9.507423400878906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.640522003173828,
      "loss_g0": 20.924053192138672,
      "loss_g1": 3.7636449337005615,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.78494930267334,
      "loss_g0": 15.825392723083496,
      "loss_g1": 13.377592086791992,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.500008583068848,
      "loss_g0": 10.488569259643555,
      "loss_g1": 11.180912971496582,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522489488124847,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.73660945892334,
      "loss_g0": 3.6889841556549072,
      "loss_g1": 6.571385383605957,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.2559356689453125,
      "loss_g0": 5.212927341461182,
      "loss_g1": 7.815850257873535,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523055732250214,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.555964469909668,
      "loss_g0": 8.505080223083496,
      "loss_g1": 11.584617614746094,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523372381925583,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.537325859069824,
      "loss_g0": 11.505496978759766,
      "loss_g1": 13.431781768798828,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.0165235735476017,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 29.902202606201172,
      "loss_g0": 30.334815979003906,
      "loss_g1": 4.15133810043335,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.1748087406158447,
      "loss_g0": 3.1348416805267334,
      "loss_g1": 5.553763389587402,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522666439414024,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.940332412719727,
      "loss_g0": 24.253883361816406,
      "loss_g1": 5.276534080505371,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.383995056152344,
      "loss_g0": 13.302573204040527,
      "loss_g1": 18.23044204711914,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652292162179947,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.01006507873535,
      "loss_g0": 20.246620178222656,
      "loss_g1": 5.929324626922607,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.94732666015625,
      "loss_g0": 14.060827255249023,
      "loss_g1": 7.191361427307129,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.653932571411133,
      "loss_g0": 10.569002151489258,
      "loss_g1": 15.709178924560547,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652294583618641,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.549330711364746,
      "loss_g0": 12.72123908996582,
      "loss_g1": 2.3166451454162598,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.751053810119629,
      "loss_g0": 14.929398536682129,
      "loss_g1": 4.135313987731934,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.86868953704834,
      "loss_g0": 5.9224653244018555,
      "loss_g1": 2.667726993560791,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.294412136077881,
      "loss_g0": 6.372463226318359,
      "loss_g1": 1.6485211849212646,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.1850557327270508,
      "loss_g0": 1.0608958005905151,
      "loss_g1": 8.575172424316406,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523193567991257,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.898595809936523,
      "loss_g0": 14.04730224609375,
      "loss_g1": 5.047036647796631,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.12756633758545,
      "loss_g0": 11.080667495727539,
      "loss_g1": 13.919113159179688,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522713005542755,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.037962913513184,
      "loss_g0": 8.037962913513184,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.483251571655273,
      "loss_g0": 4.547798156738281,
      "loss_g1": 0.6411903500556946,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.715874671936035,
      "loss_g0": 9.83985424041748,
      "loss_g1": 2.3361268043518066,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.367887496948242,
      "loss_g0": 7.487721920013428,
      "loss_g1": 0.2348891645669937,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.098768472671509,
      "loss_g0": 2.0739359855651855,
      "loss_g1": 3.576881170272827,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.249426364898682,
      "loss_g0": 4.266626358032227,
      "loss_g1": 3.2256059646606445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522472724318504,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 25.898759841918945,
      "loss_g0": 26.052043914794922,
      "loss_g1": 16.77469253540039,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 19.083030700683594,
      "loss_g0": 0.0,
      "loss_g1": 19.083030700683594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.348771095275879,
      "loss_g0": 29.714740753173828,
      "loss_g1": 10.023423194885254,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.530158042907715,
      "loss_g0": 11.390824317932129,
      "loss_g1": 15.599699020385742,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.251562118530273,
      "loss_g0": 18.920883178710938,
      "loss_g1": 20.27391815185547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 25.4081974029541,
      "loss_g0": 13.395204544067383,
      "loss_g1": 25.610015869140625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 43.47294616699219,
      "loss_g0": 20.589385986328125,
      "loss_g1": 43.85738754272461,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 29.32038116455078,
      "loss_g0": 69.01454162597656,
      "loss_g1": 28.653348922729492,
      "q_g0": 0.01652655005455017,
      "q_g1": 0.9834734797477722,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.099508285522461,
      "loss_g0": 25.88208770751953,
      "loss_g1": 14.918302536010742,
      "q_g0": 0.01652766577899456,
      "q_g1": 0.9834723472595215,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.267245292663574,
      "loss_g0": 14.106958389282227,
      "loss_g1": 12.236327171325684,
      "q_g0": 0.016527850180864334,
      "q_g1": 0.9834721088409424,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.163886070251465,
      "loss_g0": 33.85871124267578,
      "loss_g1": 12.81605052947998,
      "q_g0": 0.016530007123947144,
      "q_g1": 0.9834700226783752,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.576013565063477,
      "loss_g0": 14.576013565063477,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 28.043642044067383,
      "loss_g0": 28.467227935791016,
      "loss_g1": 2.8301517963409424,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.816703796386719,
      "loss_g0": 11.66568374633789,
      "loss_g1": 20.805503845214844,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652335561811924,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.269793510437012,
      "loss_g0": 12.166864395141602,
      "loss_g1": 18.39594841003418,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523992642760277,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.531041145324707,
      "loss_g0": 15.553922653198242,
      "loss_g1": 14.169174194335938,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523851081728935,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.136917114257812,
      "loss_g0": 16.325119018554688,
      "loss_g1": 4.934687614440918,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 25.671175003051758,
      "loss_g0": 25.881729125976562,
      "loss_g1": 13.138111114501953,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.949243545532227,
      "loss_g0": 6.77055549621582,
      "loss_g1": 17.584699630737305,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652352698147297,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.757835388183594,
      "loss_g0": 10.527462005615234,
      "loss_g1": 24.46844482421875,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.01652495190501213,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.723373413085938,
      "loss_g0": 16.8659610748291,
      "loss_g1": 8.236907958984375,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524076461791992,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.554706573486328,
      "loss_g0": 20.713743209838867,
      "loss_g1": 11.088606834411621,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523092985153198,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.07950210571289,
      "loss_g0": 12.14416790008545,
      "loss_g1": 8.23044204711914,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652269810438156,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.31066131591797,
      "loss_g0": 16.388927459716797,
      "loss_g1": 11.651886940002441,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.192840099334717,
      "loss_g0": 7.082331657409668,
      "loss_g1": 13.770444869995117,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523106023669243,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.029644966125488,
      "loss_g0": 8.008394241333008,
      "loss_g1": 9.294498443603516,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523240134119987,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.641282081604004,
      "loss_g0": 11.358695030212402,
      "loss_g1": 28.459291458129883,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016524991020560265,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.379610061645508,
      "loss_g0": 10.379610061645508,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.52580738067627,
      "loss_g0": 15.652621269226074,
      "loss_g1": 7.977384567260742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.57963752746582,
      "loss_g0": 10.675250053405762,
      "loss_g1": 4.888398170471191,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.538939476013184,
      "loss_g0": 6.506778240203857,
      "loss_g1": 8.453248023986816,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652262918651104,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.8052167892456055,
      "loss_g0": 2.683098554611206,
      "loss_g1": 10.073718070983887,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652337983250618,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.331494331359863,
      "loss_g0": 13.489097595214844,
      "loss_g1": 3.95033597946167,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.486391067504883,
      "loss_g0": 23.764911651611328,
      "loss_g1": 6.907858848571777,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.560498237609863,
      "loss_g0": 11.560498237609863,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.137931823730469,
      "loss_g0": 15.183094024658203,
      "loss_g1": 12.449706077575684,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.449775695800781,
      "loss_g0": 13.449775695800781,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 24.63710594177246,
      "loss_g0": 24.63710594177246,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.659961223602295,
      "loss_g0": 4.708728790283203,
      "loss_g1": 1.757115364074707,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.159266710281372,
      "loss_g0": 1.0800089836120605,
      "loss_g1": 5.876845359802246,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522914171218872,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.5096540451049805,
      "loss_g0": 0.0,
      "loss_g1": 1.5096540451049805,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.6586356163024902,
      "loss_g0": 17.58807373046875,
      "loss_g1": 2.4078209400177,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.8301626443862915,
      "loss_g0": 3.713120222091675,
      "loss_g1": 1.7985285520553589,
      "q_g0": 0.016522621735930443,
      "q_g1": 0.9834774136543274,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.142379760742188,
      "loss_g0": 4.8908586502075195,
      "loss_g1": 12.264204978942871,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.8388803005218506,
      "loss_g0": 17.53952407836914,
      "loss_g1": 2.591886520385742,
      "q_g0": 0.016523944213986397,
      "q_g1": 0.9834760427474976,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.733067274093628,
      "loss_g0": 11.549973487854004,
      "loss_g1": 1.5681174993515015,
      "q_g0": 0.016524961218237877,
      "q_g1": 0.9834749698638916,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.995539665222168,
      "loss_g0": 17.35718536376953,
      "loss_g1": 8.855034828186035,
      "q_g0": 0.01652582734823227,
      "q_g1": 0.9834741950035095,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.908906936645508,
      "loss_g0": 13.228914260864258,
      "loss_g1": 5.7858991622924805,
      "q_g0": 0.016526587307453156,
      "q_g1": 0.9834733605384827,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.9969053268432617,
      "loss_g0": 1.9969053268432617,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.708700180053711,
      "loss_g0": 0.0,
      "loss_g1": 7.708700180053711,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.604900360107422,
      "loss_g0": 0.0,
      "loss_g1": 17.604900360107422,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.8790340423584,
      "loss_g0": 14.99426555633545,
      "loss_g1": 24.028297424316406,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.364419937133789,
      "loss_g0": 17.39728546142578,
      "loss_g1": 9.229460716247559,
      "q_g0": 0.01652325503528118,
      "q_g1": 0.9834766983985901,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.53062629699707,
      "loss_g0": 21.46277618408203,
      "loss_g1": 20.514965057373047,
      "q_g0": 0.01652335561811924,
      "q_g1": 0.9834766387939453,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 63.694026947021484,
      "loss_g0": 39.072898864746094,
      "loss_g1": 64.10765838623047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.491110801696777,
      "loss_g0": 32.25536346435547,
      "loss_g1": 10.125420570373535,
      "q_g0": 0.016524681821465492,
      "q_g1": 0.9834753274917603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.876422882080078,
      "loss_g0": 21.876422882080078,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.003067016601562,
      "loss_g0": 18.105510711669922,
      "loss_g1": 11.905229568481445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.998339653015137,
      "loss_g0": 15.998734474182129,
      "loss_g1": 15.974810600280762,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522426158189774,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.150225639343262,
      "loss_g0": 13.905218124389648,
      "loss_g1": 28.7326602935791,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.0165239367634058,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.856245040893555,
      "loss_g0": 21.08354377746582,
      "loss_g1": 7.326643943786621,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.936988830566406,
      "loss_g0": 17.999820709228516,
      "loss_g1": 14.196883201599121,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 24.98513412475586,
      "loss_g0": 25.22356414794922,
      "loss_g1": 10.792887687683105,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.23229217529297,
      "loss_g0": 16.39470863342285,
      "loss_g1": 6.564668655395508,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.6674861907959,
      "loss_g0": 23.911983489990234,
      "loss_g1": 9.11412239074707,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.153011322021484,
      "loss_g0": 21.153011322021484,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.007712364196777,
      "loss_g0": 13.907391548156738,
      "loss_g1": 19.97917938232422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.706361770629883,
      "loss_g0": 0.0,
      "loss_g1": 17.706361770629883,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 29.95291519165039,
      "loss_g0": 3.408994674682617,
      "loss_g1": 30.398853302001953,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.761655807495117,
      "loss_g0": 14.5778169631958,
      "loss_g1": 4.596734046936035,
      "q_g0": 0.016523443162441254,
      "q_g1": 0.9834766387939453,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.55612850189209,
      "loss_g0": 11.59488582611084,
      "loss_g1": 9.521873474121094,
      "q_g0": 0.016523659229278564,
      "q_g1": 0.9834764003753662,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.866268157958984,
      "loss_g0": 28.72203254699707,
      "loss_g1": 16.66706085205078,
      "q_g0": 0.016524890437722206,
      "q_g1": 0.9834750890731812,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.43866491317749,
      "loss_g0": 4.295213222503662,
      "loss_g1": 6.474679946899414,
      "q_g0": 0.01652466505765915,
      "q_g1": 0.9834753274917603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.533851623535156,
      "loss_g0": 25.64208984375,
      "loss_g1": 8.24636173248291,
      "q_g0": 0.016526442021131516,
      "q_g1": 0.9834735989570618,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.165278434753418,
      "loss_g0": 26.13037109375,
      "loss_g1": 9.896970748901367,
      "q_g0": 0.01652809977531433,
      "q_g1": 0.9834718704223633,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.012568473815918,
      "loss_g0": 19.589658737182617,
      "loss_g1": 6.801182270050049,
      "q_g0": 0.016529399901628494,
      "q_g1": 0.983470618724823,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.7933831214904785,
      "loss_g0": 14.283552169799805,
      "loss_g1": 5.650679111480713,
      "q_g0": 0.016530277207493782,
      "q_g1": 0.9834696650505066,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.9334263801574707,
      "loss_g0": 15.862924575805664,
      "loss_g1": 1.6992766857147217,
      "q_g0": 0.016531728208065033,
      "q_g1": 0.9834682941436768,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.932229995727539,
      "loss_g0": 4.34198522567749,
      "loss_g1": 10.026195526123047,
      "q_g0": 0.016531147062778473,
      "q_g1": 0.9834688901901245,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.4260621070861816,
      "loss_g0": 9.33615493774414,
      "loss_g1": 2.3099050521850586,
      "q_g0": 0.016531866043806076,
      "q_g1": 0.9834680557250977,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.9116430282592773,
      "loss_g0": 10.356622695922852,
      "loss_g1": 3.803300142288208,
      "q_g0": 0.016532542183995247,
      "q_g1": 0.9834675192832947,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.639895915985107,
      "loss_g0": 9.454704284667969,
      "loss_g1": 7.6093878746032715,
      "q_g0": 0.016532735899090767,
      "q_g1": 0.9834672808647156,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.79119873046875,
      "loss_g0": 13.276227951049805,
      "loss_g1": 5.6653642654418945,
      "q_g0": 0.016533518210053444,
      "q_g1": 0.9834664463996887,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.791520118713379,
      "loss_g0": 4.834478378295898,
      "loss_g1": 5.807609558105469,
      "q_g0": 0.016533425077795982,
      "q_g1": 0.9834665656089783,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.956366539001465,
      "loss_g0": 7.640857219696045,
      "loss_g1": 14.062533378601074,
      "q_g0": 0.016532765701413155,
      "q_g1": 0.9834672808647156,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.722297668457031,
      "loss_g0": 17.061050415039062,
      "loss_g1": 6.548484802246094,
      "q_g0": 0.016533834859728813,
      "q_g1": 0.9834660887718201,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.811079025268555,
      "loss_g0": 17.19271469116211,
      "loss_g1": 4.602904796600342,
      "q_g0": 0.01653512381017208,
      "q_g1": 0.9834648370742798,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.59416389465332,
      "loss_g0": 6.653242588043213,
      "loss_g1": 8.626795768737793,
      "q_g0": 0.01653493009507656,
      "q_g1": 0.9834650754928589,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.6090264320373535,
      "loss_g0": 8.524413108825684,
      "loss_g1": 4.543195724487305,
      "q_g0": 0.016535332426428795,
      "q_g1": 0.9834645986557007,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.408714294433594,
      "loss_g0": 0.0,
      "loss_g1": 4.408714294433594,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.5764007568359375,
      "loss_g0": 9.317927360534668,
      "loss_g1": 3.479943037033081,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.549513816833496,
      "loss_g0": 32.116554260253906,
      "loss_g1": 10.187137603759766,
      "q_g0": 0.01652466505765915,
      "q_g1": 0.9834753274917603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.824026107788086,
      "loss_g0": 25.828664779663086,
      "loss_g1": 13.622304916381836,
      "q_g0": 0.01652592048048973,
      "q_g1": 0.98347407579422,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.533728122711182,
      "loss_g0": 10.493868827819824,
      "loss_g1": 4.433572769165039,
      "q_g0": 0.016526535153388977,
      "q_g1": 0.9834734797477722,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.645174264907837,
      "loss_g0": 3.9775140285491943,
      "loss_g1": 2.6227850914001465,
      "q_g0": 0.01652667298913002,
      "q_g1": 0.9834733605384827,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.449882984161377,
      "loss_g0": 4.770972728729248,
      "loss_g1": 6.478096008300781,
      "q_g0": 0.016526494175195694,
      "q_g1": 0.9834734797477722,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.062461853027344,
      "loss_g0": 7.246640682220459,
      "loss_g1": 8.076170921325684,
      "q_g0": 0.016526419669389725,
      "q_g1": 0.9834735989570618,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 0.9718624949455261,
      "loss_g0": 3.7385571002960205,
      "loss_g1": 0.9253697395324707,
      "q_g0": 0.016526702791452408,
      "q_g1": 0.9834732413291931,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.356663465499878,
      "loss_g0": 3.7415080070495605,
      "loss_g1": 3.350196361541748,
      "q_g0": 0.01652674376964569,
      "q_g1": 0.9834732413291931,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.872413635253906,
      "loss_g0": 15.484710693359375,
      "loss_g1": 7.7444868087768555,
      "q_g0": 0.016527533531188965,
      "q_g1": 0.983472466468811,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.306772232055664,
      "loss_g0": 30.402374267578125,
      "loss_g1": 18.103487014770508,
      "q_g0": 0.016528796404600143,
      "q_g1": 0.9834712147712708,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 50.13258361816406,
      "loss_g0": 34.335533142089844,
      "loss_g1": 50.39805221557617,
      "q_g0": 0.01652715355157852,
      "q_g1": 0.9834728240966797,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 42.138916015625,
      "loss_g0": 42.138916015625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 28.625452041625977,
      "loss_g0": 28.60680389404297,
      "loss_g1": 29.735383987426758,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.96431827545166,
      "loss_g0": 8.772768020629883,
      "loss_g1": 20.365257263183594,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523612663149834,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 32.86717987060547,
      "loss_g0": 33.189964294433594,
      "loss_g1": 13.653691291809082,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.464685440063477,
      "loss_g0": 17.477638244628906,
      "loss_g1": 16.693683624267578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.78824520111084,
      "loss_g0": 14.843603134155273,
      "loss_g1": 11.493110656738281,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 29.671659469604492,
      "loss_g0": 29.995168685913086,
      "loss_g1": 10.415112495422363,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 22.797927856445312,
      "loss_g0": 22.972253799438477,
      "loss_g1": 12.421308517456055,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.149653434753418,
      "loss_g0": 10.100053787231445,
      "loss_g1": 13.10195541381836,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522729769349098,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.295549392700195,
      "loss_g0": 9.084047317504883,
      "loss_g1": 21.883689880371094,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652403734624386,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 27.91263198852539,
      "loss_g0": 28.18048858642578,
      "loss_g1": 11.968709945678711,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.535308837890625,
      "loss_g0": 9.645698547363281,
      "loss_g1": 2.9644947052001953,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.39586353302002,
      "loss_g0": 11.240777015686035,
      "loss_g1": 20.62664222717285,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652337983250618,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.023698806762695,
      "loss_g0": 11.988243103027344,
      "loss_g1": 14.1339693069458,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523588448762894,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.182905197143555,
      "loss_g0": 23.5164794921875,
      "loss_g1": 3.327312707901001,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.266838550567627,
      "loss_g0": 7.256026268005371,
      "loss_g1": 7.91039514541626,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522489488124847,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.883562088012695,
      "loss_g0": 0.0,
      "loss_g1": 9.883562088012695,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.843311786651611,
      "loss_g0": 25.14602279663086,
      "loss_g1": 5.519025802612305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.585020065307617,
      "loss_g0": 8.223165512084961,
      "loss_g1": 7.574298858642578,
      "q_g0": 0.016522489488124847,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.780492782592773,
      "loss_g0": 8.345844268798828,
      "loss_g1": 13.871794700622559,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.34532642364502,
      "loss_g0": 13.232136726379395,
      "loss_g1": 12.330427169799805,
      "q_g0": 0.016522521153092384,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.647863388061523,
      "loss_g0": 10.21250057220459,
      "loss_g1": 9.638378143310547,
      "q_g0": 0.016522573307156563,
      "q_g1": 0.9834774136543274,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.768424034118652,
      "loss_g0": 23.580123901367188,
      "loss_g1": 15.637179374694824,
      "q_g0": 0.016523387283086777,
      "q_g1": 0.9834766387939453,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.6264824867248535,
      "loss_g0": 29.052783966064453,
      "loss_g1": 7.266449928283691,
      "q_g0": 0.01652561128139496,
      "q_g1": 0.9834744334220886,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.108086585998535,
      "loss_g0": 9.193060874938965,
      "loss_g1": 10.123462677001953,
      "q_g0": 0.016525518149137497,
      "q_g1": 0.9834745526313782,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 22.98649024963379,
      "loss_g0": 11.50139331817627,
      "loss_g1": 23.179462432861328,
      "q_g0": 0.01652432605624199,
      "q_g1": 0.9834756851196289,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.829704284667969,
      "loss_g0": 15.819270133972168,
      "loss_g1": 8.712260246276855,
      "q_g0": 0.016525061801075935,
      "q_g1": 0.983474850654602,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.459028482437134,
      "loss_g0": 5.109744548797607,
      "loss_g1": 2.4144885540008545,
      "q_g0": 0.016525331884622574,
      "q_g1": 0.9834746718406677,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.2824878692626953,
      "loss_g0": 3.8282172679901123,
      "loss_g1": 3.273317813873291,
      "q_g0": 0.016525380313396454,
      "q_g1": 0.9834746718406677,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.9493634700775146,
      "loss_g0": 22.245990753173828,
      "loss_g1": 2.6250813007354736,
      "q_g0": 0.01652737893164158,
      "q_g1": 0.9834725856781006,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.005435466766357,
      "loss_g0": 2.044215202331543,
      "loss_g1": 6.072002410888672,
      "q_g0": 0.0165269672870636,
      "q_g1": 0.9834730625152588,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.684206008911133,
      "loss_g0": 19.175674438476562,
      "loss_g1": 20.70955467224121,
      "q_g0": 0.016526812687516212,
      "q_g1": 0.9834731221199036,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.24702262878418,
      "loss_g0": 7.548400402069092,
      "loss_g1": 9.275566101074219,
      "q_g0": 0.016526643186807632,
      "q_g1": 0.9834733605384827,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.05914044380188,
      "loss_g0": 11.280662536621094,
      "loss_g1": 2.920975685119629,
      "q_g0": 0.016527501866221428,
      "q_g1": 0.983472466468811,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.3007588386535645,
      "loss_g0": 3.7859609127044678,
      "loss_g1": 2.275799512863159,
      "q_g0": 0.016527658328413963,
      "q_g1": 0.9834723472595215,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.544589042663574,
      "loss_g0": 13.626906394958496,
      "loss_g1": 4.391948223114014,
      "q_g0": 0.016528604552149773,
      "q_g1": 0.9834713935852051,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.8324888944625854,
      "loss_g0": 4.784177303314209,
      "loss_g1": 1.7828806638717651,
      "q_g0": 0.0165289044380188,
      "q_g1": 0.9834710955619812,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.434448719024658,
      "loss_g0": 9.210152626037598,
      "loss_g1": 4.354182243347168,
      "q_g0": 0.016529399901628494,
      "q_g1": 0.983470618724823,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.0661933422088623,
      "loss_g0": 7.308956146240234,
      "loss_g1": 1.978074073791504,
      "q_g0": 0.016529936343431473,
      "q_g1": 0.9834700226783752,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.6037895679473877,
      "loss_g0": 2.7495431900024414,
      "loss_g1": 3.618147611618042,
      "q_g0": 0.01652984321117401,
      "q_g1": 0.9834701418876648,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.505497932434082,
      "loss_g0": 7.357587814331055,
      "loss_g1": 9.541598320007324,
      "q_g0": 0.016529617831110954,
      "q_g1": 0.9834703803062439,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.8124566078186035,
      "loss_g0": 6.058998107910156,
      "loss_g1": 6.825119972229004,
      "q_g0": 0.016529541462659836,
      "q_g1": 0.9834704995155334,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.6547811031341553,
      "loss_g0": 1.4272029399871826,
      "loss_g1": 2.675413131713867,
      "q_g0": 0.01652940921485424,
      "q_g1": 0.983470618724823,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.7647428512573242,
      "loss_g0": 8.463903427124023,
      "loss_g1": 1.6521437168121338,
      "q_g0": 0.0165301151573658,
      "q_g1": 0.9834699034690857,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.9002559185028076,
      "loss_g0": 6.911490440368652,
      "loss_g1": 3.849642038345337,
      "q_g0": 0.01653042621910572,
      "q_g1": 0.9834696054458618,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.5814437866210938,
      "loss_g0": 4.340465545654297,
      "loss_g1": 3.56868577003479,
      "q_g0": 0.01653050258755684,
      "q_g1": 0.9834694862365723,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.072455406188965,
      "loss_g0": 2.5293898582458496,
      "loss_g1": 9.182429313659668,
      "q_g0": 0.01652981899678707,
      "q_g1": 0.9834701418876648,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.8364105224609375,
      "loss_g0": 7.474942207336426,
      "loss_g1": 6.825677871704102,
      "q_g0": 0.01652989163994789,
      "q_g1": 0.9834701418876648,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 31.29596519470215,
      "loss_g0": 34.36125946044922,
      "loss_g1": 31.244443893432617,
      "q_g0": 0.01653021015226841,
      "q_g1": 0.9834697842597961,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 27.774507522583008,
      "loss_g0": 17.290111541748047,
      "loss_g1": 27.950719833374023,
      "q_g0": 0.016529114916920662,
      "q_g1": 0.9834708571434021,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 39.58124923706055,
      "loss_g0": 30.41084098815918,
      "loss_g1": 39.7353630065918,
      "q_g0": 0.016528161242604256,
      "q_g1": 0.9834718704223633,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.898051261901855,
      "loss_g0": 21.07613754272461,
      "loss_g1": 15.811025619506836,
      "q_g0": 0.016528695821762085,
      "q_g1": 0.9834713339805603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.510860443115234,
      "loss_g0": 30.379806518554688,
      "loss_g1": 16.277751922607422,
      "q_g0": 0.01653013750910759,
      "q_g1": 0.9834697842597961,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.450468063354492,
      "loss_g0": 0.0,
      "loss_g1": 20.450468063354492,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 25.491962432861328,
      "loss_g0": 5.944809913635254,
      "loss_g1": 25.820354461669922,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 33.303993225097656,
      "loss_g0": 13.293317794799805,
      "loss_g1": 33.64017105102539,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.748585224151611,
      "loss_g0": 17.429271697998047,
      "loss_g1": 6.569137096405029,
      "q_g0": 0.016523534432053566,
      "q_g1": 0.9834765195846558,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 26.56142807006836,
      "loss_g0": 28.634267807006836,
      "loss_g1": 26.526599884033203,
      "q_g0": 0.016523750498890877,
      "q_g1": 0.9834762811660767,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.774993896484375,
      "loss_g0": 25.16704750061035,
      "loss_g1": 16.633987426757812,
      "q_g0": 0.016524627804756165,
      "q_g1": 0.9834753274917603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.897623538970947,
      "loss_g0": 7.008989334106445,
      "loss_g1": 7.912554740905762,
      "q_g0": 0.016524534672498703,
      "q_g1": 0.9834754467010498,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.406668663024902,
      "loss_g0": 9.482375144958496,
      "loss_g1": 15.506205558776855,
      "q_g0": 0.01652391441166401,
      "q_g1": 0.9834760427474976,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.692893981933594,
      "loss_g0": 38.84223556518555,
      "loss_g1": 13.270278930664062,
      "q_g0": 0.01652652584016323,
      "q_g1": 0.9834734797477722,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.28408145904541,
      "loss_g0": 16.69725227355957,
      "loss_g1": 9.159502983093262,
      "q_g0": 0.016527293249964714,
      "q_g1": 0.9834727048873901,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.815479278564453,
      "loss_g0": 40.40578079223633,
      "loss_g1": 21.503032684326172,
      "q_g0": 0.016529222950339317,
      "q_g1": 0.9834707379341125,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 32.098182678222656,
      "loss_g0": 11.835411071777344,
      "loss_g1": 32.43869400024414,
      "q_g0": 0.016527116298675537,
      "q_g1": 0.9834729433059692,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.388171195983887,
      "loss_g0": 14.827380180358887,
      "loss_g1": 13.363986015319824,
      "q_g0": 0.016527269035577774,
      "q_g1": 0.9834727048873901,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.67015552520752,
      "loss_g0": 26.767711639404297,
      "loss_g1": 12.433225631713867,
      "q_g0": 0.016528736799955368,
      "q_g1": 0.9834713339805603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.462225914001465,
      "loss_g0": 11.279794692993164,
      "loss_g1": 9.431678771972656,
      "q_g0": 0.016528921201825142,
      "q_g1": 0.9834710955619812,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.2539448738098145,
      "loss_g0": 37.423500061035156,
      "loss_g1": 5.729982376098633,
      "q_g0": 0.0165321696549654,
      "q_g1": 0.9834678173065186,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.52717399597168,
      "loss_g0": 8.467966079711914,
      "loss_g1": 23.780296325683594,
      "q_g0": 0.016530610620975494,
      "q_g1": 0.9834693670272827,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.113154411315918,
      "loss_g0": 10.178814888000488,
      "loss_g1": 8.078433990478516,
      "q_g0": 0.016530821099877357,
      "q_g1": 0.9834692478179932,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.9402647018432617,
      "loss_g0": 15.73234748840332,
      "loss_g1": 2.725229024887085,
      "q_g0": 0.016532152891159058,
      "q_g1": 0.9834678173065186,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.293924331665039,
      "loss_g0": 24.142051696777344,
      "loss_g1": 2.9434194564819336,
      "q_g0": 0.016534315422177315,
      "q_g1": 0.9834656715393066,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.159797668457031,
      "loss_g0": 15.013312339782715,
      "loss_g1": 8.04456901550293,
      "q_g0": 0.01653503067791462,
      "q_g1": 0.9834649562835693,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.05965805053711,
      "loss_g0": 18.01429557800293,
      "loss_g1": 7.892279148101807,
      "q_g0": 0.01653607003390789,
      "q_g1": 0.9834639430046082,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.787220001220703,
      "loss_g0": 29.921092987060547,
      "loss_g1": 10.465460777282715,
      "q_g0": 0.016538070514798164,
      "q_g1": 0.9834619760513306,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.886896133422852,
      "loss_g0": 5.438932418823242,
      "loss_g1": 9.961690902709961,
      "q_g0": 0.016537604853510857,
      "q_g1": 0.9834624528884888,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.607221603393555,
      "loss_g0": 5.607221603393555,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.222653865814209,
      "loss_g0": 6.154067516326904,
      "loss_g1": 10.305142402648926,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.29758644104004,
      "loss_g0": 23.537607192993164,
      "loss_g1": 9.010698318481445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.341394424438477,
      "loss_g0": 8.406364440917969,
      "loss_g1": 4.47409725189209,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.4994916915893555,
      "loss_g0": 6.482094764709473,
      "loss_g1": 7.534999847412109,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522526741027832,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.955639839172363,
      "loss_g0": 4.890394687652588,
      "loss_g1": 8.839174270629883,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522930935025215,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 29.212160110473633,
      "loss_g0": 29.6159610748291,
      "loss_g1": 5.176301956176758,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.498591899871826,
      "loss_g0": 6.415423393249512,
      "loss_g1": 11.448960304260254,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522938385605812,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.373085975646973,
      "loss_g0": 8.311430931091309,
      "loss_g1": 12.042792320251465,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523318365216255,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.554625034332275,
      "loss_g0": 4.617392063140869,
      "loss_g1": 0.8186224699020386,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522930935025215,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.562671184539795,
      "loss_g0": 1.531343698501587,
      "loss_g1": 3.4273173809051514,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523122787475586,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.723777770996094,
      "loss_g0": 21.036895751953125,
      "loss_g1": 2.085768938064575,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.9131228923797607,
      "loss_g0": 2.897920608520508,
      "loss_g1": 3.8180124759674072,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522511839866638,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.394272804260254,
      "loss_g0": 8.494007110595703,
      "loss_g1": 2.4577112197875977,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.85771369934082,
      "loss_g0": 5.89570426940918,
      "loss_g1": 3.5963590145111084,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.382617950439453,
      "loss_g0": 21.382617950439453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.139890670776367,
      "loss_g0": 10.287214279174805,
      "loss_g1": 1.370603084564209,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.9824811220169067,
      "loss_g0": 0.0,
      "loss_g1": 1.9824811220169067,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.0776193141937256,
      "loss_g0": 0.9558624029159546,
      "loss_g1": 2.0964648723602295,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 0.5037932395935059,
      "loss_g0": 1.0654842853546143,
      "loss_g1": 0.49435678124427795,
      "q_g0": 0.0165224801748991,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 0.5180168747901917,
      "loss_g0": 2.2023730278015137,
      "loss_g1": 0.4897192716598511,
      "q_g0": 0.01652265153825283,
      "q_g1": 0.9834772944450378,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 26.719213485717773,
      "loss_g0": 27.991519927978516,
      "loss_g1": 26.697837829589844,
      "q_g0": 0.016522781923413277,
      "q_g1": 0.9834771752357483,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.743982315063477,
      "loss_g0": 15.743982315063477,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 19.726776123046875,
      "loss_g0": 19.45021629333496,
      "loss_g1": 36.1885986328125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.12672233581543,
      "loss_g0": 18.045862197875977,
      "loss_g1": 22.93963623046875,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522930935025215,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.179003715515137,
      "loss_g0": 15.274347305297852,
      "loss_g1": 9.503754615783691,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 24.9783878326416,
      "loss_g0": 25.17548942565918,
      "loss_g1": 13.246162414550781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 39.0681266784668,
      "loss_g0": 39.379676818847656,
      "loss_g1": 20.52339744567871,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.946603775024414,
      "loss_g0": 17.772232055664062,
      "loss_g1": 28.325204849243164,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652350276708603,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.198974370956421,
      "loss_g0": 1.9994066953659058,
      "loss_g1": 14.076314926147461,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652473583817482,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 24.274208068847656,
      "loss_g0": 0.0,
      "loss_g1": 24.274208068847656,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 37.81714630126953,
      "loss_g0": 18.259292602539062,
      "loss_g1": 38.14571762084961,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 41.01262664794922,
      "loss_g0": 0.0,
      "loss_g1": 41.01262664794922,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.869434356689453,
      "loss_g0": 19.521326065063477,
      "loss_g1": 18.858482360839844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 20.652368545532227,
      "loss_g0": 0.0,
      "loss_g1": 20.652368545532227,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 69.54529571533203,
      "loss_g0": 53.02381134033203,
      "loss_g1": 69.8228530883789,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 21.10373878479004,
      "loss_g0": 21.10373878479004,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 15.755792617797852,
      "loss_g0": 15.812838554382324,
      "loss_g1": 12.360158920288086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.053610801696777,
      "loss_g0": 10.053610801696777,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 35.55223083496094,
      "loss_g0": 35.9189453125,
      "loss_g1": 13.723976135253906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.021368980407715,
      "loss_g0": 14.167851448059082,
      "loss_g1": 5.302178382873535,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 17.48383140563965,
      "loss_g0": 17.49296760559082,
      "loss_g1": 16.93998908996582,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.845346450805664,
      "loss_g0": 12.742327690124512,
      "loss_g1": 18.977205276489258,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523055732250214,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.204823017120361,
      "loss_g0": 7.232083320617676,
      "loss_g1": 5.582229137420654,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.975955963134766,
      "loss_g0": 11.014204978942871,
      "loss_g1": 8.699241638183594,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.154754638671875,
      "loss_g0": 9.148608207702637,
      "loss_g1": 9.520620346069336,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.0933361053466797,
      "loss_g0": 3.1005492210388184,
      "loss_g1": 2.6639797687530518,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.011894226074219,
      "loss_g0": 10.104622840881348,
      "loss_g1": 4.492351531982422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 35.427223205566406,
      "loss_g0": 35.81538772583008,
      "loss_g1": 12.32214641571045,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 27.060365676879883,
      "loss_g0": 27.480609893798828,
      "loss_g1": 2.045823097229004,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.345787525177002,
      "loss_g0": 7.176773548126221,
      "loss_g1": 17.405479431152344,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652347296476364,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.297786712646484,
      "loss_g0": 8.067201614379883,
      "loss_g1": 22.0209903717041,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524890437722206,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.661602973937988,
      "loss_g0": 14.827411651611328,
      "loss_g1": 4.792855262756348,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652386039495468,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.208498001098633,
      "loss_g0": 9.245563507080078,
      "loss_g1": 7.002391815185547,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523627564311028,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.959175109863281,
      "loss_g0": 13.017084121704102,
      "loss_g1": 9.512338638305664,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523264348506927,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 25.059146881103516,
      "loss_g0": 25.29077911376953,
      "loss_g1": 11.27147102355957,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.520490646362305,
      "loss_g0": 16.78110694885254,
      "loss_g1": 1.0075150728225708,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.254432678222656,
      "loss_g0": 14.41897964477539,
      "loss_g1": 4.45996618270874,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.752457618713379,
      "loss_g0": 5.731977462768555,
      "loss_g1": 6.971487522125244,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 19.469892501831055,
      "loss_g0": 19.6636962890625,
      "loss_g1": 7.933980941772461,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 16.678232192993164,
      "loss_g0": 16.678232192993164,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.697592735290527,
      "loss_g0": 10.823309898376465,
      "loss_g1": 3.2144246101379395,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.101585865020752,
      "loss_g0": 3.1049814224243164,
      "loss_g1": 2.899470329284668,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.76865816116333,
      "loss_g0": 2.8054962158203125,
      "loss_g1": 0.5759117603302002,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.415773630142212,
      "loss_g0": 3.470262050628662,
      "loss_g1": 0.1724134087562561,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.563769340515137,
      "loss_g0": 8.318442344665527,
      "loss_g1": 23.16522216796875,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523944213986397,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 2.7746262550354004,
      "loss_g0": 2.7656607627868652,
      "loss_g1": 3.3082268238067627,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524000093340874,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.040721416473389,
      "loss_g0": 7.026139736175537,
      "loss_g1": 7.908583164215088,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652408577501774,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 23.699983596801758,
      "loss_g0": 24.006534576416016,
      "loss_g1": 5.452915191650391,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 9.560406684875488,
      "loss_g0": 9.607198715209961,
      "loss_g1": 6.7751851081848145,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.732330799102783,
      "loss_g0": 7.85179328918457,
      "loss_g1": 0.6214622259140015,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 12.414466857910156,
      "loss_g0": 12.358745574951172,
      "loss_g1": 15.731110572814941,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652277447283268,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 32.24583053588867,
      "loss_g0": 32.226863861083984,
      "loss_g1": 33.374813079833984,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522889956831932,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 18.81256103515625,
      "loss_g0": 19.011680603027344,
      "loss_g1": 6.960276126861572,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 11.732409477233887,
      "loss_g0": 11.611000061035156,
      "loss_g1": 18.95885467529297,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523171216249466,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 19.321779251098633,
      "loss_g0": 19.256179809570312,
      "loss_g1": 23.22632598876953,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.0165235735476017,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.269824743270874,
      "loss_g0": 2.92482590675354,
      "loss_g1": 23.80132293701172,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.01652570627629757,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 38.49625015258789,
      "loss_g0": 38.49625015258789,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 19.34111785888672,
      "loss_g0": 0.0,
      "loss_g1": 19.34111785888672,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 14.945146560668945,
      "loss_g0": 15.292586326599121,
      "loss_g1": 14.939310073852539,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 13.691886901855469,
      "loss_g0": 6.389045715332031,
      "loss_g1": 13.814574241638184,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 5.33245849609375,
      "loss_g0": 12.564356803894043,
      "loss_g1": 5.210957050323486,
      "q_g0": 0.016523178666830063,
      "q_g1": 0.9834768176078796,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 8.724019050598145,
      "loss_g0": 0.0,
      "loss_g1": 8.724019050598145,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 4.667445182800293,
      "loss_g0": 3.6392569541931152,
      "loss_g1": 4.684719085693359,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 6.650892734527588,
      "loss_g0": 0.0,
      "loss_g1": 6.650892734527588,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 7.849517345428467,
      "loss_g0": 0.0,
      "loss_g1": 7.849517345428467,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 3.1241390705108643,
      "loss_g0": 9.947065353393555,
      "loss_g1": 3.0095138549804688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 10.72518253326416,
      "loss_g0": 15.009002685546875,
      "loss_g1": 10.653212547302246,
      "q_g0": 0.01652287133038044,
      "q_g1": 0.9834771752357483,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 0.89300936460495,
      "loss_g0": 4.412405014038086,
      "loss_g1": 0.8338805437088013,
      "q_g0": 0.01652323268353939,
      "q_g1": 0.9834766983985901,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 1.6071851253509521,
      "loss_g0": 1.76971435546875,
      "loss_g1": 1.604454517364502,
      "q_g0": 0.016523247584700584,
      "q_g1": 0.9834766983985901,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "eval_cer": 0.06188833001910555,
      "eval_loss": 13.985843658447266,
      "eval_runtime": 81.1459,
      "eval_samples_per_second": 20.654,
      "eval_steps_per_second": 5.164,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 40.416404724121094,
      "loss_g0": 37.23225784301758,
      "loss_g1": 40.46990203857422,
      "q_g0": 0.01652292162179947,
      "q_g1": 0.9834770560264587,
      "step": 800
    },
    {
      "epoch": 4.060913705583756,
      "gdro_loss": 47.727657318115234,
      "loss_g0": 11.588445663452148,
      "loss_g1": 48.33479309082031,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 800
    },
    {
      "epoch": 4.568527918781726,
      "grad_norm": 384.799560546875,
      "learning_rate": 0.00026616042780748663,
      "loss": 31.4498,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.785594940185547,
      "loss_g0": 18.762544631958008,
      "loss_g1": 20.155189514160156,
      "q_g0": 0.9834483861923218,
      "q_g1": 0.016551679000258446,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 30.916791915893555,
      "loss_g0": 30.758852005004883,
      "loss_g1": 40.30050277709961,
      "q_g0": 0.9834473133087158,
      "q_g1": 0.016552655026316643,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.154939651489258,
      "loss_g0": 23.272480010986328,
      "loss_g1": 16.171268463134766,
      "q_g0": 0.9834481477737427,
      "q_g1": 0.016551926732063293,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 30.555591583251953,
      "loss_g0": 30.789846420288086,
      "loss_g1": 16.63585662841797,
      "q_g0": 0.9834495186805725,
      "q_g1": 0.016550475731492043,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.11219024658203,
      "loss_g0": 26.194442749023438,
      "loss_g1": 21.22454071044922,
      "q_g0": 0.9834499955177307,
      "q_g1": 0.01654996909201145,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.164400100708008,
      "loss_g0": 17.289058685302734,
      "loss_g1": 9.756423950195312,
      "q_g0": 0.9834507703781128,
      "q_g1": 0.01654919423162937,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.446020126342773,
      "loss_g0": 23.314241409301758,
      "loss_g1": 31.276721954345703,
      "q_g0": 0.9834499955177307,
      "q_g1": 0.016550010070204735,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.030027389526367,
      "loss_g0": 19.01366424560547,
      "loss_g1": 20.0023136138916,
      "q_g0": 0.9834498763084412,
      "q_g1": 0.016550110653042793,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.496957778930664,
      "loss_g0": 19.068418502807617,
      "loss_g1": 44.9576416015625,
      "q_g0": 0.9834471940994263,
      "q_g1": 0.01655275747179985,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 81.7951431274414,
      "loss_g0": 82.76067352294922,
      "loss_g1": 24.409088134765625,
      "q_g0": 0.9834532141685486,
      "q_g1": 0.016546789556741714,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.403871536254883,
      "loss_g0": 13.280692100524902,
      "loss_g1": 20.724651336669922,
      "q_g0": 0.9834523797035217,
      "q_g1": 0.0165475495159626,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 22.794776916503906,
      "loss_g0": 22.970678329467773,
      "loss_g1": 12.339872360229492,
      "q_g0": 0.9834535717964172,
      "q_g1": 0.0165464635938406,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 61.20728302001953,
      "loss_g0": 62.074546813964844,
      "loss_g1": 9.643906593322754,
      "q_g0": 0.9834588766098022,
      "q_g1": 0.016541095450520515,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.174480438232422,
      "loss_g0": 16.242570877075195,
      "loss_g1": 12.126068115234375,
      "q_g0": 0.9834592938423157,
      "q_g1": 0.01654067635536194,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.192217826843262,
      "loss_g0": 4.120488166809082,
      "loss_g1": 8.456951141357422,
      "q_g0": 0.9834588766098022,
      "q_g1": 0.016541127115488052,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 21.414691925048828,
      "loss_g0": 21.230215072631836,
      "loss_g1": 32.38213348388672,
      "q_g0": 0.9834576845169067,
      "q_g1": 0.01654226705431938,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.38151741027832,
      "loss_g0": 17.383665084838867,
      "loss_g1": 17.253816604614258,
      "q_g0": 0.9834576845169067,
      "q_g1": 0.016542257741093636,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 21.27776527404785,
      "loss_g0": 20.739845275878906,
      "loss_g1": 53.251163482666016,
      "q_g0": 0.9834544658660889,
      "q_g1": 0.016545578837394714,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 29.72012710571289,
      "loss_g0": 29.72012710571289,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 21.983264923095703,
      "loss_g0": 22.19875717163086,
      "loss_g1": 9.15636920928955,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.911693096160889,
      "loss_g0": 7.900554180145264,
      "loss_g1": 8.574716567993164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522489488124847,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.010580062866211,
      "loss_g0": 15.169184684753418,
      "loss_g1": 5.569823265075684,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.233354568481445,
      "loss_g0": 8.253472328186035,
      "loss_g1": 7.035813808441162,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.549769401550293,
      "loss_g0": 12.6839017868042,
      "loss_g1": 4.5657196044921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.653447151184082,
      "loss_g0": 10.625041007995605,
      "loss_g1": 12.344261169433594,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522597521543503,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.459648609161377,
      "loss_g0": 0.0,
      "loss_g1": 7.459648609161377,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.18071460723877,
      "loss_g0": 10.18071460723877,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.616464138031006,
      "loss_g0": 7.53521203994751,
      "loss_g1": 12.452898979187012,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.610328674316406,
      "loss_g0": 12.682355880737305,
      "loss_g1": 8.32298755645752,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.239389419555664,
      "loss_g0": 25.239389419555664,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.907255172729492,
      "loss_g0": 9.838130950927734,
      "loss_g1": 14.021797180175781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.1912202835083,
      "loss_g0": 9.1912202835083,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.071507453918457,
      "loss_g0": 11.040423393249512,
      "loss_g1": 12.92170238494873,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.300008773803711,
      "loss_g0": 2.1738975048065186,
      "loss_g1": 9.806262016296387,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523201018571854,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.015018463134766,
      "loss_g0": 13.110960006713867,
      "loss_g1": 7.304271221160889,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.0165226049721241,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.145164489746094,
      "loss_g0": 5.131426811218262,
      "loss_g1": 5.96286678314209,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522690653800964,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.157648086547852,
      "loss_g0": 6.190588474273682,
      "loss_g1": 4.196907997131348,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.0165224801748991,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.231156349182129,
      "loss_g0": 12.304738998413086,
      "loss_g1": 7.851245880126953,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 40.38786315917969,
      "loss_g0": 40.92378234863281,
      "loss_g1": 8.487751007080078,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.307624340057373,
      "loss_g0": 6.362678527832031,
      "loss_g1": 3.0305895805358887,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.429817199707031,
      "loss_g0": 4.429817199707031,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.458019256591797,
      "loss_g0": 0.0,
      "loss_g1": 9.458019256591797,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.357155799865723,
      "loss_g0": 4.571608543395996,
      "loss_g1": 7.4039530754089355,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.821880340576172,
      "loss_g0": 8.821880340576172,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.94759750366211,
      "loss_g0": 9.064558029174805,
      "loss_g1": 1.9856359958648682,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.022152423858643,
      "loss_g0": 5.060791492462158,
      "loss_g1": 2.7221972942352295,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.287083148956299,
      "loss_g0": 6.360955238342285,
      "loss_g1": 1.8899272680282593,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.954014778137207,
      "loss_g0": 8.069613456726074,
      "loss_g1": 1.0731186866760254,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.6972111463546753,
      "loss_g0": 1.6242263317108154,
      "loss_g1": 6.041426181793213,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652287319302559,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.3886514902114868,
      "loss_g0": 1.4076862335205078,
      "loss_g1": 0.2556535601615906,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522753983736038,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 55.39949035644531,
      "loss_g0": 55.91010665893555,
      "loss_g1": 25.00579071044922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 35.79483413696289,
      "loss_g0": 35.29784393310547,
      "loss_g1": 65.37191009521484,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525497660040855,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.421436309814453,
      "loss_g0": 25.421436309814453,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 28.460651397705078,
      "loss_g0": 28.51862144470215,
      "loss_g1": 25.01007080078125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.810022354125977,
      "loss_g0": 16.793346405029297,
      "loss_g1": 17.80263328552246,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522526741027832,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.64830780029297,
      "loss_g0": 26.96169662475586,
      "loss_g1": 7.994198322296143,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.009933471679688,
      "loss_g0": 25.092815399169922,
      "loss_g1": 20.076478958129883,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.13039207458496,
      "loss_g0": 16.22382164001465,
      "loss_g1": 10.569158554077148,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 27.528837203979492,
      "loss_g0": 27.797679901123047,
      "loss_g1": 11.526193618774414,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 22.051084518432617,
      "loss_g0": 22.32061767578125,
      "loss_g1": 6.0074310302734375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 24.504478454589844,
      "loss_g0": 24.458179473876953,
      "loss_g1": 27.26034927368164,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522713005542755,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.261576652526855,
      "loss_g0": 14.346242904663086,
      "loss_g1": 9.221922874450684,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.89660358428955,
      "loss_g0": 12.830349922180176,
      "loss_g1": 16.840147018432617,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522830352187157,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.126117706298828,
      "loss_g0": 25.635204315185547,
      "loss_g1": 55.341041564941406,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.016525868326425552,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.161908149719238,
      "loss_g0": 13.979646682739258,
      "loss_g1": 25.00771713256836,
      "q_g0": 0.9834730625152588,
      "q_g1": 0.01652699150145054,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.972089767456055,
      "loss_g0": 17.034162521362305,
      "loss_g1": 13.27820873260498,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.0165266040712595,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.147233963012695,
      "loss_g0": 8.164231300354004,
      "loss_g1": 7.135777950286865,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.01652650348842144,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.596238136291504,
      "loss_g0": 14.46413516998291,
      "loss_g1": 22.45708465576172,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527315601706505,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.080172538757324,
      "loss_g0": 5.956414699554443,
      "loss_g1": 13.444158554077148,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.01652807556092739,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.17841148376465,
      "loss_g0": 18.309589385986328,
      "loss_g1": 10.372526168823242,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527263447642326,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.761240005493164,
      "loss_g0": 17.855518341064453,
      "loss_g1": 12.150897026062012,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.01652667298913002,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.545835494995117,
      "loss_g0": 7.617194175720215,
      "loss_g1": 3.29929518699646,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526229679584503,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.921711444854736,
      "loss_g0": 7.724996566772461,
      "loss_g1": 19.627275466918945,
      "q_g0": 0.9834725856781006,
      "q_g1": 0.0165274478495121,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.782268524169922,
      "loss_g0": 0.0,
      "loss_g1": 15.782268524169922,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.5462284088134766,
      "loss_g0": 22.62561798095703,
      "loss_g1": 3.2256946563720703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.012202262878418,
      "loss_g0": 3.0138940811157227,
      "loss_g1": 5.045773506164551,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.282360076904297,
      "loss_g0": 12.235650062561035,
      "loss_g1": 10.249544143676758,
      "q_g0": 0.016522621735930443,
      "q_g1": 0.9834774136543274,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.933351516723633,
      "loss_g0": 1.2598870992660522,
      "loss_g1": 12.112665176391602,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.561491012573242,
      "loss_g0": 15.614110946655273,
      "loss_g1": 25.728607177734375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.238238334655762,
      "loss_g0": 5.8706793785095215,
      "loss_g1": 10.311613082885742,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.784149169921875,
      "loss_g0": 5.0507731437683105,
      "loss_g1": 16.98126983642578,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.0003461837768555,
      "loss_g0": 8.676569938659668,
      "loss_g1": 5.955385208129883,
      "q_g0": 0.01652269810438156,
      "q_g1": 0.9834772944450378,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.613874435424805,
      "loss_g0": 0.0,
      "loss_g1": 5.613874435424805,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.631890773773193,
      "loss_g0": 1.8043395280838013,
      "loss_g1": 5.696193695068359,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 34.32081604003906,
      "loss_g0": 10.861915588378906,
      "loss_g1": 34.71492385864258,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.569239854812622,
      "loss_g0": 20.839519500732422,
      "loss_g1": 3.279067039489746,
      "q_g0": 0.016524218022823334,
      "q_g1": 0.9834758043289185,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.61113452911377,
      "loss_g0": 35.90018081665039,
      "loss_g1": 14.253390312194824,
      "q_g0": 0.01652643457055092,
      "q_g1": 0.9834735989570618,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.932258129119873,
      "loss_g0": 5.603180885314941,
      "loss_g1": 5.937788009643555,
      "q_g0": 0.01652640290558338,
      "q_g1": 0.9834735989570618,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.585326671600342,
      "loss_g0": 0.0,
      "loss_g1": 7.585326671600342,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.273778915405273,
      "loss_g0": 12.273778915405273,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 28.566221237182617,
      "loss_g0": 28.566221237182617,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.710976600646973,
      "loss_g0": 8.823647499084473,
      "loss_g1": 2.004377603530884,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.8984408378601074,
      "loss_g0": 1.8554725646972656,
      "loss_g1": 4.456032752990723,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522690653800964,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.153684616088867,
      "loss_g0": 19.40932273864746,
      "loss_g1": 3.9370546340942383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.384231567382812,
      "loss_g0": 10.55537223815918,
      "loss_g1": 0.19731405377388,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.794218063354492,
      "loss_g0": 4.8426923751831055,
      "loss_g1": 1.9088232517242432,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.966943740844727,
      "loss_g0": 9.966943740844727,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.7485198974609375,
      "loss_g0": 5.823659420013428,
      "loss_g1": 1.2759279012680054,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.9913620948791504,
      "loss_g0": 3.035872220993042,
      "loss_g1": 0.3419455587863922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.370368957519531,
      "loss_g0": 6.2817888259887695,
      "loss_g1": 11.642797470092773,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522977501153946,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.738637924194336,
      "loss_g0": 23.824411392211914,
      "loss_g1": 18.633169174194336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 31.09902000427246,
      "loss_g0": 31.240304946899414,
      "loss_g1": 22.68910026550293,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.006196022033691,
      "loss_g0": 3.806429386138916,
      "loss_g1": 15.896162033081055,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523659229278564,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.716115951538086,
      "loss_g0": 20.659122467041016,
      "loss_g1": 24.108108520507812,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.01652400940656662,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 34.98296356201172,
      "loss_g0": 0.0,
      "loss_g1": 34.98296356201172,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.688657760620117,
      "loss_g0": 11.998817443847656,
      "loss_g1": 19.817846298217773,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 30.931686401367188,
      "loss_g0": 28.538610458374023,
      "loss_g1": 30.97188949584961,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.094425201416016,
      "loss_g0": 33.838584899902344,
      "loss_g1": 10.712267875671387,
      "q_g0": 0.01652478240430355,
      "q_g1": 0.9834752082824707,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 30.964540481567383,
      "loss_g0": 16.360641479492188,
      "loss_g1": 31.209897994995117,
      "q_g0": 0.016523264348506927,
      "q_g1": 0.9834766983985901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.585212707519531,
      "loss_g0": 1.3798493146896362,
      "loss_g1": 9.723062515258789,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.522388458251953,
      "loss_g0": 37.226783752441406,
      "loss_g1": 12.107288360595703,
      "q_g0": 0.016524991020560265,
      "q_g1": 0.9834749698638916,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.75537395477295,
      "loss_g0": 31.701831817626953,
      "loss_g1": 11.420177459716797,
      "q_g0": 0.01652706228196621,
      "q_g1": 0.9834729433059692,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.488188743591309,
      "loss_g0": 9.00619125366211,
      "loss_g1": 11.529897689819336,
      "q_g0": 0.016526803374290466,
      "q_g1": 0.9834731221199036,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.286555290222168,
      "loss_g0": 12.760734558105469,
      "loss_g1": 8.211366653442383,
      "q_g0": 0.016527269035577774,
      "q_g1": 0.9834727048873901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.172743320465088,
      "loss_g0": 15.633415222167969,
      "loss_g1": 3.980132579803467,
      "q_g0": 0.016528455540537834,
      "q_g1": 0.9834715127944946,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.67574691772461,
      "loss_g0": 10.852794647216797,
      "loss_g1": 11.689577102661133,
      "q_g0": 0.01652837172150612,
      "q_g1": 0.9834716320037842,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.534910202026367,
      "loss_g0": 9.875694274902344,
      "loss_g1": 19.697235107421875,
      "q_g0": 0.016527365893125534,
      "q_g1": 0.9834727048873901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.083178520202637,
      "loss_g0": 0.0,
      "loss_g1": 7.083178520202637,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.94328498840332,
      "loss_g0": 6.300477504730225,
      "loss_g1": 21.18928337097168,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.887880325317383,
      "loss_g0": 71.9711685180664,
      "loss_g1": 23.079833984375,
      "q_g0": 0.016527416184544563,
      "q_g1": 0.9834725856781006,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.429244995117188,
      "loss_g0": 29.566696166992188,
      "loss_g1": 13.158023834228516,
      "q_g0": 0.01652909815311432,
      "q_g1": 0.9834708571434021,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.965085983276367,
      "loss_g0": 10.9504976272583,
      "loss_g1": 6.898101806640625,
      "q_g0": 0.0165295097976923,
      "q_g1": 0.9834704995155334,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.293423652648926,
      "loss_g0": 20.010189056396484,
      "loss_g1": 4.029240131378174,
      "q_g0": 0.016531139612197876,
      "q_g1": 0.9834688901901245,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.858936309814453,
      "loss_g0": 11.131706237792969,
      "loss_g1": 9.837541580200195,
      "q_g0": 0.01653127186000347,
      "q_g1": 0.983468770980835,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.328543663024902,
      "loss_g0": 2.490009069442749,
      "loss_g1": 6.393064498901367,
      "q_g0": 0.016530873253941536,
      "q_g1": 0.9834691286087036,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.603111982345581,
      "loss_g0": 4.256399154663086,
      "loss_g1": 2.575321912765503,
      "q_g0": 0.016531046479940414,
      "q_g1": 0.9834690093994141,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.53900146484375,
      "loss_g0": 28.0135555267334,
      "loss_g1": 12.278864860534668,
      "q_g0": 0.0165326576679945,
      "q_g1": 0.9834674000740051,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.593019008636475,
      "loss_g0": 9.021810531616211,
      "loss_g1": 4.518566131591797,
      "q_g0": 0.016533123329281807,
      "q_g1": 0.9834669232368469,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.7425991296768188,
      "loss_g0": 7.777369499206543,
      "loss_g1": 1.6411442756652832,
      "q_g0": 0.016533760353922844,
      "q_g1": 0.9834662079811096,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.040477752685547,
      "loss_g0": 26.040477752685547,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.8694214820861816,
      "loss_g0": 2.7875137329101562,
      "loss_g1": 7.744879722595215,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.670440673828125,
      "loss_g0": 11.80555534362793,
      "loss_g1": 3.6278653144836426,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.507786750793457,
      "loss_g0": 8.608924865722656,
      "loss_g1": 2.487689971923828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.053399085998535,
      "loss_g0": 9.032646179199219,
      "loss_g1": 10.288670539855957,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.903177261352539,
      "loss_g0": 8.556415557861328,
      "loss_g1": 29.54084587097168,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524696722626686,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.548247337341309,
      "loss_g0": 12.732478141784668,
      "loss_g1": 1.5829167366027832,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523558646440506,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.601999282836914,
      "loss_g0": 8.699485778808594,
      "loss_g1": 2.7994065284729004,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522955149412155,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.57239055633545,
      "loss_g0": 9.711551666259766,
      "loss_g1": 1.2890005111694336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.215589761734009,
      "loss_g0": 3.220686197280884,
      "loss_g1": 2.9122347831726074,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.386251449584961,
      "loss_g0": 9.386251449584961,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.643536567687988,
      "loss_g0": 0.0,
      "loss_g1": 9.643536567687988,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.7073681354522705,
      "loss_g0": 13.268354415893555,
      "loss_g1": 1.513143539428711,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.344973087310791,
      "loss_g0": 5.733799934387207,
      "loss_g1": 4.321640491485596,
      "q_g0": 0.016522573307156563,
      "q_g1": 0.9834774136543274,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.879150152206421,
      "loss_g0": 17.982223510742188,
      "loss_g1": 2.625391721725464,
      "q_g0": 0.016524137929081917,
      "q_g1": 0.9834758043289185,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.783998489379883,
      "loss_g0": 2.5684354305267334,
      "loss_g1": 2.7876203060150146,
      "q_g0": 0.016524117439985275,
      "q_g1": 0.983475923538208,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.0589792728424072,
      "loss_g0": 3.6357383728027344,
      "loss_g1": 1.0156844854354858,
      "q_g0": 0.016524387523531914,
      "q_g1": 0.9834755659103394,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.2331892251968384,
      "loss_g0": 3.8768868446350098,
      "loss_g1": 1.188768982887268,
      "q_g0": 0.016524657607078552,
      "q_g1": 0.9834753274917603,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.593382358551025,
      "loss_g0": 11.293774604797363,
      "loss_g1": 5.497598648071289,
      "q_g0": 0.016525255516171455,
      "q_g1": 0.9834747910499573,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.664255142211914,
      "loss_g0": 1.3347450494766235,
      "loss_g1": 2.6865947246551514,
      "q_g0": 0.01652512326836586,
      "q_g1": 0.983474850654602,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.046614170074463,
      "loss_g0": 2.095531940460205,
      "loss_g1": 1.028989315032959,
      "q_g0": 0.016525238752365112,
      "q_g1": 0.9834747910499573,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 29.44420623779297,
      "loss_g0": 24.432750701904297,
      "loss_g1": 29.528409957885742,
      "q_g0": 0.01652471348643303,
      "q_g1": 0.9834753274917603,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 29.168989181518555,
      "loss_g0": 24.114744186401367,
      "loss_g1": 29.253910064697266,
      "q_g0": 0.016524186357855797,
      "q_g1": 0.9834758043289185,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.3687105178833,
      "loss_g0": 27.41562271118164,
      "loss_g1": 8.048648834228516,
      "q_g0": 0.016526170074939728,
      "q_g1": 0.9834738373756409,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 31.844507217407227,
      "loss_g0": 20.311960220336914,
      "loss_g1": 32.03828430175781,
      "q_g0": 0.01652497425675392,
      "q_g1": 0.9834749698638916,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.502574920654297,
      "loss_g0": 17.458986282348633,
      "loss_g1": 18.520109176635742,
      "q_g0": 0.016524866223335266,
      "q_g1": 0.9834750890731812,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.63751220703125,
      "loss_g0": 18.482410430908203,
      "loss_g1": 11.52249526977539,
      "q_g0": 0.01652558147907257,
      "q_g1": 0.9834744334220886,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.149894714355469,
      "loss_g0": 11.149894714355469,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.100611686706543,
      "loss_g0": 12.981178283691406,
      "loss_g1": 20.209747314453125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.330995559692383,
      "loss_g0": 14.330995559692383,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.575407028198242,
      "loss_g0": 0.0,
      "loss_g1": 10.575407028198242,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 27.219982147216797,
      "loss_g0": 27.219982147216797,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.561660766601562,
      "loss_g0": 18.734376907348633,
      "loss_g1": 8.28097915649414,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.16485023498535,
      "loss_g0": 20.303691864013672,
      "loss_g1": 11.900446891784668,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.152294397354126,
      "loss_g0": 3.0611703395843506,
      "loss_g1": 8.576148986816406,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522984951734543,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.718859672546387,
      "loss_g0": 7.682210922241211,
      "loss_g1": 9.90021800994873,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.0165232103317976,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.267505645751953,
      "loss_g0": 10.285484313964844,
      "loss_g1": 9.197379112243652,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523092985153198,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.249970436096191,
      "loss_g0": 9.300103187561035,
      "loss_g1": 6.265974998474121,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522781923413277,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 60.99340057373047,
      "loss_g0": 61.75394821166992,
      "loss_g1": 15.722770690917969,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 70.70481872558594,
      "loss_g0": 71.56744384765625,
      "loss_g1": 19.358089447021484,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.683792114257812,
      "loss_g0": 13.783556938171387,
      "loss_g1": 7.745394706726074,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.202613830566406,
      "loss_g0": 12.367465019226074,
      "loss_g1": 2.3899905681610107,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.159793853759766,
      "loss_g0": 17.046735763549805,
      "loss_g1": 23.889156341552734,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523122787475586,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.15903663635254,
      "loss_g0": 17.38056755065918,
      "loss_g1": 3.9726617336273193,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.857860565185547,
      "loss_g0": 19.09089469909668,
      "loss_g1": 4.986722469329834,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.3632755279541,
      "loss_g0": 0.0,
      "loss_g1": 17.3632755279541,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.927078247070312,
      "loss_g0": 32.22885513305664,
      "loss_g1": 14.636407852172852,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.183008193969727,
      "loss_g0": 24.994237899780273,
      "loss_g1": 5.866941452026367,
      "q_g0": 0.016524378210306168,
      "q_g1": 0.9834755659103394,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.016315460205078,
      "loss_g0": 7.261071681976318,
      "loss_g1": 16.16341209411621,
      "q_g0": 0.01652347296476364,
      "q_g1": 0.9834765195846558,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.020299911499023,
      "loss_g0": 6.686558723449707,
      "loss_g1": 9.05950927734375,
      "q_g0": 0.016523223370313644,
      "q_g1": 0.9834766983985901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.901016712188721,
      "loss_g0": 0.0,
      "loss_g1": 6.901016712188721,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.966146469116211,
      "loss_g0": 43.3143424987793,
      "loss_g1": 4.321896553039551,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.12162208557129,
      "loss_g0": 2.212364435195923,
      "loss_g1": 16.355297088623047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.221144676208496,
      "loss_g0": 21.591609954833984,
      "loss_g1": 10.030106544494629,
      "q_g0": 0.016523605212569237,
      "q_g1": 0.9834764003753662,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.5671002864837646,
      "loss_g0": 5.3524322509765625,
      "loss_g1": 3.537104368209839,
      "q_g0": 0.01652379147708416,
      "q_g1": 0.9834762811660767,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.121083974838257,
      "loss_g0": 5.9823317527771,
      "loss_g1": 2.0562081336975098,
      "q_g0": 0.016524193808436394,
      "q_g1": 0.9834758043289185,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.459572792053223,
      "loss_g0": 4.784588813781738,
      "loss_g1": 7.5045166015625,
      "q_g0": 0.01652391441166401,
      "q_g1": 0.9834760427474976,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.237042427062988,
      "loss_g0": 3.3966236114501953,
      "loss_g1": 7.301565647125244,
      "q_g0": 0.016523512080311775,
      "q_g1": 0.9834765195846558,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.699894905090332,
      "loss_g0": 6.699894905090332,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.112003326416016,
      "loss_g0": 10.214705467224121,
      "loss_g1": 3.9987757205963135,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.474994659423828,
      "loss_g0": 25.846532821655273,
      "loss_g1": 3.359623432159424,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.1108174324035645,
      "loss_g0": 4.1108174324035645,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.1270830631256104,
      "loss_g0": 2.0932955741882324,
      "loss_g1": 4.138242244720459,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.512810230255127,
      "loss_g0": 5.532924652099609,
      "loss_g1": 4.315536022186279,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.061802864074707,
      "loss_g0": 5.100337505340576,
      "loss_g1": 2.768070936203003,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.57926082611084,
      "loss_g0": 11.57926082611084,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.6050443649292,
      "loss_g0": 10.74090576171875,
      "loss_g1": 2.518052577972412,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.821581840515137,
      "loss_g0": 10.996236801147461,
      "loss_g1": 0.4254632294178009,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.898683547973633,
      "loss_g0": 2.9380202293395996,
      "loss_g1": 0.5572133660316467,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.7195969820022583,
      "loss_g0": 1.6270332336425781,
      "loss_g1": 7.229154109954834,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652299053966999,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.1985344886779785,
      "loss_g0": 5.278024673461914,
      "loss_g1": 0.4669780433177948,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 21.578977584838867,
      "loss_g0": 21.594989776611328,
      "loss_g1": 20.625864028930664,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.78264617919922,
      "loss_g0": 16.6919002532959,
      "loss_g1": 22.183944702148438,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522984951734543,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 51.5251350402832,
      "loss_g0": 51.5251350402832,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.165813446044922,
      "loss_g0": 17.90544319152832,
      "loss_g1": 33.664058685302734,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 46.957252502441406,
      "loss_g0": 47.338096618652344,
      "loss_g1": 24.287809371948242,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.816888809204102,
      "loss_g0": 9.670153617858887,
      "loss_g1": 18.550642013549805,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652333326637745,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.524072647094727,
      "loss_g0": 19.592941284179688,
      "loss_g1": 15.42487907409668,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522906720638275,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.08958625793457,
      "loss_g0": 18.98778533935547,
      "loss_g1": 25.14865493774414,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523534432053566,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 46.73200988769531,
      "loss_g0": 47.33270263671875,
      "loss_g1": 10.976499557495117,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.260534286499023,
      "loss_g0": 19.260534286499023,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.682085990905762,
      "loss_g0": 0.0,
      "loss_g1": 13.682085990905762,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.609336853027344,
      "loss_g0": 15.874629020690918,
      "loss_g1": 10.520879745483398,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.643328666687012,
      "loss_g0": 6.607908248901367,
      "loss_g1": 14.77832317352295,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.560823440551758,
      "loss_g0": 13.430778503417969,
      "loss_g1": 13.563008308410645,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.064218521118164,
      "loss_g0": 0.0,
      "loss_g1": 13.064218521118164,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.150209426879883,
      "loss_g0": 26.943782806396484,
      "loss_g1": 9.86807632446289,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.562674522399902,
      "loss_g0": 20.002466201782227,
      "loss_g1": 9.387273788452148,
      "q_g0": 0.016523512080311775,
      "q_g1": 0.9834765195846558,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.793935775756836,
      "loss_g0": 9.793935775756836,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.727455139160156,
      "loss_g0": 13.755167007446289,
      "loss_g1": 12.077884674072266,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 30.594478607177734,
      "loss_g0": 30.953510284423828,
      "loss_g1": 9.223532676696777,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.921791076660156,
      "loss_g0": 0.0,
      "loss_g1": 18.921791076660156,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.285146713256836,
      "loss_g0": 19.57573890686035,
      "loss_g1": 5.045064449310303,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.227664947509766,
      "loss_g0": 10.227533340454102,
      "loss_g1": 11.244466781616211,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.144137382507324,
      "loss_g0": 6.01406717300415,
      "loss_g1": 12.247121810913086,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.782413482666016,
      "loss_g0": 17.68459701538086,
      "loss_g1": 17.784055709838867,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.597671508789062,
      "loss_g0": 0.0,
      "loss_g1": 8.597671508789062,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.07622528076172,
      "loss_g0": 12.471307754516602,
      "loss_g1": 16.13678741455078,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.432114601135254,
      "loss_g0": 19.170377731323242,
      "loss_g1": 14.352509498596191,
      "q_g0": 0.016522914171218872,
      "q_g1": 0.9834770560264587,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.896336555480957,
      "loss_g0": 26.153112411499023,
      "loss_g1": 12.67359733581543,
      "q_g0": 0.016524286940693855,
      "q_g1": 0.9834756851196289,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.2352882623672485,
      "loss_g0": 12.079794883728027,
      "loss_g1": 1.0530668497085571,
      "q_g0": 0.01652541570365429,
      "q_g1": 0.9834745526313782,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.686447620391846,
      "loss_g0": 9.984210014343262,
      "loss_g1": 6.631033897399902,
      "q_g0": 0.016525758430361748,
      "q_g1": 0.9834741950035095,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.972665786743164,
      "loss_g0": 4.25035285949707,
      "loss_g1": 6.001605987548828,
      "q_g0": 0.01652558147907257,
      "q_g1": 0.9834744334220886,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.4496891498565674,
      "loss_g0": 9.185365676879883,
      "loss_g1": 3.3533074855804443,
      "q_g0": 0.016526177525520325,
      "q_g1": 0.9834738373756409,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.9467357397079468,
      "loss_g0": 9.013740539550781,
      "loss_g1": 1.8279772996902466,
      "q_g0": 0.01652691327035427,
      "q_g1": 0.9834730625152588,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.756355047225952,
      "loss_g0": 5.93360710144043,
      "loss_g1": 3.71976637840271,
      "q_g0": 0.01652713678777218,
      "q_g1": 0.9834728240966797,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.51560115814209,
      "loss_g0": 16.360795974731445,
      "loss_g1": 11.434175491333008,
      "q_g0": 0.01652764156460762,
      "q_g1": 0.9834723472595215,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.443118095397949,
      "loss_g0": 10.026947021484375,
      "loss_g1": 6.382889270782471,
      "q_g0": 0.01652800850570202,
      "q_g1": 0.9834719896316528,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.159911632537842,
      "loss_g0": 19.287321090698242,
      "loss_g1": 6.956085205078125,
      "q_g0": 0.016529269516468048,
      "q_g1": 0.9834707379341125,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.3656110763549805,
      "loss_g0": 26.67746353149414,
      "loss_g1": 2.9737484455108643,
      "q_g0": 0.016531696543097496,
      "q_g1": 0.9834682941436768,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.347546577453613,
      "loss_g0": 16.5883731842041,
      "loss_g1": 4.1417670249938965,
      "q_g0": 0.016532976180315018,
      "q_g1": 0.9834670424461365,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.324678897857666,
      "loss_g0": 8.677999496459961,
      "loss_g1": 6.285117149353027,
      "q_g0": 0.01653321646153927,
      "q_g1": 0.9834668040275574,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.7815276384353638,
      "loss_g0": 15.082608222961426,
      "loss_g1": 1.5579018592834473,
      "q_g0": 0.016534605994820595,
      "q_g1": 0.9834654331207275,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.4608988761901855,
      "loss_g0": 4.937251091003418,
      "loss_g1": 3.436077356338501,
      "q_g0": 0.016534756869077682,
      "q_g1": 0.9834651947021484,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 0.40447765588760376,
      "loss_g0": 6.2218241691589355,
      "loss_g1": 0.30666840076446533,
      "q_g0": 0.016535362228751183,
      "q_g1": 0.9834645986557007,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 0.5685956478118896,
      "loss_g0": 10.117172241210938,
      "loss_g1": 0.4080420732498169,
      "q_g0": 0.016536356881260872,
      "q_g1": 0.983463704586029,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.918529987335205,
      "loss_g0": 7.918529987335205,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.311163902282715,
      "loss_g0": 7.371955871582031,
      "loss_g1": 3.692586898803711,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.172919273376465,
      "loss_g0": 0.0,
      "loss_g1": 6.172919273376465,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.0211403369903564,
      "loss_g0": 2.0211403369903564,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.38453197479248,
      "loss_g0": 9.379271507263184,
      "loss_g1": 9.697671890258789,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.123846054077148,
      "loss_g0": 8.711224555969238,
      "loss_g1": 33.68074417114258,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.01652497425675392,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.623472213745117,
      "loss_g0": 15.406075477600098,
      "loss_g1": 28.560590744018555,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526324674487114,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.579133987426758,
      "loss_g0": 20.7440185546875,
      "loss_g1": 10.766386032104492,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.016525300219655037,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.23733139038086,
      "loss_g0": 18.193361282348633,
      "loss_g1": 20.854106903076172,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.016525574028491974,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.956411361694336,
      "loss_g0": 25.831315994262695,
      "loss_g1": 33.40079879760742,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.016526347026228905,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.43886661529541,
      "loss_g0": 15.290349960327148,
      "loss_g1": 24.276487350463867,
      "q_g0": 0.9834727048873901,
      "q_g1": 0.016527263447642326,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.794173240661621,
      "loss_g0": 4.677480220794678,
      "loss_g1": 11.737815856933594,
      "q_g0": 0.9834719896316528,
      "q_g1": 0.01652798242866993,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.991459846496582,
      "loss_g0": 14.130020141601562,
      "loss_g1": 5.746257781982422,
      "q_g0": 0.9834729433059692,
      "q_g1": 0.01652713306248188,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.271652698516846,
      "loss_g0": 3.927290916442871,
      "loss_g1": 24.7607479095459,
      "q_g0": 0.9834707379341125,
      "q_g1": 0.016529269516468048,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.221403121948242,
      "loss_g0": 20.10726547241211,
      "loss_g1": 27.01211929321289,
      "q_g0": 0.9834700226783752,
      "q_g1": 0.016529975458979607,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.337422370910645,
      "loss_g0": 15.33820915222168,
      "loss_g1": 15.290610313415527,
      "q_g0": 0.9834700226783752,
      "q_g1": 0.01652996800839901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.763832092285156,
      "loss_g0": 11.771129608154297,
      "loss_g1": 11.32968521118164,
      "q_g0": 0.9834700226783752,
      "q_g1": 0.016529912129044533,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 24.922210693359375,
      "loss_g0": 24.910795211791992,
      "loss_g1": 25.601375579833984,
      "q_g0": 0.9834700226783752,
      "q_g1": 0.016529982909560204,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.443428993225098,
      "loss_g0": 14.398449897766113,
      "loss_g1": 17.11945915222168,
      "q_g0": 0.9834696650505066,
      "q_g1": 0.01653026044368744,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 102.9053955078125,
      "loss_g0": 104.07056427001953,
      "loss_g1": 33.553138732910156,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652306318283081,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 24.492996215820312,
      "loss_g0": 23.72442626953125,
      "loss_g1": 70.22602081298828,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.0165278110653162,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.921815872192383,
      "loss_g0": 18.051259994506836,
      "loss_g1": 10.219075202941895,
      "q_g0": 0.9834730625152588,
      "q_g1": 0.01652701571583748,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.370648384094238,
      "loss_g0": 5.2863664627075195,
      "loss_g1": 10.38582992553711,
      "q_g0": 0.983472466468811,
      "q_g1": 0.016527533531188965,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.263860702514648,
      "loss_g0": 0.0,
      "loss_g1": 10.263860702514648,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.8441810607910156,
      "loss_g0": 5.092916965484619,
      "loss_g1": 1.789602279663086,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 28.2304744720459,
      "loss_g0": 11.080268859863281,
      "loss_g1": 28.518596649169922,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.041213989257812,
      "loss_g0": 9.029019355773926,
      "loss_g1": 14.125418663024902,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.552249908447266,
      "loss_g0": 11.524810791015625,
      "loss_g1": 13.586310386657715,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.64166784286499,
      "loss_g0": 15.413047790527344,
      "loss_g1": 4.460696220397949,
      "q_g0": 0.016523543745279312,
      "q_g1": 0.9834765195846558,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.7415266036987305,
      "loss_g0": 10.036558151245117,
      "loss_g1": 5.669363021850586,
      "q_g0": 0.01652398519217968,
      "q_g1": 0.9834760427474976,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.511296272277832,
      "loss_g0": 7.344503402709961,
      "loss_g1": 10.564501762390137,
      "q_g0": 0.01652364991605282,
      "q_g1": 0.9834764003753662,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.999224662780762,
      "loss_g0": 26.95504379272461,
      "loss_g1": 5.647092819213867,
      "q_g0": 0.01652582734823227,
      "q_g1": 0.9834741950035095,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.802102565765381,
      "loss_g0": 4.650929927825928,
      "loss_g1": 6.838249206542969,
      "q_g0": 0.016525596380233765,
      "q_g1": 0.9834744334220886,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.111066818237305,
      "loss_g0": 20.597143173217773,
      "loss_g1": 11.968465805053711,
      "q_g0": 0.0165264792740345,
      "q_g1": 0.9834734797477722,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.728133678436279,
      "loss_g0": 6.086610317230225,
      "loss_g1": 4.705305576324463,
      "q_g0": 0.016526618972420692,
      "q_g1": 0.9834733605384827,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.406963348388672,
      "loss_g0": 11.628886222839355,
      "loss_g1": 4.285597801208496,
      "q_g0": 0.016527369618415833,
      "q_g1": 0.9834725856781006,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.821565628051758,
      "loss_g0": 2.155013084411621,
      "loss_g1": 7.9167890548706055,
      "q_g0": 0.016526782885193825,
      "q_g1": 0.9834732413291931,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.212043762207031,
      "loss_g0": 2.9815919399261475,
      "loss_g1": 5.2495245933532715,
      "q_g0": 0.016526542603969574,
      "q_g1": 0.9834734797477722,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.614467620849609,
      "loss_g0": 11.497406005859375,
      "loss_g1": 4.498799800872803,
      "q_g0": 0.016527263447642326,
      "q_g1": 0.9834727048873901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.1915082931518555,
      "loss_g0": 6.533886909484863,
      "loss_g1": 6.185754776000977,
      "q_g0": 0.016527293249964714,
      "q_g1": 0.9834727048873901,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.914240837097168,
      "loss_g0": 23.081924438476562,
      "loss_g1": 9.692938804626465,
      "q_g0": 0.016528654843568802,
      "q_g1": 0.9834713339805603,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 21.774656295776367,
      "loss_g0": 16.87346839904785,
      "loss_g1": 21.857025146484375,
      "q_g0": 0.016528144478797913,
      "q_g1": 0.9834718704223633,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.285681962966919,
      "loss_g0": 16.386547088623047,
      "loss_g1": 2.048682689666748,
      "q_g0": 0.016529610380530357,
      "q_g1": 0.9834703803062439,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.412226676940918,
      "loss_g0": 6.412226676940918,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.176846981048584,
      "loss_g0": 0.0,
      "loss_g1": 3.176846981048584,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.953841686248779,
      "loss_g0": 7.8556718826293945,
      "loss_g1": 5.921891212463379,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.325355529785156,
      "loss_g0": 0.0,
      "loss_g1": 12.325355529785156,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.456775426864624,
      "loss_g0": 4.050153732299805,
      "loss_g1": 2.430006504058838,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.8008170127868652,
      "loss_g0": 12.652193069458008,
      "loss_g1": 3.6521058082580566,
      "q_g0": 0.016523340716958046,
      "q_g1": 0.9834766387939453,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.765016078948975,
      "loss_g0": 6.845198154449463,
      "loss_g1": 5.746868133544922,
      "q_g0": 0.0165234562009573,
      "q_g1": 0.9834765195846558,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.16503620147705,
      "loss_g0": 2.4221410751342773,
      "loss_g1": 8.261518478393555,
      "q_g0": 0.016522862017154694,
      "q_g1": 0.9834771752357483,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.9435142278671265,
      "loss_g0": 10.403833389282227,
      "loss_g1": 1.801369309425354,
      "q_g0": 0.01652374491095543,
      "q_g1": 0.9834762811660767,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.722700119018555,
      "loss_g0": 2.531968355178833,
      "loss_g1": 4.759506702423096,
      "q_g0": 0.016523512080311775,
      "q_g1": 0.9834765195846558,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.4489858150482178,
      "loss_g0": 30.7281494140625,
      "loss_g1": 1.9737781286239624,
      "q_g0": 0.016526447609066963,
      "q_g1": 0.9834734797477722,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.082154273986816,
      "loss_g0": 4.922967910766602,
      "loss_g1": 7.11843729019165,
      "q_g0": 0.016526224091649055,
      "q_g1": 0.9834737181663513,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 28.790071487426758,
      "loss_g0": 34.76026153564453,
      "loss_g1": 28.689743041992188,
      "q_g0": 0.01652684435248375,
      "q_g1": 0.9834731221199036,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.22543716430664,
      "loss_g0": 20.186363220214844,
      "loss_g1": 26.32691764831543,
      "q_g0": 0.01652621664106846,
      "q_g1": 0.9834737181663513,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.272987365722656,
      "loss_g0": 17.409372329711914,
      "loss_g1": 23.37151336669922,
      "q_g0": 0.01652561128139496,
      "q_g1": 0.9834744334220886,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 30.048912048339844,
      "loss_g0": 12.950996398925781,
      "loss_g1": 30.336181640625,
      "q_g0": 0.01652383618056774,
      "q_g1": 0.9834761619567871,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.447607040405273,
      "loss_g0": 19.447607040405273,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.068260192871094,
      "loss_g0": 18.068260192871094,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.645559310913086,
      "loss_g0": 25.907527923583984,
      "loss_g1": 10.052159309387207,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.727903366088867,
      "loss_g0": 27.053661346435547,
      "loss_g1": 7.337525367736816,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.235602378845215,
      "loss_g0": 14.38835334777832,
      "loss_g1": 5.1432294845581055,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 31.58631134033203,
      "loss_g0": 31.898868560791016,
      "loss_g1": 12.981640815734863,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 22.06639289855957,
      "loss_g0": 0.0,
      "loss_g1": 22.06639289855957,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.531380653381348,
      "loss_g0": 29.209030151367188,
      "loss_g1": 12.251195907592773,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.0822172164917,
      "loss_g0": 4.715558052062988,
      "loss_g1": 11.189177513122559,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.13734245300293,
      "loss_g0": 10.11502456665039,
      "loss_g1": 17.25531768798828,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.66040325164795,
      "loss_g0": 18.56098175048828,
      "loss_g1": 12.561270713806152,
      "q_g0": 0.01652303896844387,
      "q_g1": 0.9834769368171692,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.359162330627441,
      "loss_g0": 16.99305534362793,
      "loss_g1": 9.230901718139648,
      "q_g0": 0.016523828729987144,
      "q_g1": 0.9834761619567871,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.34379768371582,
      "loss_g0": 15.34379768371582,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.11658477783203,
      "loss_g0": 19.19010353088379,
      "loss_g1": 14.740377426147461,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.447083473205566,
      "loss_g0": 7.435296535491943,
      "loss_g1": 8.148673057556152,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522489488124847,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.228498458862305,
      "loss_g0": 16.108707427978516,
      "loss_g1": 23.3585205078125,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523223370313644,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.491167068481445,
      "loss_g0": 12.441228866577148,
      "loss_g1": 15.463400840759277,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523534432053566,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.396547317504883,
      "loss_g0": 18.68671417236328,
      "loss_g1": 1.1247367858886719,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 15.601825714111328,
      "loss_g0": 15.746317863464355,
      "loss_g1": 7.0010833740234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.533621788024902,
      "loss_g0": 9.627528190612793,
      "loss_g1": 3.943953514099121,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.130682945251465,
      "loss_g0": 7.191960334777832,
      "loss_g1": 3.4831953048706055,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.362028121948242,
      "loss_g0": 13.567655563354492,
      "loss_g1": 1.122280240058899,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.387886047363281,
      "loss_g0": 9.374991416931152,
      "loss_g1": 10.155380249023438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652250438928604,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.996107578277588,
      "loss_g0": 3.7495627403259277,
      "loss_g1": 18.669963836669922,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524024307727814,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.294042587280273,
      "loss_g0": 19.584596633911133,
      "loss_g1": 1.9992473125457764,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.882646560668945,
      "loss_g0": 16.91792106628418,
      "loss_g1": 14.782899856567383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.8766767978668213,
      "loss_g0": 2.587635040283203,
      "loss_g1": 20.079654693603516,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524210572242737,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 14.644125938415527,
      "loss_g0": 14.816421508789062,
      "loss_g1": 4.388894081115723,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523147001862526,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.426894426345825,
      "loss_g0": 3.3885812759399414,
      "loss_g1": 5.707296371459961,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652337983250618,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.746737003326416,
      "loss_g0": 4.606392860412598,
      "loss_g1": 13.099618911743164,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652424782514572,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.590560436248779,
      "loss_g0": 4.506402492523193,
      "loss_g1": 9.599218368530273,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524773091077805,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.228053092956543,
      "loss_g0": 8.292019844055176,
      "loss_g1": 4.420938014984131,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652437075972557,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 7.439004898071289,
      "loss_g0": 7.487246513366699,
      "loss_g1": 4.567755222320557,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524069011211395,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 17.0012149810791,
      "loss_g0": 17.16938018798828,
      "loss_g1": 6.991655349731445,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523031517863274,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.486220359802246,
      "loss_g0": 0.0,
      "loss_g1": 6.486220359802246,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.60421085357666,
      "loss_g0": 14.457179069519043,
      "loss_g1": 5.455480575561523,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.4284965991973877,
      "loss_g0": 21.387537002563477,
      "loss_g1": 2.1099460124969482,
      "q_g0": 0.01652439311146736,
      "q_g1": 0.9834755659103394,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.379518508911133,
      "loss_g0": 12.067380905151367,
      "loss_g1": 9.334356307983398,
      "q_g0": 0.016524672508239746,
      "q_g1": 0.9834753274917603,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.71796989440918,
      "loss_g0": 2.3545644283294678,
      "loss_g1": 5.774481773376465,
      "q_g0": 0.01652432605624199,
      "q_g1": 0.9834756851196289,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.232781410217285,
      "loss_g0": 11.890104293823242,
      "loss_g1": 9.188133239746094,
      "q_g0": 0.016524603590369225,
      "q_g1": 0.9834753274917603,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.961517333984375,
      "loss_g0": 16.7568302154541,
      "loss_g1": 2.7297043800354004,
      "q_g0": 0.016526037827134132,
      "q_g1": 0.9834739565849304,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 0.6191233992576599,
      "loss_g0": 9.287460327148438,
      "loss_g1": 0.4734548032283783,
      "q_g0": 0.016526944935321808,
      "q_g1": 0.9834730625152588,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.136741876602173,
      "loss_g0": 20.578386306762695,
      "loss_g1": 2.8436081409454346,
      "q_g0": 0.01652875542640686,
      "q_g1": 0.9834712147712708,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.369077682495117,
      "loss_g0": 16.714073181152344,
      "loss_g1": 8.228819847106934,
      "q_g0": 0.016529617831110954,
      "q_g1": 0.9834703803062439,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.6263043880462646,
      "loss_g0": 3.639328718185425,
      "loss_g1": 2.6092779636383057,
      "q_g0": 0.016529718413949013,
      "q_g1": 0.9834702610969543,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.3991785049438477,
      "loss_g0": 1.2456903457641602,
      "loss_g1": 2.4185657501220703,
      "q_g0": 0.016529595479369164,
      "q_g1": 0.9834703803062439,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 22.545944213867188,
      "loss_g0": 0.0,
      "loss_g1": 22.545944213867188,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 21.239362716674805,
      "loss_g0": 28.047340393066406,
      "loss_g1": 21.124988555908203,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 27.72966766357422,
      "loss_g0": 29.77475357055664,
      "loss_g1": 27.695308685302734,
      "q_g0": 0.016522636637091637,
      "q_g1": 0.9834774136543274,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 22.68145751953125,
      "loss_g0": 26.35134506225586,
      "loss_g1": 22.619800567626953,
      "q_g0": 0.01652301475405693,
      "q_g1": 0.9834769368171692,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 40.796043395996094,
      "loss_g0": 23.554805755615234,
      "loss_g1": 41.085693359375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 55.68523406982422,
      "loss_g0": 40.51774215698242,
      "loss_g1": 55.94004440307617,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.696252822875977,
      "loss_g0": 39.96763610839844,
      "loss_g1": 18.338844299316406,
      "q_g0": 0.01652463525533676,
      "q_g1": 0.9834753274917603,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.780059814453125,
      "loss_g0": 15.971694946289062,
      "loss_g1": 18.827245712280273,
      "q_g0": 0.01652434840798378,
      "q_g1": 0.9834756851196289,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 16.542360305786133,
      "loss_g0": 16.542360305786133,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 18.03072166442871,
      "loss_g0": 18.03072166442871,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.19011116027832,
      "loss_g0": 8.196381568908691,
      "loss_g1": 7.816860198974609,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 36.5554313659668,
      "loss_g0": 36.74538803100586,
      "loss_g1": 25.248355865478516,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.723539352416992,
      "loss_g0": 23.958858489990234,
      "loss_g1": 9.716504096984863,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.988645553588867,
      "loss_g0": 21.132068634033203,
      "loss_g1": 12.451570510864258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.463970184326172,
      "loss_g0": 20.463970184326172,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.865102767944336,
      "loss_g0": 13.865102767944336,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.123844146728516,
      "loss_g0": 10.847189903259277,
      "loss_g1": 27.591350555419922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 32.463600158691406,
      "loss_g0": 32.57826232910156,
      "loss_g1": 25.63857078552246,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 29.693435668945312,
      "loss_g0": 29.952423095703125,
      "loss_g1": 14.277507781982422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 32.081851959228516,
      "loss_g0": 32.42086410522461,
      "loss_g1": 11.902478218078613,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 37.0944709777832,
      "loss_g0": 37.62318420410156,
      "loss_g1": 5.623421669006348,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.4206981658935547,
      "loss_g0": 3.3959310054779053,
      "loss_g1": 4.894922733306885,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652258075773716,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.088425636291504,
      "loss_g0": 5.9896240234375,
      "loss_g1": 11.969170570373535,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523193567991257,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 5.363491058349609,
      "loss_g0": 0.0,
      "loss_g1": 5.363491058349609,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.3259222507476807,
      "loss_g0": 37.84199523925781,
      "loss_g1": 1.7292520999908447,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.87686824798584,
      "loss_g0": 19.096248626708984,
      "loss_g1": 6.671566486358643,
      "q_g0": 0.0165236983448267,
      "q_g1": 0.9834762811660767,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 8.135684967041016,
      "loss_g0": 4.97001314163208,
      "loss_g1": 8.188870429992676,
      "q_g0": 0.016523372381925583,
      "q_g1": 0.9834766387939453,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.2106804847717285,
      "loss_g0": 7.899930477142334,
      "loss_g1": 3.131894111633301,
      "q_g0": 0.01652386039495468,
      "q_g1": 0.9834761619567871,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.339348793029785,
      "loss_g0": 10.543432235717773,
      "loss_g1": 9.31911849975586,
      "q_g0": 0.01652398519217968,
      "q_g1": 0.9834760427474976,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.276196479797363,
      "loss_g0": 9.563196182250977,
      "loss_g1": 13.338580131530762,
      "q_g0": 0.01652359776198864,
      "q_g1": 0.9834764003753662,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.3453521728515625,
      "loss_g0": 5.515740394592285,
      "loss_g1": 4.325687885284424,
      "q_g0": 0.01652371883392334,
      "q_g1": 0.9834762811660767,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.895610809326172,
      "loss_g0": 4.95807409286499,
      "loss_g1": 9.978565216064453,
      "q_g0": 0.0165232103317976,
      "q_g1": 0.9834768176078796,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.2550379037857056,
      "loss_g0": 9.978885650634766,
      "loss_g1": 1.108461856842041,
      "q_g0": 0.016524117439985275,
      "q_g1": 0.983475923538208,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.958297729492188,
      "loss_g0": 23.4521427154541,
      "loss_g1": 13.798774719238281,
      "q_g0": 0.01652509905397892,
      "q_g1": 0.983474850654602,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.135679244995117,
      "loss_g0": 18.470916748046875,
      "loss_g1": 8.978811264038086,
      "q_g0": 0.016526076942682266,
      "q_g1": 0.9834739565849304,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.615338325500488,
      "loss_g0": 11.841123580932617,
      "loss_g1": 4.493912220001221,
      "q_g0": 0.016526827588677406,
      "q_g1": 0.9834731221199036,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.049464702606201,
      "loss_g0": 13.036500930786133,
      "loss_g1": 3.898432970046997,
      "q_g0": 0.01652776636183262,
      "q_g1": 0.9834722280502319,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.5634284019470215,
      "loss_g0": 4.250382423400879,
      "loss_g1": 4.568689346313477,
      "q_g0": 0.016527734696865082,
      "q_g1": 0.9834722280502319,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.4201760292053223,
      "loss_g0": 14.058944702148438,
      "loss_g1": 3.2413740158081055,
      "q_g0": 0.016528837382793427,
      "q_g1": 0.9834712147712708,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.575728893280029,
      "loss_g0": 6.7300801277160645,
      "loss_g1": 6.5731353759765625,
      "q_g0": 0.01652885414659977,
      "q_g1": 0.9834712147712708,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 0.7091304659843445,
      "loss_g0": 3.6843087673187256,
      "loss_g1": 0.6591266989707947,
      "q_g0": 0.016529161483049393,
      "q_g1": 0.9834708571434021,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.271248817443848,
      "loss_g0": 1.6389540433883667,
      "loss_g1": 4.315489292144775,
      "q_g0": 0.016528887674212456,
      "q_g1": 0.9834710955619812,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.859755516052246,
      "loss_g0": 10.859755516052246,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.469762802124023,
      "loss_g0": 9.548583984375,
      "loss_g1": 4.77800989151001,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.2006161212921143,
      "loss_g0": 3.1063218116760254,
      "loss_g1": 8.813175201416016,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523005440831184,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.0610170364379883,
      "loss_g0": 2.95141863822937,
      "loss_g1": 9.584220886230469,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523681581020355,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.113486289978027,
      "loss_g0": 13.283470153808594,
      "loss_g1": 2.9954988956451416,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 4.261999607086182,
      "loss_g0": 4.111579418182373,
      "loss_g1": 13.21493911743164,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523564234375954,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 37.8245964050293,
      "loss_g0": 38.432106018066406,
      "loss_g1": 1.663230299949646,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.09677505493164,
      "loss_g0": 9.244585037231445,
      "loss_g1": 0.29850059747695923,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 82.03114318847656,
      "loss_g0": 82.99710845947266,
      "loss_g1": 24.533489227294922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 28.487125396728516,
      "loss_g0": 28.68811798095703,
      "loss_g1": 16.5233097076416,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.257965087890625,
      "loss_g0": 26.376432418823242,
      "loss_g1": 19.206382751464844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 25.99382781982422,
      "loss_g0": 25.86762237548828,
      "loss_g1": 33.50563430786133,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523201018571854,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 19.38205909729004,
      "loss_g0": 19.463001251220703,
      "loss_g1": 14.564167022705078,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652269810438156,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.651691436767578,
      "loss_g0": 26.634817123413086,
      "loss_g1": 27.656150817871094,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522806137800217,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.928616523742676,
      "loss_g0": 11.929828643798828,
      "loss_g1": 11.856513023376465,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652279868721962,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.714648246765137,
      "loss_g0": 11.591876983642578,
      "loss_g1": 19.021968841552734,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523558646440506,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 13.356990814208984,
      "loss_g0": 13.475509643554688,
      "loss_g1": 6.3024516105651855,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 20.400388717651367,
      "loss_g0": 20.65898895263672,
      "loss_g1": 5.0074896812438965,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 6.973447799682617,
      "loss_g0": 7.015198707580566,
      "loss_g1": 4.488290786743164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 9.350387573242188,
      "loss_g0": 9.379128456115723,
      "loss_g1": 7.6395769119262695,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 12.833641052246094,
      "loss_g0": 12.902304649353027,
      "loss_g1": 8.7464599609375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 24.1008243560791,
      "loss_g0": 24.406822204589844,
      "loss_g1": 5.8865766525268555,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 2.1538519859313965,
      "loss_g0": 2.1225357055664062,
      "loss_g1": 4.017877101898193,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522621735930443,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 23.522628784179688,
      "loss_g0": 23.855594635009766,
      "loss_g1": 3.7031736373901367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 3.597609758377075,
      "loss_g0": 3.6457114219665527,
      "loss_g1": 0.7344168424606323,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 10.484057426452637,
      "loss_g0": 10.583375930786133,
      "loss_g1": 4.5722503662109375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 1.7209062576293945,
      "loss_g0": 1.7179784774780273,
      "loss_g1": 1.8951823711395264,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "eval_cer": 0.06203454595079347,
      "eval_loss": 13.884725570678711,
      "eval_runtime": 81.0016,
      "eval_samples_per_second": 20.691,
      "eval_steps_per_second": 5.173,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 26.6298885345459,
      "loss_g0": 26.20008087158203,
      "loss_g1": 52.209449768066406,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525106504559517,
      "step": 900
    },
    {
      "epoch": 4.568527918781726,
      "gdro_loss": 11.387237548828125,
      "loss_g0": 11.130908966064453,
      "loss_g1": 26.640886306762695,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.016526687890291214,
      "step": 900
    },
    {
      "epoch": 5.0761421319796955,
      "grad_norm": 400.7965393066406,
      "learning_rate": 0.00024156149732620322,
      "loss": 18.1676,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 25.363750457763672,
      "loss_g0": 25.222335815429688,
      "loss_g1": 33.76264190673828,
      "q_g0": 0.9834416508674622,
      "q_g1": 0.01655837893486023,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 30.437175750732422,
      "loss_g0": 30.649864196777344,
      "loss_g1": 17.804048538208008,
      "q_g0": 0.9834429621696472,
      "q_g1": 0.01655706577003002,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 27.78239631652832,
      "loss_g0": 27.78289794921875,
      "loss_g1": 27.75262451171875,
      "q_g0": 0.9834429621696472,
      "q_g1": 0.01655706577003002,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.96289825439453,
      "loss_g0": 17.006324768066406,
      "loss_g1": 14.383472442626953,
      "q_g0": 0.9834432005882263,
      "q_g1": 0.01655680313706398,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 28.7857723236084,
      "loss_g0": 28.90688133239746,
      "loss_g1": 21.5916805267334,
      "q_g0": 0.9834439754486084,
      "q_g1": 0.01655604876577854,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.882673263549805,
      "loss_g0": 10.55624771118164,
      "loss_g1": 30.270267486572266,
      "q_g0": 0.9834418892860413,
      "q_g1": 0.016558067873120308,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.361007690429688,
      "loss_g0": 9.331613540649414,
      "loss_g1": 11.10682487487793,
      "q_g0": 0.9834417700767517,
      "q_g1": 0.016558246687054634,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 21.11859130859375,
      "loss_g0": 21.279956817626953,
      "loss_g1": 11.533984184265137,
      "q_g0": 0.9834427237510681,
      "q_g1": 0.016557244583964348,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 61.110877990722656,
      "loss_g0": 61.91569137573242,
      "loss_g1": 13.292859077453613,
      "q_g0": 0.983447790145874,
      "q_g1": 0.016552262008190155,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.75255012512207,
      "loss_g0": 17.89552116394043,
      "loss_g1": 9.257543563842773,
      "q_g0": 0.9834485650062561,
      "q_g1": 0.01655137538909912,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.433082580566406,
      "loss_g0": 15.549566268920898,
      "loss_g1": 8.511528015136719,
      "q_g0": 0.983449399471283,
      "q_g1": 0.01655065454542637,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.905251979827881,
      "loss_g0": 7.831866264343262,
      "loss_g1": 12.265729904174805,
      "q_g0": 0.9834489226341248,
      "q_g1": 0.016551105305552483,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.531173706054688,
      "loss_g0": 17.577138900756836,
      "loss_g1": 14.799917221069336,
      "q_g0": 0.9834491610527039,
      "q_g1": 0.016550816595554352,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.658658981323242,
      "loss_g0": 8.53738021850586,
      "loss_g1": 15.86474609375,
      "q_g0": 0.9834485054016113,
      "q_g1": 0.016551561653614044,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.545902252197266,
      "loss_g0": 17.626972198486328,
      "loss_g1": 12.72891902923584,
      "q_g0": 0.9834489226341248,
      "q_g1": 0.016551056876778603,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.799904823303223,
      "loss_g0": 15.875202178955078,
      "loss_g1": 11.325690269470215,
      "q_g0": 0.983449399471283,
      "q_g1": 0.016550583764910698,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.722559928894043,
      "loss_g0": 12.722559928894043,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.80137825012207,
      "loss_g0": 18.97745132446289,
      "loss_g1": 8.320833206176758,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.363019943237305,
      "loss_g0": 18.282501220703125,
      "loss_g1": 23.155677795410156,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522914171218872,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.588868618011475,
      "loss_g0": 6.378676891326904,
      "loss_g1": 19.098888397216797,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524210572242737,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 65.61439514160156,
      "loss_g0": 66.52210235595703,
      "loss_g1": 11.584254264831543,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.174360275268555,
      "loss_g0": 13.252016067504883,
      "loss_g1": 8.551965713500977,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 20.946685791015625,
      "loss_g0": 20.946685791015625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.87091588973999,
      "loss_g0": 4.856057643890381,
      "loss_g1": 5.755321502685547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.761581420898438,
      "loss_g0": 0.0,
      "loss_g1": 8.761581420898438,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.150612831115723,
      "loss_g0": 11.946321487426758,
      "loss_g1": 6.0532450675964355,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.329957008361816,
      "loss_g0": 5.53272008895874,
      "loss_g1": 14.477750778198242,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 41.88108444213867,
      "loss_g0": 13.066737174987793,
      "loss_g1": 42.36516571044922,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.225255012512207,
      "loss_g0": 7.667660713195801,
      "loss_g1": 8.234622955322266,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.37082576751709,
      "loss_g0": 7.37082576751709,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.035879611968994,
      "loss_g0": 7.0510759353637695,
      "loss_g1": 6.131329536437988,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.092382431030273,
      "loss_g0": 0.0,
      "loss_g1": 9.092382431030273,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.015854835510254,
      "loss_g0": 12.765081405639648,
      "loss_g1": 4.885667324066162,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.872295618057251,
      "loss_g0": 8.62551498413086,
      "loss_g1": 3.7924392223358154,
      "q_g0": 0.01652292162179947,
      "q_g1": 0.9834770560264587,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.766952514648438,
      "loss_g0": 12.111848831176758,
      "loss_g1": 12.777958869934082,
      "q_g0": 0.016522854566574097,
      "q_g1": 0.9834771752357483,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.1921546459198,
      "loss_g0": 15.361677169799805,
      "loss_g1": 1.9708820581436157,
      "q_g0": 0.01652422547340393,
      "q_g1": 0.9834758043289185,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.889079093933105,
      "loss_g0": 14.522418022155762,
      "loss_g1": 10.828031539916992,
      "q_g0": 0.016524603590369225,
      "q_g1": 0.9834753274917603,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.134481430053711,
      "loss_g0": 20.043346405029297,
      "loss_g1": 3.867149591445923,
      "q_g0": 0.016526253893971443,
      "q_g1": 0.9834737181663513,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.370196342468262,
      "loss_g0": 11.281333923339844,
      "loss_g1": 7.304471969604492,
      "q_g0": 0.016526658087968826,
      "q_g1": 0.9834733605384827,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.04015588760376,
      "loss_g0": 1.1406898498535156,
      "loss_g1": 4.088878631591797,
      "q_g0": 0.0165263619273901,
      "q_g1": 0.9834735989570618,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.751924514770508,
      "loss_g0": 2.441632032394409,
      "loss_g1": 8.857958793640137,
      "q_g0": 0.01652570627629757,
      "q_g1": 0.9834743142127991,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.1891062259674072,
      "loss_g0": 15.23333740234375,
      "loss_g1": 2.9867067337036133,
      "q_g0": 0.01652696169912815,
      "q_g1": 0.9834730625152588,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.0976017713546753,
      "loss_g0": 7.947738170623779,
      "loss_g1": 0.9824823141098022,
      "q_g0": 0.016527673229575157,
      "q_g1": 0.9834723472595215,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.6892991065979,
      "loss_g0": 0.0,
      "loss_g1": 4.6892991065979,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.1848580837249756,
      "loss_g0": 1.251541256904602,
      "loss_g1": 2.20053768157959,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.670467376708984,
      "loss_g0": 8.670467376708984,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.116241455078125,
      "loss_g0": 8.2169189453125,
      "loss_g1": 2.1234960556030273,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.925481796264648,
      "loss_g0": 13.002278327941895,
      "loss_g1": 8.354241371154785,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.2917871475219727,
      "loss_g0": 1.280434012413025,
      "loss_g1": 1.9675681591033936,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.57024097442627,
      "loss_g0": 8.535430908203125,
      "loss_g1": 10.64221477508545,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522705554962158,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.577858924865723,
      "loss_g0": 14.417439460754395,
      "loss_g1": 24.125892639160156,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.0165236983448267,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 30.8002986907959,
      "loss_g0": 30.89923858642578,
      "loss_g1": 24.9111385345459,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523092985153198,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 41.510406494140625,
      "loss_g0": 41.657623291015625,
      "loss_g1": 32.747650146484375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 31.318944931030273,
      "loss_g0": 31.25570297241211,
      "loss_g1": 35.08324432373047,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522813588380814,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.203805923461914,
      "loss_g0": 24.1396484375,
      "loss_g1": 28.022506713867188,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523215919733047,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.618255615234375,
      "loss_g0": 18.814510345458984,
      "loss_g1": 6.9364728927612305,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.388739824295044,
      "loss_g0": 2.962918281555176,
      "loss_g1": 28.731151580810547,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.01652505435049534,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.823890209197998,
      "loss_g0": 4.493001461029053,
      "loss_g1": 24.51396369934082,
      "q_g0": 0.9834729433059692,
      "q_g1": 0.016527099534869194,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 27.442590713500977,
      "loss_g0": 27.558704376220703,
      "loss_g1": 20.532760620117188,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.016526388004422188,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 25.603744506835938,
      "loss_g0": 25.655126571655273,
      "loss_g1": 22.545833587646484,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652606949210167,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.027189254760742,
      "loss_g0": 0.0,
      "loss_g1": 14.027189254760742,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.752887725830078,
      "loss_g0": 10.942045211791992,
      "loss_g1": 15.833709716796875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.26149559020996,
      "loss_g0": 12.855392456054688,
      "loss_g1": 19.369117736816406,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 46.249305725097656,
      "loss_g0": 15.226297378540039,
      "loss_g1": 46.77049255371094,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.733434677124023,
      "loss_g0": 30.660106658935547,
      "loss_g1": 10.398624420166016,
      "q_g0": 0.016524488106369972,
      "q_g1": 0.9834754467010498,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.10236930847168,
      "loss_g0": 36.544837951660156,
      "loss_g1": 16.775653839111328,
      "q_g0": 0.016526510939002037,
      "q_g1": 0.9834734797477722,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.904319763183594,
      "loss_g0": 20.22126007080078,
      "loss_g1": 10.747746467590332,
      "q_g0": 0.016527481377124786,
      "q_g1": 0.9834725856781006,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.37335205078125,
      "loss_g0": 26.232608795166016,
      "loss_g1": 17.224462509155273,
      "q_g0": 0.016528403386473656,
      "q_g1": 0.9834716320037842,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 23.546958923339844,
      "loss_g0": 10.860380172729492,
      "loss_g1": 23.76015281677246,
      "q_g0": 0.01652708277106285,
      "q_g1": 0.9834729433059692,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.418828964233398,
      "loss_g0": 40.73773956298828,
      "loss_g1": 10.926037788391113,
      "q_g0": 0.016530122607946396,
      "q_g1": 0.9834699034690857,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.093650817871094,
      "loss_g0": 9.019454002380371,
      "loss_g1": 17.229352951049805,
      "q_g0": 0.016529284417629242,
      "q_g1": 0.9834707379341125,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.664268493652344,
      "loss_g0": 12.657005310058594,
      "loss_g1": 10.630775451660156,
      "q_g0": 0.016529494896531105,
      "q_g1": 0.9834704995155334,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.368062496185303,
      "loss_g0": 7.857247829437256,
      "loss_g1": 7.359840393066406,
      "q_g0": 0.016529541462659836,
      "q_g1": 0.9834704995155334,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.787887573242188,
      "loss_g0": 0.0,
      "loss_g1": 14.787887573242188,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.84335470199585,
      "loss_g0": 6.373515605926514,
      "loss_g1": 6.851247787475586,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.7316107749938965,
      "loss_g0": 10.21179485321045,
      "loss_g1": 5.656341552734375,
      "q_g0": 0.016522889956831932,
      "q_g1": 0.9834770560264587,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.448654174804688,
      "loss_g0": 17.448654174804688,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.68128204345703,
      "loss_g0": 17.619407653808594,
      "loss_g1": 21.364276885986328,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.434187889099121,
      "loss_g0": 11.381004333496094,
      "loss_g1": 14.599733352661133,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522753983736038,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 34.84471130371094,
      "loss_g0": 35.286128997802734,
      "loss_g1": 8.5698823928833,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.501723289489746,
      "loss_g0": 6.421990871429443,
      "loss_g1": 11.247557640075684,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522914171218872,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.2966716289520264,
      "loss_g0": 3.311434030532837,
      "loss_g1": 2.417980194091797,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.556589126586914,
      "loss_g0": 13.724467277526855,
      "loss_g1": 3.5637917518615723,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.41680145263672,
      "loss_g0": 17.16689109802246,
      "loss_g1": 32.29092788696289,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523968428373337,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.006834030151367,
      "loss_g0": 11.057709693908691,
      "loss_g1": 7.978716850280762,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652364991605282,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.194354057312012,
      "loss_g0": 15.334335327148438,
      "loss_g1": 6.862311363220215,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522789373993874,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 26.373899459838867,
      "loss_g0": 26.70195198059082,
      "loss_g1": 6.846936225891113,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.780941009521484,
      "loss_g0": 6.767593860626221,
      "loss_g1": 7.575407981872559,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652250438928604,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.908735275268555,
      "loss_g0": 6.971398830413818,
      "loss_g1": 3.1787471771240234,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.98673677444458,
      "loss_g0": 4.800871849060059,
      "loss_g1": 16.049345016479492,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.0165235735476017,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.034540176391602,
      "loss_g0": 12.1283540725708,
      "loss_g1": 6.450578212738037,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652299053966999,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.370977401733398,
      "loss_g0": 9.249638557434082,
      "loss_g1": 16.59295654296875,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652374491095543,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.636088848114014,
      "loss_g0": 4.630499362945557,
      "loss_g1": 4.968799591064453,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523784026503563,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.380635261535645,
      "loss_g0": 13.577568054199219,
      "loss_g1": 1.6585521697998047,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522565856575966,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.273490905761719,
      "loss_g0": 4.323225021362305,
      "loss_g1": 1.313132643699646,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.478670120239258,
      "loss_g0": 12.641752243041992,
      "loss_g1": 2.77138614654541,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.822991371154785,
      "loss_g0": 16.06818389892578,
      "loss_g1": 1.2281899452209473,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 0.6806389689445496,
      "loss_g0": 0.6446215510368347,
      "loss_g1": 2.8245062828063965,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.790804862976074,
      "loss_g0": 16.022226333618164,
      "loss_g1": 2.0157086849212646,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.888852119445801,
      "loss_g0": 5.888852119445801,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 52.40540313720703,
      "loss_g0": 52.986717224121094,
      "loss_g1": 17.803302764892578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.015178680419922,
      "loss_g0": 11.986587524414062,
      "loss_g1": 13.716995239257812,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522597521543503,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.132320404052734,
      "loss_g0": 13.905632972717285,
      "loss_g1": 27.62428855895996,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523992642760277,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 20.0686092376709,
      "loss_g0": 20.128435134887695,
      "loss_g1": 16.507783889770508,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652361825108528,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.923974990844727,
      "loss_g0": 0.0,
      "loss_g1": 24.923974990844727,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 31.622493743896484,
      "loss_g0": 30.099960327148438,
      "loss_g1": 31.6480712890625,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.185036659240723,
      "loss_g0": 34.1092529296875,
      "loss_g1": 10.79985237121582,
      "q_g0": 0.01652480475604534,
      "q_g1": 0.9834752082824707,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 33.026424407958984,
      "loss_g0": 41.06464767456055,
      "loss_g1": 32.891353607177734,
      "q_g0": 0.016525642946362495,
      "q_g1": 0.9834743142127991,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 26.48830223083496,
      "loss_g0": 25.790115356445312,
      "loss_g1": 26.500032424926758,
      "q_g0": 0.016525574028491974,
      "q_g1": 0.9834744334220886,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.76102066040039,
      "loss_g0": 22.483707427978516,
      "loss_g1": 17.68166160583496,
      "q_g0": 0.01652606949210167,
      "q_g1": 0.9834739565849304,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.196296691894531,
      "loss_g0": 22.820974349975586,
      "loss_g1": 8.967330932617188,
      "q_g0": 0.016527486965060234,
      "q_g1": 0.983472466468811,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.41474151611328,
      "loss_g0": 17.10727882385254,
      "loss_g1": 16.40310287475586,
      "q_g0": 0.016527565196156502,
      "q_g1": 0.983472466468811,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.997934341430664,
      "loss_g0": 0.0,
      "loss_g1": 16.997934341430664,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.49095344543457,
      "loss_g0": 18.158084869384766,
      "loss_g1": 8.328545570373535,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.573226928710938,
      "loss_g0": 7.251836776733398,
      "loss_g1": 16.729825973510742,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 42.16709899902344,
      "loss_g0": 12.468526840209961,
      "loss_g1": 42.66603469848633,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 20.518869400024414,
      "loss_g0": 16.894763946533203,
      "loss_g1": 20.579753875732422,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.115156173706055,
      "loss_g0": 5.793840408325195,
      "loss_g1": 13.238154411315918,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.61929988861084,
      "loss_g0": 29.27741050720215,
      "loss_g1": 6.238588333129883,
      "q_g0": 0.016524773091077805,
      "q_g1": 0.9834752082824707,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.525762557983398,
      "loss_g0": 5.322088718414307,
      "loss_g1": 9.596392631530762,
      "q_g0": 0.016524333506822586,
      "q_g1": 0.9834756851196289,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.711096286773682,
      "loss_g0": 27.772523880004883,
      "loss_g1": 6.357174873352051,
      "q_g0": 0.01652652584016323,
      "q_g1": 0.9834734797477722,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.24912166595459,
      "loss_g0": 29.230180740356445,
      "loss_g1": 12.980545043945312,
      "q_g0": 0.016528191044926643,
      "q_g1": 0.9834717512130737,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.484004974365234,
      "loss_g0": 11.911398887634277,
      "loss_g1": 5.375982284545898,
      "q_g0": 0.016528859734535217,
      "q_g1": 0.9834712147712708,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.486405849456787,
      "loss_g0": 7.81651496887207,
      "loss_g1": 6.464051246643066,
      "q_g0": 0.01652899757027626,
      "q_g1": 0.9834709763526917,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.634413719177246,
      "loss_g0": 13.82765007019043,
      "loss_g1": 7.53032112121582,
      "q_g0": 0.016529632732272148,
      "q_g1": 0.9834703803062439,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.75044059753418,
      "loss_g0": 12.029836654663086,
      "loss_g1": 15.812973022460938,
      "q_g0": 0.016529247164726257,
      "q_g1": 0.9834707379341125,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.872052192687988,
      "loss_g0": 15.443262100219727,
      "loss_g1": 6.727987766265869,
      "q_g0": 0.01653013750910759,
      "q_g1": 0.9834697842597961,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.844076156616211,
      "loss_g0": 4.767666816711426,
      "loss_g1": 15.01343059539795,
      "q_g0": 0.01652909256517887,
      "q_g1": 0.9834709763526917,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.218419551849365,
      "loss_g0": 4.156757354736328,
      "loss_g1": 4.219456195831299,
      "q_g0": 0.016529085114598274,
      "q_g1": 0.9834709763526917,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.642153263092041,
      "loss_g0": 17.375648498535156,
      "loss_g1": 4.428125381469727,
      "q_g0": 0.01653040200471878,
      "q_g1": 0.9834696054458618,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.013631343841553,
      "loss_g0": 7.140806198120117,
      "loss_g1": 7.011493682861328,
      "q_g0": 0.016530416905879974,
      "q_g1": 0.9834696054458618,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.615950584411621,
      "loss_g0": 0.0,
      "loss_g1": 9.615950584411621,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.16626262664795,
      "loss_g0": 24.424217224121094,
      "loss_g1": 7.893128871917725,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.6131768226623535,
      "loss_g0": 12.219866752624512,
      "loss_g1": 7.5357818603515625,
      "q_g0": 0.016522899270057678,
      "q_g1": 0.9834770560264587,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.457289218902588,
      "loss_g0": 18.125978469848633,
      "loss_g1": 5.244431495666504,
      "q_g0": 0.016524210572242737,
      "q_g1": 0.9834758043289185,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.736729860305786,
      "loss_g0": 5.692541122436523,
      "loss_g1": 2.687065601348877,
      "q_g0": 0.01652451977133751,
      "q_g1": 0.9834754467010498,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.911103248596191,
      "loss_g0": 24.25787353515625,
      "loss_g1": 12.720438957214355,
      "q_g0": 0.016525696963071823,
      "q_g1": 0.9834743142127991,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.643383979797363,
      "loss_g0": 7.738502502441406,
      "loss_g1": 5.608178615570068,
      "q_g0": 0.016525914892554283,
      "q_g1": 0.98347407579422,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.323740005493164,
      "loss_g0": 7.102441787719727,
      "loss_g1": 7.327458381652832,
      "q_g0": 0.016525890678167343,
      "q_g1": 0.98347407579422,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.861801147460938,
      "loss_g0": 15.861801147460938,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.029727458953857,
      "loss_g0": 7.083258628845215,
      "loss_g1": 3.843350887298584,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.539090156555176,
      "loss_g0": 12.658066749572754,
      "loss_g1": 5.457109451293945,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.5068283081054688,
      "loss_g0": 3.5084388256073,
      "loss_g1": 3.410970449447632,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.963673114776611,
      "loss_g0": 8.09036922454834,
      "loss_g1": 0.4222121834754944,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.45208740234375,
      "loss_g0": 4.45208740234375,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.628571510314941,
      "loss_g0": 7.735340118408203,
      "loss_g1": 1.273289442062378,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.159407615661621,
      "loss_g0": 8.279687881469727,
      "loss_g1": 0.9998548626899719,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.392090082168579,
      "loss_g0": 3.3544206619262695,
      "loss_g1": 5.634284496307373,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.33455753326416,
      "loss_g0": 9.423013687133789,
      "loss_g1": 4.0693182945251465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 0.9442704916000366,
      "loss_g0": 0.8401466608047485,
      "loss_g1": 7.141875743865967,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652306318283081,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 100.29459381103516,
      "loss_g0": 101.23596954345703,
      "loss_g1": 44.260337829589844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 23.027328491210938,
      "loss_g0": 23.12599754333496,
      "loss_g1": 17.15416717529297,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 25.881328582763672,
      "loss_g0": 25.881328582763672,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 31.08538818359375,
      "loss_g0": 31.324913024902344,
      "loss_g1": 16.827877044677734,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.3045654296875,
      "loss_g0": 17.36186981201172,
      "loss_g1": 13.893579483032227,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.24980926513672,
      "loss_g0": 19.1312255859375,
      "loss_g1": 26.30807876586914,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523154452443123,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 21.647977828979492,
      "loss_g0": 21.634401321411133,
      "loss_g1": 22.456045150756836,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523240134119987,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 32.99601364135742,
      "loss_g0": 33.263179779052734,
      "loss_g1": 17.09311294555664,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 32.18003463745117,
      "loss_g0": 32.479881286621094,
      "loss_g1": 14.331876754760742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.29300308227539,
      "loss_g0": 14.29300308227539,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 55.898006439208984,
      "loss_g0": 56.58509826660156,
      "loss_g1": 14.999614715576172,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 21.79974365234375,
      "loss_g0": 21.9703369140625,
      "loss_g1": 11.645383834838867,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.961047172546387,
      "loss_g0": 0.0,
      "loss_g1": 14.961047172546387,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.8212661743164062,
      "loss_g0": 20.32012176513672,
      "loss_g1": 3.5440852642059326,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.348176956176758,
      "loss_g0": 18.014482498168945,
      "loss_g1": 11.236178398132324,
      "q_g0": 0.01652311533689499,
      "q_g1": 0.9834768176078796,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.772777557373047,
      "loss_g0": 0.0,
      "loss_g1": 13.772777557373047,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.323063850402832,
      "loss_g0": 59.254825592041016,
      "loss_g1": 11.5346097946167,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.38491439819336,
      "loss_g0": 16.38491439819336,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.989643096923828,
      "loss_g0": 4.9559326171875,
      "loss_g1": 6.996201515197754,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.8587216138839722,
      "loss_g0": 1.7608691453933716,
      "loss_g1": 7.683052062988281,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523022204637527,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.149356842041016,
      "loss_g0": 0.0,
      "loss_g1": 10.149356842041016,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.105314254760742,
      "loss_g0": 10.817424774169922,
      "loss_g1": 8.0597505569458,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.927505970001221,
      "loss_g0": 13.64526653289795,
      "loss_g1": 6.814642906188965,
      "q_g0": 0.016523122787475586,
      "q_g1": 0.9834768176078796,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.9618687629699707,
      "loss_g0": 16.72905158996582,
      "loss_g1": 2.7305498123168945,
      "q_g0": 0.016524557024240494,
      "q_g1": 0.9834754467010498,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.622605323791504,
      "loss_g0": 0.0,
      "loss_g1": 12.622605323791504,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.694169998168945,
      "loss_g0": 9.694169998168945,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.885185241699219,
      "loss_g0": 5.958333969116211,
      "loss_g1": 1.5310933589935303,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.840312957763672,
      "loss_g0": 12.000313758850098,
      "loss_g1": 2.3164405822753906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.065010070800781,
      "loss_g0": 10.090126991271973,
      "loss_g1": 8.569911003112793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.218521118164062,
      "loss_g0": 16.46305274963379,
      "loss_g1": 1.6630127429962158,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.590898513793945,
      "loss_g0": 12.664794921875,
      "loss_g1": 8.192306518554688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.003801345825195,
      "loss_g0": 7.003801345825195,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.5837738513946533,
      "loss_g0": 3.5408332347869873,
      "loss_g1": 6.139769554138184,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.13248348236084,
      "loss_g0": 13.248788833618164,
      "loss_g1": 6.209532737731934,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.941797256469727,
      "loss_g0": 10.967905044555664,
      "loss_g1": 9.387746810913086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.154906272888184,
      "loss_g0": 12.157045364379883,
      "loss_g1": 12.027583122253418,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.6890549659729,
      "loss_g0": 0.0,
      "loss_g1": 5.6890549659729,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.299999237060547,
      "loss_g0": 12.820147514343262,
      "loss_g1": 2.123260736465454,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.681868553161621,
      "loss_g0": 35.358253479003906,
      "loss_g1": 14.33445930480957,
      "q_g0": 0.01652456447482109,
      "q_g1": 0.9834754467010498,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.1962125301361084,
      "loss_g0": 3.7498602867126465,
      "loss_g1": 1.1533048152923584,
      "q_g0": 0.01652483083307743,
      "q_g1": 0.9834752082824707,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.6056129932403564,
      "loss_g0": 5.436160087585449,
      "loss_g1": 3.574854850769043,
      "q_g0": 0.0165250226855278,
      "q_g1": 0.9834749698638916,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.365665435791016,
      "loss_g0": 5.683374404907227,
      "loss_g1": 4.343523979187012,
      "q_g0": 0.016525164246559143,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.9134504795074463,
      "loss_g0": 5.6644439697265625,
      "loss_g1": 2.867225170135498,
      "q_g0": 0.016525447368621826,
      "q_g1": 0.9834745526313782,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.744856357574463,
      "loss_g0": 0.0,
      "loss_g1": 2.744856357574463,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.2178361415863037,
      "loss_g0": 5.491615295410156,
      "loss_g1": 2.162836790084839,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.3883273601531982,
      "loss_g0": 4.541029453277588,
      "loss_g1": 2.352161407470703,
      "q_g0": 0.01652265153825283,
      "q_g1": 0.9834772944450378,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.9626739025115967,
      "loss_g0": 1.0821915864944458,
      "loss_g1": 2.9942660331726074,
      "q_g0": 0.01652245782315731,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.861861228942871,
      "loss_g0": 8.728118896484375,
      "loss_g1": 2.7633042335510254,
      "q_g0": 0.01652306318283081,
      "q_g1": 0.9834769368171692,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.109490394592285,
      "loss_g0": 5.492877006530762,
      "loss_g1": 2.0526459217071533,
      "q_g0": 0.016523411497473717,
      "q_g1": 0.9834766387939453,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 0.16698162257671356,
      "loss_g0": 3.239985466003418,
      "loss_g1": 0.11535099148750305,
      "q_g0": 0.016523735597729683,
      "q_g1": 0.9834762811660767,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.071081161499023,
      "loss_g0": 34.54935073852539,
      "loss_g1": 8.64294147491455,
      "q_g0": 0.016526388004422188,
      "q_g1": 0.9834735989570618,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 22.59947967529297,
      "loss_g0": 22.14541244506836,
      "loss_g1": 22.60711097717285,
      "q_g0": 0.01652633771300316,
      "q_g1": 0.9834735989570618,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 58.00615692138672,
      "loss_g0": 10.298348426818848,
      "loss_g1": 58.807647705078125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.123361587524414,
      "loss_g0": 21.354698181152344,
      "loss_g1": 16.035472869873047,
      "q_g0": 0.016522971913218498,
      "q_g1": 0.9834770560264587,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 131.77239990234375,
      "loss_g0": 32.31682586669922,
      "loss_g1": 133.44325256347656,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 27.916746139526367,
      "loss_g0": 27.916746139526367,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.032024383544922,
      "loss_g0": 12.793867111206055,
      "loss_g1": 27.208036422729492,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.393738746643066,
      "loss_g0": 15.393738746643066,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.030208587646484,
      "loss_g0": 23.99106216430664,
      "loss_g1": 26.360347747802734,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 22.435169219970703,
      "loss_g0": 22.265438079833984,
      "loss_g1": 32.53759765625,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652347296476364,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.337823867797852,
      "loss_g0": 0.0,
      "loss_g1": 14.337823867797852,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.054606437683105,
      "loss_g0": 2.1104795932769775,
      "loss_g1": 12.221667289733887,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.631752014160156,
      "loss_g0": 25.031295776367188,
      "loss_g1": 10.389817237854004,
      "q_g0": 0.016523919999599457,
      "q_g1": 0.9834760427474976,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.622344017028809,
      "loss_g0": 18.683656692504883,
      "loss_g1": 11.503698348999023,
      "q_g0": 0.016524652019143105,
      "q_g1": 0.9834753274917603,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 23.108564376831055,
      "loss_g0": 27.88753318786621,
      "loss_g1": 23.028263092041016,
      "q_g0": 0.016525154933333397,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 23.77627182006836,
      "loss_g0": 29.32659149169922,
      "loss_g1": 23.683006286621094,
      "q_g0": 0.016525736078619957,
      "q_g1": 0.9834743142127991,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.828914642333984,
      "loss_g0": 6.260043621063232,
      "loss_g1": 17.00649642944336,
      "q_g0": 0.01652464270591736,
      "q_g1": 0.9834753274917603,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.114230632781982,
      "loss_g0": 10.762629508972168,
      "loss_g1": 4.0025177001953125,
      "q_g0": 0.016525331884622574,
      "q_g1": 0.9834746718406677,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.62639331817627,
      "loss_g0": 10.697025299072266,
      "loss_g1": 12.658811569213867,
      "q_g0": 0.016525132581591606,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.885000705718994,
      "loss_g0": 8.163252830505371,
      "loss_g1": 7.8803253173828125,
      "q_g0": 0.016525164246559143,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.964059829711914,
      "loss_g0": 8.808012008666992,
      "loss_g1": 25.23550033569336,
      "q_g0": 0.01652348041534424,
      "q_g1": 0.9834765195846558,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.997275352478027,
      "loss_g0": 8.971537590026855,
      "loss_g1": 8.99770736694336,
      "q_g0": 0.01652347296476364,
      "q_g1": 0.9834765195846558,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.639060974121094,
      "loss_g0": 21.202816009521484,
      "loss_g1": 17.579185485839844,
      "q_g0": 0.01652384363114834,
      "q_g1": 0.9834761619567871,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.219499588012695,
      "loss_g0": 5.890852928161621,
      "loss_g1": 10.292224884033203,
      "q_g0": 0.016523394733667374,
      "q_g1": 0.9834766387939453,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.337433099746704,
      "loss_g0": 14.32429027557373,
      "loss_g1": 1.1192227602005005,
      "q_g0": 0.016524743288755417,
      "q_g1": 0.9834752082824707,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.41769790649414,
      "loss_g0": 5.320101261138916,
      "loss_g1": 11.520148277282715,
      "q_g0": 0.016524117439985275,
      "q_g1": 0.983475923538208,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.31804084777832,
      "loss_g0": 16.540176391601562,
      "loss_g1": 18.34791374206543,
      "q_g0": 0.016523927450180054,
      "q_g1": 0.9834760427474976,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.961733818054199,
      "loss_g0": 20.171459197998047,
      "loss_g1": 7.756575584411621,
      "q_g0": 0.01652519218623638,
      "q_g1": 0.9834747910499573,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.155845642089844,
      "loss_g0": 0.0,
      "loss_g1": 5.155845642089844,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.8480260372161865,
      "loss_g0": 3.44038987159729,
      "loss_g1": 3.8548743724823,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.382669448852539,
      "loss_g0": 20.286264419555664,
      "loss_g1": 9.19947624206543,
      "q_g0": 0.016523558646440506,
      "q_g1": 0.9834765195846558,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.302258491516113,
      "loss_g0": 3.1165378093719482,
      "loss_g1": 7.372581481933594,
      "q_g0": 0.01652311533689499,
      "q_g1": 0.9834768176078796,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.612990140914917,
      "loss_g0": 4.586841583251953,
      "loss_g1": 3.5966286659240723,
      "q_g0": 0.016523223370313644,
      "q_g1": 0.9834766983985901,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.547268390655518,
      "loss_g0": 2.9207491874694824,
      "loss_g1": 6.6081953048706055,
      "q_g0": 0.016522854566574097,
      "q_g1": 0.9834771752357483,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.004624366760254,
      "loss_g0": 6.828030586242676,
      "loss_g1": 9.041191101074219,
      "q_g0": 0.01652262918651104,
      "q_g1": 0.9834774136543274,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.560901641845703,
      "loss_g0": 0.0,
      "loss_g1": 7.560901641845703,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.104513168334961,
      "loss_g0": 0.0,
      "loss_g1": 12.104513168334961,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.327890396118164,
      "loss_g0": 7.915072441101074,
      "loss_g1": 11.385225296020508,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.366250038146973,
      "loss_g0": 15.94059944152832,
      "loss_g1": 10.272597312927246,
      "q_g0": 0.016522999852895737,
      "q_g1": 0.9834769368171692,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 0.8165817260742188,
      "loss_g0": 9.16968059539795,
      "loss_g1": 0.676237165927887,
      "q_g0": 0.01652386784553528,
      "q_g1": 0.9834761619567871,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.204076290130615,
      "loss_g0": 5.5523223876953125,
      "loss_g1": 6.21502685546875,
      "q_g0": 0.016523797065019608,
      "q_g1": 0.9834761619567871,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.4189653396606445,
      "loss_g0": 1.298954963684082,
      "loss_g1": 6.504985809326172,
      "q_g0": 0.016523264348506927,
      "q_g1": 0.9834766983985901,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 0.3141711354255676,
      "loss_g0": 5.105068683624268,
      "loss_g1": 0.23367741703987122,
      "q_g0": 0.016523759812116623,
      "q_g1": 0.9834762811660767,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.62447452545166,
      "loss_g0": 6.62447452545166,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.293891429901123,
      "loss_g0": 7.3982439041137695,
      "loss_g1": 1.0824366807937622,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.826844215393066,
      "loss_g0": 4.871233940124512,
      "loss_g1": 2.1845879554748535,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.4608042240142822,
      "loss_g0": 3.4141671657562256,
      "loss_g1": 6.236761093139648,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522713005542755,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 28.851917266845703,
      "loss_g0": 29.284854888916016,
      "loss_g1": 3.0818066596984863,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.140436053276062,
      "loss_g0": 1.0784670114517212,
      "loss_g1": 4.828981876373291,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522806137800217,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.646862268447876,
      "loss_g0": 0.0,
      "loss_g1": 1.646862268447876,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.78542709350586,
      "loss_g0": 0.0,
      "loss_g1": 24.78542709350586,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 26.75121307373047,
      "loss_g0": 28.329509735107422,
      "loss_g1": 26.72469711303711,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 28.554973602294922,
      "loss_g0": 0.0,
      "loss_g1": 28.554973602294922,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 26.364334106445312,
      "loss_g0": 12.101099967956543,
      "loss_g1": 26.60395622253418,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 31.311786651611328,
      "loss_g0": 31.311786651611328,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 30.919124603271484,
      "loss_g0": 0.0,
      "loss_g1": 30.919124603271484,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.895816802978516,
      "loss_g0": 13.065292358398438,
      "loss_g1": 17.97696876525879,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.495288848876953,
      "loss_g0": 4.052539825439453,
      "loss_g1": 19.75472640991211,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.44927215576172,
      "loss_g0": 33.149295806884766,
      "loss_g1": 19.219091415405273,
      "q_g0": 0.01652384363114834,
      "q_g1": 0.9834761619567871,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 32.379703521728516,
      "loss_g0": 22.166196823120117,
      "loss_g1": 32.55129623413086,
      "q_g0": 0.01652277447283268,
      "q_g1": 0.9834771752357483,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.080163955688477,
      "loss_g0": 13.112822532653809,
      "loss_g1": 18.16361427307129,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 26.75749397277832,
      "loss_g0": 20.642621994018555,
      "loss_g1": 26.8602237701416,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.844380378723145,
      "loss_g0": 9.4801025390625,
      "loss_g1": 9.850500106811523,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.273232460021973,
      "loss_g0": 19.779706954956055,
      "loss_g1": 12.147117614746094,
      "q_g0": 0.016523201018571854,
      "q_g1": 0.9834768176078796,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.024465560913086,
      "loss_g0": 25.691368103027344,
      "loss_g1": 6.710808753967285,
      "q_g0": 0.0165251474827528,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.542239665985107,
      "loss_g0": 13.09574031829834,
      "loss_g1": 7.448921203613281,
      "q_g0": 0.01652572862803936,
      "q_g1": 0.9834743142127991,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.043731689453125,
      "loss_g0": 18.034543991088867,
      "loss_g1": 17.027080535888672,
      "q_g0": 0.016525836661458015,
      "q_g1": 0.9834741950035095,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.445470809936523,
      "loss_g0": 18.481597900390625,
      "loss_g1": 18.444862365722656,
      "q_g0": 0.016525836661458015,
      "q_g1": 0.9834741950035095,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 21.842262268066406,
      "loss_g0": 21.842262268066406,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.264399528503418,
      "loss_g0": 12.32480239868164,
      "loss_g1": 8.668984413146973,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.600274085998535,
      "loss_g0": 7.620748996734619,
      "loss_g1": 6.3815131187438965,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.951761245727539,
      "loss_g0": 15.143356323242188,
      "loss_g1": 3.5472912788391113,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.970892906188965,
      "loss_g0": 0.0,
      "loss_g1": 10.970892906188965,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.457283020019531,
      "loss_g0": 7.49135160446167,
      "loss_g1": 5.423110485076904,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.582380294799805,
      "loss_g0": 7.75996732711792,
      "loss_g1": 5.5457963943481445,
      "q_g0": 0.016522642225027084,
      "q_g1": 0.9834772944450378,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.764135837554932,
      "loss_g0": 18.290529251098633,
      "loss_g1": 7.587277889251709,
      "q_g0": 0.016523735597729683,
      "q_g1": 0.9834762811660767,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.9139404296875,
      "loss_g0": 0.0,
      "loss_g1": 13.9139404296875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.846095323562622,
      "loss_g0": 21.370803833007812,
      "loss_g1": 1.5180801153182983,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.831781387329102,
      "loss_g0": 15.795499801635742,
      "loss_g1": 14.815589904785156,
      "q_g0": 0.016522521153092384,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.247414588928223,
      "loss_g0": 7.206166744232178,
      "loss_g1": 9.281707763671875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.34157943725586,
      "loss_g0": 3.6242361068725586,
      "loss_g1": 9.437630653381348,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.856752395629883,
      "loss_g0": 2.963604211807251,
      "loss_g1": 6.922157287597656,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.388795852661133,
      "loss_g0": 20.96566390991211,
      "loss_g1": 11.227893829345703,
      "q_g0": 0.016523417085409164,
      "q_g1": 0.9834766387939453,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.73534107208252,
      "loss_g0": 28.71156120300293,
      "loss_g1": 9.416481018066406,
      "q_g0": 0.01652538776397705,
      "q_g1": 0.9834746718406677,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.66839599609375,
      "loss_g0": 8.294561386108398,
      "loss_g1": 6.641071319580078,
      "q_g0": 0.016525547951459885,
      "q_g1": 0.9834744334220886,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.893566131591797,
      "loss_g0": 4.707669734954834,
      "loss_g1": 9.980703353881836,
      "q_g0": 0.016525015234947205,
      "q_g1": 0.9834749698638916,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.57794189453125,
      "loss_g0": 15.014276504516602,
      "loss_g1": 13.553807258605957,
      "q_g0": 0.016525164246559143,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.7451460361480713,
      "loss_g0": 11.250762939453125,
      "loss_g1": 3.6190242767333984,
      "q_g0": 0.01652594469487667,
      "q_g1": 0.98347407579422,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.1678885221481323,
      "loss_g0": 9.705107688903809,
      "loss_g1": 1.0244243144989014,
      "q_g0": 0.016526827588677406,
      "q_g1": 0.9834731221199036,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.423935890197754,
      "loss_g0": 27.140600204467773,
      "loss_g1": 4.0421366691589355,
      "q_g0": 0.01652919314801693,
      "q_g1": 0.9834708571434021,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.664844036102295,
      "loss_g0": 7.7502241134643555,
      "loss_g1": 2.579371452331543,
      "q_g0": 0.016529718413949013,
      "q_g1": 0.9834702610969543,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.2261269092559814,
      "loss_g0": 11.929759979248047,
      "loss_g1": 1.046212911605835,
      "q_g0": 0.016530826687812805,
      "q_g1": 0.9834691286087036,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.317760467529297,
      "loss_g0": 5.547366142272949,
      "loss_g1": 17.51559066772461,
      "q_g0": 0.016529595479369164,
      "q_g1": 0.9834703803062439,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.728144645690918,
      "loss_g0": 6.86073112487793,
      "loss_g1": 5.709108352661133,
      "q_g0": 0.016529718413949013,
      "q_g1": 0.9834702610969543,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.8767595291137695,
      "loss_g0": 19.398550033569336,
      "loss_g1": 2.5990395545959473,
      "q_g0": 0.016531439498066902,
      "q_g1": 0.9834685325622559,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.742752552032471,
      "loss_g0": 4.742752552032471,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.577078819274902,
      "loss_g0": 9.68002986907959,
      "loss_g1": 3.449061870574951,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 22.36368751525879,
      "loss_g0": 22.693281173706055,
      "loss_g1": 2.7450292110443115,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.0459365844726562,
      "loss_g0": 1.9106502532958984,
      "loss_g1": 10.098276138305664,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652325503528118,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.641989231109619,
      "loss_g0": 2.6401748657226562,
      "loss_g1": 2.7499797344207764,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523271799087524,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.5695219039917,
      "loss_g0": 13.037057876586914,
      "loss_g1": 45.255767822265625,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526563093066216,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.308448791503906,
      "loss_g0": 17.019933700561523,
      "loss_g1": 34.47572708129883,
      "q_g0": 0.9834716320037842,
      "q_g1": 0.01652834750711918,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 37.820491790771484,
      "loss_g0": 38.05965042114258,
      "loss_g1": 23.588733673095703,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.01652686670422554,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 25.144760131835938,
      "loss_g0": 0.0,
      "loss_g1": 25.144760131835938,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.582347869873047,
      "loss_g0": 22.565841674804688,
      "loss_g1": 11.39782428741455,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.44118309020996,
      "loss_g0": 11.401792526245117,
      "loss_g1": 18.559444427490234,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.410091400146484,
      "loss_g0": 21.0767822265625,
      "loss_g1": 18.36528968811035,
      "q_g0": 0.01652269810438156,
      "q_g1": 0.9834772944450378,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.455310821533203,
      "loss_g0": 23.131406784057617,
      "loss_g1": 16.3431453704834,
      "q_g0": 0.016523387283086777,
      "q_g1": 0.9834766387939453,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.146512985229492,
      "loss_g0": 0.0,
      "loss_g1": 17.146512985229492,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 18.073991775512695,
      "loss_g0": 27.045860290527344,
      "loss_g1": 17.923263549804688,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 31.83961296081543,
      "loss_g0": 17.616737365722656,
      "loss_g1": 32.078556060791016,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.36655044555664,
      "loss_g0": 17.36655044555664,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.429831504821777,
      "loss_g0": 8.091450691223145,
      "loss_g1": 28.57152557373047,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 36.88855743408203,
      "loss_g0": 36.88855743408203,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.817028045654297,
      "loss_g0": 16.834712982177734,
      "loss_g1": 15.764341354370117,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.034421920776367,
      "loss_g0": 6.949136734008789,
      "loss_g1": 12.110742568969727,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522955149412155,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 47.118473052978516,
      "loss_g0": 47.676292419433594,
      "loss_g1": 13.91499137878418,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 49.1507453918457,
      "loss_g0": 49.73714828491211,
      "loss_g1": 14.245807647705078,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.650999069213867,
      "loss_g0": 0.0,
      "loss_g1": 24.650999069213867,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 15.540766716003418,
      "loss_g0": 0.0,
      "loss_g1": 15.540766716003418,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.81630802154541,
      "loss_g0": 61.58892822265625,
      "loss_g1": 10.980128288269043,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.304438591003418,
      "loss_g0": 19.165611267089844,
      "loss_g1": 8.12195873260498,
      "q_g0": 0.01652355119585991,
      "q_g1": 0.9834765195846558,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.67581844329834,
      "loss_g0": 1.187017798423767,
      "loss_g1": 5.751233100891113,
      "q_g0": 0.0165230855345726,
      "q_g1": 0.9834769368171692,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.112247467041016,
      "loss_g0": 6.829036712646484,
      "loss_g1": 10.16740608215332,
      "q_g0": 0.016522744670510292,
      "q_g1": 0.9834772944450378,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.5762925148010254,
      "loss_g0": 18.95151138305664,
      "loss_g1": 3.317958116531372,
      "q_g0": 0.016524342820048332,
      "q_g1": 0.9834756851196289,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.093887329101562,
      "loss_g0": 15.575498580932617,
      "loss_g1": 7.968174934387207,
      "q_g0": 0.016525115817785263,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 22.166685104370117,
      "loss_g0": 6.975938320159912,
      "loss_g1": 22.42190933227539,
      "q_g0": 0.016523543745279312,
      "q_g1": 0.9834765195846558,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.5620477199554443,
      "loss_g0": 3.5620477199554443,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.59514045715332,
      "loss_g0": 12.724101066589355,
      "loss_g1": 4.918921947479248,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.26686954498291,
      "loss_g0": 8.236753463745117,
      "loss_g1": 10.059466361999512,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522612422704697,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.308159828186035,
      "loss_g0": 2.2021408081054688,
      "loss_g1": 8.618492126464844,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523271799087524,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.32705020904541,
      "loss_g0": 8.30117416381836,
      "loss_g1": 9.867208480834961,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523433849215508,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.03583335876465,
      "loss_g0": 19.313356399536133,
      "loss_g1": 2.5166430473327637,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.457257270812988,
      "loss_g0": 6.493377685546875,
      "loss_g1": 4.307248592376709,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.757823944091797,
      "loss_g0": 0.0,
      "loss_g1": 4.757823944091797,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.555915832519531,
      "loss_g0": 5.691295146942139,
      "loss_g1": 8.60404109954834,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.258909225463867,
      "loss_g0": 14.544271469116211,
      "loss_g1": 2.0524988174438477,
      "q_g0": 0.0165236983448267,
      "q_g1": 0.9834762811660767,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.09185791015625,
      "loss_g0": 5.020512580871582,
      "loss_g1": 3.0594537258148193,
      "q_g0": 0.016523899510502815,
      "q_g1": 0.9834761619567871,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.858804941177368,
      "loss_g0": 5.339912414550781,
      "loss_g1": 3.8339197635650635,
      "q_g0": 0.016524044796824455,
      "q_g1": 0.983475923538208,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.675368309020996,
      "loss_g0": 3.966794967651367,
      "loss_g1": 6.7208757400512695,
      "q_g0": 0.016523759812116623,
      "q_g1": 0.9834762811660767,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.415975570678711,
      "loss_g0": 11.056598663330078,
      "loss_g1": 6.338004112243652,
      "q_g0": 0.01652424782514572,
      "q_g1": 0.9834758043289185,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.395218372344971,
      "loss_g0": 9.804134368896484,
      "loss_g1": 6.3379411697387695,
      "q_g0": 0.016524603590369225,
      "q_g1": 0.9834753274917603,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.070154190063477,
      "loss_g0": 5.180136680603027,
      "loss_g1": 7.101910591125488,
      "q_g0": 0.016524409875273705,
      "q_g1": 0.9834755659103394,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.026573181152344,
      "loss_g0": 7.026573181152344,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.75113868713379,
      "loss_g0": 16.99824333190918,
      "loss_g1": 2.0424387454986572,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 1.7750427722930908,
      "loss_g0": 1.7486026287078857,
      "loss_g1": 3.3488476276397705,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522590070962906,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 31.34975242614746,
      "loss_g0": 31.792919158935547,
      "loss_g1": 4.970728397369385,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 2.4987375736236572,
      "loss_g0": 2.5217738151550293,
      "loss_g1": 1.1275371313095093,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.6525168418884277,
      "loss_g0": 3.6525168418884277,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.58965539932251,
      "loss_g0": 5.645731449127197,
      "loss_g1": 2.2517948150634766,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 32.9202995300293,
      "loss_g0": 32.96186065673828,
      "loss_g1": 30.446407318115234,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 57.954994201660156,
      "loss_g0": 58.74290466308594,
      "loss_g1": 11.055484771728516,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 40.89112854003906,
      "loss_g0": 40.995887756347656,
      "loss_g1": 34.655216217041016,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 26.77793312072754,
      "loss_g0": 27.083969116210938,
      "loss_g1": 8.561456680297852,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.436514854431152,
      "loss_g0": 7.213172435760498,
      "loss_g1": 20.72957992553711,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523797065019608,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 20.789915084838867,
      "loss_g0": 20.879976272583008,
      "loss_g1": 15.429388046264648,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523240134119987,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 50.26878356933594,
      "loss_g0": 50.63936996459961,
      "loss_g1": 28.21010971069336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.298011779785156,
      "loss_g0": 19.387466430664062,
      "loss_g1": 13.973329544067383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 30.492403030395508,
      "loss_g0": 30.573104858398438,
      "loss_g1": 25.6887149810791,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.043495178222656,
      "loss_g0": 8.920734405517578,
      "loss_g1": 16.350318908691406,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652318611741066,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 20.777694702148438,
      "loss_g0": 20.777694702148438,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 36.212684631347656,
      "loss_g0": 36.50904846191406,
      "loss_g1": 18.571914672851562,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.785429000854492,
      "loss_g0": 17.932228088378906,
      "loss_g1": 9.047372817993164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.200730323791504,
      "loss_g0": 4.870738506317139,
      "loss_g1": 24.84063720703125,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524463891983032,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 22.673948287963867,
      "loss_g0": 0.0,
      "loss_g1": 22.673948287963867,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 27.64824676513672,
      "loss_g0": 22.296995162963867,
      "loss_g1": 27.738147735595703,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.783979415893555,
      "loss_g0": 22.637006759643555,
      "loss_g1": 12.618437767028809,
      "q_g0": 0.0165234487503767,
      "q_g1": 0.9834765195846558,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.257284164428711,
      "loss_g0": 12.897945404052734,
      "loss_g1": 14.280122756958008,
      "q_g0": 0.016523301601409912,
      "q_g1": 0.9834766983985901,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.054990291595459,
      "loss_g0": 22.979969024658203,
      "loss_g1": 4.753799915313721,
      "q_g0": 0.01652517169713974,
      "q_g1": 0.983474850654602,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 42.71559524536133,
      "loss_g0": 14.86945629119873,
      "loss_g1": 43.18341064453125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.94614601135254,
      "loss_g0": 2.940568685531616,
      "loss_g1": 18.198238372802734,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 27.34567642211914,
      "loss_g0": 17.90806007385254,
      "loss_g1": 27.504226684570312,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.453731060028076,
      "loss_g0": 7.874496936798096,
      "loss_g1": 7.446662425994873,
      "q_g0": 0.016522465273737907,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.730896234512329,
      "loss_g0": 20.662181854248047,
      "loss_g1": 3.4464190006256104,
      "q_g0": 0.01652422547340393,
      "q_g1": 0.9834758043289185,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.447449684143066,
      "loss_g0": 12.015878677368164,
      "loss_g1": 10.421096801757812,
      "q_g0": 0.01652439311146736,
      "q_g1": 0.9834755659103394,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 25.018335342407227,
      "loss_g0": 10.7515287399292,
      "loss_g1": 25.258026123046875,
      "q_g0": 0.016522914171218872,
      "q_g1": 0.9834770560264587,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.508193016052246,
      "loss_g0": 23.695602416992188,
      "loss_g1": 6.2194037437438965,
      "q_g0": 0.016524706035852432,
      "q_g1": 0.9834753274917603,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.899425506591797,
      "loss_g0": 8.854060173034668,
      "loss_g1": 25.16899871826172,
      "q_g0": 0.01652303896844387,
      "q_g1": 0.9834769368171692,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 14.785712242126465,
      "loss_g0": 58.65407180786133,
      "loss_g1": 14.04848861694336,
      "q_g0": 0.016527602449059486,
      "q_g1": 0.9834723472595215,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.225992202758789,
      "loss_g0": 0.0,
      "loss_g1": 8.225992202758789,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 11.167790412902832,
      "loss_g0": 19.752498626708984,
      "loss_g1": 11.023567199707031,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.557868957519531,
      "loss_g0": 24.30667495727539,
      "loss_g1": 6.259654998779297,
      "q_g0": 0.016524270176887512,
      "q_g1": 0.9834756851196289,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.649613380432129,
      "loss_g0": 14.791221618652344,
      "loss_g1": 4.479204177856445,
      "q_g0": 0.01652531512081623,
      "q_g1": 0.9834746718406677,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.7718772888183594,
      "loss_g0": 19.648221969604492,
      "loss_g1": 3.505080223083496,
      "q_g0": 0.01652696169912815,
      "q_g1": 0.9834730625152588,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.91652774810791,
      "loss_g0": 22.22470474243164,
      "loss_g1": 13.77690315246582,
      "q_g0": 0.016527820378541946,
      "q_g1": 0.9834722280502319,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.302473068237305,
      "loss_g0": 5.728785991668701,
      "loss_g1": 10.37933349609375,
      "q_g0": 0.01652734912931919,
      "q_g1": 0.9834727048873901,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.4681050777435303,
      "loss_g0": 0.0,
      "loss_g1": 3.4681050777435303,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.230071067810059,
      "loss_g0": 1.528329849243164,
      "loss_g1": 5.29226016998291,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.868119716644287,
      "loss_g0": 24.38698959350586,
      "loss_g1": 5.5569658279418945,
      "q_g0": 0.01652434840798378,
      "q_g1": 0.9834756851196289,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.386041641235352,
      "loss_g0": 13.81032943725586,
      "loss_g1": 8.294899940490723,
      "q_g0": 0.0165249053388834,
      "q_g1": 0.9834750890731812,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.2375833988189697,
      "loss_g0": 12.931901931762695,
      "loss_g1": 3.074683904647827,
      "q_g0": 0.016525914892554283,
      "q_g1": 0.98347407579422,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.154583930969238,
      "loss_g0": 37.50386047363281,
      "loss_g1": 3.594076156616211,
      "q_g0": 0.0165293850004673,
      "q_g1": 0.983470618724823,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.527387619018555,
      "loss_g0": 18.700347900390625,
      "loss_g1": 9.373207092285156,
      "q_g0": 0.016530346125364304,
      "q_g1": 0.9834696650505066,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.505050659179688,
      "loss_g0": 2.089660406112671,
      "loss_g1": 13.696908950805664,
      "q_g0": 0.016529161483049393,
      "q_g1": 0.9834708571434021,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.0877366065979,
      "loss_g0": 3.730095148086548,
      "loss_g1": 4.093747615814209,
      "q_g0": 0.01652912236750126,
      "q_g1": 0.9834708571434021,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.020190238952637,
      "loss_g0": 10.020190238952637,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.211832046508789,
      "loss_g0": 9.267345428466797,
      "loss_g1": 5.907468795776367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 6.62862491607666,
      "loss_g0": 6.62862491607666,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.082590103149414,
      "loss_g0": 4.138575553894043,
      "loss_g1": 0.7501150369644165,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.0750837326049805,
      "loss_g0": 3.0750837326049805,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 32.22376251220703,
      "loss_g0": 32.59973907470703,
      "loss_g1": 9.844165802001953,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.863784313201904,
      "loss_g0": 7.5794477462768555,
      "loss_g1": 24.7867431640625,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524186357855797,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.922290802001953,
      "loss_g0": 8.804920196533203,
      "loss_g1": 15.90752124786377,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524914652109146,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 13.985563278198242,
      "loss_g0": 13.985563278198242,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.692329406738281,
      "loss_g0": 10.663538932800293,
      "loss_g1": 12.406050682067871,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 10.056482315063477,
      "loss_g0": 9.955564498901367,
      "loss_g1": 16.063257217407227,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523046419024467,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 8.800541877746582,
      "loss_g0": 8.418972969055176,
      "loss_g1": 31.508790969848633,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525408253073692,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.458812713623047,
      "loss_g0": 7.22785758972168,
      "loss_g1": 21.202396392822266,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.01652684435248375,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 21.432464599609375,
      "loss_g0": 21.72489356994629,
      "loss_g1": 4.028787612915039,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016525031998753548,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 17.31183624267578,
      "loss_g0": 17.485136032104492,
      "loss_g1": 6.997348785400391,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523951664566994,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.09943962097168,
      "loss_g0": 24.33498764038086,
      "loss_g1": 10.078817367553711,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 7.49144983291626,
      "loss_g0": 7.49144983291626,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.279902458190918,
      "loss_g0": 9.338751792907715,
      "loss_g1": 5.776943206787109,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 16.053619384765625,
      "loss_g0": 16.107383728027344,
      "loss_g1": 12.85327434539795,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 12.108170509338379,
      "loss_g0": 12.2366361618042,
      "loss_g1": 4.461399555206299,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 4.583703994750977,
      "loss_g0": 4.429573059082031,
      "loss_g1": 13.75763988494873,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523372381925583,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 9.909554481506348,
      "loss_g0": 9.95986270904541,
      "loss_g1": 6.915102005004883,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652306318283081,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 3.0361413955688477,
      "loss_g0": 2.95904803276062,
      "loss_g1": 7.624736309051514,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523543745279312,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 5.3121771812438965,
      "loss_g0": 5.3121771812438965,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "eval_cer": 0.06218076188248138,
      "eval_loss": 14.027531623840332,
      "eval_runtime": 80.785,
      "eval_samples_per_second": 20.746,
      "eval_steps_per_second": 5.187,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 19.866619110107422,
      "loss_g0": 19.373504638671875,
      "loss_g1": 49.2186279296875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1000
    },
    {
      "epoch": 5.0761421319796955,
      "gdro_loss": 24.90645408630371,
      "loss_g0": 24.537837982177734,
      "loss_g1": 46.8447265625,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524706035852432,
      "step": 1000
    },
    {
      "epoch": 5.583756345177665,
      "grad_norm": 119.45448303222656,
      "learning_rate": 0.0002169625668449198,
      "loss": 20.9698,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.56870460510254,
      "loss_g0": 28.099767684936523,
      "loss_g1": 18.408571243286133,
      "q_g0": 0.016523417085409164,
      "q_g1": 0.9834766387939453,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 41.700687408447266,
      "loss_g0": 41.700687408447266,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 39.097381591796875,
      "loss_g0": 39.31739807128906,
      "loss_g1": 26.000896453857422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.001304626464844,
      "loss_g0": 4.771565914154053,
      "loss_g1": 18.675016403198242,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652384363114834,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.203679084777832,
      "loss_g0": 11.999225616455078,
      "loss_g1": 24.371490478515625,
      "q_g0": 0.983474850654602,
      "q_g1": 0.01652509905397892,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.76123046875,
      "loss_g0": 17.52089500427246,
      "loss_g1": 32.063255310058594,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526587307453156,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.112545013427734,
      "loss_g0": 23.817012786865234,
      "loss_g1": 41.69723129272461,
      "q_g0": 0.9834715127944946,
      "q_g1": 0.0165284164249897,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.6785888671875,
      "loss_g0": 10.73497486114502,
      "loss_g1": 7.323409080505371,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.01652807556092739,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.9837646484375,
      "loss_g0": 10.871482849121094,
      "loss_g1": 17.66460418701172,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.016528764739632607,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.323883056640625,
      "loss_g0": 18.445159912109375,
      "loss_g1": 11.107501029968262,
      "q_g0": 0.9834719896316528,
      "q_g1": 0.016528021544218063,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 31.594919204711914,
      "loss_g0": 31.878137588500977,
      "loss_g1": 14.740659713745117,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526270657777786,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.76536750793457,
      "loss_g0": 11.548933982849121,
      "loss_g1": 24.644201278686523,
      "q_g0": 0.9834723472595215,
      "q_g1": 0.016527602449059486,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.837151527404785,
      "loss_g0": 12.017477035522461,
      "loss_g1": 1.1061640977859497,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.016526486724615097,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 31.246721267700195,
      "loss_g0": 31.218807220458984,
      "loss_g1": 32.90780258178711,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526658087968826,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.841826438903809,
      "loss_g0": 13.902033805847168,
      "loss_g1": 10.258885383605957,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.01652628742158413,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.38900375366211,
      "loss_g0": 10.438520431518555,
      "loss_g1": 7.442205905914307,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525976359844208,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.24489974975586,
      "loss_g0": 0.0,
      "loss_g1": 17.24489974975586,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.993412017822266,
      "loss_g0": 12.871553421020508,
      "loss_g1": 25.197059631347656,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.25261116027832,
      "loss_g0": 9.16418170928955,
      "loss_g1": 26.539695739746094,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.406817436218262,
      "loss_g0": 8.000524520874023,
      "loss_g1": 5.3632426261901855,
      "q_g0": 0.01652269810438156,
      "q_g1": 0.9834772944450378,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.098767280578613,
      "loss_g0": 7.0280609130859375,
      "loss_g1": 9.133554458618164,
      "q_g0": 0.016522489488124847,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.510163307189941,
      "loss_g0": 11.05801773071289,
      "loss_g1": 8.467358589172363,
      "q_g0": 0.016522753983736038,
      "q_g1": 0.9834772944450378,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.812506675720215,
      "loss_g0": 8.511396408081055,
      "loss_g1": 5.767163276672363,
      "q_g0": 0.016523031517863274,
      "q_g1": 0.9834769368171692,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.063225746154785,
      "loss_g0": 18.181285858154297,
      "loss_g1": 7.893225193023682,
      "q_g0": 0.01652408577501774,
      "q_g1": 0.983475923538208,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.273611068725586,
      "loss_g0": 8.653016090393066,
      "loss_g1": 11.317641258239746,
      "q_g0": 0.01652381382882595,
      "q_g1": 0.9834761619567871,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.1525936126709,
      "loss_g0": 7.643560409545898,
      "loss_g1": 17.312349319458008,
      "q_g0": 0.016522830352187157,
      "q_g1": 0.9834771752357483,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.703078269958496,
      "loss_g0": 8.786392211914062,
      "loss_g1": 12.768878936767578,
      "q_g0": 0.016522426158189774,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.419735431671143,
      "loss_g0": 14.527054786682129,
      "loss_g1": 5.2667236328125,
      "q_g0": 0.016523372381925583,
      "q_g1": 0.9834766387939453,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.673048496246338,
      "loss_g0": 5.261613368988037,
      "loss_g1": 7.713562488555908,
      "q_g0": 0.016523122787475586,
      "q_g1": 0.9834768176078796,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.87498664855957,
      "loss_g0": 26.41184425354004,
      "loss_g1": 8.580322265625,
      "q_g0": 0.016524946317076683,
      "q_g1": 0.9834750890731812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.5178804397583,
      "loss_g0": 11.557703018188477,
      "loss_g1": 8.466801643371582,
      "q_g0": 0.016525255516171455,
      "q_g1": 0.9834747910499573,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.412158012390137,
      "loss_g0": 11.54956340789795,
      "loss_g1": 10.393045425415039,
      "q_g0": 0.016525372862815857,
      "q_g1": 0.9834746718406677,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.110283851623535,
      "loss_g0": 1.5282284021377563,
      "loss_g1": 5.170472145080566,
      "q_g0": 0.016524991020560265,
      "q_g1": 0.9834749698638916,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.431784629821777,
      "loss_g0": 7.846843719482422,
      "loss_g1": 12.508821487426758,
      "q_g0": 0.01652451977133751,
      "q_g1": 0.9834754467010498,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.686319351196289,
      "loss_g0": 3.78564715385437,
      "loss_g1": 11.819061279296875,
      "q_g0": 0.0165236983448267,
      "q_g1": 0.9834762811660767,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.097204208374023,
      "loss_g0": 23.792339324951172,
      "loss_g1": 14.951107025146484,
      "q_g0": 0.016524603590369225,
      "q_g1": 0.9834753274917603,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.9323313236236572,
      "loss_g0": 3.618914842605591,
      "loss_g1": 2.920795202255249,
      "q_g0": 0.016524681821465492,
      "q_g1": 0.9834753274917603,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.294901847839355,
      "loss_g0": 8.294901847839355,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.15865707397461,
      "loss_g0": 25.55767250061035,
      "loss_g1": 1.4076635837554932,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.262558937072754,
      "loss_g0": 8.301688194274902,
      "loss_g1": 5.9334564208984375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.067339897155762,
      "loss_g0": 6.149777412414551,
      "loss_g1": 1.1603350639343262,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.90692138671875,
      "loss_g0": 13.90692138671875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.240850925445557,
      "loss_g0": 5.965293884277344,
      "loss_g1": 22.643043518066406,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.322203636169434,
      "loss_g0": 10.481721878051758,
      "loss_g1": 0.8270457983016968,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.885241508483887,
      "loss_g0": 5.885241508483887,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.742098808288574,
      "loss_g0": 12.90184211730957,
      "loss_g1": 3.2335362434387207,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.833449363708496,
      "loss_g0": 11.973499298095703,
      "loss_g1": 3.497134208679199,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.8490090370178223,
      "loss_g0": 3.8114395141601562,
      "loss_g1": 6.085268974304199,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.418181300163269,
      "loss_g0": 1.4311463832855225,
      "loss_g1": 0.6464588642120361,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.1072874069213867,
      "loss_g0": 3.157444715499878,
      "loss_g1": 0.12172838300466537,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 78.92320251464844,
      "loss_g0": 79.82537078857422,
      "loss_g1": 25.22260093688965,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.234524726867676,
      "loss_g0": 7.028623580932617,
      "loss_g1": 19.489587783813477,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.0165236983448267,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 33.40413284301758,
      "loss_g0": 33.53702926635742,
      "loss_g1": 25.493696212768555,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652287319302559,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.106610298156738,
      "loss_g0": 14.048851013183594,
      "loss_g1": 17.544546127319336,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523223370313644,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 43.314300537109375,
      "loss_g0": 43.56767654418945,
      "loss_g1": 28.232168197631836,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 28.82413673400879,
      "loss_g0": 0.0,
      "loss_g1": 28.82413673400879,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.96260643005371,
      "loss_g0": 12.603423118591309,
      "loss_g1": 17.03584098815918,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.197958469390869,
      "loss_g0": 21.366004943847656,
      "loss_g1": 6.959914207458496,
      "q_g0": 0.016523899510502815,
      "q_g1": 0.9834761619567871,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 34.427459716796875,
      "loss_g0": 33.825469970703125,
      "loss_g1": 34.43757629394531,
      "q_g0": 0.016523828729987144,
      "q_g1": 0.9834761619567871,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.851600646972656,
      "loss_g0": 19.020458221435547,
      "loss_g1": 16.815160751342773,
      "q_g0": 0.0165240541100502,
      "q_g1": 0.983475923538208,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.460366249084473,
      "loss_g0": 26.17197036743164,
      "loss_g1": 6.129135608673096,
      "q_g0": 0.016526101157069206,
      "q_g1": 0.9834739565849304,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.165803909301758,
      "loss_g0": 30.446331024169922,
      "loss_g1": 26.0938720703125,
      "q_g0": 0.01652655005455017,
      "q_g1": 0.9834734797477722,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.06952667236328,
      "loss_g0": 36.08610916137695,
      "loss_g1": 20.81715965270996,
      "q_g0": 0.016528114676475525,
      "q_g1": 0.9834718704223633,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.321969985961914,
      "loss_g0": 24.059423446655273,
      "loss_g1": 16.19192886352539,
      "q_g0": 0.016528913751244545,
      "q_g1": 0.9834710955619812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 45.40242004394531,
      "loss_g0": 19.723495483398438,
      "loss_g1": 45.833927154541016,
      "q_g0": 0.016526246443390846,
      "q_g1": 0.9834737181663513,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.477296829223633,
      "loss_g0": 9.477296829223633,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.44676399230957,
      "loss_g0": 8.379080772399902,
      "loss_g1": 12.475543975830078,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.813664436340332,
      "loss_g0": 10.436494827270508,
      "loss_g1": 33.260986328125,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652476005256176,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 31.384838104248047,
      "loss_g0": 31.73343276977539,
      "loss_g1": 10.635401725769043,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.0165226049721241,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 28.67485237121582,
      "loss_g0": 28.978252410888672,
      "loss_g1": 10.61532974243164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.671177864074707,
      "loss_g0": 0.0,
      "loss_g1": 15.671177864074707,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.909828186035156,
      "loss_g0": 11.375151634216309,
      "loss_g1": 12.935609817504883,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.506369590759277,
      "loss_g0": 0.0,
      "loss_g1": 12.506369590759277,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.1891591548919678,
      "loss_g0": 17.645034790039062,
      "loss_g1": 1.9295003414154053,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.6559031009674072,
      "loss_g0": 14.462844848632812,
      "loss_g1": 3.474334239959717,
      "q_g0": 0.01652355119585991,
      "q_g1": 0.9834765195846558,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.123923301696777,
      "loss_g0": 23.449417114257812,
      "loss_g1": 9.900020599365234,
      "q_g0": 0.01652493141591549,
      "q_g1": 0.9834750890731812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.994918823242188,
      "loss_g0": 13.806639671325684,
      "loss_g1": 16.031686782836914,
      "q_g0": 0.016524696722626686,
      "q_g1": 0.9834753274917603,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.8321926593780518,
      "loss_g0": 11.461772918701172,
      "loss_g1": 3.703991413116455,
      "q_g0": 0.01652548834681511,
      "q_g1": 0.9834745526313782,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.453279495239258,
      "loss_g0": 11.99911117553711,
      "loss_g1": 12.46091079711914,
      "q_g0": 0.01652543991804123,
      "q_g1": 0.9834745526313782,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.456165313720703,
      "loss_g0": 24.981813430786133,
      "loss_g1": 5.128032207489014,
      "q_g0": 0.01652747206389904,
      "q_g1": 0.9834725856781006,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.864938735961914,
      "loss_g0": 4.833408355712891,
      "loss_g1": 8.932687759399414,
      "q_g0": 0.016527052968740463,
      "q_g1": 0.9834729433059692,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.8189271688461304,
      "loss_g0": 14.914645195007324,
      "loss_g1": 1.598837971687317,
      "q_g0": 0.0165284164249897,
      "q_g1": 0.9834715127944946,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.2141480445861816,
      "loss_g0": 22.371610641479492,
      "loss_g1": 2.8921444416046143,
      "q_g0": 0.016530409455299377,
      "q_g1": 0.9834696054458618,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.614948272705078,
      "loss_g0": 4.614948272705078,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.766131401062012,
      "loss_g0": 6.82334566116333,
      "loss_g1": 3.360524892807007,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.6342902183532715,
      "loss_g0": 5.599283695220947,
      "loss_g1": 7.717994213104248,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.78544807434082,
      "loss_g0": 15.82865047454834,
      "loss_g1": 13.213868141174316,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.007022857666016,
      "loss_g0": 11.007022857666016,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.50040340423584,
      "loss_g0": 14.624652862548828,
      "loss_g1": 7.104597568511963,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.716829776763916,
      "loss_g0": 7.79193639755249,
      "loss_g1": 3.246211528778076,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.284048080444336,
      "loss_g0": 16.406841278076172,
      "loss_g1": 8.974869728088379,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.39704418182373,
      "loss_g0": 8.522533416748047,
      "loss_g1": 0.9274564385414124,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.3916596174240112,
      "loss_g0": 1.1787312030792236,
      "loss_g1": 14.064935684204102,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652374491095543,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.1023857593536377,
      "loss_g0": 1.110967755317688,
      "loss_g1": 0.5915900468826294,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.0165236983448267,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.117863655090332,
      "loss_g0": 2.073514461517334,
      "loss_g1": 4.757433891296387,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523977741599083,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.791324615478516,
      "loss_g0": 4.7996625900268555,
      "loss_g1": 4.295053482055664,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523919999599457,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.878657341003418,
      "loss_g0": 13.086675643920898,
      "loss_g1": 0.4967549443244934,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.41659164428711,
      "loss_g0": 10.581596374511719,
      "loss_g1": 0.5948811173439026,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.1883841753005981,
      "loss_g0": 1.1383781433105469,
      "loss_g1": 4.164876937866211,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522737219929695,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.38517689704895,
      "loss_g0": 2.393714666366577,
      "loss_g1": 1.8769874572753906,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522690653800964,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.0609130859375,
      "loss_g0": 25.162275314331055,
      "loss_g1": 19.027433395385742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.539289474487305,
      "loss_g0": 20.63677978515625,
      "loss_g1": 14.736222267150879,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.864981174468994,
      "loss_g0": 5.050393581390381,
      "loss_g1": 54.3372688293457,
      "q_g0": 0.9834725856781006,
      "q_g1": 0.016527464613318443,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 30.336929321289062,
      "loss_g0": 30.478757858276367,
      "loss_g1": 21.89695167541504,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526587307453156,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 33.00391387939453,
      "loss_g0": 32.50276565551758,
      "loss_g1": 62.82068634033203,
      "q_g0": 0.9834702610969543,
      "q_g1": 0.016529696062207222,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 58.80801773071289,
      "loss_g0": 59.531890869140625,
      "loss_g1": 15.727660179138184,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.01652521640062332,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.89741039276123,
      "loss_g0": 12.778867721557617,
      "loss_g1": 19.951950073242188,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525952145457268,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.20406436920166,
      "loss_g0": 13.24955940246582,
      "loss_g1": 10.496549606323242,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.016525665298104286,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 46.51520538330078,
      "loss_g0": 47.089134216308594,
      "loss_g1": 12.352876663208008,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.306201934814453,
      "loss_g0": 16.353689193725586,
      "loss_g1": 13.479637145996094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.053226947784424,
      "loss_g0": 6.901594161987305,
      "loss_g1": 16.078466415405273,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523364931344986,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.114019393920898,
      "loss_g0": 13.290209770202637,
      "loss_g1": 2.626481056213379,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 28.31052017211914,
      "loss_g0": 28.519168853759766,
      "loss_g1": 15.89093017578125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.353872299194336,
      "loss_g0": 7.353872299194336,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.022506713867188,
      "loss_g0": 8.774421691894531,
      "loss_g1": 23.789472579956055,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.08585262298584,
      "loss_g0": 15.294520378112793,
      "loss_g1": 2.665153741836548,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.439987182617188,
      "loss_g0": 16.439987182617188,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.117477416992188,
      "loss_g0": 21.117477416992188,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.282480239868164,
      "loss_g0": 4.008701324462891,
      "loss_g1": 20.57883071899414,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.836201190948486,
      "loss_g0": 0.0,
      "loss_g1": 6.836201190948486,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.759779930114746,
      "loss_g0": 10.548881530761719,
      "loss_g1": 8.729722023010254,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.122989654541016,
      "loss_g0": 19.709304809570312,
      "loss_g1": 14.029136657714844,
      "q_g0": 0.016523005440831184,
      "q_g1": 0.9834769368171692,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.256372928619385,
      "loss_g0": 9.686606407165527,
      "loss_g1": 5.18194055557251,
      "q_g0": 0.016523465514183044,
      "q_g1": 0.9834765195846558,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.891356468200684,
      "loss_g0": 10.095647811889648,
      "loss_g1": 8.871122360229492,
      "q_g0": 0.016523588448762894,
      "q_g1": 0.9834764003753662,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.927122592926025,
      "loss_g0": 2.5474207401275635,
      "loss_g1": 4.967103958129883,
      "q_g0": 0.016523340716958046,
      "q_g1": 0.9834766387939453,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.621784210205078,
      "loss_g0": 40.78355407714844,
      "loss_g1": 25.367027282714844,
      "q_g0": 0.016524922102689743,
      "q_g1": 0.9834750890731812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.945088386535645,
      "loss_g0": 12.000396728515625,
      "loss_g1": 11.944158554077148,
      "q_g0": 0.016524922102689743,
      "q_g1": 0.9834750890731812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.908556938171387,
      "loss_g0": 10.123969078063965,
      "loss_g1": 8.888134002685547,
      "q_g0": 0.01652505435049534,
      "q_g1": 0.9834749698638916,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.999780654907227,
      "loss_g0": 6.733793258666992,
      "loss_g1": 15.138664245605469,
      "q_g0": 0.016524193808436394,
      "q_g1": 0.9834758043289185,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.318384170532227,
      "loss_g0": 13.577882766723633,
      "loss_g1": 9.246813774108887,
      "q_g0": 0.01652463525533676,
      "q_g1": 0.9834753274917603,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.464103698730469,
      "loss_g0": 4.93629789352417,
      "loss_g1": 8.523377418518066,
      "q_g0": 0.016524270176887512,
      "q_g1": 0.9834756851196289,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.38736343383789,
      "loss_g0": 14.42527961730957,
      "loss_g1": 13.369924545288086,
      "q_g0": 0.01652437075972557,
      "q_g1": 0.9834755659103394,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.888508319854736,
      "loss_g0": 10.669854164123535,
      "loss_g1": 4.7913665771484375,
      "q_g0": 0.01652497425675392,
      "q_g1": 0.9834749698638916,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.380596160888672,
      "loss_g0": 13.759658813476562,
      "loss_g1": 5.239797592163086,
      "q_g0": 0.01652585156261921,
      "q_g1": 0.9834741950035095,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.615630149841309,
      "loss_g0": 7.398702144622803,
      "loss_g1": 5.585668087005615,
      "q_g0": 0.016526037827134132,
      "q_g1": 0.9834739565849304,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.892514705657959,
      "loss_g0": 5.935000896453857,
      "loss_g1": 6.908604145050049,
      "q_g0": 0.016525937244296074,
      "q_g1": 0.98347407579422,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.813039302825928,
      "loss_g0": 20.68648910522461,
      "loss_g1": 4.5462799072265625,
      "q_g0": 0.016527589410543442,
      "q_g1": 0.983472466468811,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.01914405822754,
      "loss_g0": 8.927879333496094,
      "loss_g1": 21.222326278686523,
      "q_g0": 0.01652633212506771,
      "q_g1": 0.9834735989570618,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.450817108154297,
      "loss_g0": 36.189327239990234,
      "loss_g1": 1.8837517499923706,
      "q_g0": 0.016529835760593414,
      "q_g1": 0.9834701418876648,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.422938823699951,
      "loss_g0": 5.2156596183776855,
      "loss_g1": 7.460037708282471,
      "q_g0": 0.016529610380530357,
      "q_g1": 0.9834703803062439,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.8819799423217773,
      "loss_g0": 5.238246917724609,
      "loss_g1": 2.842376708984375,
      "q_g0": 0.016529850661754608,
      "q_g1": 0.9834701418876648,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.123628616333008,
      "loss_g0": 4.085394859313965,
      "loss_g1": 9.208307266235352,
      "q_g0": 0.016529330983757973,
      "q_g1": 0.983470618724823,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.426882028579712,
      "loss_g0": 16.0804386138916,
      "loss_g1": 3.2141947746276855,
      "q_g0": 0.01653064787387848,
      "q_g1": 0.9834693670272827,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.8108129501342773,
      "loss_g0": 5.938976287841797,
      "loss_g1": 2.758232355117798,
      "q_g0": 0.016530973836779594,
      "q_g1": 0.9834690093994141,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.30759859085083,
      "loss_g0": 1.5363972187042236,
      "loss_g1": 7.40460205078125,
      "q_g0": 0.016530370339751244,
      "q_g1": 0.9834696054458618,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.116796016693115,
      "loss_g0": 4.116796016693115,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.4193434715270996,
      "loss_g0": 3.427295684814453,
      "loss_g1": 2.9460012912750244,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.6765612363815308,
      "loss_g0": 1.6724437475204468,
      "loss_g1": 1.9216526746749878,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522448509931564,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.164580821990967,
      "loss_g0": 6.164580821990967,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.3504226207733154,
      "loss_g0": 2.3504226207733154,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.915678024291992,
      "loss_g0": 0.0,
      "loss_g1": 18.915678024291992,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.204299926757812,
      "loss_g0": 10.344034194946289,
      "loss_g1": 18.336353302001953,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 23.542531967163086,
      "loss_g0": 0.0,
      "loss_g1": 23.542531967163086,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.567960739135742,
      "loss_g0": 17.540088653564453,
      "loss_g1": 8.417229652404785,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.156997680664062,
      "loss_g0": 21.206769943237305,
      "loss_g1": 17.088958740234375,
      "q_g0": 0.01652284525334835,
      "q_g1": 0.9834771752357483,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 32.144561767578125,
      "loss_g0": 20.668689727783203,
      "loss_g1": 32.33735656738281,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.00206184387207,
      "loss_g0": 30.474411010742188,
      "loss_g1": 10.674885749816895,
      "q_g0": 0.01652444899082184,
      "q_g1": 0.9834755659103394,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 28.495485305786133,
      "loss_g0": 14.689420700073242,
      "loss_g1": 28.727434158325195,
      "q_g0": 0.01652301475405693,
      "q_g1": 0.9834769368171692,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.986692428588867,
      "loss_g0": 19.715614318847656,
      "loss_g1": 21.008047103881836,
      "q_g0": 0.016522882506251335,
      "q_g1": 0.9834770560264587,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 22.346410751342773,
      "loss_g0": 11.500969886779785,
      "loss_g1": 22.528614044189453,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 19.2146053314209,
      "loss_g0": 21.07481575012207,
      "loss_g1": 19.183353424072266,
      "q_g0": 0.016522621735930443,
      "q_g1": 0.9834774136543274,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.346513748168945,
      "loss_g0": 11.82697582244873,
      "loss_g1": 9.304841041564941,
      "q_g0": 0.01652287319302559,
      "q_g1": 0.9834770560264587,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.384653091430664,
      "loss_g0": 14.94653606414795,
      "loss_g1": 10.30801010131836,
      "q_g0": 0.016523348167538643,
      "q_g1": 0.9834766387939453,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.864063262939453,
      "loss_g0": 0.0,
      "loss_g1": 17.864063262939453,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.43010711669922,
      "loss_g0": 4.414066314697266,
      "loss_g1": 16.631975173950195,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.577648162841797,
      "loss_g0": 74.61803436279297,
      "loss_g1": 14.585394859313965,
      "q_g0": 0.016528556123375893,
      "q_g1": 0.9834713935852051,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.27069091796875,
      "loss_g0": 16.487857818603516,
      "loss_g1": 4.065349102020264,
      "q_g0": 0.01652981899678707,
      "q_g1": 0.9834701418876648,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.248794555664062,
      "loss_g0": 28.124435424804688,
      "loss_g1": 11.981935501098633,
      "q_g0": 0.01653147302567959,
      "q_g1": 0.9834685325622559,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.157049179077148,
      "loss_g0": 21.071125030517578,
      "loss_g1": 4.889516353607178,
      "q_g0": 0.016533123329281807,
      "q_g1": 0.9834669232368469,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.805703639984131,
      "loss_g0": 11.914051055908203,
      "loss_g1": 5.703011512756348,
      "q_g0": 0.016533760353922844,
      "q_g1": 0.9834662079811096,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.525028228759766,
      "loss_g0": 6.392005920410156,
      "loss_g1": 6.527264595031738,
      "q_g0": 0.016533751040697098,
      "q_g1": 0.9834662079811096,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 19.67960548400879,
      "loss_g0": 17.4801025390625,
      "loss_g1": 19.716583251953125,
      "q_g0": 0.01653352566063404,
      "q_g1": 0.9834664463996887,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.774848937988281,
      "loss_g0": 13.901833534240723,
      "loss_g1": 10.722278594970703,
      "q_g0": 0.016533851623535156,
      "q_g1": 0.9834660887718201,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.66718864440918,
      "loss_g0": 15.10315227508545,
      "loss_g1": 9.575798034667969,
      "q_g0": 0.016534417867660522,
      "q_g1": 0.9834655523300171,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.091065406799316,
      "loss_g0": 0.0,
      "loss_g1": 12.091065406799316,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.499871253967285,
      "loss_g0": 8.08378791809082,
      "loss_g1": 10.540460586547852,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.867094039916992,
      "loss_g0": 8.722025871276855,
      "loss_g1": 14.970331192016602,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.459051132202148,
      "loss_g0": 23.575946807861328,
      "loss_g1": 12.272274017333984,
      "q_g0": 0.016523580998182297,
      "q_g1": 0.9834764003753662,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.68246841430664,
      "loss_g0": 18.68246841430664,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.46303653717041,
      "loss_g0": 9.584837913513184,
      "loss_g1": 2.2129385471343994,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.8307294845581055,
      "loss_g0": 7.793475151062012,
      "loss_g1": 10.048226356506348,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522657126188278,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.213824272155762,
      "loss_g0": 7.196022987365723,
      "loss_g1": 8.273407936096191,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522767022252083,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.8670814037323,
      "loss_g0": 2.7536063194274902,
      "loss_g1": 9.621116638183594,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652347296476364,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.7813122272491455,
      "loss_g0": 2.733096122741699,
      "loss_g1": 5.65108060836792,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523776575922966,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.35278034210205,
      "loss_g0": 13.35278034210205,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.439858436584473,
      "loss_g0": 12.52490234375,
      "loss_g1": 7.377684116363525,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.6262712478637695,
      "loss_g0": 4.6262712478637695,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 40.85594940185547,
      "loss_g0": 41.48629379272461,
      "loss_g1": 3.3355422019958496,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.245578765869141,
      "loss_g0": 5.282085418701172,
      "loss_g1": 3.072537422180176,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.308821678161621,
      "loss_g0": 2.308821678161621,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.652288436889648,
      "loss_g0": 0.0,
      "loss_g1": 6.652288436889648,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.55238676071167,
      "loss_g0": 6.613658905029297,
      "loss_g1": 5.534556865692139,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.251490592956543,
      "loss_g0": 19.51763916015625,
      "loss_g1": 3.9949941635131836,
      "q_g0": 0.01652400940656662,
      "q_g1": 0.9834760427474976,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.7399061918258667,
      "loss_g0": 3.7162582874298096,
      "loss_g1": 1.7066997289657593,
      "q_g0": 0.016524210572242737,
      "q_g1": 0.9834758043289185,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.382839202880859,
      "loss_g0": 5.382839202880859,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.73988151550293,
      "loss_g0": 6.848326683044434,
      "loss_g1": 0.2848305106163025,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.931730270385742,
      "loss_g0": 8.982460021972656,
      "loss_g1": 5.912076473236084,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.216564655303955,
      "loss_g0": 7.149608135223389,
      "loss_g1": 11.201957702636719,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522839665412903,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.407583236694336,
      "loss_g0": 1.2888723611831665,
      "loss_g1": 8.473209381103516,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.0165235735476017,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.771729469299316,
      "loss_g0": 0.0,
      "loss_g1": 10.771729469299316,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 32.206153869628906,
      "loss_g0": 61.702796936035156,
      "loss_g1": 31.710607528686523,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.64146614074707,
      "loss_g0": 22.51464080810547,
      "loss_g1": 17.559595108032227,
      "q_g0": 0.016522930935025215,
      "q_g1": 0.9834770560264587,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.550792694091797,
      "loss_g0": 25.052017211914062,
      "loss_g1": 18.441564559936523,
      "q_g0": 0.016523605212569237,
      "q_g1": 0.9834764003753662,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 37.96096420288086,
      "loss_g0": 30.552734375,
      "loss_g1": 38.085426330566406,
      "q_g0": 0.016522839665412903,
      "q_g1": 0.9834771752357483,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.172615051269531,
      "loss_g0": 10.172615051269531,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.696929931640625,
      "loss_g0": 24.802343368530273,
      "loss_g1": 18.4222469329834,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.152727127075195,
      "loss_g0": 15.124746322631836,
      "loss_g1": 16.818195343017578,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522597521543503,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 49.00303649902344,
      "loss_g0": 49.222564697265625,
      "loss_g1": 35.93568420410156,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.069433212280273,
      "loss_g0": 0.0,
      "loss_g1": 16.069433212280273,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.858394622802734,
      "loss_g0": 26.771116256713867,
      "loss_g1": 10.591060638427734,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.263315200805664,
      "loss_g0": 33.06898498535156,
      "loss_g1": 18.014556884765625,
      "q_g0": 0.01652396097779274,
      "q_g1": 0.9834760427474976,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.340890884399414,
      "loss_g0": 11.62979507446289,
      "loss_g1": 7.268828392028809,
      "q_g0": 0.016524402424693108,
      "q_g1": 0.9834755659103394,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.090576171875,
      "loss_g0": 30.414527893066406,
      "loss_g1": 13.81627082824707,
      "q_g0": 0.016526101157069206,
      "q_g1": 0.9834739565849304,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.716842651367188,
      "loss_g0": 3.1129140853881836,
      "loss_g1": 9.82780933380127,
      "q_g0": 0.01652541570365429,
      "q_g1": 0.9834745526313782,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.807908058166504,
      "loss_g0": 22.825057983398438,
      "loss_g1": 14.673188209533691,
      "q_g0": 0.016526246443390846,
      "q_g1": 0.9834737181663513,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.510651588439941,
      "loss_g0": 5.897741317749023,
      "loss_g1": 10.588164329528809,
      "q_g0": 0.016525765880942345,
      "q_g1": 0.9834741950035095,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.279888153076172,
      "loss_g0": 6.189624786376953,
      "loss_g1": 17.466228485107422,
      "q_g0": 0.01652461104094982,
      "q_g1": 0.9834753274917603,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.216463088989258,
      "loss_g0": 21.216463088989258,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.492267608642578,
      "loss_g0": 9.539078712463379,
      "loss_g1": 6.7059173583984375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 66.39398956298828,
      "loss_g0": 67.3848876953125,
      "loss_g1": 7.411507606506348,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.619158744812012,
      "loss_g0": 8.4127197265625,
      "loss_g1": 20.906238555908203,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523705795407295,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.563112735748291,
      "loss_g0": 5.499305725097656,
      "loss_g1": 9.360734939575195,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524100676178932,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 64.95401763916016,
      "loss_g0": 65.99252319335938,
      "loss_g1": 3.138025999069214,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.374969482421875,
      "loss_g0": 10.366134643554688,
      "loss_g1": 10.900839805603027,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.0165224801748991,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.451435089111328,
      "loss_g0": 0.0,
      "loss_g1": 6.451435089111328,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.07455062866211,
      "loss_g0": 11.07455062866211,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.427319526672363,
      "loss_g0": 4.3472208976745605,
      "loss_g1": 9.195083618164062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.9521992206573486,
      "loss_g0": 2.9565696716308594,
      "loss_g1": 2.6920430660247803,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.36157488822937,
      "loss_g0": 3.324049234390259,
      "loss_g1": 5.595213890075684,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522657126188278,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.823848247528076,
      "loss_g0": 7.890144348144531,
      "loss_g1": 3.8776650428771973,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 22.63703155517578,
      "loss_g0": 22.929630279541016,
      "loss_g1": 5.22039794921875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.876936912536621,
      "loss_g0": 7.962169647216797,
      "loss_g1": 2.8035495281219482,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.508544921875,
      "loss_g0": 18.661212921142578,
      "loss_g1": 9.42122745513916,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.493485927581787,
      "loss_g0": 0.0,
      "loss_g1": 5.493485927581787,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.080692291259766,
      "loss_g0": 23.893753051757812,
      "loss_g1": 3.747832775115967,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.5770583152771,
      "loss_g0": 4.308206558227539,
      "loss_g1": 7.631974697113037,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.297019958496094,
      "loss_g0": 20.029075622558594,
      "loss_g1": 8.099905967712402,
      "q_g0": 0.01652364432811737,
      "q_g1": 0.9834764003753662,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.2125558853149414,
      "loss_g0": 17.04638671875,
      "loss_g1": 1.963305115699768,
      "q_g0": 0.016525182873010635,
      "q_g1": 0.9834747910499573,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.255342960357666,
      "loss_g0": 10.182723999023438,
      "loss_g1": 6.189350128173828,
      "q_g0": 0.016525588929653168,
      "q_g1": 0.9834744334220886,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.608368873596191,
      "loss_g0": 41.198482513427734,
      "loss_g1": 7.0438232421875,
      "q_g0": 0.016529075801372528,
      "q_g1": 0.9834709763526917,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.824150085449219,
      "loss_g0": 9.64469051361084,
      "loss_g1": 4.743129730224609,
      "q_g0": 0.016529571264982224,
      "q_g1": 0.9834703803062439,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.5685319900512695,
      "loss_g0": 11.196572303771973,
      "loss_g1": 5.473936080932617,
      "q_g0": 0.01653016172349453,
      "q_g1": 0.9834697842597961,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.072608947753906,
      "loss_g0": 1.8192565441131592,
      "loss_g1": 14.278547286987305,
      "q_g0": 0.016528887674212456,
      "q_g1": 0.9834710955619812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.9139981269836426,
      "loss_g0": 3.8191189765930176,
      "loss_g1": 2.8987860679626465,
      "q_g0": 0.01652897521853447,
      "q_g1": 0.9834710955619812,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.4337384700775146,
      "loss_g0": 2.4337384700775146,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.538607120513916,
      "loss_g0": 1.5489870309829712,
      "loss_g1": 0.9207590818405151,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.26470947265625,
      "loss_g0": 18.49382209777832,
      "loss_g1": 4.627063274383545,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.499879837036133,
      "loss_g0": 14.707075119018555,
      "loss_g1": 2.1668286323547363,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.0069319009780884,
      "loss_g0": 0.9477275609970093,
      "loss_g1": 4.530918598175049,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522789373993874,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.0399065017700195,
      "loss_g0": 6.109304428100586,
      "loss_g1": 1.909055471420288,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 36.33671188354492,
      "loss_g0": 36.303836822509766,
      "loss_g1": 38.29352569580078,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652262918651104,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.202316284179688,
      "loss_g0": 16.946683883666992,
      "loss_g1": 32.41689682006836,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524210572242737,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.264122009277344,
      "loss_g0": 26.318958282470703,
      "loss_g1": 23.000301361083984,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652386784553528,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 31.464439392089844,
      "loss_g0": 31.464439392089844,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.941755294799805,
      "loss_g0": 25.064956665039062,
      "loss_g1": 17.608293533325195,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.324403285980225,
      "loss_g0": 6.957436561584473,
      "loss_g1": 29.164581298828125,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524696722626686,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.273591995239258,
      "loss_g0": 20.412227630615234,
      "loss_g1": 12.022200584411621,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652384363114834,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.689828872680664,
      "loss_g0": 12.494466781616211,
      "loss_g1": 24.316627502441406,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.01652505435049534,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 36.27484130859375,
      "loss_g0": 36.649078369140625,
      "loss_g1": 13.999110221862793,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522744670510292,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.839473724365234,
      "loss_g0": 22.073989868164062,
      "loss_g1": 7.880168914794922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.84539031982422,
      "loss_g0": 19.040468215942383,
      "loss_g1": 7.233582496643066,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.629451751708984,
      "loss_g0": 17.61690330505371,
      "loss_g1": 18.376371383666992,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 23.98695182800293,
      "loss_g0": 24.149871826171875,
      "loss_g1": 14.289306640625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.21158790588379,
      "loss_g0": 20.4520320892334,
      "loss_g1": 5.8994927406311035,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.07941436767578,
      "loss_g0": 20.208282470703125,
      "loss_g1": 12.408641815185547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.8665771484375,
      "loss_g0": 0.0,
      "loss_g1": 25.8665771484375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 12.634780883789062,
      "loss_g0": 12.477399826049805,
      "loss_g1": 12.63742446899414,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.377115249633789,
      "loss_g0": 5.4856367111206055,
      "loss_g1": 15.543292045593262,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.82632827758789,
      "loss_g0": 22.439579010009766,
      "loss_g1": 15.715221405029297,
      "q_g0": 0.016523106023669243,
      "q_g1": 0.9834768176078796,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.35927152633667,
      "loss_g0": 7.455932140350342,
      "loss_g1": 6.340846538543701,
      "q_g0": 0.016523215919733047,
      "q_g1": 0.9834766983985901,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.40795612335205,
      "loss_g0": 9.156072616577148,
      "loss_g1": 14.496188163757324,
      "q_g0": 0.01652267388999462,
      "q_g1": 0.9834772944450378,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.406820297241211,
      "loss_g0": 0.0,
      "loss_g1": 10.406820297241211,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.973532199859619,
      "loss_g0": 16.49531364440918,
      "loss_g1": 6.813566207885742,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.670708179473877,
      "loss_g0": 15.002006530761719,
      "loss_g1": 5.513932704925537,
      "q_g0": 0.016523394733667374,
      "q_g1": 0.9834766387939453,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.8004831075668335,
      "loss_g0": 8.326284408569336,
      "loss_g1": 1.6908385753631592,
      "q_g0": 0.016524069011211395,
      "q_g1": 0.983475923538208,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.148682594299316,
      "loss_g0": 35.81544494628906,
      "loss_g1": 5.650136470794678,
      "q_g0": 0.016527146100997925,
      "q_g1": 0.9834728240966797,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.61679458618164,
      "loss_g0": 6.3597869873046875,
      "loss_g1": 13.738741874694824,
      "q_g0": 0.016526388004422188,
      "q_g1": 0.9834735989570618,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 23.108203887939453,
      "loss_g0": 13.704374313354492,
      "loss_g1": 23.266216278076172,
      "q_g0": 0.016525408253073692,
      "q_g1": 0.9834745526313782,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.360160827636719,
      "loss_g0": 6.49704122543335,
      "loss_g1": 6.357860565185547,
      "q_g0": 0.016525425016880035,
      "q_g1": 0.9834745526313782,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.219002723693848,
      "loss_g0": 12.889451026916504,
      "loss_g1": 9.157325744628906,
      "q_g0": 0.016525806859135628,
      "q_g1": 0.9834741950035095,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.915602684020996,
      "loss_g0": 0.0,
      "loss_g1": 8.915602684020996,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.089615345001221,
      "loss_g0": 2.442434310913086,
      "loss_g1": 6.150887966156006,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.223321914672852,
      "loss_g0": 19.934078216552734,
      "loss_g1": 9.04336929321289,
      "q_g0": 0.016523534432053566,
      "q_g1": 0.9834765195846558,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.552849769592285,
      "loss_g0": 9.583391189575195,
      "loss_g1": 5.485130786895752,
      "q_g0": 0.016523944213986397,
      "q_g1": 0.9834760427474976,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.809321403503418,
      "loss_g0": 6.176435470581055,
      "loss_g1": 5.803153038024902,
      "q_g0": 0.016523977741599083,
      "q_g1": 0.9834760427474976,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.1621856689453125,
      "loss_g0": 6.43441104888916,
      "loss_g1": 7.174414157867432,
      "q_g0": 0.016523899510502815,
      "q_g1": 0.9834761619567871,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.4310126304626465,
      "loss_g0": 12.184717178344727,
      "loss_g1": 5.317535400390625,
      "q_g0": 0.016524596139788628,
      "q_g1": 0.9834754467010498,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.3289666175842285,
      "loss_g0": 7.289072036743164,
      "loss_g1": 3.2624263763427734,
      "q_g0": 0.01652500592172146,
      "q_g1": 0.9834749698638916,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.586979866027832,
      "loss_g0": 14.15816593170166,
      "loss_g1": 7.476561546325684,
      "q_g0": 0.016525689512491226,
      "q_g1": 0.9834743142127991,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.5589606761932373,
      "loss_g0": 1.2387375831604004,
      "loss_g1": 2.5811448097229004,
      "q_g0": 0.016525547951459885,
      "q_g1": 0.9834744334220886,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.594468116760254,
      "loss_g0": 4.630910873413086,
      "loss_g1": 5.610658645629883,
      "q_g0": 0.016525447368621826,
      "q_g1": 0.9834745526313782,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 0.7779190540313721,
      "loss_g0": 7.454524040222168,
      "loss_g1": 0.6657264232635498,
      "q_g0": 0.01652613840997219,
      "q_g1": 0.9834738373756409,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.5868258476257324,
      "loss_g0": 1.1647379398345947,
      "loss_g1": 2.610722064971924,
      "q_g0": 0.0165259912610054,
      "q_g1": 0.9834739565849304,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.1153037548065186,
      "loss_g0": 14.673757553100586,
      "loss_g1": 2.921064615249634,
      "q_g0": 0.016527200117707253,
      "q_g1": 0.9834728240966797,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.834257125854492,
      "loss_g0": 5.834257125854492,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 27.178054809570312,
      "loss_g0": 27.614398956298828,
      "loss_g1": 1.2051563262939453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.9883394241333,
      "loss_g0": 10.990655899047852,
      "loss_g1": 10.850443840026855,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 23.520893096923828,
      "loss_g0": 23.738698959350586,
      "loss_g1": 10.556131362915039,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.3672449588775635,
      "loss_g0": 2.3945608139038086,
      "loss_g1": 0.7412973046302795,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.906116485595703,
      "loss_g0": 14.131770133972168,
      "loss_g1": 0.4743266701698303,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 22.82025909423828,
      "loss_g0": 22.641353607177734,
      "loss_g1": 33.468589782714844,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652352698147297,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 33.78401184082031,
      "loss_g0": 33.781639099121094,
      "loss_g1": 33.92521667480469,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523543745279312,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.221129417419434,
      "loss_g0": 13.688806533813477,
      "loss_g1": 45.898414611816406,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.016526827588677406,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 19.58540916442871,
      "loss_g0": 19.38365936279297,
      "loss_g1": 31.590049743652344,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.01652807556092739,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 35.814754486083984,
      "loss_g0": 36.17344665527344,
      "loss_g1": 14.468421936035156,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.01652585156261921,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.521669387817383,
      "loss_g0": 26.715957641601562,
      "loss_g1": 14.958555221557617,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.01652464270591736,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.400596618652344,
      "loss_g0": 0.0,
      "loss_g1": 25.400596618652344,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.607492446899414,
      "loss_g0": 14.560909271240234,
      "loss_g1": 17.658674240112305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.076523780822754,
      "loss_g0": 16.623332977294922,
      "loss_g1": 10.983333587646484,
      "q_g0": 0.016522999852895737,
      "q_g1": 0.9834769368171692,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.009252548217773,
      "loss_g0": 22.007892608642578,
      "loss_g1": 26.07647705078125,
      "q_g0": 0.01652258075773716,
      "q_g1": 0.9834774136543274,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.734560012817383,
      "loss_g0": 15.734560012817383,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.350373268127441,
      "loss_g0": 8.29241943359375,
      "loss_g1": 11.79998779296875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.821911811828613,
      "loss_g0": 0.0,
      "loss_g1": 13.821911811828613,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.821855545043945,
      "loss_g0": 14.418542861938477,
      "loss_g1": 9.744630813598633,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.519111633300781,
      "loss_g0": 10.519111633300781,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.072206497192383,
      "loss_g0": 12.756521224975586,
      "loss_g1": 92.38680267333984,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.657452583312988,
      "loss_g0": 15.728565216064453,
      "loss_g1": 11.424535751342773,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.405764579772949,
      "loss_g0": 7.154296875,
      "loss_g1": 22.37265396118164,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523977741599083,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.952089309692383,
      "loss_g0": 21.191993713378906,
      "loss_g1": 6.672022819519043,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522489488124847,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 22.362224578857422,
      "loss_g0": 22.471229553222656,
      "loss_g1": 15.873847961425781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.6316864490509033,
      "loss_g0": 3.4994304180145264,
      "loss_g1": 11.503682136535645,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523240134119987,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.304821014404297,
      "loss_g0": 11.204821586608887,
      "loss_g1": 17.256580352783203,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652386784553528,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.602591037750244,
      "loss_g0": 3.5456411838531494,
      "loss_g1": 6.9920973777771,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524218022823334,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.253562927246094,
      "loss_g0": 5.1561174392700195,
      "loss_g1": 11.053017616271973,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524821519851685,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.146941184997559,
      "loss_g0": 13.158794403076172,
      "loss_g1": 12.441469192504883,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524743288755417,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.146499633789062,
      "loss_g0": 16.278684616088867,
      "loss_g1": 8.279091835021973,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523927450180054,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.306146621704102,
      "loss_g0": 15.422975540161133,
      "loss_g1": 8.352405548095703,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523201018571854,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 15.099282264709473,
      "loss_g0": 0.0,
      "loss_g1": 15.099282264709473,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.790525436401367,
      "loss_g0": 16.13670539855957,
      "loss_g1": 13.75110912322998,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.141942977905273,
      "loss_g0": 8.30699348449707,
      "loss_g1": 11.189569473266602,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.322650909423828,
      "loss_g0": 17.035266876220703,
      "loss_g1": 3.0922586917877197,
      "q_g0": 0.016523851081728935,
      "q_g1": 0.9834761619567871,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.043292999267578,
      "loss_g0": 28.349882125854492,
      "loss_g1": 9.735679626464844,
      "q_g0": 0.01652575097978115,
      "q_g1": 0.9834743142127991,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.277474880218506,
      "loss_g0": 19.481517791748047,
      "loss_g1": 7.072389125823975,
      "q_g0": 0.01652701571583748,
      "q_g1": 0.9834730625152588,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.189053773880005,
      "loss_g0": 23.89253807067871,
      "loss_g1": 2.841090679168701,
      "q_g0": 0.01652917079627514,
      "q_g1": 0.9834708571434021,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.044971227645874,
      "loss_g0": 8.75456714630127,
      "loss_g1": 2.9490067958831787,
      "q_g0": 0.016529759392142296,
      "q_g1": 0.9834702610969543,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.846923351287842,
      "loss_g0": 17.154579162597656,
      "loss_g1": 7.6904754638671875,
      "q_g0": 0.016530726104974747,
      "q_g1": 0.9834692478179932,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.763582468032837,
      "loss_g0": 12.271570205688477,
      "loss_g1": 2.603756904602051,
      "q_g0": 0.016531718894839287,
      "q_g1": 0.9834682941436768,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.32198429107666,
      "loss_g0": 8.32198429107666,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.7380263805389404,
      "loss_g0": 3.597808837890625,
      "loss_g1": 12.08430004119873,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.245158195495605,
      "loss_g0": 8.344171524047852,
      "loss_g1": 2.351541042327881,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.689111709594727,
      "loss_g0": 16.91215705871582,
      "loss_g1": 3.412630558013916,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 36.27208709716797,
      "loss_g0": 36.80019760131836,
      "loss_g1": 4.83686637878418,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.389618873596191,
      "loss_g0": 14.598079681396484,
      "loss_g1": 1.9812469482421875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.196405410766602,
      "loss_g0": 10.284676551818848,
      "loss_g1": 4.942162990570068,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.2217793464660645,
      "loss_g0": 7.28383207321167,
      "loss_g1": 3.5281777381896973,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.262832164764404,
      "loss_g0": 4.137933254241943,
      "loss_g1": 11.696943283081055,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523193567991257,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 0.9970887303352356,
      "loss_g0": 0.9251555800437927,
      "loss_g1": 5.278499603271484,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652364432811737,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.86959457397461,
      "loss_g0": 19.15517234802246,
      "loss_g1": 1.87091863155365,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.377248764038086,
      "loss_g0": 20.649457931518555,
      "loss_g1": 4.174293518066406,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.804337501525879,
      "loss_g0": 2.847781181335449,
      "loss_g1": 0.2183992862701416,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 57.49811935424805,
      "loss_g0": 58.16828155517578,
      "loss_g1": 17.607263565063477,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.808734893798828,
      "loss_g0": 26.791000366210938,
      "loss_g1": 27.86441993713379,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.427139282226562,
      "loss_g0": 13.347197532653809,
      "loss_g1": 18.185401916503906,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523031517863274,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.830181121826172,
      "loss_g0": 18.830181121826172,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 20.6861515045166,
      "loss_g0": 20.631206512451172,
      "loss_g1": 23.956689834594727,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 23.756858825683594,
      "loss_g0": 23.817272186279297,
      "loss_g1": 20.160743713378906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 30.54085922241211,
      "loss_g0": 30.862255096435547,
      "loss_g1": 11.410089492797852,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.71011734008789,
      "loss_g0": 24.846019744873047,
      "loss_g1": 16.6207332611084,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 41.328453063964844,
      "loss_g0": 41.374725341796875,
      "loss_g1": 38.57415008544922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.695192337036133,
      "loss_g0": 25.81087875366211,
      "loss_g1": 18.809139251708984,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.833011627197266,
      "loss_g0": 16.91257095336914,
      "loss_g1": 12.097269058227539,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.477294921875,
      "loss_g0": 4.367682933807373,
      "loss_g1": 11.001521110534668,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523102298378944,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 27.149995803833008,
      "loss_g0": 27.149995803833008,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 28.748432159423828,
      "loss_g0": 29.020587921142578,
      "loss_g1": 12.548659324645996,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.304040908813477,
      "loss_g0": 0.0,
      "loss_g1": 11.304040908813477,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.660828590393066,
      "loss_g0": 11.031099319458008,
      "loss_g1": 14.721807479858398,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.548473358154297,
      "loss_g0": 20.981340408325195,
      "loss_g1": 16.4739990234375,
      "q_g0": 0.016522882506251335,
      "q_g1": 0.9834770560264587,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 37.11112976074219,
      "loss_g0": 20.294172286987305,
      "loss_g1": 37.393653869628906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.960147857666016,
      "loss_g0": 11.268839836120605,
      "loss_g1": 7.904560565948486,
      "q_g0": 0.016522767022252083,
      "q_g1": 0.9834771752357483,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.429357528686523,
      "loss_g0": 21.429357528686523,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 41.24272155761719,
      "loss_g0": 41.742252349853516,
      "loss_g1": 11.508651733398438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.31416130065918,
      "loss_g0": 24.673564910888672,
      "loss_g1": 2.9211273193359375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.734169960021973,
      "loss_g0": 14.817535400390625,
      "loss_g1": 9.771951675415039,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 21.98408317565918,
      "loss_g0": 22.329715728759766,
      "loss_g1": 1.410731554031372,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 14.082777976989746,
      "loss_g0": 14.055809020996094,
      "loss_g1": 15.688102722167969,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522590070962906,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.514086723327637,
      "loss_g0": 8.514086723327637,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.31468963623047,
      "loss_g0": 16.411741256713867,
      "loss_g1": 10.537847518920898,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 16.51980209350586,
      "loss_g0": 16.71083641052246,
      "loss_g1": 5.14879035949707,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.16848373413086,
      "loss_g0": 8.230010986328125,
      "loss_g1": 4.506148815155029,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.174102783203125,
      "loss_g0": 4.201597213745117,
      "loss_g1": 2.5375068187713623,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 26.738021850585938,
      "loss_g0": 27.077726364135742,
      "loss_g1": 6.517490386962891,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.078457832336426,
      "loss_g0": 8.003362655639648,
      "loss_g1": 12.548301696777344,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522889956831932,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.363163471221924,
      "loss_g0": 3.114813804626465,
      "loss_g1": 18.144062042236328,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524426639080048,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.8862762451171875,
      "loss_g0": 0.0,
      "loss_g1": 5.8862762451171875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 4.951663017272949,
      "loss_g0": 26.638404846191406,
      "loss_g1": 4.587325572967529,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.808280944824219,
      "loss_g0": 0.0,
      "loss_g1": 11.808280944824219,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.011842727661133,
      "loss_g0": 13.758514404296875,
      "loss_g1": 10.9656982421875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.8737807273864746,
      "loss_g0": 13.126422882080078,
      "loss_g1": 2.7015249729156494,
      "q_g0": 0.016523487865924835,
      "q_g1": 0.9834765195846558,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.3118391036987305,
      "loss_g0": 0.0,
      "loss_g1": 5.3118391036987305,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.4888293743133545,
      "loss_g0": 7.021522045135498,
      "loss_g1": 3.4294800758361816,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.149799346923828,
      "loss_g0": 11.101617813110352,
      "loss_g1": 8.100207328796387,
      "q_g0": 0.016522729769349098,
      "q_g1": 0.9834772944450378,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 3.1369504928588867,
      "loss_g0": 8.22480583190918,
      "loss_g1": 3.0514702796936035,
      "q_g0": 0.01652325503528118,
      "q_g1": 0.9834766983985901,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 10.110016822814941,
      "loss_g0": 10.110016822814941,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 18.72027587890625,
      "loss_g0": 18.982032775878906,
      "loss_g1": 3.1395301818847656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.511874198913574,
      "loss_g0": 8.616358757019043,
      "loss_g1": 2.2925174236297607,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.941988945007324,
      "loss_g0": 10.069679260253906,
      "loss_g1": 2.341334342956543,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.184453964233398,
      "loss_g0": 11.357872009277344,
      "loss_g1": 0.8619327545166016,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.6288933753967285,
      "loss_g0": 0.0,
      "loss_g1": 2.6288933753967285,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.020066738128662,
      "loss_g0": 6.241400718688965,
      "loss_g1": 1.9491482973098755,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 0.8326910138130188,
      "loss_g0": 4.19261360168457,
      "loss_g1": 0.7762430906295776,
      "q_g0": 0.016522767022252083,
      "q_g1": 0.9834771752357483,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.589872360229492,
      "loss_g0": 9.500563621520996,
      "loss_g1": 25.860172271728516,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 17.761940002441406,
      "loss_g0": 18.545846939086914,
      "loss_g1": 17.748769760131836,
      "q_g0": 0.01652250438928604,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.067261695861816,
      "loss_g0": 0.0,
      "loss_g1": 7.067261695861816,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.760404586791992,
      "loss_g0": 22.907642364501953,
      "loss_g1": 11.573131561279297,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 24.54339027404785,
      "loss_g0": 14.507303237915039,
      "loss_g1": 24.71199607849121,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 19.998069763183594,
      "loss_g0": 15.834259033203125,
      "loss_g1": 20.068021774291992,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 25.35243797302246,
      "loss_g0": 37.625213623046875,
      "loss_g1": 25.146240234375,
      "q_g0": 0.0165236983448267,
      "q_g1": 0.9834762811660767,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.404979705810547,
      "loss_g0": 6.693911075592041,
      "loss_g1": 8.433727264404297,
      "q_g0": 0.016523517668247223,
      "q_g1": 0.9834765195846558,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 19.475406646728516,
      "loss_g0": 18.519763946533203,
      "loss_g1": 19.49146270751953,
      "q_g0": 0.016523417085409164,
      "q_g1": 0.9834766387939453,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.594185829162598,
      "loss_g0": 18.076004028320312,
      "loss_g1": 13.51888370513916,
      "q_g0": 0.016523877158761024,
      "q_g1": 0.9834761619567871,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.239896774291992,
      "loss_g0": 45.78993606567383,
      "loss_g1": 12.692893028259277,
      "q_g0": 0.016527263447642326,
      "q_g1": 0.9834727048873901,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.993220329284668,
      "loss_g0": 2.655010223388672,
      "loss_g1": 12.150139808654785,
      "q_g0": 0.016526293009519577,
      "q_g1": 0.9834737181663513,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 9.69083309173584,
      "loss_g0": 44.67341232299805,
      "loss_g1": 9.10285472869873,
      "q_g0": 0.016529928892850876,
      "q_g1": 0.9834700226783752,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 5.039255619049072,
      "loss_g0": 7.284224033355713,
      "loss_g1": 5.001522064208984,
      "q_g0": 0.016530169174075127,
      "q_g1": 0.9834697842597961,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 6.987278938293457,
      "loss_g0": 8.260467529296875,
      "loss_g1": 6.965878963470459,
      "q_g0": 0.016530301421880722,
      "q_g1": 0.9834696650505066,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 1.4527171850204468,
      "loss_g0": 9.181291580200195,
      "loss_g1": 1.322807788848877,
      "q_g0": 0.01653110608458519,
      "q_g1": 0.9834688901901245,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 2.145216226577759,
      "loss_g0": 4.780805587768555,
      "loss_g1": 2.1009137630462646,
      "q_g0": 0.016531378030776978,
      "q_g1": 0.9834686517715454,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 7.935704231262207,
      "loss_g0": 7.935704231262207,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 8.546916961669922,
      "loss_g0": 8.546916961669922,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "eval_cer": 0.06067961165048544,
      "eval_loss": 13.589922904968262,
      "eval_runtime": 81.2784,
      "eval_samples_per_second": 20.62,
      "eval_steps_per_second": 5.155,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 11.699625015258789,
      "loss_g0": 11.114060401916504,
      "loss_g1": 46.55463409423828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1100
    },
    {
      "epoch": 5.583756345177665,
      "gdro_loss": 13.800586700439453,
      "loss_g0": 13.48279094696045,
      "loss_g1": 32.71465301513672,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652439311146736,
      "step": 1100
    },
    {
      "epoch": 6.091370558375634,
      "grad_norm": 57.09098434448242,
      "learning_rate": 0.00019236363636363635,
      "loss": 20.05,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 35.99641799926758,
      "loss_g0": 28.151897430419922,
      "loss_g1": 36.128204345703125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.39615249633789,
      "loss_g0": 18.39615249633789,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.79543399810791,
      "loss_g0": 15.629783630371094,
      "loss_g1": 25.65555763244629,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 28.41619873046875,
      "loss_g0": 0.0,
      "loss_g1": 28.41619873046875,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 25.368574142456055,
      "loss_g0": 21.261554718017578,
      "loss_g1": 25.437572479248047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.342418670654297,
      "loss_g0": 8.28596305847168,
      "loss_g1": 20.544967651367188,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.532042980194092,
      "loss_g0": 23.074491500854492,
      "loss_g1": 6.254100322723389,
      "q_g0": 0.016524137929081917,
      "q_g1": 0.9834758043289185,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 30.64635467529297,
      "loss_g0": 49.898681640625,
      "loss_g1": 30.32284164428711,
      "q_g0": 0.01652613840997219,
      "q_g1": 0.9834738373756409,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.31245231628418,
      "loss_g0": 17.988285064697266,
      "loss_g1": 11.20026683807373,
      "q_g0": 0.016526836901903152,
      "q_g1": 0.9834731221199036,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.621047973632812,
      "loss_g0": 7.882821083068848,
      "loss_g1": 17.784683227539062,
      "q_g0": 0.016525819897651672,
      "q_g1": 0.9834741950035095,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.92293643951416,
      "loss_g0": 18.04050064086914,
      "loss_g1": 11.820136070251465,
      "q_g0": 0.01652645505964756,
      "q_g1": 0.9834734797477722,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.746646881103516,
      "loss_g0": 23.30064582824707,
      "loss_g1": 18.67011833190918,
      "q_g0": 0.016526928171515465,
      "q_g1": 0.9834730625152588,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 26.084548950195312,
      "loss_g0": 100.27391052246094,
      "loss_g1": 24.837230682373047,
      "q_g0": 0.01653464138507843,
      "q_g1": 0.983465313911438,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 67.56437683105469,
      "loss_g0": 13.389717102050781,
      "loss_g1": 68.4748764038086,
      "q_g0": 0.016529005020856857,
      "q_g1": 0.9834709763526917,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.808838844299316,
      "loss_g0": 13.151016235351562,
      "loss_g1": 13.819894790649414,
      "q_g0": 0.01652892865240574,
      "q_g1": 0.9834710955619812,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.280744552612305,
      "loss_g0": 24.728065490722656,
      "loss_g1": 12.071529388427734,
      "q_g0": 0.016530223190784454,
      "q_g1": 0.9834697842597961,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.4702677726745605,
      "loss_g0": 4.961643218994141,
      "loss_g1": 4.462008476257324,
      "q_g0": 0.016530269756913185,
      "q_g1": 0.9834696650505066,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.638616561889648,
      "loss_g0": 11.286766052246094,
      "loss_g1": 6.560487747192383,
      "q_g0": 0.016530757769942284,
      "q_g1": 0.9834692478179932,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.90272331237793,
      "loss_g0": 15.357637405395508,
      "loss_g1": 20.99592399597168,
      "q_g0": 0.01653018407523632,
      "q_g1": 0.9834697842597961,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.05925750732422,
      "loss_g0": 6.538591384887695,
      "loss_g1": 20.286495208740234,
      "q_g0": 0.016528772190213203,
      "q_g1": 0.9834712147712708,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.388690948486328,
      "loss_g0": 10.835124969482422,
      "loss_g1": 10.381187438964844,
      "q_g0": 0.016528820618987083,
      "q_g1": 0.9834712147712708,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.548182010650635,
      "loss_g0": 3.171855926513672,
      "loss_g1": 5.588119029998779,
      "q_g0": 0.016528571024537086,
      "q_g1": 0.9834713935852051,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 23.84088897705078,
      "loss_g0": 1.7986236810684204,
      "loss_g1": 24.211286544799805,
      "q_g0": 0.016526278108358383,
      "q_g1": 0.9834737181663513,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.3377997875213623,
      "loss_g0": 21.049409866333008,
      "loss_g1": 2.0233325958251953,
      "q_g0": 0.01652822457253933,
      "q_g1": 0.9834717512130737,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.464726448059082,
      "loss_g0": 11.045663833618164,
      "loss_g1": 14.522186279296875,
      "q_g0": 0.016527866944670677,
      "q_g1": 0.9834721088409424,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.155301094055176,
      "loss_g0": 31.62470245361328,
      "loss_g1": 2.676767587661743,
      "q_g0": 0.016530826687812805,
      "q_g1": 0.9834691286087036,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.068087577819824,
      "loss_g0": 4.890925884246826,
      "loss_g1": 5.071065425872803,
      "q_g0": 0.01653081364929676,
      "q_g1": 0.9834692478179932,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.999661445617676,
      "loss_g0": 18.808679580688477,
      "loss_g1": 8.8347749710083,
      "q_g0": 0.01653183624148369,
      "q_g1": 0.9834681749343872,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.4160850048065186,
      "loss_g0": 13.318862915039062,
      "loss_g1": 3.2496113777160645,
      "q_g0": 0.016532866284251213,
      "q_g1": 0.983467161655426,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.80234146118164,
      "loss_g0": 6.841504096984863,
      "loss_g1": 9.8521146774292,
      "q_g0": 0.01653255708515644,
      "q_g1": 0.9834675192832947,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.890618801116943,
      "loss_g0": 8.579458236694336,
      "loss_g1": 7.8790388107299805,
      "q_g0": 0.01653262786567211,
      "q_g1": 0.9834674000740051,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 19.549148559570312,
      "loss_g0": 17.152219772338867,
      "loss_g1": 19.589441299438477,
      "q_g0": 0.016532378271222115,
      "q_g1": 0.9834676384925842,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.06494140625,
      "loss_g0": 2.3420629501342773,
      "loss_g1": 10.194758415222168,
      "q_g0": 0.016531571745872498,
      "q_g1": 0.9834684133529663,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.98643684387207,
      "loss_g0": 19.161514282226562,
      "loss_g1": 10.849011421203613,
      "q_g0": 0.016532426699995995,
      "q_g1": 0.9834676384925842,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.8700313568115234,
      "loss_g0": 9.788752555847168,
      "loss_g1": 3.7705318927764893,
      "q_g0": 0.016533037647604942,
      "q_g1": 0.9834669232368469,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.5252182483673096,
      "loss_g0": 5.442172050476074,
      "loss_g1": 2.4761805534362793,
      "q_g0": 0.016533348709344864,
      "q_g1": 0.9834666848182678,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.823153018951416,
      "loss_g0": 17.411977767944336,
      "loss_g1": 5.6283159255981445,
      "q_g0": 0.01653454825282097,
      "q_g1": 0.9834654331207275,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.733442306518555,
      "loss_g0": 9.733442306518555,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.282398223876953,
      "loss_g0": 15.460939407348633,
      "loss_g1": 4.654973030090332,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.722589492797852,
      "loss_g0": 8.718366622924805,
      "loss_g1": 8.973944664001465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522448509931564,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.571501731872559,
      "loss_g0": 0.0,
      "loss_g1": 7.571501731872559,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.2168364524841309,
      "loss_g0": 0.0,
      "loss_g1": 1.2168364524841309,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.242180824279785,
      "loss_g0": 0.0,
      "loss_g1": 2.242180824279785,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.715441703796387,
      "loss_g0": 4.928420066833496,
      "loss_g1": 5.728663444519043,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.614943027496338,
      "loss_g0": 10.566286087036133,
      "loss_g1": 2.481353759765625,
      "q_g0": 0.016523247584700584,
      "q_g1": 0.9834766983985901,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.7221735715866089,
      "loss_g0": 8.209684371948242,
      "loss_g1": 0.596371054649353,
      "q_g0": 0.016524024307727814,
      "q_g1": 0.9834760427474976,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.4083486795425415,
      "loss_g0": 4.514197826385498,
      "loss_g1": 1.3561642169952393,
      "q_g0": 0.01652434840798378,
      "q_g1": 0.9834756851196289,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.150872707366943,
      "loss_g0": 2.1322734355926514,
      "loss_g1": 7.23519229888916,
      "q_g0": 0.0165238194167614,
      "q_g1": 0.9834761619567871,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.0060704946517944,
      "loss_g0": 6.0695648193359375,
      "loss_g1": 0.9209936857223511,
      "q_g0": 0.016524342820048332,
      "q_g1": 0.9834756851196289,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.8191863298416138,
      "loss_g0": 0.21917425096035004,
      "loss_g1": 0.8292677402496338,
      "q_g0": 0.016524286940693855,
      "q_g1": 0.9834756851196289,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 25.98375701904297,
      "loss_g0": 51.380455017089844,
      "loss_g1": 25.556974411010742,
      "q_g0": 0.016526928171515465,
      "q_g1": 0.9834730625152588,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.017684936523438,
      "loss_g0": 14.470799446105957,
      "loss_g1": 16.04368019104004,
      "q_g0": 0.01652676612138748,
      "q_g1": 0.9834732413291931,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 38.12537384033203,
      "loss_g0": 16.88958740234375,
      "loss_g1": 38.482181549072266,
      "q_g0": 0.016524557024240494,
      "q_g1": 0.9834754467010498,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.150182723999023,
      "loss_g0": 18.557910919189453,
      "loss_g1": 15.092924118041992,
      "q_g0": 0.016524914652109146,
      "q_g1": 0.9834750890731812,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 29.438467025756836,
      "loss_g0": 11.536493301391602,
      "loss_g1": 29.73923110961914,
      "q_g0": 0.016523046419024467,
      "q_g1": 0.9834769368171692,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 37.56843948364258,
      "loss_g0": 3.3934032917022705,
      "loss_g1": 38.142578125,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 25.533599853515625,
      "loss_g0": 22.86490821838379,
      "loss_g1": 25.578433990478516,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 21.821022033691406,
      "loss_g0": 0.0,
      "loss_g1": 21.821022033691406,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.549970626831055,
      "loss_g0": 44.06980895996094,
      "loss_g1": 18.12123680114746,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.562851905822754,
      "loss_g0": 1.7539865970611572,
      "loss_g1": 11.727640151977539,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.033895492553711,
      "loss_g0": 9.506369590759277,
      "loss_g1": 14.109957695007324,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.298717498779297,
      "loss_g0": 0.0,
      "loss_g1": 10.298717498779297,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.975959777832031,
      "loss_g0": 20.934480667114258,
      "loss_g1": 9.79185676574707,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 50.04647445678711,
      "loss_g0": 8.76199722290039,
      "loss_g1": 50.740055084228516,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.363621711730957,
      "loss_g0": 27.976917266845703,
      "loss_g1": 11.084488868713379,
      "q_g0": 0.016524145379662514,
      "q_g1": 0.9834758043289185,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 43.09718704223633,
      "loss_g0": 11.797255516052246,
      "loss_g1": 43.623023986816406,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 26.608606338500977,
      "loss_g0": 22.349777221679688,
      "loss_g1": 26.68015480041504,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.81839942932129,
      "loss_g0": 20.51115608215332,
      "loss_g1": 16.75636100769043,
      "q_g0": 0.016522813588380814,
      "q_g1": 0.9834771752357483,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 36.447696685791016,
      "loss_g0": 9.777505874633789,
      "loss_g1": 36.895755767822266,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.901918411254883,
      "loss_g0": 5.958590984344482,
      "loss_g1": 19.119367599487305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.71462345123291,
      "loss_g0": 65.6600341796875,
      "loss_g1": 9.791216850280762,
      "q_g0": 0.01652812957763672,
      "q_g1": 0.9834718704223633,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.09943389892578,
      "loss_g0": 66.75312805175781,
      "loss_g1": 21.348772048950195,
      "q_g0": 0.0165327750146389,
      "q_g1": 0.9834672808647156,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.219940185546875,
      "loss_g0": 16.28011131286621,
      "loss_g1": 15.202116966247559,
      "q_g0": 0.016532890498638153,
      "q_g1": 0.983467161655426,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.816681861877441,
      "loss_g0": 17.788209915161133,
      "loss_g1": 6.632228374481201,
      "q_g0": 0.016534030437469482,
      "q_g1": 0.9834660291671753,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.607479095458984,
      "loss_g0": 12.779180526733398,
      "loss_g1": 14.638216018676758,
      "q_g0": 0.01653384417295456,
      "q_g1": 0.9834660887718201,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.117794036865234,
      "loss_g0": 13.3218412399292,
      "loss_g1": 10.063926696777344,
      "q_g0": 0.016534175723791122,
      "q_g1": 0.9834657907485962,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.628629684448242,
      "loss_g0": 1.1992698907852173,
      "loss_g1": 7.736716270446777,
      "q_g0": 0.016533510759472847,
      "q_g1": 0.9834664463996887,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.696140766143799,
      "loss_g0": 7.4329047203063965,
      "loss_g1": 6.6837544441223145,
      "q_g0": 0.016533588990569115,
      "q_g1": 0.9834664463996887,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.97337532043457,
      "loss_g0": 13.787063598632812,
      "loss_g1": 8.892447471618652,
      "q_g0": 0.016534091904759407,
      "q_g1": 0.9834659099578857,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.92947769165039,
      "loss_g0": 6.496158123016357,
      "loss_g1": 12.020819664001465,
      "q_g0": 0.016533533111214638,
      "q_g1": 0.9834664463996887,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.241855144500732,
      "loss_g0": 10.492424011230469,
      "loss_g1": 5.153582572937012,
      "q_g0": 0.016534077003598213,
      "q_g1": 0.9834659099578857,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 19.661846160888672,
      "loss_g0": 19.661846160888672,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.599448204040527,
      "loss_g0": 12.694350242614746,
      "loss_g1": 6.950510501861572,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.341686248779297,
      "loss_g0": 12.341686248779297,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.569003105163574,
      "loss_g0": 10.631800651550293,
      "loss_g1": 6.831023216247559,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.3420467376709,
      "loss_g0": 16.541152954101562,
      "loss_g1": 4.4904985427856445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.91503381729126,
      "loss_g0": 6.932114124298096,
      "loss_g1": 5.898358345031738,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 27.256940841674805,
      "loss_g0": 27.667558670043945,
      "loss_g1": 2.815293073654175,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.307643890380859,
      "loss_g0": 5.334892749786377,
      "loss_g1": 3.6856722831726074,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.427292823791504,
      "loss_g0": 7.453500747680664,
      "loss_g1": 5.86729621887207,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.440317153930664,
      "loss_g0": 9.5462007522583,
      "loss_g1": 3.1377434730529785,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.2436124086380005,
      "loss_g0": 1.1810053586959839,
      "loss_g1": 4.970128059387207,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522813588380814,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.026302337646484,
      "loss_g0": 13.163650512695312,
      "loss_g1": 4.8507843017578125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.976292133331299,
      "loss_g0": 2.9808647632598877,
      "loss_g1": 2.7041029930114746,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.009087085723877,
      "loss_g0": 5.8911027908325195,
      "loss_g1": 13.031646728515625,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523154452443123,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.956844329833984,
      "loss_g0": 10.079496383666992,
      "loss_g1": 2.6561059951782227,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 39.214683532714844,
      "loss_g0": 39.849700927734375,
      "loss_g1": 1.4160072803497314,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.4321421384811401,
      "loss_g0": 1.4361982345581055,
      "loss_g1": 1.1907072067260742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.3464958667755127,
      "loss_g0": 2.2690024375915527,
      "loss_g1": 6.959054946899414,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522906720638275,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.570707321166992,
      "loss_g0": 9.724193572998047,
      "loss_g1": 0.4346168339252472,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.26368522644043,
      "loss_g0": 9.048090934753418,
      "loss_g1": 22.095653533935547,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523750498890877,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.75306510925293,
      "loss_g0": 12.407416343688965,
      "loss_g1": 33.32303237915039,
      "q_g0": 0.98347407579422,
      "q_g1": 0.01652589812874794,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 62.94605255126953,
      "loss_g0": 63.636146545410156,
      "loss_g1": 21.86884117126465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 33.784217834472656,
      "loss_g0": 33.784217834472656,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 24.297239303588867,
      "loss_g0": 24.400291442871094,
      "loss_g1": 18.1632080078125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 41.83588790893555,
      "loss_g0": 42.110042572021484,
      "loss_g1": 25.517004013061523,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.82313346862793,
      "loss_g0": 11.680513381958008,
      "loss_g1": 20.311946868896484,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523301601409912,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.242278099060059,
      "loss_g0": 10.300537109375,
      "loss_g1": 6.774604797363281,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522938385605812,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.025395393371582,
      "loss_g0": 15.025395393371582,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.872039794921875,
      "loss_g0": 23.12277603149414,
      "loss_g1": 7.947268962860107,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.346237182617188,
      "loss_g0": 0.0,
      "loss_g1": 13.346237182617188,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.824243545532227,
      "loss_g0": 33.09697341918945,
      "loss_g1": 16.550861358642578,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 27.786434173583984,
      "loss_g0": 12.570295333862305,
      "loss_g1": 28.042064666748047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.61166763305664,
      "loss_g0": 0.0,
      "loss_g1": 16.61166763305664,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 37.61482620239258,
      "loss_g0": 35.7271842956543,
      "loss_g1": 37.64653778076172,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.680912017822266,
      "loss_g0": 15.680912017822266,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.938301086425781,
      "loss_g0": 0.0,
      "loss_g1": 10.938301086425781,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.4117045402526855,
      "loss_g0": 11.864871978759766,
      "loss_g1": 3.2696914672851562,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.622518539428711,
      "loss_g0": 16.44969367980957,
      "loss_g1": 5.440609931945801,
      "q_g0": 0.01652355119585991,
      "q_g1": 0.9834765195846558,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.713902235031128,
      "loss_g0": 5.17491340637207,
      "loss_g1": 2.672553539276123,
      "q_g0": 0.016523797065019608,
      "q_g1": 0.9834761619567871,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.322620391845703,
      "loss_g0": 12.604405403137207,
      "loss_g1": 11.301084518432617,
      "q_g0": 0.016523919999599457,
      "q_g1": 0.9834760427474976,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.586116790771484,
      "loss_g0": 8.919431686401367,
      "loss_g1": 11.630919456481934,
      "q_g0": 0.01652364432811737,
      "q_g1": 0.9834764003753662,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.984618663787842,
      "loss_g0": 19.970252990722656,
      "loss_g1": 7.783229827880859,
      "q_g0": 0.01652488298714161,
      "q_g1": 0.9834750890731812,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.839152336120605,
      "loss_g0": 20.767366409301758,
      "loss_g1": 8.638711929321289,
      "q_g0": 0.016526123508810997,
      "q_g1": 0.9834738373756409,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.599821090698242,
      "loss_g0": 10.509326934814453,
      "loss_g1": 8.567733764648438,
      "q_g0": 0.016526324674487114,
      "q_g1": 0.9834737181663513,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.185520887374878,
      "loss_g0": 5.831382751464844,
      "loss_g1": 1.1074492931365967,
      "q_g0": 0.016526803374290466,
      "q_g1": 0.9834731221199036,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.58608055114746,
      "loss_g0": 39.449974060058594,
      "loss_g1": 22.30265998840332,
      "q_g0": 0.016528556123375893,
      "q_g1": 0.9834713935852051,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.789143085479736,
      "loss_g0": 0.0,
      "loss_g1": 6.789143085479736,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 21.72886848449707,
      "loss_g0": 4.407679080963135,
      "loss_g1": 22.01986312866211,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.421451568603516,
      "loss_g0": 4.805474758148193,
      "loss_g1": 15.599800109863281,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.693996906280518,
      "loss_g0": 7.0902018547058105,
      "loss_g1": 4.653739929199219,
      "q_g0": 0.01652267388999462,
      "q_g1": 0.9834772944450378,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.9692378044128418,
      "loss_g0": 2.7967002391815186,
      "loss_g1": 1.955336093902588,
      "q_g0": 0.016522759571671486,
      "q_g1": 0.9834771752357483,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.470926284790039,
      "loss_g0": 4.160469055175781,
      "loss_g1": 5.492941856384277,
      "q_g0": 0.01652262918651104,
      "q_g1": 0.9834774136543274,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.774583339691162,
      "loss_g0": 0.0,
      "loss_g1": 3.774583339691162,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.043391227722168,
      "loss_g0": 11.551374435424805,
      "loss_g1": 4.934056758880615,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.260997772216797,
      "loss_g0": 16.26087188720703,
      "loss_g1": 2.0257787704467773,
      "q_g0": 0.016523877158761024,
      "q_g1": 0.9834761619567871,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.648089408874512,
      "loss_g0": 12.714406967163086,
      "loss_g1": 5.529359340667725,
      "q_g0": 0.016524620354175568,
      "q_g1": 0.9834753274917603,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.375285148620605,
      "loss_g0": 14.375285148620605,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.9828054904937744,
      "loss_g0": 4.006978511810303,
      "loss_g1": 2.543938159942627,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.310415267944336,
      "loss_g0": 4.238247394561768,
      "loss_g1": 8.606013298034668,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652287319302559,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.04289436340332,
      "loss_g0": 14.979090690612793,
      "loss_g1": 18.840564727783203,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523271799087524,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.8941429257392883,
      "loss_g0": 0.8465296626091003,
      "loss_g1": 3.7280662059783936,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523564234375954,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.3225836753845215,
      "loss_g0": 5.072004318237305,
      "loss_g1": 20.235519409179688,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525115817785263,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.8069329261779785,
      "loss_g0": 6.864834785461426,
      "loss_g1": 3.360889196395874,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652476005256176,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.531357765197754,
      "loss_g0": 5.5831708908081055,
      "loss_g1": 2.447622776031494,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652444154024124,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.07544994354248,
      "loss_g0": 11.248265266418457,
      "loss_g1": 0.7894118428230286,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523364931344986,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.77673864364624,
      "loss_g0": 7.83089017868042,
      "loss_g1": 4.55354118347168,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523031517863274,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.635848999023438,
      "loss_g0": 8.712116241455078,
      "loss_g1": 4.0962018966674805,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652255840599537,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.338582992553711,
      "loss_g0": 1.3232814073562622,
      "loss_g1": 2.2493717670440674,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.263648986816406,
      "loss_g0": 5.263648986816406,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 36.03345489501953,
      "loss_g0": 36.169498443603516,
      "loss_g1": 27.935476303100586,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 25.490463256835938,
      "loss_g0": 25.672348022460938,
      "loss_g1": 14.664007186889648,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.1456937789917,
      "loss_g0": 10.738604545593262,
      "loss_g1": 35.37346267700195,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524946317076683,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 36.88737869262695,
      "loss_g0": 37.324825286865234,
      "loss_g1": 10.848878860473633,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.441770553588867,
      "loss_g0": 20.423601150512695,
      "loss_g1": 21.52324104309082,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.550369262695312,
      "loss_g0": 15.401371002197266,
      "loss_g1": 24.41878890991211,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.0165234562009573,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.52757453918457,
      "loss_g0": 18.557723999023438,
      "loss_g1": 16.733083724975586,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523271799087524,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 50.62762451171875,
      "loss_g0": 51.260684967041016,
      "loss_g1": 12.945552825927734,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.28765869140625,
      "loss_g0": 7.093667984008789,
      "loss_g1": 18.833852767944336,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652361825108528,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.353074073791504,
      "loss_g0": 10.35412883758545,
      "loss_g1": 10.290265083312988,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523612663149834,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.917470932006836,
      "loss_g0": 14.963699340820312,
      "loss_g1": 12.165925025939941,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523325815796852,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.23717498779297,
      "loss_g0": 22.54677963256836,
      "loss_g1": 3.808237314224243,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.22993278503418,
      "loss_g0": 17.385101318359375,
      "loss_g1": 7.99367618560791,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.316287994384766,
      "loss_g0": 14.291990280151367,
      "loss_g1": 15.762542724609375,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522573307156563,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 27.45611572265625,
      "loss_g0": 27.58346176147461,
      "loss_g1": 19.876018524169922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.881176948547363,
      "loss_g0": 13.815633773803711,
      "loss_g1": 17.782432556152344,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522830352187157,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.100622177124023,
      "loss_g0": 11.100622177124023,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.718125820159912,
      "loss_g0": 4.596896648406982,
      "loss_g1": 11.934127807617188,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.988569259643555,
      "loss_g0": 5.7364630699157715,
      "loss_g1": 20.993431091308594,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.01652398519217968,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.152484893798828,
      "loss_g0": 10.201234817504883,
      "loss_g1": 7.250919342041016,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523681581020355,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 67.53472900390625,
      "loss_g0": 68.55957794189453,
      "loss_g1": 6.531786918640137,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.98879337310791,
      "loss_g0": 16.229244232177734,
      "loss_g1": 1.6762278079986572,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.28974151611328,
      "loss_g0": 18.523406982421875,
      "loss_g1": 4.381119251251221,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.94505500793457,
      "loss_g0": 13.05184555053711,
      "loss_g1": 6.588438987731934,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.811864852905273,
      "loss_g0": 20.979032516479492,
      "loss_g1": 10.86138916015625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.624894142150879,
      "loss_g0": 13.76650619506836,
      "loss_g1": 5.195615768432617,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.443222999572754,
      "loss_g0": 11.619253158569336,
      "loss_g1": 0.965268611907959,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.078774452209473,
      "loss_g0": 14.186051368713379,
      "loss_g1": 7.693215370178223,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.135651588439941,
      "loss_g0": 13.235637664794922,
      "loss_g1": 7.184083938598633,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.57080364227295,
      "loss_g0": 14.691384315490723,
      "loss_g1": 7.393326759338379,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.472429275512695,
      "loss_g0": 14.675878524780273,
      "loss_g1": 2.3623852729797363,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.375300884246826,
      "loss_g0": 0.0,
      "loss_g1": 4.375300884246826,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.473688125610352,
      "loss_g0": 7.852826118469238,
      "loss_g1": 4.4169182777404785,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.589625358581543,
      "loss_g0": 11.787692070007324,
      "loss_g1": 10.569498062133789,
      "q_g0": 0.016522543504834175,
      "q_g1": 0.9834774136543274,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.923981189727783,
      "loss_g0": 1.628182053565979,
      "loss_g1": 4.9793500900268555,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.2343220710754395,
      "loss_g0": 16.83481788635254,
      "loss_g1": 4.022616386413574,
      "q_g0": 0.016523735597729683,
      "q_g1": 0.9834762811660767,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.020849704742432,
      "loss_g0": 15.291929244995117,
      "loss_g1": 4.848270416259766,
      "q_g0": 0.01652480475604534,
      "q_g1": 0.9834752082824707,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.3048715591430664,
      "loss_g0": 5.766018867492676,
      "loss_g1": 3.2635176181793213,
      "q_g0": 0.01652505435049534,
      "q_g1": 0.9834749698638916,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 29.462026596069336,
      "loss_g0": 29.462026596069336,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.8478279113769531,
      "loss_g0": 0.8083963990211487,
      "loss_g1": 3.194939374923706,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.410518169403076,
      "loss_g0": 3.3998708724975586,
      "loss_g1": 4.044279098510742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522489488124847,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.4442138671875,
      "loss_g0": 17.70020866394043,
      "loss_g1": 2.2063839435577393,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.697765350341797,
      "loss_g0": 3.6852035522460938,
      "loss_g1": 4.4454874992370605,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652250438928604,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.59460711479187,
      "loss_g0": 3.601536750793457,
      "loss_g1": 3.1821324825286865,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652245782315731,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.723382949829102,
      "loss_g0": 9.793266296386719,
      "loss_g1": 5.563612937927246,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.274833679199219,
      "loss_g0": 6.3754777908325195,
      "loss_g1": 0.2840885818004608,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.979158878326416,
      "loss_g0": 7.924563884735107,
      "loss_g1": 11.228803634643555,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522759571671486,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.3967368602752686,
      "loss_g0": 2.435546398162842,
      "loss_g1": 0.08664800971746445,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522521153092384,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.705772876739502,
      "loss_g0": 5.705772876739502,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.090862274169922,
      "loss_g0": 1.9811320304870605,
      "loss_g1": 8.622417449951172,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.54349136352539,
      "loss_g0": 0.0,
      "loss_g1": 22.54349136352539,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.178534507751465,
      "loss_g0": 16.81945037841797,
      "loss_g1": 13.117366790771484,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.968286514282227,
      "loss_g0": 19.71727752685547,
      "loss_g1": 8.787692070007324,
      "q_g0": 0.016523543745279312,
      "q_g1": 0.9834765195846558,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.703205108642578,
      "loss_g0": 0.0,
      "loss_g1": 20.703205108642578,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 25.369857788085938,
      "loss_g0": 22.442428588867188,
      "loss_g1": 25.419036865234375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.977170944213867,
      "loss_g0": 23.615596771240234,
      "loss_g1": 7.714419364929199,
      "q_g0": 0.0165240541100502,
      "q_g1": 0.983475923538208,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.811758518218994,
      "loss_g0": 32.059974670410156,
      "loss_g1": 7.404284477233887,
      "q_g0": 0.016526570543646812,
      "q_g1": 0.9834733605384827,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 37.471885681152344,
      "loss_g0": 33.244407653808594,
      "loss_g1": 37.54292297363281,
      "q_g0": 0.016526129096746445,
      "q_g1": 0.9834738373756409,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 27.063356399536133,
      "loss_g0": 48.68365478515625,
      "loss_g1": 26.700002670288086,
      "q_g0": 0.01652837172150612,
      "q_g1": 0.9834716320037842,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.436129570007324,
      "loss_g0": 18.762834548950195,
      "loss_g1": 9.279373168945312,
      "q_g0": 0.016529347747564316,
      "q_g1": 0.983470618724823,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.492290496826172,
      "loss_g0": 0.0,
      "loss_g1": 16.492290496826172,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 35.18440628051758,
      "loss_g0": 16.407611846923828,
      "loss_g1": 35.499855041503906,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.520401954650879,
      "loss_g0": 16.725696563720703,
      "loss_g1": 9.399347305297852,
      "q_g0": 0.016523178666830063,
      "q_g1": 0.9834768176078796,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.26827049255371,
      "loss_g0": 2.646811008453369,
      "loss_g1": 17.5139102935791,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.463282585144043,
      "loss_g0": 28.17477798461914,
      "loss_g1": 9.14889144897461,
      "q_g0": 0.01652437075972557,
      "q_g1": 0.9834755659103394,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.141192436218262,
      "loss_g0": 19.51797866821289,
      "loss_g1": 11.00043773651123,
      "q_g0": 0.016525238752365112,
      "q_g1": 0.9834747910499573,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.976892471313477,
      "loss_g0": 34.460655212402344,
      "loss_g1": 10.582237243652344,
      "q_g0": 0.016527680680155754,
      "q_g1": 0.9834723472595215,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.114654541015625,
      "loss_g0": 32.086387634277344,
      "loss_g1": 10.762168884277344,
      "q_g0": 0.016529859974980354,
      "q_g1": 0.9834701418876648,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 21.694198608398438,
      "loss_g0": 19.471759796142578,
      "loss_g1": 21.731552124023438,
      "q_g0": 0.016529632732272148,
      "q_g1": 0.9834703803062439,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.42018699645996,
      "loss_g0": 6.587164878845215,
      "loss_g1": 16.5854434967041,
      "q_g0": 0.01652861200273037,
      "q_g1": 0.9834713935852051,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.0310170650482178,
      "loss_g0": 25.8249568939209,
      "loss_g1": 1.6310659646987915,
      "q_g0": 0.0165310837328434,
      "q_g1": 0.9834688901901245,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.24413537979126,
      "loss_g0": 8.638402938842773,
      "loss_g1": 5.187080383300781,
      "q_g0": 0.016531439498066902,
      "q_g1": 0.9834685325622559,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.053508758544922,
      "loss_g0": 16.550994873046875,
      "loss_g1": 7.9106645584106445,
      "q_g0": 0.01653231866657734,
      "q_g1": 0.9834677577018738,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.012848854064941,
      "loss_g0": 2.9472498893737793,
      "loss_g1": 15.215658187866211,
      "q_g0": 0.016531066969037056,
      "q_g1": 0.9834688901901245,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.144659519195557,
      "loss_g0": 20.218198776245117,
      "loss_g1": 5.908076763153076,
      "q_g0": 0.0165325328707695,
      "q_g1": 0.9834675192832947,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.057845115661621,
      "loss_g0": 26.03270721435547,
      "loss_g1": 12.839714050292969,
      "q_g0": 0.016533883288502693,
      "q_g1": 0.9834660887718201,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.05841064453125,
      "loss_g0": 5.004824638366699,
      "loss_g1": 10.143367767333984,
      "q_g0": 0.01653335615992546,
      "q_g1": 0.9834666848182678,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.518919467926025,
      "loss_g0": 6.960268020629883,
      "loss_g1": 6.511499404907227,
      "q_g0": 0.016533400863409042,
      "q_g1": 0.9834665656089783,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.846890926361084,
      "loss_g0": 3.9767796993255615,
      "loss_g1": 5.878329753875732,
      "q_g0": 0.016533199697732925,
      "q_g1": 0.9834668040275574,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.902148962020874,
      "loss_g0": 12.2648344039917,
      "loss_g1": 2.7447423934936523,
      "q_g0": 0.016534170135855675,
      "q_g1": 0.9834659099578857,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.03868293762207,
      "loss_g0": 7.745198726654053,
      "loss_g1": 16.1781063079834,
      "q_g0": 0.01653330959379673,
      "q_g1": 0.9834666848182678,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.805467128753662,
      "loss_g0": 12.358665466308594,
      "loss_g1": 5.6952948570251465,
      "q_g0": 0.016533993184566498,
      "q_g1": 0.9834660291671753,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.2505428791046143,
      "loss_g0": 2.385586738586426,
      "loss_g1": 2.24827241897583,
      "q_g0": 0.016533998772501945,
      "q_g1": 0.9834660291671753,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.430963039398193,
      "loss_g0": 14.484503746032715,
      "loss_g1": 5.278746128082275,
      "q_g0": 0.016534939408302307,
      "q_g1": 0.9834650754928589,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.257587909698486,
      "loss_g0": 9.160385131835938,
      "loss_g1": 5.19196891784668,
      "q_g0": 0.016535349190235138,
      "q_g1": 0.9834645986557007,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.911457061767578,
      "loss_g0": 11.923141479492188,
      "loss_g1": 10.89444637298584,
      "q_g0": 0.016535449773073196,
      "q_g1": 0.9834644794464111,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.426568508148193,
      "loss_g0": 8.55451488494873,
      "loss_g1": 6.390789985656738,
      "q_g0": 0.016535673290491104,
      "q_g1": 0.9834643006324768,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.86946964263916,
      "loss_g0": 27.22081184387207,
      "loss_g1": 3.4767885208129883,
      "q_g0": 0.016538100317120552,
      "q_g1": 0.983461856842041,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.98153829574585,
      "loss_g0": 7.98153829574585,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.165804386138916,
      "loss_g0": 4.179819583892822,
      "loss_g1": 3.331557035446167,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 41.45440673828125,
      "loss_g0": 42.07328796386719,
      "loss_g1": 4.616156101226807,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.4755747318267822,
      "loss_g0": 2.462831735610962,
      "loss_g1": 3.2340762615203857,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652250438928604,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.64505386352539,
      "loss_g0": 18.92603302001953,
      "loss_g1": 1.9201632738113403,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.3883633613586426,
      "loss_g0": 3.3883633613586426,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.143212795257568,
      "loss_g0": 7.231603145599365,
      "loss_g1": 1.881879448890686,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.4078845977783203,
      "loss_g0": 2.379460573196411,
      "loss_g1": 4.099765777587891,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522597521543503,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.381599426269531,
      "loss_g0": 8.497310638427734,
      "loss_g1": 1.4940465688705444,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.880918025970459,
      "loss_g0": 7.950719833374023,
      "loss_g1": 3.726017951965332,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.0919636487960815,
      "loss_g0": 0.9154407382011414,
      "loss_g1": 11.598572731018066,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523512080311775,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.0007824897766113,
      "loss_g0": 1.9055094718933105,
      "loss_g1": 7.671201229095459,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524100676178932,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.459765911102295,
      "loss_g0": 7.165390491485596,
      "loss_g1": 24.978361129760742,
      "q_g0": 0.98347407579422,
      "q_g1": 0.016525914892554283,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 29.32762908935547,
      "loss_g0": 28.995264053344727,
      "loss_g1": 49.10455322265625,
      "q_g0": 0.9834719896316528,
      "q_g1": 0.016527967527508736,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.148149490356445,
      "loss_g0": 22.186767578125,
      "loss_g1": 19.850217819213867,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.016527734696865082,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.269111633300781,
      "loss_g0": 14.304055213928223,
      "loss_g1": 12.189823150634766,
      "q_g0": 0.983472466468811,
      "q_g1": 0.016527516767382622,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 31.83817481994629,
      "loss_g0": 30.186798095703125,
      "loss_g1": 130.04185485839844,
      "q_g0": 0.9834623336791992,
      "q_g1": 0.01653772033751011,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.541196823120117,
      "loss_g0": 22.6632137298584,
      "loss_g1": 15.284743309020996,
      "q_g0": 0.9834629893302917,
      "q_g1": 0.01653696782886982,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 33.266971588134766,
      "loss_g0": 33.36742401123047,
      "loss_g1": 27.292680740356445,
      "q_g0": 0.983463704586029,
      "q_g1": 0.01653634011745453,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.380364418029785,
      "loss_g0": 10.158464431762695,
      "loss_g1": 23.576265335083008,
      "q_g0": 0.9834623336791992,
      "q_g1": 0.01653771474957466,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.285140991210938,
      "loss_g0": 15.356767654418945,
      "loss_g1": 11.025503158569336,
      "q_g0": 0.9834627509117126,
      "q_g1": 0.016537263989448547,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 24.059560775756836,
      "loss_g0": 24.059560775756836,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 36.17572021484375,
      "loss_g0": 36.53876876831055,
      "loss_g1": 14.565746307373047,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.559822082519531,
      "loss_g0": 9.65808391571045,
      "loss_g1": 3.710857391357422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.826576232910156,
      "loss_g0": 15.026424407958984,
      "loss_g1": 2.9308669567108154,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.540606498718262,
      "loss_g0": 14.263662338256836,
      "loss_g1": 31.02358627319336,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652413234114647,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 23.83823013305664,
      "loss_g0": 24.077211380004883,
      "loss_g1": 9.613363265991211,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.261587142944336,
      "loss_g0": 16.261587142944336,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.490453720092773,
      "loss_g0": 14.633227348327637,
      "loss_g1": 5.992029666900635,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.168867111206055,
      "loss_g0": 16.309398651123047,
      "loss_g1": 7.803893089294434,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.219874382019043,
      "loss_g0": 2.1058003902435303,
      "loss_g1": 9.009685516357422,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523132100701332,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 21.953128814697266,
      "loss_g0": 22.030105590820312,
      "loss_g1": 17.371246337890625,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.502575874328613,
      "loss_g0": 11.502575874328613,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.669191360473633,
      "loss_g0": 9.247623443603516,
      "loss_g1": 34.76251220703125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.969008445739746,
      "loss_g0": 4.967266082763672,
      "loss_g1": 5.072704315185547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.048236846923828,
      "loss_g0": 18.235015869140625,
      "loss_g1": 6.930419445037842,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.4380598068237305,
      "loss_g0": 0.0,
      "loss_g1": 6.4380598068237305,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.9534666538238525,
      "loss_g0": 7.637158393859863,
      "loss_g1": 2.8747804164886475,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.666775703430176,
      "loss_g0": 16.399999618530273,
      "loss_g1": 12.604056358337402,
      "q_g0": 0.016522806137800217,
      "q_g1": 0.9834771752357483,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.060027122497559,
      "loss_g0": 19.661014556884766,
      "loss_g1": 9.89871597290039,
      "q_g0": 0.016523806378245354,
      "q_g1": 0.9834761619567871,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.7121000289917,
      "loss_g0": 8.529863357543945,
      "loss_g1": 12.782364845275879,
      "q_g0": 0.01652337983250618,
      "q_g1": 0.9834766387939453,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.353514194488525,
      "loss_g0": 9.941917419433594,
      "loss_g1": 4.259620189666748,
      "q_g0": 0.016523951664566994,
      "q_g1": 0.9834760427474976,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.388559818267822,
      "loss_g0": 7.388559818267822,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.2116923332214355,
      "loss_g0": 7.191043853759766,
      "loss_g1": 8.440759658813477,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.241610050201416,
      "loss_g0": 7.107566833496094,
      "loss_g1": 15.219970703125,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652325503528118,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.375229835510254,
      "loss_g0": 12.414668083190918,
      "loss_g1": 10.027811050415039,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652301475405693,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.947830200195312,
      "loss_g0": 0.0,
      "loss_g1": 8.947830200195312,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.748196601867676,
      "loss_g0": 4.642358303070068,
      "loss_g1": 4.749974727630615,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.328173637390137,
      "loss_g0": 11.823309898376465,
      "loss_g1": 6.235852241516113,
      "q_g0": 0.01652299053966999,
      "q_g1": 0.9834769368171692,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.039483070373535,
      "loss_g0": 15.039483070373535,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.032380104064941,
      "loss_g0": 9.136946678161621,
      "loss_g1": 2.808211088180542,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.264610290527344,
      "loss_g0": 8.216629028320312,
      "loss_g1": 11.120575904846191,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.0165227223187685,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.409833908081055,
      "loss_g0": 8.53108024597168,
      "loss_g1": 1.192774772644043,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.409207344055176,
      "loss_g0": 12.397205352783203,
      "loss_g1": 13.123544692993164,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.915151596069336,
      "loss_g0": 8.928461074829102,
      "loss_g1": 8.122937202453613,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.304370880126953,
      "loss_g0": 11.304370880126953,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.371135711669922,
      "loss_g0": 5.413871765136719,
      "loss_g1": 2.8273375034332275,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.66892409324646,
      "loss_g0": 3.6780683994293213,
      "loss_g1": 3.1246166229248047,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.6786733269691467,
      "loss_g0": 0.6568071842193604,
      "loss_g1": 1.9802191257476807,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652255840599537,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 38.561126708984375,
      "loss_g0": 39.123863220214844,
      "loss_g1": 5.064723968505859,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.8780955076217651,
      "loss_g0": 0.8854914903640747,
      "loss_g1": 0.43785569071769714,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 1.3609528541564941,
      "loss_g0": 1.3661388158798218,
      "loss_g1": 1.0522677898406982,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 56.743656158447266,
      "loss_g0": 57.17440414428711,
      "loss_g1": 31.10394859313965,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.9977388381958,
      "loss_g0": 10.80988597869873,
      "loss_g1": 22.178680419921875,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523588448762894,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 23.636266708374023,
      "loss_g0": 23.7255916595459,
      "loss_g1": 18.31950569152832,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.01652303896844387,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 31.596450805664062,
      "loss_g0": 31.390972137451172,
      "loss_g1": 43.8259391784668,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524309292435646,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 32.707305908203125,
      "loss_g0": 33.01276779174805,
      "loss_g1": 14.525063514709473,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 29.631938934326172,
      "loss_g0": 29.490875244140625,
      "loss_g1": 38.02820587158203,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523301601409912,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.678833961486816,
      "loss_g0": 14.511894226074219,
      "loss_g1": 24.61456871032715,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524342820048332,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.77798843383789,
      "loss_g0": 13.960063934326172,
      "loss_g1": 63.44308090209961,
      "q_g0": 0.983470618724823,
      "q_g1": 0.016529399901628494,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.264575958251953,
      "loss_g0": 18.468856811523438,
      "loss_g1": 6.109236717224121,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.016528138890862465,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 34.630638122558594,
      "loss_g0": 34.89003372192383,
      "loss_g1": 19.19426155090332,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.01652652584016323,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.669293403625488,
      "loss_g0": 10.475936889648438,
      "loss_g1": 22.17487144470215,
      "q_g0": 0.9834722280502319,
      "q_g1": 0.016527727246284485,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 19.96529197692871,
      "loss_g0": 20.054386138916016,
      "loss_g1": 14.66358470916748,
      "q_g0": 0.9834728240966797,
      "q_g1": 0.016527168452739716,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 30.231996536254883,
      "loss_g0": 30.31304168701172,
      "loss_g1": 25.409088134765625,
      "q_g0": 0.9834733605384827,
      "q_g1": 0.016526665538549423,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.183477401733398,
      "loss_g0": 15.325393676757812,
      "loss_g1": 6.737852096557617,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.01652578078210354,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.52392315864563,
      "loss_g0": 3.371915340423584,
      "loss_g1": 12.569602012634277,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.01652671955525875,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 20.18410873413086,
      "loss_g0": 20.16492462158203,
      "loss_g1": 21.325658798217773,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.016526836901903152,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.65976333618164,
      "loss_g0": 9.731616973876953,
      "loss_g1": 5.383803367614746,
      "q_g0": 0.9834735989570618,
      "q_g1": 0.016526393592357635,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.290376663208008,
      "loss_g0": 15.22904109954834,
      "loss_g1": 18.94031524658203,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.01652677357196808,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.9341402053833,
      "loss_g0": 13.009958267211914,
      "loss_g1": 8.422186851501465,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.016526302322745323,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.932682991027832,
      "loss_g0": 12.974506378173828,
      "loss_g1": 10.44378662109375,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.01652604527771473,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.235817909240723,
      "loss_g0": 11.353697776794434,
      "loss_g1": 4.220465660095215,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.01652531512081623,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 23.278223037719727,
      "loss_g0": 23.548110961914062,
      "loss_g1": 7.214591979980469,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652364432811737,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 32.510826110839844,
      "loss_g0": 32.510826110839844,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.272473335266113,
      "loss_g0": 14.181097030639648,
      "loss_g1": 19.71154022216797,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.477845191955566,
      "loss_g0": 11.596322059631348,
      "loss_g1": 4.425637722015381,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.500828742980957,
      "loss_g0": 13.516529083251953,
      "loss_g1": 12.566267967224121,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.793570041656494,
      "loss_g0": 2.675522804260254,
      "loss_g1": 9.819877624511719,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523154452443123,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.3493645191192627,
      "loss_g0": 2.1690499782562256,
      "loss_g1": 13.081148147583008,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524270176887512,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.644268035888672,
      "loss_g0": 17.864957809448242,
      "loss_g1": 4.508432388305664,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522899270057678,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 26.653404235839844,
      "loss_g0": 26.791934967041016,
      "loss_g1": 18.40746307373047,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.2350575923919678,
      "loss_g0": 3.0828847885131836,
      "loss_g1": 12.292434692382812,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523364931344986,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.249539852142334,
      "loss_g0": 7.327629089355469,
      "loss_g1": 2.601524591445923,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.396030902862549,
      "loss_g0": 4.372512340545654,
      "loss_g1": 5.795882701873779,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523022204637527,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.233469009399414,
      "loss_g0": 12.35146427154541,
      "loss_g1": 5.209933757781982,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.914908409118652,
      "loss_g0": 8.972177505493164,
      "loss_g1": 5.506048202514648,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.328510284423828,
      "loss_g0": 2.294363498687744,
      "loss_g1": 4.361021041870117,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 24.173009872436523,
      "loss_g0": 24.347055435180664,
      "loss_g1": 13.813068389892578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.055782318115234,
      "loss_g0": 14.143647193908691,
      "loss_g1": 8.825743675231934,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.408923864364624,
      "loss_g0": 3.3192131519317627,
      "loss_g1": 8.748664855957031,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522977501153946,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 19.327085494995117,
      "loss_g0": 19.578489303588867,
      "loss_g1": 4.362617015838623,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.5145182609558105,
      "loss_g0": 3.2042975425720215,
      "loss_g1": 21.97785186767578,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.016524342820048332,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.981012344360352,
      "loss_g0": 9.081567764282227,
      "loss_g1": 2.9960458278656006,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523728147149086,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.135457038879395,
      "loss_g0": 8.121624946594238,
      "loss_g1": 8.958745956420898,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652381382882595,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.053680419921875,
      "loss_g0": 7.104919910430908,
      "loss_g1": 4.003879547119141,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652350276708603,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.13779354095459,
      "loss_g0": 4.199911117553711,
      "loss_g1": 0.4404846429824829,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652311533689499,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.9155187606811523,
      "loss_g0": 3.900087594985962,
      "loss_g1": 4.834010601043701,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523215919733047,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.944408416748047,
      "loss_g0": 12.109905242919922,
      "loss_g1": 2.0933609008789062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.355480194091797,
      "loss_g0": 5.392262935638428,
      "loss_g1": 3.1660385131835938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.086257934570312,
      "loss_g0": 8.170980453491211,
      "loss_g1": 3.0432071685791016,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.3037643432617188,
      "loss_g0": 2.273292303085327,
      "loss_g1": 4.117555618286133,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522612422704697,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 30.907617568969727,
      "loss_g0": 31.17870330810547,
      "loss_g1": 14.771553039550781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.511770248413086,
      "loss_g0": 22.33228874206543,
      "loss_g1": 33.19437789916992,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523534432053566,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.46004295349121,
      "loss_g0": 16.328140258789062,
      "loss_g1": 24.310413360595703,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652434840798378,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 24.44059944152832,
      "loss_g0": 24.485368728637695,
      "loss_g1": 21.776023864746094,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524069011211395,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 29.39170265197754,
      "loss_g0": 29.80127716064453,
      "loss_g1": 5.012283802032471,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 27.47588539123535,
      "loss_g0": 27.61874771118164,
      "loss_g1": 18.972118377685547,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.218161582946777,
      "loss_g0": 13.199810028076172,
      "loss_g1": 14.310457229614258,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652253419160843,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.989782333374023,
      "loss_g0": 11.989782333374023,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.158632278442383,
      "loss_g0": 16.10982894897461,
      "loss_g1": 19.063600540161133,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 26.23981475830078,
      "loss_g0": 26.540611267089844,
      "loss_g1": 8.335262298583984,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 31.586097717285156,
      "loss_g0": 31.75984001159668,
      "loss_g1": 21.24431610107422,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 28.903783798217773,
      "loss_g0": 29.2000789642334,
      "loss_g1": 11.267173767089844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 24.436203002929688,
      "loss_g0": 24.606210708618164,
      "loss_g1": 14.31658935546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.237607955932617,
      "loss_g0": 17.2503719329834,
      "loss_g1": 16.477746963500977,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.8895263671875,
      "loss_g0": 22.8895263671875,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.249309539794922,
      "loss_g0": 5.093695640563965,
      "loss_g1": 14.512052536010742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 29.617109298706055,
      "loss_g0": 29.89997100830078,
      "loss_g1": 12.780067443847656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 22.279754638671875,
      "loss_g0": 22.47152328491211,
      "loss_g1": 10.86502742767334,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.164892196655273,
      "loss_g0": 8.048107147216797,
      "loss_g1": 15.116069793701172,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652313955128193,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.977062225341797,
      "loss_g0": 4.908762454986572,
      "loss_g1": 9.042247772216797,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523558646440506,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.679757118225098,
      "loss_g0": 13.761519432067871,
      "loss_g1": 8.813175201416016,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523046419024467,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.332650184631348,
      "loss_g0": 9.11148452758789,
      "loss_g1": 22.495656967163086,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.01652441918849945,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.03354549407959,
      "loss_g0": 0.0,
      "loss_g1": 11.03354549407959,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 15.520931243896484,
      "loss_g0": 14.55406665802002,
      "loss_g1": 15.537175178527832,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.942366361618042,
      "loss_g0": 15.586149215698242,
      "loss_g1": 0.696327805519104,
      "q_g0": 0.016523944213986397,
      "q_g1": 0.9834760427474976,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.691659927368164,
      "loss_g0": 10.327685356140137,
      "loss_g1": 2.563356399536133,
      "q_g0": 0.01652473583817482,
      "q_g1": 0.9834752082824707,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.554542064666748,
      "loss_g0": 20.97983169555664,
      "loss_g1": 6.312139987945557,
      "q_g0": 0.01652623899281025,
      "q_g1": 0.9834737181663513,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 26.39594078063965,
      "loss_g0": 18.832195281982422,
      "loss_g1": 26.523035049438477,
      "q_g0": 0.016525447368621826,
      "q_g1": 0.9834745526313782,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.344423294067383,
      "loss_g0": 10.344423294067383,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.378259658813477,
      "loss_g0": 7.876601219177246,
      "loss_g1": 38.238853454589844,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.175018310546875,
      "loss_g0": 13.157690048217773,
      "loss_g1": 14.206488609313965,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522526741027832,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 13.983370780944824,
      "loss_g0": 14.03469467163086,
      "loss_g1": 10.928342819213867,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 17.782085418701172,
      "loss_g0": 17.992284774780273,
      "loss_g1": 5.2701873779296875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.9780426025390625,
      "loss_g0": 3.7590184211730957,
      "loss_g1": 17.014095306396484,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523776575922966,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 19.78550148010254,
      "loss_g0": 20.01308250427246,
      "loss_g1": 6.238900184631348,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.599691867828369,
      "loss_g0": 0.0,
      "loss_g1": 5.599691867828369,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.066293716430664,
      "loss_g0": 7.676170825958252,
      "loss_g1": 8.072847366333008,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.525711536407471,
      "loss_g0": 7.4773077964782715,
      "loss_g1": 4.476123332977295,
      "q_g0": 0.016522737219929695,
      "q_g1": 0.9834772944450378,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.147790431976318,
      "loss_g0": 7.849487781524658,
      "loss_g1": 4.085598945617676,
      "q_g0": 0.016523122787475586,
      "q_g1": 0.9834768176078796,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.773979187011719,
      "loss_g0": 9.505266189575195,
      "loss_g1": 5.711289405822754,
      "q_g0": 0.016523512080311775,
      "q_g1": 0.9834765195846558,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.609106540679932,
      "loss_g0": 3.9459259510040283,
      "loss_g1": 7.670650959014893,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 4.712158679962158,
      "loss_g0": 4.712158679962158,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.36515736579895,
      "loss_g0": 2.3720245361328125,
      "loss_g1": 1.9563965797424316,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.119454383850098,
      "loss_g0": 11.259164810180664,
      "loss_g1": 2.8033275604248047,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.339268922805786,
      "loss_g0": 2.339268922805786,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 8.547780990600586,
      "loss_g0": 8.547780990600586,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 2.1083054542541504,
      "loss_g0": 2.118645191192627,
      "loss_g1": 1.4928407669067383,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.855281829833984,
      "loss_g0": 14.971111297607422,
      "loss_g1": 7.960689067840576,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 25.603511810302734,
      "loss_g0": 25.603511810302734,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.094596862792969,
      "loss_g0": 0.0,
      "loss_g1": 6.094596862792969,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 16.90241241455078,
      "loss_g0": 32.143367767333984,
      "loss_g1": 16.646364212036133,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 38.00780487060547,
      "loss_g0": 0.0,
      "loss_g1": 38.00780487060547,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.31544303894043,
      "loss_g0": 12.789358139038086,
      "loss_g1": 11.290680885314941,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.255218505859375,
      "loss_g0": 0.0,
      "loss_g1": 11.255218505859375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.541118621826172,
      "loss_g0": 11.251448631286621,
      "loss_g1": 10.529184341430664,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 18.013059616088867,
      "loss_g0": 23.5924129486084,
      "loss_g1": 17.919322967529297,
      "q_g0": 0.016522999852895737,
      "q_g1": 0.9834769368171692,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 30.728071212768555,
      "loss_g0": 13.01888656616211,
      "loss_g1": 31.025585174560547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.562113285064697,
      "loss_g0": 24.310001373291016,
      "loss_g1": 7.280718803405762,
      "q_g0": 0.016524169594049454,
      "q_g1": 0.9834758043289185,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 6.058482646942139,
      "loss_g0": 4.876395225524902,
      "loss_g1": 6.078343391418457,
      "q_g0": 0.016524044796824455,
      "q_g1": 0.983475923538208,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.3091959953308105,
      "loss_g0": 12.276973724365234,
      "loss_g1": 5.192120552062988,
      "q_g0": 0.016524773091077805,
      "q_g1": 0.9834752082824707,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 12.680984497070312,
      "loss_g0": 14.310663223266602,
      "loss_g1": 12.65360164642334,
      "q_g0": 0.016524946317076683,
      "q_g1": 0.9834750890731812,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 11.179034233093262,
      "loss_g0": 13.59383773803711,
      "loss_g1": 11.138459205627441,
      "q_g0": 0.01652519218623638,
      "q_g1": 0.9834747910499573,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 7.128073692321777,
      "loss_g0": 8.467601776123047,
      "loss_g1": 7.105565547943115,
      "q_g0": 0.016525331884622574,
      "q_g1": 0.9834746718406677,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 19.208209991455078,
      "loss_g0": 10.375545501708984,
      "loss_g1": 19.356616973876953,
      "q_g0": 0.016524409875273705,
      "q_g1": 0.9834755659103394,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 5.283294677734375,
      "loss_g0": 10.417912483215332,
      "loss_g1": 5.197019577026367,
      "q_g0": 0.016524938866496086,
      "q_g1": 0.9834750890731812,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 9.573959350585938,
      "loss_g0": 41.773929595947266,
      "loss_g1": 9.032804489135742,
      "q_g0": 0.016528286039829254,
      "q_g1": 0.9834717512130737,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 3.714902400970459,
      "loss_g0": 3.714902400970459,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 0.9993663430213928,
      "loss_g0": 0.9526463747024536,
      "loss_g1": 3.7803168296813965,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 23.225563049316406,
      "loss_g0": 23.61236572265625,
      "loss_g1": 0.2016148418188095,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "eval_cer": 0.060016766093500214,
      "eval_loss": 13.612616539001465,
      "eval_runtime": 80.8918,
      "eval_samples_per_second": 20.719,
      "eval_steps_per_second": 5.18,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 14.706605911254883,
      "loss_g0": 14.219215393066406,
      "loss_g1": 43.71253204345703,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.01652543991804123,
      "step": 1200
    },
    {
      "epoch": 6.091370558375634,
      "gdro_loss": 10.952299118041992,
      "loss_g0": 10.811888694763184,
      "loss_g1": 19.30803680419922,
      "q_g0": 0.9834737181663513,
      "q_g1": 0.01652630977332592,
      "step": 1200
    },
    {
      "epoch": 6.598984771573604,
      "grad_norm": 65.60956573486328,
      "learning_rate": 0.00016776470588235294,
      "loss": 27.9095,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.04648208618164,
      "loss_g0": 26.344186782836914,
      "loss_g1": 9.772651672363281,
      "q_g0": 0.016524124890565872,
      "q_g1": 0.983475923538208,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 36.315277099609375,
      "loss_g0": 0.0,
      "loss_g1": 36.315277099609375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 28.231578826904297,
      "loss_g0": 0.0,
      "loss_g1": 28.231578826904297,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 20.208070755004883,
      "loss_g0": 25.58837127685547,
      "loss_g1": 20.1176815032959,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 17.159862518310547,
      "loss_g0": 16.70513153076172,
      "loss_g1": 17.16750144958496,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 24.60431480407715,
      "loss_g0": 24.933149337768555,
      "loss_g1": 24.598791122436523,
      "q_g0": 0.01652245782315731,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.221054077148438,
      "loss_g0": 10.221054077148438,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 40.51918029785156,
      "loss_g0": 40.804691314697266,
      "loss_g1": 23.52440071105957,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.84293556213379,
      "loss_g0": 26.156673431396484,
      "loss_g1": 7.168045997619629,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.17220401763916,
      "loss_g0": 15.291244506835938,
      "loss_g1": 8.0864839553833,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.014055252075195,
      "loss_g0": 18.092205047607422,
      "loss_g1": 13.362296104431152,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.339637756347656,
      "loss_g0": 11.091402053833008,
      "loss_g1": 26.11423110961914,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523951664566994,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.258931636810303,
      "loss_g0": 6.06393575668335,
      "loss_g1": 17.863866806030273,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525164246559143,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.071539878845215,
      "loss_g0": 8.141582489013672,
      "loss_g1": 3.9029574394226074,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524728387594223,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.178879737854004,
      "loss_g0": 14.06558895111084,
      "loss_g1": 20.92112159729004,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525432467460632,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 28.058841705322266,
      "loss_g0": 28.364521026611328,
      "loss_g1": 9.86494255065918,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523543745279312,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.837209701538086,
      "loss_g0": 25.945785522460938,
      "loss_g1": 19.374446868896484,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652287133038044,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.0162353515625,
      "loss_g0": 19.254690170288086,
      "loss_g1": 4.822467803955078,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.748854160308838,
      "loss_g0": 4.4913434982299805,
      "loss_g1": 20.075353622436523,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524016857147217,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.45844841003418,
      "loss_g0": 18.34110450744629,
      "loss_g1": 25.44221305847168,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524743288755417,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.859562635421753,
      "loss_g0": 3.7168960571289062,
      "loss_g1": 12.349959373474121,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.0165256280452013,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.49459171295166,
      "loss_g0": 10.540478706359863,
      "loss_g1": 7.763704299926758,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.01652533933520317,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.204632759094238,
      "loss_g0": 14.428378105163574,
      "loss_g1": 0.887715756893158,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523951664566994,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.87997055053711,
      "loss_g0": 26.259563446044922,
      "loss_g1": 3.2850894927978516,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.677786350250244,
      "loss_g0": 2.586836576461792,
      "loss_g1": 8.091269493103027,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522984951734543,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.759523868560791,
      "loss_g0": 5.769136428833008,
      "loss_g1": 5.187384605407715,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522914171218872,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.285079956054688,
      "loss_g0": 22.46007537841797,
      "loss_g1": 11.868583679199219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.375857353210449,
      "loss_g0": 7.3832478523254395,
      "loss_g1": 6.935964107513428,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.480359077453613,
      "loss_g0": 13.634319305419922,
      "loss_g1": 4.316072940826416,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.08280086517334,
      "loss_g0": 15.207048416137695,
      "loss_g1": 7.687087535858154,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.74854278564453,
      "loss_g0": 16.892560958862305,
      "loss_g1": 8.176005363464355,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.571575164794922,
      "loss_g0": 0.0,
      "loss_g1": 5.571575164794922,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.229974746704102,
      "loss_g0": 21.51729393005371,
      "loss_g1": 9.023547172546387,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.927333831787109,
      "loss_g0": 9.684679985046387,
      "loss_g1": 7.897809982299805,
      "q_g0": 0.0165226049721241,
      "q_g1": 0.9834774136543274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.382511138916016,
      "loss_g0": 23.072113037109375,
      "loss_g1": 5.085289001464844,
      "q_g0": 0.01652444154024124,
      "q_g1": 0.9834755659103394,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.9242498874664307,
      "loss_g0": 7.678045749664307,
      "loss_g1": 3.8611764907836914,
      "q_g0": 0.01652483083307743,
      "q_g1": 0.9834752082824707,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.510793685913086,
      "loss_g0": 3.4344537258148193,
      "loss_g1": 25.88167953491211,
      "q_g0": 0.01652253419160843,
      "q_g1": 0.9834774136543274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.872370719909668,
      "loss_g0": 15.390881538391113,
      "loss_g1": 2.662041425704956,
      "q_g0": 0.01652383618056774,
      "q_g1": 0.9834761619567871,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.363255739212036,
      "loss_g0": 9.290853500366211,
      "loss_g1": 3.2636594772338867,
      "q_g0": 0.01652444899082184,
      "q_g1": 0.9834755659103394,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.549125671386719,
      "loss_g0": 36.6347770690918,
      "loss_g1": 9.093951225280762,
      "q_g0": 0.016527263447642326,
      "q_g1": 0.9834727048873901,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.2843587398529053,
      "loss_g0": 2.989366292953491,
      "loss_g1": 2.2725110054016113,
      "q_g0": 0.016527332365512848,
      "q_g1": 0.9834727048873901,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.35536789894104,
      "loss_g0": 0.7940649390220642,
      "loss_g1": 3.3984100818634033,
      "q_g0": 0.01652706228196621,
      "q_g1": 0.9834729433059692,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.16521167755127,
      "loss_g0": 10.16521167755127,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.5962672233581543,
      "loss_g0": 3.5962672233581543,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.976942539215088,
      "loss_g0": 3.943233013153076,
      "loss_g1": 5.983457565307617,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.744061470031738,
      "loss_g0": 11.863187789916992,
      "loss_g1": 4.653238773345947,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.383663177490234,
      "loss_g0": 12.44513988494873,
      "loss_g1": 8.724320411682129,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.416959285736084,
      "loss_g0": 3.411668300628662,
      "loss_g1": 3.7318997383117676,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652245782315731,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.861677169799805,
      "loss_g0": 16.114444732666016,
      "loss_g1": 0.8159849643707275,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.0485479831695557,
      "loss_g0": 3.096893310546875,
      "loss_g1": 0.1708546280860901,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 20.85312271118164,
      "loss_g0": 20.819705963134766,
      "loss_g1": 22.842220306396484,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652262918651104,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 24.869096755981445,
      "loss_g0": 0.0,
      "loss_g1": 24.869096755981445,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 64.80906677246094,
      "loss_g0": 31.956228256225586,
      "loss_g1": 65.36099243164062,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.416650772094727,
      "loss_g0": 20.732120513916016,
      "loss_g1": 18.377750396728516,
      "q_g0": 0.016522657126188278,
      "q_g1": 0.9834772944450378,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.381092071533203,
      "loss_g0": 20.842308044433594,
      "loss_g1": 18.33974266052246,
      "q_g0": 0.016522914171218872,
      "q_g1": 0.9834770560264587,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.99944496154785,
      "loss_g0": 18.955196380615234,
      "loss_g1": 16.96658706665039,
      "q_g0": 0.01652311533689499,
      "q_g1": 0.9834768176078796,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 23.171749114990234,
      "loss_g0": 22.630550384521484,
      "loss_g1": 23.18084144592285,
      "q_g0": 0.01652306318283081,
      "q_g1": 0.9834769368171692,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.448352813720703,
      "loss_g0": 20.358327865600586,
      "loss_g1": 9.265045166015625,
      "q_g0": 0.01652420125901699,
      "q_g1": 0.9834758043289185,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 21.2142391204834,
      "loss_g0": 34.674407958984375,
      "loss_g1": 20.98806381225586,
      "q_g0": 0.016525596380233765,
      "q_g1": 0.9834744334220886,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.84939956665039,
      "loss_g0": 26.84939956665039,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.471424102783203,
      "loss_g0": 0.0,
      "loss_g1": 15.471424102783203,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.063254356384277,
      "loss_g0": 0.0,
      "loss_g1": 14.063254356384277,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.97506046295166,
      "loss_g0": 8.637231826782227,
      "loss_g1": 12.031135559082031,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 34.47230911254883,
      "loss_g0": 23.892791748046875,
      "loss_g1": 34.65004348754883,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.28292465209961,
      "loss_g0": 15.28292465209961,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.862241744995117,
      "loss_g0": 12.738285064697266,
      "loss_g1": 20.2406005859375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.430217742919922,
      "loss_g0": 7.007722854614258,
      "loss_g1": 32.57466506958008,
      "q_g0": 0.9834749698638916,
      "q_g1": 0.016525037586688995,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.164546012878418,
      "loss_g0": 9.937247276306152,
      "loss_g1": 23.690860748291016,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.016526447609066963,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.915334701538086,
      "loss_g0": 26.915334701538086,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.91400909423828,
      "loss_g0": 16.98965835571289,
      "loss_g1": 12.411020278930664,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.08005714416504,
      "loss_g0": 18.97810173034668,
      "loss_g1": 25.148529052734375,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523055732250214,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.005293846130371,
      "loss_g0": 11.12790584564209,
      "loss_g1": 3.7069449424743652,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.943456649780273,
      "loss_g0": 8.882732391357422,
      "loss_g1": 12.55793285369873,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652279868721962,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.90455436706543,
      "loss_g0": 15.951712608337402,
      "loss_g1": 13.097465515136719,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522511839866638,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.217808723449707,
      "loss_g0": 14.373805046081543,
      "loss_g1": 4.932300090789795,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.847990036010742,
      "loss_g0": 15.800249099731445,
      "loss_g1": 18.68964195251465,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.0165227223187685,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.565189361572266,
      "loss_g0": 10.468708038330078,
      "loss_g1": 16.307767868041992,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523318365216255,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.943944931030273,
      "loss_g0": 11.943944931030273,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.716219902038574,
      "loss_g0": 8.805240631103516,
      "loss_g1": 3.417397975921631,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.953452110290527,
      "loss_g0": 10.964746475219727,
      "loss_g1": 10.281129837036133,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.541738510131836,
      "loss_g0": 13.609501838684082,
      "loss_g1": 9.508237838745117,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.761315822601318,
      "loss_g0": 5.761315822601318,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.7939016819000244,
      "loss_g0": 2.7852962017059326,
      "loss_g1": 3.3061282634735107,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.044427871704102,
      "loss_g0": 0.0,
      "loss_g1": 8.044427871704102,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.818422317504883,
      "loss_g0": 0.0,
      "loss_g1": 8.818422317504883,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.5667831897735596,
      "loss_g0": 13.418354034423828,
      "loss_g1": 3.4012765884399414,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.345190525054932,
      "loss_g0": 13.440897941589355,
      "loss_g1": 5.2091755867004395,
      "q_g0": 0.016523264348506927,
      "q_g1": 0.9834766983985901,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.222915172576904,
      "loss_g0": 6.489318370819092,
      "loss_g1": 7.2352399826049805,
      "q_g0": 0.016523193567991257,
      "q_g1": 0.9834768176078796,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.304530143737793,
      "loss_g0": 15.807877540588379,
      "loss_g1": 7.161659240722656,
      "q_g0": 0.01652408577501774,
      "q_g1": 0.983475923538208,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.46509075164795,
      "loss_g0": 11.46509075164795,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 30.068683624267578,
      "loss_g0": 30.068683624267578,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.527466773986816,
      "loss_g0": 5.587904930114746,
      "loss_g1": 1.9299509525299072,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.963507652282715,
      "loss_g0": 11.08358097076416,
      "loss_g1": 3.816253900527954,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.112684726715088,
      "loss_g0": 7.112684726715088,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.496062278747559,
      "loss_g0": 8.496062278747559,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.896979331970215,
      "loss_g0": 4.917144775390625,
      "loss_g1": 3.6966466903686523,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.618859767913818,
      "loss_g0": 7.735238552093506,
      "loss_g1": 0.691551685333252,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 32.85514450073242,
      "loss_g0": 33.257076263427734,
      "loss_g1": 8.93076229095459,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.48600959777832,
      "loss_g0": 6.48600959777832,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.834125518798828,
      "loss_g0": 9.986713409423828,
      "loss_g1": 0.751510500907898,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.334582328796387,
      "loss_g0": 8.998810768127441,
      "loss_g1": 29.31844139099121,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524503007531166,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.426852226257324,
      "loss_g0": 12.496912956237793,
      "loss_g1": 8.256994247436523,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524069011211395,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 21.040231704711914,
      "loss_g0": 20.95976448059082,
      "loss_g1": 25.82929229736328,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524571925401688,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 71.4716796875,
      "loss_g0": 72.31312561035156,
      "loss_g1": 21.385549545288086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 28.240110397338867,
      "loss_g0": 28.304367065429688,
      "loss_g1": 24.415306091308594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 28.00492286682129,
      "loss_g0": 28.00286293029785,
      "loss_g1": 28.127548217773438,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.787599563598633,
      "loss_g0": 22.85820770263672,
      "loss_g1": 18.58473014831543,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 20.586048126220703,
      "loss_g0": 20.64445686340332,
      "loss_g1": 17.10933494567871,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 17.79876136779785,
      "loss_g0": 17.79876136779785,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.882669448852539,
      "loss_g0": 8.934977531433105,
      "loss_g1": 5.769058704376221,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 27.02309799194336,
      "loss_g0": 27.304763793945312,
      "loss_g1": 10.25725269317627,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.592949867248535,
      "loss_g0": 11.56027603149414,
      "loss_g1": 13.537758827209473,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652262918651104,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 33.890037536621094,
      "loss_g0": 34.303585052490234,
      "loss_g1": 9.274059295654297,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.031126022338867,
      "loss_g0": 25.437192916870117,
      "loss_g1": 0.8604943156242371,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.006196975708008,
      "loss_g0": 22.32796859741211,
      "loss_g1": 2.8531012535095215,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.265276908874512,
      "loss_g0": 8.262850761413574,
      "loss_g1": 8.40969467163086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.05547046661377,
      "loss_g0": 10.103533744812012,
      "loss_g1": 7.194540023803711,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 31.326383590698242,
      "loss_g0": 31.69133758544922,
      "loss_g1": 9.602846145629883,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.121541976928711,
      "loss_g0": 15.13021469116211,
      "loss_g1": 14.605303764343262,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.59625244140625,
      "loss_g0": 14.59625244140625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 31.739713668823242,
      "loss_g0": 32.13248825073242,
      "loss_g1": 8.360331535339355,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 63.11547088623047,
      "loss_g0": 63.92039489746094,
      "loss_g1": 15.203211784362793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.3355073928833,
      "loss_g0": 13.190872192382812,
      "loss_g1": 21.94424819946289,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523318365216255,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.43767261505127,
      "loss_g0": 15.54062271118164,
      "loss_g1": 9.309786796569824,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.889064311981201,
      "loss_g0": 7.865344047546387,
      "loss_g1": 9.30093765258789,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522830352187157,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.513428688049316,
      "loss_g0": 8.547966957092285,
      "loss_g1": 6.457592010498047,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522621735930443,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.554447174072266,
      "loss_g0": 15.68509578704834,
      "loss_g1": 7.7777099609375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.820249557495117,
      "loss_g0": 12.006187438964844,
      "loss_g1": 0.752475917339325,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.829785346984863,
      "loss_g0": 9.973296165466309,
      "loss_g1": 1.2874685525894165,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.700899124145508,
      "loss_g0": 14.863046646118164,
      "loss_g1": 5.049249172210693,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.364971160888672,
      "loss_g0": 16.60552215576172,
      "loss_g1": 2.046426296234131,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.6814863681793213,
      "loss_g0": 2.483304023742676,
      "loss_g1": 14.47714614868164,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652364991605282,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 32.26359939575195,
      "loss_g0": 32.58646011352539,
      "loss_g1": 13.045608520507812,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.91718864440918,
      "loss_g0": 23.180334091186523,
      "loss_g1": 7.253762245178223,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.5056633949279785,
      "loss_g0": 2.463545322418213,
      "loss_g1": 5.012658596038818,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522681340575218,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.577524185180664,
      "loss_g0": 6.493758201599121,
      "loss_g1": 11.563346862792969,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523201018571854,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.98434591293335,
      "loss_g0": 4.966480255126953,
      "loss_g1": 6.0477142333984375,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523318365216255,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.821295261383057,
      "loss_g0": 5.875186443328857,
      "loss_g1": 2.613600015640259,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522984951734543,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.244437217712402,
      "loss_g0": 14.254196166992188,
      "loss_g1": 13.663580894470215,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652292162179947,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.820785045623779,
      "loss_g0": 5.7945756912231445,
      "loss_g1": 7.3808064460754395,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523078083992004,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.570309162139893,
      "loss_g0": 5.590444564819336,
      "loss_g1": 4.371812343597412,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522955149412155,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.774470806121826,
      "loss_g0": 2.721832275390625,
      "loss_g1": 5.907543659210205,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523286700248718,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.098183631896973,
      "loss_g0": 13.307828903198242,
      "loss_g1": 0.6192561984062195,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.191936016082764,
      "loss_g0": 5.234763145446777,
      "loss_g1": 2.6427016258239746,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.755568027496338,
      "loss_g0": 4.801219463348389,
      "loss_g1": 2.0382003784179688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.72446346282959,
      "loss_g0": 2.72446346282959,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.041964054107666,
      "loss_g0": 6.123147487640381,
      "loss_g1": 1.2096062898635864,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.488235950469971,
      "loss_g0": 4.465558052062988,
      "loss_g1": 5.838109016418457,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522565856575966,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.848621368408203,
      "loss_g0": 7.888516426086426,
      "loss_g1": 5.47390079498291,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.9315956830978394,
      "loss_g0": 1.914223074913025,
      "loss_g1": 2.9656708240509033,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522526741027832,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 77.05565643310547,
      "loss_g0": 77.89866638183594,
      "loss_g1": 26.876190185546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.667856216430664,
      "loss_g0": 25.419925689697266,
      "loss_g1": 40.4243278503418,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523951664566994,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.211374282836914,
      "loss_g0": 26.211374282836914,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 33.623077392578125,
      "loss_g0": 33.930015563964844,
      "loss_g1": 15.352832794189453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.06019115447998,
      "loss_g0": 8.920283317565918,
      "loss_g1": 17.38761329650879,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523286700248718,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.430862426757812,
      "loss_g0": 14.547374725341797,
      "loss_g1": 7.495682239532471,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522565856575966,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 45.257843017578125,
      "loss_g0": 45.30295944213867,
      "loss_g1": 42.572303771972656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.286134719848633,
      "loss_g0": 6.000457286834717,
      "loss_g1": 23.288902282714844,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524186357855797,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.788763999938965,
      "loss_g0": 11.822128295898438,
      "loss_g1": 9.802999496459961,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523977741599083,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.040183067321777,
      "loss_g0": 11.906940460205078,
      "loss_g1": 19.970163345336914,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524799168109894,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.04408073425293,
      "loss_g0": 14.063690185546875,
      "loss_g1": 12.876989364624023,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524681821465492,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.257954120635986,
      "loss_g0": 7.224880695343018,
      "loss_g1": 9.22633171081543,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524890437722206,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 55.06344223022461,
      "loss_g0": 55.69491958618164,
      "loss_g1": 17.475444793701172,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.257274627685547,
      "loss_g0": 10.274377822875977,
      "loss_g1": 9.239256858825684,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 23.499500274658203,
      "loss_g0": 23.499500274658203,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.375000476837158,
      "loss_g0": 5.36929988861084,
      "loss_g1": 5.714297771453857,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.261075973510742,
      "loss_g0": 26.234100341796875,
      "loss_g1": 27.866741180419922,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522597521543503,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.538089752197266,
      "loss_g0": 18.48428726196289,
      "loss_g1": 21.74043846130371,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522930935025215,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.577138900756836,
      "loss_g0": 26.737058639526367,
      "loss_g1": 17.058088302612305,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.70600700378418,
      "loss_g0": 0.0,
      "loss_g1": 10.70600700378418,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.46531867980957,
      "loss_g0": 2.694183349609375,
      "loss_g1": 8.562273025512695,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.426861763000488,
      "loss_g0": 24.232154846191406,
      "loss_g1": 15.278924942016602,
      "q_g0": 0.016523340716958046,
      "q_g1": 0.9834766387939453,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.3487958908081055,
      "loss_g0": 20.717859268188477,
      "loss_g1": 7.124163627624512,
      "q_g0": 0.01652473583817482,
      "q_g1": 0.9834752082824707,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.184748649597168,
      "loss_g0": 0.0,
      "loss_g1": 6.184748649597168,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.004449844360352,
      "loss_g0": 3.5456695556640625,
      "loss_g1": 11.129756927490234,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 17.968332290649414,
      "loss_g0": 19.909950256347656,
      "loss_g1": 17.935712814331055,
      "q_g0": 0.01652262918651104,
      "q_g1": 0.9834774136543274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.7415552139282227,
      "loss_g0": 10.635913848876953,
      "loss_g1": 2.608921766281128,
      "q_g0": 0.016523443162441254,
      "q_g1": 0.9834766387939453,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.756403923034668,
      "loss_g0": 4.36553430557251,
      "loss_g1": 7.813372611999512,
      "q_g0": 0.016523092985153198,
      "q_g1": 0.9834769368171692,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.195501327514648,
      "loss_g0": 40.12638854980469,
      "loss_g1": 9.692545890808105,
      "q_g0": 0.016526201739907265,
      "q_g1": 0.9834738373756409,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.51352596282959,
      "loss_g0": 8.322664260864258,
      "loss_g1": 8.516733169555664,
      "q_g0": 0.01652618497610092,
      "q_g1": 0.9834738373756409,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.8171396255493164,
      "loss_g0": 5.98883581161499,
      "loss_g1": 2.7638418674468994,
      "q_g0": 0.016526510939002037,
      "q_g1": 0.9834734797477722,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.828563213348389,
      "loss_g0": 3.8638720512390137,
      "loss_g1": 4.844773769378662,
      "q_g0": 0.01652641035616398,
      "q_g1": 0.9834735989570618,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.9881832599639893,
      "loss_g0": 8.082653045654297,
      "loss_g1": 2.9025721549987793,
      "q_g0": 0.016526944935321808,
      "q_g1": 0.9834730625152588,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.5730926990509033,
      "loss_g0": 23.322322845458984,
      "loss_g1": 2.224362373352051,
      "q_g0": 0.016529105603694916,
      "q_g1": 0.9834708571434021,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.944110155105591,
      "loss_g0": 9.851384162902832,
      "loss_g1": 3.8448233604431152,
      "q_g0": 0.016529718413949013,
      "q_g1": 0.9834702610969543,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.074200630187988,
      "loss_g0": 7.1563239097595215,
      "loss_g1": 5.0392045974731445,
      "q_g0": 0.016529936343431473,
      "q_g1": 0.9834700226783752,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.246140003204346,
      "loss_g0": 4.277138710021973,
      "loss_g1": 4.245619297027588,
      "q_g0": 0.016529936343431473,
      "q_g1": 0.9834700226783752,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.8854055404663086,
      "loss_g0": 4.417376518249512,
      "loss_g1": 2.8596560955047607,
      "q_g0": 0.016530092805624008,
      "q_g1": 0.9834699034690857,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.452409267425537,
      "loss_g0": 8.487385749816895,
      "loss_g1": 7.435013294219971,
      "q_g0": 0.016530200839042664,
      "q_g1": 0.9834697842597961,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.353536605834961,
      "loss_g0": 7.8600263595581055,
      "loss_g1": 7.3450236320495605,
      "q_g0": 0.016530252993106842,
      "q_g1": 0.9834696650505066,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.680734157562256,
      "loss_g0": 71.58666229248047,
      "loss_g1": 5.589335918426514,
      "q_g0": 0.016537006944417953,
      "q_g1": 0.9834629893302917,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.320258617401123,
      "loss_g0": 11.70482063293457,
      "loss_g1": 5.212897300720215,
      "q_g0": 0.01653766818344593,
      "q_g1": 0.9834623336791992,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.984593391418457,
      "loss_g0": 4.598730087280273,
      "loss_g1": 6.00789737701416,
      "q_g0": 0.01653752662241459,
      "q_g1": 0.9834624528884888,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.740610599517822,
      "loss_g0": 5.486291408538818,
      "loss_g1": 4.728071212768555,
      "q_g0": 0.01653759740293026,
      "q_g1": 0.9834624528884888,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.8196605443954468,
      "loss_g0": 3.630861520767212,
      "loss_g1": 1.7892035245895386,
      "q_g0": 0.016537781804800034,
      "q_g1": 0.9834622144699097,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.4583908319473267,
      "loss_g0": 4.505123138427734,
      "loss_g1": 1.4071563482284546,
      "q_g0": 0.016538100317120552,
      "q_g1": 0.983461856842041,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.651669025421143,
      "loss_g0": 7.895617961883545,
      "loss_g1": 7.647567272186279,
      "q_g0": 0.016538122668862343,
      "q_g1": 0.983461856842041,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.3617029190063477,
      "loss_g0": 3.4207966327667236,
      "loss_g1": 2.343893051147461,
      "q_g0": 0.016538232564926147,
      "q_g1": 0.9834617376327515,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.809669494628906,
      "loss_g0": 20.68651580810547,
      "loss_g1": 8.609929084777832,
      "q_g0": 0.016539474949240685,
      "q_g1": 0.9834606051445007,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.003629684448242,
      "loss_g0": 0.605678141117096,
      "loss_g1": 8.128039360046387,
      "q_g0": 0.016538698226213455,
      "q_g1": 0.9834612607955933,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.404560089111328,
      "loss_g0": 10.672823905944824,
      "loss_g1": 19.551393508911133,
      "q_g0": 0.01653778925538063,
      "q_g1": 0.9834622144699097,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.083765029907227,
      "loss_g0": 24.07777976989746,
      "loss_g1": 25.100679397583008,
      "q_g0": 0.016537688672542572,
      "q_g1": 0.9834623336791992,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.36177635192871,
      "loss_g0": 22.36177635192871,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 20.154560089111328,
      "loss_g0": 20.20142936706543,
      "loss_g1": 17.364727020263672,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 30.249164581298828,
      "loss_g0": 30.33106803894043,
      "loss_g1": 25.373981475830078,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.474605560302734,
      "loss_g0": 14.656420707702637,
      "loss_g1": 64.16096496582031,
      "q_g0": 0.9834725856781006,
      "q_g1": 0.016527481377124786,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.779861450195312,
      "loss_g0": 11.873241424560547,
      "loss_g1": 6.223040580749512,
      "q_g0": 0.9834731221199036,
      "q_g1": 0.016526905819773674,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.945501804351807,
      "loss_g0": 6.631173610687256,
      "loss_g1": 25.64815330505371,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.01652885414659977,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 24.62127113342285,
      "loss_g0": 24.630903244018555,
      "loss_g1": 24.048219680786133,
      "q_g0": 0.9834712147712708,
      "q_g1": 0.016528788954019547,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.0376033782959,
      "loss_g0": 22.0893611907959,
      "loss_g1": 18.957963943481445,
      "q_g0": 0.9834715127944946,
      "q_g1": 0.016528472304344177,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 29.721174240112305,
      "loss_g0": 30.093578338623047,
      "loss_g1": 7.5593366622924805,
      "q_g0": 0.9834738373756409,
      "q_g1": 0.016526170074939728,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 28.70355224609375,
      "loss_g0": 28.966251373291016,
      "loss_g1": 13.068794250488281,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.0165245421230793,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.350395202636719,
      "loss_g0": 9.137900352478027,
      "loss_g1": 21.996246337890625,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.01652585156261921,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.861373901367188,
      "loss_g0": 18.91240119934082,
      "loss_g1": 15.824535369873047,
      "q_g0": 0.9834744334220886,
      "q_g1": 0.01652553305029869,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 23.862747192382812,
      "loss_g0": 23.86884307861328,
      "loss_g1": 23.4998722076416,
      "q_g0": 0.9834745526313782,
      "q_g1": 0.016525497660040855,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.613519668579102,
      "loss_g0": 13.654911994934082,
      "loss_g1": 11.150041580200195,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.016525238752365112,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.121572494506836,
      "loss_g0": 16.317596435546875,
      "loss_g1": 4.454620361328125,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524024307727814,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.0496551990509033,
      "loss_g0": 2.904489040374756,
      "loss_g1": 11.689178466796875,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524914652109146,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.785476684570312,
      "loss_g0": 9.635872840881348,
      "loss_g1": 18.688602447509766,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.016525844112038612,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.059492111206055,
      "loss_g0": 6.081871032714844,
      "loss_g1": 4.727686882019043,
      "q_g0": 0.9834743142127991,
      "q_g1": 0.016525696963071823,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 21.441911697387695,
      "loss_g0": 21.679006576538086,
      "loss_g1": 7.330696105957031,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524232923984528,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.05711555480957,
      "loss_g0": 18.177860260009766,
      "loss_g1": 10.870441436767578,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652348041534424,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.60060453414917,
      "loss_g0": 6.605072498321533,
      "loss_g1": 6.33469295501709,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.0165234562009573,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.425674438476562,
      "loss_g0": 19.70016098022461,
      "loss_g1": 3.0872297286987305,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.508049011230469,
      "loss_g0": 8.482660293579102,
      "loss_g1": 10.019299507141113,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.01652258075773716,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.5235153436660767,
      "loss_g0": 1.4332493543624878,
      "loss_g1": 6.896249294281006,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652313955128193,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.953071594238281,
      "loss_g0": 9.012258529663086,
      "loss_g1": 5.430139541625977,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652277447283268,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.87536907196045,
      "loss_g0": 13.656734466552734,
      "loss_g1": 26.887916564941406,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652413234114647,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.482501983642578,
      "loss_g0": 11.482501983642578,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.4503021240234375,
      "loss_g0": 6.394914627075195,
      "loss_g1": 9.747176170349121,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.055843353271484,
      "loss_g0": 12.167259216308594,
      "loss_g1": 5.4239654541015625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.834543228149414,
      "loss_g0": 6.77803897857666,
      "loss_g1": 10.197813034057617,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652277447283268,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.3577299118042,
      "loss_g0": 15.45831298828125,
      "loss_g1": 9.370582580566406,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.6382150650024414,
      "loss_g0": 2.574657917022705,
      "loss_g1": 6.421282768249512,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.782543182373047,
      "loss_g0": 0.0,
      "loss_g1": 9.782543182373047,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.673460960388184,
      "loss_g0": 18.898035049438477,
      "loss_g1": 9.518487930297852,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.853038311004639,
      "loss_g0": 6.639403343200684,
      "loss_g1": 7.873426914215088,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.072182655334473,
      "loss_g0": 11.507889747619629,
      "loss_g1": 4.964057922363281,
      "q_g0": 0.016523092985153198,
      "q_g1": 0.9834769368171692,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.039941310882568,
      "loss_g0": 18.577953338623047,
      "loss_g1": 3.7956700325012207,
      "q_g0": 0.016524603590369225,
      "q_g1": 0.9834753274917603,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.9173667430877686,
      "loss_g0": 9.705162048339844,
      "loss_g1": 3.820115089416504,
      "q_g0": 0.016525207087397575,
      "q_g1": 0.9834747910499573,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 0.4915740489959717,
      "loss_g0": 4.931098937988281,
      "loss_g1": 0.4169751703739166,
      "q_g0": 0.016525665298104286,
      "q_g1": 0.9834743142127991,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.683300018310547,
      "loss_g0": 3.3895182609558105,
      "loss_g1": 12.839456558227539,
      "q_g0": 0.016524706035852432,
      "q_g1": 0.9834753274917603,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.40802526473999,
      "loss_g0": 15.927963256835938,
      "loss_g1": 4.214448928833008,
      "q_g0": 0.016525907441973686,
      "q_g1": 0.98347407579422,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.535205841064453,
      "loss_g0": 1.4242690801620483,
      "loss_g1": 10.68829345703125,
      "q_g0": 0.016524961218237877,
      "q_g1": 0.9834749698638916,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.248908042907715,
      "loss_g0": 4.544020175933838,
      "loss_g1": 11.361562728881836,
      "q_g0": 0.016524260863661766,
      "q_g1": 0.9834756851196289,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.388437271118164,
      "loss_g0": 22.03719139099121,
      "loss_g1": 1.0414530038833618,
      "q_g0": 0.01652641035616398,
      "q_g1": 0.9834735989570618,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.821183443069458,
      "loss_g0": 5.765775680541992,
      "loss_g1": 2.7717013359069824,
      "q_g0": 0.016526712104678154,
      "q_g1": 0.9834732413291931,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.861032485961914,
      "loss_g0": 8.048894882202148,
      "loss_g1": 8.874679565429688,
      "q_g0": 0.01652662642300129,
      "q_g1": 0.9834733605384827,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.3470041751861572,
      "loss_g0": 1.3470041751861572,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.29008674621582,
      "loss_g0": 4.214651107788086,
      "loss_g1": 8.780295372009277,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 56.281829833984375,
      "loss_g0": 57.0218505859375,
      "loss_g1": 12.232844352722168,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.200701713562012,
      "loss_g0": 12.084771156311035,
      "loss_g1": 19.10101890563965,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.01652313955128193,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.48537254333496,
      "loss_g0": 16.014129638671875,
      "loss_g1": 44.52933120727539,
      "q_g0": 0.9834739565849304,
      "q_g1": 0.016526052728295326,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 20.19561767578125,
      "loss_g0": 20.19561767578125,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 17.164400100708008,
      "loss_g0": 17.083051681518555,
      "loss_g1": 22.006603240966797,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.768540382385254,
      "loss_g0": 13.684450149536133,
      "loss_g1": 18.773733139038086,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652294583618641,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.46017074584961,
      "loss_g0": 15.263251304626465,
      "loss_g1": 27.180248260498047,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524160280823708,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 21.182315826416016,
      "loss_g0": 21.184232711791992,
      "loss_g1": 21.068267822265625,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652415469288826,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 31.022192001342773,
      "loss_g0": 31.121723175048828,
      "loss_g1": 25.09823226928711,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523543745279312,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.347164154052734,
      "loss_g0": 26.402109146118164,
      "loss_g1": 23.07680892944336,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523193567991257,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.648712158203125,
      "loss_g0": 16.67452621459961,
      "loss_g1": 15.11215877532959,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523031517863274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.88687801361084,
      "loss_g0": 5.845712184906006,
      "loss_g1": 8.337113380432129,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523286700248718,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.239482879638672,
      "loss_g0": 0.0,
      "loss_g1": 11.239482879638672,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 29.27690315246582,
      "loss_g0": 5.655154228210449,
      "loss_g1": 29.673748016357422,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.423429489135742,
      "loss_g0": 9.353913307189941,
      "loss_g1": 22.64299774169922,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 26.686790466308594,
      "loss_g0": 26.686790466308594,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.040658950805664,
      "loss_g0": 5.945956230163574,
      "loss_g1": 11.677706718444824,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.19141674041748,
      "loss_g0": 15.116270065307617,
      "loss_g1": 19.66429328918457,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522882506251335,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.4896970987319946,
      "loss_g0": 1.3842284679412842,
      "loss_g1": 7.767167568206787,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652352698147297,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.512619018554688,
      "loss_g0": 9.62643814086914,
      "loss_g1": 2.7378158569335938,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.018163681030273,
      "loss_g0": 16.21231460571289,
      "loss_g1": 4.461533546447754,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.092841625213623,
      "loss_g0": 6.921838760375977,
      "loss_g1": 17.27091407775879,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.01652348041534424,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 20.20948600769043,
      "loss_g0": 20.388931274414062,
      "loss_g1": 9.52815055847168,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.918331146240234,
      "loss_g0": 11.90420150756836,
      "loss_g1": 12.759324073791504,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522511839866638,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.65404224395752,
      "loss_g0": 13.700900077819824,
      "loss_g1": 10.864858627319336,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.635597229003906,
      "loss_g0": 19.806774139404297,
      "loss_g1": 9.446432113647461,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.419027328491211,
      "loss_g0": 15.595947265625,
      "loss_g1": 4.888099670410156,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.871317386627197,
      "loss_g0": 7.643711090087891,
      "loss_g1": 21.418149948120117,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.01652383618056774,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.365081787109375,
      "loss_g0": 0.0,
      "loss_g1": 15.365081787109375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.469674587249756,
      "loss_g0": 2.477598190307617,
      "loss_g1": 6.536741256713867,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.443326950073242,
      "loss_g0": 6.908851623535156,
      "loss_g1": 14.569906234741211,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.7114224433898926,
      "loss_g0": 3.721451759338379,
      "loss_g1": 1.677653431892395,
      "q_g0": 0.01652262918651104,
      "q_g1": 0.9834774136543274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.973426818847656,
      "loss_g0": 10.171499252319336,
      "loss_g1": 9.970098495483398,
      "q_g0": 0.016522642225027084,
      "q_g1": 0.9834772944450378,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.63209342956543,
      "loss_g0": 16.63209342956543,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.215186595916748,
      "loss_g0": 3.1883015632629395,
      "loss_g1": 4.815472602844238,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 27.506486892700195,
      "loss_g0": 27.877653121948242,
      "loss_g1": 5.413227081298828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.862437725067139,
      "loss_g0": 5.848121643066406,
      "loss_g1": 6.714574813842773,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522511839866638,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 17.008535385131836,
      "loss_g0": 17.213211059570312,
      "loss_g1": 4.825412750244141,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.8332223892211914,
      "loss_g0": 2.8120529651641846,
      "loss_g1": 4.093300819396973,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522550955414772,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.941265106201172,
      "loss_g0": 23.250934600830078,
      "loss_g1": 4.508566379547119,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.8345112800598145,
      "loss_g0": 7.935787200927734,
      "loss_g1": 1.8061827421188354,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.374030113220215,
      "loss_g0": 4.346440315246582,
      "loss_g1": 6.016260147094727,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522590070962906,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.863435745239258,
      "loss_g0": 11.043193817138672,
      "loss_g1": 0.1635003536939621,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.359750747680664,
      "loss_g0": 10.464151382446289,
      "loss_g1": 4.145419120788574,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.341570854187012,
      "loss_g0": 10.381065368652344,
      "loss_g1": 7.990708351135254,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.070825099945068,
      "loss_g0": 5.107769966125488,
      "loss_g1": 2.871721029281616,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.88698673248291,
      "loss_g0": 5.9664716720581055,
      "loss_g1": 1.1557135581970215,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.321362495422363,
      "loss_g0": 10.409988403320312,
      "loss_g1": 5.045990943908691,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.7901687622070312,
      "loss_g0": 1.7948206663131714,
      "loss_g1": 1.513269305229187,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.0483975410461426,
      "loss_g0": 3.0483975410461426,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.283703327178955,
      "loss_g0": 7.075299263000488,
      "loss_g1": 19.688690185546875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.67234992980957,
      "loss_g0": 8.44930648803711,
      "loss_g1": 21.947566986083984,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523806378245354,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.416522979736328,
      "loss_g0": 25.416522979736328,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.569648742675781,
      "loss_g0": 12.538911819458008,
      "loss_g1": 14.399208068847656,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 38.4459342956543,
      "loss_g0": 38.5445442199707,
      "loss_g1": 32.57632827758789,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.387825965881348,
      "loss_g0": 12.174445152282715,
      "loss_g1": 25.0880184173584,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.01652374491095543,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 39.5777702331543,
      "loss_g0": 40.06843566894531,
      "loss_g1": 10.371379852294922,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 22.986570358276367,
      "loss_g0": 23.08992576599121,
      "loss_g1": 16.834396362304688,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 28.802404403686523,
      "loss_g0": 29.165328979492188,
      "loss_g1": 7.199747085571289,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 31.832653045654297,
      "loss_g0": 32.10759353637695,
      "loss_g1": 15.467056274414062,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 25.113285064697266,
      "loss_g0": 25.260114669799805,
      "loss_g1": 16.373445510864258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 30.401037216186523,
      "loss_g0": 30.776954650878906,
      "loss_g1": 8.024869918823242,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.083769798278809,
      "loss_g0": 0.0,
      "loss_g1": 9.083769798278809,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.21744155883789,
      "loss_g0": 16.21744155883789,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 27.00750732421875,
      "loss_g0": 27.309677124023438,
      "loss_g1": 9.021220207214355,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 23.519987106323242,
      "loss_g0": 23.615537643432617,
      "loss_g1": 17.83250617980957,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.423855781555176,
      "loss_g0": 14.499048233032227,
      "loss_g1": 9.948101997375488,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.7356181144714355,
      "loss_g0": 5.6711745262146,
      "loss_g1": 9.571446418762207,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652282290160656,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 49.93038558959961,
      "loss_g0": 50.549747467041016,
      "loss_g1": 13.063647270202637,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.305810928344727,
      "loss_g0": 18.393402099609375,
      "loss_g1": 13.092065811157227,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.533502578735352,
      "loss_g0": 5.336779594421387,
      "loss_g1": 17.24233627319336,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523635014891624,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.865984916687012,
      "loss_g0": 11.865984916687012,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 59.608150482177734,
      "loss_g0": 60.5184326171875,
      "loss_g1": 5.424555778503418,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.443708419799805,
      "loss_g0": 0.0,
      "loss_g1": 14.443708419799805,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.001812934875488,
      "loss_g0": 8.927605628967285,
      "loss_g1": 13.070259094238281,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.088520050048828,
      "loss_g0": 7.748573303222656,
      "loss_g1": 5.043830394744873,
      "q_g0": 0.01652269810438156,
      "q_g1": 0.9834772944450378,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.3376946449279785,
      "loss_g0": 24.22915267944336,
      "loss_g1": 7.053882598876953,
      "q_g0": 0.016524456441402435,
      "q_g1": 0.9834755659103394,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.641841888427734,
      "loss_g0": 12.782188415527344,
      "loss_g1": 19.757091522216797,
      "q_g0": 0.01652374491095543,
      "q_g1": 0.9834762811660767,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.602054595947266,
      "loss_g0": 9.967324256896973,
      "loss_g1": 9.595918655395508,
      "q_g0": 0.016523784026503563,
      "q_g1": 0.9834762811660767,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 13.539743423461914,
      "loss_g0": 8.625348091125488,
      "loss_g1": 13.622309684753418,
      "q_g0": 0.01652327924966812,
      "q_g1": 0.9834766983985901,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.163785934448242,
      "loss_g0": 7.9531965255737305,
      "loss_g1": 6.133721828460693,
      "q_g0": 0.016523465514183044,
      "q_g1": 0.9834765195846558,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.703198432922363,
      "loss_g0": 11.28590202331543,
      "loss_g1": 6.626201629638672,
      "q_g0": 0.0165239367634058,
      "q_g1": 0.9834760427474976,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.1260249614715576,
      "loss_g0": 11.04995346069336,
      "loss_g1": 2.9928836822509766,
      "q_g0": 0.01652476005256176,
      "q_g1": 0.9834752082824707,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.397655487060547,
      "loss_g0": 7.99182653427124,
      "loss_g1": 3.320460081100464,
      "q_g0": 0.016525238752365112,
      "q_g1": 0.9834747910499573,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.674019813537598,
      "loss_g0": 5.468594551086426,
      "loss_g1": 11.778284072875977,
      "q_g0": 0.01652458868920803,
      "q_g1": 0.9834754467010498,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.002731323242188,
      "loss_g0": 13.770400047302246,
      "loss_g1": 8.922621726989746,
      "q_g0": 0.016525082290172577,
      "q_g1": 0.983474850654602,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.262903213500977,
      "loss_g0": 3.8457274436950684,
      "loss_g1": 9.353923797607422,
      "q_g0": 0.016524527221918106,
      "q_g1": 0.9834754467010498,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.259481430053711,
      "loss_g0": 11.990242004394531,
      "loss_g1": 10.230400085449219,
      "q_g0": 0.016524706035852432,
      "q_g1": 0.9834753274917603,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.011343955993652,
      "loss_g0": 0.0,
      "loss_g1": 5.011343955993652,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.500063419342041,
      "loss_g0": 4.833829879760742,
      "loss_g1": 4.4944562911987305,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.320061683654785,
      "loss_g0": 11.378684997558594,
      "loss_g1": 7.251874923706055,
      "q_g0": 0.01652284525334835,
      "q_g1": 0.9834771752357483,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.536252021789551,
      "loss_g0": 6.536252021789551,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.42658519744873,
      "loss_g0": 8.229519844055176,
      "loss_g1": 20.156652450561523,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.129758834838867,
      "loss_g0": 12.285999298095703,
      "loss_g1": 2.829695701599121,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.397385597229004,
      "loss_g0": 6.394774436950684,
      "loss_g1": 6.552811622619629,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.532756805419922,
      "loss_g0": 16.532756805419922,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.0148024559021,
      "loss_g0": 6.0815300941467285,
      "loss_g1": 2.0429012775421143,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.763931274414062,
      "loss_g0": 17.01428985595703,
      "loss_g1": 1.8615825176239014,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 0.7350258827209473,
      "loss_g0": 0.6745772361755371,
      "loss_g1": 4.333080768585205,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652279868721962,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 0.7002982497215271,
      "loss_g0": 0.6948031783103943,
      "loss_g1": 1.0273758172988892,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522839665412903,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 37.66142272949219,
      "loss_g0": 37.762081146240234,
      "loss_g1": 31.669713973999023,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 29.54228973388672,
      "loss_g0": 29.54228973388672,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 29.175264358520508,
      "loss_g0": 29.51228141784668,
      "loss_g1": 9.11473274230957,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 23.88471031188965,
      "loss_g0": 23.88471031188965,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 24.605764389038086,
      "loss_g0": 24.083738327026367,
      "loss_g1": 55.67869567871094,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.853103637695312,
      "loss_g0": 9.62228775024414,
      "loss_g1": 23.590938568115234,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.016523851081728935,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.965670585632324,
      "loss_g0": 15.82176399230957,
      "loss_g1": 24.530338287353516,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524743288755417,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 14.24209976196289,
      "loss_g0": 14.345918655395508,
      "loss_g1": 8.063011169433594,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524100676178932,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.088410377502441,
      "loss_g0": 10.063864707946777,
      "loss_g1": 11.549222946166992,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.01652424782514572,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 46.109771728515625,
      "loss_g0": 46.42090606689453,
      "loss_g1": 27.589984893798828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.598233222961426,
      "loss_g0": 12.659788131713867,
      "loss_g1": 8.93426513671875,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.412874221801758,
      "loss_g0": 0.0,
      "loss_g1": 16.412874221801758,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 24.582014083862305,
      "loss_g0": 19.003952026367188,
      "loss_g1": 24.675724029541016,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 37.299232482910156,
      "loss_g0": 0.0,
      "loss_g1": 37.299232482910156,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.438370704650879,
      "loss_g0": 8.865324020385742,
      "loss_g1": 9.447998046875,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.598721504211426,
      "loss_g0": 24.34967803955078,
      "loss_g1": 7.317275047302246,
      "q_g0": 0.016524160280823708,
      "q_g1": 0.9834758043289185,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.746648788452148,
      "loss_g0": 4.325238227844238,
      "loss_g1": 12.88813591003418,
      "q_g0": 0.016523286700248718,
      "q_g1": 0.9834766983985901,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.492162704467773,
      "loss_g0": 14.7052001953125,
      "loss_g1": 19.57258415222168,
      "q_g0": 0.016522789373993874,
      "q_g1": 0.9834771752357483,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.304585456848145,
      "loss_g0": 26.085899353027344,
      "loss_g1": 9.022623062133789,
      "q_g0": 0.016524534672498703,
      "q_g1": 0.9834754467010498,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.146127700805664,
      "loss_g0": 0.0,
      "loss_g1": 11.146127700805664,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.76247787475586,
      "loss_g0": 15.827788352966309,
      "loss_g1": 8.643779754638672,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.439968109130859,
      "loss_g0": 9.287107467651367,
      "loss_g1": 7.408935546875,
      "q_g0": 0.016522621735930443,
      "q_g1": 0.9834774136543274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.37629222869873,
      "loss_g0": 48.49292755126953,
      "loss_g1": 7.702151775360107,
      "q_g0": 0.01652679033577442,
      "q_g1": 0.9834732413291931,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.1538777351379395,
      "loss_g0": 7.821197986602783,
      "loss_g1": 4.092248439788818,
      "q_g0": 0.016527168452739716,
      "q_g1": 0.9834728240966797,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.808622360229492,
      "loss_g0": 8.577510833740234,
      "loss_g1": 19.997344970703125,
      "q_g0": 0.016525998711586,
      "q_g1": 0.9834739565849304,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.884919166564941,
      "loss_g0": 14.340984344482422,
      "loss_g1": 6.759624004364014,
      "q_g0": 0.01652677357196808,
      "q_g1": 0.9834732413291931,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.462461471557617,
      "loss_g0": 0.0,
      "loss_g1": 16.462461471557617,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 19.164472579956055,
      "loss_g0": 8.669511795043945,
      "loss_g1": 19.340787887573242,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.581478118896484,
      "loss_g0": 21.294532775878906,
      "loss_g1": 11.418289184570312,
      "q_g0": 0.016523433849215508,
      "q_g1": 0.9834766387939453,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.33407211303711,
      "loss_g0": 11.811342239379883,
      "loss_g1": 12.342853546142578,
      "q_g0": 0.01652337983250618,
      "q_g1": 0.9834766387939453,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.366740226745605,
      "loss_g0": 25.650970458984375,
      "loss_g1": 12.143532752990723,
      "q_g0": 0.016524752601981163,
      "q_g1": 0.9834752082824707,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.600353240966797,
      "loss_g0": 0.0,
      "loss_g1": 6.600353240966797,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.5852816104888916,
      "loss_g0": 0.0,
      "loss_g1": 3.5852816104888916,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 7.90184211730957,
      "loss_g0": 7.90184211730957,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.8339433670043945,
      "loss_g0": 4.766307830810547,
      "loss_g1": 8.85988998413086,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.254011154174805,
      "loss_g0": 8.254011154174805,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 10.308173179626465,
      "loss_g0": 10.34957504272461,
      "loss_g1": 7.84376859664917,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.554582595825195,
      "loss_g0": 11.554582595825195,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 12.649186134338379,
      "loss_g0": 12.68018627166748,
      "loss_g1": 10.803899765014648,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.622961044311523,
      "loss_g0": 9.737198829650879,
      "loss_g1": 2.823066473007202,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.432641983032227,
      "loss_g0": 6.468153953552246,
      "loss_g1": 4.3188300132751465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.118479609489441,
      "loss_g0": 1.0094044208526611,
      "loss_g1": 7.610779762268066,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523092985153198,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.564513206481934,
      "loss_g0": 9.631699562072754,
      "loss_g1": 5.565398693084717,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652267388999462,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.14097785949707,
      "loss_g0": 18.423002243041992,
      "loss_g1": 1.3537137508392334,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.487858295440674,
      "loss_g0": 5.472232818603516,
      "loss_g1": 6.417935371398926,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522521153092384,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 3.2169322967529297,
      "loss_g0": 0.0,
      "loss_g1": 3.2169322967529297,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.175544023513794,
      "loss_g0": 2.5813443660736084,
      "loss_g1": 1.1519265174865723,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 4.560724258422852,
      "loss_g0": 5.942519664764404,
      "loss_g1": 4.537509918212891,
      "q_g0": 0.016522565856575966,
      "q_g1": 0.9834774136543274,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.9975473880767822,
      "loss_g0": 1.0865063667297363,
      "loss_g1": 2.012852907180786,
      "q_g0": 0.016522465273737907,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 16.559295654296875,
      "loss_g0": 4.3978142738342285,
      "loss_g1": 16.763608932495117,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 40.66401290893555,
      "loss_g0": 0.0,
      "loss_g1": 40.66401290893555,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.29686164855957,
      "loss_g0": 12.512629508972168,
      "loss_g1": 18.39403533935547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 27.523174285888672,
      "loss_g0": 34.70224380493164,
      "loss_g1": 27.40256118774414,
      "q_g0": 0.016523171216249466,
      "q_g1": 0.9834768176078796,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.359368324279785,
      "loss_g0": 25.793983459472656,
      "loss_g1": 15.184046745300293,
      "q_g0": 0.01652425527572632,
      "q_g1": 0.9834756851196289,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.133459091186523,
      "loss_g0": 17.93302345275879,
      "loss_g1": 18.136825561523438,
      "q_g0": 0.016524240374565125,
      "q_g1": 0.9834758043289185,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.89486312866211,
      "loss_g0": 24.226043701171875,
      "loss_g1": 18.805286407470703,
      "q_g0": 0.016524799168109894,
      "q_g1": 0.9834752082824707,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 27.71434783935547,
      "loss_g0": 8.591614723205566,
      "loss_g1": 28.03561782836914,
      "q_g0": 0.016522813588380814,
      "q_g1": 0.9834771752357483,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.1691951751709,
      "loss_g0": 19.60263442993164,
      "loss_g1": 18.145111083984375,
      "q_g0": 0.016522971913218498,
      "q_g1": 0.9834770560264587,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 8.667179107666016,
      "loss_g0": 7.076107978820801,
      "loss_g1": 8.693909645080566,
      "q_g0": 0.016522806137800217,
      "q_g1": 0.9834771752357483,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.845534801483154,
      "loss_g0": 9.573429107666016,
      "loss_g1": 6.799704074859619,
      "q_g0": 0.016523092985153198,
      "q_g1": 0.9834769368171692,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.756668090820312,
      "loss_g0": 7.477593421936035,
      "loss_g1": 9.794957160949707,
      "q_g0": 0.016522862017154694,
      "q_g1": 0.9834771752357483,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 9.03696060180664,
      "loss_g0": 10.4605712890625,
      "loss_g1": 9.013043403625488,
      "q_g0": 0.016523005440831184,
      "q_g1": 0.9834769368171692,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 18.8456974029541,
      "loss_g0": 25.644338607788086,
      "loss_g1": 18.731470108032227,
      "q_g0": 0.016523711383342743,
      "q_g1": 0.9834762811660767,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 11.20383358001709,
      "loss_g0": 14.867959976196289,
      "loss_g1": 11.142269134521484,
      "q_g0": 0.016524093225598335,
      "q_g1": 0.983475923538208,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.9641366004943848,
      "loss_g0": 15.164497375488281,
      "loss_g1": 1.7423290014266968,
      "q_g0": 0.01652546413242817,
      "q_g1": 0.9834745526313782,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 5.468133449554443,
      "loss_g0": 9.292262077331543,
      "loss_g1": 5.40387487411499,
      "q_g0": 0.016525868326425552,
      "q_g1": 0.9834741950035095,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 1.5433317422866821,
      "loss_g0": 2.7886290550231934,
      "loss_g1": 1.5224061012268066,
      "q_g0": 0.016525998711586,
      "q_g1": 0.9834739565849304,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 6.407381057739258,
      "loss_g0": 6.407381057739258,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 2.2598915100097656,
      "loss_g0": 2.271566390991211,
      "loss_g1": 1.5649685859680176,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "eval_cer": 0.05948064101064452,
      "eval_loss": 13.978235244750977,
      "eval_runtime": 80.5988,
      "eval_samples_per_second": 20.794,
      "eval_steps_per_second": 5.199,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 15.773266792297363,
      "loss_g0": 15.778036117553711,
      "loss_g1": 15.489359855651855,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1300
    },
    {
      "epoch": 6.598984771573604,
      "gdro_loss": 36.36517333984375,
      "loss_g0": 0.0,
      "loss_g1": 36.36517333984375,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1300
    },
    {
      "epoch": 7.106598984771574,
      "grad_norm": 47.619503021240234,
      "learning_rate": 0.00014316577540106953,
      "loss": 26.9286,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.642637252807617,
      "loss_g0": 17.36247444152832,
      "loss_g1": 19.68094253540039,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.583605766296387,
      "loss_g0": 19.601896286010742,
      "loss_g1": 14.499295234680176,
      "q_g0": 0.01652294583618641,
      "q_g1": 0.9834770560264587,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 67.02316284179688,
      "loss_g0": 38.06760787963867,
      "loss_g1": 67.5096206665039,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.066701889038086,
      "loss_g0": 23.80095100402832,
      "loss_g1": 8.819143295288086,
      "q_g0": 0.016523951664566994,
      "q_g1": 0.9834760427474976,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.588096618652344,
      "loss_g0": 33.97517776489258,
      "loss_g1": 23.413564682006836,
      "q_g0": 0.016525031998753548,
      "q_g1": 0.9834749698638916,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.563386917114258,
      "loss_g0": 14.05142593383789,
      "loss_g1": 19.65599822998047,
      "q_g0": 0.016524456441402435,
      "q_g1": 0.9834755659103394,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.870859146118164,
      "loss_g0": 8.13499927520752,
      "loss_g1": 24.135229110717773,
      "q_g0": 0.016522813588380814,
      "q_g1": 0.9834771752357483,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.842845916748047,
      "loss_g0": 15.563541412353516,
      "loss_g1": 11.780335426330566,
      "q_g0": 0.016523201018571854,
      "q_g1": 0.9834768176078796,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 35.008731842041016,
      "loss_g0": 17.994483947753906,
      "loss_g1": 35.29457092285156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.851600646972656,
      "loss_g0": 19.311586380004883,
      "loss_g1": 22.91107177734375,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.506534576416016,
      "loss_g0": 13.900386810302734,
      "loss_g1": 19.600717544555664,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.701921463012695,
      "loss_g0": 9.416653633117676,
      "loss_g1": 16.82431411743164,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.58414077758789,
      "loss_g0": 24.70918083190918,
      "loss_g1": 21.53163719177246,
      "q_g0": 0.016522753983736038,
      "q_g1": 0.9834772944450378,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.014198303222656,
      "loss_g0": 24.076967239379883,
      "loss_g1": 20.96274185180664,
      "q_g0": 0.016523078083992004,
      "q_g1": 0.9834769368171692,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.083677291870117,
      "loss_g0": 10.083677291870117,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 33.67915344238281,
      "loss_g0": 33.67915344238281,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.87603235244751,
      "loss_g0": 5.525378227233887,
      "loss_g1": 26.748275756835938,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.893215656280518,
      "loss_g0": 5.833512306213379,
      "loss_g1": 9.446897506713867,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652279868721962,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.814327716827393,
      "loss_g0": 6.675196170806885,
      "loss_g1": 15.095304489135742,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.016523659229278564,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.164548873901367,
      "loss_g0": 7.109729290008545,
      "loss_g1": 10.42728328704834,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524000093340874,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.712018966674805,
      "loss_g0": 5.65533971786499,
      "loss_g1": 9.085372924804688,
      "q_g0": 0.9834756851196289,
      "q_g1": 0.01652434840798378,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 36.27822494506836,
      "loss_g0": 36.75981521606445,
      "loss_g1": 7.612140655517578,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.23844337463379,
      "loss_g0": 21.357112884521484,
      "loss_g1": 14.174731254577637,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.155941486358643,
      "loss_g0": 6.185685157775879,
      "loss_g1": 4.385479927062988,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.803081035614014,
      "loss_g0": 0.0,
      "loss_g1": 5.803081035614014,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.600187301635742,
      "loss_g0": 0.0,
      "loss_g1": 12.600187301635742,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.468534469604492,
      "loss_g0": 17.654743194580078,
      "loss_g1": 6.280605792999268,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.449302911758423,
      "loss_g0": 19.058040618896484,
      "loss_g1": 2.1702466011047363,
      "q_g0": 0.016524145379662514,
      "q_g1": 0.9834758043289185,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.9458208084106445,
      "loss_g0": 6.9458208084106445,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.020543098449707,
      "loss_g0": 10.1249418258667,
      "loss_g1": 3.8063437938690186,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.18069076538086,
      "loss_g0": 13.302507400512695,
      "loss_g1": 5.929697513580322,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.9947363138198853,
      "loss_g0": 1.9258400201797485,
      "loss_g1": 6.095599174499512,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652284525334835,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.120813369750977,
      "loss_g0": 22.43323516845703,
      "loss_g1": 3.5242536067962646,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.03930950164795,
      "loss_g0": 8.136604309082031,
      "loss_g1": 2.2479641437530518,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.944385528564453,
      "loss_g0": 22.17409324645996,
      "loss_g1": 8.271308898925781,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.739400863647461,
      "loss_g0": 8.619141578674316,
      "loss_g1": 15.89739990234375,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523171216249466,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.2774248123168945,
      "loss_g0": 6.350372314453125,
      "loss_g1": 1.9353801012039185,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.0165227223187685,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.62273645401001,
      "loss_g0": 6.62869930267334,
      "loss_g1": 6.267819404602051,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522690653800964,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.3127975463867188,
      "loss_g0": 3.2953827381134033,
      "loss_g1": 4.349375247955322,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652279868721962,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 24.24785614013672,
      "loss_g0": 24.597911834716797,
      "loss_g1": 3.411113977432251,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 27.128068923950195,
      "loss_g0": 27.128068923950195,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.1962385177612305,
      "loss_g0": 5.154477596282959,
      "loss_g1": 7.682007789611816,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.6643242835998535,
      "loss_g0": 6.72580623626709,
      "loss_g1": 3.004667043685913,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.01511573791504,
      "loss_g0": 16.153079986572266,
      "loss_g1": 7.802903175354004,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.938683271408081,
      "loss_g0": 3.787747859954834,
      "loss_g1": 12.922420501708984,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652335561811924,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.6702880859375,
      "loss_g0": 11.819023132324219,
      "loss_g1": 2.817035675048828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.455421447753906,
      "loss_g0": 13.590710639953613,
      "loss_g1": 5.4024763107299805,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.242886543273926,
      "loss_g0": 5.257972717285156,
      "loss_g1": 4.344897747039795,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.1027326583862305,
      "loss_g0": 0.0,
      "loss_g1": 7.1027326583862305,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.029913902282715,
      "loss_g0": 2.029913902282715,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 53.95389175415039,
      "loss_g0": 54.4206428527832,
      "loss_g1": 26.171016693115234,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.763144493103027,
      "loss_g0": 14.547566413879395,
      "loss_g1": 27.594097137451172,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523759812116623,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.593060493469238,
      "loss_g0": 13.466361999511719,
      "loss_g1": 21.133638381958008,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.016524551436305046,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.445271492004395,
      "loss_g0": 13.432422637939453,
      "loss_g1": 14.209994316101074,
      "q_g0": 0.9834753274917603,
      "q_g1": 0.016524627804756165,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 69.74018096923828,
      "loss_g0": 70.36164093017578,
      "loss_g1": 32.7486572265625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 33.34818649291992,
      "loss_g0": 33.739341735839844,
      "loss_g1": 10.06522274017334,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.422876358032227,
      "loss_g0": 0.0,
      "loss_g1": 15.422876358032227,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.17544174194336,
      "loss_g0": 13.890192031860352,
      "loss_g1": 11.129833221435547,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.30600643157959,
      "loss_g0": 15.4859619140625,
      "loss_g1": 9.202178955078125,
      "q_g0": 0.01652306318283081,
      "q_g1": 0.9834769368171692,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 40.531612396240234,
      "loss_g0": 20.619266510009766,
      "loss_g1": 40.86613845825195,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.30468463897705,
      "loss_g0": 21.11861801147461,
      "loss_g1": 14.190205574035645,
      "q_g0": 0.016523132100701332,
      "q_g1": 0.9834768176078796,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 26.348526000976562,
      "loss_g0": 11.213848114013672,
      "loss_g1": 26.6027889251709,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.357208251953125,
      "loss_g0": 22.551597595214844,
      "loss_g1": 7.101917743682861,
      "q_g0": 0.016524000093340874,
      "q_g1": 0.9834760427474976,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.179306030273438,
      "loss_g0": 15.936991691589355,
      "loss_g1": 18.21697998046875,
      "q_g0": 0.016523759812116623,
      "q_g1": 0.9834762811660767,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.102861404418945,
      "loss_g0": 21.552597045898438,
      "loss_g1": 19.06169891357422,
      "q_g0": 0.016524016857147217,
      "q_g1": 0.9834760427474976,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.744686126708984,
      "loss_g0": 101.39606475830078,
      "loss_g1": 10.237530708312988,
      "q_g0": 0.016533341258764267,
      "q_g1": 0.9834666848182678,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.805891990661621,
      "loss_g0": 14.106098175048828,
      "loss_g1": 14.817655563354492,
      "q_g0": 0.016533263027668,
      "q_g1": 0.9834666848182678,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.061800003051758,
      "loss_g0": 17.296260833740234,
      "loss_g1": 16.041048049926758,
      "q_g0": 0.01653338596224785,
      "q_g1": 0.9834665656089783,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.747270584106445,
      "loss_g0": 7.445690155029297,
      "loss_g1": 6.735528945922852,
      "q_g0": 0.01653345674276352,
      "q_g1": 0.9834665656089783,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.193216323852539,
      "loss_g0": 35.806270599365234,
      "loss_g1": 10.77937126159668,
      "q_g0": 0.016536016017198563,
      "q_g1": 0.9834640622138977,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.367923736572266,
      "loss_g0": 16.778797149658203,
      "loss_g1": 20.42827033996582,
      "q_g0": 0.016535643488168716,
      "q_g1": 0.9834643602371216,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.996987342834473,
      "loss_g0": 13.44681453704834,
      "loss_g1": 8.92216682434082,
      "q_g0": 0.016536099836230278,
      "q_g1": 0.9834639430046082,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.108355522155762,
      "loss_g0": 9.076385498046875,
      "loss_g1": 12.159334182739258,
      "q_g0": 0.016535785049200058,
      "q_g1": 0.9834643006324768,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 0.9354052543640137,
      "loss_g0": 15.195566177368164,
      "loss_g1": 0.6956157088279724,
      "q_g0": 0.016537263989448547,
      "q_g1": 0.9834627509117126,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.03213882446289,
      "loss_g0": 10.587041854858398,
      "loss_g1": 10.022808074951172,
      "q_g0": 0.016537325456738472,
      "q_g1": 0.9834626317024231,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.8354973793029785,
      "loss_g0": 8.73369026184082,
      "loss_g1": 3.753129482269287,
      "q_g0": 0.01653783768415451,
      "q_g1": 0.9834622144699097,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.05739974975586,
      "loss_g0": 27.94420051574707,
      "loss_g1": 7.722941875457764,
      "q_g0": 0.016539908945560455,
      "q_g1": 0.9834601283073425,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.527378082275391,
      "loss_g0": 5.527378082275391,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.4218831062316895,
      "loss_g0": 5.4089674949646,
      "loss_g1": 6.19066047668457,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.744929313659668,
      "loss_g0": 6.601012229919434,
      "loss_g1": 15.310934066772461,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523310914635658,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 32.934532165527344,
      "loss_g0": 33.407188415527344,
      "loss_g1": 4.800172328948975,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.687325477600098,
      "loss_g0": 12.84733772277832,
      "loss_g1": 3.162773609161377,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.915188312530518,
      "loss_g0": 5.862348556518555,
      "loss_g1": 9.060325622558594,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522753983736038,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.924771308898926,
      "loss_g0": 6.928757190704346,
      "loss_g1": 6.687521934509277,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522729769349098,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.292588233947754,
      "loss_g0": 14.476162910461426,
      "loss_g1": 3.3655271530151367,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.327248573303223,
      "loss_g0": 14.414637565612793,
      "loss_g1": 9.125541687011719,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.444693088531494,
      "loss_g0": 5.44495964050293,
      "loss_g1": 5.428836345672607,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522426158189774,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.28499984741211,
      "loss_g0": 10.43520736694336,
      "loss_g1": 1.3440614938735962,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.5817363262176514,
      "loss_g0": 3.6080801486968994,
      "loss_g1": 2.0136544704437256,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.13342809677124,
      "loss_g0": 7.176555633544922,
      "loss_g1": 4.566303253173828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.44021987915039,
      "loss_g0": 18.730470657348633,
      "loss_g1": 1.1633992195129395,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.3931055068969727,
      "loss_g0": 3.123746395111084,
      "loss_g1": 19.424739837646484,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652408577501774,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.410855770111084,
      "loss_g0": 6.4061150550842285,
      "loss_g1": 6.69299840927124,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524117439985275,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.598182678222656,
      "loss_g0": 6.477217674255371,
      "loss_g1": 13.797402381896973,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524866223335266,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.057981252670288,
      "loss_g0": 0.0,
      "loss_g1": 2.057981252670288,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.637415885925293,
      "loss_g0": 1.9348931312561035,
      "loss_g1": 4.682818412780762,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.9528956413269043,
      "loss_g0": 13.478121757507324,
      "loss_g1": 3.7928619384765625,
      "q_g0": 0.016523417085409164,
      "q_g1": 0.9834766387939453,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.007728099822998,
      "loss_g0": 6.924646854400635,
      "loss_g1": 5.9923224449157715,
      "q_g0": 0.01652350276708603,
      "q_g1": 0.9834765195846558,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.38840389251709,
      "loss_g0": 1.4526939392089844,
      "loss_g1": 6.4713263511657715,
      "q_g0": 0.016522984951734543,
      "q_g1": 0.9834770560264587,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.125828266143799,
      "loss_g0": 1.9830876588821411,
      "loss_g1": 2.1282262802124023,
      "q_g0": 0.016522971913218498,
      "q_g1": 0.9834770560264587,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.891773223876953,
      "loss_g0": 0.0,
      "loss_g1": 19.891773223876953,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 90.28753662109375,
      "loss_g0": 23.075557708740234,
      "loss_g1": 91.41669464111328,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.244741439819336,
      "loss_g0": 36.425323486328125,
      "loss_g1": 22.006484985351562,
      "q_g0": 0.01652389205992222,
      "q_g1": 0.9834761619567871,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.993308067321777,
      "loss_g0": 24.220417022705078,
      "loss_g1": 12.804661750793457,
      "q_g0": 0.01652505435049534,
      "q_g1": 0.9834749698638916,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 61.68699645996094,
      "loss_g0": 38.14895248413086,
      "loss_g1": 62.082435607910156,
      "q_g0": 0.0165226049721241,
      "q_g1": 0.9834774136543274,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 27.4686336517334,
      "loss_g0": 18.42134666442871,
      "loss_g1": 27.620628356933594,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.316222190856934,
      "loss_g0": 4.430387496948242,
      "loss_g1": 12.44870376586914,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.827306747436523,
      "loss_g0": 10.221647262573242,
      "loss_g1": 12.871081352233887,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 24.81669807434082,
      "loss_g0": 59.49774932861328,
      "loss_g1": 24.233928680419922,
      "q_g0": 0.016526028513908386,
      "q_g1": 0.9834739565849304,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.711952209472656,
      "loss_g0": 26.585765838623047,
      "loss_g1": 15.529218673706055,
      "q_g0": 0.016527162864804268,
      "q_g1": 0.9834728240966797,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.312368392944336,
      "loss_g0": 7.34794807434082,
      "loss_g1": 15.446203231811523,
      "q_g0": 0.01652633771300316,
      "q_g1": 0.9834735989570618,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.450095176696777,
      "loss_g0": 19.019386291503906,
      "loss_g1": 7.255670070648193,
      "q_g0": 0.016527540981769562,
      "q_g1": 0.983472466468811,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.647663116455078,
      "loss_g0": 14.15876293182373,
      "loss_g1": 4.48781681060791,
      "q_g0": 0.016528533771634102,
      "q_g1": 0.9834713935852051,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.6004416942596436,
      "loss_g0": 18.675006866455078,
      "loss_g1": 2.3302595615386963,
      "q_g0": 0.01653021015226841,
      "q_g1": 0.9834697842597961,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.464759826660156,
      "loss_g0": 18.648109436035156,
      "loss_g1": 23.54571533203125,
      "q_g0": 0.016529710963368416,
      "q_g1": 0.9834702610969543,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.222654342651367,
      "loss_g0": 0.0,
      "loss_g1": 12.222654342651367,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 30.320117950439453,
      "loss_g0": 10.403003692626953,
      "loss_g1": 30.654726028442383,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.100605010986328,
      "loss_g0": 16.100605010986328,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.89453887939453,
      "loss_g0": 18.106956481933594,
      "loss_g1": 5.250699520111084,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.4558727741241455,
      "loss_g0": 3.4214141368865967,
      "loss_g1": 5.506953716278076,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.407957077026367,
      "loss_g0": 11.407957077026367,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.794147491455078,
      "loss_g0": 10.872162818908691,
      "loss_g1": 6.150340557098389,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 65.0884780883789,
      "loss_g0": 66.09507751464844,
      "loss_g1": 5.1718268394470215,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.080421447753906,
      "loss_g0": 0.0,
      "loss_g1": 20.080421447753906,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.577774047851562,
      "loss_g0": 0.0,
      "loss_g1": 13.577774047851562,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.984940528869629,
      "loss_g0": 0.0,
      "loss_g1": 5.984940528869629,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.90709114074707,
      "loss_g0": 3.2892346382141113,
      "loss_g1": 18.152671813964844,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.109909057617188,
      "loss_g0": 5.093764781951904,
      "loss_g1": 21.378978729248047,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.375473499298096,
      "loss_g0": 10.959383010864258,
      "loss_g1": 4.264859199523926,
      "q_g0": 0.016523106023669243,
      "q_g1": 0.9834768176078796,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.98104190826416,
      "loss_g0": 10.630090713500977,
      "loss_g1": 6.91973352432251,
      "q_g0": 0.016523487865924835,
      "q_g1": 0.9834765195846558,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.167969703674316,
      "loss_g0": 9.339527130126953,
      "loss_g1": 8.148285865783691,
      "q_g0": 0.016523605212569237,
      "q_g1": 0.9834764003753662,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.8415727615356445,
      "loss_g0": 28.34453582763672,
      "loss_g1": 5.463441371917725,
      "q_g0": 0.01652594469487667,
      "q_g1": 0.98347407579422,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.357632160186768,
      "loss_g0": 2.525278329849243,
      "loss_g1": 4.388422012329102,
      "q_g0": 0.01652575097978115,
      "q_g1": 0.9834743142127991,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.6049435138702393,
      "loss_g0": 5.07110595703125,
      "loss_g1": 2.563502788543701,
      "q_g0": 0.016526008024811745,
      "q_g1": 0.9834739565849304,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.348780632019043,
      "loss_g0": 11.196022033691406,
      "loss_g1": 5.250521659851074,
      "q_g0": 0.016526618972420692,
      "q_g1": 0.9834733605384827,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.998462677001953,
      "loss_g0": 14.09398078918457,
      "loss_g1": 21.114482879638672,
      "q_g0": 0.016525907441973686,
      "q_g1": 0.98347407579422,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.787979125976562,
      "loss_g0": 16.176496505737305,
      "loss_g1": 11.7142333984375,
      "q_g0": 0.0165263544768095,
      "q_g1": 0.9834735989570618,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.532012939453125,
      "loss_g0": 5.963838577270508,
      "loss_g1": 12.642380714416504,
      "q_g0": 0.016525674611330032,
      "q_g1": 0.9834743142127991,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.499983310699463,
      "loss_g0": 13.85653018951416,
      "loss_g1": 7.393167495727539,
      "q_g0": 0.01652633771300316,
      "q_g1": 0.9834735989570618,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.442739009857178,
      "loss_g0": 24.604904174804688,
      "loss_g1": 6.137506008148193,
      "q_g0": 0.016528230160474777,
      "q_g1": 0.9834717512130737,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.381799697875977,
      "loss_g0": 10.381799697875977,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.834654808044434,
      "loss_g0": 10.951240539550781,
      "loss_g1": 3.8949968814849854,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.260830402374268,
      "loss_g0": 7.261888027191162,
      "loss_g1": 7.1978583335876465,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.9985625743865967,
      "loss_g0": 2.8679046630859375,
      "loss_g1": 10.775429725646973,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.01652323268353939,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.333900451660156,
      "loss_g0": 14.506855010986328,
      "loss_g1": 4.038931846618652,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.758063316345215,
      "loss_g0": 8.860092163085938,
      "loss_g1": 2.6849169731140137,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.15727710723877,
      "loss_g0": 10.162172317504883,
      "loss_g1": 9.8659086227417,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.9715840816497803,
      "loss_g0": 2.999124526977539,
      "loss_g1": 1.3322657346725464,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.659520149230957,
      "loss_g0": 2.659520149230957,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.447741508483887,
      "loss_g0": 8.587762832641602,
      "loss_g1": 0.11310871690511703,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 32.79783630371094,
      "loss_g0": 0.0,
      "loss_g1": 32.79783630371094,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 24.918315887451172,
      "loss_g0": 27.47261619567871,
      "loss_g1": 24.875402450561523,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 30.678010940551758,
      "loss_g0": 27.349369049072266,
      "loss_g1": 30.733932495117188,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.990697860717773,
      "loss_g0": 34.970191955566406,
      "loss_g1": 21.77262306213379,
      "q_g0": 0.01652376726269722,
      "q_g1": 0.9834762811660767,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.657304763793945,
      "loss_g0": 24.57118797302246,
      "loss_g1": 22.62514877319336,
      "q_g0": 0.016523968428373337,
      "q_g1": 0.9834760427474976,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 33.81041717529297,
      "loss_g0": 18.141828536987305,
      "loss_g1": 34.073646545410156,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.50497817993164,
      "loss_g0": 21.765443801879883,
      "loss_g1": 15.399798393249512,
      "q_g0": 0.016523070633411407,
      "q_g1": 0.9834769368171692,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 33.60469436645508,
      "loss_g0": 19.594345092773438,
      "loss_g1": 33.84006881713867,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 25.100955963134766,
      "loss_g0": 24.33985710144043,
      "loss_g1": 25.11374282836914,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.955323219299316,
      "loss_g0": 35.175533294677734,
      "loss_g1": 8.514750480651855,
      "q_g0": 0.0165251474827528,
      "q_g1": 0.983474850654602,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.358875274658203,
      "loss_g0": 0.0,
      "loss_g1": 20.358875274658203,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.1546220779418945,
      "loss_g0": 21.425151824951172,
      "loss_g1": 6.914876937866211,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 24.649723052978516,
      "loss_g0": 0.0,
      "loss_g1": 24.649723052978516,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 21.663307189941406,
      "loss_g0": 21.663307189941406,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.557624816894531,
      "loss_g0": 9.628437042236328,
      "loss_g1": 5.342591762542725,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.539005279541016,
      "loss_g0": 20.684478759765625,
      "loss_g1": 11.879768371582031,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.5879693031311035,
      "loss_g0": 7.490077018737793,
      "loss_g1": 13.414684295654297,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523031517863274,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.616022109985352,
      "loss_g0": 11.616022109985352,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.525606155395508,
      "loss_g0": 18.2926025390625,
      "loss_g1": 32.394874572753906,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.637839317321777,
      "loss_g0": 14.637839317321777,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.42363166809082,
      "loss_g0": 8.487828254699707,
      "loss_g1": 4.602436065673828,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 25.188758850097656,
      "loss_g0": 25.3748722076416,
      "loss_g1": 14.11056137084961,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.21484088897705,
      "loss_g0": 13.315567970275879,
      "loss_g1": 7.219180583953857,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.546977996826172,
      "loss_g0": 4.57844352722168,
      "loss_g1": 2.6740195751190186,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.38222312927246,
      "loss_g0": 23.571258544921875,
      "loss_g1": 12.130123138427734,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.735272407531738,
      "loss_g0": 5.519117832183838,
      "loss_g1": 18.600542068481445,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523759812116623,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 28.002887725830078,
      "loss_g0": 28.308563232421875,
      "loss_g1": 9.807884216308594,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.454103708267212,
      "loss_g0": 3.3025033473968506,
      "loss_g1": 12.477413177490234,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523364931344986,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.794746398925781,
      "loss_g0": 4.754650115966797,
      "loss_g1": 7.181249141693115,
      "q_g0": 0.9834764003753662,
      "q_g1": 0.01652361825108528,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.01278829574585,
      "loss_g0": 5.915900707244873,
      "loss_g1": 11.779267311096191,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524218022823334,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.657997131347656,
      "loss_g0": 8.741090774536133,
      "loss_g1": 3.712325096130371,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523705795407295,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.873807907104492,
      "loss_g0": 7.8750996589660645,
      "loss_g1": 7.796931266784668,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.0165236983448267,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.443321228027344,
      "loss_g0": 4.181586265563965,
      "loss_g1": 20.02002716064453,
      "q_g0": 0.9834746718406677,
      "q_g1": 0.01652531512081623,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.91029930114746,
      "loss_g0": 17.0892391204834,
      "loss_g1": 6.260313034057617,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524210572242737,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.3582489490509033,
      "loss_g0": 3.3149967193603516,
      "loss_g1": 5.9324445724487305,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524480655789375,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.848348617553711,
      "loss_g0": 3.7814624309539795,
      "loss_g1": 7.82907247543335,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.016524890437722206,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.1741704940795898,
      "loss_g0": 1.127842903137207,
      "loss_g1": 3.9312949180603027,
      "q_g0": 0.983474850654602,
      "q_g1": 0.016525179147720337,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.1364688873291,
      "loss_g0": 22.482040405273438,
      "loss_g1": 1.567527413368225,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523046419024467,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.164180755615234,
      "loss_g0": 11.21381950378418,
      "loss_g1": 8.209505081176758,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522737219929695,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.8421051502227783,
      "loss_g0": 2.750328540802002,
      "loss_g1": 8.304696083068848,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523310914635658,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 71.49044036865234,
      "loss_g0": 72.64068603515625,
      "loss_g1": 3.0233707427978516,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.225971221923828,
      "loss_g0": 8.321770668029785,
      "loss_g1": 2.523634433746338,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.7982677221298218,
      "loss_g0": 1.80888831615448,
      "loss_g1": 1.166088581085205,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.30155611038208,
      "loss_g0": 4.299333572387695,
      "loss_g1": 4.433843612670898,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.01652243360877037,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.257045269012451,
      "loss_g0": 4.31797981262207,
      "loss_g1": 0.6299931406974792,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.1585123538970947,
      "loss_g0": 2.1605772972106934,
      "loss_g1": 2.035593271255493,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.645106315612793,
      "loss_g0": 3.646231174468994,
      "loss_g1": 3.578148365020752,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.6101473569869995,
      "loss_g0": 1.5743634700775146,
      "loss_g1": 3.740114688873291,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.03151798248291,
      "loss_g0": 5.890133857727051,
      "loss_g1": 14.446642875671387,
      "q_g0": 0.9834765195846558,
      "q_g1": 0.016523517668247223,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.1061339378356934,
      "loss_g0": 2.1306142807006836,
      "loss_g1": 0.6490588784217834,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.016523372381925583,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 35.03140640258789,
      "loss_g0": 34.81077194213867,
      "loss_g1": 48.16242218017578,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652473583817482,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.483858108520508,
      "loss_g0": 22.600618362426758,
      "loss_g1": 15.53441333770752,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524016857147217,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.656829833984375,
      "loss_g0": 12.250905990600586,
      "loss_g1": 36.812896728515625,
      "q_g0": 0.9834734797477722,
      "q_g1": 0.01652652584016323,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.600936889648438,
      "loss_g0": 17.50290298461914,
      "loss_g1": 23.43473243713379,
      "q_g0": 0.9834729433059692,
      "q_g1": 0.01652713306248188,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 38.67342758178711,
      "loss_g0": 39.045650482177734,
      "loss_g1": 16.52060317993164,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.01652483083307743,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.97434425354004,
      "loss_g0": 19.97710418701172,
      "loss_g1": 19.810075759887695,
      "q_g0": 0.9834752082824707,
      "q_g1": 0.016524814069271088,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.775494575500488,
      "loss_g0": 15.769739151000977,
      "loss_g1": 16.11800193786621,
      "q_g0": 0.9834750890731812,
      "q_g1": 0.01652486063539982,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.860544204711914,
      "loss_g0": 18.974918365478516,
      "loss_g1": 12.053193092346191,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524145379662514,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 33.786766052246094,
      "loss_g0": 34.182796478271484,
      "loss_g1": 10.213468551635742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.092222213745117,
      "loss_g0": 22.20511817932129,
      "loss_g1": 15.37224006652832,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.266977310180664,
      "loss_g0": 17.396774291992188,
      "loss_g1": 9.541004180908203,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.7751522064209,
      "loss_g0": 16.98678970336914,
      "loss_g1": 4.1776957511901855,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.67719841003418,
      "loss_g0": 8.621294021606445,
      "loss_g1": 12.004741668701172,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522767022252083,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.62607192993164,
      "loss_g0": 9.085189819335938,
      "loss_g1": 102.2895278930664,
      "q_g0": 0.9834677577018738,
      "q_g1": 0.0165322944521904,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.445882320404053,
      "loss_g0": 6.218988418579102,
      "loss_g1": 19.942106246948242,
      "q_g0": 0.9834663271903992,
      "q_g1": 0.01653369516134262,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.8138480186462402,
      "loss_g0": 2.439093589782715,
      "loss_g1": 25.102039337158203,
      "q_g0": 0.9834640622138977,
      "q_g1": 0.016536006703972816,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.33510684967041,
      "loss_g0": 10.352507591247559,
      "loss_g1": 9.300220489501953,
      "q_g0": 0.9834641814231873,
      "q_g1": 0.01653589867055416,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 36.94540023803711,
      "loss_g0": 36.94540023803711,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.696121215820312,
      "loss_g0": 15.89469051361084,
      "loss_g1": 3.876546859741211,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.630511283874512,
      "loss_g0": 13.657745361328125,
      "loss_g1": 12.009395599365234,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.070237159729004,
      "loss_g0": 10.968315124511719,
      "loss_g1": 17.136791229248047,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.016523055732250214,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.285162448883057,
      "loss_g0": 5.276134490966797,
      "loss_g1": 5.822510719299316,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523106023669243,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.669178009033203,
      "loss_g0": 5.437379837036133,
      "loss_g1": 19.46490478515625,
      "q_g0": 0.9834754467010498,
      "q_g1": 0.0165245421230793,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.117922306060791,
      "loss_g0": 7.173834323883057,
      "loss_g1": 3.790168285369873,
      "q_g0": 0.9834758043289185,
      "q_g1": 0.016524193808436394,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.3777494430542,
      "loss_g0": 9.339366912841797,
      "loss_g1": 11.662121772766113,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524434089660645,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.316729545593262,
      "loss_g0": 12.3693265914917,
      "loss_g1": 9.186256408691406,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652410998940468,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 29.314537048339844,
      "loss_g0": 29.383047103881836,
      "loss_g1": 25.236787796020508,
      "q_g0": 0.9834762811660767,
      "q_g1": 0.016523681581020355,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.447265625,
      "loss_g0": 13.334084510803223,
      "loss_g1": 20.18346405029297,
      "q_g0": 0.9834755659103394,
      "q_g1": 0.016524378210306168,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.32960033416748,
      "loss_g0": 8.390398025512695,
      "loss_g1": 4.711027145385742,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524000093340874,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.616907119750977,
      "loss_g0": 10.595457077026367,
      "loss_g1": 11.893525123596191,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652413234114647,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.932012557983398,
      "loss_g0": 10.067405700683594,
      "loss_g1": 1.8733307123184204,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523294150829315,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.005489349365234,
      "loss_g0": 15.10974407196045,
      "loss_g1": 8.799925804138184,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.01652265153825283,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.796890735626221,
      "loss_g0": 7.799859046936035,
      "loss_g1": 7.620170593261719,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.597681999206543,
      "loss_g0": 6.622663497924805,
      "loss_g1": 5.110695838928223,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522472724318504,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.979570388793945,
      "loss_g0": 17.1527042388916,
      "loss_g1": 6.67389440536499,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.827865600585938,
      "loss_g0": 24.19702911376953,
      "loss_g1": 1.8538258075714111,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.26226806640625,
      "loss_g0": 16.486141204833984,
      "loss_g1": 2.9364612102508545,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.8155536651611328,
      "loss_g0": 1.775848388671875,
      "loss_g1": 4.178924560546875,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522666439414024,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.856620788574219,
      "loss_g0": 6.879251480102539,
      "loss_g1": 5.509566307067871,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522526741027832,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.683476448059082,
      "loss_g0": 6.627655506134033,
      "loss_g1": 10.006040573120117,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652287133038044,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.854122638702393,
      "loss_g0": 5.923684597015381,
      "loss_g1": 1.7135342359542847,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522442921996117,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.154921531677246,
      "loss_g0": 12.345720291137695,
      "loss_g1": 0.7978374361991882,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.573331832885742,
      "loss_g0": 0.0,
      "loss_g1": 5.573331832885742,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.806586265563965,
      "loss_g0": 0.0,
      "loss_g1": 2.806586265563965,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.5129423141479492,
      "loss_g0": 1.0358006954193115,
      "loss_g1": 1.5209583044052124,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.335642099380493,
      "loss_g0": 3.057107925415039,
      "loss_g1": 2.323521375656128,
      "q_g0": 0.016522496938705444,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.078036904335022,
      "loss_g0": 1.078036904335022,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.0805888175964355,
      "loss_g0": 4.11598539352417,
      "loss_g1": 1.9736543893814087,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.9855828285217285,
      "loss_g0": 2.9506897926330566,
      "loss_g1": 5.062527656555176,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522642225027084,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.665555000305176,
      "loss_g0": 4.730382442474365,
      "loss_g1": 0.8067934513092041,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 67.79900360107422,
      "loss_g0": 68.6673812866211,
      "loss_g1": 16.109771728515625,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.50927734375,
      "loss_g0": 22.49738311767578,
      "loss_g1": 23.2171630859375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522496938705444,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.24254608154297,
      "loss_g0": 17.24254608154297,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 25.283042907714844,
      "loss_g0": 24.997211456298828,
      "loss_g1": 42.29677963256836,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.265769958496094,
      "loss_g0": 23.04037094116211,
      "loss_g1": 36.68122482299805,
      "q_g0": 0.9834761619567871,
      "q_g1": 0.0165238194167614,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.517937183380127,
      "loss_g0": 5.190915584564209,
      "loss_g1": 24.979385375976562,
      "q_g0": 0.9834741950035095,
      "q_g1": 0.016525844112038612,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 34.689971923828125,
      "loss_g0": 35.112979888916016,
      "loss_g1": 9.512349128723145,
      "q_g0": 0.9834766983985901,
      "q_g1": 0.016523223370313644,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.2691338062286377,
      "loss_g0": 2.9481611251831055,
      "loss_g1": 22.371379852294922,
      "q_g0": 0.9834747910499573,
      "q_g1": 0.016525207087397575,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 29.259248733520508,
      "loss_g0": 29.435617446899414,
      "loss_g1": 18.76218032836914,
      "q_g0": 0.983475923538208,
      "q_g1": 0.01652410998940468,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 31.006946563720703,
      "loss_g0": 31.389963150024414,
      "loss_g1": 8.208338737487793,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 24.040483474731445,
      "loss_g0": 24.147775650024414,
      "loss_g1": 17.653963088989258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.452255249023438,
      "loss_g0": 16.566255569458008,
      "loss_g1": 9.666439056396484,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.65833282470703,
      "loss_g0": 18.71885108947754,
      "loss_g1": 15.056068420410156,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.834941864013672,
      "loss_g0": 4.679959297180176,
      "loss_g1": 14.059539794921875,
      "q_g0": 0.9834766387939453,
      "q_g1": 0.01652337983250618,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 26.438255310058594,
      "loss_g0": 26.72694969177246,
      "loss_g1": 9.253995895385742,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.443845748901367,
      "loss_g0": 16.202030181884766,
      "loss_g1": 30.836286544799805,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016523919999599457,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.402214050292969,
      "loss_g0": 12.94947338104248,
      "loss_g1": 40.343910217285156,
      "q_g0": 0.9834732413291931,
      "q_g1": 0.01652671955525875,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.510644912719727,
      "loss_g0": 9.283987998962402,
      "loss_g1": 22.99742889404297,
      "q_g0": 0.9834718704223633,
      "q_g1": 0.016528114676475525,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.861481666564941,
      "loss_g0": 0.0,
      "loss_g1": 10.861481666564941,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.979524612426758,
      "loss_g0": 22.858184814453125,
      "loss_g1": 15.86396312713623,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.063400268554688,
      "loss_g0": 0.0,
      "loss_g1": 14.063400268554688,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.409279823303223,
      "loss_g0": 2.265695810317993,
      "loss_g1": 8.512492179870605,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.823458194732666,
      "loss_g0": 20.983312606811523,
      "loss_g1": 5.568747520446777,
      "q_g0": 0.016524000093340874,
      "q_g1": 0.9834760427474976,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.80723237991333,
      "loss_g0": 6.984992980957031,
      "loss_g1": 2.737037420272827,
      "q_g0": 0.016524434089660645,
      "q_g1": 0.9834755659103394,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 0.9753307104110718,
      "loss_g0": 18.146949768066406,
      "loss_g1": 0.6867801547050476,
      "q_g0": 0.01652621664106846,
      "q_g1": 0.9834737181663513,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.277074337005615,
      "loss_g0": 13.446226119995117,
      "loss_g1": 7.173404693603516,
      "q_g0": 0.016526861116290092,
      "q_g1": 0.9834731221199036,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.844566345214844,
      "loss_g0": 8.262432098388672,
      "loss_g1": 13.938368797302246,
      "q_g0": 0.016526278108358383,
      "q_g1": 0.9834737181663513,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.313888549804688,
      "loss_g0": 8.932777404785156,
      "loss_g1": 20.5051212310791,
      "q_g0": 0.016525091603398323,
      "q_g1": 0.983474850654602,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.492120742797852,
      "loss_g0": 5.1199140548706055,
      "loss_g1": 10.582386016845703,
      "q_g0": 0.0165245421230793,
      "q_g1": 0.9834754467010498,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.5782253742218018,
      "loss_g0": 7.7098212242126465,
      "loss_g1": 3.508803606033325,
      "q_g0": 0.01652497425675392,
      "q_g1": 0.9834749698638916,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.767647743225098,
      "loss_g0": 18.37955093383789,
      "loss_g1": 13.690154075622559,
      "q_g0": 0.016525447368621826,
      "q_g1": 0.9834745526313782,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.5928263664245605,
      "loss_g0": 21.983009338378906,
      "loss_g1": 4.300584316253662,
      "q_g0": 0.01652725413441658,
      "q_g1": 0.9834727048873901,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.573966979980469,
      "loss_g0": 7.588349342346191,
      "loss_g1": 8.590530395507812,
      "q_g0": 0.01652715355157852,
      "q_g1": 0.9834728240966797,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.2887210845947266,
      "loss_g0": 16.45311164855957,
      "loss_g1": 3.067476272583008,
      "q_g0": 0.016528518870472908,
      "q_g1": 0.9834715127944946,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.459158897399902,
      "loss_g0": 25.298751831054688,
      "loss_g1": 5.125686168670654,
      "q_g0": 0.016530588269233704,
      "q_g1": 0.9834694862365723,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.349316120147705,
      "loss_g0": 25.141305923461914,
      "loss_g1": 5.016601085662842,
      "q_g0": 0.016532650217413902,
      "q_g1": 0.9834674000740051,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.767165184020996,
      "loss_g0": 5.792581558227539,
      "loss_g1": 7.800358295440674,
      "q_g0": 0.01653243973851204,
      "q_g1": 0.9834675192832947,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.1076428890228271,
      "loss_g0": 12.668594360351562,
      "loss_g1": 0.9132848381996155,
      "q_g0": 0.016533643007278442,
      "q_g1": 0.9834663271903992,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.428894519805908,
      "loss_g0": 10.599100112915039,
      "loss_g1": 3.3083465099334717,
      "q_g0": 0.01653439551591873,
      "q_g1": 0.9834656715393066,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.095229148864746,
      "loss_g0": 9.908007621765137,
      "loss_g1": 9.081563949584961,
      "q_g0": 0.016534479334950447,
      "q_g1": 0.9834655523300171,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.345400333404541,
      "loss_g0": 16.281518936157227,
      "loss_g1": 4.144709587097168,
      "q_g0": 0.016535719856619835,
      "q_g1": 0.9834643006324768,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.0207468271255493,
      "loss_g0": 2.3637301921844482,
      "loss_g1": 0.9981659650802612,
      "q_g0": 0.016535859555006027,
      "q_g1": 0.9834641814231873,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.269645690917969,
      "loss_g0": 0.0,
      "loss_g1": 7.269645690917969,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.655519723892212,
      "loss_g0": 5.607308864593506,
      "loss_g1": 3.622729539871216,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.85187292098999,
      "loss_g0": 4.85187292098999,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 16.393051147460938,
      "loss_g0": 16.581958770751953,
      "loss_g1": 5.148509979248047,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.750447750091553,
      "loss_g0": 5.750447750091553,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 33.065940856933594,
      "loss_g0": 33.58784484863281,
      "loss_g1": 2.000074625015259,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.2343661785125732,
      "loss_g0": 3.237823486328125,
      "loss_g1": 3.0285680294036865,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.357472896575928,
      "loss_g0": 4.408839702606201,
      "loss_g1": 1.2999167442321777,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.283625602722168,
      "loss_g0": 14.025872230529785,
      "loss_g1": 29.624614715576172,
      "q_g0": 0.9834760427474976,
      "q_g1": 0.016524024307727814,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.118623733520508,
      "loss_g0": 0.0,
      "loss_g1": 13.118623733520508,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.729190826416016,
      "loss_g0": 23.729190826416016,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.702898025512695,
      "loss_g0": 18.67554473876953,
      "loss_g1": 20.33106231689453,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 65.06285858154297,
      "loss_g0": 65.7664566040039,
      "loss_g1": 23.18171501159668,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.013917922973633,
      "loss_g0": 17.971342086791992,
      "loss_g1": 20.5482234954834,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522690653800964,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.934463024139404,
      "loss_g0": 7.979315757751465,
      "loss_g1": 5.264647483825684,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 25.647705078125,
      "loss_g0": 25.710105895996094,
      "loss_g1": 21.9333438873291,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.064263343811035,
      "loss_g0": 12.988436698913574,
      "loss_g1": 17.57759666442871,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.016522889956831932,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.00395393371582,
      "loss_g0": 8.804430961608887,
      "loss_g1": 20.8790283203125,
      "q_g0": 0.983475923538208,
      "q_g1": 0.016524117439985275,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 43.850215911865234,
      "loss_g0": 44.454734802246094,
      "loss_g1": 7.86701774597168,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.885191917419434,
      "loss_g0": 11.829298973083496,
      "loss_g1": 15.21209716796875,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652277447283268,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 25.372241973876953,
      "loss_g0": 25.554126739501953,
      "loss_g1": 14.54578685760498,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 39.653560638427734,
      "loss_g0": 40.08961868286133,
      "loss_g1": 13.697680473327637,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.386863708496094,
      "loss_g0": 20.635456085205078,
      "loss_g1": 5.589716911315918,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 23.243471145629883,
      "loss_g0": 23.283912658691406,
      "loss_g1": 20.836244583129883,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.597614288330078,
      "loss_g0": 20.758188247680664,
      "loss_g1": 11.039670944213867,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.404631614685059,
      "loss_g0": 11.370776176452637,
      "loss_g1": 13.419784545898438,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522636637091637,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 27.534467697143555,
      "loss_g0": 27.534467697143555,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.473228454589844,
      "loss_g0": 9.374186515808105,
      "loss_g1": 15.368562698364258,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 56.28868865966797,
      "loss_g0": 57.08611297607422,
      "loss_g1": 8.822890281677246,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 34.06827926635742,
      "loss_g0": 34.53669357299805,
      "loss_g1": 6.186463832855225,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.737317085266113,
      "loss_g0": 10.75760269165039,
      "loss_g1": 9.529786109924316,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.3823442459106445,
      "loss_g0": 7.385372161865234,
      "loss_g1": 7.20208740234375,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.777437210083008,
      "loss_g0": 18.797395706176758,
      "loss_g1": 17.589385986328125,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.608261108398438,
      "loss_g0": 17.855628967285156,
      "loss_g1": 2.8839168548583984,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.993587493896484,
      "loss_g0": 19.98677635192871,
      "loss_g1": 20.399051666259766,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522465273737907,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.25763511657715,
      "loss_g0": 17.422405242919922,
      "loss_g1": 7.449973106384277,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.306591033935547,
      "loss_g0": 19.44329071044922,
      "loss_g1": 11.16968822479248,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.769974708557129,
      "loss_g0": 5.772798538208008,
      "loss_g1": 5.601861953735352,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.274068832397461,
      "loss_g0": 0.0,
      "loss_g1": 5.274068832397461,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.9352598190307617,
      "loss_g0": 5.276727676391602,
      "loss_g1": 3.9127230644226074,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.4982187747955322,
      "loss_g0": 0.0,
      "loss_g1": 2.4982187747955322,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.611332893371582,
      "loss_g0": 7.428632736206055,
      "loss_g1": 11.681602478027344,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.363198757171631,
      "loss_g0": 25.28387451171875,
      "loss_g1": 6.045292377471924,
      "q_g0": 0.01652439311146736,
      "q_g1": 0.9834755659103394,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.438750267028809,
      "loss_g0": 12.805595397949219,
      "loss_g1": 5.314966201782227,
      "q_g0": 0.016525164246559143,
      "q_g1": 0.983474850654602,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.8237648010253906,
      "loss_g0": 26.11811065673828,
      "loss_g1": 3.4491028785705566,
      "q_g0": 0.016527486965060234,
      "q_g1": 0.983472466468811,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.541487693786621,
      "loss_g0": 5.9501190185546875,
      "loss_g1": 7.568230628967285,
      "q_g0": 0.01652732491493225,
      "q_g1": 0.9834727048873901,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 18.033180236816406,
      "loss_g0": 13.499944686889648,
      "loss_g1": 18.109359741210938,
      "q_g0": 0.016526851803064346,
      "q_g1": 0.9834731221199036,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.055119514465332,
      "loss_g0": 3.806342363357544,
      "loss_g1": 5.076104640960693,
      "q_g0": 0.01652671955525875,
      "q_g1": 0.9834732413291931,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.674848556518555,
      "loss_g0": 5.218895435333252,
      "loss_g1": 12.800135612487793,
      "q_g0": 0.01652594469487667,
      "q_g1": 0.98347407579422,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.748359680175781,
      "loss_g0": 4.865699291229248,
      "loss_g1": 4.746387481689453,
      "q_g0": 0.016525952145457268,
      "q_g1": 0.98347407579422,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.3107092380523682,
      "loss_g0": 5.310553073883057,
      "loss_g1": 1.2434955835342407,
      "q_g0": 0.016526371240615845,
      "q_g1": 0.9834735989570618,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.5604881048202515,
      "loss_g0": 2.541515827178955,
      "loss_g1": 1.544002652168274,
      "q_g0": 0.0165264792740345,
      "q_g1": 0.9834734797477722,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.9861884117126465,
      "loss_g0": 6.047152519226074,
      "loss_g1": 1.917945384979248,
      "q_g0": 0.016526905819773674,
      "q_g1": 0.9834731221199036,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.12448787689209,
      "loss_g0": 10.12448787689209,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.93161678314209,
      "loss_g0": 15.13868522644043,
      "loss_g1": 2.6061370372772217,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.439617156982422,
      "loss_g0": 10.609853744506836,
      "loss_g1": 0.30648842453956604,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.376611709594727,
      "loss_g0": 8.481913566589355,
      "loss_g1": 2.108621597290039,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.2917962074279785,
      "loss_g0": 2.178114175796509,
      "loss_g1": 9.0582914352417,
      "q_g0": 0.9834768176078796,
      "q_g1": 0.016523122787475586,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 30.767162322998047,
      "loss_g0": 30.799772262573242,
      "loss_g1": 28.826284408569336,
      "q_g0": 0.9834770560264587,
      "q_g1": 0.01652292162179947,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 30.6528377532959,
      "loss_g0": 30.898244857788086,
      "loss_g1": 16.045225143432617,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.904443740844727,
      "loss_g0": 15.856925010681152,
      "loss_g1": 18.73287582397461,
      "q_g0": 0.9834772944450378,
      "q_g1": 0.016522713005542755,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.015840530395508,
      "loss_g0": 14.955536842346191,
      "loss_g1": 18.605178833007812,
      "q_g0": 0.9834769368171692,
      "q_g1": 0.0165230855345726,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 24.632246017456055,
      "loss_g0": 24.632246017456055,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 35.402976989746094,
      "loss_g0": 35.402976989746094,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 38.18678665161133,
      "loss_g0": 38.714168548583984,
      "loss_g1": 6.794985771179199,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 17.963293075561523,
      "loss_g0": 18.065750122070312,
      "loss_g1": 11.86466121673584,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.288431167602539,
      "loss_g0": 9.343588829040527,
      "loss_g1": 6.005242824554443,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 15.683470726013184,
      "loss_g0": 15.813940048217773,
      "loss_g1": 7.9174604415893555,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.94947624206543,
      "loss_g0": 0.0,
      "loss_g1": 9.94947624206543,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.094902038574219,
      "loss_g0": 25.891036987304688,
      "loss_g1": 6.77912712097168,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.18683910369873,
      "loss_g0": 10.801900863647461,
      "loss_g1": 10.176506042480469,
      "q_g0": 0.0165224801748991,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.58897590637207,
      "loss_g0": 14.264178276062012,
      "loss_g1": 6.460025787353516,
      "q_g0": 0.016523271799087524,
      "q_g1": 0.9834766983985901,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.092412948608398,
      "loss_g0": 13.092412948608398,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 22.083415985107422,
      "loss_g0": 22.035778045654297,
      "loss_g1": 24.918989181518555,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 28.824718475341797,
      "loss_g0": 29.17609405517578,
      "loss_g1": 7.9094953536987305,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.538084030151367,
      "loss_g0": 5.542581081390381,
      "loss_g1": 5.270382881164551,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.982280731201172,
      "loss_g0": 8.009464263916016,
      "loss_g1": 6.3642168045043945,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 12.464509963989258,
      "loss_g0": 0.0,
      "loss_g1": 12.464509963989258,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.747285842895508,
      "loss_g0": 9.02778434753418,
      "loss_g1": 8.742572784423828,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.253321409225464,
      "loss_g0": 23.0733699798584,
      "loss_g1": 1.9034980535507202,
      "q_g0": 0.016524581238627434,
      "q_g1": 0.9834754467010498,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.671017646789551,
      "loss_g0": 5.238346576690674,
      "loss_g1": 5.678287506103516,
      "q_g0": 0.016524534672498703,
      "q_g1": 0.9834754467010498,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.251127243041992,
      "loss_g0": 14.477823257446289,
      "loss_g1": 6.112893104553223,
      "q_g0": 0.01652538776397705,
      "q_g1": 0.9834746718406677,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.184975624084473,
      "loss_g0": 3.5803699493408203,
      "loss_g1": 4.19513463973999,
      "q_g0": 0.01652531512081623,
      "q_g1": 0.9834746718406677,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.5312180519104,
      "loss_g0": 18.60729217529297,
      "loss_g1": 7.345093727111816,
      "q_g0": 0.016526462510228157,
      "q_g1": 0.9834734797477722,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.310333251953125,
      "loss_g0": 12.915666580200195,
      "loss_g1": 5.182525634765625,
      "q_g0": 0.01652725413441658,
      "q_g1": 0.9834727048873901,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.773795127868652,
      "loss_g0": 12.471923828125,
      "loss_g1": 11.762063026428223,
      "q_g0": 0.016527332365512848,
      "q_g1": 0.9834727048873901,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 14.680523872375488,
      "loss_g0": 16.367055892944336,
      "loss_g1": 14.652181625366211,
      "q_g0": 0.016527509316802025,
      "q_g1": 0.983472466468811,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.951205253601074,
      "loss_g0": 20.786739349365234,
      "loss_g1": 2.651440143585205,
      "q_g0": 0.01652936451137066,
      "q_g1": 0.983470618724823,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.876808166503906,
      "loss_g0": 5.869298934936523,
      "loss_g1": 7.910548210144043,
      "q_g0": 0.016529154032468796,
      "q_g1": 0.9834708571434021,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.191372871398926,
      "loss_g0": 16.077171325683594,
      "loss_g1": 2.9747841358184814,
      "q_g0": 0.016530493274331093,
      "q_g1": 0.9834694862365723,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 2.7328364849090576,
      "loss_g0": 5.7972612380981445,
      "loss_g1": 2.6813275814056396,
      "q_g0": 0.016530806198716164,
      "q_g1": 0.9834692478179932,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.419203758239746,
      "loss_g0": 8.080130577087402,
      "loss_g1": 9.44171142578125,
      "q_g0": 0.016530673950910568,
      "q_g1": 0.9834693670272827,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.563135147094727,
      "loss_g0": 6.4681196212768555,
      "loss_g1": 9.615157127380371,
      "q_g0": 0.016530346125364304,
      "q_g1": 0.9834696650505066,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.228058815002441,
      "loss_g0": 2.51550030708313,
      "loss_g1": 4.256843566894531,
      "q_g0": 0.01653016172349453,
      "q_g1": 0.9834697842597961,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 11.15336799621582,
      "loss_g0": 11.15336799621582,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.910665035247803,
      "loss_g0": 7.905240058898926,
      "loss_g1": 8.233573913574219,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.456137657165527,
      "loss_g0": 7.512310028076172,
      "loss_g1": 4.112536907196045,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 26.223419189453125,
      "loss_g0": 26.662023544311523,
      "loss_g1": 0.11598683148622513,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.762199401855469,
      "loss_g0": 10.916304588317871,
      "loss_g1": 1.5892891883850098,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.310429573059082,
      "loss_g0": 10.310429573059082,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.136978626251221,
      "loss_g0": 6.889939308166504,
      "loss_g1": 21.841684341430664,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.718088626861572,
      "loss_g0": 4.718088626861572,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.765533447265625,
      "loss_g0": 8.765533447265625,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 25.062362670898438,
      "loss_g0": 25.461517333984375,
      "loss_g1": 1.3030648231506348,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.695426940917969,
      "loss_g0": 10.846785545349121,
      "loss_g1": 1.6859917640686035,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 34.21070861816406,
      "loss_g0": 34.78146743774414,
      "loss_g1": 0.2368766963481903,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 0.9366901516914368,
      "loss_g0": 0.8677968382835388,
      "loss_g1": 5.037372589111328,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.01652284525334835,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 9.464378356933594,
      "loss_g0": 9.50550651550293,
      "loss_g1": 7.0163092613220215,
      "q_g0": 0.9834774136543274,
      "q_g1": 0.016522597521543503,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 20.61673355102539,
      "loss_g0": 20.577392578125,
      "loss_g1": 22.95834732055664,
      "q_g0": 0.9834771752357483,
      "q_g1": 0.016522839665412903,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 26.156360626220703,
      "loss_g0": 26.156360626220703,
      "loss_g1": 0.0,
      "q_g0": 1.0,
      "q_g1": 0.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.354697227478027,
      "loss_g0": 9.926275253295898,
      "loss_g1": 35.85600280761719,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 19.417551040649414,
      "loss_g0": 19.56722068786621,
      "loss_g1": 10.508591651916504,
      "q_g0": 0.9834775328636169,
      "q_g1": 0.016522422432899475,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 13.200075149536133,
      "loss_g0": 0.0,
      "loss_g1": 13.200075149536133,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 8.555368423461914,
      "loss_g0": 25.734107971191406,
      "loss_g1": 8.266765594482422,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 10.34322452545166,
      "loss_g0": 15.004528045654297,
      "loss_g1": 10.264912605285645,
      "q_g0": 0.016522906720638275,
      "q_g1": 0.9834770560264587,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.075135707855225,
      "loss_g0": 22.915197372436523,
      "loss_g1": 4.775379180908203,
      "q_g0": 0.01652476005256176,
      "q_g1": 0.9834752082824707,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 30.346033096313477,
      "loss_g0": 67.61344146728516,
      "loss_g1": 29.719701766967773,
      "q_g0": 0.01652863621711731,
      "q_g1": 0.9834713935852051,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 3.9693353176116943,
      "loss_g0": 0.0,
      "loss_g1": 3.9693353176116943,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.0532755851745605,
      "loss_g0": 8.211894989013672,
      "loss_g1": 6.017010688781738,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.513273239135742,
      "loss_g0": 10.14857006072998,
      "loss_g1": 7.468999862670898,
      "q_g0": 0.01652269810438156,
      "q_g1": 0.9834772944450378,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 5.9567484855651855,
      "loss_g0": 3.0027058124542236,
      "loss_g1": 6.006376266479492,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 6.692985534667969,
      "loss_g0": 0.0,
      "loss_g1": 6.692985534667969,
      "q_g0": 0.0,
      "q_g1": 1.0,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.7978428602218628,
      "loss_g0": 10.629594802856445,
      "loss_g1": 1.6494693756103516,
      "q_g0": 0.016522422432899475,
      "q_g1": 0.9834775328636169,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 4.029397487640381,
      "loss_g0": 7.755870819091797,
      "loss_g1": 3.9667916297912598,
      "q_g0": 0.016522806137800217,
      "q_g1": 0.9834771752357483,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 1.4685897827148438,
      "loss_g0": 12.347294807434082,
      "loss_g1": 1.2858104705810547,
      "q_g0": 0.0165239367634058,
      "q_g1": 0.9834760427474976,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 7.416388988494873,
      "loss_g0": 9.255908012390137,
      "loss_g1": 7.385481357574463,
      "q_g0": 0.016524124890565872,
      "q_g1": 0.983475923538208,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "gdro_loss": 0.47764042019844055,
      "loss_g0": 1.348243236541748,
      "loss_g1": 0.4630126655101776,
      "q_g0": 0.016524218022823334,
      "q_g1": 0.9834758043289185,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "eval_cer": 0.05962685694233243,
      "eval_loss": 13.847650527954102,
      "eval_runtime": 80.9809,
      "eval_samples_per_second": 20.696,
      "eval_steps_per_second": 5.174,
      "step": 1400
    },
    {
      "epoch": 7.106598984771574,
      "step": 1400,
      "total_flos": 1.3869661565632915e+19,
      "train_loss": 109.38554827008929,
      "train_runtime": 2644.8438,
      "train_samples_per_second": 11.906,
      "train_steps_per_second": 0.745
    }
  ],
  "logging_steps": 100,
  "max_steps": 1970,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 200,
  "stateful_callbacks": {
    "EarlyStoppingCallback": {
      "args": {
        "early_stopping_patience": 3,
        "early_stopping_threshold": 0.0
      },
      "attributes": {
        "early_stopping_patience_counter": 3
      }
    },
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 1.3869661565632915e+19,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}