{
  "best_metric": 1.3928166627883911,
  "best_model_checkpoint": "jungledude23/mixtral-barlow-gsm8k-scale2e-6/checkpoint-466",
  "epoch": 1.9935828877005348,
  "eval_steps": 500,
  "global_step": 466,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "barlow_loss": 0.1585599184036255,
      "epoch": 0,
      "loss": 1.6077316999435425,
      "step": 0
    },
    {
      "barlow_loss": 0.16877791285514832,
      "epoch": 0,
      "loss": 1.7065032720565796,
      "step": 0
    },
    {
      "barlow_loss": 0.18340139091014862,
      "epoch": 0,
      "loss": 1.4923725128173828,
      "step": 0
    },
    {
      "barlow_loss": 0.163997620344162,
      "epoch": 0,
      "loss": 1.9266834259033203,
      "step": 0
    },
    {
      "barlow_loss": 0.15060921013355255,
      "epoch": 0.0042780748663101605,
      "loss": 1.582384467124939,
      "step": 1
    },
    {
      "barlow_loss": 0.1526964008808136,
      "epoch": 0.0042780748663101605,
      "loss": 1.6403454542160034,
      "step": 1
    },
    {
      "barlow_loss": 0.16772675514221191,
      "epoch": 0.0042780748663101605,
      "loss": 1.6037108898162842,
      "step": 1
    },
    {
      "barlow_loss": 0.16591890156269073,
      "epoch": 0.0042780748663101605,
      "loss": 1.6657137870788574,
      "step": 1
    },
    {
      "barlow_loss": 0.16248787939548492,
      "epoch": 0.008556149732620321,
      "loss": 1.655889868736267,
      "step": 2
    },
    {
      "barlow_loss": 0.1527155488729477,
      "epoch": 0.008556149732620321,
      "loss": 1.572197437286377,
      "step": 2
    },
    {
      "barlow_loss": 0.15598638355731964,
      "epoch": 0.008556149732620321,
      "loss": 1.8180546760559082,
      "step": 2
    },
    {
      "barlow_loss": 0.15999066829681396,
      "epoch": 0.008556149732620321,
      "loss": 1.576218605041504,
      "step": 2
    },
    {
      "barlow_loss": 0.1580725610256195,
      "epoch": 0.012834224598930482,
      "loss": 1.9403448104858398,
      "step": 3
    },
    {
      "barlow_loss": 0.14824599027633667,
      "epoch": 0.012834224598930482,
      "loss": 1.7365310192108154,
      "step": 3
    },
    {
      "barlow_loss": 0.1688745766878128,
      "epoch": 0.012834224598930482,
      "loss": 1.5892987251281738,
      "step": 3
    },
    {
      "barlow_loss": 0.1772894263267517,
      "epoch": 0.012834224598930482,
      "loss": 1.6161361932754517,
      "step": 3
    },
    {
      "epoch": 0.017112299465240642,
      "grad_norm": 0.23516832292079926,
      "learning_rate": 4.166666666666667e-06,
      "loss": 1.6706,
      "step": 4
    },
    {
      "barlow_loss": 0.1493869572877884,
      "epoch": 0.017112299465240642,
      "loss": 1.7811100482940674,
      "step": 4
    },
    {
      "barlow_loss": 0.16708625853061676,
      "epoch": 0.017112299465240642,
      "loss": 1.602239727973938,
      "step": 4
    },
    {
      "barlow_loss": 0.17603515088558197,
      "epoch": 0.017112299465240642,
      "loss": 1.5734937191009521,
      "step": 4
    },
    {
      "barlow_loss": 0.16268548369407654,
      "epoch": 0.017112299465240642,
      "loss": 1.9233371019363403,
      "step": 4
    },
    {
      "barlow_loss": 0.17499813437461853,
      "epoch": 0.0213903743315508,
      "loss": 1.7165064811706543,
      "step": 5
    },
    {
      "barlow_loss": 0.16469089686870575,
      "epoch": 0.0213903743315508,
      "loss": 1.5351988077163696,
      "step": 5
    },
    {
      "barlow_loss": 0.17805194854736328,
      "epoch": 0.0213903743315508,
      "loss": 1.6616038084030151,
      "step": 5
    },
    {
      "barlow_loss": 0.16441243886947632,
      "epoch": 0.0213903743315508,
      "loss": 1.519624948501587,
      "step": 5
    },
    {
      "barlow_loss": 0.15703794360160828,
      "epoch": 0.025668449197860963,
      "loss": 1.5024290084838867,
      "step": 6
    },
    {
      "barlow_loss": 0.17620772123336792,
      "epoch": 0.025668449197860963,
      "loss": 1.4058555364608765,
      "step": 6
    },
    {
      "barlow_loss": 0.16154858469963074,
      "epoch": 0.025668449197860963,
      "loss": 1.9035557508468628,
      "step": 6
    },
    {
      "barlow_loss": 0.16517719626426697,
      "epoch": 0.025668449197860963,
      "loss": 1.8484210968017578,
      "step": 6
    },
    {
      "barlow_loss": 0.1607310175895691,
      "epoch": 0.029946524064171122,
      "loss": 1.6375558376312256,
      "step": 7
    },
    {
      "barlow_loss": 0.17218199372291565,
      "epoch": 0.029946524064171122,
      "loss": 1.6924121379852295,
      "step": 7
    },
    {
      "barlow_loss": 0.17651046812534332,
      "epoch": 0.029946524064171122,
      "loss": 1.9043635129928589,
      "step": 7
    },
    {
      "barlow_loss": 0.16861680150032043,
      "epoch": 0.029946524064171122,
      "loss": 1.802912950515747,
      "step": 7
    },
    {
      "epoch": 0.034224598930481284,
      "grad_norm": 0.23444822430610657,
      "learning_rate": 8.333333333333334e-06,
      "loss": 1.6882,
      "step": 8
    },
    {
      "barlow_loss": 0.1612589806318283,
      "epoch": 0.034224598930481284,
      "loss": 1.4539847373962402,
      "step": 8
    },
    {
      "barlow_loss": 0.16069044172763824,
      "epoch": 0.034224598930481284,
      "loss": 1.6162855625152588,
      "step": 8
    },
    {
      "barlow_loss": 0.1587289422750473,
      "epoch": 0.034224598930481284,
      "loss": 1.737745761871338,
      "step": 8
    },
    {
      "barlow_loss": 0.1623297780752182,
      "epoch": 0.034224598930481284,
      "loss": 1.7921684980392456,
      "step": 8
    },
    {
      "barlow_loss": 0.1790577471256256,
      "epoch": 0.038502673796791446,
      "loss": 1.7282754182815552,
      "step": 9
    },
    {
      "barlow_loss": 0.14387206733226776,
      "epoch": 0.038502673796791446,
      "loss": 1.8310436010360718,
      "step": 9
    },
    {
      "barlow_loss": 0.1601451337337494,
      "epoch": 0.038502673796791446,
      "loss": 1.577781081199646,
      "step": 9
    },
    {
      "barlow_loss": 0.17837581038475037,
      "epoch": 0.038502673796791446,
      "loss": 1.7797036170959473,
      "step": 9
    },
    {
      "barlow_loss": 0.1686144918203354,
      "epoch": 0.0427807486631016,
      "loss": 1.685567855834961,
      "step": 10
    },
    {
      "barlow_loss": 0.16616277396678925,
      "epoch": 0.0427807486631016,
      "loss": 1.5781502723693848,
      "step": 10
    },
    {
      "barlow_loss": 0.16896234452724457,
      "epoch": 0.0427807486631016,
      "loss": 1.642483115196228,
      "step": 10
    },
    {
      "barlow_loss": 0.1667196899652481,
      "epoch": 0.0427807486631016,
      "loss": 1.7646623849868774,
      "step": 10
    },
    {
      "barlow_loss": 0.16737215220928192,
      "epoch": 0.047058823529411764,
      "loss": 1.6144684553146362,
      "step": 11
    },
    {
      "barlow_loss": 0.15302208065986633,
      "epoch": 0.047058823529411764,
      "loss": 1.609765887260437,
      "step": 11
    },
    {
      "barlow_loss": 0.18187567591667175,
      "epoch": 0.047058823529411764,
      "loss": 1.803045630455017,
      "step": 11
    },
    {
      "barlow_loss": 0.15781933069229126,
      "epoch": 0.047058823529411764,
      "loss": 1.6364264488220215,
      "step": 11
    },
    {
      "epoch": 0.051336898395721926,
      "grad_norm": 0.2205454409122467,
      "learning_rate": 1.25e-05,
      "loss": 1.6782,
      "step": 12
    },
    {
      "barlow_loss": 0.16361591219902039,
      "epoch": 0.051336898395721926,
      "loss": 1.8468031883239746,
      "step": 12
    },
    {
      "barlow_loss": 0.17066185176372528,
      "epoch": 0.051336898395721926,
      "loss": 1.8121522665023804,
      "step": 12
    },
    {
      "barlow_loss": 0.1573106199502945,
      "epoch": 0.051336898395721926,
      "loss": 1.5622649192810059,
      "step": 12
    },
    {
      "barlow_loss": 0.1640360951423645,
      "epoch": 0.051336898395721926,
      "loss": 1.7879736423492432,
      "step": 12
    },
    {
      "barlow_loss": 0.1635410040616989,
      "epoch": 0.05561497326203209,
      "loss": 1.822780728340149,
      "step": 13
    },
    {
      "barlow_loss": 0.15114203095436096,
      "epoch": 0.05561497326203209,
      "loss": 1.7811460494995117,
      "step": 13
    },
    {
      "barlow_loss": 0.15519177913665771,
      "epoch": 0.05561497326203209,
      "loss": 1.67121422290802,
      "step": 13
    },
    {
      "barlow_loss": 0.16310343146324158,
      "epoch": 0.05561497326203209,
      "loss": 1.6594802141189575,
      "step": 13
    },
    {
      "barlow_loss": 0.16564993560314178,
      "epoch": 0.059893048128342244,
      "loss": 1.6326580047607422,
      "step": 14
    },
    {
      "barlow_loss": 0.15832221508026123,
      "epoch": 0.059893048128342244,
      "loss": 1.4730528593063354,
      "step": 14
    },
    {
      "barlow_loss": 0.16867950558662415,
      "epoch": 0.059893048128342244,
      "loss": 1.6341816186904907,
      "step": 14
    },
    {
      "barlow_loss": 0.17301930487155914,
      "epoch": 0.059893048128342244,
      "loss": 1.8583589792251587,
      "step": 14
    },
    {
      "barlow_loss": 0.15810857713222504,
      "epoch": 0.06417112299465241,
      "loss": 1.7832087278366089,
      "step": 15
    },
    {
      "barlow_loss": 0.16231968998908997,
      "epoch": 0.06417112299465241,
      "loss": 1.642933964729309,
      "step": 15
    },
    {
      "barlow_loss": 0.1644217073917389,
      "epoch": 0.06417112299465241,
      "loss": 1.719920039176941,
      "step": 15
    },
    {
      "barlow_loss": 0.16730813682079315,
      "epoch": 0.06417112299465241,
      "loss": 2.13230299949646,
      "step": 15
    },
    {
      "epoch": 0.06844919786096257,
      "grad_norm": 0.28154295682907104,
      "learning_rate": 1.6666666666666667e-05,
      "loss": 1.7388,
      "step": 16
    },
    {
      "barlow_loss": 0.16664528846740723,
      "epoch": 0.06844919786096257,
      "loss": 1.4975132942199707,
      "step": 16
    },
    {
      "barlow_loss": 0.18340373039245605,
      "epoch": 0.06844919786096257,
      "loss": 1.6137131452560425,
      "step": 16
    },
    {
      "barlow_loss": 0.16500025987625122,
      "epoch": 0.06844919786096257,
      "loss": 1.6954190731048584,
      "step": 16
    },
    {
      "barlow_loss": 0.15429849922657013,
      "epoch": 0.06844919786096257,
      "loss": 1.4773802757263184,
      "step": 16
    },
    {
      "barlow_loss": 0.14937271177768707,
      "epoch": 0.07272727272727272,
      "loss": 1.6289373636245728,
      "step": 17
    },
    {
      "barlow_loss": 0.1711861789226532,
      "epoch": 0.07272727272727272,
      "loss": 1.8005648851394653,
      "step": 17
    },
    {
      "barlow_loss": 0.1820562183856964,
      "epoch": 0.07272727272727272,
      "loss": 1.556789517402649,
      "step": 17
    },
    {
      "barlow_loss": 0.16246338188648224,
      "epoch": 0.07272727272727272,
      "loss": 1.8150590658187866,
      "step": 17
    },
    {
      "barlow_loss": 0.1733948439359665,
      "epoch": 0.07700534759358289,
      "loss": 1.7747148275375366,
      "step": 18
    },
    {
      "barlow_loss": 0.1618361622095108,
      "epoch": 0.07700534759358289,
      "loss": 1.6261860132217407,
      "step": 18
    },
    {
      "barlow_loss": 0.17435228824615479,
      "epoch": 0.07700534759358289,
      "loss": 1.7253001928329468,
      "step": 18
    },
    {
      "barlow_loss": 0.16540586948394775,
      "epoch": 0.07700534759358289,
      "loss": 1.8024779558181763,
      "step": 18
    },
    {
      "barlow_loss": 0.16136477887630463,
      "epoch": 0.08128342245989305,
      "loss": 1.5588748455047607,
      "step": 19
    },
    {
      "barlow_loss": 0.17014120519161224,
      "epoch": 0.08128342245989305,
      "loss": 1.7027559280395508,
      "step": 19
    },
    {
      "barlow_loss": 0.17533418536186218,
      "epoch": 0.08128342245989305,
      "loss": 1.7209923267364502,
      "step": 19
    },
    {
      "barlow_loss": 0.16413331031799316,
      "epoch": 0.08128342245989305,
      "loss": 1.5688037872314453,
      "step": 19
    },
    {
      "epoch": 0.0855614973262032,
      "grad_norm": 0.24900399148464203,
      "learning_rate": 2.0833333333333336e-05,
      "loss": 1.6603,
      "step": 20
    },
    {
      "barlow_loss": 0.15105070173740387,
      "epoch": 0.0855614973262032,
      "loss": 1.829237937927246,
      "step": 20
    },
    {
      "barlow_loss": 0.15005707740783691,
      "epoch": 0.0855614973262032,
      "loss": 1.458463191986084,
      "step": 20
    },
    {
      "barlow_loss": 0.15261860191822052,
      "epoch": 0.0855614973262032,
      "loss": 1.6228259801864624,
      "step": 20
    },
    {
      "barlow_loss": 0.1543506681919098,
      "epoch": 0.0855614973262032,
      "loss": 1.8120239973068237,
      "step": 20
    },
    {
      "barlow_loss": 0.15330223739147186,
      "epoch": 0.08983957219251337,
      "loss": 1.6085469722747803,
      "step": 21
    },
    {
      "barlow_loss": 0.1680625081062317,
      "epoch": 0.08983957219251337,
      "loss": 1.6553059816360474,
      "step": 21
    },
    {
      "barlow_loss": 0.16542162001132965,
      "epoch": 0.08983957219251337,
      "loss": 1.5700926780700684,
      "step": 21
    },
    {
      "barlow_loss": 0.16289600729942322,
      "epoch": 0.08983957219251337,
      "loss": 1.5583313703536987,
      "step": 21
    },
    {
      "barlow_loss": 0.17457102239131927,
      "epoch": 0.09411764705882353,
      "loss": 1.5780285596847534,
      "step": 22
    },
    {
      "barlow_loss": 0.17424727976322174,
      "epoch": 0.09411764705882353,
      "loss": 1.8477423191070557,
      "step": 22
    },
    {
      "barlow_loss": 0.15577517449855804,
      "epoch": 0.09411764705882353,
      "loss": 1.8091905117034912,
      "step": 22
    },
    {
      "barlow_loss": 0.15419355034828186,
      "epoch": 0.09411764705882353,
      "loss": 1.7258429527282715,
      "step": 22
    },
    {
      "barlow_loss": 0.16497600078582764,
      "epoch": 0.09839572192513368,
      "loss": 1.5959159135818481,
      "step": 23
    },
    {
      "barlow_loss": 0.16613352298736572,
      "epoch": 0.09839572192513368,
      "loss": 1.6860661506652832,
      "step": 23
    },
    {
      "barlow_loss": 0.15948894619941711,
      "epoch": 0.09839572192513368,
      "loss": 1.6180912256240845,
      "step": 23
    },
    {
      "barlow_loss": 0.1724734902381897,
      "epoch": 0.09839572192513368,
      "loss": 1.6086125373840332,
      "step": 23
    },
    {
      "epoch": 0.10267379679144385,
      "grad_norm": 0.245857372879982,
      "learning_rate": 2.5e-05,
      "loss": 1.6615,
      "step": 24
    },
    {
      "barlow_loss": 0.15647085011005402,
      "epoch": 0.10267379679144385,
      "loss": 2.041674852371216,
      "step": 24
    },
    {
      "barlow_loss": 0.1719466745853424,
      "epoch": 0.10267379679144385,
      "loss": 1.7762763500213623,
      "step": 24
    },
    {
      "barlow_loss": 0.1642158329486847,
      "epoch": 0.10267379679144385,
      "loss": 1.8694415092468262,
      "step": 24
    },
    {
      "barlow_loss": 0.1530241072177887,
      "epoch": 0.10267379679144385,
      "loss": 1.8207184076309204,
      "step": 24
    },
    {
      "barlow_loss": 0.1639435887336731,
      "epoch": 0.10695187165775401,
      "loss": 1.6309581995010376,
      "step": 25
    },
    {
      "barlow_loss": 0.15103672444820404,
      "epoch": 0.10695187165775401,
      "loss": 1.7647839784622192,
      "step": 25
    },
    {
      "barlow_loss": 0.16237720847129822,
      "epoch": 0.10695187165775401,
      "loss": 1.8172286748886108,
      "step": 25
    },
    {
      "barlow_loss": 0.1558689922094345,
      "epoch": 0.10695187165775401,
      "loss": 1.633702278137207,
      "step": 25
    },
    {
      "barlow_loss": 0.15474200248718262,
      "epoch": 0.11122994652406418,
      "loss": 1.8892121315002441,
      "step": 26
    },
    {
      "barlow_loss": 0.17366531491279602,
      "epoch": 0.11122994652406418,
      "loss": 1.7578625679016113,
      "step": 26
    },
    {
      "barlow_loss": 0.16684305667877197,
      "epoch": 0.11122994652406418,
      "loss": 1.7037280797958374,
      "step": 26
    },
    {
      "barlow_loss": 0.17122101783752441,
      "epoch": 0.11122994652406418,
      "loss": 1.9668108224868774,
      "step": 26
    },
    {
      "barlow_loss": 0.1557237207889557,
      "epoch": 0.11550802139037433,
      "loss": 1.4604514837265015,
      "step": 27
    },
    {
      "barlow_loss": 0.16505755484104156,
      "epoch": 0.11550802139037433,
      "loss": 1.7197670936584473,
      "step": 27
    },
    {
      "barlow_loss": 0.17105066776275635,
      "epoch": 0.11550802139037433,
      "loss": 1.697708249092102,
      "step": 27
    },
    {
      "barlow_loss": 0.16861124336719513,
      "epoch": 0.11550802139037433,
      "loss": 1.8586238622665405,
      "step": 27
    },
    {
      "epoch": 0.11978609625668449,
      "grad_norm": 0.27866652607917786,
      "learning_rate": 2.4994948434913852e-05,
      "loss": 1.7756,
      "step": 28
    },
    {
      "barlow_loss": 0.1448787897825241,
      "epoch": 0.11978609625668449,
      "loss": 1.6587568521499634,
      "step": 28
    },
    {
      "barlow_loss": 0.15055593848228455,
      "epoch": 0.11978609625668449,
      "loss": 1.3539716005325317,
      "step": 28
    },
    {
      "barlow_loss": 0.15717601776123047,
      "epoch": 0.11978609625668449,
      "loss": 1.6498923301696777,
      "step": 28
    },
    {
      "barlow_loss": 0.16670680046081543,
      "epoch": 0.11978609625668449,
      "loss": 1.663672685623169,
      "step": 28
    },
    {
      "barlow_loss": 0.15749046206474304,
      "epoch": 0.12406417112299466,
      "loss": 1.6205414533615112,
      "step": 29
    },
    {
      "barlow_loss": 0.1583046317100525,
      "epoch": 0.12406417112299466,
      "loss": 1.6505569219589233,
      "step": 29
    },
    {
      "barlow_loss": 0.1472414880990982,
      "epoch": 0.12406417112299466,
      "loss": 1.6455942392349243,
      "step": 29
    },
    {
      "barlow_loss": 0.16031493246555328,
      "epoch": 0.12406417112299466,
      "loss": 1.7063720226287842,
      "step": 29
    },
    {
      "barlow_loss": 0.15760846436023712,
      "epoch": 0.12834224598930483,
      "loss": 1.6912870407104492,
      "step": 30
    },
    {
      "barlow_loss": 0.1561989188194275,
      "epoch": 0.12834224598930483,
      "loss": 1.830762505531311,
      "step": 30
    },
    {
      "barlow_loss": 0.156636580824852,
      "epoch": 0.12834224598930483,
      "loss": 1.7782241106033325,
      "step": 30
    },
    {
      "barlow_loss": 0.1539766788482666,
      "epoch": 0.12834224598930483,
      "loss": 1.6385033130645752,
      "step": 30
    },
    {
      "barlow_loss": 0.16341504454612732,
      "epoch": 0.13262032085561498,
      "loss": 1.7587453126907349,
      "step": 31
    },
    {
      "barlow_loss": 0.17107301950454712,
      "epoch": 0.13262032085561498,
      "loss": 1.7462949752807617,
      "step": 31
    },
    {
      "barlow_loss": 0.16206081211566925,
      "epoch": 0.13262032085561498,
      "loss": 1.527721881866455,
      "step": 31
    },
    {
      "barlow_loss": 0.15588338673114777,
      "epoch": 0.13262032085561498,
      "loss": 1.748080849647522,
      "step": 31
    },
    {
      "epoch": 0.13689839572192514,
      "grad_norm": 0.3388544023036957,
      "learning_rate": 2.4979797822584984e-05,
      "loss": 1.6668,
      "step": 32
    },
    {
      "barlow_loss": 0.16415488719940186,
      "epoch": 0.13689839572192514,
      "loss": 1.7937355041503906,
      "step": 32
    },
    {
      "barlow_loss": 0.14962893724441528,
      "epoch": 0.13689839572192514,
      "loss": 1.7251248359680176,
      "step": 32
    },
    {
      "barlow_loss": 0.18225455284118652,
      "epoch": 0.13689839572192514,
      "loss": 1.7323604822158813,
      "step": 32
    },
    {
      "barlow_loss": 0.16859182715415955,
      "epoch": 0.13689839572192514,
      "loss": 1.662642478942871,
      "step": 32
    },
    {
      "barlow_loss": 0.17220677435398102,
      "epoch": 0.1411764705882353,
      "loss": 1.6563365459442139,
      "step": 33
    },
    {
      "barlow_loss": 0.17591622471809387,
      "epoch": 0.1411764705882353,
      "loss": 1.572616457939148,
      "step": 33
    },
    {
      "barlow_loss": 0.18220311403274536,
      "epoch": 0.1411764705882353,
      "loss": 1.6334190368652344,
      "step": 33
    },
    {
      "barlow_loss": 0.163389191031456,
      "epoch": 0.1411764705882353,
      "loss": 1.6118062734603882,
      "step": 33
    },
    {
      "barlow_loss": 0.15492798388004303,
      "epoch": 0.14545454545454545,
      "loss": 1.6966552734375,
      "step": 34
    },
    {
      "barlow_loss": 0.1632213294506073,
      "epoch": 0.14545454545454545,
      "loss": 1.8359284400939941,
      "step": 34
    },
    {
      "barlow_loss": 0.16788245737552643,
      "epoch": 0.14545454545454545,
      "loss": 1.740525484085083,
      "step": 34
    },
    {
      "barlow_loss": 0.1531849205493927,
      "epoch": 0.14545454545454545,
      "loss": 1.6319278478622437,
      "step": 34
    },
    {
      "barlow_loss": 0.1581384241580963,
      "epoch": 0.1497326203208556,
      "loss": 1.8486456871032715,
      "step": 35
    },
    {
      "barlow_loss": 0.1526244729757309,
      "epoch": 0.1497326203208556,
      "loss": 1.4847880601882935,
      "step": 35
    },
    {
      "barlow_loss": 0.1665688455104828,
      "epoch": 0.1497326203208556,
      "loss": 1.699148416519165,
      "step": 35
    },
    {
      "barlow_loss": 0.16243930160999298,
      "epoch": 0.1497326203208556,
      "loss": 1.5482001304626465,
      "step": 35
    },
    {
      "epoch": 0.15401069518716579,
      "grad_norm": 0.3178236782550812,
      "learning_rate": 2.4954560408502066e-05,
      "loss": 1.6796,
      "step": 36
    },
    {
      "barlow_loss": 0.14855574071407318,
      "epoch": 0.15401069518716579,
      "loss": 1.5175374746322632,
      "step": 36
    },
    {
      "barlow_loss": 0.1571977138519287,
      "epoch": 0.15401069518716579,
      "loss": 1.4532427787780762,
      "step": 36
    },
    {
      "barlow_loss": 0.1670173555612564,
      "epoch": 0.15401069518716579,
      "loss": 1.6794296503067017,
      "step": 36
    },
    {
      "barlow_loss": 0.16200628876686096,
      "epoch": 0.15401069518716579,
      "loss": 1.6790173053741455,
      "step": 36
    },
    {
      "barlow_loss": 0.16843099892139435,
      "epoch": 0.15828877005347594,
      "loss": 1.6696730852127075,
      "step": 37
    },
    {
      "barlow_loss": 0.15435117483139038,
      "epoch": 0.15828877005347594,
      "loss": 1.5999253988265991,
      "step": 37
    },
    {
      "barlow_loss": 0.1623205989599228,
      "epoch": 0.15828877005347594,
      "loss": 1.9163687229156494,
      "step": 37
    },
    {
      "barlow_loss": 0.16284434497356415,
      "epoch": 0.15828877005347594,
      "loss": 1.527005910873413,
      "step": 37
    },
    {
      "barlow_loss": 0.1620793342590332,
      "epoch": 0.1625668449197861,
      "loss": 1.7334372997283936,
      "step": 38
    },
    {
      "barlow_loss": 0.16382403671741486,
      "epoch": 0.1625668449197861,
      "loss": 1.684836745262146,
      "step": 38
    },
    {
      "barlow_loss": 0.166356161236763,
      "epoch": 0.1625668449197861,
      "loss": 1.4024871587753296,
      "step": 38
    },
    {
      "barlow_loss": 0.149616077542305,
      "epoch": 0.1625668449197861,
      "loss": 1.8562943935394287,
      "step": 38
    },
    {
      "barlow_loss": 0.17990975081920624,
      "epoch": 0.16684491978609625,
      "loss": 1.5777137279510498,
      "step": 39
    },
    {
      "barlow_loss": 0.17378996312618256,
      "epoch": 0.16684491978609625,
      "loss": 2.023001194000244,
      "step": 39
    },
    {
      "barlow_loss": 0.1544182300567627,
      "epoch": 0.16684491978609625,
      "loss": 1.6466169357299805,
      "step": 39
    },
    {
      "barlow_loss": 0.16323333978652954,
      "epoch": 0.16684491978609625,
      "loss": 1.7313926219940186,
      "step": 39
    },
    {
      "epoch": 0.1711229946524064,
      "grad_norm": 0.36824730038642883,
      "learning_rate": 2.4919256590815486e-05,
      "loss": 1.6686,
      "step": 40
    },
    {
      "barlow_loss": 0.16520389914512634,
      "epoch": 0.1711229946524064,
      "loss": 1.8408037424087524,
      "step": 40
    },
    {
      "barlow_loss": 0.17006729543209076,
      "epoch": 0.1711229946524064,
      "loss": 1.662778377532959,
      "step": 40
    },
    {
      "barlow_loss": 0.1853155493736267,
      "epoch": 0.1711229946524064,
      "loss": 1.6776635646820068,
      "step": 40
    },
    {
      "barlow_loss": 0.16311444342136383,
      "epoch": 0.1711229946524064,
      "loss": 1.4881106615066528,
      "step": 40
    },
    {
      "barlow_loss": 0.14668023586273193,
      "epoch": 0.1754010695187166,
      "loss": 1.7121309041976929,
      "step": 41
    },
    {
      "barlow_loss": 0.1672130525112152,
      "epoch": 0.1754010695187166,
      "loss": 1.6183340549468994,
      "step": 41
    },
    {
      "barlow_loss": 0.15681864321231842,
      "epoch": 0.1754010695187166,
      "loss": 1.6616437435150146,
      "step": 41
    },
    {
      "barlow_loss": 0.18166722357273102,
      "epoch": 0.1754010695187166,
      "loss": 1.6139909029006958,
      "step": 41
    },
    {
      "barlow_loss": 0.15547063946723938,
      "epoch": 0.17967914438502675,
      "loss": 1.6197926998138428,
      "step": 42
    },
    {
      "barlow_loss": 0.16404621303081512,
      "epoch": 0.17967914438502675,
      "loss": 1.734810709953308,
      "step": 42
    },
    {
      "barlow_loss": 0.16853784024715424,
      "epoch": 0.17967914438502675,
      "loss": 1.7151261568069458,
      "step": 42
    },
    {
      "barlow_loss": 0.14535783231258392,
      "epoch": 0.17967914438502675,
      "loss": 1.563204288482666,
      "step": 42
    },
    {
      "barlow_loss": 0.16207252442836761,
      "epoch": 0.1839572192513369,
      "loss": 1.710302472114563,
      "step": 43
    },
    {
      "barlow_loss": 0.15921103954315186,
      "epoch": 0.1839572192513369,
      "loss": 1.5956672430038452,
      "step": 43
    },
    {
      "barlow_loss": 0.16454145312309265,
      "epoch": 0.1839572192513369,
      "loss": 1.719834327697754,
      "step": 43
    },
    {
      "barlow_loss": 0.15133118629455566,
      "epoch": 0.1839572192513369,
      "loss": 1.8209890127182007,
      "step": 43
    },
    {
      "epoch": 0.18823529411764706,
      "grad_norm": 0.3723647892475128,
      "learning_rate": 2.48739149038505e-05,
      "loss": 1.6722,
      "step": 44
    },
    {
      "barlow_loss": 0.17030441761016846,
      "epoch": 0.18823529411764706,
      "loss": 1.7632513046264648,
      "step": 44
    },
    {
      "barlow_loss": 0.17599354684352875,
      "epoch": 0.18823529411764706,
      "loss": 1.7938435077667236,
      "step": 44
    },
    {
      "barlow_loss": 0.15399537980556488,
      "epoch": 0.18823529411764706,
      "loss": 1.6355829238891602,
      "step": 44
    },
    {
      "barlow_loss": 0.16696800291538239,
      "epoch": 0.18823529411764706,
      "loss": 1.688047170639038,
      "step": 44
    },
    {
      "barlow_loss": 0.16660334169864655,
      "epoch": 0.1925133689839572,
      "loss": 1.937117576599121,
      "step": 45
    },
    {
      "barlow_loss": 0.1656961441040039,
      "epoch": 0.1925133689839572,
      "loss": 1.4777483940124512,
      "step": 45
    },
    {
      "barlow_loss": 0.15889710187911987,
      "epoch": 0.1925133689839572,
      "loss": 1.4034969806671143,
      "step": 45
    },
    {
      "barlow_loss": 0.16849839687347412,
      "epoch": 0.1925133689839572,
      "loss": 1.722142219543457,
      "step": 45
    },
    {
      "barlow_loss": 0.1656748652458191,
      "epoch": 0.19679144385026737,
      "loss": 1.3900363445281982,
      "step": 46
    },
    {
      "barlow_loss": 0.1624937355518341,
      "epoch": 0.19679144385026737,
      "loss": 1.6809660196304321,
      "step": 46
    },
    {
      "barlow_loss": 0.16892464458942413,
      "epoch": 0.19679144385026737,
      "loss": 1.7349110841751099,
      "step": 46
    },
    {
      "barlow_loss": 0.17394572496414185,
      "epoch": 0.19679144385026737,
      "loss": 1.8635200262069702,
      "step": 46
    },
    {
      "barlow_loss": 0.1683494746685028,
      "epoch": 0.20106951871657755,
      "loss": 1.602893590927124,
      "step": 47
    },
    {
      "barlow_loss": 0.1700470894575119,
      "epoch": 0.20106951871657755,
      "loss": 1.8583382368087769,
      "step": 47
    },
    {
      "barlow_loss": 0.17315849661827087,
      "epoch": 0.20106951871657755,
      "loss": 1.7582087516784668,
      "step": 47
    },
    {
      "barlow_loss": 0.1675821989774704,
      "epoch": 0.20106951871657755,
      "loss": 1.4242912530899048,
      "step": 47
    },
    {
      "epoch": 0.2053475935828877,
      "grad_norm": 0.4102187156677246,
      "learning_rate": 2.4818571995044346e-05,
      "loss": 1.6709,
      "step": 48
    },
    {
      "barlow_loss": 0.15118470788002014,
      "epoch": 0.2053475935828877,
      "loss": 1.4334715604782104,
      "step": 48
    },
    {
      "barlow_loss": 0.17196477949619293,
      "epoch": 0.2053475935828877,
      "loss": 1.5671590566635132,
      "step": 48
    },
    {
      "barlow_loss": 0.16472674906253815,
      "epoch": 0.2053475935828877,
      "loss": 1.8914384841918945,
      "step": 48
    },
    {
      "barlow_loss": 0.16977357864379883,
      "epoch": 0.2053475935828877,
      "loss": 1.6278672218322754,
      "step": 48
    },
    {
      "barlow_loss": 0.15749713778495789,
      "epoch": 0.20962566844919786,
      "loss": 1.6682417392730713,
      "step": 49
    },
    {
      "barlow_loss": 0.16315527260303497,
      "epoch": 0.20962566844919786,
      "loss": 1.6177152395248413,
      "step": 49
    },
    {
      "barlow_loss": 0.16695979237556458,
      "epoch": 0.20962566844919786,
      "loss": 1.7157902717590332,
      "step": 49
    },
    {
      "barlow_loss": 0.16415473818778992,
      "epoch": 0.20962566844919786,
      "loss": 1.661434292793274,
      "step": 49
    },
    {
      "barlow_loss": 0.18272045254707336,
      "epoch": 0.21390374331550802,
      "loss": 1.5857294797897339,
      "step": 50
    },
    {
      "barlow_loss": 0.18641123175621033,
      "epoch": 0.21390374331550802,
      "loss": 1.7893203496932983,
      "step": 50
    },
    {
      "barlow_loss": 0.16921687126159668,
      "epoch": 0.21390374331550802,
      "loss": 1.8841160535812378,
      "step": 50
    },
    {
      "barlow_loss": 0.15025575459003448,
      "epoch": 0.21390374331550802,
      "loss": 1.6366788148880005,
      "step": 50
    },
    {
      "barlow_loss": 0.1745861917734146,
      "epoch": 0.21818181818181817,
      "loss": 1.7268013954162598,
      "step": 51
    },
    {
      "barlow_loss": 0.156719371676445,
      "epoch": 0.21818181818181817,
      "loss": 1.630128264427185,
      "step": 51
    },
    {
      "barlow_loss": 0.17507393658161163,
      "epoch": 0.21818181818181817,
      "loss": 1.6867221593856812,
      "step": 51
    },
    {
      "barlow_loss": 0.17788220942020416,
      "epoch": 0.21818181818181817,
      "loss": 1.5838754177093506,
      "step": 51
    },
    {
      "epoch": 0.22245989304812835,
      "grad_norm": 0.3930147886276245,
      "learning_rate": 2.4753272595325976e-05,
      "loss": 1.6692,
      "step": 52
    },
    {
      "barlow_loss": 0.1746867597103119,
      "epoch": 0.22245989304812835,
      "loss": 1.611845850944519,
      "step": 52
    },
    {
      "barlow_loss": 0.15941151976585388,
      "epoch": 0.22245989304812835,
      "loss": 1.7059894800186157,
      "step": 52
    },
    {
      "barlow_loss": 0.15882128477096558,
      "epoch": 0.22245989304812835,
      "loss": 1.7706851959228516,
      "step": 52
    },
    {
      "barlow_loss": 0.14734356105327606,
      "epoch": 0.22245989304812835,
      "loss": 1.7196292877197266,
      "step": 52
    },
    {
      "barlow_loss": 0.17121373116970062,
      "epoch": 0.2267379679144385,
      "loss": 1.632250428199768,
      "step": 53
    },
    {
      "barlow_loss": 0.16420862078666687,
      "epoch": 0.2267379679144385,
      "loss": 1.8511340618133545,
      "step": 53
    },
    {
      "barlow_loss": 0.16979339718818665,
      "epoch": 0.2267379679144385,
      "loss": 1.6204756498336792,
      "step": 53
    },
    {
      "barlow_loss": 0.1665061116218567,
      "epoch": 0.2267379679144385,
      "loss": 1.7034058570861816,
      "step": 53
    },
    {
      "barlow_loss": 0.15157772600650787,
      "epoch": 0.23101604278074866,
      "loss": 1.3843978643417358,
      "step": 54
    },
    {
      "barlow_loss": 0.1664179563522339,
      "epoch": 0.23101604278074866,
      "loss": 1.780599594116211,
      "step": 54
    },
    {
      "barlow_loss": 0.18062923848628998,
      "epoch": 0.23101604278074866,
      "loss": 1.7096967697143555,
      "step": 54
    },
    {
      "barlow_loss": 0.16063818335533142,
      "epoch": 0.23101604278074866,
      "loss": 1.6743810176849365,
      "step": 54
    },
    {
      "barlow_loss": 0.17024219036102295,
      "epoch": 0.23529411764705882,
      "loss": 1.761144995689392,
      "step": 55
    },
    {
      "barlow_loss": 0.15758037567138672,
      "epoch": 0.23529411764705882,
      "loss": 1.4423748254776,
      "step": 55
    },
    {
      "barlow_loss": 0.17529235780239105,
      "epoch": 0.23529411764705882,
      "loss": 1.6715879440307617,
      "step": 55
    },
    {
      "barlow_loss": 0.16181297600269318,
      "epoch": 0.23529411764705882,
      "loss": 1.5712162256240845,
      "step": 55
    },
    {
      "epoch": 0.23957219251336898,
      "grad_norm": 0.4056495428085327,
      "learning_rate": 2.467806948296223e-05,
      "loss": 1.6632,
      "step": 56
    },
    {
      "barlow_loss": 0.1546171009540558,
      "epoch": 0.23957219251336898,
      "loss": 1.8778797388076782,
      "step": 56
    },
    {
      "barlow_loss": 0.15422707796096802,
      "epoch": 0.23957219251336898,
      "loss": 1.773299217224121,
      "step": 56
    },
    {
      "barlow_loss": 0.15896932780742645,
      "epoch": 0.23957219251336898,
      "loss": 1.564217448234558,
      "step": 56
    },
    {
      "barlow_loss": 0.15872028470039368,
      "epoch": 0.23957219251336898,
      "loss": 1.5201245546340942,
      "step": 56
    },
    {
      "barlow_loss": 0.16673491895198822,
      "epoch": 0.24385026737967913,
      "loss": 1.619876742362976,
      "step": 57
    },
    {
      "barlow_loss": 0.17192408442497253,
      "epoch": 0.24385026737967913,
      "loss": 1.6997820138931274,
      "step": 57
    },
    {
      "barlow_loss": 0.16818903386592865,
      "epoch": 0.24385026737967913,
      "loss": 1.6625871658325195,
      "step": 57
    },
    {
      "barlow_loss": 0.1776949018239975,
      "epoch": 0.24385026737967913,
      "loss": 1.777388095855713,
      "step": 57
    },
    {
      "barlow_loss": 0.1693827509880066,
      "epoch": 0.24812834224598931,
      "loss": 1.6497406959533691,
      "step": 58
    },
    {
      "barlow_loss": 0.16086909174919128,
      "epoch": 0.24812834224598931,
      "loss": 1.559453010559082,
      "step": 58
    },
    {
      "barlow_loss": 0.1821274757385254,
      "epoch": 0.24812834224598931,
      "loss": 1.5788769721984863,
      "step": 58
    },
    {
      "barlow_loss": 0.1395736038684845,
      "epoch": 0.24812834224598931,
      "loss": 1.5262309312820435,
      "step": 58
    },
    {
      "barlow_loss": 0.16184790432453156,
      "epoch": 0.25240641711229944,
      "loss": 1.6383435726165771,
      "step": 59
    },
    {
      "barlow_loss": 0.16542984545230865,
      "epoch": 0.25240641711229944,
      "loss": 1.8743122816085815,
      "step": 59
    },
    {
      "barlow_loss": 0.15865930914878845,
      "epoch": 0.25240641711229944,
      "loss": 1.6166536808013916,
      "step": 59
    },
    {
      "barlow_loss": 0.17299897968769073,
      "epoch": 0.25240641711229944,
      "loss": 1.8130767345428467,
      "step": 59
    },
    {
      "epoch": 0.25668449197860965,
      "grad_norm": 0.5045633912086487,
      "learning_rate": 2.459302344089979e-05,
      "loss": 1.672,
      "step": 60
    },
    {
      "barlow_loss": 0.14702536165714264,
      "epoch": 0.25668449197860965,
      "loss": 1.5300711393356323,
      "step": 60
    },
    {
      "barlow_loss": 0.1661507487297058,
      "epoch": 0.25668449197860965,
      "loss": 1.6484835147857666,
      "step": 60
    },
    {
      "barlow_loss": 0.14241595566272736,
      "epoch": 0.25668449197860965,
      "loss": 1.679790735244751,
      "step": 60
    },
    {
      "barlow_loss": 0.14461328089237213,
      "epoch": 0.25668449197860965,
      "loss": 1.6623977422714233,
      "step": 60
    },
    {
      "barlow_loss": 0.1570630669593811,
      "epoch": 0.2609625668449198,
      "loss": 1.4895479679107666,
      "step": 61
    },
    {
      "barlow_loss": 0.1671610027551651,
      "epoch": 0.2609625668449198,
      "loss": 1.4977598190307617,
      "step": 61
    },
    {
      "barlow_loss": 0.17437195777893066,
      "epoch": 0.2609625668449198,
      "loss": 1.7785601615905762,
      "step": 61
    },
    {
      "barlow_loss": 0.16043245792388916,
      "epoch": 0.2609625668449198,
      "loss": 1.5768665075302124,
      "step": 61
    },
    {
      "barlow_loss": 0.1673382818698883,
      "epoch": 0.26524064171122996,
      "loss": 1.7108575105667114,
      "step": 62
    },
    {
      "barlow_loss": 0.16054287552833557,
      "epoch": 0.26524064171122996,
      "loss": 1.5529330968856812,
      "step": 62
    },
    {
      "barlow_loss": 0.1623287945985794,
      "epoch": 0.26524064171122996,
      "loss": 1.6611227989196777,
      "step": 62
    },
    {
      "barlow_loss": 0.1521051526069641,
      "epoch": 0.26524064171122996,
      "loss": 1.6539918184280396,
      "step": 62
    },
    {
      "barlow_loss": 0.1669902503490448,
      "epoch": 0.2695187165775401,
      "loss": 1.7227333784103394,
      "step": 63
    },
    {
      "barlow_loss": 0.16960734128952026,
      "epoch": 0.2695187165775401,
      "loss": 1.7157281637191772,
      "step": 63
    },
    {
      "barlow_loss": 0.1684102863073349,
      "epoch": 0.2695187165775401,
      "loss": 1.7408638000488281,
      "step": 63
    },
    {
      "barlow_loss": 0.1659426987171173,
      "epoch": 0.2695187165775401,
      "loss": 1.7793700695037842,
      "step": 63
    },
    {
      "epoch": 0.2737967914438503,
      "grad_norm": 0.5523879528045654,
      "learning_rate": 2.449820320763735e-05,
      "loss": 1.6501,
      "step": 64
    },
    {
      "barlow_loss": 0.1676473766565323,
      "epoch": 0.2737967914438503,
      "loss": 1.647491216659546,
      "step": 64
    },
    {
      "barlow_loss": 0.1662994921207428,
      "epoch": 0.2737967914438503,
      "loss": 1.5934703350067139,
      "step": 64
    },
    {
      "barlow_loss": 0.1655750274658203,
      "epoch": 0.2737967914438503,
      "loss": 1.7252857685089111,
      "step": 64
    },
    {
      "barlow_loss": 0.1577088087797165,
      "epoch": 0.2737967914438503,
      "loss": 1.564756989479065,
      "step": 64
    },
    {
      "barlow_loss": 0.1573389172554016,
      "epoch": 0.27807486631016043,
      "loss": 1.749711275100708,
      "step": 65
    },
    {
      "barlow_loss": 0.15923874080181122,
      "epoch": 0.27807486631016043,
      "loss": 1.5985392332077026,
      "step": 65
    },
    {
      "barlow_loss": 0.16215190291404724,
      "epoch": 0.27807486631016043,
      "loss": 1.796593189239502,
      "step": 65
    },
    {
      "barlow_loss": 0.17042309045791626,
      "epoch": 0.27807486631016043,
      "loss": 1.7676894664764404,
      "step": 65
    },
    {
      "barlow_loss": 0.14912953972816467,
      "epoch": 0.2823529411764706,
      "loss": 1.5891350507736206,
      "step": 66
    },
    {
      "barlow_loss": 0.16341611742973328,
      "epoch": 0.2823529411764706,
      "loss": 1.844146728515625,
      "step": 66
    },
    {
      "barlow_loss": 0.15056465566158295,
      "epoch": 0.2823529411764706,
      "loss": 1.5575562715530396,
      "step": 66
    },
    {
      "barlow_loss": 0.1613842397928238,
      "epoch": 0.2823529411764706,
      "loss": 1.4749189615249634,
      "step": 66
    },
    {
      "barlow_loss": 0.1699976921081543,
      "epoch": 0.28663101604278074,
      "loss": 1.7963787317276,
      "step": 67
    },
    {
      "barlow_loss": 0.17151038348674774,
      "epoch": 0.28663101604278074,
      "loss": 1.7424514293670654,
      "step": 67
    },
    {
      "barlow_loss": 0.15893328189849854,
      "epoch": 0.28663101604278074,
      "loss": 1.8765177726745605,
      "step": 67
    },
    {
      "barlow_loss": 0.15353800356388092,
      "epoch": 0.28663101604278074,
      "loss": 1.6389739513397217,
      "step": 67
    },
    {
      "epoch": 0.2909090909090909,
      "grad_norm": 0.5557935833930969,
      "learning_rate": 2.4393685421667677e-05,
      "loss": 1.6852,
      "step": 68
    },
    {
      "barlow_loss": 0.16258592903614044,
      "epoch": 0.2909090909090909,
      "loss": 1.6506996154785156,
      "step": 68
    },
    {
      "barlow_loss": 0.1822413206100464,
      "epoch": 0.2909090909090909,
      "loss": 1.7162235975265503,
      "step": 68
    },
    {
      "barlow_loss": 0.1560390591621399,
      "epoch": 0.2909090909090909,
      "loss": 1.8098477125167847,
      "step": 68
    },
    {
      "barlow_loss": 0.1722884178161621,
      "epoch": 0.2909090909090909,
      "loss": 1.6791046857833862,
      "step": 68
    },
    {
      "barlow_loss": 0.16191378235816956,
      "epoch": 0.29518716577540105,
      "loss": 1.525629997253418,
      "step": 69
    },
    {
      "barlow_loss": 0.16873304545879364,
      "epoch": 0.29518716577540105,
      "loss": 1.6913870573043823,
      "step": 69
    },
    {
      "barlow_loss": 0.16460488736629486,
      "epoch": 0.29518716577540105,
      "loss": 1.4852555990219116,
      "step": 69
    },
    {
      "barlow_loss": 0.1661253422498703,
      "epoch": 0.29518716577540105,
      "loss": 1.7679693698883057,
      "step": 69
    },
    {
      "barlow_loss": 0.15689118206501007,
      "epoch": 0.2994652406417112,
      "loss": 1.5529489517211914,
      "step": 70
    },
    {
      "barlow_loss": 0.16426092386245728,
      "epoch": 0.2994652406417112,
      "loss": 1.3777538537979126,
      "step": 70
    },
    {
      "barlow_loss": 0.18176884949207306,
      "epoch": 0.2994652406417112,
      "loss": 1.6948890686035156,
      "step": 70
    },
    {
      "barlow_loss": 0.16438813507556915,
      "epoch": 0.2994652406417112,
      "loss": 1.6100175380706787,
      "step": 70
    },
    {
      "barlow_loss": 0.16257739067077637,
      "epoch": 0.3037433155080214,
      "loss": 1.7111226320266724,
      "step": 71
    },
    {
      "barlow_loss": 0.158912792801857,
      "epoch": 0.3037433155080214,
      "loss": 1.581896185874939,
      "step": 71
    },
    {
      "barlow_loss": 0.15168407559394836,
      "epoch": 0.3037433155080214,
      "loss": 1.7556819915771484,
      "step": 71
    },
    {
      "barlow_loss": 0.151200070977211,
      "epoch": 0.3037433155080214,
      "loss": 1.6859999895095825,
      "step": 71
    },
    {
      "epoch": 0.30802139037433157,
      "grad_norm": 0.5373147130012512,
      "learning_rate": 2.427955455953453e-05,
      "loss": 1.6435,
      "step": 72
    },
    {
      "barlow_loss": 0.1571812927722931,
      "epoch": 0.30802139037433157,
      "loss": 1.6518144607543945,
      "step": 72
    },
    {
      "barlow_loss": 0.16813497245311737,
      "epoch": 0.30802139037433157,
      "loss": 1.497619867324829,
      "step": 72
    },
    {
      "barlow_loss": 0.14594432711601257,
      "epoch": 0.30802139037433157,
      "loss": 1.4196484088897705,
      "step": 72
    },
    {
      "barlow_loss": 0.16348160803318024,
      "epoch": 0.30802139037433157,
      "loss": 1.8398410081863403,
      "step": 72
    },
    {
      "barlow_loss": 0.17378953099250793,
      "epoch": 0.3122994652406417,
      "loss": 1.671456217765808,
      "step": 73
    },
    {
      "barlow_loss": 0.16685524582862854,
      "epoch": 0.3122994652406417,
      "loss": 1.5790425539016724,
      "step": 73
    },
    {
      "barlow_loss": 0.16752614080905914,
      "epoch": 0.3122994652406417,
      "loss": 1.7327203750610352,
      "step": 73
    },
    {
      "barlow_loss": 0.16559304296970367,
      "epoch": 0.3122994652406417,
      "loss": 1.820792317390442,
      "step": 73
    },
    {
      "barlow_loss": 0.16035322844982147,
      "epoch": 0.3165775401069519,
      "loss": 1.5625611543655396,
      "step": 74
    },
    {
      "barlow_loss": 0.15720324218273163,
      "epoch": 0.3165775401069519,
      "loss": 1.7397788763046265,
      "step": 74
    },
    {
      "barlow_loss": 0.1506081074476242,
      "epoch": 0.3165775401069519,
      "loss": 1.6041369438171387,
      "step": 74
    },
    {
      "barlow_loss": 0.17234838008880615,
      "epoch": 0.3165775401069519,
      "loss": 1.728597640991211,
      "step": 74
    },
    {
      "barlow_loss": 0.15772005915641785,
      "epoch": 0.32085561497326204,
      "loss": 1.5676183700561523,
      "step": 75
    },
    {
      "barlow_loss": 0.17499001324176788,
      "epoch": 0.32085561497326204,
      "loss": 1.6605027914047241,
      "step": 75
    },
    {
      "barlow_loss": 0.16962599754333496,
      "epoch": 0.32085561497326204,
      "loss": 1.766018271446228,
      "step": 75
    },
    {
      "barlow_loss": 0.13991861045360565,
      "epoch": 0.32085561497326204,
      "loss": 1.5415972471237183,
      "step": 75
    },
    {
      "epoch": 0.3251336898395722,
      "grad_norm": 0.525235652923584,
      "learning_rate": 2.4155902867554447e-05,
      "loss": 1.649,
      "step": 76
    },
    {
      "barlow_loss": 0.17605067789554596,
      "epoch": 0.3251336898395722,
      "loss": 1.841238021850586,
      "step": 76
    },
    {
      "barlow_loss": 0.169071227312088,
      "epoch": 0.3251336898395722,
      "loss": 1.880579948425293,
      "step": 76
    },
    {
      "barlow_loss": 0.15920963883399963,
      "epoch": 0.3251336898395722,
      "loss": 1.8004504442214966,
      "step": 76
    },
    {
      "barlow_loss": 0.17413705587387085,
      "epoch": 0.3251336898395722,
      "loss": 1.5877491235733032,
      "step": 76
    },
    {
      "barlow_loss": 0.15649473667144775,
      "epoch": 0.32941176470588235,
      "loss": 1.7068980932235718,
      "step": 77
    },
    {
      "barlow_loss": 0.15414363145828247,
      "epoch": 0.32941176470588235,
      "loss": 1.6242761611938477,
      "step": 77
    },
    {
      "barlow_loss": 0.1706146001815796,
      "epoch": 0.32941176470588235,
      "loss": 1.6200345754623413,
      "step": 77
    },
    {
      "barlow_loss": 0.16227774322032928,
      "epoch": 0.32941176470588235,
      "loss": 1.7221654653549194,
      "step": 77
    },
    {
      "barlow_loss": 0.14975041151046753,
      "epoch": 0.3336898395721925,
      "loss": 1.7510122060775757,
      "step": 78
    },
    {
      "barlow_loss": 0.17424245178699493,
      "epoch": 0.3336898395721925,
      "loss": 1.6223715543746948,
      "step": 78
    },
    {
      "barlow_loss": 0.1717989295721054,
      "epoch": 0.3336898395721925,
      "loss": 1.4894354343414307,
      "step": 78
    },
    {
      "barlow_loss": 0.17341038584709167,
      "epoch": 0.3336898395721925,
      "loss": 1.523658275604248,
      "step": 78
    },
    {
      "barlow_loss": 0.15611328184604645,
      "epoch": 0.33796791443850266,
      "loss": 1.5432237386703491,
      "step": 79
    },
    {
      "barlow_loss": 0.14972448348999023,
      "epoch": 0.33796791443850266,
      "loss": 1.4384770393371582,
      "step": 79
    },
    {
      "barlow_loss": 0.16935020685195923,
      "epoch": 0.33796791443850266,
      "loss": 1.6158287525177002,
      "step": 79
    },
    {
      "barlow_loss": 0.15065133571624756,
      "epoch": 0.33796791443850266,
      "loss": 1.4449650049209595,
      "step": 79
    },
    {
      "epoch": 0.3422459893048128,
      "grad_norm": 0.5081396698951721,
      "learning_rate": 2.4022830287258646e-05,
      "loss": 1.6383,
      "step": 80
    },
    {
      "barlow_loss": 0.16398173570632935,
      "epoch": 0.3422459893048128,
      "loss": 1.6651028394699097,
      "step": 80
    },
    {
      "barlow_loss": 0.16368262469768524,
      "epoch": 0.3422459893048128,
      "loss": 1.584667682647705,
      "step": 80
    },
    {
      "barlow_loss": 0.15717267990112305,
      "epoch": 0.3422459893048128,
      "loss": 1.6741849184036255,
      "step": 80
    },
    {
      "barlow_loss": 0.16012999415397644,
      "epoch": 0.3422459893048128,
      "loss": 1.6585325002670288,
      "step": 80
    },
    {
      "barlow_loss": 0.15835067629814148,
      "epoch": 0.34652406417112297,
      "loss": 1.751094102859497,
      "step": 81
    },
    {
      "barlow_loss": 0.14053408801555634,
      "epoch": 0.34652406417112297,
      "loss": 1.6372278928756714,
      "step": 81
    },
    {
      "barlow_loss": 0.17709480226039886,
      "epoch": 0.34652406417112297,
      "loss": 1.84829580783844,
      "step": 81
    },
    {
      "barlow_loss": 0.16200761497020721,
      "epoch": 0.34652406417112297,
      "loss": 1.89885675907135,
      "step": 81
    },
    {
      "barlow_loss": 0.1633906215429306,
      "epoch": 0.3508021390374332,
      "loss": 1.6401863098144531,
      "step": 82
    },
    {
      "barlow_loss": 0.18481212854385376,
      "epoch": 0.3508021390374332,
      "loss": 1.8483372926712036,
      "step": 82
    },
    {
      "barlow_loss": 0.1823607236146927,
      "epoch": 0.3508021390374332,
      "loss": 1.660907506942749,
      "step": 82
    },
    {
      "barlow_loss": 0.16827046871185303,
      "epoch": 0.3508021390374332,
      "loss": 1.6253180503845215,
      "step": 82
    },
    {
      "barlow_loss": 0.16590216755867004,
      "epoch": 0.35508021390374334,
      "loss": 1.8279173374176025,
      "step": 83
    },
    {
      "barlow_loss": 0.16860514879226685,
      "epoch": 0.35508021390374334,
      "loss": 1.785931944847107,
      "step": 83
    },
    {
      "barlow_loss": 0.13492904603481293,
      "epoch": 0.35508021390374334,
      "loss": 1.7283858060836792,
      "step": 83
    },
    {
      "barlow_loss": 0.15387167036533356,
      "epoch": 0.35508021390374334,
      "loss": 1.5901944637298584,
      "step": 83
    },
    {
      "epoch": 0.3593582887700535,
      "grad_norm": 0.6334472894668579,
      "learning_rate": 2.3880444374615208e-05,
      "loss": 1.7141,
      "step": 84
    },
    {
      "barlow_loss": 0.1600218117237091,
      "epoch": 0.3593582887700535,
      "loss": 1.4954262971878052,
      "step": 84
    },
    {
      "barlow_loss": 0.18623362481594086,
      "epoch": 0.3593582887700535,
      "loss": 1.5882664918899536,
      "step": 84
    },
    {
      "barlow_loss": 0.16072221100330353,
      "epoch": 0.3593582887700535,
      "loss": 1.5907753705978394,
      "step": 84
    },
    {
      "barlow_loss": 0.16705524921417236,
      "epoch": 0.3593582887700535,
      "loss": 1.6383450031280518,
      "step": 84
    },
    {
      "barlow_loss": 0.16776926815509796,
      "epoch": 0.36363636363636365,
      "loss": 1.6602816581726074,
      "step": 85
    },
    {
      "barlow_loss": 0.16087746620178223,
      "epoch": 0.36363636363636365,
      "loss": 1.6186431646347046,
      "step": 85
    },
    {
      "barlow_loss": 0.16176648437976837,
      "epoch": 0.36363636363636365,
      "loss": 1.4512255191802979,
      "step": 85
    },
    {
      "barlow_loss": 0.1646393984556198,
      "epoch": 0.36363636363636365,
      "loss": 1.7621599435806274,
      "step": 85
    },
    {
      "barlow_loss": 0.16251057386398315,
      "epoch": 0.3679144385026738,
      "loss": 1.457538366317749,
      "step": 86
    },
    {
      "barlow_loss": 0.15853163599967957,
      "epoch": 0.3679144385026738,
      "loss": 1.7606936693191528,
      "step": 86
    },
    {
      "barlow_loss": 0.16013330221176147,
      "epoch": 0.3679144385026738,
      "loss": 1.5479881763458252,
      "step": 86
    },
    {
      "barlow_loss": 0.16978669166564941,
      "epoch": 0.3679144385026738,
      "loss": 1.8404866456985474,
      "step": 86
    },
    {
      "barlow_loss": 0.16077587008476257,
      "epoch": 0.37219251336898396,
      "loss": 1.6156474351882935,
      "step": 87
    },
    {
      "barlow_loss": 0.16075465083122253,
      "epoch": 0.37219251336898396,
      "loss": 1.5930768251419067,
      "step": 87
    },
    {
      "barlow_loss": 0.16548261046409607,
      "epoch": 0.37219251336898396,
      "loss": 1.7049682140350342,
      "step": 87
    },
    {
      "barlow_loss": 0.1627115160226822,
      "epoch": 0.37219251336898396,
      "loss": 1.7351033687591553,
      "step": 87
    },
    {
      "epoch": 0.3764705882352941,
      "grad_norm": 0.6128621697425842,
      "learning_rate": 2.372886021309694e-05,
      "loss": 1.6288,
      "step": 88
    },
    {
      "barlow_loss": 0.17880256474018097,
      "epoch": 0.3764705882352941,
      "loss": 1.895533561706543,
      "step": 88
    },
    {
      "barlow_loss": 0.1837157905101776,
      "epoch": 0.3764705882352941,
      "loss": 1.5632797479629517,
      "step": 88
    },
    {
      "barlow_loss": 0.17011809349060059,
      "epoch": 0.3764705882352941,
      "loss": 1.7628364562988281,
      "step": 88
    },
    {
      "barlow_loss": 0.16453199088573456,
      "epoch": 0.3764705882352941,
      "loss": 1.6353366374969482,
      "step": 88
    },
    {
      "barlow_loss": 0.17552538216114044,
      "epoch": 0.38074866310160427,
      "loss": 1.7840886116027832,
      "step": 89
    },
    {
      "barlow_loss": 0.16900596022605896,
      "epoch": 0.38074866310160427,
      "loss": 1.7063047885894775,
      "step": 89
    },
    {
      "barlow_loss": 0.1715012788772583,
      "epoch": 0.38074866310160427,
      "loss": 1.6402209997177124,
      "step": 89
    },
    {
      "barlow_loss": 0.14392243325710297,
      "epoch": 0.38074866310160427,
      "loss": 1.6254167556762695,
      "step": 89
    },
    {
      "barlow_loss": 0.14862790703773499,
      "epoch": 0.3850267379679144,
      "loss": 1.7518028020858765,
      "step": 90
    },
    {
      "barlow_loss": 0.15608692169189453,
      "epoch": 0.3850267379679144,
      "loss": 1.568796157836914,
      "step": 90
    },
    {
      "barlow_loss": 0.16175740957260132,
      "epoch": 0.3850267379679144,
      "loss": 1.4662153720855713,
      "step": 90
    },
    {
      "barlow_loss": 0.172115758061409,
      "epoch": 0.3850267379679144,
      "loss": 1.5971050262451172,
      "step": 90
    },
    {
      "barlow_loss": 0.1541074514389038,
      "epoch": 0.3893048128342246,
      "loss": 1.538048505783081,
      "step": 91
    },
    {
      "barlow_loss": 0.16185617446899414,
      "epoch": 0.3893048128342246,
      "loss": 1.681467056274414,
      "step": 91
    },
    {
      "barlow_loss": 0.1580023467540741,
      "epoch": 0.3893048128342246,
      "loss": 1.773501992225647,
      "step": 91
    },
    {
      "barlow_loss": 0.1566670835018158,
      "epoch": 0.3893048128342246,
      "loss": 1.5533735752105713,
      "step": 91
    },
    {
      "epoch": 0.39358288770053473,
      "grad_norm": 0.6049765348434448,
      "learning_rate": 2.3568200320665124e-05,
      "loss": 1.659,
      "step": 92
    },
    {
      "barlow_loss": 0.1650390774011612,
      "epoch": 0.39358288770053473,
      "loss": 1.439446210861206,
      "step": 92
    },
    {
      "barlow_loss": 0.17763935029506683,
      "epoch": 0.39358288770053473,
      "loss": 1.6903328895568848,
      "step": 92
    },
    {
      "barlow_loss": 0.16928786039352417,
      "epoch": 0.39358288770053473,
      "loss": 1.71549391746521,
      "step": 92
    },
    {
      "barlow_loss": 0.17508326470851898,
      "epoch": 0.39358288770053473,
      "loss": 1.6649088859558105,
      "step": 92
    },
    {
      "barlow_loss": 0.14421017467975616,
      "epoch": 0.39786096256684494,
      "loss": 1.7588549852371216,
      "step": 93
    },
    {
      "barlow_loss": 0.17496471107006073,
      "epoch": 0.39786096256684494,
      "loss": 1.784633755683899,
      "step": 93
    },
    {
      "barlow_loss": 0.16093887388706207,
      "epoch": 0.39786096256684494,
      "loss": 1.8419842720031738,
      "step": 93
    },
    {
      "barlow_loss": 0.14051063358783722,
      "epoch": 0.39786096256684494,
      "loss": 1.6459461450576782,
      "step": 93
    },
    {
      "barlow_loss": 0.16113802790641785,
      "epoch": 0.4021390374331551,
      "loss": 1.741296410560608,
      "step": 94
    },
    {
      "barlow_loss": 0.1716879904270172,
      "epoch": 0.4021390374331551,
      "loss": 1.602450966835022,
      "step": 94
    },
    {
      "barlow_loss": 0.1548032909631729,
      "epoch": 0.4021390374331551,
      "loss": 1.4480512142181396,
      "step": 94
    },
    {
      "barlow_loss": 0.15407928824424744,
      "epoch": 0.4021390374331551,
      "loss": 1.8744561672210693,
      "step": 94
    },
    {
      "barlow_loss": 0.1611781269311905,
      "epoch": 0.40641711229946526,
      "loss": 1.6407215595245361,
      "step": 95
    },
    {
      "barlow_loss": 0.16404759883880615,
      "epoch": 0.40641711229946526,
      "loss": 1.6807295083999634,
      "step": 95
    },
    {
      "barlow_loss": 0.17476731538772583,
      "epoch": 0.40641711229946526,
      "loss": 1.5689812898635864,
      "step": 95
    },
    {
      "barlow_loss": 0.16736538708209991,
      "epoch": 0.40641711229946526,
      "loss": 1.6799068450927734,
      "step": 95
    },
    {
      "epoch": 0.4106951871657754,
      "grad_norm": 0.6646336913108826,
      "learning_rate": 2.3398594550744292e-05,
      "loss": 1.6736,
      "step": 96
    },
    {
      "barlow_loss": 0.16323865950107574,
      "epoch": 0.4106951871657754,
      "loss": 1.6416916847229004,
      "step": 96
    },
    {
      "barlow_loss": 0.14024928212165833,
      "epoch": 0.4106951871657754,
      "loss": 1.450959324836731,
      "step": 96
    },
    {
      "barlow_loss": 0.17963306605815887,
      "epoch": 0.4106951871657754,
      "loss": 1.5517644882202148,
      "step": 96
    },
    {
      "barlow_loss": 0.17471584677696228,
      "epoch": 0.4106951871657754,
      "loss": 1.4278680086135864,
      "step": 96
    },
    {
      "barlow_loss": 0.17684724926948547,
      "epoch": 0.41497326203208557,
      "loss": 1.5937904119491577,
      "step": 97
    },
    {
      "barlow_loss": 0.16947387158870697,
      "epoch": 0.41497326203208557,
      "loss": 1.409781813621521,
      "step": 97
    },
    {
      "barlow_loss": 0.15488949418067932,
      "epoch": 0.41497326203208557,
      "loss": 1.3995857238769531,
      "step": 97
    },
    {
      "barlow_loss": 0.1586073786020279,
      "epoch": 0.41497326203208557,
      "loss": 1.6534597873687744,
      "step": 97
    },
    {
      "barlow_loss": 0.15865355730056763,
      "epoch": 0.4192513368983957,
      "loss": 1.643000841140747,
      "step": 98
    },
    {
      "barlow_loss": 0.16012407839298248,
      "epoch": 0.4192513368983957,
      "loss": 1.5359362363815308,
      "step": 98
    },
    {
      "barlow_loss": 0.16166305541992188,
      "epoch": 0.4192513368983957,
      "loss": 1.7306519746780396,
      "step": 98
    },
    {
      "barlow_loss": 0.17927353084087372,
      "epoch": 0.4192513368983957,
      "loss": 1.7109830379486084,
      "step": 98
    },
    {
      "barlow_loss": 0.1628662794828415,
      "epoch": 0.4235294117647059,
      "loss": 1.9153053760528564,
      "step": 99
    },
    {
      "barlow_loss": 0.16493569314479828,
      "epoch": 0.4235294117647059,
      "loss": 1.4035500288009644,
      "step": 99
    },
    {
      "barlow_loss": 0.1571175456047058,
      "epoch": 0.4235294117647059,
      "loss": 1.640413522720337,
      "step": 99
    },
    {
      "barlow_loss": 0.1678721308708191,
      "epoch": 0.4235294117647059,
      "loss": 1.9162118434906006,
      "step": 99
    },
    {
      "epoch": 0.42780748663101603,
      "grad_norm": 0.7058869004249573,
      "learning_rate": 2.3220179987268153e-05,
      "loss": 1.6016,
      "step": 100
    },
    {
      "barlow_loss": 0.15645745396614075,
      "epoch": 0.42780748663101603,
      "loss": 1.6595842838287354,
      "step": 100
    },
    {
      "barlow_loss": 0.1721504032611847,
      "epoch": 0.42780748663101603,
      "loss": 1.666886806488037,
      "step": 100
    },
    {
      "barlow_loss": 0.1643727868795395,
      "epoch": 0.42780748663101603,
      "loss": 1.4570834636688232,
      "step": 100
    },
    {
      "barlow_loss": 0.1600043624639511,
      "epoch": 0.42780748663101603,
      "loss": 1.4666709899902344,
      "step": 100
    },
    {
      "barlow_loss": 0.16735738515853882,
      "epoch": 0.4320855614973262,
      "loss": 1.4435211420059204,
      "step": 101
    },
    {
      "barlow_loss": 0.16837336122989655,
      "epoch": 0.4320855614973262,
      "loss": 1.6272304058074951,
      "step": 101
    },
    {
      "barlow_loss": 0.15639623999595642,
      "epoch": 0.4320855614973262,
      "loss": 1.555080533027649,
      "step": 101
    },
    {
      "barlow_loss": 0.14727447926998138,
      "epoch": 0.4320855614973262,
      "loss": 1.6708219051361084,
      "step": 101
    },
    {
      "barlow_loss": 0.16466355323791504,
      "epoch": 0.43636363636363634,
      "loss": 1.871424913406372,
      "step": 102
    },
    {
      "barlow_loss": 0.1489398181438446,
      "epoch": 0.43636363636363634,
      "loss": 1.570081114768982,
      "step": 102
    },
    {
      "barlow_loss": 0.16587640345096588,
      "epoch": 0.43636363636363634,
      "loss": 1.436240792274475,
      "step": 102
    },
    {
      "barlow_loss": 0.16537460684776306,
      "epoch": 0.43636363636363634,
      "loss": 1.8211233615875244,
      "step": 102
    },
    {
      "barlow_loss": 0.153898686170578,
      "epoch": 0.4406417112299465,
      "loss": 1.6616439819335938,
      "step": 103
    },
    {
      "barlow_loss": 0.17819693684577942,
      "epoch": 0.4406417112299465,
      "loss": 1.4671001434326172,
      "step": 103
    },
    {
      "barlow_loss": 0.17207303643226624,
      "epoch": 0.4406417112299465,
      "loss": 1.8522084951400757,
      "step": 103
    },
    {
      "barlow_loss": 0.1748334765434265,
      "epoch": 0.4406417112299465,
      "loss": 1.4609217643737793,
      "step": 103
    },
    {
      "epoch": 0.4449197860962567,
      "grad_norm": 0.6507449150085449,
      "learning_rate": 2.303310083388147e-05,
      "loss": 1.6055,
      "step": 104
    },
    {
      "barlow_loss": 0.1770409643650055,
      "epoch": 0.4449197860962567,
      "loss": 1.689785361289978,
      "step": 104
    },
    {
      "barlow_loss": 0.19272619485855103,
      "epoch": 0.4449197860962567,
      "loss": 1.5958216190338135,
      "step": 104
    },
    {
      "barlow_loss": 0.15616288781166077,
      "epoch": 0.4449197860962567,
      "loss": 1.6571943759918213,
      "step": 104
    },
    {
      "barlow_loss": 0.14411109685897827,
      "epoch": 0.4449197860962567,
      "loss": 1.4788886308670044,
      "step": 104
    },
    {
      "barlow_loss": 0.1726577877998352,
      "epoch": 0.44919786096256686,
      "loss": 1.6511458158493042,
      "step": 105
    },
    {
      "barlow_loss": 0.16541913151741028,
      "epoch": 0.44919786096256686,
      "loss": 1.8400694131851196,
      "step": 105
    },
    {
      "barlow_loss": 0.17455711960792542,
      "epoch": 0.44919786096256686,
      "loss": 1.5601246356964111,
      "step": 105
    },
    {
      "barlow_loss": 0.16302289068698883,
      "epoch": 0.44919786096256686,
      "loss": 1.57891047000885,
      "step": 105
    },
    {
      "barlow_loss": 0.159755676984787,
      "epoch": 0.453475935828877,
      "loss": 1.5049322843551636,
      "step": 106
    },
    {
      "barlow_loss": 0.16570670902729034,
      "epoch": 0.453475935828877,
      "loss": 1.6580688953399658,
      "step": 106
    },
    {
      "barlow_loss": 0.17059820890426636,
      "epoch": 0.453475935828877,
      "loss": 1.4984554052352905,
      "step": 106
    },
    {
      "barlow_loss": 0.17171388864517212,
      "epoch": 0.453475935828877,
      "loss": 1.804961919784546,
      "step": 106
    },
    {
      "barlow_loss": 0.15877437591552734,
      "epoch": 0.4577540106951872,
      "loss": 1.367188811302185,
      "step": 107
    },
    {
      "barlow_loss": 0.16314637660980225,
      "epoch": 0.4577540106951872,
      "loss": 1.563667893409729,
      "step": 107
    },
    {
      "barlow_loss": 0.1780811995267868,
      "epoch": 0.4577540106951872,
      "loss": 1.6045358180999756,
      "step": 107
    },
    {
      "barlow_loss": 0.16822071373462677,
      "epoch": 0.4577540106951872,
      "loss": 1.5702869892120361,
      "step": 107
    },
    {
      "epoch": 0.46203208556149733,
      "grad_norm": 0.6089738607406616,
      "learning_rate": 2.2837508297387384e-05,
      "loss": 1.6015,
      "step": 108
    },
    {
      "barlow_loss": 0.1717865765094757,
      "epoch": 0.46203208556149733,
      "loss": 1.5617985725402832,
      "step": 108
    },
    {
      "barlow_loss": 0.15536054968833923,
      "epoch": 0.46203208556149733,
      "loss": 1.602267861366272,
      "step": 108
    },
    {
      "barlow_loss": 0.17703953385353088,
      "epoch": 0.46203208556149733,
      "loss": 1.930182933807373,
      "step": 108
    },
    {
      "barlow_loss": 0.16947846114635468,
      "epoch": 0.46203208556149733,
      "loss": 1.6522846221923828,
      "step": 108
    },
    {
      "barlow_loss": 0.1620883047580719,
      "epoch": 0.4663101604278075,
      "loss": 1.6608963012695312,
      "step": 109
    },
    {
      "barlow_loss": 0.17615433037281036,
      "epoch": 0.4663101604278075,
      "loss": 1.736676812171936,
      "step": 109
    },
    {
      "barlow_loss": 0.16644740104675293,
      "epoch": 0.4663101604278075,
      "loss": 1.4646540880203247,
      "step": 109
    },
    {
      "barlow_loss": 0.1554500013589859,
      "epoch": 0.4663101604278075,
      "loss": 1.4877732992172241,
      "step": 109
    },
    {
      "barlow_loss": 0.14925993978977203,
      "epoch": 0.47058823529411764,
      "loss": 1.4760220050811768,
      "step": 110
    },
    {
      "barlow_loss": 0.14630642533302307,
      "epoch": 0.47058823529411764,
      "loss": 1.4942257404327393,
      "step": 110
    },
    {
      "barlow_loss": 0.16735586524009705,
      "epoch": 0.47058823529411764,
      "loss": 1.473475694656372,
      "step": 110
    },
    {
      "barlow_loss": 0.15473830699920654,
      "epoch": 0.47058823529411764,
      "loss": 1.6148402690887451,
      "step": 110
    },
    {
      "barlow_loss": 0.16978006064891815,
      "epoch": 0.4748663101604278,
      "loss": 1.6411669254302979,
      "step": 111
    },
    {
      "barlow_loss": 0.1636914610862732,
      "epoch": 0.4748663101604278,
      "loss": 1.3257462978363037,
      "step": 111
    },
    {
      "barlow_loss": 0.1566513180732727,
      "epoch": 0.4748663101604278,
      "loss": 1.6269562244415283,
      "step": 111
    },
    {
      "barlow_loss": 0.15606990456581116,
      "epoch": 0.4748663101604278,
      "loss": 1.4046180248260498,
      "step": 111
    },
    {
      "epoch": 0.47914438502673795,
      "grad_norm": 0.6345927715301514,
      "learning_rate": 2.2633560465534438e-05,
      "loss": 1.5721,
      "step": 112
    },
    {
      "barlow_loss": 0.17722520232200623,
      "epoch": 0.47914438502673795,
      "loss": 1.490165114402771,
      "step": 112
    },
    {
      "barlow_loss": 0.18189507722854614,
      "epoch": 0.47914438502673795,
      "loss": 1.5165762901306152,
      "step": 112
    },
    {
      "barlow_loss": 0.16500739753246307,
      "epoch": 0.47914438502673795,
      "loss": 1.6100355386734009,
      "step": 112
    },
    {
      "barlow_loss": 0.15079323947429657,
      "epoch": 0.47914438502673795,
      "loss": 1.4463458061218262,
      "step": 112
    },
    {
      "barlow_loss": 0.16120579838752747,
      "epoch": 0.4834224598930481,
      "loss": 1.531149983406067,
      "step": 113
    },
    {
      "barlow_loss": 0.160549595952034,
      "epoch": 0.4834224598930481,
      "loss": 1.5562065839767456,
      "step": 113
    },
    {
      "barlow_loss": 0.1654520481824875,
      "epoch": 0.4834224598930481,
      "loss": 1.9496368169784546,
      "step": 113
    },
    {
      "barlow_loss": 0.16209936141967773,
      "epoch": 0.4834224598930481,
      "loss": 1.6067856550216675,
      "step": 113
    },
    {
      "barlow_loss": 0.15959830582141876,
      "epoch": 0.48770053475935826,
      "loss": 1.5856454372406006,
      "step": 114
    },
    {
      "barlow_loss": 0.16712737083435059,
      "epoch": 0.48770053475935826,
      "loss": 1.6038488149642944,
      "step": 114
    },
    {
      "barlow_loss": 0.1712852269411087,
      "epoch": 0.48770053475935826,
      "loss": 1.4043760299682617,
      "step": 114
    },
    {
      "barlow_loss": 0.1708967536687851,
      "epoch": 0.48770053475935826,
      "loss": 1.5916452407836914,
      "step": 114
    },
    {
      "barlow_loss": 0.1556926667690277,
      "epoch": 0.4919786096256685,
      "loss": 1.5938349962234497,
      "step": 115
    },
    {
      "barlow_loss": 0.17129963636398315,
      "epoch": 0.4919786096256685,
      "loss": 1.4034948348999023,
      "step": 115
    },
    {
      "barlow_loss": 0.1615298092365265,
      "epoch": 0.4919786096256685,
      "loss": 1.4772542715072632,
      "step": 115
    },
    {
      "barlow_loss": 0.18384672701358795,
      "epoch": 0.4919786096256685,
      "loss": 1.718433141708374,
      "step": 115
    },
    {
      "epoch": 0.49625668449197863,
      "grad_norm": 0.6031079888343811,
      "learning_rate": 2.2421422179242118e-05,
      "loss": 1.5678,
      "step": 116
    },
    {
      "barlow_loss": 0.17677146196365356,
      "epoch": 0.49625668449197863,
      "loss": 1.5698268413543701,
      "step": 116
    },
    {
      "barlow_loss": 0.17374162375926971,
      "epoch": 0.49625668449197863,
      "loss": 1.7295057773590088,
      "step": 116
    },
    {
      "barlow_loss": 0.18084610998630524,
      "epoch": 0.49625668449197863,
      "loss": 1.7078709602355957,
      "step": 116
    },
    {
      "barlow_loss": 0.17015832662582397,
      "epoch": 0.49625668449197863,
      "loss": 1.631987452507019,
      "step": 116
    },
    {
      "barlow_loss": 0.17356643080711365,
      "epoch": 0.5005347593582887,
      "loss": 1.5124465227127075,
      "step": 117
    },
    {
      "barlow_loss": 0.17417724430561066,
      "epoch": 0.5005347593582887,
      "loss": 1.6042885780334473,
      "step": 117
    },
    {
      "barlow_loss": 0.15851779282093048,
      "epoch": 0.5005347593582887,
      "loss": 1.5551366806030273,
      "step": 117
    },
    {
      "barlow_loss": 0.17982886731624603,
      "epoch": 0.5005347593582887,
      "loss": 1.5242774486541748,
      "step": 117
    },
    {
      "barlow_loss": 0.17722490429878235,
      "epoch": 0.5048128342245989,
      "loss": 1.6771560907363892,
      "step": 118
    },
    {
      "barlow_loss": 0.17501097917556763,
      "epoch": 0.5048128342245989,
      "loss": 1.782839298248291,
      "step": 118
    },
    {
      "barlow_loss": 0.15827623009681702,
      "epoch": 0.5048128342245989,
      "loss": 1.5677423477172852,
      "step": 118
    },
    {
      "barlow_loss": 0.167347252368927,
      "epoch": 0.5048128342245989,
      "loss": 1.6109164953231812,
      "step": 118
    },
    {
      "barlow_loss": 0.15949702262878418,
      "epoch": 0.509090909090909,
      "loss": 1.482805609703064,
      "step": 119
    },
    {
      "barlow_loss": 0.16967907547950745,
      "epoch": 0.509090909090909,
      "loss": 1.6540474891662598,
      "step": 119
    },
    {
      "barlow_loss": 0.14680543541908264,
      "epoch": 0.509090909090909,
      "loss": 1.6152913570404053,
      "step": 119
    },
    {
      "barlow_loss": 0.15678881108760834,
      "epoch": 0.509090909090909,
      "loss": 1.541663646697998,
      "step": 119
    },
    {
      "epoch": 0.5133689839572193,
      "grad_norm": 0.6493864059448242,
      "learning_rate": 2.220126489936811e-05,
      "loss": 1.6105,
      "step": 120
    },
    {
      "barlow_loss": 0.154567152261734,
      "epoch": 0.5133689839572193,
      "loss": 1.4139970541000366,
      "step": 120
    },
    {
      "barlow_loss": 0.17079229652881622,
      "epoch": 0.5133689839572193,
      "loss": 1.667179822921753,
      "step": 120
    },
    {
      "barlow_loss": 0.17020989954471588,
      "epoch": 0.5133689839572193,
      "loss": 1.3575206995010376,
      "step": 120
    },
    {
      "barlow_loss": 0.17584270238876343,
      "epoch": 0.5133689839572193,
      "loss": 1.764635443687439,
      "step": 120
    },
    {
      "barlow_loss": 0.13982662558555603,
      "epoch": 0.5176470588235295,
      "loss": 1.4368503093719482,
      "step": 121
    },
    {
      "barlow_loss": 0.16414809226989746,
      "epoch": 0.5176470588235295,
      "loss": 1.5907645225524902,
      "step": 121
    },
    {
      "barlow_loss": 0.16258057951927185,
      "epoch": 0.5176470588235295,
      "loss": 1.7207870483398438,
      "step": 121
    },
    {
      "barlow_loss": 0.1739169955253601,
      "epoch": 0.5176470588235295,
      "loss": 1.6684023141860962,
      "step": 121
    },
    {
      "barlow_loss": 0.15149284899234772,
      "epoch": 0.5219251336898396,
      "loss": 1.6013332605361938,
      "step": 122
    },
    {
      "barlow_loss": 0.161963552236557,
      "epoch": 0.5219251336898396,
      "loss": 1.5163909196853638,
      "step": 122
    },
    {
      "barlow_loss": 0.15965156257152557,
      "epoch": 0.5219251336898396,
      "loss": 1.5123286247253418,
      "step": 122
    },
    {
      "barlow_loss": 0.1463104635477066,
      "epoch": 0.5219251336898396,
      "loss": 1.6864314079284668,
      "step": 122
    },
    {
      "barlow_loss": 0.16780449450016022,
      "epoch": 0.5262032085561498,
      "loss": 1.707782506942749,
      "step": 123
    },
    {
      "barlow_loss": 0.16657595336437225,
      "epoch": 0.5262032085561498,
      "loss": 1.6678200960159302,
      "step": 123
    },
    {
      "barlow_loss": 0.15554828941822052,
      "epoch": 0.5262032085561498,
      "loss": 1.4612224102020264,
      "step": 123
    },
    {
      "barlow_loss": 0.1663541942834854,
      "epoch": 0.5262032085561498,
      "loss": 1.5541656017303467,
      "step": 123
    },
    {
      "epoch": 0.5304812834224599,
      "grad_norm": 0.7045292258262634,
      "learning_rate": 2.1973266568124963e-05,
      "loss": 1.583,
      "step": 124
    },
    {
      "barlow_loss": 0.17585265636444092,
      "epoch": 0.5304812834224599,
      "loss": 1.573535680770874,
      "step": 124
    },
    {
      "barlow_loss": 0.15862546861171722,
      "epoch": 0.5304812834224599,
      "loss": 1.7758550643920898,
      "step": 124
    },
    {
      "barlow_loss": 0.16091616451740265,
      "epoch": 0.5304812834224599,
      "loss": 1.651956558227539,
      "step": 124
    },
    {
      "barlow_loss": 0.1659775674343109,
      "epoch": 0.5304812834224599,
      "loss": 1.4608937501907349,
      "step": 124
    },
    {
      "barlow_loss": 0.1682448387145996,
      "epoch": 0.5347593582887701,
      "loss": 1.62762451171875,
      "step": 125
    },
    {
      "barlow_loss": 0.15835168957710266,
      "epoch": 0.5347593582887701,
      "loss": 1.5489037036895752,
      "step": 125
    },
    {
      "barlow_loss": 0.1832101047039032,
      "epoch": 0.5347593582887701,
      "loss": 1.5818963050842285,
      "step": 125
    },
    {
      "barlow_loss": 0.17364291846752167,
      "epoch": 0.5347593582887701,
      "loss": 1.4044431447982788,
      "step": 125
    },
    {
      "barlow_loss": 0.16714511811733246,
      "epoch": 0.5390374331550802,
      "loss": 1.4203650951385498,
      "step": 126
    },
    {
      "barlow_loss": 0.1694854348897934,
      "epoch": 0.5390374331550802,
      "loss": 1.4802876710891724,
      "step": 126
    },
    {
      "barlow_loss": 0.16341206431388855,
      "epoch": 0.5390374331550802,
      "loss": 1.6686944961547852,
      "step": 126
    },
    {
      "barlow_loss": 0.17555324733257294,
      "epoch": 0.5390374331550802,
      "loss": 1.9986225366592407,
      "step": 126
    },
    {
      "barlow_loss": 0.1657291054725647,
      "epoch": 0.5433155080213904,
      "loss": 1.657192349433899,
      "step": 127
    },
    {
      "barlow_loss": 0.14153806865215302,
      "epoch": 0.5433155080213904,
      "loss": 1.469849944114685,
      "step": 127
    },
    {
      "barlow_loss": 0.15508483350276947,
      "epoch": 0.5433155080213904,
      "loss": 1.5377893447875977,
      "step": 127
    },
    {
      "barlow_loss": 0.16026854515075684,
      "epoch": 0.5433155080213904,
      "loss": 1.629863977432251,
      "step": 127
    },
    {
      "epoch": 0.5475935828877005,
      "grad_norm": 0.6963449120521545,
      "learning_rate": 2.173761146525824e-05,
      "loss": 1.593,
      "step": 128
    },
    {
      "barlow_loss": 0.16451025009155273,
      "epoch": 0.5475935828877005,
      "loss": 1.2096487283706665,
      "step": 128
    },
    {
      "barlow_loss": 0.15896734595298767,
      "epoch": 0.5475935828877005,
      "loss": 1.5354342460632324,
      "step": 128
    },
    {
      "barlow_loss": 0.17262008786201477,
      "epoch": 0.5475935828877005,
      "loss": 1.602356195449829,
      "step": 128
    },
    {
      "barlow_loss": 0.1606743335723877,
      "epoch": 0.5475935828877005,
      "loss": 1.6118289232254028,
      "step": 128
    },
    {
      "barlow_loss": 0.17121726274490356,
      "epoch": 0.5518716577540107,
      "loss": 1.7432317733764648,
      "step": 129
    },
    {
      "barlow_loss": 0.15160314738750458,
      "epoch": 0.5518716577540107,
      "loss": 1.5480347871780396,
      "step": 129
    },
    {
      "barlow_loss": 0.1452227532863617,
      "epoch": 0.5518716577540107,
      "loss": 1.4544330835342407,
      "step": 129
    },
    {
      "barlow_loss": 0.16471752524375916,
      "epoch": 0.5518716577540107,
      "loss": 1.8396855592727661,
      "step": 129
    },
    {
      "barlow_loss": 0.1637190878391266,
      "epoch": 0.5561497326203209,
      "loss": 1.5648458003997803,
      "step": 130
    },
    {
      "barlow_loss": 0.15458542108535767,
      "epoch": 0.5561497326203209,
      "loss": 1.4222259521484375,
      "step": 130
    },
    {
      "barlow_loss": 0.16443203389644623,
      "epoch": 0.5561497326203209,
      "loss": 1.5168914794921875,
      "step": 130
    },
    {
      "barlow_loss": 0.16589748859405518,
      "epoch": 0.5561497326203209,
      "loss": 1.5166815519332886,
      "step": 130
    },
    {
      "barlow_loss": 0.15999488532543182,
      "epoch": 0.560427807486631,
      "loss": 1.6749564409255981,
      "step": 131
    },
    {
      "barlow_loss": 0.15844281017780304,
      "epoch": 0.560427807486631,
      "loss": 1.4414341449737549,
      "step": 131
    },
    {
      "barlow_loss": 0.15555426478385925,
      "epoch": 0.560427807486631,
      "loss": 1.4897058010101318,
      "step": 131
    },
    {
      "barlow_loss": 0.18514740467071533,
      "epoch": 0.560427807486631,
      "loss": 1.7083170413970947,
      "step": 131
    },
    {
      "epoch": 0.5647058823529412,
      "grad_norm": 0.7076470851898193,
      "learning_rate": 2.1494490059102352e-05,
      "loss": 1.555,
      "step": 132
    },
    {
      "barlow_loss": 0.16332432627677917,
      "epoch": 0.5647058823529412,
      "loss": 1.2521008253097534,
      "step": 132
    },
    {
      "barlow_loss": 0.1671813428401947,
      "epoch": 0.5647058823529412,
      "loss": 1.4990627765655518,
      "step": 132
    },
    {
      "barlow_loss": 0.15929155051708221,
      "epoch": 0.5647058823529412,
      "loss": 1.521970272064209,
      "step": 132
    },
    {
      "barlow_loss": 0.17039372026920319,
      "epoch": 0.5647058823529412,
      "loss": 1.5365374088287354,
      "step": 132
    },
    {
      "barlow_loss": 0.15981367230415344,
      "epoch": 0.5689839572192513,
      "loss": 1.5038930177688599,
      "step": 133
    },
    {
      "barlow_loss": 0.1625518649816513,
      "epoch": 0.5689839572192513,
      "loss": 1.6252899169921875,
      "step": 133
    },
    {
      "barlow_loss": 0.16938789188861847,
      "epoch": 0.5689839572192513,
      "loss": 1.532901406288147,
      "step": 133
    },
    {
      "barlow_loss": 0.1530674695968628,
      "epoch": 0.5689839572192513,
      "loss": 1.5873006582260132,
      "step": 133
    },
    {
      "barlow_loss": 0.15171991288661957,
      "epoch": 0.5732620320855615,
      "loss": 1.4961007833480835,
      "step": 134
    },
    {
      "barlow_loss": 0.16806112229824066,
      "epoch": 0.5732620320855615,
      "loss": 1.456265926361084,
      "step": 134
    },
    {
      "barlow_loss": 0.16986165940761566,
      "epoch": 0.5732620320855615,
      "loss": 1.7237768173217773,
      "step": 134
    },
    {
      "barlow_loss": 0.151210218667984,
      "epoch": 0.5732620320855615,
      "loss": 1.4882293939590454,
      "step": 134
    },
    {
      "barlow_loss": 0.15857717394828796,
      "epoch": 0.5775401069518716,
      "loss": 1.6005675792694092,
      "step": 135
    },
    {
      "barlow_loss": 0.149105966091156,
      "epoch": 0.5775401069518716,
      "loss": 1.672706127166748,
      "step": 135
    },
    {
      "barlow_loss": 0.16746555268764496,
      "epoch": 0.5775401069518716,
      "loss": 1.5505762100219727,
      "step": 135
    },
    {
      "barlow_loss": 0.1639406383037567,
      "epoch": 0.5775401069518716,
      "loss": 1.6427228450775146,
      "step": 135
    },
    {
      "epoch": 0.5818181818181818,
      "grad_norm": 0.7805615663528442,
      "learning_rate": 2.1244098852634424e-05,
      "loss": 1.5431,
      "step": 136
    },
    {
      "barlow_loss": 0.16385430097579956,
      "epoch": 0.5818181818181818,
      "loss": 1.3573002815246582,
      "step": 136
    },
    {
      "barlow_loss": 0.16987046599388123,
      "epoch": 0.5818181818181818,
      "loss": 1.457895040512085,
      "step": 136
    },
    {
      "barlow_loss": 0.17057761549949646,
      "epoch": 0.5818181818181818,
      "loss": 1.5813801288604736,
      "step": 136
    },
    {
      "barlow_loss": 0.16819988191127777,
      "epoch": 0.5818181818181818,
      "loss": 1.4895635843276978,
      "step": 136
    },
    {
      "barlow_loss": 0.15750476717948914,
      "epoch": 0.586096256684492,
      "loss": 1.430631399154663,
      "step": 137
    },
    {
      "barlow_loss": 0.17432759702205658,
      "epoch": 0.586096256684492,
      "loss": 1.3911088705062866,
      "step": 137
    },
    {
      "barlow_loss": 0.1717047393321991,
      "epoch": 0.586096256684492,
      "loss": 1.515979528427124,
      "step": 137
    },
    {
      "barlow_loss": 0.160686194896698,
      "epoch": 0.586096256684492,
      "loss": 1.5253474712371826,
      "step": 137
    },
    {
      "barlow_loss": 0.16903749108314514,
      "epoch": 0.5903743315508021,
      "loss": 1.8020987510681152,
      "step": 138
    },
    {
      "barlow_loss": 0.14902658760547638,
      "epoch": 0.5903743315508021,
      "loss": 1.5265388488769531,
      "step": 138
    },
    {
      "barlow_loss": 0.16472575068473816,
      "epoch": 0.5903743315508021,
      "loss": 1.5601500272750854,
      "step": 138
    },
    {
      "barlow_loss": 0.1709531843662262,
      "epoch": 0.5903743315508021,
      "loss": 1.5836323499679565,
      "step": 138
    },
    {
      "barlow_loss": 0.15983743965625763,
      "epoch": 0.5946524064171123,
      "loss": 1.4345282316207886,
      "step": 139
    },
    {
      "barlow_loss": 0.1815115511417389,
      "epoch": 0.5946524064171123,
      "loss": 1.6147379875183105,
      "step": 139
    },
    {
      "barlow_loss": 0.1619776487350464,
      "epoch": 0.5946524064171123,
      "loss": 1.6552228927612305,
      "step": 139
    },
    {
      "barlow_loss": 0.17436841130256653,
      "epoch": 0.5946524064171123,
      "loss": 1.5345964431762695,
      "step": 139
    },
    {
      "epoch": 0.5989304812834224,
      "grad_norm": 0.7162126302719116,
      "learning_rate": 2.0986640224650684e-05,
      "loss": 1.5288,
      "step": 140
    },
    {
      "barlow_loss": 0.14823848009109497,
      "epoch": 0.5989304812834224,
      "loss": 1.545431137084961,
      "step": 140
    },
    {
      "barlow_loss": 0.1657588928937912,
      "epoch": 0.5989304812834224,
      "loss": 1.694096326828003,
      "step": 140
    },
    {
      "barlow_loss": 0.16199660301208496,
      "epoch": 0.5989304812834224,
      "loss": 1.7005828619003296,
      "step": 140
    },
    {
      "barlow_loss": 0.17242993414402008,
      "epoch": 0.5989304812834224,
      "loss": 1.6524399518966675,
      "step": 140
    },
    {
      "barlow_loss": 0.16285046935081482,
      "epoch": 0.6032085561497326,
      "loss": 1.7262600660324097,
      "step": 141
    },
    {
      "barlow_loss": 0.14329443871974945,
      "epoch": 0.6032085561497326,
      "loss": 1.4400790929794312,
      "step": 141
    },
    {
      "barlow_loss": 0.17116527259349823,
      "epoch": 0.6032085561497326,
      "loss": 1.736707091331482,
      "step": 141
    },
    {
      "barlow_loss": 0.15935495495796204,
      "epoch": 0.6032085561497326,
      "loss": 1.523126482963562,
      "step": 141
    },
    {
      "barlow_loss": 0.13971176743507385,
      "epoch": 0.6074866310160428,
      "loss": 1.5044218301773071,
      "step": 142
    },
    {
      "barlow_loss": 0.1625044047832489,
      "epoch": 0.6074866310160428,
      "loss": 1.4841312170028687,
      "step": 142
    },
    {
      "barlow_loss": 0.15166091918945312,
      "epoch": 0.6074866310160428,
      "loss": 1.5087652206420898,
      "step": 142
    },
    {
      "barlow_loss": 0.16370102763175964,
      "epoch": 0.6074866310160428,
      "loss": 1.6240613460540771,
      "step": 142
    },
    {
      "barlow_loss": 0.16149574518203735,
      "epoch": 0.611764705882353,
      "loss": 1.7302647829055786,
      "step": 143
    },
    {
      "barlow_loss": 0.1528588980436325,
      "epoch": 0.611764705882353,
      "loss": 1.5703563690185547,
      "step": 143
    },
    {
      "barlow_loss": 0.17448808252811432,
      "epoch": 0.611764705882353,
      "loss": 1.736674189567566,
      "step": 143
    },
    {
      "barlow_loss": 0.15873731672763824,
      "epoch": 0.611764705882353,
      "loss": 1.6704381704330444,
      "step": 143
    },
    {
      "epoch": 0.6160427807486631,
      "grad_norm": 0.7825184464454651,
      "learning_rate": 2.072232226619374e-05,
      "loss": 1.6155,
      "step": 144
    },
    {
      "barlow_loss": 0.17037351429462433,
      "epoch": 0.6160427807486631,
      "loss": 1.5969620943069458,
      "step": 144
    },
    {
      "barlow_loss": 0.15725064277648926,
      "epoch": 0.6160427807486631,
      "loss": 1.4941178560256958,
      "step": 144
    },
    {
      "barlow_loss": 0.15297818183898926,
      "epoch": 0.6160427807486631,
      "loss": 1.4334269762039185,
      "step": 144
    },
    {
      "barlow_loss": 0.17377227544784546,
      "epoch": 0.6160427807486631,
      "loss": 1.5652310848236084,
      "step": 144
    },
    {
      "barlow_loss": 0.16433705389499664,
      "epoch": 0.6203208556149733,
      "loss": 1.4267733097076416,
      "step": 145
    },
    {
      "barlow_loss": 0.1642569601535797,
      "epoch": 0.6203208556149733,
      "loss": 1.700101613998413,
      "step": 145
    },
    {
      "barlow_loss": 0.16193734109401703,
      "epoch": 0.6203208556149733,
      "loss": 1.377098798751831,
      "step": 145
    },
    {
      "barlow_loss": 0.1500864326953888,
      "epoch": 0.6203208556149733,
      "loss": 1.4876834154129028,
      "step": 145
    },
    {
      "barlow_loss": 0.17250150442123413,
      "epoch": 0.6245989304812835,
      "loss": 1.2556686401367188,
      "step": 146
    },
    {
      "barlow_loss": 0.15404151380062103,
      "epoch": 0.6245989304812835,
      "loss": 1.4247040748596191,
      "step": 146
    },
    {
      "barlow_loss": 0.15675371885299683,
      "epoch": 0.6245989304812835,
      "loss": 1.5805095434188843,
      "step": 146
    },
    {
      "barlow_loss": 0.16165666282176971,
      "epoch": 0.6245989304812835,
      "loss": 1.4926508665084839,
      "step": 146
    },
    {
      "barlow_loss": 0.16711080074310303,
      "epoch": 0.6288770053475936,
      "loss": 1.3717273473739624,
      "step": 147
    },
    {
      "barlow_loss": 0.17002272605895996,
      "epoch": 0.6288770053475936,
      "loss": 1.738415241241455,
      "step": 147
    },
    {
      "barlow_loss": 0.17825134098529816,
      "epoch": 0.6288770053475936,
      "loss": 1.404173731803894,
      "step": 147
    },
    {
      "barlow_loss": 0.15062174201011658,
      "epoch": 0.6288770053475936,
      "loss": 1.6008456945419312,
      "step": 147
    },
    {
      "epoch": 0.6331550802139038,
      "grad_norm": 0.6998313069343567,
      "learning_rate": 2.0451358612362883e-05,
      "loss": 1.4969,
      "step": 148
    },
    {
      "barlow_loss": 0.15929673612117767,
      "epoch": 0.6331550802139038,
      "loss": 1.70541512966156,
      "step": 148
    },
    {
      "barlow_loss": 0.15342853963375092,
      "epoch": 0.6331550802139038,
      "loss": 1.3996219635009766,
      "step": 148
    },
    {
      "barlow_loss": 0.17143186926841736,
      "epoch": 0.6331550802139038,
      "loss": 1.6596885919570923,
      "step": 148
    },
    {
      "barlow_loss": 0.162881001830101,
      "epoch": 0.6331550802139038,
      "loss": 1.6748631000518799,
      "step": 148
    },
    {
      "barlow_loss": 0.17072774469852448,
      "epoch": 0.6374331550802139,
      "loss": 1.3188661336898804,
      "step": 149
    },
    {
      "barlow_loss": 0.1778217852115631,
      "epoch": 0.6374331550802139,
      "loss": 1.5770962238311768,
      "step": 149
    },
    {
      "barlow_loss": 0.16184189915657043,
      "epoch": 0.6374331550802139,
      "loss": 1.5070860385894775,
      "step": 149
    },
    {
      "barlow_loss": 0.14924001693725586,
      "epoch": 0.6374331550802139,
      "loss": 1.68938410282135,
      "step": 149
    },
    {
      "barlow_loss": 0.15570850670337677,
      "epoch": 0.6417112299465241,
      "loss": 1.5982747077941895,
      "step": 150
    },
    {
      "barlow_loss": 0.16932153701782227,
      "epoch": 0.6417112299465241,
      "loss": 1.5545823574066162,
      "step": 150
    },
    {
      "barlow_loss": 0.15747059881687164,
      "epoch": 0.6417112299465241,
      "loss": 1.4777922630310059,
      "step": 150
    },
    {
      "barlow_loss": 0.17064344882965088,
      "epoch": 0.6417112299465241,
      "loss": 1.5250838994979858,
      "step": 150
    },
    {
      "barlow_loss": 0.16563023626804352,
      "epoch": 0.6459893048128342,
      "loss": 1.5494065284729004,
      "step": 151
    },
    {
      "barlow_loss": 0.17262566089630127,
      "epoch": 0.6459893048128342,
      "loss": 1.504891276359558,
      "step": 151
    },
    {
      "barlow_loss": 0.16540934145450592,
      "epoch": 0.6459893048128342,
      "loss": 1.7189024686813354,
      "step": 151
    },
    {
      "barlow_loss": 0.15931236743927002,
      "epoch": 0.6459893048128342,
      "loss": 1.396087408065796,
      "step": 151
    },
    {
      "epoch": 0.6502673796791444,
      "grad_norm": 0.7364256381988525,
      "learning_rate": 2.0173968269643438e-05,
      "loss": 1.5536,
      "step": 152
    },
    {
      "barlow_loss": 0.14984233677387238,
      "epoch": 0.6502673796791444,
      "loss": 1.590454339981079,
      "step": 152
    },
    {
      "barlow_loss": 0.1776231825351715,
      "epoch": 0.6502673796791444,
      "loss": 1.6354070901870728,
      "step": 152
    },
    {
      "barlow_loss": 0.16068781912326813,
      "epoch": 0.6502673796791444,
      "loss": 1.6709917783737183,
      "step": 152
    },
    {
      "barlow_loss": 0.1683586984872818,
      "epoch": 0.6502673796791444,
      "loss": 1.441332221031189,
      "step": 152
    },
    {
      "barlow_loss": 0.1651637852191925,
      "epoch": 0.6545454545454545,
      "loss": 1.4240593910217285,
      "step": 153
    },
    {
      "barlow_loss": 0.16271115839481354,
      "epoch": 0.6545454545454545,
      "loss": 1.5369417667388916,
      "step": 153
    },
    {
      "barlow_loss": 0.16494370996952057,
      "epoch": 0.6545454545454545,
      "loss": 1.4803680181503296,
      "step": 153
    },
    {
      "barlow_loss": 0.1810602843761444,
      "epoch": 0.6545454545454545,
      "loss": 1.772995948791504,
      "step": 153
    },
    {
      "barlow_loss": 0.16994865238666534,
      "epoch": 0.6588235294117647,
      "loss": 1.377069115638733,
      "step": 154
    },
    {
      "barlow_loss": 0.15653131902217865,
      "epoch": 0.6588235294117647,
      "loss": 1.616522192955017,
      "step": 154
    },
    {
      "barlow_loss": 0.1783367246389389,
      "epoch": 0.6588235294117647,
      "loss": 1.5746066570281982,
      "step": 154
    },
    {
      "barlow_loss": 0.1546972393989563,
      "epoch": 0.6588235294117647,
      "loss": 1.4817487001419067,
      "step": 154
    },
    {
      "barlow_loss": 0.16523511707782745,
      "epoch": 0.6631016042780749,
      "loss": 1.2921663522720337,
      "step": 155
    },
    {
      "barlow_loss": 0.1674201637506485,
      "epoch": 0.6631016042780749,
      "loss": 1.6684503555297852,
      "step": 155
    },
    {
      "barlow_loss": 0.16175463795661926,
      "epoch": 0.6631016042780749,
      "loss": 1.6701604127883911,
      "step": 155
    },
    {
      "barlow_loss": 0.16252227127552032,
      "epoch": 0.6631016042780749,
      "loss": 1.5836595296859741,
      "step": 155
    },
    {
      "epoch": 0.667379679144385,
      "grad_norm": 0.7751290798187256,
      "learning_rate": 1.98903754388947e-05,
      "loss": 1.5511,
      "step": 156
    },
    {
      "barlow_loss": 0.1643076241016388,
      "epoch": 0.667379679144385,
      "loss": 1.7459198236465454,
      "step": 156
    },
    {
      "barlow_loss": 0.15748943388462067,
      "epoch": 0.667379679144385,
      "loss": 1.9495866298675537,
      "step": 156
    },
    {
      "barlow_loss": 0.1696619838476181,
      "epoch": 0.667379679144385,
      "loss": 1.646153450012207,
      "step": 156
    },
    {
      "barlow_loss": 0.17812222242355347,
      "epoch": 0.667379679144385,
      "loss": 1.4911930561065674,
      "step": 156
    },
    {
      "barlow_loss": 0.17009615898132324,
      "epoch": 0.6716577540106952,
      "loss": 1.3728302717208862,
      "step": 157
    },
    {
      "barlow_loss": 0.15764199197292328,
      "epoch": 0.6716577540106952,
      "loss": 1.4285191297531128,
      "step": 157
    },
    {
      "barlow_loss": 0.1653091162443161,
      "epoch": 0.6716577540106952,
      "loss": 1.5601868629455566,
      "step": 157
    },
    {
      "barlow_loss": 0.14998020231723785,
      "epoch": 0.6716577540106952,
      "loss": 1.5279417037963867,
      "step": 157
    },
    {
      "barlow_loss": 0.1514916718006134,
      "epoch": 0.6759358288770053,
      "loss": 1.4181630611419678,
      "step": 158
    },
    {
      "barlow_loss": 0.16607192158699036,
      "epoch": 0.6759358288770053,
      "loss": 1.5546566247940063,
      "step": 158
    },
    {
      "barlow_loss": 0.164521723985672,
      "epoch": 0.6759358288770053,
      "loss": 1.628808617591858,
      "step": 158
    },
    {
      "barlow_loss": 0.16545294225215912,
      "epoch": 0.6759358288770053,
      "loss": 1.4409235715866089,
      "step": 158
    },
    {
      "barlow_loss": 0.16759447753429413,
      "epoch": 0.6802139037433155,
      "loss": 1.5378702878952026,
      "step": 159
    },
    {
      "barlow_loss": 0.16659224033355713,
      "epoch": 0.6802139037433155,
      "loss": 1.6652231216430664,
      "step": 159
    },
    {
      "barlow_loss": 0.16627228260040283,
      "epoch": 0.6802139037433155,
      "loss": 1.6008676290512085,
      "step": 159
    },
    {
      "barlow_loss": 0.1670747846364975,
      "epoch": 0.6802139037433155,
      "loss": 1.5970646142959595,
      "step": 159
    },
    {
      "epoch": 0.6844919786096256,
      "grad_norm": 0.8076488375663757,
      "learning_rate": 1.9600809334139446e-05,
      "loss": 1.5729,
      "step": 160
    },
    {
      "barlow_loss": 0.15990948677062988,
      "epoch": 0.6844919786096256,
      "loss": 1.2365812063217163,
      "step": 160
    },
    {
      "barlow_loss": 0.16896936297416687,
      "epoch": 0.6844919786096256,
      "loss": 1.633705973625183,
      "step": 160
    },
    {
      "barlow_loss": 0.1522538959980011,
      "epoch": 0.6844919786096256,
      "loss": 1.5744277238845825,
      "step": 160
    },
    {
      "barlow_loss": 0.1683734804391861,
      "epoch": 0.6844919786096256,
      "loss": 1.466213345527649,
      "step": 160
    },
    {
      "barlow_loss": 0.14953762292861938,
      "epoch": 0.6887700534759358,
      "loss": 1.6991257667541504,
      "step": 161
    },
    {
      "barlow_loss": 0.16517087817192078,
      "epoch": 0.6887700534759358,
      "loss": 1.5380820035934448,
      "step": 161
    },
    {
      "barlow_loss": 0.1577536016702652,
      "epoch": 0.6887700534759358,
      "loss": 1.4748564958572388,
      "step": 161
    },
    {
      "barlow_loss": 0.1510472148656845,
      "epoch": 0.6887700534759358,
      "loss": 1.464156985282898,
      "step": 161
    },
    {
      "barlow_loss": 0.1557035595178604,
      "epoch": 0.6930481283422459,
      "loss": 1.5717599391937256,
      "step": 162
    },
    {
      "barlow_loss": 0.15272223949432373,
      "epoch": 0.6930481283422459,
      "loss": 1.5699818134307861,
      "step": 162
    },
    {
      "barlow_loss": 0.16470572352409363,
      "epoch": 0.6930481283422459,
      "loss": 1.4072067737579346,
      "step": 162
    },
    {
      "barlow_loss": 0.16356992721557617,
      "epoch": 0.6930481283422459,
      "loss": 1.4926750659942627,
      "step": 162
    },
    {
      "barlow_loss": 0.16640397906303406,
      "epoch": 0.6973262032085561,
      "loss": 1.5579484701156616,
      "step": 163
    },
    {
      "barlow_loss": 0.1666896939277649,
      "epoch": 0.6973262032085561,
      "loss": 1.4976333379745483,
      "step": 163
    },
    {
      "barlow_loss": 0.16991540789604187,
      "epoch": 0.6973262032085561,
      "loss": 1.8414894342422485,
      "step": 163
    },
    {
      "barlow_loss": 0.17313224077224731,
      "epoch": 0.6973262032085561,
      "loss": 1.4432940483093262,
      "step": 163
    },
    {
      "epoch": 0.7016042780748664,
      "grad_norm": 0.8423817753791809,
      "learning_rate": 1.9305503997301683e-05,
      "loss": 1.5293,
      "step": 164
    },
    {
      "barlow_loss": 0.16564452648162842,
      "epoch": 0.7016042780748664,
      "loss": 1.48248291015625,
      "step": 164
    },
    {
      "barlow_loss": 0.16601262986660004,
      "epoch": 0.7016042780748664,
      "loss": 1.5582211017608643,
      "step": 164
    },
    {
      "barlow_loss": 0.1555272936820984,
      "epoch": 0.7016042780748664,
      "loss": 1.4633535146713257,
      "step": 164
    },
    {
      "barlow_loss": 0.15092149376869202,
      "epoch": 0.7016042780748664,
      "loss": 1.3795582056045532,
      "step": 164
    },
    {
      "barlow_loss": 0.17307229340076447,
      "epoch": 0.7058823529411765,
      "loss": 1.6234965324401855,
      "step": 165
    },
    {
      "barlow_loss": 0.17223533987998962,
      "epoch": 0.7058823529411765,
      "loss": 1.468836784362793,
      "step": 165
    },
    {
      "barlow_loss": 0.1621953696012497,
      "epoch": 0.7058823529411765,
      "loss": 1.5316087007522583,
      "step": 165
    },
    {
      "barlow_loss": 0.15892793238162994,
      "epoch": 0.7058823529411765,
      "loss": 1.4015347957611084,
      "step": 165
    },
    {
      "barlow_loss": 0.17215536534786224,
      "epoch": 0.7101604278074867,
      "loss": 1.7589812278747559,
      "step": 166
    },
    {
      "barlow_loss": 0.154892697930336,
      "epoch": 0.7101604278074867,
      "loss": 1.5179580450057983,
      "step": 166
    },
    {
      "barlow_loss": 0.1583665907382965,
      "epoch": 0.7101604278074867,
      "loss": 1.5798966884613037,
      "step": 166
    },
    {
      "barlow_loss": 0.17401327192783356,
      "epoch": 0.7101604278074867,
      "loss": 1.599365472793579,
      "step": 166
    },
    {
      "barlow_loss": 0.15607766807079315,
      "epoch": 0.7144385026737968,
      "loss": 1.6001638174057007,
      "step": 167
    },
    {
      "barlow_loss": 0.16552072763442993,
      "epoch": 0.7144385026737968,
      "loss": 1.5079772472381592,
      "step": 167
    },
    {
      "barlow_loss": 0.14951002597808838,
      "epoch": 0.7144385026737968,
      "loss": 1.3434123992919922,
      "step": 167
    },
    {
      "barlow_loss": 0.16549499332904816,
      "epoch": 0.7144385026737968,
      "loss": 1.3491089344024658,
      "step": 167
    },
    {
      "epoch": 0.718716577540107,
      "grad_norm": 0.7214534878730774,
      "learning_rate": 1.9004698109042098e-05,
      "loss": 1.5104,
      "step": 168
    },
    {
      "barlow_loss": 0.15528662502765656,
      "epoch": 0.718716577540107,
      "loss": 1.3199784755706787,
      "step": 168
    },
    {
      "barlow_loss": 0.1605503410100937,
      "epoch": 0.718716577540107,
      "loss": 1.5329419374465942,
      "step": 168
    },
    {
      "barlow_loss": 0.1697731465101242,
      "epoch": 0.718716577540107,
      "loss": 1.612616777420044,
      "step": 168
    },
    {
      "barlow_loss": 0.17531657218933105,
      "epoch": 0.718716577540107,
      "loss": 1.7198518514633179,
      "step": 168
    },
    {
      "barlow_loss": 0.14694547653198242,
      "epoch": 0.7229946524064171,
      "loss": 1.4934654235839844,
      "step": 169
    },
    {
      "barlow_loss": 0.1863529235124588,
      "epoch": 0.7229946524064171,
      "loss": 1.394136905670166,
      "step": 169
    },
    {
      "barlow_loss": 0.1528354287147522,
      "epoch": 0.7229946524064171,
      "loss": 1.5450576543807983,
      "step": 169
    },
    {
      "barlow_loss": 0.16891977190971375,
      "epoch": 0.7229946524064171,
      "loss": 1.5498356819152832,
      "step": 169
    },
    {
      "barlow_loss": 0.15071669220924377,
      "epoch": 0.7272727272727273,
      "loss": 1.5255502462387085,
      "step": 170
    },
    {
      "barlow_loss": 0.17036139965057373,
      "epoch": 0.7272727272727273,
      "loss": 1.5451571941375732,
      "step": 170
    },
    {
      "barlow_loss": 0.1596231311559677,
      "epoch": 0.7272727272727273,
      "loss": 1.3812382221221924,
      "step": 170
    },
    {
      "barlow_loss": 0.1692906618118286,
      "epoch": 0.7272727272727273,
      "loss": 1.5186437368392944,
      "step": 170
    },
    {
      "barlow_loss": 0.16580578684806824,
      "epoch": 0.7315508021390374,
      "loss": 1.5200988054275513,
      "step": 171
    },
    {
      "barlow_loss": 0.1677360087633133,
      "epoch": 0.7315508021390374,
      "loss": 1.393332839012146,
      "step": 171
    },
    {
      "barlow_loss": 0.16885100305080414,
      "epoch": 0.7315508021390374,
      "loss": 1.5767266750335693,
      "step": 171
    },
    {
      "barlow_loss": 0.14446425437927246,
      "epoch": 0.7315508021390374,
      "loss": 1.6997956037521362,
      "step": 171
    },
    {
      "epoch": 0.7358288770053476,
      "grad_norm": 0.8361346125602722,
      "learning_rate": 1.8698634795844334e-05,
      "loss": 1.5205,
      "step": 172
    },
    {
      "barlow_loss": 0.1605420708656311,
      "epoch": 0.7358288770053476,
      "loss": 1.5094459056854248,
      "step": 172
    },
    {
      "barlow_loss": 0.14515040814876556,
      "epoch": 0.7358288770053476,
      "loss": 1.417251706123352,
      "step": 172
    },
    {
      "barlow_loss": 0.17763498425483704,
      "epoch": 0.7358288770053476,
      "loss": 1.5533936023712158,
      "step": 172
    },
    {
      "barlow_loss": 0.1671605259180069,
      "epoch": 0.7358288770053476,
      "loss": 1.5708826780319214,
      "step": 172
    },
    {
      "barlow_loss": 0.1692424714565277,
      "epoch": 0.7401069518716578,
      "loss": 1.4777910709381104,
      "step": 173
    },
    {
      "barlow_loss": 0.1673140972852707,
      "epoch": 0.7401069518716578,
      "loss": 1.5132638216018677,
      "step": 173
    },
    {
      "barlow_loss": 0.1659003645181656,
      "epoch": 0.7401069518716578,
      "loss": 1.52165949344635,
      "step": 173
    },
    {
      "barlow_loss": 0.16515615582466125,
      "epoch": 0.7401069518716578,
      "loss": 1.4271782636642456,
      "step": 173
    },
    {
      "barlow_loss": 0.1565697342157364,
      "epoch": 0.7443850267379679,
      "loss": 1.384207844734192,
      "step": 174
    },
    {
      "barlow_loss": 0.1625346839427948,
      "epoch": 0.7443850267379679,
      "loss": 1.4319226741790771,
      "step": 174
    },
    {
      "barlow_loss": 0.16072410345077515,
      "epoch": 0.7443850267379679,
      "loss": 1.60719895362854,
      "step": 174
    },
    {
      "barlow_loss": 0.15877848863601685,
      "epoch": 0.7443850267379679,
      "loss": 1.653768539428711,
      "step": 174
    },
    {
      "barlow_loss": 0.1638067364692688,
      "epoch": 0.7486631016042781,
      "loss": 1.8074322938919067,
      "step": 175
    },
    {
      "barlow_loss": 0.15103553235530853,
      "epoch": 0.7486631016042781,
      "loss": 1.4314228296279907,
      "step": 175
    },
    {
      "barlow_loss": 0.16527695953845978,
      "epoch": 0.7486631016042781,
      "loss": 1.4442178010940552,
      "step": 175
    },
    {
      "barlow_loss": 0.15947285294532776,
      "epoch": 0.7486631016042781,
      "loss": 1.7032605409622192,
      "step": 175
    },
    {
      "epoch": 0.7529411764705882,
      "grad_norm": 0.870806872844696,
      "learning_rate": 1.8387561433507935e-05,
      "loss": 1.5284,
      "step": 176
    },
    {
      "barlow_loss": 0.16980765759944916,
      "epoch": 0.7529411764705882,
      "loss": 1.4185514450073242,
      "step": 176
    },
    {
      "barlow_loss": 0.15583772957324982,
      "epoch": 0.7529411764705882,
      "loss": 1.5512032508850098,
      "step": 176
    },
    {
      "barlow_loss": 0.1572200059890747,
      "epoch": 0.7529411764705882,
      "loss": 1.4716449975967407,
      "step": 176
    },
    {
      "barlow_loss": 0.1474006175994873,
      "epoch": 0.7529411764705882,
      "loss": 1.533575415611267,
      "step": 176
    },
    {
      "barlow_loss": 0.17629875242710114,
      "epoch": 0.7572192513368984,
      "loss": 1.358595848083496,
      "step": 177
    },
    {
      "barlow_loss": 0.16544023156166077,
      "epoch": 0.7572192513368984,
      "loss": 1.525640606880188,
      "step": 177
    },
    {
      "barlow_loss": 0.17306843400001526,
      "epoch": 0.7572192513368984,
      "loss": 1.7309869527816772,
      "step": 177
    },
    {
      "barlow_loss": 0.18157030642032623,
      "epoch": 0.7572192513368984,
      "loss": 1.7392773628234863,
      "step": 177
    },
    {
      "barlow_loss": 0.17126956582069397,
      "epoch": 0.7614973262032085,
      "loss": 1.8134516477584839,
      "step": 178
    },
    {
      "barlow_loss": 0.17197458446025848,
      "epoch": 0.7614973262032085,
      "loss": 1.4065461158752441,
      "step": 178
    },
    {
      "barlow_loss": 0.16823755204677582,
      "epoch": 0.7614973262032085,
      "loss": 1.3416107892990112,
      "step": 178
    },
    {
      "barlow_loss": 0.1672968566417694,
      "epoch": 0.7614973262032085,
      "loss": 1.4988274574279785,
      "step": 178
    },
    {
      "barlow_loss": 0.17195436358451843,
      "epoch": 0.7657754010695187,
      "loss": 1.5286332368850708,
      "step": 179
    },
    {
      "barlow_loss": 0.18571288883686066,
      "epoch": 0.7657754010695187,
      "loss": 1.6393464803695679,
      "step": 179
    },
    {
      "barlow_loss": 0.14829064905643463,
      "epoch": 0.7657754010695187,
      "loss": 1.4576386213302612,
      "step": 179
    },
    {
      "barlow_loss": 0.15792177617549896,
      "epoch": 0.7657754010695187,
      "loss": 1.5582445859909058,
      "step": 179
    },
    {
      "epoch": 0.7700534759358288,
      "grad_norm": 0.8129858374595642,
      "learning_rate": 1.8071729447206733e-05,
      "loss": 1.5359,
      "step": 180
    },
    {
      "barlow_loss": 0.15354563295841217,
      "epoch": 0.7700534759358288,
      "loss": 1.6380497217178345,
      "step": 180
    },
    {
      "barlow_loss": 0.15887515246868134,
      "epoch": 0.7700534759358288,
      "loss": 1.6269192695617676,
      "step": 180
    },
    {
      "barlow_loss": 0.17773310840129852,
      "epoch": 0.7700534759358288,
      "loss": 1.482541799545288,
      "step": 180
    },
    {
      "barlow_loss": 0.14692501723766327,
      "epoch": 0.7700534759358288,
      "loss": 1.713883638381958,
      "step": 180
    },
    {
      "barlow_loss": 0.16280481219291687,
      "epoch": 0.774331550802139,
      "loss": 1.517274260520935,
      "step": 181
    },
    {
      "barlow_loss": 0.16978299617767334,
      "epoch": 0.774331550802139,
      "loss": 1.5974794626235962,
      "step": 181
    },
    {
      "barlow_loss": 0.18233539164066315,
      "epoch": 0.774331550802139,
      "loss": 1.466374158859253,
      "step": 181
    },
    {
      "barlow_loss": 0.1587313711643219,
      "epoch": 0.774331550802139,
      "loss": 1.7532706260681152,
      "step": 181
    },
    {
      "barlow_loss": 0.17542192339897156,
      "epoch": 0.7786096256684492,
      "loss": 1.3346247673034668,
      "step": 182
    },
    {
      "barlow_loss": 0.1619897186756134,
      "epoch": 0.7786096256684492,
      "loss": 1.7643612623214722,
      "step": 182
    },
    {
      "barlow_loss": 0.17516155540943146,
      "epoch": 0.7786096256684492,
      "loss": 1.351156234741211,
      "step": 182
    },
    {
      "barlow_loss": 0.16250933706760406,
      "epoch": 0.7786096256684492,
      "loss": 1.4738188982009888,
      "step": 182
    },
    {
      "barlow_loss": 0.17327257990837097,
      "epoch": 0.7828877005347593,
      "loss": 2.067648410797119,
      "step": 183
    },
    {
      "barlow_loss": 0.16442033648490906,
      "epoch": 0.7828877005347593,
      "loss": 1.529763102531433,
      "step": 183
    },
    {
      "barlow_loss": 0.17436619102954865,
      "epoch": 0.7828877005347593,
      "loss": 1.4258652925491333,
      "step": 183
    },
    {
      "barlow_loss": 0.17925062775611877,
      "epoch": 0.7828877005347593,
      "loss": 1.4907898902893066,
      "step": 183
    },
    {
      "epoch": 0.7871657754010695,
      "grad_norm": 0.9454470872879028,
      "learning_rate": 1.775139410827433e-05,
      "loss": 1.5771,
      "step": 184
    },
    {
      "barlow_loss": 0.18075227737426758,
      "epoch": 0.7871657754010695,
      "loss": 1.6687160730361938,
      "step": 184
    },
    {
      "barlow_loss": 0.16777470707893372,
      "epoch": 0.7871657754010695,
      "loss": 1.5152783393859863,
      "step": 184
    },
    {
      "barlow_loss": 0.16266043484210968,
      "epoch": 0.7871657754010695,
      "loss": 1.6852861642837524,
      "step": 184
    },
    {
      "barlow_loss": 0.16112153232097626,
      "epoch": 0.7871657754010695,
      "loss": 1.3669331073760986,
      "step": 184
    },
    {
      "barlow_loss": 0.15866301953792572,
      "epoch": 0.7914438502673797,
      "loss": 1.4186270236968994,
      "step": 185
    },
    {
      "barlow_loss": 0.14988932013511658,
      "epoch": 0.7914438502673797,
      "loss": 1.6264803409576416,
      "step": 185
    },
    {
      "barlow_loss": 0.1695926934480667,
      "epoch": 0.7914438502673797,
      "loss": 1.625841498374939,
      "step": 185
    },
    {
      "barlow_loss": 0.16845542192459106,
      "epoch": 0.7914438502673797,
      "loss": 1.4679250717163086,
      "step": 185
    },
    {
      "barlow_loss": 0.1670072376728058,
      "epoch": 0.7957219251336899,
      "loss": 1.612380862236023,
      "step": 186
    },
    {
      "barlow_loss": 0.177758127450943,
      "epoch": 0.7957219251336899,
      "loss": 1.6552177667617798,
      "step": 186
    },
    {
      "barlow_loss": 0.16068041324615479,
      "epoch": 0.7957219251336899,
      "loss": 1.476547360420227,
      "step": 186
    },
    {
      "barlow_loss": 0.15395022928714752,
      "epoch": 0.7957219251336899,
      "loss": 1.3628968000411987,
      "step": 186
    },
    {
      "barlow_loss": 0.14837872982025146,
      "epoch": 0.8,
      "loss": 1.461277961730957,
      "step": 187
    },
    {
      "barlow_loss": 0.1714535504579544,
      "epoch": 0.8,
      "loss": 1.472104549407959,
      "step": 187
    },
    {
      "barlow_loss": 0.1855722814798355,
      "epoch": 0.8,
      "loss": 1.646425724029541,
      "step": 187
    },
    {
      "barlow_loss": 0.17940106987953186,
      "epoch": 0.8,
      "loss": 1.564677357673645,
      "step": 187
    },
    {
      "epoch": 0.8042780748663102,
      "grad_norm": 0.8154215216636658,
      "learning_rate": 1.742681432788097e-05,
      "loss": 1.5392,
      "step": 188
    },
    {
      "barlow_loss": 0.1445707380771637,
      "epoch": 0.8042780748663102,
      "loss": 1.442139744758606,
      "step": 188
    },
    {
      "barlow_loss": 0.17086264491081238,
      "epoch": 0.8042780748663102,
      "loss": 1.5731977224349976,
      "step": 188
    },
    {
      "barlow_loss": 0.16091081500053406,
      "epoch": 0.8042780748663102,
      "loss": 1.3424921035766602,
      "step": 188
    },
    {
      "barlow_loss": 0.15003365278244019,
      "epoch": 0.8042780748663102,
      "loss": 1.5054030418395996,
      "step": 188
    },
    {
      "barlow_loss": 0.17713132500648499,
      "epoch": 0.8085561497326204,
      "loss": 1.4232109785079956,
      "step": 189
    },
    {
      "barlow_loss": 0.17599190771579742,
      "epoch": 0.8085561497326204,
      "loss": 1.3727182149887085,
      "step": 189
    },
    {
      "barlow_loss": 0.1550561785697937,
      "epoch": 0.8085561497326204,
      "loss": 1.410263180732727,
      "step": 189
    },
    {
      "barlow_loss": 0.1774967908859253,
      "epoch": 0.8085561497326204,
      "loss": 1.5417921543121338,
      "step": 189
    },
    {
      "barlow_loss": 0.1580280363559723,
      "epoch": 0.8128342245989305,
      "loss": 1.385986566543579,
      "step": 190
    },
    {
      "barlow_loss": 0.17119315266609192,
      "epoch": 0.8128342245989305,
      "loss": 1.2897114753723145,
      "step": 190
    },
    {
      "barlow_loss": 0.17042049765586853,
      "epoch": 0.8128342245989305,
      "loss": 1.3521009683609009,
      "step": 190
    },
    {
      "barlow_loss": 0.1742289811372757,
      "epoch": 0.8128342245989305,
      "loss": 1.4429950714111328,
      "step": 190
    },
    {
      "barlow_loss": 0.16149598360061646,
      "epoch": 0.8171122994652407,
      "loss": 1.469161033630371,
      "step": 191
    },
    {
      "barlow_loss": 0.17074993252754211,
      "epoch": 0.8171122994652407,
      "loss": 1.4629018306732178,
      "step": 191
    },
    {
      "barlow_loss": 0.15322667360305786,
      "epoch": 0.8171122994652407,
      "loss": 1.2953928709030151,
      "step": 191
    },
    {
      "barlow_loss": 0.1576496809720993,
      "epoch": 0.8171122994652407,
      "loss": 1.4415489435195923,
      "step": 191
    },
    {
      "epoch": 0.8213903743315508,
      "grad_norm": 0.764717161655426,
      "learning_rate": 1.7098252447768474e-05,
      "loss": 1.4219,
      "step": 192
    },
    {
      "barlow_loss": 0.15285499393939972,
      "epoch": 0.8213903743315508,
      "loss": 1.4633545875549316,
      "step": 192
    },
    {
      "barlow_loss": 0.16703090071678162,
      "epoch": 0.8213903743315508,
      "loss": 1.4630006551742554,
      "step": 192
    },
    {
      "barlow_loss": 0.1563686579465866,
      "epoch": 0.8213903743315508,
      "loss": 1.540819764137268,
      "step": 192
    },
    {
      "barlow_loss": 0.16685011982917786,
      "epoch": 0.8213903743315508,
      "loss": 1.6324912309646606,
      "step": 192
    },
    {
      "barlow_loss": 0.15770545601844788,
      "epoch": 0.825668449197861,
      "loss": 1.4886085987091064,
      "step": 193
    },
    {
      "barlow_loss": 0.15971511602401733,
      "epoch": 0.825668449197861,
      "loss": 1.5121855735778809,
      "step": 193
    },
    {
      "barlow_loss": 0.15748293697834015,
      "epoch": 0.825668449197861,
      "loss": 1.2933911085128784,
      "step": 193
    },
    {
      "barlow_loss": 0.16434040665626526,
      "epoch": 0.825668449197861,
      "loss": 1.711201786994934,
      "step": 193
    },
    {
      "barlow_loss": 0.15965703129768372,
      "epoch": 0.8299465240641711,
      "loss": 1.495228886604309,
      "step": 194
    },
    {
      "barlow_loss": 0.1518017053604126,
      "epoch": 0.8299465240641711,
      "loss": 1.4379847049713135,
      "step": 194
    },
    {
      "barlow_loss": 0.15958769619464874,
      "epoch": 0.8299465240641711,
      "loss": 1.428547739982605,
      "step": 194
    },
    {
      "barlow_loss": 0.1635461151599884,
      "epoch": 0.8299465240641711,
      "loss": 1.474283218383789,
      "step": 194
    },
    {
      "barlow_loss": 0.16919425129890442,
      "epoch": 0.8342245989304813,
      "loss": 1.583293080329895,
      "step": 195
    },
    {
      "barlow_loss": 0.16699761152267456,
      "epoch": 0.8342245989304813,
      "loss": 1.4314337968826294,
      "step": 195
    },
    {
      "barlow_loss": 0.15603601932525635,
      "epoch": 0.8342245989304813,
      "loss": 1.381665587425232,
      "step": 195
    },
    {
      "barlow_loss": 0.1509171575307846,
      "epoch": 0.8342245989304813,
      "loss": 1.381334900856018,
      "step": 195
    },
    {
      "epoch": 0.8385026737967914,
      "grad_norm": 0.7887957692146301,
      "learning_rate": 1.6765974028212384e-05,
      "loss": 1.4824,
      "step": 196
    },
    {
      "barlow_loss": 0.17089733481407166,
      "epoch": 0.8385026737967914,
      "loss": 1.5804550647735596,
      "step": 196
    },
    {
      "barlow_loss": 0.1740933209657669,
      "epoch": 0.8385026737967914,
      "loss": 1.3448337316513062,
      "step": 196
    },
    {
      "barlow_loss": 0.16747958958148956,
      "epoch": 0.8385026737967914,
      "loss": 1.7062325477600098,
      "step": 196
    },
    {
      "barlow_loss": 0.1710062474012375,
      "epoch": 0.8385026737967914,
      "loss": 1.5819698572158813,
      "step": 196
    },
    {
      "barlow_loss": 0.17592856287956238,
      "epoch": 0.8427807486631016,
      "loss": 1.5814749002456665,
      "step": 197
    },
    {
      "barlow_loss": 0.16961157321929932,
      "epoch": 0.8427807486631016,
      "loss": 1.5254812240600586,
      "step": 197
    },
    {
      "barlow_loss": 0.14491808414459229,
      "epoch": 0.8427807486631016,
      "loss": 1.4427927732467651,
      "step": 197
    },
    {
      "barlow_loss": 0.1562959998846054,
      "epoch": 0.8427807486631016,
      "loss": 1.458741307258606,
      "step": 197
    },
    {
      "barlow_loss": 0.17059287428855896,
      "epoch": 0.8470588235294118,
      "loss": 1.4829026460647583,
      "step": 198
    },
    {
      "barlow_loss": 0.16391418874263763,
      "epoch": 0.8470588235294118,
      "loss": 1.523331880569458,
      "step": 198
    },
    {
      "barlow_loss": 0.16603586077690125,
      "epoch": 0.8470588235294118,
      "loss": 1.5661251544952393,
      "step": 198
    },
    {
      "barlow_loss": 0.17198243737220764,
      "epoch": 0.8470588235294118,
      "loss": 1.5726466178894043,
      "step": 198
    },
    {
      "barlow_loss": 0.15778334438800812,
      "epoch": 0.8513368983957219,
      "loss": 1.39322829246521,
      "step": 199
    },
    {
      "barlow_loss": 0.14658544957637787,
      "epoch": 0.8513368983957219,
      "loss": 1.216208815574646,
      "step": 199
    },
    {
      "barlow_loss": 0.1626572608947754,
      "epoch": 0.8513368983957219,
      "loss": 1.5979241132736206,
      "step": 199
    },
    {
      "barlow_loss": 0.16294154524803162,
      "epoch": 0.8513368983957219,
      "loss": 1.4726885557174683,
      "step": 199
    },
    {
      "epoch": 0.8556149732620321,
      "grad_norm": 0.8482218384742737,
      "learning_rate": 1.64302476333828e-05,
      "loss": 1.5029,
      "step": 200
    },
    {
      "barlow_loss": 0.15740053355693817,
      "epoch": 0.8556149732620321,
      "loss": 1.2847217321395874,
      "step": 200
    },
    {
      "barlow_loss": 0.16856835782527924,
      "epoch": 0.8556149732620321,
      "loss": 1.300281047821045,
      "step": 200
    },
    {
      "barlow_loss": 0.15918512642383575,
      "epoch": 0.8556149732620321,
      "loss": 1.4677956104278564,
      "step": 200
    },
    {
      "barlow_loss": 0.1416899859905243,
      "epoch": 0.8556149732620321,
      "loss": 1.5127310752868652,
      "step": 200
    },
    {
      "barlow_loss": 0.1512111872434616,
      "epoch": 0.8598930481283422,
      "loss": 1.4053198099136353,
      "step": 201
    },
    {
      "barlow_loss": 0.16876691579818726,
      "epoch": 0.8598930481283422,
      "loss": 1.4431220293045044,
      "step": 201
    },
    {
      "barlow_loss": 0.16827018558979034,
      "epoch": 0.8598930481283422,
      "loss": 1.5585386753082275,
      "step": 201
    },
    {
      "barlow_loss": 0.1606370210647583,
      "epoch": 0.8598930481283422,
      "loss": 1.2917506694793701,
      "step": 201
    },
    {
      "barlow_loss": 0.17728567123413086,
      "epoch": 0.8641711229946524,
      "loss": 1.4851795434951782,
      "step": 202
    },
    {
      "barlow_loss": 0.16511081159114838,
      "epoch": 0.8641711229946524,
      "loss": 1.6566845178604126,
      "step": 202
    },
    {
      "barlow_loss": 0.14821743965148926,
      "epoch": 0.8641711229946524,
      "loss": 1.5238397121429443,
      "step": 202
    },
    {
      "barlow_loss": 0.1711300164461136,
      "epoch": 0.8641711229946524,
      "loss": 1.494301438331604,
      "step": 202
    },
    {
      "barlow_loss": 0.15040011703968048,
      "epoch": 0.8684491978609625,
      "loss": 1.4951646327972412,
      "step": 203
    },
    {
      "barlow_loss": 0.1793406754732132,
      "epoch": 0.8684491978609625,
      "loss": 1.4138076305389404,
      "step": 203
    },
    {
      "barlow_loss": 0.16951851546764374,
      "epoch": 0.8684491978609625,
      "loss": 1.449324607849121,
      "step": 203
    },
    {
      "barlow_loss": 0.16430912911891937,
      "epoch": 0.8684491978609625,
      "loss": 1.4476677179336548,
      "step": 203
    },
    {
      "epoch": 0.8727272727272727,
      "grad_norm": 0.8205013871192932,
      "learning_rate": 1.6091344614277258e-05,
      "loss": 1.4519,
      "step": 204
    },
    {
      "barlow_loss": 0.16103285551071167,
      "epoch": 0.8727272727272727,
      "loss": 1.5534886121749878,
      "step": 204
    },
    {
      "barlow_loss": 0.16593074798583984,
      "epoch": 0.8727272727272727,
      "loss": 1.5787615776062012,
      "step": 204
    },
    {
      "barlow_loss": 0.1602737009525299,
      "epoch": 0.8727272727272727,
      "loss": 1.2727614641189575,
      "step": 204
    },
    {
      "barlow_loss": 0.16754943132400513,
      "epoch": 0.8727272727272727,
      "loss": 1.4939152002334595,
      "step": 204
    },
    {
      "barlow_loss": 0.1692991405725479,
      "epoch": 0.8770053475935828,
      "loss": 1.368924617767334,
      "step": 205
    },
    {
      "barlow_loss": 0.16313305497169495,
      "epoch": 0.8770053475935828,
      "loss": 1.2597209215164185,
      "step": 205
    },
    {
      "barlow_loss": 0.17084039747714996,
      "epoch": 0.8770053475935828,
      "loss": 1.6544684171676636,
      "step": 205
    },
    {
      "barlow_loss": 0.1786745935678482,
      "epoch": 0.8770053475935828,
      "loss": 1.6403464078903198,
      "step": 205
    },
    {
      "barlow_loss": 0.15015646815299988,
      "epoch": 0.881283422459893,
      "loss": 1.4241136312484741,
      "step": 206
    },
    {
      "barlow_loss": 0.15726681053638458,
      "epoch": 0.881283422459893,
      "loss": 1.4408397674560547,
      "step": 206
    },
    {
      "barlow_loss": 0.15111488103866577,
      "epoch": 0.881283422459893,
      "loss": 1.405232548713684,
      "step": 206
    },
    {
      "barlow_loss": 0.1774623990058899,
      "epoch": 0.881283422459893,
      "loss": 1.5679022073745728,
      "step": 206
    },
    {
      "barlow_loss": 0.1552904099225998,
      "epoch": 0.8855614973262033,
      "loss": 1.324870228767395,
      "step": 207
    },
    {
      "barlow_loss": 0.16926321387290955,
      "epoch": 0.8855614973262033,
      "loss": 1.4854081869125366,
      "step": 207
    },
    {
      "barlow_loss": 0.16289100050926208,
      "epoch": 0.8855614973262033,
      "loss": 1.3910768032073975,
      "step": 207
    },
    {
      "barlow_loss": 0.168021097779274,
      "epoch": 0.8855614973262033,
      "loss": 1.4610447883605957,
      "step": 207
    },
    {
      "epoch": 0.8898395721925134,
      "grad_norm": 0.8189829587936401,
      "learning_rate": 1.5749538889401193e-05,
      "loss": 1.4577,
      "step": 208
    },
    {
      "barlow_loss": 0.1661664843559265,
      "epoch": 0.8898395721925134,
      "loss": 1.5559418201446533,
      "step": 208
    },
    {
      "barlow_loss": 0.1599818766117096,
      "epoch": 0.8898395721925134,
      "loss": 1.7641083002090454,
      "step": 208
    },
    {
      "barlow_loss": 0.17575043439865112,
      "epoch": 0.8898395721925134,
      "loss": 1.3698441982269287,
      "step": 208
    },
    {
      "barlow_loss": 0.16711486876010895,
      "epoch": 0.8898395721925134,
      "loss": 1.3079913854599,
      "step": 208
    },
    {
      "barlow_loss": 0.1609489917755127,
      "epoch": 0.8941176470588236,
      "loss": 1.4383912086486816,
      "step": 209
    },
    {
      "barlow_loss": 0.16190004348754883,
      "epoch": 0.8941176470588236,
      "loss": 1.407212495803833,
      "step": 209
    },
    {
      "barlow_loss": 0.1483662724494934,
      "epoch": 0.8941176470588236,
      "loss": 1.290302038192749,
      "step": 209
    },
    {
      "barlow_loss": 0.1519496887922287,
      "epoch": 0.8941176470588236,
      "loss": 1.3889186382293701,
      "step": 209
    },
    {
      "barlow_loss": 0.17253831028938293,
      "epoch": 0.8983957219251337,
      "loss": 1.5523532629013062,
      "step": 210
    },
    {
      "barlow_loss": 0.17655488848686218,
      "epoch": 0.8983957219251337,
      "loss": 1.5533736944198608,
      "step": 210
    },
    {
      "barlow_loss": 0.15463146567344666,
      "epoch": 0.8983957219251337,
      "loss": 1.474972128868103,
      "step": 210
    },
    {
      "barlow_loss": 0.1646728515625,
      "epoch": 0.8983957219251337,
      "loss": 1.5057637691497803,
      "step": 210
    },
    {
      "barlow_loss": 0.17113302648067474,
      "epoch": 0.9026737967914439,
      "loss": 1.4260566234588623,
      "step": 211
    },
    {
      "barlow_loss": 0.17631566524505615,
      "epoch": 0.9026737967914439,
      "loss": 1.523959994316101,
      "step": 211
    },
    {
      "barlow_loss": 0.17336899042129517,
      "epoch": 0.9026737967914439,
      "loss": 1.2867684364318848,
      "step": 211
    },
    {
      "barlow_loss": 0.1736525297164917,
      "epoch": 0.9026737967914439,
      "loss": 1.1885663270950317,
      "step": 211
    },
    {
      "epoch": 0.906951871657754,
      "grad_norm": 0.7519401907920837,
      "learning_rate": 1.540510672337316e-05,
      "loss": 1.4397,
      "step": 212
    },
    {
      "barlow_loss": 0.1579951047897339,
      "epoch": 0.906951871657754,
      "loss": 1.5370335578918457,
      "step": 212
    },
    {
      "barlow_loss": 0.16582100093364716,
      "epoch": 0.906951871657754,
      "loss": 1.4909372329711914,
      "step": 212
    },
    {
      "barlow_loss": 0.1681341528892517,
      "epoch": 0.906951871657754,
      "loss": 1.4995168447494507,
      "step": 212
    },
    {
      "barlow_loss": 0.1537277102470398,
      "epoch": 0.906951871657754,
      "loss": 1.49972403049469,
      "step": 212
    },
    {
      "barlow_loss": 0.15525324642658234,
      "epoch": 0.9112299465240642,
      "loss": 1.4759628772735596,
      "step": 213
    },
    {
      "barlow_loss": 0.16929998993873596,
      "epoch": 0.9112299465240642,
      "loss": 1.5890893936157227,
      "step": 213
    },
    {
      "barlow_loss": 0.17053790390491486,
      "epoch": 0.9112299465240642,
      "loss": 1.465419888496399,
      "step": 213
    },
    {
      "barlow_loss": 0.15480607748031616,
      "epoch": 0.9112299465240642,
      "loss": 1.365796685218811,
      "step": 213
    },
    {
      "barlow_loss": 0.15261276066303253,
      "epoch": 0.9155080213903743,
      "loss": 1.4497346878051758,
      "step": 214
    },
    {
      "barlow_loss": 0.16999934613704681,
      "epoch": 0.9155080213903743,
      "loss": 1.8107244968414307,
      "step": 214
    },
    {
      "barlow_loss": 0.1645309329032898,
      "epoch": 0.9155080213903743,
      "loss": 1.4910529851913452,
      "step": 214
    },
    {
      "barlow_loss": 0.16849882900714874,
      "epoch": 0.9155080213903743,
      "loss": 1.5711445808410645,
      "step": 214
    },
    {
      "barlow_loss": 0.16660769283771515,
      "epoch": 0.9197860962566845,
      "loss": 1.2861415147781372,
      "step": 215
    },
    {
      "barlow_loss": 0.16972915828227997,
      "epoch": 0.9197860962566845,
      "loss": 1.317273497581482,
      "step": 215
    },
    {
      "barlow_loss": 0.16738677024841309,
      "epoch": 0.9197860962566845,
      "loss": 1.4170862436294556,
      "step": 215
    },
    {
      "barlow_loss": 0.16058054566383362,
      "epoch": 0.9197860962566845,
      "loss": 1.4469919204711914,
      "step": 215
    },
    {
      "epoch": 0.9240641711229947,
      "grad_norm": 0.779980480670929,
      "learning_rate": 1.5058326503633877e-05,
      "loss": 1.4821,
      "step": 216
    },
    {
      "barlow_loss": 0.165887251496315,
      "epoch": 0.9240641711229947,
      "loss": 1.4558188915252686,
      "step": 216
    },
    {
      "barlow_loss": 0.1551951766014099,
      "epoch": 0.9240641711229947,
      "loss": 1.4547317028045654,
      "step": 216
    },
    {
      "barlow_loss": 0.17293457686901093,
      "epoch": 0.9240641711229947,
      "loss": 1.404870867729187,
      "step": 216
    },
    {
      "barlow_loss": 0.15555962920188904,
      "epoch": 0.9240641711229947,
      "loss": 1.463140845298767,
      "step": 216
    },
    {
      "barlow_loss": 0.1682032346725464,
      "epoch": 0.9283422459893048,
      "loss": 1.5067222118377686,
      "step": 217
    },
    {
      "barlow_loss": 0.15719924867153168,
      "epoch": 0.9283422459893048,
      "loss": 1.3860998153686523,
      "step": 217
    },
    {
      "barlow_loss": 0.15797021985054016,
      "epoch": 0.9283422459893048,
      "loss": 1.5007219314575195,
      "step": 217
    },
    {
      "barlow_loss": 0.13939501345157623,
      "epoch": 0.9283422459893048,
      "loss": 1.3700437545776367,
      "step": 217
    },
    {
      "barlow_loss": 0.15353628993034363,
      "epoch": 0.932620320855615,
      "loss": 1.3613300323486328,
      "step": 218
    },
    {
      "barlow_loss": 0.16899895668029785,
      "epoch": 0.932620320855615,
      "loss": 1.4976444244384766,
      "step": 218
    },
    {
      "barlow_loss": 0.16479933261871338,
      "epoch": 0.932620320855615,
      "loss": 1.4251494407653809,
      "step": 218
    },
    {
      "barlow_loss": 0.1679021269083023,
      "epoch": 0.932620320855615,
      "loss": 1.489304542541504,
      "step": 218
    },
    {
      "barlow_loss": 0.16334612667560577,
      "epoch": 0.9368983957219251,
      "loss": 1.5018094778060913,
      "step": 219
    },
    {
      "barlow_loss": 0.16517601907253265,
      "epoch": 0.9368983957219251,
      "loss": 1.4699697494506836,
      "step": 219
    },
    {
      "barlow_loss": 0.1808105707168579,
      "epoch": 0.9368983957219251,
      "loss": 1.670850396156311,
      "step": 219
    },
    {
      "barlow_loss": 0.17373618483543396,
      "epoch": 0.9368983957219251,
      "loss": 1.607096791267395,
      "step": 219
    },
    {
      "epoch": 0.9411764705882353,
      "grad_norm": 0.964859664440155,
      "learning_rate": 1.4709478515439437e-05,
      "loss": 1.4728,
      "step": 220
    },
    {
      "barlow_loss": 0.1512172967195511,
      "epoch": 0.9411764705882353,
      "loss": 1.4238420724868774,
      "step": 220
    },
    {
      "barlow_loss": 0.15907222032546997,
      "epoch": 0.9411764705882353,
      "loss": 1.4170199632644653,
      "step": 220
    },
    {
      "barlow_loss": 0.17755214869976044,
      "epoch": 0.9411764705882353,
      "loss": 1.550123929977417,
      "step": 220
    },
    {
      "barlow_loss": 0.16465498507022858,
      "epoch": 0.9411764705882353,
      "loss": 1.5421305894851685,
      "step": 220
    },
    {
      "barlow_loss": 0.15274466574192047,
      "epoch": 0.9454545454545454,
      "loss": 1.419433832168579,
      "step": 221
    },
    {
      "barlow_loss": 0.14772750437259674,
      "epoch": 0.9454545454545454,
      "loss": 1.4055367708206177,
      "step": 221
    },
    {
      "barlow_loss": 0.1644344925880432,
      "epoch": 0.9454545454545454,
      "loss": 1.4482988119125366,
      "step": 221
    },
    {
      "barlow_loss": 0.1755521148443222,
      "epoch": 0.9454545454545454,
      "loss": 1.7120236158370972,
      "step": 221
    },
    {
      "barlow_loss": 0.16208507120609283,
      "epoch": 0.9497326203208556,
      "loss": 1.3447637557983398,
      "step": 222
    },
    {
      "barlow_loss": 0.16892337799072266,
      "epoch": 0.9497326203208556,
      "loss": 1.588897466659546,
      "step": 222
    },
    {
      "barlow_loss": 0.1672305166721344,
      "epoch": 0.9497326203208556,
      "loss": 1.5018230676651,
      "step": 222
    },
    {
      "barlow_loss": 0.15274541079998016,
      "epoch": 0.9497326203208556,
      "loss": 1.421297550201416,
      "step": 222
    },
    {
      "barlow_loss": 0.1678066998720169,
      "epoch": 0.9540106951871657,
      "loss": 1.4676616191864014,
      "step": 223
    },
    {
      "barlow_loss": 0.16958953440189362,
      "epoch": 0.9540106951871657,
      "loss": 1.527853012084961,
      "step": 223
    },
    {
      "barlow_loss": 0.16920532286167145,
      "epoch": 0.9540106951871657,
      "loss": 1.4974465370178223,
      "step": 223
    },
    {
      "barlow_loss": 0.17707090079784393,
      "epoch": 0.9540106951871657,
      "loss": 1.6658525466918945,
      "step": 223
    },
    {
      "epoch": 0.9582887700534759,
      "grad_norm": 0.9037023186683655,
      "learning_rate": 1.4358844715320646e-05,
      "loss": 1.4959,
      "step": 224
    },
    {
      "barlow_loss": 0.1628016233444214,
      "epoch": 0.9582887700534759,
      "loss": 1.3369925022125244,
      "step": 224
    },
    {
      "barlow_loss": 0.1567959487438202,
      "epoch": 0.9582887700534759,
      "loss": 1.3913987874984741,
      "step": 224
    },
    {
      "barlow_loss": 0.17096126079559326,
      "epoch": 0.9582887700534759,
      "loss": 1.4672731161117554,
      "step": 224
    },
    {
      "barlow_loss": 0.18110859394073486,
      "epoch": 0.9582887700534759,
      "loss": 1.4628468751907349,
      "step": 224
    },
    {
      "barlow_loss": 0.17341864109039307,
      "epoch": 0.9625668449197861,
      "loss": 1.5502707958221436,
      "step": 225
    },
    {
      "barlow_loss": 0.16316431760787964,
      "epoch": 0.9625668449197861,
      "loss": 1.5728759765625,
      "step": 225
    },
    {
      "barlow_loss": 0.15572576224803925,
      "epoch": 0.9625668449197861,
      "loss": 1.585825800895691,
      "step": 225
    },
    {
      "barlow_loss": 0.1536334753036499,
      "epoch": 0.9625668449197861,
      "loss": 1.4982017278671265,
      "step": 225
    },
    {
      "barlow_loss": 0.16197945177555084,
      "epoch": 0.9668449197860962,
      "loss": 1.3871264457702637,
      "step": 226
    },
    {
      "barlow_loss": 0.1642138659954071,
      "epoch": 0.9668449197860962,
      "loss": 1.3686013221740723,
      "step": 226
    },
    {
      "barlow_loss": 0.18397648632526398,
      "epoch": 0.9668449197860962,
      "loss": 1.4628443717956543,
      "step": 226
    },
    {
      "barlow_loss": 0.16425667703151703,
      "epoch": 0.9668449197860962,
      "loss": 1.2709975242614746,
      "step": 226
    },
    {
      "barlow_loss": 0.1719474196434021,
      "epoch": 0.9711229946524064,
      "loss": 1.5065052509307861,
      "step": 227
    },
    {
      "barlow_loss": 0.171735942363739,
      "epoch": 0.9711229946524064,
      "loss": 1.685876727104187,
      "step": 227
    },
    {
      "barlow_loss": 0.17384770512580872,
      "epoch": 0.9711229946524064,
      "loss": 1.320768117904663,
      "step": 227
    },
    {
      "barlow_loss": 0.1711738109588623,
      "epoch": 0.9711229946524064,
      "loss": 1.4751875400543213,
      "step": 227
    },
    {
      "epoch": 0.9754010695187165,
      "grad_norm": 0.9077315330505371,
      "learning_rate": 1.4006708503191538e-05,
      "loss": 1.459,
      "step": 228
    },
    {
      "barlow_loss": 0.1644524186849594,
      "epoch": 0.9754010695187165,
      "loss": 1.3927298784255981,
      "step": 228
    },
    {
      "barlow_loss": 0.1571546494960785,
      "epoch": 0.9754010695187165,
      "loss": 1.4194045066833496,
      "step": 228
    },
    {
      "barlow_loss": 0.16687175631523132,
      "epoch": 0.9754010695187165,
      "loss": 1.472839593887329,
      "step": 228
    },
    {
      "barlow_loss": 0.1547909379005432,
      "epoch": 0.9754010695187165,
      "loss": 1.300218939781189,
      "step": 228
    },
    {
      "barlow_loss": 0.16171559691429138,
      "epoch": 0.9796791443850268,
      "loss": 1.4989243745803833,
      "step": 229
    },
    {
      "barlow_loss": 0.16600175201892853,
      "epoch": 0.9796791443850268,
      "loss": 1.4636698961257935,
      "step": 229
    },
    {
      "barlow_loss": 0.16148945689201355,
      "epoch": 0.9796791443850268,
      "loss": 1.4532151222229004,
      "step": 229
    },
    {
      "barlow_loss": 0.17521817982196808,
      "epoch": 0.9796791443850268,
      "loss": 1.6418538093566895,
      "step": 229
    },
    {
      "barlow_loss": 0.15551398694515228,
      "epoch": 0.983957219251337,
      "loss": 1.4294490814208984,
      "step": 230
    },
    {
      "barlow_loss": 0.1448078155517578,
      "epoch": 0.983957219251337,
      "loss": 1.3540139198303223,
      "step": 230
    },
    {
      "barlow_loss": 0.17111890017986298,
      "epoch": 0.983957219251337,
      "loss": 1.429170846939087,
      "step": 230
    },
    {
      "barlow_loss": 0.15904994308948517,
      "epoch": 0.983957219251337,
      "loss": 1.4079792499542236,
      "step": 230
    },
    {
      "barlow_loss": 0.16759131848812103,
      "epoch": 0.9882352941176471,
      "loss": 1.5384013652801514,
      "step": 231
    },
    {
      "barlow_loss": 0.15156246721744537,
      "epoch": 0.9882352941176471,
      "loss": 1.3691767454147339,
      "step": 231
    },
    {
      "barlow_loss": 0.1634349673986435,
      "epoch": 0.9882352941176471,
      "loss": 1.3174028396606445,
      "step": 231
    },
    {
      "barlow_loss": 0.17215333878993988,
      "epoch": 0.9882352941176471,
      "loss": 1.5387215614318848,
      "step": 231
    },
    {
      "epoch": 0.9925133689839573,
      "grad_norm": 0.9200155138969421,
      "learning_rate": 1.3653354493291276e-05,
      "loss": 1.4392,
      "step": 232
    },
    {
      "barlow_loss": 0.17612993717193604,
      "epoch": 0.9925133689839573,
      "loss": 1.3958773612976074,
      "step": 232
    },
    {
      "barlow_loss": 0.17812564969062805,
      "epoch": 0.9925133689839573,
      "loss": 1.6408734321594238,
      "step": 232
    },
    {
      "barlow_loss": 0.1501082479953766,
      "epoch": 0.9925133689839573,
      "loss": 1.4088884592056274,
      "step": 232
    },
    {
      "barlow_loss": 0.15733683109283447,
      "epoch": 0.9925133689839573,
      "loss": 1.3252452611923218,
      "step": 232
    },
    {
      "barlow_loss": 0.1744495928287506,
      "epoch": 0.9967914438502674,
      "loss": 1.6487751007080078,
      "step": 233
    },
    {
      "barlow_loss": 0.16216614842414856,
      "epoch": 0.9967914438502674,
      "loss": 1.5386134386062622,
      "step": 233
    },
    {
      "barlow_loss": 0.19031141698360443,
      "epoch": 0.9967914438502674,
      "loss": 1.3002268075942993,
      "step": 233
    },
    {
      "barlow_loss": 0.1583918035030365,
      "epoch": 0.9967914438502674,
      "loss": 1.6660665273666382,
      "step": 233
    },
    {
      "barlow_loss": 0.16803504526615143,
      "epoch": 0.9967914438502674,
      "loss": 1.3519378900527954,
      "step": 233
    },
    {
      "barlow_loss": 0.1783069223165512,
      "epoch": 0.9967914438502674,
      "loss": 1.2980672121047974,
      "step": 233
    },
    {
      "barlow_loss": 0.16431361436843872,
      "epoch": 0.9967914438502674,
      "loss": 1.474358081817627,
      "step": 233
    },
    {
      "barlow_loss": 0.17729540169239044,
      "epoch": 0.9967914438502674,
      "loss": 1.20277738571167,
      "step": 233
    },
    {
      "barlow_loss": 0.17542177438735962,
      "epoch": 0.9967914438502674,
      "loss": 1.6685874462127686,
      "step": 233
    },
    {
      "barlow_loss": 0.17129358649253845,
      "epoch": 0.9967914438502674,
      "loss": 1.3128035068511963,
      "step": 233
    },
    {
      "barlow_loss": 0.17662745714187622,
      "epoch": 0.9967914438502674,
      "loss": 1.4720872640609741,
      "step": 233
    },
    {
      "barlow_loss": 0.16455203294754028,
      "epoch": 0.9967914438502674,
      "loss": 1.2989826202392578,
      "step": 233
    },
    {
      "barlow_loss": 0.16177532076835632,
      "epoch": 0.9967914438502674,
      "loss": 1.345260500907898,
      "step": 233
    },
    {
      "barlow_loss": 0.18159276247024536,
      "epoch": 0.9967914438502674,
      "loss": 1.7443525791168213,
      "step": 233
    },
    {
      "barlow_loss": 0.18973073363304138,
      "epoch": 0.9967914438502674,
      "loss": 1.3096450567245483,
      "step": 233
    },
    {
      "barlow_loss": 0.1631690263748169,
      "epoch": 0.9967914438502674,
      "loss": 1.4565736055374146,
      "step": 233
    },
    {
      "barlow_loss": 0.1702689528465271,
      "epoch": 0.9967914438502674,
      "loss": 1.4127514362335205,
      "step": 233
    },
    {
      "barlow_loss": 0.1682599037885666,
      "epoch": 0.9967914438502674,
      "loss": 1.3411723375320435,
      "step": 233
    },
    {
      "barlow_loss": 0.1664944291114807,
      "epoch": 0.9967914438502674,
      "loss": 1.6337579488754272,
      "step": 233
    },
    {
      "barlow_loss": 0.1685115098953247,
      "epoch": 0.9967914438502674,
      "loss": 1.6190648078918457,
      "step": 233
    },
    {
      "barlow_loss": 0.17192557454109192,
      "epoch": 0.9967914438502674,
      "loss": 1.3419766426086426,
      "step": 233
    },
    {
      "barlow_loss": 0.1624436229467392,
      "epoch": 0.9967914438502674,
      "loss": 1.414273977279663,
      "step": 233
    },
    {
      "barlow_loss": 0.16493189334869385,
      "epoch": 0.9967914438502674,
      "loss": 1.553446650505066,
      "step": 233
    },
    {
      "barlow_loss": 0.16925333440303802,
      "epoch": 0.9967914438502674,
      "loss": 1.7366689443588257,
      "step": 233
    },
    {
      "barlow_loss": 0.16403193771839142,
      "epoch": 0.9967914438502674,
      "loss": 1.3401660919189453,
      "step": 233
    },
    {
      "barlow_loss": 0.15886980295181274,
      "epoch": 0.9967914438502674,
      "loss": 1.4022645950317383,
      "step": 233
    },
    {
      "barlow_loss": 0.1992405205965042,
      "epoch": 0.9967914438502674,
      "loss": 1.337121844291687,
      "step": 233
    },
    {
      "barlow_loss": 0.1805429458618164,
      "epoch": 0.9967914438502674,
      "loss": 1.6673917770385742,
      "step": 233
    },
    {
      "barlow_loss": 0.16685496270656586,
      "epoch": 0.9967914438502674,
      "loss": 1.5751100778579712,
      "step": 233
    },
    {
      "barlow_loss": 0.16261327266693115,
      "epoch": 0.9967914438502674,
      "loss": 1.1781134605407715,
      "step": 233
    },
    {
      "barlow_loss": 0.16691376268863678,
      "epoch": 0.9967914438502674,
      "loss": 1.6411293745040894,
      "step": 233
    },
    {
      "barlow_loss": 0.1724374145269394,
      "epoch": 0.9967914438502674,
      "loss": 1.2124301195144653,
      "step": 233
    },
    {
      "barlow_loss": 0.18267890810966492,
      "epoch": 0.9967914438502674,
      "loss": 1.7501134872436523,
      "step": 233
    },
    {
      "barlow_loss": 0.17425470054149628,
      "epoch": 0.9967914438502674,
      "loss": 1.2801209688186646,
      "step": 233
    },
    {
      "barlow_loss": 0.17997942864894867,
      "epoch": 0.9967914438502674,
      "loss": 1.4266459941864014,
      "step": 233
    },
    {
      "barlow_loss": 0.16822141408920288,
      "epoch": 0.9967914438502674,
      "loss": 1.3219982385635376,
      "step": 233
    },
    {
      "barlow_loss": 0.17997273802757263,
      "epoch": 0.9967914438502674,
      "loss": 1.550864577293396,
      "step": 233
    },
    {
      "barlow_loss": 0.178277850151062,
      "epoch": 0.9967914438502674,
      "loss": 1.5985835790634155,
      "step": 233
    },
    {
      "barlow_loss": 0.18048317730426788,
      "epoch": 0.9967914438502674,
      "loss": 1.531161904335022,
      "step": 233
    },
    {
      "barlow_loss": 0.17169995605945587,
      "epoch": 0.9967914438502674,
      "loss": 1.54717218875885,
      "step": 233
    },
    {
      "barlow_loss": 0.16300427913665771,
      "epoch": 0.9967914438502674,
      "loss": 1.2754948139190674,
      "step": 233
    },
    {
      "barlow_loss": 0.15687322616577148,
      "epoch": 0.9967914438502674,
      "loss": 1.5237786769866943,
      "step": 233
    },
    {
      "barlow_loss": 0.18052265048027039,
      "epoch": 0.9967914438502674,
      "loss": 1.6070098876953125,
      "step": 233
    },
    {
      "barlow_loss": 0.16794292628765106,
      "epoch": 0.9967914438502674,
      "loss": 1.50627601146698,
      "step": 233
    },
    {
      "barlow_loss": 0.16492120921611786,
      "epoch": 0.9967914438502674,
      "loss": 1.3607884645462036,
      "step": 233
    },
    {
      "barlow_loss": 0.17213647067546844,
      "epoch": 0.9967914438502674,
      "loss": 1.5001503229141235,
      "step": 233
    },
    {
      "barlow_loss": 0.17984643578529358,
      "epoch": 0.9967914438502674,
      "loss": 1.2266675233840942,
      "step": 233
    },
    {
      "barlow_loss": 0.17990970611572266,
      "epoch": 0.9967914438502674,
      "loss": 1.3105361461639404,
      "step": 233
    },
    {
      "barlow_loss": 0.17379550635814667,
      "epoch": 0.9967914438502674,
      "loss": 1.3793236017227173,
      "step": 233
    },
    {
      "barlow_loss": 0.17182019352912903,
      "epoch": 0.9967914438502674,
      "loss": 1.5805115699768066,
      "step": 233
    },
    {
      "barlow_loss": 0.16919386386871338,
      "epoch": 0.9967914438502674,
      "loss": 1.208288311958313,
      "step": 233
    },
    {
      "barlow_loss": 0.17444494366645813,
      "epoch": 0.9967914438502674,
      "loss": 1.47515070438385,
      "step": 233
    },
    {
      "barlow_loss": 0.17575740814208984,
      "epoch": 0.9967914438502674,
      "loss": 1.458329677581787,
      "step": 233
    },
    {
      "barlow_loss": 0.1765114814043045,
      "epoch": 0.9967914438502674,
      "loss": 1.294682502746582,
      "step": 233
    },
    {
      "barlow_loss": 0.18103483319282532,
      "epoch": 0.9967914438502674,
      "loss": 1.6066111326217651,
      "step": 233
    },
    {
      "barlow_loss": 0.18312589824199677,
      "epoch": 0.9967914438502674,
      "loss": 1.3752139806747437,
      "step": 233
    },
    {
      "barlow_loss": 0.16956911981105804,
      "epoch": 0.9967914438502674,
      "loss": 1.433899998664856,
      "step": 233
    },
    {
      "barlow_loss": 0.17655104398727417,
      "epoch": 0.9967914438502674,
      "loss": 1.429350733757019,
      "step": 233
    },
    {
      "barlow_loss": 0.1696513444185257,
      "epoch": 0.9967914438502674,
      "loss": 1.3709527254104614,
      "step": 233
    },
    {
      "barlow_loss": 0.17395611107349396,
      "epoch": 0.9967914438502674,
      "loss": 1.4184317588806152,
      "step": 233
    },
    {
      "barlow_loss": 0.15725842118263245,
      "epoch": 0.9967914438502674,
      "loss": 1.2891217470169067,
      "step": 233
    },
    {
      "barlow_loss": 0.17583754658699036,
      "epoch": 0.9967914438502674,
      "loss": 1.4583940505981445,
      "step": 233
    },
    {
      "barlow_loss": 0.17584216594696045,
      "epoch": 0.9967914438502674,
      "loss": 1.2808194160461426,
      "step": 233
    },
    {
      "barlow_loss": 0.17918461561203003,
      "epoch": 0.9967914438502674,
      "loss": 1.4098626375198364,
      "step": 233
    },
    {
      "barlow_loss": 0.183903306722641,
      "epoch": 0.9967914438502674,
      "loss": 1.3399062156677246,
      "step": 233
    },
    {
      "barlow_loss": 0.15390022099018097,
      "epoch": 0.9967914438502674,
      "loss": 1.3944003582000732,
      "step": 233
    },
    {
      "barlow_loss": 0.16807687282562256,
      "epoch": 0.9967914438502674,
      "loss": 1.5585782527923584,
      "step": 233
    },
    {
      "barlow_loss": 0.17921724915504456,
      "epoch": 0.9967914438502674,
      "loss": 1.4132885932922363,
      "step": 233
    },
    {
      "barlow_loss": 0.17240937054157257,
      "epoch": 0.9967914438502674,
      "loss": 1.2954983711242676,
      "step": 233
    },
    {
      "barlow_loss": 0.17588236927986145,
      "epoch": 0.9967914438502674,
      "loss": 1.4417425394058228,
      "step": 233
    },
    {
      "barlow_loss": 0.18280605971813202,
      "epoch": 0.9967914438502674,
      "loss": 1.5107475519180298,
      "step": 233
    },
    {
      "barlow_loss": 0.18712429702281952,
      "epoch": 0.9967914438502674,
      "loss": 1.5258020162582397,
      "step": 233
    },
    {
      "barlow_loss": 0.16800928115844727,
      "epoch": 0.9967914438502674,
      "loss": 1.5914970636367798,
      "step": 233
    },
    {
      "barlow_loss": 0.170953169465065,
      "epoch": 0.9967914438502674,
      "loss": 1.4497257471084595,
      "step": 233
    },
    {
      "barlow_loss": 0.1711990088224411,
      "epoch": 0.9967914438502674,
      "loss": 1.5367767810821533,
      "step": 233
    },
    {
      "barlow_loss": 0.1864774376153946,
      "epoch": 0.9967914438502674,
      "loss": 1.2303317785263062,
      "step": 233
    },
    {
      "barlow_loss": 0.17433834075927734,
      "epoch": 0.9967914438502674,
      "loss": 1.6004160642623901,
      "step": 233
    },
    {
      "barlow_loss": 0.1839454174041748,
      "epoch": 0.9967914438502674,
      "loss": 1.4057079553604126,
      "step": 233
    },
    {
      "barlow_loss": 0.15940994024276733,
      "epoch": 0.9967914438502674,
      "loss": 1.3533669710159302,
      "step": 233
    },
    {
      "barlow_loss": 0.1767035573720932,
      "epoch": 0.9967914438502674,
      "loss": 1.5808392763137817,
      "step": 233
    },
    {
      "barlow_loss": 0.15765081346035004,
      "epoch": 0.9967914438502674,
      "loss": 1.2894009351730347,
      "step": 233
    },
    {
      "barlow_loss": 0.14608131349086761,
      "epoch": 0.9967914438502674,
      "loss": 1.2159514427185059,
      "step": 233
    },
    {
      "barlow_loss": 0.19498182833194733,
      "epoch": 0.9967914438502674,
      "loss": 1.5404763221740723,
      "step": 233
    },
    {
      "barlow_loss": 0.16051943600177765,
      "epoch": 0.9967914438502674,
      "loss": 1.5830086469650269,
      "step": 233
    },
    {
      "barlow_loss": 0.1772376447916031,
      "epoch": 0.9967914438502674,
      "loss": 1.4173378944396973,
      "step": 233
    },
    {
      "barlow_loss": 0.1871369332075119,
      "epoch": 0.9967914438502674,
      "loss": 1.5411045551300049,
      "step": 233
    },
    {
      "barlow_loss": 0.15700551867485046,
      "epoch": 0.9967914438502674,
      "loss": 1.2518614530563354,
      "step": 233
    },
    {
      "barlow_loss": 0.17044830322265625,
      "epoch": 0.9967914438502674,
      "loss": 1.5408622026443481,
      "step": 233
    },
    {
      "barlow_loss": 0.16776823997497559,
      "epoch": 0.9967914438502674,
      "loss": 1.3073240518569946,
      "step": 233
    },
    {
      "barlow_loss": 0.19362914562225342,
      "epoch": 0.9967914438502674,
      "loss": 2.0632190704345703,
      "step": 233
    },
    {
      "barlow_loss": 0.17230495810508728,
      "epoch": 0.9967914438502674,
      "loss": 1.4623693227767944,
      "step": 233
    },
    {
      "barlow_loss": 0.16420236229896545,
      "epoch": 0.9967914438502674,
      "loss": 1.335249423980713,
      "step": 233
    },
    {
      "barlow_loss": 0.17606526613235474,
      "epoch": 0.9967914438502674,
      "loss": 1.544936180114746,
      "step": 233
    },
    {
      "barlow_loss": 0.17471450567245483,
      "epoch": 0.9967914438502674,
      "loss": 1.7059926986694336,
      "step": 233
    },
    {
      "barlow_loss": 0.18731194734573364,
      "epoch": 0.9967914438502674,
      "loss": 1.4482386112213135,
      "step": 233
    },
    {
      "barlow_loss": 0.1824161410331726,
      "epoch": 0.9967914438502674,
      "loss": 1.3129987716674805,
      "step": 233
    },
    {
      "barlow_loss": 0.1731521338224411,
      "epoch": 0.9967914438502674,
      "loss": 1.371876835823059,
      "step": 233
    },
    {
      "barlow_loss": 0.18543265759944916,
      "epoch": 0.9967914438502674,
      "loss": 1.806872010231018,
      "step": 233
    },
    {
      "barlow_loss": 0.1771698147058487,
      "epoch": 0.9967914438502674,
      "loss": 1.5745762586593628,
      "step": 233
    },
    {
      "barlow_loss": 0.18546685576438904,
      "epoch": 0.9967914438502674,
      "loss": 1.2853010892868042,
      "step": 233
    },
    {
      "barlow_loss": 0.1590326875448227,
      "epoch": 0.9967914438502674,
      "loss": 1.6212573051452637,
      "step": 233
    },
    {
      "barlow_loss": 0.1851566731929779,
      "epoch": 0.9967914438502674,
      "loss": 1.5512806177139282,
      "step": 233
    },
    {
      "barlow_loss": 0.1629382073879242,
      "epoch": 0.9967914438502674,
      "loss": 1.578324317932129,
      "step": 233
    },
    {
      "barlow_loss": 0.1689993143081665,
      "epoch": 0.9967914438502674,
      "loss": 1.6629778146743774,
      "step": 233
    },
    {
      "barlow_loss": 0.16942401230335236,
      "epoch": 0.9967914438502674,
      "loss": 1.2752275466918945,
      "step": 233
    },
    {
      "barlow_loss": 0.15602947771549225,
      "epoch": 0.9967914438502674,
      "loss": 1.272578477859497,
      "step": 233
    },
    {
      "barlow_loss": 0.17961092293262482,
      "epoch": 0.9967914438502674,
      "loss": 1.251308798789978,
      "step": 233
    },
    {
      "barlow_loss": 0.17410261929035187,
      "epoch": 0.9967914438502674,
      "loss": 1.4376500844955444,
      "step": 233
    },
    {
      "barlow_loss": 0.161423459649086,
      "epoch": 0.9967914438502674,
      "loss": 1.2733081579208374,
      "step": 233
    },
    {
      "barlow_loss": 0.15011069178581238,
      "epoch": 0.9967914438502674,
      "loss": 1.441217303276062,
      "step": 233
    },
    {
      "barlow_loss": 0.18931333720684052,
      "epoch": 0.9967914438502674,
      "loss": 1.6695293188095093,
      "step": 233
    },
    {
      "barlow_loss": 0.16837556660175323,
      "epoch": 0.9967914438502674,
      "loss": 1.4879839420318604,
      "step": 233
    },
    {
      "barlow_loss": 0.1694118082523346,
      "epoch": 0.9967914438502674,
      "loss": 1.39909029006958,
      "step": 233
    },
    {
      "barlow_loss": 0.16777589917182922,
      "epoch": 0.9967914438502674,
      "loss": 1.4272327423095703,
      "step": 233
    },
    {
      "barlow_loss": 0.1765153706073761,
      "epoch": 0.9967914438502674,
      "loss": 1.6197396516799927,
      "step": 233
    },
    {
      "barlow_loss": 0.17550067603588104,
      "epoch": 0.9967914438502674,
      "loss": 1.6691029071807861,
      "step": 233
    },
    {
      "barlow_loss": 0.17357154190540314,
      "epoch": 0.9967914438502674,
      "loss": 1.2675268650054932,
      "step": 233
    },
    {
      "barlow_loss": 0.18234986066818237,
      "epoch": 0.9967914438502674,
      "loss": 1.3933649063110352,
      "step": 233
    },
    {
      "barlow_loss": 0.16921395063400269,
      "epoch": 0.9967914438502674,
      "loss": 1.7127690315246582,
      "step": 233
    },
    {
      "barlow_loss": 0.1604708731174469,
      "epoch": 0.9967914438502674,
      "loss": 1.5052109956741333,
      "step": 233
    },
    {
      "barlow_loss": 0.16609135270118713,
      "epoch": 0.9967914438502674,
      "loss": 1.6326314210891724,
      "step": 233
    },
    {
      "barlow_loss": 0.18402469158172607,
      "epoch": 0.9967914438502674,
      "loss": 1.2235065698623657,
      "step": 233
    },
    {
      "barlow_loss": 0.14488932490348816,
      "epoch": 0.9967914438502674,
      "loss": 1.2382886409759521,
      "step": 233
    },
    {
      "barlow_loss": 0.17745134234428406,
      "epoch": 0.9967914438502674,
      "loss": 1.5365809202194214,
      "step": 233
    },
    {
      "barlow_loss": 0.16297952830791473,
      "epoch": 0.9967914438502674,
      "loss": 1.8973503112792969,
      "step": 233
    },
    {
      "barlow_loss": 0.18299917876720428,
      "epoch": 0.9967914438502674,
      "loss": 1.5844770669937134,
      "step": 233
    },
    {
      "barlow_loss": 0.176910862326622,
      "epoch": 0.9967914438502674,
      "loss": 1.3330131769180298,
      "step": 233
    },
    {
      "barlow_loss": 0.17543362081050873,
      "epoch": 0.9967914438502674,
      "loss": 1.5614863634109497,
      "step": 233
    },
    {
      "barlow_loss": 0.1729087084531784,
      "epoch": 0.9967914438502674,
      "loss": 1.6092995405197144,
      "step": 233
    },
    {
      "barlow_loss": 0.18374177813529968,
      "epoch": 0.9967914438502674,
      "loss": 1.6041803359985352,
      "step": 233
    },
    {
      "barlow_loss": 0.1635703295469284,
      "epoch": 0.9967914438502674,
      "loss": 1.5422964096069336,
      "step": 233
    },
    {
      "barlow_loss": 0.17351067066192627,
      "epoch": 0.9967914438502674,
      "loss": 1.6022636890411377,
      "step": 233
    },
    {
      "barlow_loss": 0.19196327030658722,
      "epoch": 0.9967914438502674,
      "loss": 1.5680186748504639,
      "step": 233
    },
    {
      "barlow_loss": 0.18775585293769836,
      "epoch": 0.9967914438502674,
      "loss": 1.6567407846450806,
      "step": 233
    },
    {
      "barlow_loss": 0.14934490621089935,
      "epoch": 0.9967914438502674,
      "loss": 1.3923343420028687,
      "step": 233
    },
    {
      "barlow_loss": 0.16949795186519623,
      "epoch": 0.9967914438502674,
      "loss": 1.4721781015396118,
      "step": 233
    },
    {
      "barlow_loss": 0.18004347383975983,
      "epoch": 0.9967914438502674,
      "loss": 1.6751744747161865,
      "step": 233
    },
    {
      "barlow_loss": 0.18269629776477814,
      "epoch": 0.9967914438502674,
      "loss": 1.2924340963363647,
      "step": 233
    },
    {
      "barlow_loss": 0.17708396911621094,
      "epoch": 0.9967914438502674,
      "loss": 1.535682201385498,
      "step": 233
    },
    {
      "barlow_loss": 0.17945007979869843,
      "epoch": 0.9967914438502674,
      "loss": 1.4205718040466309,
      "step": 233
    },
    {
      "barlow_loss": 0.17245236039161682,
      "epoch": 0.9967914438502674,
      "loss": 1.1978511810302734,
      "step": 233
    },
    {
      "barlow_loss": 0.1690610945224762,
      "epoch": 0.9967914438502674,
      "loss": 1.2601876258850098,
      "step": 233
    },
    {
      "barlow_loss": 0.17526763677597046,
      "epoch": 0.9967914438502674,
      "loss": 1.314894676208496,
      "step": 233
    },
    {
      "barlow_loss": 0.16310007870197296,
      "epoch": 0.9967914438502674,
      "loss": 1.5219074487686157,
      "step": 233
    },
    {
      "barlow_loss": 0.17293287813663483,
      "epoch": 0.9967914438502674,
      "loss": 1.3271183967590332,
      "step": 233
    },
    {
      "barlow_loss": 0.1630888432264328,
      "epoch": 0.9967914438502674,
      "loss": 1.5589710474014282,
      "step": 233
    },
    {
      "barlow_loss": 0.16386264562606812,
      "epoch": 0.9967914438502674,
      "loss": 1.489474892616272,
      "step": 233
    },
    {
      "barlow_loss": 0.17821510136127472,
      "epoch": 0.9967914438502674,
      "loss": 1.3877393007278442,
      "step": 233
    },
    {
      "barlow_loss": 0.1485891342163086,
      "epoch": 0.9967914438502674,
      "loss": 1.2722115516662598,
      "step": 233
    },
    {
      "barlow_loss": 0.17364828288555145,
      "epoch": 0.9967914438502674,
      "loss": 1.6913437843322754,
      "step": 233
    },
    {
      "barlow_loss": 0.1689678430557251,
      "epoch": 0.9967914438502674,
      "loss": 1.5510236024856567,
      "step": 233
    },
    {
      "barlow_loss": 0.1850588470697403,
      "epoch": 0.9967914438502674,
      "loss": 1.484261393547058,
      "step": 233
    },
    {
      "barlow_loss": 0.17230205237865448,
      "epoch": 0.9967914438502674,
      "loss": 1.6250742673873901,
      "step": 233
    },
    {
      "barlow_loss": 0.17282651364803314,
      "epoch": 0.9967914438502674,
      "loss": 1.365293025970459,
      "step": 233
    },
    {
      "barlow_loss": 0.17262350022792816,
      "epoch": 0.9967914438502674,
      "loss": 1.4311109781265259,
      "step": 233
    },
    {
      "barlow_loss": 0.1669156700372696,
      "epoch": 0.9967914438502674,
      "loss": 1.411784291267395,
      "step": 233
    },
    {
      "barlow_loss": 0.18508218228816986,
      "epoch": 0.9967914438502674,
      "loss": 1.5093872547149658,
      "step": 233
    },
    {
      "barlow_loss": 0.16549986600875854,
      "epoch": 0.9967914438502674,
      "loss": 1.4332908391952515,
      "step": 233
    },
    {
      "barlow_loss": 0.18072360754013062,
      "epoch": 0.9967914438502674,
      "loss": 1.3231018781661987,
      "step": 233
    },
    {
      "barlow_loss": 0.1678248941898346,
      "epoch": 0.9967914438502674,
      "loss": 1.2733216285705566,
      "step": 233
    },
    {
      "barlow_loss": 0.17073848843574524,
      "epoch": 0.9967914438502674,
      "loss": 1.2254647016525269,
      "step": 233
    },
    {
      "barlow_loss": 0.17525723576545715,
      "epoch": 0.9967914438502674,
      "loss": 1.3842639923095703,
      "step": 233
    },
    {
      "barlow_loss": 0.18474693596363068,
      "epoch": 0.9967914438502674,
      "loss": 1.3099582195281982,
      "step": 233
    },
    {
      "barlow_loss": 0.1552879363298416,
      "epoch": 0.9967914438502674,
      "loss": 1.5592097043991089,
      "step": 233
    },
    {
      "barlow_loss": 0.18723788857460022,
      "epoch": 0.9967914438502674,
      "loss": 1.6762415170669556,
      "step": 233
    },
    {
      "barlow_loss": 0.15721088647842407,
      "epoch": 0.9967914438502674,
      "loss": 1.3418530225753784,
      "step": 233
    },
    {
      "barlow_loss": 0.19435562193393707,
      "epoch": 0.9967914438502674,
      "loss": 1.4109578132629395,
      "step": 233
    },
    {
      "barlow_loss": 0.17579911649227142,
      "epoch": 0.9967914438502674,
      "loss": 1.2608470916748047,
      "step": 233
    },
    {
      "barlow_loss": 0.18219992518424988,
      "epoch": 0.9967914438502674,
      "loss": 1.5304092168807983,
      "step": 233
    },
    {
      "barlow_loss": 0.1606438010931015,
      "epoch": 0.9967914438502674,
      "loss": 1.3843064308166504,
      "step": 233
    },
    {
      "barlow_loss": 0.1641792356967926,
      "epoch": 0.9967914438502674,
      "loss": 1.5527997016906738,
      "step": 233
    },
    {
      "barlow_loss": 0.15028463304042816,
      "epoch": 0.9967914438502674,
      "loss": 1.3133946657180786,
      "step": 233
    },
    {
      "barlow_loss": 0.1578560620546341,
      "epoch": 0.9967914438502674,
      "loss": 1.5332255363464355,
      "step": 233
    },
    {
      "barlow_loss": 0.1727428138256073,
      "epoch": 0.9967914438502674,
      "loss": 1.4314439296722412,
      "step": 233
    },
    {
      "barlow_loss": 0.18857938051223755,
      "epoch": 0.9967914438502674,
      "loss": 1.494422197341919,
      "step": 233
    },
    {
      "barlow_loss": 0.1684594601392746,
      "epoch": 0.9967914438502674,
      "loss": 1.5391185283660889,
      "step": 233
    },
    {
      "barlow_loss": 0.14511866867542267,
      "epoch": 0.9967914438502674,
      "loss": 1.109243631362915,
      "step": 233
    },
    {
      "barlow_loss": 0.15591372549533844,
      "epoch": 0.9967914438502674,
      "loss": 1.2799501419067383,
      "step": 233
    },
    {
      "barlow_loss": 0.19063526391983032,
      "epoch": 0.9967914438502674,
      "loss": 1.1350685358047485,
      "step": 233
    },
    {
      "barlow_loss": 0.15889012813568115,
      "epoch": 0.9967914438502674,
      "loss": 1.4462515115737915,
      "step": 233
    },
    {
      "barlow_loss": 0.1851123571395874,
      "epoch": 0.9967914438502674,
      "loss": 1.513041377067566,
      "step": 233
    },
    {
      "barlow_loss": 0.15958134829998016,
      "epoch": 0.9967914438502674,
      "loss": 1.3439620733261108,
      "step": 233
    },
    {
      "barlow_loss": 0.16761767864227295,
      "epoch": 0.9967914438502674,
      "loss": 1.2480823993682861,
      "step": 233
    },
    {
      "barlow_loss": 0.177927166223526,
      "epoch": 0.9967914438502674,
      "loss": 1.62682044506073,
      "step": 233
    },
    {
      "barlow_loss": 0.174891397356987,
      "epoch": 0.9967914438502674,
      "loss": 1.3774785995483398,
      "step": 233
    },
    {
      "barlow_loss": 0.17722608149051666,
      "epoch": 0.9967914438502674,
      "loss": 1.11354660987854,
      "step": 233
    },
    {
      "barlow_loss": 0.18019750714302063,
      "epoch": 0.9967914438502674,
      "loss": 1.5463610887527466,
      "step": 233
    },
    {
      "barlow_loss": 0.15984080731868744,
      "epoch": 0.9967914438502674,
      "loss": 1.387148141860962,
      "step": 233
    },
    {
      "barlow_loss": 0.17473863065242767,
      "epoch": 0.9967914438502674,
      "loss": 1.5062798261642456,
      "step": 233
    },
    {
      "barlow_loss": 0.15714825689792633,
      "epoch": 0.9967914438502674,
      "loss": 1.7308576107025146,
      "step": 233
    },
    {
      "barlow_loss": 0.16608363389968872,
      "epoch": 0.9967914438502674,
      "loss": 1.3490492105484009,
      "step": 233
    },
    {
      "barlow_loss": 0.19109228253364563,
      "epoch": 0.9967914438502674,
      "loss": 1.4339643716812134,
      "step": 233
    },
    {
      "barlow_loss": 0.18042540550231934,
      "epoch": 0.9967914438502674,
      "loss": 1.4017207622528076,
      "step": 233
    },
    {
      "barlow_loss": 0.16342619061470032,
      "epoch": 0.9967914438502674,
      "loss": 1.2950985431671143,
      "step": 233
    },
    {
      "barlow_loss": 0.1750188171863556,
      "epoch": 0.9967914438502674,
      "loss": 1.2614290714263916,
      "step": 233
    },
    {
      "barlow_loss": 0.16214510798454285,
      "epoch": 0.9967914438502674,
      "loss": 1.683698296546936,
      "step": 233
    },
    {
      "barlow_loss": 0.17735834419727325,
      "epoch": 0.9967914438502674,
      "loss": 1.6723804473876953,
      "step": 233
    },
    {
      "barlow_loss": 0.1821528971195221,
      "epoch": 0.9967914438502674,
      "loss": 1.391794204711914,
      "step": 233
    },
    {
      "barlow_loss": 0.18147368729114532,
      "epoch": 0.9967914438502674,
      "loss": 1.6031550168991089,
      "step": 233
    },
    {
      "barlow_loss": 0.17553271353244781,
      "epoch": 0.9967914438502674,
      "loss": 1.5345969200134277,
      "step": 233
    },
    {
      "barlow_loss": 0.17541605234146118,
      "epoch": 0.9967914438502674,
      "loss": 1.4910459518432617,
      "step": 233
    },
    {
      "barlow_loss": 0.17353083193302155,
      "epoch": 0.9967914438502674,
      "loss": 1.5747616291046143,
      "step": 233
    },
    {
      "barlow_loss": 0.1845594197511673,
      "epoch": 0.9967914438502674,
      "loss": 1.4714069366455078,
      "step": 233
    },
    {
      "barlow_loss": 0.1771775484085083,
      "epoch": 0.9967914438502674,
      "loss": 1.822436809539795,
      "step": 233
    },
    {
      "barlow_loss": 0.1799197494983673,
      "epoch": 0.9967914438502674,
      "loss": 1.5170000791549683,
      "step": 233
    },
    {
      "barlow_loss": 0.17993059754371643,
      "epoch": 0.9967914438502674,
      "loss": 1.5851950645446777,
      "step": 233
    },
    {
      "barlow_loss": 0.15684114396572113,
      "epoch": 0.9967914438502674,
      "loss": 1.5858479738235474,
      "step": 233
    },
    {
      "barlow_loss": 0.17552244663238525,
      "epoch": 0.9967914438502674,
      "loss": 1.30880606174469,
      "step": 233
    },
    {
      "barlow_loss": 0.18026746809482574,
      "epoch": 0.9967914438502674,
      "loss": 2.6038079261779785,
      "step": 233
    },
    {
      "barlow_loss": 0.1753031462430954,
      "epoch": 0.9967914438502674,
      "loss": 1.6718052625656128,
      "step": 233
    },
    {
      "barlow_loss": 0.16617093980312347,
      "epoch": 0.9967914438502674,
      "loss": 1.5067998170852661,
      "step": 233
    },
    {
      "barlow_loss": 0.18302252888679504,
      "epoch": 0.9967914438502674,
      "loss": 1.7378177642822266,
      "step": 233
    },
    {
      "barlow_loss": 0.19009914994239807,
      "epoch": 0.9967914438502674,
      "loss": 1.4280413389205933,
      "step": 233
    },
    {
      "barlow_loss": 0.18113000690937042,
      "epoch": 0.9967914438502674,
      "loss": 1.522109031677246,
      "step": 233
    },
    {
      "barlow_loss": 0.1762990653514862,
      "epoch": 0.9967914438502674,
      "loss": 1.3516948223114014,
      "step": 233
    },
    {
      "barlow_loss": 0.19013765454292297,
      "epoch": 0.9967914438502674,
      "loss": 1.2518973350524902,
      "step": 233
    },
    {
      "barlow_loss": 0.1657903492450714,
      "epoch": 0.9967914438502674,
      "loss": 1.615709662437439,
      "step": 233
    },
    {
      "barlow_loss": 0.17047226428985596,
      "epoch": 0.9967914438502674,
      "loss": 1.587114930152893,
      "step": 233
    },
    {
      "barlow_loss": 0.17627640068531036,
      "epoch": 0.9967914438502674,
      "loss": 1.350862979888916,
      "step": 233
    },
    {
      "barlow_loss": 0.1820695400238037,
      "epoch": 0.9967914438502674,
      "loss": 1.212502121925354,
      "step": 233
    },
    {
      "barlow_loss": 0.1657891571521759,
      "epoch": 0.9967914438502674,
      "loss": 1.3951616287231445,
      "step": 233
    },
    {
      "barlow_loss": 0.17465800046920776,
      "epoch": 0.9967914438502674,
      "loss": 1.2918623685836792,
      "step": 233
    },
    {
      "barlow_loss": 0.17533935606479645,
      "epoch": 0.9967914438502674,
      "loss": 1.5088502168655396,
      "step": 233
    },
    {
      "barlow_loss": 0.17161332070827484,
      "epoch": 0.9967914438502674,
      "loss": 1.4670518636703491,
      "step": 233
    },
    {
      "barlow_loss": 0.1642177551984787,
      "epoch": 0.9967914438502674,
      "loss": 1.5745898485183716,
      "step": 233
    },
    {
      "barlow_loss": 0.17811724543571472,
      "epoch": 0.9967914438502674,
      "loss": 1.5273141860961914,
      "step": 233
    },
    {
      "barlow_loss": 0.1781366765499115,
      "epoch": 0.9967914438502674,
      "loss": 1.6098639965057373,
      "step": 233
    },
    {
      "barlow_loss": 0.1560046672821045,
      "epoch": 0.9967914438502674,
      "loss": 1.452190637588501,
      "step": 233
    },
    {
      "barlow_loss": 0.1669861078262329,
      "epoch": 0.9967914438502674,
      "loss": 1.7144986391067505,
      "step": 233
    },
    {
      "barlow_loss": 0.180605947971344,
      "epoch": 0.9967914438502674,
      "loss": 1.3686622381210327,
      "step": 233
    },
    {
      "barlow_loss": 0.16481907665729523,
      "epoch": 0.9967914438502674,
      "loss": 1.5692778825759888,
      "step": 233
    },
    {
      "barlow_loss": 0.17448467016220093,
      "epoch": 0.9967914438502674,
      "loss": 1.601749300956726,
      "step": 233
    },
    {
      "barlow_loss": 0.18300336599349976,
      "epoch": 0.9967914438502674,
      "loss": 1.4805563688278198,
      "step": 233
    },
    {
      "barlow_loss": 0.18373052775859833,
      "epoch": 0.9967914438502674,
      "loss": 1.277369737625122,
      "step": 233
    },
    {
      "barlow_loss": 0.15214937925338745,
      "epoch": 0.9967914438502674,
      "loss": 1.892348289489746,
      "step": 233
    },
    {
      "barlow_loss": 0.17280784249305725,
      "epoch": 0.9967914438502674,
      "loss": 1.636748194694519,
      "step": 233
    },
    {
      "barlow_loss": 0.18345977365970612,
      "epoch": 0.9967914438502674,
      "loss": 1.3957403898239136,
      "step": 233
    },
    {
      "barlow_loss": 0.18642692267894745,
      "epoch": 0.9967914438502674,
      "loss": 1.5704946517944336,
      "step": 233
    },
    {
      "barlow_loss": 0.17997892200946808,
      "epoch": 0.9967914438502674,
      "loss": 1.635587215423584,
      "step": 233
    },
    {
      "barlow_loss": 0.17112919688224792,
      "epoch": 0.9967914438502674,
      "loss": 1.6585173606872559,
      "step": 233
    },
    {
      "barlow_loss": 0.1765541136264801,
      "epoch": 0.9967914438502674,
      "loss": 1.512661337852478,
      "step": 233
    },
    {
      "barlow_loss": 0.1737109273672104,
      "epoch": 0.9967914438502674,
      "loss": 1.41684091091156,
      "step": 233
    },
    {
      "barlow_loss": 0.15764054656028748,
      "epoch": 0.9967914438502674,
      "loss": 1.2801209688186646,
      "step": 233
    },
    {
      "barlow_loss": 0.1812153458595276,
      "epoch": 0.9967914438502674,
      "loss": 1.4292066097259521,
      "step": 233
    },
    {
      "barlow_loss": 0.17410799860954285,
      "epoch": 0.9967914438502674,
      "loss": 1.6749869585037231,
      "step": 233
    },
    {
      "barlow_loss": 0.1715717315673828,
      "epoch": 0.9967914438502674,
      "loss": 1.4949100017547607,
      "step": 233
    },
    {
      "barlow_loss": 0.17140765488147736,
      "epoch": 0.9967914438502674,
      "loss": 1.562168002128601,
      "step": 233
    },
    {
      "barlow_loss": 0.153492271900177,
      "epoch": 0.9967914438502674,
      "loss": 1.6059986352920532,
      "step": 233
    },
    {
      "barlow_loss": 0.14893873035907745,
      "epoch": 0.9967914438502674,
      "loss": 1.3933396339416504,
      "step": 233
    },
    {
      "barlow_loss": 0.17389674484729767,
      "epoch": 0.9967914438502674,
      "loss": 1.3374009132385254,
      "step": 233
    },
    {
      "barlow_loss": 0.16245347261428833,
      "epoch": 0.9967914438502674,
      "loss": 1.5151699781417847,
      "step": 233
    },
    {
      "barlow_loss": 0.17842094600200653,
      "epoch": 0.9967914438502674,
      "loss": 1.5466175079345703,
      "step": 233
    },
    {
      "barlow_loss": 0.1850438266992569,
      "epoch": 0.9967914438502674,
      "loss": 1.5063966512680054,
      "step": 233
    },
    {
      "barlow_loss": 0.17110338807106018,
      "epoch": 0.9967914438502674,
      "loss": 1.310268521308899,
      "step": 233
    },
    {
      "barlow_loss": 0.17997749149799347,
      "epoch": 0.9967914438502674,
      "loss": 1.4666461944580078,
      "step": 233
    },
    {
      "barlow_loss": 0.18487544357776642,
      "epoch": 0.9967914438502674,
      "loss": 1.3203952312469482,
      "step": 233
    },
    {
      "barlow_loss": 0.16496236622333527,
      "epoch": 0.9967914438502674,
      "loss": 1.5024511814117432,
      "step": 233
    },
    {
      "barlow_loss": 0.14907705783843994,
      "epoch": 0.9967914438502674,
      "loss": 1.3630682229995728,
      "step": 233
    },
    {
      "barlow_loss": 0.18344657123088837,
      "epoch": 0.9967914438502674,
      "loss": 1.2694206237792969,
      "step": 233
    },
    {
      "barlow_loss": 0.17467644810676575,
      "epoch": 0.9967914438502674,
      "loss": 1.4173287153244019,
      "step": 233
    },
    {
      "barlow_loss": 0.17140381038188934,
      "epoch": 0.9967914438502674,
      "loss": 1.4000447988510132,
      "step": 233
    },
    {
      "barlow_loss": 0.1890028864145279,
      "epoch": 0.9967914438502674,
      "loss": 1.4839859008789062,
      "step": 233
    },
    {
      "barlow_loss": 0.1687573492527008,
      "epoch": 0.9967914438502674,
      "loss": 1.3949345350265503,
      "step": 233
    },
    {
      "barlow_loss": 0.16317419707775116,
      "epoch": 0.9967914438502674,
      "loss": 1.2332794666290283,
      "step": 233
    },
    {
      "barlow_loss": 0.16865991055965424,
      "epoch": 0.9967914438502674,
      "loss": 1.3286309242248535,
      "step": 233
    },
    {
      "barlow_loss": 0.1821410208940506,
      "epoch": 0.9967914438502674,
      "loss": 1.485621690750122,
      "step": 233
    },
    {
      "barlow_loss": 0.16974776983261108,
      "epoch": 0.9967914438502674,
      "loss": 1.5366673469543457,
      "step": 233
    },
    {
      "barlow_loss": 0.1860034018754959,
      "epoch": 0.9967914438502674,
      "loss": 1.3456661701202393,
      "step": 233
    },
    {
      "barlow_loss": 0.18652485311031342,
      "epoch": 0.9967914438502674,
      "loss": 1.3799934387207031,
      "step": 233
    },
    {
      "barlow_loss": 0.1711248755455017,
      "epoch": 0.9967914438502674,
      "loss": 1.365430474281311,
      "step": 233
    },
    {
      "barlow_loss": 0.16084955632686615,
      "epoch": 0.9967914438502674,
      "loss": 1.3441206216812134,
      "step": 233
    },
    {
      "barlow_loss": 0.1839408427476883,
      "epoch": 0.9967914438502674,
      "loss": 1.4155402183532715,
      "step": 233
    },
    {
      "barlow_loss": 0.16640247404575348,
      "epoch": 0.9967914438502674,
      "loss": 1.299885630607605,
      "step": 233
    },
    {
      "barlow_loss": 0.1829802691936493,
      "epoch": 0.9967914438502674,
      "loss": 1.2320903539657593,
      "step": 233
    },
    {
      "barlow_loss": 0.15897944569587708,
      "epoch": 0.9967914438502674,
      "loss": 1.5300750732421875,
      "step": 233
    },
    {
      "barlow_loss": 0.1655956506729126,
      "epoch": 0.9967914438502674,
      "loss": 1.331110954284668,
      "step": 233
    },
    {
      "barlow_loss": 0.17380459606647491,
      "epoch": 0.9967914438502674,
      "loss": 1.5794633626937866,
      "step": 233
    },
    {
      "barlow_loss": 0.18264240026474,
      "epoch": 0.9967914438502674,
      "loss": 1.3249473571777344,
      "step": 233
    },
    {
      "barlow_loss": 0.17608113586902618,
      "epoch": 0.9967914438502674,
      "loss": 1.211821436882019,
      "step": 233
    },
    {
      "barlow_loss": 0.16081547737121582,
      "epoch": 0.9967914438502674,
      "loss": 1.4171857833862305,
      "step": 233
    },
    {
      "barlow_loss": 0.1860131323337555,
      "epoch": 0.9967914438502674,
      "loss": 1.5425984859466553,
      "step": 233
    },
    {
      "barlow_loss": 0.1711762547492981,
      "epoch": 0.9967914438502674,
      "loss": 1.7892447710037231,
      "step": 233
    },
    {
      "barlow_loss": 0.17785605788230896,
      "epoch": 0.9967914438502674,
      "loss": 1.4911527633666992,
      "step": 233
    },
    {
      "barlow_loss": 0.17961212992668152,
      "epoch": 0.9967914438502674,
      "loss": 1.5300110578536987,
      "step": 233
    },
    {
      "barlow_loss": 0.17599089443683624,
      "epoch": 0.9967914438502674,
      "loss": 1.7588436603546143,
      "step": 233
    },
    {
      "barlow_loss": 0.17402225732803345,
      "epoch": 0.9967914438502674,
      "loss": 1.6248211860656738,
      "step": 233
    },
    {
      "barlow_loss": 0.17455989122390747,
      "epoch": 0.9967914438502674,
      "loss": 1.3141627311706543,
      "step": 233
    },
    {
      "barlow_loss": 0.17908883094787598,
      "epoch": 0.9967914438502674,
      "loss": 1.6230871677398682,
      "step": 233
    },
    {
      "barlow_loss": 0.16762618720531464,
      "epoch": 0.9967914438502674,
      "loss": 1.4969947338104248,
      "step": 233
    },
    {
      "barlow_loss": 0.17124204337596893,
      "epoch": 0.9967914438502674,
      "loss": 1.4379671812057495,
      "step": 233
    },
    {
      "barlow_loss": 0.18226301670074463,
      "epoch": 0.9967914438502674,
      "loss": 1.599489688873291,
      "step": 233
    },
    {
      "barlow_loss": 0.17779281735420227,
      "epoch": 0.9967914438502674,
      "loss": 1.5539926290512085,
      "step": 233
    },
    {
      "barlow_loss": 0.16088129580020905,
      "epoch": 0.9967914438502674,
      "loss": 1.2163482904434204,
      "step": 233
    },
    {
      "barlow_loss": 0.17300088703632355,
      "epoch": 0.9967914438502674,
      "loss": 1.2762936353683472,
      "step": 233
    },
    {
      "barlow_loss": 0.19042660295963287,
      "epoch": 0.9967914438502674,
      "loss": 1.485611081123352,
      "step": 233
    },
    {
      "barlow_loss": 0.184158593416214,
      "epoch": 0.9967914438502674,
      "loss": 1.4887059926986694,
      "step": 233
    },
    {
      "barlow_loss": 0.17058581113815308,
      "epoch": 0.9967914438502674,
      "loss": 1.6366432905197144,
      "step": 233
    },
    {
      "barlow_loss": 0.1817641258239746,
      "epoch": 0.9967914438502674,
      "loss": 1.4175605773925781,
      "step": 233
    },
    {
      "barlow_loss": 0.16929545998573303,
      "epoch": 0.9967914438502674,
      "loss": 1.315251111984253,
      "step": 233
    },
    {
      "barlow_loss": 0.1744055449962616,
      "epoch": 0.9967914438502674,
      "loss": 1.4230936765670776,
      "step": 233
    },
    {
      "barlow_loss": 0.17759090662002563,
      "epoch": 0.9967914438502674,
      "loss": 1.2454017400741577,
      "step": 233
    },
    {
      "barlow_loss": 0.18395954370498657,
      "epoch": 0.9967914438502674,
      "loss": 1.4904412031173706,
      "step": 233
    },
    {
      "barlow_loss": 0.19979329407215118,
      "epoch": 0.9967914438502674,
      "loss": 1.6103143692016602,
      "step": 233
    },
    {
      "barlow_loss": 0.17074477672576904,
      "epoch": 0.9967914438502674,
      "loss": 1.387979507446289,
      "step": 233
    },
    {
      "barlow_loss": 0.16851140558719635,
      "epoch": 0.9967914438502674,
      "loss": 1.5849100351333618,
      "step": 233
    },
    {
      "barlow_loss": 0.17102137207984924,
      "epoch": 0.9967914438502674,
      "loss": 1.4492087364196777,
      "step": 233
    },
    {
      "barlow_loss": 0.1807238608598709,
      "epoch": 0.9967914438502674,
      "loss": 1.501302719116211,
      "step": 233
    },
    {
      "barlow_loss": 0.1793709099292755,
      "epoch": 0.9967914438502674,
      "loss": 1.3254088163375854,
      "step": 233
    },
    {
      "barlow_loss": 0.17315205931663513,
      "epoch": 0.9967914438502674,
      "loss": 1.493369221687317,
      "step": 233
    },
    {
      "barlow_loss": 0.16731354594230652,
      "epoch": 0.9967914438502674,
      "loss": 1.4063252210617065,
      "step": 233
    },
    {
      "barlow_loss": 0.1622677445411682,
      "epoch": 0.9967914438502674,
      "loss": 1.376042127609253,
      "step": 233
    },
    {
      "barlow_loss": 0.1714337170124054,
      "epoch": 0.9967914438502674,
      "loss": 1.5637527704238892,
      "step": 233
    },
    {
      "barlow_loss": 0.16601602733135223,
      "epoch": 0.9967914438502674,
      "loss": 1.5414291620254517,
      "step": 233
    },
    {
      "barlow_loss": 0.18308153748512268,
      "epoch": 0.9967914438502674,
      "loss": 1.3341639041900635,
      "step": 233
    },
    {
      "barlow_loss": 0.17611879110336304,
      "epoch": 0.9967914438502674,
      "loss": 1.4022918939590454,
      "step": 233
    },
    {
      "barlow_loss": 0.18218854069709778,
      "epoch": 0.9967914438502674,
      "loss": 1.4862990379333496,
      "step": 233
    },
    {
      "barlow_loss": 0.16426695883274078,
      "epoch": 0.9967914438502674,
      "loss": 1.5278830528259277,
      "step": 233
    },
    {
      "barlow_loss": 0.17951582372188568,
      "epoch": 0.9967914438502674,
      "loss": 1.467275857925415,
      "step": 233
    },
    {
      "barlow_loss": 0.1637917160987854,
      "epoch": 0.9967914438502674,
      "loss": 1.4074748754501343,
      "step": 233
    },
    {
      "barlow_loss": 0.17014078795909882,
      "epoch": 0.9967914438502674,
      "loss": 1.3908138275146484,
      "step": 233
    },
    {
      "barlow_loss": 0.17229944467544556,
      "epoch": 0.9967914438502674,
      "loss": 1.3670952320098877,
      "step": 233
    },
    {
      "barlow_loss": 0.17426717281341553,
      "epoch": 0.9967914438502674,
      "loss": 1.5473886728286743,
      "step": 233
    },
    {
      "barlow_loss": 0.1768171340227127,
      "epoch": 0.9967914438502674,
      "loss": 1.408127784729004,
      "step": 233
    },
    {
      "barlow_loss": 0.1704457849264145,
      "epoch": 0.9967914438502674,
      "loss": 1.3785954713821411,
      "step": 233
    },
    {
      "barlow_loss": 0.162406787276268,
      "epoch": 0.9967914438502674,
      "loss": 1.4813930988311768,
      "step": 233
    },
    {
      "barlow_loss": 0.18233957886695862,
      "epoch": 0.9967914438502674,
      "loss": 1.265499234199524,
      "step": 233
    },
    {
      "barlow_loss": 0.17842014133930206,
      "epoch": 0.9967914438502674,
      "loss": 1.158975601196289,
      "step": 233
    },
    {
      "barlow_loss": 0.17275692522525787,
      "epoch": 0.9967914438502674,
      "loss": 1.408730149269104,
      "step": 233
    },
    {
      "barlow_loss": 0.15706944465637207,
      "epoch": 0.9967914438502674,
      "loss": 1.3360846042633057,
      "step": 233
    },
    {
      "barlow_loss": 0.18729369342327118,
      "epoch": 0.9967914438502674,
      "loss": 1.388216257095337,
      "step": 233
    },
    {
      "barlow_loss": 0.16411978006362915,
      "epoch": 0.9967914438502674,
      "loss": 1.3703336715698242,
      "step": 233
    },
    {
      "barlow_loss": 0.17853371798992157,
      "epoch": 0.9967914438502674,
      "loss": 1.4093480110168457,
      "step": 233
    },
    {
      "barlow_loss": 0.18798023462295532,
      "epoch": 0.9967914438502674,
      "loss": 1.485351324081421,
      "step": 233
    },
    {
      "epoch": 0.9967914438502674,
      "eval_loss": 1.4584516286849976,
      "eval_runtime": 840.7523,
      "eval_samples_per_second": 1.569,
      "eval_steps_per_second": 0.393,
      "step": 233
    },
    {
      "barlow_loss": 0.17760981619358063,
      "epoch": 0.9967914438502674,
      "loss": 1.495605707168579,
      "step": 233
    },
    {
      "barlow_loss": 0.1770021915435791,
      "epoch": 1.0010695187165775,
      "loss": 1.37521231174469,
      "step": 234
    },
    {
      "barlow_loss": 0.1618765890598297,
      "epoch": 1.0010695187165775,
      "loss": 1.2537994384765625,
      "step": 234
    },
    {
      "barlow_loss": 0.15927155315876007,
      "epoch": 1.0010695187165775,
      "loss": 1.4466516971588135,
      "step": 234
    },
    {
      "barlow_loss": 0.16015425324440002,
      "epoch": 1.0010695187165775,
      "loss": 1.6957645416259766,
      "step": 234
    },
    {
      "barlow_loss": 0.1753191351890564,
      "epoch": 1.0053475935828877,
      "loss": 1.4519840478897095,
      "step": 235
    },
    {
      "barlow_loss": 0.16390766203403473,
      "epoch": 1.0053475935828877,
      "loss": 1.4446734189987183,
      "step": 235
    },
    {
      "barlow_loss": 0.17977489531040192,
      "epoch": 1.0053475935828877,
      "loss": 1.3993191719055176,
      "step": 235
    },
    {
      "barlow_loss": 0.1527116894721985,
      "epoch": 1.0053475935828877,
      "loss": 1.402759075164795,
      "step": 235
    },
    {
      "epoch": 1.0096256684491978,
      "grad_norm": 0.8739461302757263,
      "learning_rate": 1.3299068284144572e-05,
      "loss": 1.4515,
      "step": 236
    },
    {
      "barlow_loss": 0.17256242036819458,
      "epoch": 1.0096256684491978,
      "loss": 1.6333814859390259,
      "step": 236
    },
    {
      "barlow_loss": 0.15991930663585663,
      "epoch": 1.0096256684491978,
      "loss": 1.5326086282730103,
      "step": 236
    },
    {
      "barlow_loss": 0.15620875358581543,
      "epoch": 1.0096256684491978,
      "loss": 1.5435751676559448,
      "step": 236
    },
    {
      "barlow_loss": 0.16620385646820068,
      "epoch": 1.0096256684491978,
      "loss": 1.2445136308670044,
      "step": 236
    },
    {
      "barlow_loss": 0.15490929782390594,
      "epoch": 1.013903743315508,
      "loss": 1.472212791442871,
      "step": 237
    },
    {
      "barlow_loss": 0.18204693496227264,
      "epoch": 1.013903743315508,
      "loss": 1.767061471939087,
      "step": 237
    },
    {
      "barlow_loss": 0.15682095289230347,
      "epoch": 1.013903743315508,
      "loss": 1.45550537109375,
      "step": 237
    },
    {
      "barlow_loss": 0.1619194895029068,
      "epoch": 1.013903743315508,
      "loss": 1.5612564086914062,
      "step": 237
    },
    {
      "barlow_loss": 0.1598099172115326,
      "epoch": 1.018181818181818,
      "loss": 1.6322846412658691,
      "step": 238
    },
    {
      "barlow_loss": 0.1617528796195984,
      "epoch": 1.018181818181818,
      "loss": 1.4850975275039673,
      "step": 238
    },
    {
      "barlow_loss": 0.17365753650665283,
      "epoch": 1.018181818181818,
      "loss": 1.4400417804718018,
      "step": 238
    },
    {
      "barlow_loss": 0.16030852496623993,
      "epoch": 1.018181818181818,
      "loss": 1.392042636871338,
      "step": 238
    },
    {
      "barlow_loss": 0.1657123565673828,
      "epoch": 1.0224598930481283,
      "loss": 1.3602609634399414,
      "step": 239
    },
    {
      "barlow_loss": 0.14172640442848206,
      "epoch": 1.0224598930481283,
      "loss": 1.343631386756897,
      "step": 239
    },
    {
      "barlow_loss": 0.15575334429740906,
      "epoch": 1.0224598930481283,
      "loss": 1.3385177850723267,
      "step": 239
    },
    {
      "barlow_loss": 0.14397960901260376,
      "epoch": 1.0224598930481283,
      "loss": 1.344821810722351,
      "step": 239
    },
    {
      "epoch": 1.0267379679144386,
      "grad_norm": 0.8215503692626953,
      "learning_rate": 1.2944136227726567e-05,
      "loss": 1.4717,
      "step": 240
    },
    {
      "barlow_loss": 0.17379097640514374,
      "epoch": 1.0267379679144386,
      "loss": 1.2634937763214111,
      "step": 240
    },
    {
      "barlow_loss": 0.15732093155384064,
      "epoch": 1.0267379679144386,
      "loss": 1.313191294670105,
      "step": 240
    },
    {
      "barlow_loss": 0.15981045365333557,
      "epoch": 1.0267379679144386,
      "loss": 1.3922816514968872,
      "step": 240
    },
    {
      "barlow_loss": 0.15303580462932587,
      "epoch": 1.0267379679144386,
      "loss": 1.2472560405731201,
      "step": 240
    },
    {
      "barlow_loss": 0.16058093309402466,
      "epoch": 1.0310160427807487,
      "loss": 1.360917329788208,
      "step": 241
    },
    {
      "barlow_loss": 0.163498193025589,
      "epoch": 1.0310160427807487,
      "loss": 1.4376444816589355,
      "step": 241
    },
    {
      "barlow_loss": 0.15655428171157837,
      "epoch": 1.0310160427807487,
      "loss": 1.3975940942764282,
      "step": 241
    },
    {
      "barlow_loss": 0.15592484176158905,
      "epoch": 1.0310160427807487,
      "loss": 1.4692643880844116,
      "step": 241
    },
    {
      "barlow_loss": 0.17216235399246216,
      "epoch": 1.035294117647059,
      "loss": 1.5662492513656616,
      "step": 242
    },
    {
      "barlow_loss": 0.16527658700942993,
      "epoch": 1.035294117647059,
      "loss": 1.5184812545776367,
      "step": 242
    },
    {
      "barlow_loss": 0.16231462359428406,
      "epoch": 1.035294117647059,
      "loss": 1.3897230625152588,
      "step": 242
    },
    {
      "barlow_loss": 0.16479256749153137,
      "epoch": 1.035294117647059,
      "loss": 1.3957170248031616,
      "step": 242
    },
    {
      "barlow_loss": 0.17735330760478973,
      "epoch": 1.039572192513369,
      "loss": 1.4187320470809937,
      "step": 243
    },
    {
      "barlow_loss": 0.1615438461303711,
      "epoch": 1.039572192513369,
      "loss": 1.4900352954864502,
      "step": 243
    },
    {
      "barlow_loss": 0.16044123470783234,
      "epoch": 1.039572192513369,
      "loss": 1.4709337949752808,
      "step": 243
    },
    {
      "barlow_loss": 0.15958544611930847,
      "epoch": 1.039572192513369,
      "loss": 1.315226674079895,
      "step": 243
    },
    {
      "epoch": 1.0438502673796792,
      "grad_norm": 0.9059010744094849,
      "learning_rate": 1.258884519801872e-05,
      "loss": 1.4029,
      "step": 244
    },
    {
      "barlow_loss": 0.16668781638145447,
      "epoch": 1.0438502673796792,
      "loss": 1.2319051027297974,
      "step": 244
    },
    {
      "barlow_loss": 0.15802766382694244,
      "epoch": 1.0438502673796792,
      "loss": 1.322811484336853,
      "step": 244
    },
    {
      "barlow_loss": 0.17932656407356262,
      "epoch": 1.0438502673796792,
      "loss": 1.7333186864852905,
      "step": 244
    },
    {
      "barlow_loss": 0.16772834956645966,
      "epoch": 1.0438502673796792,
      "loss": 1.6405940055847168,
      "step": 244
    },
    {
      "barlow_loss": 0.1721983551979065,
      "epoch": 1.0481283422459893,
      "loss": 1.4397430419921875,
      "step": 245
    },
    {
      "barlow_loss": 0.1591833084821701,
      "epoch": 1.0481283422459893,
      "loss": 1.4089272022247314,
      "step": 245
    },
    {
      "barlow_loss": 0.1691402941942215,
      "epoch": 1.0481283422459893,
      "loss": 1.3878803253173828,
      "step": 245
    },
    {
      "barlow_loss": 0.15059806406497955,
      "epoch": 1.0481283422459893,
      "loss": 1.229685664176941,
      "step": 245
    },
    {
      "barlow_loss": 0.16536009311676025,
      "epoch": 1.0524064171122995,
      "loss": 1.3698903322219849,
      "step": 246
    },
    {
      "barlow_loss": 0.16620613634586334,
      "epoch": 1.0524064171122995,
      "loss": 1.3071494102478027,
      "step": 246
    },
    {
      "barlow_loss": 0.1387304663658142,
      "epoch": 1.0524064171122995,
      "loss": 1.4055017232894897,
      "step": 246
    },
    {
      "barlow_loss": 0.15389147400856018,
      "epoch": 1.0524064171122995,
      "loss": 1.5183066129684448,
      "step": 246
    },
    {
      "barlow_loss": 0.17300055921077728,
      "epoch": 1.0566844919786096,
      "loss": 1.592128038406372,
      "step": 247
    },
    {
      "barlow_loss": 0.16730310022830963,
      "epoch": 1.0566844919786096,
      "loss": 1.3967235088348389,
      "step": 247
    },
    {
      "barlow_loss": 0.16248361766338348,
      "epoch": 1.0566844919786096,
      "loss": 1.4095792770385742,
      "step": 247
    },
    {
      "barlow_loss": 0.149921715259552,
      "epoch": 1.0566844919786096,
      "loss": 1.5692179203033447,
      "step": 247
    },
    {
      "epoch": 1.0609625668449199,
      "grad_norm": 0.9840031266212463,
      "learning_rate": 1.2233482359142818e-05,
      "loss": 1.4352,
      "step": 248
    },
    {
      "barlow_loss": 0.17121738195419312,
      "epoch": 1.0609625668449199,
      "loss": 1.3420912027359009,
      "step": 248
    },
    {
      "barlow_loss": 0.16799113154411316,
      "epoch": 1.0609625668449199,
      "loss": 1.5078260898590088,
      "step": 248
    },
    {
      "barlow_loss": 0.15496057271957397,
      "epoch": 1.0609625668449199,
      "loss": 1.4320710897445679,
      "step": 248
    },
    {
      "barlow_loss": 0.1515318602323532,
      "epoch": 1.0609625668449199,
      "loss": 1.3323590755462646,
      "step": 248
    },
    {
      "barlow_loss": 0.15720070898532867,
      "epoch": 1.06524064171123,
      "loss": 1.4600650072097778,
      "step": 249
    },
    {
      "barlow_loss": 0.17505259811878204,
      "epoch": 1.06524064171123,
      "loss": 1.3716164827346802,
      "step": 249
    },
    {
      "barlow_loss": 0.15706601738929749,
      "epoch": 1.06524064171123,
      "loss": 1.1958537101745605,
      "step": 249
    },
    {
      "barlow_loss": 0.15830230712890625,
      "epoch": 1.06524064171123,
      "loss": 1.2902858257293701,
      "step": 249
    },
    {
      "barlow_loss": 0.15381848812103271,
      "epoch": 1.0695187165775402,
      "loss": 1.4509085416793823,
      "step": 250
    },
    {
      "barlow_loss": 0.16416174173355103,
      "epoch": 1.0695187165775402,
      "loss": 1.3907841444015503,
      "step": 250
    },
    {
      "barlow_loss": 0.14609314501285553,
      "epoch": 1.0695187165775402,
      "loss": 1.5639441013336182,
      "step": 250
    },
    {
      "barlow_loss": 0.14777447283267975,
      "epoch": 1.0695187165775402,
      "loss": 1.4668254852294922,
      "step": 250
    },
    {
      "barlow_loss": 0.15234807133674622,
      "epoch": 1.0737967914438502,
      "loss": 1.3047447204589844,
      "step": 251
    },
    {
      "barlow_loss": 0.1747499406337738,
      "epoch": 1.0737967914438502,
      "loss": 1.508074402809143,
      "step": 251
    },
    {
      "barlow_loss": 0.15781623125076294,
      "epoch": 1.0737967914438502,
      "loss": 1.5071665048599243,
      "step": 251
    },
    {
      "barlow_loss": 0.15943670272827148,
      "epoch": 1.0737967914438502,
      "loss": 1.5669846534729004,
      "step": 251
    },
    {
      "epoch": 1.0780748663101605,
      "grad_norm": 0.967465877532959,
      "learning_rate": 1.1878334933260416e-05,
      "loss": 1.4182,
      "step": 252
    },
    {
      "barlow_loss": 0.17097768187522888,
      "epoch": 1.0780748663101605,
      "loss": 1.2706961631774902,
      "step": 252
    },
    {
      "barlow_loss": 0.17592373490333557,
      "epoch": 1.0780748663101605,
      "loss": 1.593865990638733,
      "step": 252
    },
    {
      "barlow_loss": 0.16645929217338562,
      "epoch": 1.0780748663101605,
      "loss": 1.5250213146209717,
      "step": 252
    },
    {
      "barlow_loss": 0.16728998720645905,
      "epoch": 1.0780748663101605,
      "loss": 1.3460700511932373,
      "step": 252
    },
    {
      "barlow_loss": 0.17925667762756348,
      "epoch": 1.0823529411764705,
      "loss": 1.4082930088043213,
      "step": 253
    },
    {
      "barlow_loss": 0.1719592809677124,
      "epoch": 1.0823529411764705,
      "loss": 1.266131043434143,
      "step": 253
    },
    {
      "barlow_loss": 0.17360498011112213,
      "epoch": 1.0823529411764705,
      "loss": 1.6167924404144287,
      "step": 253
    },
    {
      "barlow_loss": 0.16632238030433655,
      "epoch": 1.0823529411764705,
      "loss": 1.447889804840088,
      "step": 253
    },
    {
      "barlow_loss": 0.17434918880462646,
      "epoch": 1.0866310160427808,
      "loss": 1.4074900150299072,
      "step": 254
    },
    {
      "barlow_loss": 0.1634155958890915,
      "epoch": 1.0866310160427808,
      "loss": 1.4404031038284302,
      "step": 254
    },
    {
      "barlow_loss": 0.17723998427391052,
      "epoch": 1.0866310160427808,
      "loss": 1.5849215984344482,
      "step": 254
    },
    {
      "barlow_loss": 0.1560637503862381,
      "epoch": 1.0866310160427808,
      "loss": 1.4583419561386108,
      "step": 254
    },
    {
      "barlow_loss": 0.17206016182899475,
      "epoch": 1.0909090909090908,
      "loss": 1.4271372556686401,
      "step": 255
    },
    {
      "barlow_loss": 0.1751297563314438,
      "epoch": 1.0909090909090908,
      "loss": 1.4196243286132812,
      "step": 255
    },
    {
      "barlow_loss": 0.16426630318164825,
      "epoch": 1.0909090909090908,
      "loss": 1.3696261644363403,
      "step": 255
    },
    {
      "barlow_loss": 0.15349221229553223,
      "epoch": 1.0909090909090908,
      "loss": 1.4722611904144287,
      "step": 255
    },
    {
      "epoch": 1.095187165775401,
      "grad_norm": 0.9370867609977722,
      "learning_rate": 1.1523689968425443e-05,
      "loss": 1.4409,
      "step": 256
    },
    {
      "barlow_loss": 0.16523516178131104,
      "epoch": 1.095187165775401,
      "loss": 1.3902937173843384,
      "step": 256
    },
    {
      "barlow_loss": 0.16184644401073456,
      "epoch": 1.095187165775401,
      "loss": 1.4387580156326294,
      "step": 256
    },
    {
      "barlow_loss": 0.15298563241958618,
      "epoch": 1.095187165775401,
      "loss": 1.4297237396240234,
      "step": 256
    },
    {
      "barlow_loss": 0.16542424261569977,
      "epoch": 1.095187165775401,
      "loss": 1.4412453174591064,
      "step": 256
    },
    {
      "barlow_loss": 0.16545121371746063,
      "epoch": 1.0994652406417111,
      "loss": 1.5035098791122437,
      "step": 257
    },
    {
      "barlow_loss": 0.15654733777046204,
      "epoch": 1.0994652406417111,
      "loss": 1.3378052711486816,
      "step": 257
    },
    {
      "barlow_loss": 0.16954047977924347,
      "epoch": 1.0994652406417111,
      "loss": 1.221287727355957,
      "step": 257
    },
    {
      "barlow_loss": 0.15437647700309753,
      "epoch": 1.0994652406417111,
      "loss": 1.4195960760116577,
      "step": 257
    },
    {
      "barlow_loss": 0.1689256876707077,
      "epoch": 1.1037433155080214,
      "loss": 1.4297829866409302,
      "step": 258
    },
    {
      "barlow_loss": 0.1751781553030014,
      "epoch": 1.1037433155080214,
      "loss": 1.5124045610427856,
      "step": 258
    },
    {
      "barlow_loss": 0.1785656213760376,
      "epoch": 1.1037433155080214,
      "loss": 1.4305580854415894,
      "step": 258
    },
    {
      "barlow_loss": 0.17346255481243134,
      "epoch": 1.1037433155080214,
      "loss": 1.4205302000045776,
      "step": 258
    },
    {
      "barlow_loss": 0.14340746402740479,
      "epoch": 1.1080213903743314,
      "loss": 1.4965227842330933,
      "step": 259
    },
    {
      "barlow_loss": 0.17149440944194794,
      "epoch": 1.1080213903743314,
      "loss": 1.4416711330413818,
      "step": 259
    },
    {
      "barlow_loss": 0.14939440786838531,
      "epoch": 1.1080213903743314,
      "loss": 1.3479440212249756,
      "step": 259
    },
    {
      "barlow_loss": 0.15859746932983398,
      "epoch": 1.1080213903743314,
      "loss": 1.4329644441604614,
      "step": 259
    },
    {
      "epoch": 1.1122994652406417,
      "grad_norm": 0.9736074805259705,
      "learning_rate": 1.1169834106577472e-05,
      "loss": 1.4184,
      "step": 260
    },
    {
      "barlow_loss": 0.1621178686618805,
      "epoch": 1.1122994652406417,
      "loss": 1.464511752128601,
      "step": 260
    },
    {
      "barlow_loss": 0.14673882722854614,
      "epoch": 1.1122994652406417,
      "loss": 1.5073546171188354,
      "step": 260
    },
    {
      "barlow_loss": 0.16896873712539673,
      "epoch": 1.1122994652406417,
      "loss": 1.3755971193313599,
      "step": 260
    },
    {
      "barlow_loss": 0.16996178030967712,
      "epoch": 1.1122994652406417,
      "loss": 1.435895323753357,
      "step": 260
    },
    {
      "barlow_loss": 0.17383816838264465,
      "epoch": 1.116577540106952,
      "loss": 1.2761198282241821,
      "step": 261
    },
    {
      "barlow_loss": 0.15967662632465363,
      "epoch": 1.116577540106952,
      "loss": 1.4158236980438232,
      "step": 261
    },
    {
      "barlow_loss": 0.14631417393684387,
      "epoch": 1.116577540106952,
      "loss": 1.2795515060424805,
      "step": 261
    },
    {
      "barlow_loss": 0.15893222391605377,
      "epoch": 1.116577540106952,
      "loss": 1.530005931854248,
      "step": 261
    },
    {
      "barlow_loss": 0.16346020996570587,
      "epoch": 1.120855614973262,
      "loss": 1.487589955329895,
      "step": 262
    },
    {
      "barlow_loss": 0.1842881590127945,
      "epoch": 1.120855614973262,
      "loss": 1.4969477653503418,
      "step": 262
    },
    {
      "barlow_loss": 0.16229817271232605,
      "epoch": 1.120855614973262,
      "loss": 1.2853692770004272,
      "step": 262
    },
    {
      "barlow_loss": 0.15094435214996338,
      "epoch": 1.120855614973262,
      "loss": 1.4146772623062134,
      "step": 262
    },
    {
      "barlow_loss": 0.16562435030937195,
      "epoch": 1.125133689839572,
      "loss": 1.3744735717773438,
      "step": 263
    },
    {
      "barlow_loss": 0.17069223523139954,
      "epoch": 1.125133689839572,
      "loss": 1.392030119895935,
      "step": 263
    },
    {
      "barlow_loss": 0.1683955043554306,
      "epoch": 1.125133689839572,
      "loss": 1.2934447526931763,
      "step": 263
    },
    {
      "barlow_loss": 0.18869560956954956,
      "epoch": 1.125133689839572,
      "loss": 1.5340265035629272,
      "step": 263
    },
    {
      "epoch": 1.1294117647058823,
      "grad_norm": 0.9064357280731201,
      "learning_rate": 1.081705335186326e-05,
      "loss": 1.4102,
      "step": 264
    },
    {
      "barlow_loss": 0.15547077357769012,
      "epoch": 1.1294117647058823,
      "loss": 1.5563838481903076,
      "step": 264
    },
    {
      "barlow_loss": 0.16792365908622742,
      "epoch": 1.1294117647058823,
      "loss": 1.3217110633850098,
      "step": 264
    },
    {
      "barlow_loss": 0.1744866669178009,
      "epoch": 1.1294117647058823,
      "loss": 1.4842538833618164,
      "step": 264
    },
    {
      "barlow_loss": 0.16570602357387543,
      "epoch": 1.1294117647058823,
      "loss": 1.5661921501159668,
      "step": 264
    },
    {
      "barlow_loss": 0.15599387884140015,
      "epoch": 1.1336898395721926,
      "loss": 1.420258641242981,
      "step": 265
    },
    {
      "barlow_loss": 0.15386901795864105,
      "epoch": 1.1336898395721926,
      "loss": 1.474305510520935,
      "step": 265
    },
    {
      "barlow_loss": 0.16921061277389526,
      "epoch": 1.1336898395721926,
      "loss": 1.431146264076233,
      "step": 265
    },
    {
      "barlow_loss": 0.17879550158977509,
      "epoch": 1.1336898395721926,
      "loss": 1.4600324630737305,
      "step": 265
    },
    {
      "barlow_loss": 0.15995445847511292,
      "epoch": 1.1379679144385026,
      "loss": 1.55377995967865,
      "step": 266
    },
    {
      "barlow_loss": 0.16181936860084534,
      "epoch": 1.1379679144385026,
      "loss": 1.3951478004455566,
      "step": 266
    },
    {
      "barlow_loss": 0.17559006810188293,
      "epoch": 1.1379679144385026,
      "loss": 1.3460596799850464,
      "step": 266
    },
    {
      "barlow_loss": 0.15351298451423645,
      "epoch": 1.1379679144385026,
      "loss": 1.2801997661590576,
      "step": 266
    },
    {
      "barlow_loss": 0.16363829374313354,
      "epoch": 1.142245989304813,
      "loss": 1.2984014749526978,
      "step": 267
    },
    {
      "barlow_loss": 0.16076958179473877,
      "epoch": 1.142245989304813,
      "loss": 1.5234451293945312,
      "step": 267
    },
    {
      "barlow_loss": 0.16184550523757935,
      "epoch": 1.142245989304813,
      "loss": 1.5300605297088623,
      "step": 267
    },
    {
      "barlow_loss": 0.14909180998802185,
      "epoch": 1.142245989304813,
      "loss": 1.4202876091003418,
      "step": 267
    },
    {
      "epoch": 1.146524064171123,
      "grad_norm": 1.0107258558273315,
      "learning_rate": 1.0465632839473777e-05,
      "loss": 1.4414,
      "step": 268
    },
    {
      "barlow_loss": 0.14854606986045837,
      "epoch": 1.146524064171123,
      "loss": 1.4784746170043945,
      "step": 268
    },
    {
      "barlow_loss": 0.16444335877895355,
      "epoch": 1.146524064171123,
      "loss": 1.3136060237884521,
      "step": 268
    },
    {
      "barlow_loss": 0.16549842059612274,
      "epoch": 1.146524064171123,
      "loss": 1.490563154220581,
      "step": 268
    },
    {
      "barlow_loss": 0.1639695167541504,
      "epoch": 1.146524064171123,
      "loss": 1.5012558698654175,
      "step": 268
    },
    {
      "barlow_loss": 0.17907066643238068,
      "epoch": 1.1508021390374332,
      "loss": 1.240760087966919,
      "step": 269
    },
    {
      "barlow_loss": 0.18694011867046356,
      "epoch": 1.1508021390374332,
      "loss": 1.5283491611480713,
      "step": 269
    },
    {
      "barlow_loss": 0.15518230199813843,
      "epoch": 1.1508021390374332,
      "loss": 1.3410053253173828,
      "step": 269
    },
    {
      "barlow_loss": 0.16858524084091187,
      "epoch": 1.1508021390374332,
      "loss": 1.4896938800811768,
      "step": 269
    },
    {
      "barlow_loss": 0.16173246502876282,
      "epoch": 1.1550802139037433,
      "loss": 1.2748607397079468,
      "step": 270
    },
    {
      "barlow_loss": 0.16112463176250458,
      "epoch": 1.1550802139037433,
      "loss": 1.6271631717681885,
      "step": 270
    },
    {
      "barlow_loss": 0.1741667091846466,
      "epoch": 1.1550802139037433,
      "loss": 1.2794195413589478,
      "step": 270
    },
    {
      "barlow_loss": 0.17859145998954773,
      "epoch": 1.1550802139037433,
      "loss": 1.4833102226257324,
      "step": 270
    },
    {
      "barlow_loss": 0.16357053816318512,
      "epoch": 1.1593582887700535,
      "loss": 1.3840961456298828,
      "step": 271
    },
    {
      "barlow_loss": 0.16731810569763184,
      "epoch": 1.1593582887700535,
      "loss": 1.6155661344528198,
      "step": 271
    },
    {
      "barlow_loss": 0.16467829048633575,
      "epoch": 1.1593582887700535,
      "loss": 1.4674819707870483,
      "step": 271
    },
    {
      "barlow_loss": 0.1776798814535141,
      "epoch": 1.1593582887700535,
      "loss": 1.3697023391723633,
      "step": 271
    },
    {
      "epoch": 1.1636363636363636,
      "grad_norm": 0.957348108291626,
      "learning_rate": 1.0115856605183581e-05,
      "loss": 1.4303,
      "step": 272
    },
    {
      "barlow_loss": 0.1636219024658203,
      "epoch": 1.1636363636363636,
      "loss": 1.4549992084503174,
      "step": 272
    },
    {
      "barlow_loss": 0.15554849803447723,
      "epoch": 1.1636363636363636,
      "loss": 1.5152366161346436,
      "step": 272
    },
    {
      "barlow_loss": 0.1396462768316269,
      "epoch": 1.1636363636363636,
      "loss": 1.4166847467422485,
      "step": 272
    },
    {
      "barlow_loss": 0.1755114495754242,
      "epoch": 1.1636363636363636,
      "loss": 1.4314275979995728,
      "step": 272
    },
    {
      "barlow_loss": 0.16570839285850525,
      "epoch": 1.1679144385026738,
      "loss": 1.4317413568496704,
      "step": 273
    },
    {
      "barlow_loss": 0.16959494352340698,
      "epoch": 1.1679144385026738,
      "loss": 1.4989380836486816,
      "step": 273
    },
    {
      "barlow_loss": 0.1669914275407791,
      "epoch": 1.1679144385026738,
      "loss": 1.2889835834503174,
      "step": 273
    },
    {
      "barlow_loss": 0.1643492728471756,
      "epoch": 1.1679144385026738,
      "loss": 1.4485000371932983,
      "step": 273
    },
    {
      "barlow_loss": 0.14395427703857422,
      "epoch": 1.172192513368984,
      "loss": 1.2654823064804077,
      "step": 274
    },
    {
      "barlow_loss": 0.15663430094718933,
      "epoch": 1.172192513368984,
      "loss": 1.553335428237915,
      "step": 274
    },
    {
      "barlow_loss": 0.1671992540359497,
      "epoch": 1.172192513368984,
      "loss": 1.615455985069275,
      "step": 274
    },
    {
      "barlow_loss": 0.15297414362430573,
      "epoch": 1.172192513368984,
      "loss": 1.3963165283203125,
      "step": 274
    },
    {
      "barlow_loss": 0.15696510672569275,
      "epoch": 1.1764705882352942,
      "loss": 1.4779388904571533,
      "step": 275
    },
    {
      "barlow_loss": 0.15214046835899353,
      "epoch": 1.1764705882352942,
      "loss": 1.3589365482330322,
      "step": 275
    },
    {
      "barlow_loss": 0.1461433619260788,
      "epoch": 1.1764705882352942,
      "loss": 1.4539991617202759,
      "step": 275
    },
    {
      "barlow_loss": 0.1489359438419342,
      "epoch": 1.1764705882352942,
      "loss": 1.2199369668960571,
      "step": 275
    },
    {
      "epoch": 1.1807486631016042,
      "grad_norm": 0.9641424417495728,
      "learning_rate": 9.768007355778764e-06,
      "loss": 1.4267,
      "step": 276
    },
    {
      "barlow_loss": 0.16302280128002167,
      "epoch": 1.1807486631016042,
      "loss": 1.4038667678833008,
      "step": 276
    },
    {
      "barlow_loss": 0.15567271411418915,
      "epoch": 1.1807486631016042,
      "loss": 1.5252588987350464,
      "step": 276
    },
    {
      "barlow_loss": 0.16339200735092163,
      "epoch": 1.1807486631016042,
      "loss": 1.4447475671768188,
      "step": 276
    },
    {
      "barlow_loss": 0.17018766701221466,
      "epoch": 1.1807486631016042,
      "loss": 1.3472226858139038,
      "step": 276
    },
    {
      "barlow_loss": 0.1455286741256714,
      "epoch": 1.1850267379679145,
      "loss": 1.4392876625061035,
      "step": 277
    },
    {
      "barlow_loss": 0.15976853668689728,
      "epoch": 1.1850267379679145,
      "loss": 1.2675848007202148,
      "step": 277
    },
    {
      "barlow_loss": 0.17369145154953003,
      "epoch": 1.1850267379679145,
      "loss": 1.335842251777649,
      "step": 277
    },
    {
      "barlow_loss": 0.17372359335422516,
      "epoch": 1.1850267379679145,
      "loss": 1.2919878959655762,
      "step": 277
    },
    {
      "barlow_loss": 0.16659727692604065,
      "epoch": 1.1893048128342245,
      "loss": 1.4749261140823364,
      "step": 278
    },
    {
      "barlow_loss": 0.15589278936386108,
      "epoch": 1.1893048128342245,
      "loss": 1.4421523809432983,
      "step": 278
    },
    {
      "barlow_loss": 0.16749665141105652,
      "epoch": 1.1893048128342245,
      "loss": 1.3130996227264404,
      "step": 278
    },
    {
      "barlow_loss": 0.15777182579040527,
      "epoch": 1.1893048128342245,
      "loss": 1.5018832683563232,
      "step": 278
    },
    {
      "barlow_loss": 0.17269372940063477,
      "epoch": 1.1935828877005348,
      "loss": 1.3466044664382935,
      "step": 279
    },
    {
      "barlow_loss": 0.18381625413894653,
      "epoch": 1.1935828877005348,
      "loss": 1.3191275596618652,
      "step": 279
    },
    {
      "barlow_loss": 0.1448126882314682,
      "epoch": 1.1935828877005348,
      "loss": 1.4900484085083008,
      "step": 279
    },
    {
      "barlow_loss": 0.16990719735622406,
      "epoch": 1.1935828877005348,
      "loss": 1.4466850757598877,
      "step": 279
    },
    {
      "epoch": 1.1978609625668448,
      "grad_norm": 0.9203186631202698,
      "learning_rate": 9.422366240559092e-06,
      "loss": 1.3994,
      "step": 280
    },
    {
      "barlow_loss": 0.15646542608737946,
      "epoch": 1.1978609625668448,
      "loss": 1.4767552614212036,
      "step": 280
    },
    {
      "barlow_loss": 0.15904918313026428,
      "epoch": 1.1978609625668448,
      "loss": 1.4998247623443604,
      "step": 280
    },
    {
      "barlow_loss": 0.15924622118473053,
      "epoch": 1.1978609625668448,
      "loss": 1.4560978412628174,
      "step": 280
    },
    {
      "barlow_loss": 0.1729372888803482,
      "epoch": 1.1978609625668448,
      "loss": 1.5103791952133179,
      "step": 280
    },
    {
      "barlow_loss": 0.16095738112926483,
      "epoch": 1.202139037433155,
      "loss": 1.618930459022522,
      "step": 281
    },
    {
      "barlow_loss": 0.16308540105819702,
      "epoch": 1.202139037433155,
      "loss": 1.4167401790618896,
      "step": 281
    },
    {
      "barlow_loss": 0.16721265017986298,
      "epoch": 1.202139037433155,
      "loss": 1.2525736093521118,
      "step": 281
    },
    {
      "barlow_loss": 0.1678769290447235,
      "epoch": 1.202139037433155,
      "loss": 1.4486027956008911,
      "step": 281
    },
    {
      "barlow_loss": 0.1580558866262436,
      "epoch": 1.2064171122994654,
      "loss": 1.3225740194320679,
      "step": 282
    },
    {
      "barlow_loss": 0.14984850585460663,
      "epoch": 1.2064171122994654,
      "loss": 1.431451439857483,
      "step": 282
    },
    {
      "barlow_loss": 0.17257821559906006,
      "epoch": 1.2064171122994654,
      "loss": 1.3162751197814941,
      "step": 282
    },
    {
      "barlow_loss": 0.163250133395195,
      "epoch": 1.2064171122994654,
      "loss": 1.3791841268539429,
      "step": 282
    },
    {
      "barlow_loss": 0.15570497512817383,
      "epoch": 1.2106951871657754,
      "loss": 1.3421531915664673,
      "step": 283
    },
    {
      "barlow_loss": 0.15562905371189117,
      "epoch": 1.2106951871657754,
      "loss": 1.4662587642669678,
      "step": 283
    },
    {
      "barlow_loss": 0.16594485938549042,
      "epoch": 1.2106951871657754,
      "loss": 1.3123503923416138,
      "step": 283
    },
    {
      "barlow_loss": 0.1715538054704666,
      "epoch": 1.2106951871657754,
      "loss": 1.2762707471847534,
      "step": 283
    },
    {
      "epoch": 1.2149732620320854,
      "grad_norm": 0.9223936200141907,
      "learning_rate": 9.079212624098965e-06,
      "loss": 1.4079,
      "step": 284
    },
    {
      "barlow_loss": 0.15574117004871368,
      "epoch": 1.2149732620320854,
      "loss": 1.5061839818954468,
      "step": 284
    },
    {
      "barlow_loss": 0.1580096334218979,
      "epoch": 1.2149732620320854,
      "loss": 1.3083808422088623,
      "step": 284
    },
    {
      "barlow_loss": 0.1691766083240509,
      "epoch": 1.2149732620320854,
      "loss": 1.4137059450149536,
      "step": 284
    },
    {
      "barlow_loss": 0.17912335693836212,
      "epoch": 1.2149732620320854,
      "loss": 1.4101589918136597,
      "step": 284
    },
    {
      "barlow_loss": 0.15747788548469543,
      "epoch": 1.2192513368983957,
      "loss": 1.3835076093673706,
      "step": 285
    },
    {
      "barlow_loss": 0.16153648495674133,
      "epoch": 1.2192513368983957,
      "loss": 1.4515846967697144,
      "step": 285
    },
    {
      "barlow_loss": 0.1832638829946518,
      "epoch": 1.2192513368983957,
      "loss": 1.5712833404541016,
      "step": 285
    },
    {
      "barlow_loss": 0.16708292067050934,
      "epoch": 1.2192513368983957,
      "loss": 1.2785704135894775,
      "step": 285
    },
    {
      "barlow_loss": 0.16151279211044312,
      "epoch": 1.223529411764706,
      "loss": 1.4812335968017578,
      "step": 286
    },
    {
      "barlow_loss": 0.1631208211183548,
      "epoch": 1.223529411764706,
      "loss": 1.6383442878723145,
      "step": 286
    },
    {
      "barlow_loss": 0.15013283491134644,
      "epoch": 1.223529411764706,
      "loss": 1.3586474657058716,
      "step": 286
    },
    {
      "barlow_loss": 0.1634666621685028,
      "epoch": 1.223529411764706,
      "loss": 1.3023624420166016,
      "step": 286
    },
    {
      "barlow_loss": 0.17411386966705322,
      "epoch": 1.227807486631016,
      "loss": 1.3135627508163452,
      "step": 287
    },
    {
      "barlow_loss": 0.14731881022453308,
      "epoch": 1.227807486631016,
      "loss": 1.2654609680175781,
      "step": 287
    },
    {
      "barlow_loss": 0.179386168718338,
      "epoch": 1.227807486631016,
      "loss": 1.5270146131515503,
      "step": 287
    },
    {
      "barlow_loss": 0.177154079079628,
      "epoch": 1.227807486631016,
      "loss": 1.3244340419769287,
      "step": 287
    },
    {
      "epoch": 1.2320855614973263,
      "grad_norm": 0.9717881679534912,
      "learning_rate": 8.738823860450874e-06,
      "loss": 1.4084,
      "step": 288
    },
    {
      "barlow_loss": 0.18010151386260986,
      "epoch": 1.2320855614973263,
      "loss": 1.4226598739624023,
      "step": 288
    },
    {
      "barlow_loss": 0.16337478160858154,
      "epoch": 1.2320855614973263,
      "loss": 1.5709961652755737,
      "step": 288
    },
    {
      "barlow_loss": 0.1559692919254303,
      "epoch": 1.2320855614973263,
      "loss": 1.323628544807434,
      "step": 288
    },
    {
      "barlow_loss": 0.1699560433626175,
      "epoch": 1.2320855614973263,
      "loss": 1.511042833328247,
      "step": 288
    },
    {
      "barlow_loss": 0.16329629719257355,
      "epoch": 1.2363636363636363,
      "loss": 1.5743759870529175,
      "step": 289
    },
    {
      "barlow_loss": 0.16865794360637665,
      "epoch": 1.2363636363636363,
      "loss": 1.5927278995513916,
      "step": 289
    },
    {
      "barlow_loss": 0.17322216928005219,
      "epoch": 1.2363636363636363,
      "loss": 1.3579232692718506,
      "step": 289
    },
    {
      "barlow_loss": 0.16988921165466309,
      "epoch": 1.2363636363636363,
      "loss": 1.3899980783462524,
      "step": 289
    },
    {
      "barlow_loss": 0.1866893172264099,
      "epoch": 1.2406417112299466,
      "loss": 1.5299618244171143,
      "step": 290
    },
    {
      "barlow_loss": 0.17183439433574677,
      "epoch": 1.2406417112299466,
      "loss": 1.4968563318252563,
      "step": 290
    },
    {
      "barlow_loss": 0.15841899812221527,
      "epoch": 1.2406417112299466,
      "loss": 1.514529824256897,
      "step": 290
    },
    {
      "barlow_loss": 0.15337786078453064,
      "epoch": 1.2406417112299466,
      "loss": 1.3301174640655518,
      "step": 290
    },
    {
      "barlow_loss": 0.15584079921245575,
      "epoch": 1.2449197860962566,
      "loss": 1.4096242189407349,
      "step": 291
    },
    {
      "barlow_loss": 0.14547127485275269,
      "epoch": 1.2449197860962566,
      "loss": 1.3719409704208374,
      "step": 291
    },
    {
      "barlow_loss": 0.17049762606620789,
      "epoch": 1.2449197860962566,
      "loss": 1.6322863101959229,
      "step": 291
    },
    {
      "barlow_loss": 0.1719040423631668,
      "epoch": 1.2449197860962566,
      "loss": 1.533942699432373,
      "step": 291
    },
    {
      "epoch": 1.249197860962567,
      "grad_norm": 1.0133179426193237,
      "learning_rate": 8.40147506897389e-06,
      "loss": 1.4727,
      "step": 292
    },
    {
      "barlow_loss": 0.17978830635547638,
      "epoch": 1.249197860962567,
      "loss": 1.389291763305664,
      "step": 292
    },
    {
      "barlow_loss": 0.15538683533668518,
      "epoch": 1.249197860962567,
      "loss": 1.3658721446990967,
      "step": 292
    },
    {
      "barlow_loss": 0.16359665989875793,
      "epoch": 1.249197860962567,
      "loss": 1.3347073793411255,
      "step": 292
    },
    {
      "barlow_loss": 0.1573910117149353,
      "epoch": 1.249197860962567,
      "loss": 1.2532888650894165,
      "step": 292
    },
    {
      "barlow_loss": 0.1600763201713562,
      "epoch": 1.253475935828877,
      "loss": 1.2836253643035889,
      "step": 293
    },
    {
      "barlow_loss": 0.16958627104759216,
      "epoch": 1.253475935828877,
      "loss": 1.4088703393936157,
      "step": 293
    },
    {
      "barlow_loss": 0.16310051083564758,
      "epoch": 1.253475935828877,
      "loss": 1.5435166358947754,
      "step": 293
    },
    {
      "barlow_loss": 0.16161614656448364,
      "epoch": 1.253475935828877,
      "loss": 1.5825093984603882,
      "step": 293
    },
    {
      "barlow_loss": 0.16044411063194275,
      "epoch": 1.2577540106951872,
      "loss": 1.4825676679611206,
      "step": 294
    },
    {
      "barlow_loss": 0.15643751621246338,
      "epoch": 1.2577540106951872,
      "loss": 1.3271191120147705,
      "step": 294
    },
    {
      "barlow_loss": 0.15715721249580383,
      "epoch": 1.2577540106951872,
      "loss": 1.461525321006775,
      "step": 294
    },
    {
      "barlow_loss": 0.1643514484167099,
      "epoch": 1.2577540106951872,
      "loss": 1.3150506019592285,
      "step": 294
    },
    {
      "barlow_loss": 0.15519680082798004,
      "epoch": 1.2620320855614973,
      "loss": 1.262597918510437,
      "step": 295
    },
    {
      "barlow_loss": 0.16587772965431213,
      "epoch": 1.2620320855614973,
      "loss": 1.6835649013519287,
      "step": 295
    },
    {
      "barlow_loss": 0.16439130902290344,
      "epoch": 1.2620320855614973,
      "loss": 1.4621564149856567,
      "step": 295
    },
    {
      "barlow_loss": 0.16752536594867706,
      "epoch": 1.2620320855614973,
      "loss": 1.5206165313720703,
      "step": 295
    },
    {
      "epoch": 1.2663101604278075,
      "grad_norm": 1.166208267211914,
      "learning_rate": 8.067438911968305e-06,
      "loss": 1.4173,
      "step": 296
    },
    {
      "barlow_loss": 0.15607859194278717,
      "epoch": 1.2663101604278075,
      "loss": 1.275511622428894,
      "step": 296
    },
    {
      "barlow_loss": 0.1627466082572937,
      "epoch": 1.2663101604278075,
      "loss": 1.5545141696929932,
      "step": 296
    },
    {
      "barlow_loss": 0.15728852152824402,
      "epoch": 1.2663101604278075,
      "loss": 1.4864388704299927,
      "step": 296
    },
    {
      "barlow_loss": 0.17365142703056335,
      "epoch": 1.2663101604278075,
      "loss": 1.4332444667816162,
      "step": 296
    },
    {
      "barlow_loss": 0.15460769832134247,
      "epoch": 1.2705882352941176,
      "loss": 1.5958845615386963,
      "step": 297
    },
    {
      "barlow_loss": 0.1680685579776764,
      "epoch": 1.2705882352941176,
      "loss": 1.6586371660232544,
      "step": 297
    },
    {
      "barlow_loss": 0.1532331258058548,
      "epoch": 1.2705882352941176,
      "loss": 1.4406466484069824,
      "step": 297
    },
    {
      "barlow_loss": 0.16257484257221222,
      "epoch": 1.2705882352941176,
      "loss": 1.5697921514511108,
      "step": 297
    },
    {
      "barlow_loss": 0.17429667711257935,
      "epoch": 1.2748663101604278,
      "loss": 1.3017971515655518,
      "step": 298
    },
    {
      "barlow_loss": 0.1646052747964859,
      "epoch": 1.2748663101604278,
      "loss": 1.2502954006195068,
      "step": 298
    },
    {
      "barlow_loss": 0.16178712248802185,
      "epoch": 1.2748663101604278,
      "loss": 1.27544105052948,
      "step": 298
    },
    {
      "barlow_loss": 0.17761464416980743,
      "epoch": 1.2748663101604278,
      "loss": 1.4033979177474976,
      "step": 298
    },
    {
      "barlow_loss": 0.160198375582695,
      "epoch": 1.279144385026738,
      "loss": 1.369808316230774,
      "step": 299
    },
    {
      "barlow_loss": 0.17670147120952606,
      "epoch": 1.279144385026738,
      "loss": 1.4652498960494995,
      "step": 299
    },
    {
      "barlow_loss": 0.1624394953250885,
      "epoch": 1.279144385026738,
      "loss": 1.506310224533081,
      "step": 299
    },
    {
      "barlow_loss": 0.16033995151519775,
      "epoch": 1.279144385026738,
      "loss": 1.2892048358917236,
      "step": 299
    },
    {
      "epoch": 1.2834224598930482,
      "grad_norm": 1.1043682098388672,
      "learning_rate": 7.736985374296246e-06,
      "loss": 1.4298,
      "step": 300
    },
    {
      "barlow_loss": 0.16551290452480316,
      "epoch": 1.2834224598930482,
      "loss": 1.3235009908676147,
      "step": 300
    },
    {
      "barlow_loss": 0.17058689892292023,
      "epoch": 1.2834224598930482,
      "loss": 1.5304789543151855,
      "step": 300
    },
    {
      "barlow_loss": 0.16516752541065216,
      "epoch": 1.2834224598930482,
      "loss": 1.349576473236084,
      "step": 300
    },
    {
      "barlow_loss": 0.17694631218910217,
      "epoch": 1.2834224598930482,
      "loss": 1.3587971925735474,
      "step": 300
    },
    {
      "barlow_loss": 0.16077497601509094,
      "epoch": 1.2877005347593582,
      "loss": 1.2571754455566406,
      "step": 301
    },
    {
      "barlow_loss": 0.16418196260929108,
      "epoch": 1.2877005347593582,
      "loss": 1.3798984289169312,
      "step": 301
    },
    {
      "barlow_loss": 0.16714823246002197,
      "epoch": 1.2877005347593582,
      "loss": 1.3388923406600952,
      "step": 301
    },
    {
      "barlow_loss": 0.1714666485786438,
      "epoch": 1.2877005347593582,
      "loss": 1.5488300323486328,
      "step": 301
    },
    {
      "barlow_loss": 0.16127371788024902,
      "epoch": 1.2919786096256685,
      "loss": 1.4295345544815063,
      "step": 302
    },
    {
      "barlow_loss": 0.17459134757518768,
      "epoch": 1.2919786096256685,
      "loss": 1.3238446712493896,
      "step": 302
    },
    {
      "barlow_loss": 0.16919024288654327,
      "epoch": 1.2919786096256685,
      "loss": 1.289993405342102,
      "step": 302
    },
    {
      "barlow_loss": 0.17443245649337769,
      "epoch": 1.2919786096256685,
      "loss": 1.403639316558838,
      "step": 302
    },
    {
      "barlow_loss": 0.16960826516151428,
      "epoch": 1.2962566844919787,
      "loss": 1.6124749183654785,
      "step": 303
    },
    {
      "barlow_loss": 0.1544586569070816,
      "epoch": 1.2962566844919787,
      "loss": 1.491522192955017,
      "step": 303
    },
    {
      "barlow_loss": 0.16344287991523743,
      "epoch": 1.2962566844919787,
      "loss": 1.3802080154418945,
      "step": 303
    },
    {
      "barlow_loss": 0.14812329411506653,
      "epoch": 1.2962566844919787,
      "loss": 1.4132450819015503,
      "step": 303
    },
    {
      "epoch": 1.3005347593582888,
      "grad_norm": 1.089735507965088,
      "learning_rate": 7.41038154516627e-06,
      "loss": 1.402,
      "step": 304
    },
    {
      "barlow_loss": 0.16705097258090973,
      "epoch": 1.3005347593582888,
      "loss": 1.3889951705932617,
      "step": 304
    },
    {
      "barlow_loss": 0.16502559185028076,
      "epoch": 1.3005347593582888,
      "loss": 1.6580785512924194,
      "step": 304
    },
    {
      "barlow_loss": 0.14345484972000122,
      "epoch": 1.3005347593582888,
      "loss": 1.3567434549331665,
      "step": 304
    },
    {
      "barlow_loss": 0.15290671586990356,
      "epoch": 1.3005347593582888,
      "loss": 1.4326144456863403,
      "step": 304
    },
    {
      "barlow_loss": 0.17623275518417358,
      "epoch": 1.3048128342245988,
      "loss": 1.2773797512054443,
      "step": 305
    },
    {
      "barlow_loss": 0.17308957874774933,
      "epoch": 1.3048128342245988,
      "loss": 1.458248496055603,
      "step": 305
    },
    {
      "barlow_loss": 0.17531085014343262,
      "epoch": 1.3048128342245988,
      "loss": 1.4025449752807617,
      "step": 305
    },
    {
      "barlow_loss": 0.16140180826187134,
      "epoch": 1.3048128342245988,
      "loss": 1.3804513216018677,
      "step": 305
    },
    {
      "barlow_loss": 0.15159112215042114,
      "epoch": 1.309090909090909,
      "loss": 1.2567124366760254,
      "step": 306
    },
    {
      "barlow_loss": 0.16269414126873016,
      "epoch": 1.309090909090909,
      "loss": 1.3342959880828857,
      "step": 306
    },
    {
      "barlow_loss": 0.17329874634742737,
      "epoch": 1.309090909090909,
      "loss": 1.4371693134307861,
      "step": 306
    },
    {
      "barlow_loss": 0.15324151515960693,
      "epoch": 1.309090909090909,
      "loss": 1.4624208211898804,
      "step": 306
    },
    {
      "barlow_loss": 0.15259690582752228,
      "epoch": 1.3133689839572193,
      "loss": 1.414725422859192,
      "step": 307
    },
    {
      "barlow_loss": 0.16762718558311462,
      "epoch": 1.3133689839572193,
      "loss": 1.4930109977722168,
      "step": 307
    },
    {
      "barlow_loss": 0.15962201356887817,
      "epoch": 1.3133689839572193,
      "loss": 1.5051145553588867,
      "step": 307
    },
    {
      "barlow_loss": 0.15060235559940338,
      "epoch": 1.3133689839572193,
      "loss": 1.5083311796188354,
      "step": 307
    },
    {
      "epoch": 1.3176470588235294,
      "grad_norm": 1.1001722812652588,
      "learning_rate": 7.087891402258412e-06,
      "loss": 1.4229,
      "step": 308
    },
    {
      "barlow_loss": 0.15226440131664276,
      "epoch": 1.3176470588235294,
      "loss": 1.3886877298355103,
      "step": 308
    },
    {
      "barlow_loss": 0.1711483746767044,
      "epoch": 1.3176470588235294,
      "loss": 1.321161150932312,
      "step": 308
    },
    {
      "barlow_loss": 0.17132067680358887,
      "epoch": 1.3176470588235294,
      "loss": 1.4463709592819214,
      "step": 308
    },
    {
      "barlow_loss": 0.15636573731899261,
      "epoch": 1.3176470588235294,
      "loss": 1.4809699058532715,
      "step": 308
    },
    {
      "barlow_loss": 0.16492882370948792,
      "epoch": 1.3219251336898394,
      "loss": 1.4403736591339111,
      "step": 309
    },
    {
      "barlow_loss": 0.16459067165851593,
      "epoch": 1.3219251336898394,
      "loss": 1.5015332698822021,
      "step": 309
    },
    {
      "barlow_loss": 0.16407500207424164,
      "epoch": 1.3219251336898394,
      "loss": 1.436253309249878,
      "step": 309
    },
    {
      "barlow_loss": 0.1648886501789093,
      "epoch": 1.3219251336898394,
      "loss": 1.3678207397460938,
      "step": 309
    },
    {
      "barlow_loss": 0.17225529253482819,
      "epoch": 1.3262032085561497,
      "loss": 1.2279388904571533,
      "step": 310
    },
    {
      "barlow_loss": 0.16193845868110657,
      "epoch": 1.3262032085561497,
      "loss": 1.5029487609863281,
      "step": 310
    },
    {
      "barlow_loss": 0.15999333560466766,
      "epoch": 1.3262032085561497,
      "loss": 1.4262890815734863,
      "step": 310
    },
    {
      "barlow_loss": 0.15990857779979706,
      "epoch": 1.3262032085561497,
      "loss": 1.3552539348602295,
      "step": 310
    },
    {
      "barlow_loss": 0.17022638022899628,
      "epoch": 1.33048128342246,
      "loss": 1.4441906213760376,
      "step": 311
    },
    {
      "barlow_loss": 0.1787494421005249,
      "epoch": 1.33048128342246,
      "loss": 1.4644654989242554,
      "step": 311
    },
    {
      "barlow_loss": 0.16114279627799988,
      "epoch": 1.33048128342246,
      "loss": 1.3849101066589355,
      "step": 311
    },
    {
      "barlow_loss": 0.15941578149795532,
      "epoch": 1.33048128342246,
      "loss": 1.3086636066436768,
      "step": 311
    },
    {
      "epoch": 1.33475935828877,
      "grad_norm": 0.9415149092674255,
      "learning_rate": 6.769775598364126e-06,
      "loss": 1.4061,
      "step": 312
    },
    {
      "barlow_loss": 0.16439658403396606,
      "epoch": 1.33475935828877,
      "loss": 1.3957345485687256,
      "step": 312
    },
    {
      "barlow_loss": 0.1626112312078476,
      "epoch": 1.33475935828877,
      "loss": 1.2999829053878784,
      "step": 312
    },
    {
      "barlow_loss": 0.15411238372325897,
      "epoch": 1.33475935828877,
      "loss": 1.3141921758651733,
      "step": 312
    },
    {
      "barlow_loss": 0.17526260018348694,
      "epoch": 1.33475935828877,
      "loss": 1.5170872211456299,
      "step": 312
    },
    {
      "barlow_loss": 0.16201019287109375,
      "epoch": 1.3390374331550803,
      "loss": 1.4196733236312866,
      "step": 313
    },
    {
      "barlow_loss": 0.15171951055526733,
      "epoch": 1.3390374331550803,
      "loss": 1.2772808074951172,
      "step": 313
    },
    {
      "barlow_loss": 0.15302926301956177,
      "epoch": 1.3390374331550803,
      "loss": 1.216784119606018,
      "step": 313
    },
    {
      "barlow_loss": 0.17770211398601532,
      "epoch": 1.3390374331550803,
      "loss": 1.4499766826629639,
      "step": 313
    },
    {
      "barlow_loss": 0.147957444190979,
      "epoch": 1.3433155080213903,
      "loss": 1.1770087480545044,
      "step": 314
    },
    {
      "barlow_loss": 0.1566104292869568,
      "epoch": 1.3433155080213903,
      "loss": 1.2398148775100708,
      "step": 314
    },
    {
      "barlow_loss": 0.17030031979084015,
      "epoch": 1.3433155080213903,
      "loss": 1.2855693101882935,
      "step": 314
    },
    {
      "barlow_loss": 0.1548352986574173,
      "epoch": 1.3433155080213903,
      "loss": 1.5078543424606323,
      "step": 314
    },
    {
      "barlow_loss": 0.13713201880455017,
      "epoch": 1.3475935828877006,
      "loss": 1.4976340532302856,
      "step": 315
    },
    {
      "barlow_loss": 0.15563353896141052,
      "epoch": 1.3475935828877006,
      "loss": 1.2895441055297852,
      "step": 315
    },
    {
      "barlow_loss": 0.14380726218223572,
      "epoch": 1.3475935828877006,
      "loss": 1.4210413694381714,
      "step": 315
    },
    {
      "barlow_loss": 0.15898899734020233,
      "epoch": 1.3475935828877006,
      "loss": 1.3734915256500244,
      "step": 315
    },
    {
      "epoch": 1.3518716577540106,
      "grad_norm": 1.0363781452178955,
      "learning_rate": 6.456291250713531e-06,
      "loss": 1.3552,
      "step": 316
    },
    {
      "barlow_loss": 0.15882475674152374,
      "epoch": 1.3518716577540106,
      "loss": 1.3814775943756104,
      "step": 316
    },
    {
      "barlow_loss": 0.1676209270954132,
      "epoch": 1.3518716577540106,
      "loss": 1.3577632904052734,
      "step": 316
    },
    {
      "barlow_loss": 0.17008325457572937,
      "epoch": 1.3518716577540106,
      "loss": 1.3979911804199219,
      "step": 316
    },
    {
      "barlow_loss": 0.14193353056907654,
      "epoch": 1.3518716577540106,
      "loss": 1.3262431621551514,
      "step": 316
    },
    {
      "barlow_loss": 0.1670791506767273,
      "epoch": 1.356149732620321,
      "loss": 1.4911247491836548,
      "step": 317
    },
    {
      "barlow_loss": 0.1614382266998291,
      "epoch": 1.356149732620321,
      "loss": 1.3106114864349365,
      "step": 317
    },
    {
      "barlow_loss": 0.17261242866516113,
      "epoch": 1.356149732620321,
      "loss": 1.2235817909240723,
      "step": 317
    },
    {
      "barlow_loss": 0.16489793360233307,
      "epoch": 1.356149732620321,
      "loss": 1.4215168952941895,
      "step": 317
    },
    {
      "barlow_loss": 0.17163152992725372,
      "epoch": 1.360427807486631,
      "loss": 1.3893452882766724,
      "step": 318
    },
    {
      "barlow_loss": 0.16074590384960175,
      "epoch": 1.360427807486631,
      "loss": 1.6202038526535034,
      "step": 318
    },
    {
      "barlow_loss": 0.1737179011106491,
      "epoch": 1.360427807486631,
      "loss": 1.1719720363616943,
      "step": 318
    },
    {
      "barlow_loss": 0.16444699466228485,
      "epoch": 1.360427807486631,
      "loss": 1.488257884979248,
      "step": 318
    },
    {
      "barlow_loss": 0.1790151596069336,
      "epoch": 1.3647058823529412,
      "loss": 1.3790866136550903,
      "step": 319
    },
    {
      "barlow_loss": 0.1527070850133896,
      "epoch": 1.3647058823529412,
      "loss": 1.4421541690826416,
      "step": 319
    },
    {
      "barlow_loss": 0.1596255898475647,
      "epoch": 1.3647058823529412,
      "loss": 1.43302583694458,
      "step": 319
    },
    {
      "barlow_loss": 0.1668556034564972,
      "epoch": 1.3647058823529412,
      "loss": 1.2581379413604736,
      "step": 319
    },
    {
      "epoch": 1.3689839572192513,
      "grad_norm": 1.1148815155029297,
      "learning_rate": 6.147691733160338e-06,
      "loss": 1.3808,
      "step": 320
    },
    {
      "barlow_loss": 0.16067299246788025,
      "epoch": 1.3689839572192513,
      "loss": 1.42144775390625,
      "step": 320
    },
    {
      "barlow_loss": 0.17290851473808289,
      "epoch": 1.3689839572192513,
      "loss": 1.4896397590637207,
      "step": 320
    },
    {
      "barlow_loss": 0.15203823149204254,
      "epoch": 1.3689839572192513,
      "loss": 1.2991288900375366,
      "step": 320
    },
    {
      "barlow_loss": 0.15833567082881927,
      "epoch": 1.3689839572192513,
      "loss": 1.3969216346740723,
      "step": 320
    },
    {
      "barlow_loss": 0.1758192479610443,
      "epoch": 1.3732620320855615,
      "loss": 1.3178925514221191,
      "step": 321
    },
    {
      "barlow_loss": 0.16629870235919952,
      "epoch": 1.3732620320855615,
      "loss": 1.4358935356140137,
      "step": 321
    },
    {
      "barlow_loss": 0.17848920822143555,
      "epoch": 1.3732620320855615,
      "loss": 1.37489914894104,
      "step": 321
    },
    {
      "barlow_loss": 0.1553826779127121,
      "epoch": 1.3732620320855615,
      "loss": 1.358275294303894,
      "step": 321
    },
    {
      "barlow_loss": 0.16146261990070343,
      "epoch": 1.3775401069518716,
      "loss": 1.418050765991211,
      "step": 322
    },
    {
      "barlow_loss": 0.1733247935771942,
      "epoch": 1.3775401069518716,
      "loss": 1.5164159536361694,
      "step": 322
    },
    {
      "barlow_loss": 0.14458678662776947,
      "epoch": 1.3775401069518716,
      "loss": 1.316252589225769,
      "step": 322
    },
    {
      "barlow_loss": 0.16073982417583466,
      "epoch": 1.3775401069518716,
      "loss": 1.2816340923309326,
      "step": 322
    },
    {
      "barlow_loss": 0.16175299882888794,
      "epoch": 1.3818181818181818,
      "loss": 1.4370685815811157,
      "step": 323
    },
    {
      "barlow_loss": 0.1646631956100464,
      "epoch": 1.3818181818181818,
      "loss": 1.401346206665039,
      "step": 323
    },
    {
      "barlow_loss": 0.1522103101015091,
      "epoch": 1.3818181818181818,
      "loss": 1.3205643892288208,
      "step": 323
    },
    {
      "barlow_loss": 0.17054805159568787,
      "epoch": 1.3818181818181818,
      "loss": 1.3487253189086914,
      "step": 323
    },
    {
      "epoch": 1.386096256684492,
      "grad_norm": 1.0349311828613281,
      "learning_rate": 5.8442264713922956e-06,
      "loss": 1.3834,
      "step": 324
    },
    {
      "barlow_loss": 0.18482168018817902,
      "epoch": 1.386096256684492,
      "loss": 1.413385272026062,
      "step": 324
    },
    {
      "barlow_loss": 0.17107246816158295,
      "epoch": 1.386096256684492,
      "loss": 1.318993330001831,
      "step": 324
    },
    {
      "barlow_loss": 0.17083622515201569,
      "epoch": 1.386096256684492,
      "loss": 1.4391940832138062,
      "step": 324
    },
    {
      "barlow_loss": 0.16108264029026031,
      "epoch": 1.386096256684492,
      "loss": 1.612626075744629,
      "step": 324
    },
    {
      "barlow_loss": 0.1824459284543991,
      "epoch": 1.3903743315508021,
      "loss": 1.4007426500320435,
      "step": 325
    },
    {
      "barlow_loss": 0.17272236943244934,
      "epoch": 1.3903743315508021,
      "loss": 1.6148756742477417,
      "step": 325
    },
    {
      "barlow_loss": 0.14751897752285004,
      "epoch": 1.3903743315508021,
      "loss": 1.3087377548217773,
      "step": 325
    },
    {
      "barlow_loss": 0.16109415888786316,
      "epoch": 1.3903743315508021,
      "loss": 1.3145782947540283,
      "step": 325
    },
    {
      "barlow_loss": 0.169277161359787,
      "epoch": 1.3946524064171122,
      "loss": 1.376477837562561,
      "step": 326
    },
    {
      "barlow_loss": 0.1761491894721985,
      "epoch": 1.3946524064171122,
      "loss": 1.3827143907546997,
      "step": 326
    },
    {
      "barlow_loss": 0.1582074910402298,
      "epoch": 1.3946524064171122,
      "loss": 1.3788666725158691,
      "step": 326
    },
    {
      "barlow_loss": 0.1613713651895523,
      "epoch": 1.3946524064171122,
      "loss": 1.6022061109542847,
      "step": 326
    },
    {
      "barlow_loss": 0.17309261858463287,
      "epoch": 1.3989304812834225,
      "loss": 1.2480074167251587,
      "step": 327
    },
    {
      "barlow_loss": 0.15915311872959137,
      "epoch": 1.3989304812834225,
      "loss": 1.3674001693725586,
      "step": 327
    },
    {
      "barlow_loss": 0.15929624438285828,
      "epoch": 1.3989304812834225,
      "loss": 1.3884687423706055,
      "step": 327
    },
    {
      "barlow_loss": 0.14927883446216583,
      "epoch": 1.3989304812834225,
      "loss": 1.3579200506210327,
      "step": 327
    },
    {
      "epoch": 1.4032085561497327,
      "grad_norm": 0.9720832705497742,
      "learning_rate": 5.5461407413328005e-06,
      "loss": 1.4078,
      "step": 328
    },
    {
      "barlow_loss": 0.151846244931221,
      "epoch": 1.4032085561497327,
      "loss": 1.305336594581604,
      "step": 328
    },
    {
      "barlow_loss": 0.15919972956180573,
      "epoch": 1.4032085561497327,
      "loss": 1.1508983373641968,
      "step": 328
    },
    {
      "barlow_loss": 0.1602754294872284,
      "epoch": 1.4032085561497327,
      "loss": 1.3540921211242676,
      "step": 328
    },
    {
      "barlow_loss": 0.1573108434677124,
      "epoch": 1.4032085561497327,
      "loss": 1.437476396560669,
      "step": 328
    },
    {
      "barlow_loss": 0.16744579374790192,
      "epoch": 1.4074866310160428,
      "loss": 1.368462324142456,
      "step": 329
    },
    {
      "barlow_loss": 0.1761282980442047,
      "epoch": 1.4074866310160428,
      "loss": 1.34173583984375,
      "step": 329
    },
    {
      "barlow_loss": 0.1831836849451065,
      "epoch": 1.4074866310160428,
      "loss": 1.4566900730133057,
      "step": 329
    },
    {
      "barlow_loss": 0.16348640620708466,
      "epoch": 1.4074866310160428,
      "loss": 1.2420719861984253,
      "step": 329
    },
    {
      "barlow_loss": 0.17522504925727844,
      "epoch": 1.4117647058823528,
      "loss": 1.4514782428741455,
      "step": 330
    },
    {
      "barlow_loss": 0.15449529886245728,
      "epoch": 1.4117647058823528,
      "loss": 1.4630829095840454,
      "step": 330
    },
    {
      "barlow_loss": 0.16896945238113403,
      "epoch": 1.4117647058823528,
      "loss": 1.3633064031600952,
      "step": 330
    },
    {
      "barlow_loss": 0.15403762459754944,
      "epoch": 1.4117647058823528,
      "loss": 1.2441495656967163,
      "step": 330
    },
    {
      "barlow_loss": 0.16740381717681885,
      "epoch": 1.416042780748663,
      "loss": 1.4499642848968506,
      "step": 331
    },
    {
      "barlow_loss": 0.17362187802791595,
      "epoch": 1.416042780748663,
      "loss": 1.3603529930114746,
      "step": 331
    },
    {
      "barlow_loss": 0.16524654626846313,
      "epoch": 1.416042780748663,
      "loss": 1.339410662651062,
      "step": 331
    },
    {
      "barlow_loss": 0.15593375265598297,
      "epoch": 1.416042780748663,
      "loss": 1.3326475620269775,
      "step": 331
    },
    {
      "epoch": 1.4203208556149733,
      "grad_norm": 1.037824034690857,
      "learning_rate": 5.253675470896521e-06,
      "loss": 1.3538,
      "step": 332
    },
    {
      "barlow_loss": 0.1818309873342514,
      "epoch": 1.4203208556149733,
      "loss": 1.4392261505126953,
      "step": 332
    },
    {
      "barlow_loss": 0.1615051031112671,
      "epoch": 1.4203208556149733,
      "loss": 1.4093788862228394,
      "step": 332
    },
    {
      "barlow_loss": 0.17623668909072876,
      "epoch": 1.4203208556149733,
      "loss": 1.4799479246139526,
      "step": 332
    },
    {
      "barlow_loss": 0.15793824195861816,
      "epoch": 1.4203208556149733,
      "loss": 1.3365936279296875,
      "step": 332
    },
    {
      "barlow_loss": 0.1686384230852127,
      "epoch": 1.4245989304812834,
      "loss": 1.4201158285140991,
      "step": 333
    },
    {
      "barlow_loss": 0.16560128331184387,
      "epoch": 1.4245989304812834,
      "loss": 1.6222683191299438,
      "step": 333
    },
    {
      "barlow_loss": 0.17059975862503052,
      "epoch": 1.4245989304812834,
      "loss": 1.496461033821106,
      "step": 333
    },
    {
      "barlow_loss": 0.16526922583580017,
      "epoch": 1.4245989304812834,
      "loss": 1.3005188703536987,
      "step": 333
    },
    {
      "barlow_loss": 0.15565049648284912,
      "epoch": 1.4288770053475937,
      "loss": 1.2382608652114868,
      "step": 334
    },
    {
      "barlow_loss": 0.1644240915775299,
      "epoch": 1.4288770053475937,
      "loss": 1.403070092201233,
      "step": 334
    },
    {
      "barlow_loss": 0.16333606839179993,
      "epoch": 1.4288770053475937,
      "loss": 1.4666098356246948,
      "step": 334
    },
    {
      "barlow_loss": 0.16799655556678772,
      "epoch": 1.4288770053475937,
      "loss": 1.5247597694396973,
      "step": 334
    },
    {
      "barlow_loss": 0.17068274319171906,
      "epoch": 1.4331550802139037,
      "loss": 1.334659457206726,
      "step": 335
    },
    {
      "barlow_loss": 0.15731704235076904,
      "epoch": 1.4331550802139037,
      "loss": 1.253639817237854,
      "step": 335
    },
    {
      "barlow_loss": 0.1582518070936203,
      "epoch": 1.4331550802139037,
      "loss": 1.3593850135803223,
      "step": 335
    },
    {
      "barlow_loss": 0.15813854336738586,
      "epoch": 1.4331550802139037,
      "loss": 1.4070193767547607,
      "step": 335
    },
    {
      "epoch": 1.437433155080214,
      "grad_norm": 0.9833551049232483,
      "learning_rate": 4.967067045259296e-06,
      "loss": 1.4057,
      "step": 336
    },
    {
      "barlow_loss": 0.15022817254066467,
      "epoch": 1.437433155080214,
      "loss": 1.561268925666809,
      "step": 336
    },
    {
      "barlow_loss": 0.1703580766916275,
      "epoch": 1.437433155080214,
      "loss": 1.4425667524337769,
      "step": 336
    },
    {
      "barlow_loss": 0.15765666961669922,
      "epoch": 1.437433155080214,
      "loss": 1.3691396713256836,
      "step": 336
    },
    {
      "barlow_loss": 0.14909961819648743,
      "epoch": 1.437433155080214,
      "loss": 1.3097721338272095,
      "step": 336
    },
    {
      "barlow_loss": 0.16888222098350525,
      "epoch": 1.441711229946524,
      "loss": 1.4301077127456665,
      "step": 337
    },
    {
      "barlow_loss": 0.16195949912071228,
      "epoch": 1.441711229946524,
      "loss": 1.2099076509475708,
      "step": 337
    },
    {
      "barlow_loss": 0.15590837597846985,
      "epoch": 1.441711229946524,
      "loss": 1.4898364543914795,
      "step": 337
    },
    {
      "barlow_loss": 0.16880865395069122,
      "epoch": 1.441711229946524,
      "loss": 1.237210750579834,
      "step": 337
    },
    {
      "barlow_loss": 0.17356310784816742,
      "epoch": 1.4459893048128343,
      "loss": 1.3795673847198486,
      "step": 338
    },
    {
      "barlow_loss": 0.17372451722621918,
      "epoch": 1.4459893048128343,
      "loss": 1.5629991292953491,
      "step": 338
    },
    {
      "barlow_loss": 0.174073725938797,
      "epoch": 1.4459893048128343,
      "loss": 1.4475706815719604,
      "step": 338
    },
    {
      "barlow_loss": 0.14907848834991455,
      "epoch": 1.4459893048128343,
      "loss": 1.3676248788833618,
      "step": 338
    },
    {
      "barlow_loss": 0.1731414496898651,
      "epoch": 1.4502673796791443,
      "loss": 1.427351713180542,
      "step": 339
    },
    {
      "barlow_loss": 0.15891648828983307,
      "epoch": 1.4502673796791443,
      "loss": 1.5184779167175293,
      "step": 339
    },
    {
      "barlow_loss": 0.17527258396148682,
      "epoch": 1.4502673796791443,
      "loss": 1.278509497642517,
      "step": 339
    },
    {
      "barlow_loss": 0.16975858807563782,
      "epoch": 1.4502673796791443,
      "loss": 1.3307304382324219,
      "step": 339
    },
    {
      "epoch": 1.4545454545454546,
      "grad_norm": 1.0912426710128784,
      "learning_rate": 4.6865471157997455e-06,
      "loss": 1.3977,
      "step": 340
    },
    {
      "barlow_loss": 0.1544775664806366,
      "epoch": 1.4545454545454546,
      "loss": 1.4169507026672363,
      "step": 340
    },
    {
      "barlow_loss": 0.15470749139785767,
      "epoch": 1.4545454545454546,
      "loss": 1.5377650260925293,
      "step": 340
    },
    {
      "barlow_loss": 0.14427834749221802,
      "epoch": 1.4545454545454546,
      "loss": 1.3715479373931885,
      "step": 340
    },
    {
      "barlow_loss": 0.17056547105312347,
      "epoch": 1.4545454545454546,
      "loss": 1.1369004249572754,
      "step": 340
    },
    {
      "barlow_loss": 0.14361561834812164,
      "epoch": 1.4588235294117646,
      "loss": 1.358918309211731,
      "step": 341
    },
    {
      "barlow_loss": 0.16447120904922485,
      "epoch": 1.4588235294117646,
      "loss": 1.493674635887146,
      "step": 341
    },
    {
      "barlow_loss": 0.1570630520582199,
      "epoch": 1.4588235294117646,
      "loss": 1.145390510559082,
      "step": 341
    },
    {
      "barlow_loss": 0.17242999374866486,
      "epoch": 1.4588235294117646,
      "loss": 1.303160548210144,
      "step": 341
    },
    {
      "barlow_loss": 0.1664409339427948,
      "epoch": 1.463101604278075,
      "loss": 1.3914823532104492,
      "step": 342
    },
    {
      "barlow_loss": 0.16661041975021362,
      "epoch": 1.463101604278075,
      "loss": 1.3246859312057495,
      "step": 342
    },
    {
      "barlow_loss": 0.1731366068124771,
      "epoch": 1.463101604278075,
      "loss": 1.1632184982299805,
      "step": 342
    },
    {
      "barlow_loss": 0.1743421107530594,
      "epoch": 1.463101604278075,
      "loss": 1.4293603897094727,
      "step": 342
    },
    {
      "barlow_loss": 0.18244561553001404,
      "epoch": 1.467379679144385,
      "loss": 1.4469352960586548,
      "step": 343
    },
    {
      "barlow_loss": 0.17067895829677582,
      "epoch": 1.467379679144385,
      "loss": 1.4376293420791626,
      "step": 343
    },
    {
      "barlow_loss": 0.16263046860694885,
      "epoch": 1.467379679144385,
      "loss": 1.5346453189849854,
      "step": 343
    },
    {
      "barlow_loss": 0.16184011101722717,
      "epoch": 1.467379679144385,
      "loss": 1.4247088432312012,
      "step": 343
    },
    {
      "epoch": 1.4716577540106952,
      "grad_norm": 1.136305570602417,
      "learning_rate": 4.4123424128669325e-06,
      "loss": 1.3698,
      "step": 344
    },
    {
      "barlow_loss": 0.17306238412857056,
      "epoch": 1.4716577540106952,
      "loss": 1.4370853900909424,
      "step": 344
    },
    {
      "barlow_loss": 0.1567813754081726,
      "epoch": 1.4716577540106952,
      "loss": 1.4070285558700562,
      "step": 344
    },
    {
      "barlow_loss": 0.15805217623710632,
      "epoch": 1.4716577540106952,
      "loss": 1.2983912229537964,
      "step": 344
    },
    {
      "barlow_loss": 0.14716924726963043,
      "epoch": 1.4716577540106952,
      "loss": 1.221583366394043,
      "step": 344
    },
    {
      "barlow_loss": 0.1667984426021576,
      "epoch": 1.4759358288770055,
      "loss": 1.4663047790527344,
      "step": 345
    },
    {
      "barlow_loss": 0.17427152395248413,
      "epoch": 1.4759358288770055,
      "loss": 1.4576724767684937,
      "step": 345
    },
    {
      "barlow_loss": 0.15646156668663025,
      "epoch": 1.4759358288770055,
      "loss": 1.2281476259231567,
      "step": 345
    },
    {
      "barlow_loss": 0.17776723206043243,
      "epoch": 1.4759358288770055,
      "loss": 1.4958257675170898,
      "step": 345
    },
    {
      "barlow_loss": 0.1731806844472885,
      "epoch": 1.4802139037433155,
      "loss": 1.3555006980895996,
      "step": 346
    },
    {
      "barlow_loss": 0.16525331139564514,
      "epoch": 1.4802139037433155,
      "loss": 1.4095308780670166,
      "step": 346
    },
    {
      "barlow_loss": 0.1645641326904297,
      "epoch": 1.4802139037433155,
      "loss": 1.298234224319458,
      "step": 346
    },
    {
      "barlow_loss": 0.15710678696632385,
      "epoch": 1.4802139037433155,
      "loss": 1.3490973711013794,
      "step": 346
    },
    {
      "barlow_loss": 0.1705683469772339,
      "epoch": 1.4844919786096256,
      "loss": 1.506506085395813,
      "step": 347
    },
    {
      "barlow_loss": 0.17084170877933502,
      "epoch": 1.4844919786096256,
      "loss": 1.2995682954788208,
      "step": 347
    },
    {
      "barlow_loss": 0.1522163599729538,
      "epoch": 1.4844919786096256,
      "loss": 1.3185662031173706,
      "step": 347
    },
    {
      "barlow_loss": 0.16379977762699127,
      "epoch": 1.4844919786096256,
      "loss": 1.403045654296875,
      "step": 347
    },
    {
      "epoch": 1.4887700534759358,
      "grad_norm": 1.0705865621566772,
      "learning_rate": 4.144674562525456e-06,
      "loss": 1.372,
      "step": 348
    },
    {
      "barlow_loss": 0.1607091724872589,
      "epoch": 1.4887700534759358,
      "loss": 1.2916347980499268,
      "step": 348
    },
    {
      "barlow_loss": 0.1618138551712036,
      "epoch": 1.4887700534759358,
      "loss": 1.4244470596313477,
      "step": 348
    },
    {
      "barlow_loss": 0.15761026740074158,
      "epoch": 1.4887700534759358,
      "loss": 1.3071353435516357,
      "step": 348
    },
    {
      "barlow_loss": 0.16127003729343414,
      "epoch": 1.4887700534759358,
      "loss": 1.3780419826507568,
      "step": 348
    },
    {
      "barlow_loss": 0.1704813539981842,
      "epoch": 1.493048128342246,
      "loss": 1.3746514320373535,
      "step": 349
    },
    {
      "barlow_loss": 0.17716370522975922,
      "epoch": 1.493048128342246,
      "loss": 1.3147494792938232,
      "step": 349
    },
    {
      "barlow_loss": 0.1647396832704544,
      "epoch": 1.493048128342246,
      "loss": 1.5289981365203857,
      "step": 349
    },
    {
      "barlow_loss": 0.1595843881368637,
      "epoch": 1.493048128342246,
      "loss": 1.3080378770828247,
      "step": 349
    },
    {
      "barlow_loss": 0.1718212366104126,
      "epoch": 1.4973262032085561,
      "loss": 1.4392653703689575,
      "step": 350
    },
    {
      "barlow_loss": 0.16080275177955627,
      "epoch": 1.4973262032085561,
      "loss": 1.3272380828857422,
      "step": 350
    },
    {
      "barlow_loss": 0.16039294004440308,
      "epoch": 1.4973262032085561,
      "loss": 1.3821508884429932,
      "step": 350
    },
    {
      "barlow_loss": 0.15646159648895264,
      "epoch": 1.4973262032085561,
      "loss": 1.2084412574768066,
      "step": 350
    },
    {
      "barlow_loss": 0.17194320261478424,
      "epoch": 1.5016042780748662,
      "loss": 1.2499035596847534,
      "step": 351
    },
    {
      "barlow_loss": 0.16247810423374176,
      "epoch": 1.5016042780748662,
      "loss": 1.5125539302825928,
      "step": 351
    },
    {
      "barlow_loss": 0.15735095739364624,
      "epoch": 1.5016042780748662,
      "loss": 1.3011701107025146,
      "step": 351
    },
    {
      "barlow_loss": 0.15826109051704407,
      "epoch": 1.5016042780748662,
      "loss": 1.5000697374343872,
      "step": 351
    },
    {
      "epoch": 1.5058823529411764,
      "grad_norm": 1.0634288787841797,
      "learning_rate": 3.883759907426117e-06,
      "loss": 1.3655,
      "step": 352
    },
    {
      "barlow_loss": 0.17752288281917572,
      "epoch": 1.5058823529411764,
      "loss": 1.3616119623184204,
      "step": 352
    },
    {
      "barlow_loss": 0.16366463899612427,
      "epoch": 1.5058823529411764,
      "loss": 1.5193582773208618,
      "step": 352
    },
    {
      "barlow_loss": 0.17744271457195282,
      "epoch": 1.5058823529411764,
      "loss": 1.3024263381958008,
      "step": 352
    },
    {
      "barlow_loss": 0.17559731006622314,
      "epoch": 1.5058823529411764,
      "loss": 1.3710620403289795,
      "step": 352
    },
    {
      "barlow_loss": 0.15495112538337708,
      "epoch": 1.5101604278074867,
      "loss": 1.445385217666626,
      "step": 353
    },
    {
      "barlow_loss": 0.17648181319236755,
      "epoch": 1.5101604278074867,
      "loss": 1.3976949453353882,
      "step": 353
    },
    {
      "barlow_loss": 0.16756242513656616,
      "epoch": 1.5101604278074867,
      "loss": 1.1532552242279053,
      "step": 353
    },
    {
      "barlow_loss": 0.17734979093074799,
      "epoch": 1.5101604278074867,
      "loss": 1.2848584651947021,
      "step": 353
    },
    {
      "barlow_loss": 0.1648934781551361,
      "epoch": 1.5144385026737968,
      "loss": 1.2412714958190918,
      "step": 354
    },
    {
      "barlow_loss": 0.17484387755393982,
      "epoch": 1.5144385026737968,
      "loss": 1.3221831321716309,
      "step": 354
    },
    {
      "barlow_loss": 0.16399520635604858,
      "epoch": 1.5144385026737968,
      "loss": 1.3084765672683716,
      "step": 354
    },
    {
      "barlow_loss": 0.16492575407028198,
      "epoch": 1.5144385026737968,
      "loss": 1.4062292575836182,
      "step": 354
    },
    {
      "barlow_loss": 0.1740492880344391,
      "epoch": 1.5187165775401068,
      "loss": 1.286585807800293,
      "step": 355
    },
    {
      "barlow_loss": 0.17551425099372864,
      "epoch": 1.5187165775401068,
      "loss": 1.380715250968933,
      "step": 355
    },
    {
      "barlow_loss": 0.16213569045066833,
      "epoch": 1.5187165775401068,
      "loss": 1.430554747581482,
      "step": 355
    },
    {
      "barlow_loss": 0.1742367446422577,
      "epoch": 1.5187165775401068,
      "loss": 1.3983973264694214,
      "step": 355
    },
    {
      "epoch": 1.522994652406417,
      "grad_norm": 0.9974904656410217,
      "learning_rate": 3.629809331946862e-06,
      "loss": 1.3506,
      "step": 356
    },
    {
      "barlow_loss": 0.16880130767822266,
      "epoch": 1.522994652406417,
      "loss": 1.263501763343811,
      "step": 356
    },
    {
      "barlow_loss": 0.16264992952346802,
      "epoch": 1.522994652406417,
      "loss": 1.3913302421569824,
      "step": 356
    },
    {
      "barlow_loss": 0.17163677513599396,
      "epoch": 1.522994652406417,
      "loss": 1.3995391130447388,
      "step": 356
    },
    {
      "barlow_loss": 0.16064508259296417,
      "epoch": 1.522994652406417,
      "loss": 1.4268337488174438,
      "step": 356
    },
    {
      "barlow_loss": 0.15626883506774902,
      "epoch": 1.5272727272727273,
      "loss": 1.3894704580307007,
      "step": 357
    },
    {
      "barlow_loss": 0.1560847908258438,
      "epoch": 1.5272727272727273,
      "loss": 1.5339715480804443,
      "step": 357
    },
    {
      "barlow_loss": 0.1600819230079651,
      "epoch": 1.5272727272727273,
      "loss": 1.402121663093567,
      "step": 357
    },
    {
      "barlow_loss": 0.16549015045166016,
      "epoch": 1.5272727272727273,
      "loss": 1.219244122505188,
      "step": 357
    },
    {
      "barlow_loss": 0.17517028748989105,
      "epoch": 1.5315508021390374,
      "loss": 1.208366870880127,
      "step": 358
    },
    {
      "barlow_loss": 0.16427987813949585,
      "epoch": 1.5315508021390374,
      "loss": 1.2528504133224487,
      "step": 358
    },
    {
      "barlow_loss": 0.15649661421775818,
      "epoch": 1.5315508021390374,
      "loss": 1.463644027709961,
      "step": 358
    },
    {
      "barlow_loss": 0.16224664449691772,
      "epoch": 1.5315508021390374,
      "loss": 1.4219878911972046,
      "step": 358
    },
    {
      "barlow_loss": 0.16481883823871613,
      "epoch": 1.5358288770053476,
      "loss": 1.3467943668365479,
      "step": 359
    },
    {
      "barlow_loss": 0.18653351068496704,
      "epoch": 1.5358288770053476,
      "loss": 1.455762267112732,
      "step": 359
    },
    {
      "barlow_loss": 0.15810516476631165,
      "epoch": 1.5358288770053476,
      "loss": 1.3911062479019165,
      "step": 359
    },
    {
      "barlow_loss": 0.166363924741745,
      "epoch": 1.5358288770053476,
      "loss": 1.3365144729614258,
      "step": 359
    },
    {
      "epoch": 1.5401069518716577,
      "grad_norm": 1.0449621677398682,
      "learning_rate": 3.3830280917453996e-06,
      "loss": 1.3689,
      "step": 360
    },
    {
      "barlow_loss": 0.16527217626571655,
      "epoch": 1.5401069518716577,
      "loss": 1.517494797706604,
      "step": 360
    },
    {
      "barlow_loss": 0.17381568253040314,
      "epoch": 1.5401069518716577,
      "loss": 1.1958295106887817,
      "step": 360
    },
    {
      "barlow_loss": 0.17180046439170837,
      "epoch": 1.5401069518716577,
      "loss": 1.4203394651412964,
      "step": 360
    },
    {
      "barlow_loss": 0.16698859632015228,
      "epoch": 1.5401069518716577,
      "loss": 1.2405234575271606,
      "step": 360
    },
    {
      "barlow_loss": 0.17167143523693085,
      "epoch": 1.544385026737968,
      "loss": 1.45158052444458,
      "step": 361
    },
    {
      "barlow_loss": 0.16417059302330017,
      "epoch": 1.544385026737968,
      "loss": 1.4216976165771484,
      "step": 361
    },
    {
      "barlow_loss": 0.16978517174720764,
      "epoch": 1.544385026737968,
      "loss": 1.5553916692733765,
      "step": 361
    },
    {
      "barlow_loss": 0.1474570631980896,
      "epoch": 1.544385026737968,
      "loss": 1.2801563739776611,
      "step": 361
    },
    {
      "barlow_loss": 0.1627197265625,
      "epoch": 1.5486631016042782,
      "loss": 1.4706984758377075,
      "step": 362
    },
    {
      "barlow_loss": 0.16179412603378296,
      "epoch": 1.5486631016042782,
      "loss": 1.67202627658844,
      "step": 362
    },
    {
      "barlow_loss": 0.15390506386756897,
      "epoch": 1.5486631016042782,
      "loss": 1.2733176946640015,
      "step": 362
    },
    {
      "barlow_loss": 0.17011652886867523,
      "epoch": 1.5486631016042782,
      "loss": 1.3417762517929077,
      "step": 362
    },
    {
      "barlow_loss": 0.17068564891815186,
      "epoch": 1.5529411764705883,
      "loss": 1.5295343399047852,
      "step": 363
    },
    {
      "barlow_loss": 0.1719181090593338,
      "epoch": 1.5529411764705883,
      "loss": 1.189863920211792,
      "step": 363
    },
    {
      "barlow_loss": 0.14499028027057648,
      "epoch": 1.5529411764705883,
      "loss": 1.3424547910690308,
      "step": 363
    },
    {
      "barlow_loss": 0.1443391740322113,
      "epoch": 1.5529411764705883,
      "loss": 1.4575905799865723,
      "step": 363
    },
    {
      "epoch": 1.5572192513368983,
      "grad_norm": 1.0720150470733643,
      "learning_rate": 3.143615647861235e-06,
      "loss": 1.3975,
      "step": 364
    },
    {
      "barlow_loss": 0.1679568737745285,
      "epoch": 1.5572192513368983,
      "loss": 1.4685271978378296,
      "step": 364
    },
    {
      "barlow_loss": 0.1664041131734848,
      "epoch": 1.5572192513368983,
      "loss": 1.4160844087600708,
      "step": 364
    },
    {
      "barlow_loss": 0.1761854588985443,
      "epoch": 1.5572192513368983,
      "loss": 1.2628246545791626,
      "step": 364
    },
    {
      "barlow_loss": 0.1750512570142746,
      "epoch": 1.5572192513368983,
      "loss": 1.4331152439117432,
      "step": 364
    },
    {
      "barlow_loss": 0.17294731736183167,
      "epoch": 1.5614973262032086,
      "loss": 1.3078454732894897,
      "step": 365
    },
    {
      "barlow_loss": 0.16305863857269287,
      "epoch": 1.5614973262032086,
      "loss": 1.4975590705871582,
      "step": 365
    },
    {
      "barlow_loss": 0.15305112302303314,
      "epoch": 1.5614973262032086,
      "loss": 1.412302851676941,
      "step": 365
    },
    {
      "barlow_loss": 0.1570764183998108,
      "epoch": 1.5614973262032086,
      "loss": 1.2578961849212646,
      "step": 365
    },
    {
      "barlow_loss": 0.1663104146718979,
      "epoch": 1.5657754010695188,
      "loss": 1.323641061782837,
      "step": 366
    },
    {
      "barlow_loss": 0.172359436750412,
      "epoch": 1.5657754010695188,
      "loss": 1.4909641742706299,
      "step": 366
    },
    {
      "barlow_loss": 0.1686282902956009,
      "epoch": 1.5657754010695188,
      "loss": 1.4247713088989258,
      "step": 366
    },
    {
      "barlow_loss": 0.180951789021492,
      "epoch": 1.5657754010695188,
      "loss": 1.3464685678482056,
      "step": 366
    },
    {
      "barlow_loss": 0.1625315546989441,
      "epoch": 1.570053475935829,
      "loss": 1.4414960145950317,
      "step": 367
    },
    {
      "barlow_loss": 0.16339774429798126,
      "epoch": 1.570053475935829,
      "loss": 1.4947307109832764,
      "step": 367
    },
    {
      "barlow_loss": 0.15318085253238678,
      "epoch": 1.570053475935829,
      "loss": 1.4980915784835815,
      "step": 367
    },
    {
      "barlow_loss": 0.16477052867412567,
      "epoch": 1.570053475935829,
      "loss": 1.4868980646133423,
      "step": 367
    },
    {
      "epoch": 1.574331550802139,
      "grad_norm": 1.2551615238189697,
      "learning_rate": 2.9117655055012113e-06,
      "loss": 1.4102,
      "step": 368
    },
    {
      "barlow_loss": 0.157575324177742,
      "epoch": 1.574331550802139,
      "loss": 1.2643206119537354,
      "step": 368
    },
    {
      "barlow_loss": 0.16777503490447998,
      "epoch": 1.574331550802139,
      "loss": 1.4704124927520752,
      "step": 368
    },
    {
      "barlow_loss": 0.1687440574169159,
      "epoch": 1.574331550802139,
      "loss": 1.2841079235076904,
      "step": 368
    },
    {
      "barlow_loss": 0.16948851943016052,
      "epoch": 1.574331550802139,
      "loss": 1.349208116531372,
      "step": 368
    },
    {
      "barlow_loss": 0.16961991786956787,
      "epoch": 1.5786096256684492,
      "loss": 1.3604494333267212,
      "step": 369
    },
    {
      "barlow_loss": 0.1672685146331787,
      "epoch": 1.5786096256684492,
      "loss": 1.436644434928894,
      "step": 369
    },
    {
      "barlow_loss": 0.1749078333377838,
      "epoch": 1.5786096256684492,
      "loss": 1.3998854160308838,
      "step": 369
    },
    {
      "barlow_loss": 0.1790512353181839,
      "epoch": 1.5786096256684492,
      "loss": 1.4035464525222778,
      "step": 369
    },
    {
      "barlow_loss": 0.1642802655696869,
      "epoch": 1.5828877005347595,
      "loss": 1.3676913976669312,
      "step": 370
    },
    {
      "barlow_loss": 0.1614268273115158,
      "epoch": 1.5828877005347595,
      "loss": 1.3691070079803467,
      "step": 370
    },
    {
      "barlow_loss": 0.15366646647453308,
      "epoch": 1.5828877005347595,
      "loss": 1.3518847227096558,
      "step": 370
    },
    {
      "barlow_loss": 0.15351159870624542,
      "epoch": 1.5828877005347595,
      "loss": 1.4477304220199585,
      "step": 370
    },
    {
      "barlow_loss": 0.1686762124300003,
      "epoch": 1.5871657754010695,
      "loss": 1.409849762916565,
      "step": 371
    },
    {
      "barlow_loss": 0.14064490795135498,
      "epoch": 1.5871657754010695,
      "loss": 1.3741589784622192,
      "step": 371
    },
    {
      "barlow_loss": 0.17771050333976746,
      "epoch": 1.5871657754010695,
      "loss": 1.33860445022583,
      "step": 371
    },
    {
      "barlow_loss": 0.1442946344614029,
      "epoch": 1.5871657754010695,
      "loss": 1.2756890058517456,
      "step": 371
    },
    {
      "epoch": 1.5914438502673796,
      "grad_norm": 1.0255159139633179,
      "learning_rate": 2.6876650576388177e-06,
      "loss": 1.369,
      "step": 372
    },
    {
      "barlow_loss": 0.1526157259941101,
      "epoch": 1.5914438502673796,
      "loss": 1.2357839345932007,
      "step": 372
    },
    {
      "barlow_loss": 0.17113520205020905,
      "epoch": 1.5914438502673796,
      "loss": 1.411818265914917,
      "step": 372
    },
    {
      "barlow_loss": 0.17472204566001892,
      "epoch": 1.5914438502673796,
      "loss": 1.4103974103927612,
      "step": 372
    },
    {
      "barlow_loss": 0.16978690028190613,
      "epoch": 1.5914438502673796,
      "loss": 1.501254916191101,
      "step": 372
    },
    {
      "barlow_loss": 0.15966852009296417,
      "epoch": 1.5957219251336898,
      "loss": 1.296275019645691,
      "step": 373
    },
    {
      "barlow_loss": 0.15593025088310242,
      "epoch": 1.5957219251336898,
      "loss": 1.4189201593399048,
      "step": 373
    },
    {
      "barlow_loss": 0.1677265465259552,
      "epoch": 1.5957219251336898,
      "loss": 1.4286787509918213,
      "step": 373
    },
    {
      "barlow_loss": 0.16675616800785065,
      "epoch": 1.5957219251336898,
      "loss": 1.4128416776657104,
      "step": 373
    },
    {
      "barlow_loss": 0.17137204110622406,
      "epoch": 1.6,
      "loss": 1.4485124349594116,
      "step": 374
    },
    {
      "barlow_loss": 0.1564183384180069,
      "epoch": 1.6,
      "loss": 1.4519087076187134,
      "step": 374
    },
    {
      "barlow_loss": 0.16708043217658997,
      "epoch": 1.6,
      "loss": 1.468001127243042,
      "step": 374
    },
    {
      "barlow_loss": 0.15591242909431458,
      "epoch": 1.6,
      "loss": 1.3312674760818481,
      "step": 374
    },
    {
      "barlow_loss": 0.1749807596206665,
      "epoch": 1.6042780748663101,
      "loss": 1.3602885007858276,
      "step": 375
    },
    {
      "barlow_loss": 0.1569051742553711,
      "epoch": 1.6042780748663101,
      "loss": 1.301513671875,
      "step": 375
    },
    {
      "barlow_loss": 0.16656415164470673,
      "epoch": 1.6042780748663101,
      "loss": 1.5982941389083862,
      "step": 375
    },
    {
      "barlow_loss": 0.16418182849884033,
      "epoch": 1.6042780748663101,
      "loss": 1.5541480779647827,
      "step": 375
    },
    {
      "epoch": 1.6085561497326202,
      "grad_norm": 1.1438924074172974,
      "learning_rate": 2.4714954335537734e-06,
      "loss": 1.4144,
      "step": 376
    },
    {
      "barlow_loss": 0.173216313123703,
      "epoch": 1.6085561497326202,
      "loss": 1.4617749452590942,
      "step": 376
    },
    {
      "barlow_loss": 0.15818887948989868,
      "epoch": 1.6085561497326202,
      "loss": 1.4311116933822632,
      "step": 376
    },
    {
      "barlow_loss": 0.17624489963054657,
      "epoch": 1.6085561497326202,
      "loss": 1.2326041460037231,
      "step": 376
    },
    {
      "barlow_loss": 0.16357143223285675,
      "epoch": 1.6085561497326202,
      "loss": 1.3384325504302979,
      "step": 376
    },
    {
      "barlow_loss": 0.17234434187412262,
      "epoch": 1.6128342245989304,
      "loss": 1.4236454963684082,
      "step": 377
    },
    {
      "barlow_loss": 0.15951868891716003,
      "epoch": 1.6128342245989304,
      "loss": 1.396691083908081,
      "step": 377
    },
    {
      "barlow_loss": 0.16298744082450867,
      "epoch": 1.6128342245989304,
      "loss": 1.4536110162734985,
      "step": 377
    },
    {
      "barlow_loss": 0.18090645968914032,
      "epoch": 1.6128342245989304,
      "loss": 1.3767226934432983,
      "step": 377
    },
    {
      "barlow_loss": 0.17449437081813812,
      "epoch": 1.6171122994652407,
      "loss": 1.2070704698562622,
      "step": 378
    },
    {
      "barlow_loss": 0.16491198539733887,
      "epoch": 1.6171122994652407,
      "loss": 1.2830562591552734,
      "step": 378
    },
    {
      "barlow_loss": 0.14166146516799927,
      "epoch": 1.6171122994652407,
      "loss": 1.3600577116012573,
      "step": 378
    },
    {
      "barlow_loss": 0.1555413454771042,
      "epoch": 1.6171122994652407,
      "loss": 1.214347004890442,
      "step": 378
    },
    {
      "barlow_loss": 0.1671699583530426,
      "epoch": 1.6213903743315508,
      "loss": 1.2557001113891602,
      "step": 379
    },
    {
      "barlow_loss": 0.1604607254266739,
      "epoch": 1.6213903743315508,
      "loss": 1.456427812576294,
      "step": 379
    },
    {
      "barlow_loss": 0.15388819575309753,
      "epoch": 1.6213903743315508,
      "loss": 1.347169041633606,
      "step": 379
    },
    {
      "barlow_loss": 0.1759292334318161,
      "epoch": 1.6213903743315508,
      "loss": 1.3987574577331543,
      "step": 379
    },
    {
      "epoch": 1.6256684491978608,
      "grad_norm": 1.03034508228302,
      "learning_rate": 2.263431352434188e-06,
      "loss": 1.3523,
      "step": 380
    },
    {
      "barlow_loss": 0.16835704445838928,
      "epoch": 1.6256684491978608,
      "loss": 1.1492195129394531,
      "step": 380
    },
    {
      "barlow_loss": 0.17722640931606293,
      "epoch": 1.6256684491978608,
      "loss": 1.4321566820144653,
      "step": 380
    },
    {
      "barlow_loss": 0.1641402691602707,
      "epoch": 1.6256684491978608,
      "loss": 1.4275161027908325,
      "step": 380
    },
    {
      "barlow_loss": 0.1566484123468399,
      "epoch": 1.6256684491978608,
      "loss": 1.3477526903152466,
      "step": 380
    },
    {
      "barlow_loss": 0.1518103927373886,
      "epoch": 1.629946524064171,
      "loss": 1.5232163667678833,
      "step": 381
    },
    {
      "barlow_loss": 0.16164250671863556,
      "epoch": 1.629946524064171,
      "loss": 1.4025355577468872,
      "step": 381
    },
    {
      "barlow_loss": 0.17208130657672882,
      "epoch": 1.629946524064171,
      "loss": 1.4915506839752197,
      "step": 381
    },
    {
      "barlow_loss": 0.14882616698741913,
      "epoch": 1.629946524064171,
      "loss": 1.319931149482727,
      "step": 381
    },
    {
      "barlow_loss": 0.18244200944900513,
      "epoch": 1.6342245989304813,
      "loss": 1.3120487928390503,
      "step": 382
    },
    {
      "barlow_loss": 0.1436823308467865,
      "epoch": 1.6342245989304813,
      "loss": 1.3771857023239136,
      "step": 382
    },
    {
      "barlow_loss": 0.16138549149036407,
      "epoch": 1.6342245989304813,
      "loss": 1.4677668809890747,
      "step": 382
    },
    {
      "barlow_loss": 0.1697925627231598,
      "epoch": 1.6342245989304813,
      "loss": 1.3951457738876343,
      "step": 382
    },
    {
      "barlow_loss": 0.16675475239753723,
      "epoch": 1.6385026737967916,
      "loss": 1.5589301586151123,
      "step": 383
    },
    {
      "barlow_loss": 0.16590872406959534,
      "epoch": 1.6385026737967916,
      "loss": 1.469956398010254,
      "step": 383
    },
    {
      "barlow_loss": 0.16392488777637482,
      "epoch": 1.6385026737967916,
      "loss": 1.3021957874298096,
      "step": 383
    },
    {
      "barlow_loss": 0.16398502886295319,
      "epoch": 1.6385026737967916,
      "loss": 1.465620756149292,
      "step": 383
    },
    {
      "epoch": 1.6427807486631016,
      "grad_norm": 1.1530191898345947,
      "learning_rate": 2.063640982159723e-06,
      "loss": 1.4027,
      "step": 384
    },
    {
      "barlow_loss": 0.16786396503448486,
      "epoch": 1.6427807486631016,
      "loss": 1.4176228046417236,
      "step": 384
    },
    {
      "barlow_loss": 0.15633605420589447,
      "epoch": 1.6427807486631016,
      "loss": 1.3450618982315063,
      "step": 384
    },
    {
      "barlow_loss": 0.15378248691558838,
      "epoch": 1.6427807486631016,
      "loss": 1.4359416961669922,
      "step": 384
    },
    {
      "barlow_loss": 0.18276698887348175,
      "epoch": 1.6427807486631016,
      "loss": 1.5559808015823364,
      "step": 384
    },
    {
      "barlow_loss": 0.15952077507972717,
      "epoch": 1.6470588235294117,
      "loss": 1.2603013515472412,
      "step": 385
    },
    {
      "barlow_loss": 0.17662596702575684,
      "epoch": 1.6470588235294117,
      "loss": 1.4517874717712402,
      "step": 385
    },
    {
      "barlow_loss": 0.16596366465091705,
      "epoch": 1.6470588235294117,
      "loss": 1.3880704641342163,
      "step": 385
    },
    {
      "barlow_loss": 0.17036545276641846,
      "epoch": 1.6470588235294117,
      "loss": 1.2811766862869263,
      "step": 385
    },
    {
      "barlow_loss": 0.1716901808977127,
      "epoch": 1.651336898395722,
      "loss": 1.3749245405197144,
      "step": 386
    },
    {
      "barlow_loss": 0.1551070213317871,
      "epoch": 1.651336898395722,
      "loss": 1.2090290784835815,
      "step": 386
    },
    {
      "barlow_loss": 0.155910924077034,
      "epoch": 1.651336898395722,
      "loss": 1.4194480180740356,
      "step": 386
    },
    {
      "barlow_loss": 0.15536466240882874,
      "epoch": 1.651336898395722,
      "loss": 1.367440938949585,
      "step": 386
    },
    {
      "barlow_loss": 0.166548952460289,
      "epoch": 1.6556149732620322,
      "loss": 1.3839596509933472,
      "step": 387
    },
    {
      "barlow_loss": 0.15291725099086761,
      "epoch": 1.6556149732620322,
      "loss": 1.352308750152588,
      "step": 387
    },
    {
      "barlow_loss": 0.1740029901266098,
      "epoch": 1.6556149732620322,
      "loss": 1.4048144817352295,
      "step": 387
    },
    {
      "barlow_loss": 0.16816265881061554,
      "epoch": 1.6556149732620322,
      "loss": 1.5496588945388794,
      "step": 387
    },
    {
      "epoch": 1.6598930481283423,
      "grad_norm": 1.1626052856445312,
      "learning_rate": 1.8722858033798252e-06,
      "loss": 1.3873,
      "step": 388
    },
    {
      "barlow_loss": 0.17609933018684387,
      "epoch": 1.6598930481283423,
      "loss": 1.3661246299743652,
      "step": 388
    },
    {
      "barlow_loss": 0.17116856575012207,
      "epoch": 1.6598930481283423,
      "loss": 1.3081570863723755,
      "step": 388
    },
    {
      "barlow_loss": 0.14901866018772125,
      "epoch": 1.6598930481283423,
      "loss": 1.2951353788375854,
      "step": 388
    },
    {
      "barlow_loss": 0.15819038450717926,
      "epoch": 1.6598930481283423,
      "loss": 1.3281631469726562,
      "step": 388
    },
    {
      "barlow_loss": 0.14251820743083954,
      "epoch": 1.6641711229946523,
      "loss": 1.3255176544189453,
      "step": 389
    },
    {
      "barlow_loss": 0.15690013766288757,
      "epoch": 1.6641711229946523,
      "loss": 1.2541553974151611,
      "step": 389
    },
    {
      "barlow_loss": 0.1663934886455536,
      "epoch": 1.6641711229946523,
      "loss": 1.4123197793960571,
      "step": 389
    },
    {
      "barlow_loss": 0.1581827700138092,
      "epoch": 1.6641711229946523,
      "loss": 1.4310789108276367,
      "step": 389
    },
    {
      "barlow_loss": 0.14762598276138306,
      "epoch": 1.6684491978609626,
      "loss": 1.190331220626831,
      "step": 390
    },
    {
      "barlow_loss": 0.17517639696598053,
      "epoch": 1.6684491978609626,
      "loss": 1.5287715196609497,
      "step": 390
    },
    {
      "barlow_loss": 0.15225434303283691,
      "epoch": 1.6684491978609626,
      "loss": 1.348070740699768,
      "step": 390
    },
    {
      "barlow_loss": 0.17378099262714386,
      "epoch": 1.6684491978609626,
      "loss": 1.3278629779815674,
      "step": 390
    },
    {
      "barlow_loss": 0.158481627702713,
      "epoch": 1.6727272727272728,
      "loss": 1.2417210340499878,
      "step": 391
    },
    {
      "barlow_loss": 0.16470016539096832,
      "epoch": 1.6727272727272728,
      "loss": 1.774024486541748,
      "step": 391
    },
    {
      "barlow_loss": 0.16191494464874268,
      "epoch": 1.6727272727272728,
      "loss": 1.4703539609909058,
      "step": 391
    },
    {
      "barlow_loss": 0.15689443051815033,
      "epoch": 1.6727272727272728,
      "loss": 1.1970077753067017,
      "step": 391
    },
    {
      "epoch": 1.6770053475935829,
      "grad_norm": 1.1270455121994019,
      "learning_rate": 1.6895204789969172e-06,
      "loss": 1.3624,
      "step": 392
    },
    {
      "barlow_loss": 0.14798292517662048,
      "epoch": 1.6770053475935829,
      "loss": 1.552363634109497,
      "step": 392
    },
    {
      "barlow_loss": 0.14308013021945953,
      "epoch": 1.6770053475935829,
      "loss": 1.5563515424728394,
      "step": 392
    },
    {
      "barlow_loss": 0.16878828406333923,
      "epoch": 1.6770053475935829,
      "loss": 1.2435102462768555,
      "step": 392
    },
    {
      "barlow_loss": 0.16219082474708557,
      "epoch": 1.6770053475935829,
      "loss": 1.5148037672042847,
      "step": 392
    },
    {
      "barlow_loss": 0.16858211159706116,
      "epoch": 1.681283422459893,
      "loss": 1.3193047046661377,
      "step": 393
    },
    {
      "barlow_loss": 0.16128690540790558,
      "epoch": 1.681283422459893,
      "loss": 1.5953208208084106,
      "step": 393
    },
    {
      "barlow_loss": 0.15577992796897888,
      "epoch": 1.681283422459893,
      "loss": 1.2455538511276245,
      "step": 393
    },
    {
      "barlow_loss": 0.15956276655197144,
      "epoch": 1.681283422459893,
      "loss": 1.3425744771957397,
      "step": 393
    },
    {
      "barlow_loss": 0.16104869544506073,
      "epoch": 1.6855614973262032,
      "loss": 1.3988099098205566,
      "step": 394
    },
    {
      "barlow_loss": 0.153834730386734,
      "epoch": 1.6855614973262032,
      "loss": 1.2896089553833008,
      "step": 394
    },
    {
      "barlow_loss": 0.15635257959365845,
      "epoch": 1.6855614973262032,
      "loss": 1.3633224964141846,
      "step": 394
    },
    {
      "barlow_loss": 0.15455138683319092,
      "epoch": 1.6855614973262032,
      "loss": 1.3739292621612549,
      "step": 394
    },
    {
      "barlow_loss": 0.1674828678369522,
      "epoch": 1.6898395721925135,
      "loss": 1.3316227197647095,
      "step": 395
    },
    {
      "barlow_loss": 0.1572660654783249,
      "epoch": 1.6898395721925135,
      "loss": 1.3332875967025757,
      "step": 395
    },
    {
      "barlow_loss": 0.14838621020317078,
      "epoch": 1.6898395721925135,
      "loss": 1.2085040807724,
      "step": 395
    },
    {
      "barlow_loss": 0.15926223993301392,
      "epoch": 1.6898395721925135,
      "loss": 1.3298593759536743,
      "step": 395
    },
    {
      "epoch": 1.6941176470588235,
      "grad_norm": 1.029380202293396,
      "learning_rate": 1.515492729160059e-06,
      "loss": 1.3749,
      "step": 396
    },
    {
      "barlow_loss": 0.16278252005577087,
      "epoch": 1.6941176470588235,
      "loss": 1.461343765258789,
      "step": 396
    },
    {
      "barlow_loss": 0.15543845295906067,
      "epoch": 1.6941176470588235,
      "loss": 1.3787341117858887,
      "step": 396
    },
    {
      "barlow_loss": 0.1606270968914032,
      "epoch": 1.6941176470588235,
      "loss": 1.290077805519104,
      "step": 396
    },
    {
      "barlow_loss": 0.1584598273038864,
      "epoch": 1.6941176470588235,
      "loss": 1.3384164571762085,
      "step": 396
    },
    {
      "barlow_loss": 0.1623477339744568,
      "epoch": 1.6983957219251336,
      "loss": 1.4099575281143188,
      "step": 397
    },
    {
      "barlow_loss": 0.15557152032852173,
      "epoch": 1.6983957219251336,
      "loss": 1.5182793140411377,
      "step": 397
    },
    {
      "barlow_loss": 0.15491555631160736,
      "epoch": 1.6983957219251336,
      "loss": 1.3144164085388184,
      "step": 397
    },
    {
      "barlow_loss": 0.168035089969635,
      "epoch": 1.6983957219251336,
      "loss": 1.4764175415039062,
      "step": 397
    },
    {
      "barlow_loss": 0.16057038307189941,
      "epoch": 1.7026737967914438,
      "loss": 1.3558040857315063,
      "step": 398
    },
    {
      "barlow_loss": 0.15944252908229828,
      "epoch": 1.7026737967914438,
      "loss": 1.4717576503753662,
      "step": 398
    },
    {
      "barlow_loss": 0.1671985387802124,
      "epoch": 1.7026737967914438,
      "loss": 1.3252650499343872,
      "step": 398
    },
    {
      "barlow_loss": 0.1533321738243103,
      "epoch": 1.7026737967914438,
      "loss": 1.3123072385787964,
      "step": 398
    },
    {
      "barlow_loss": 0.1603221595287323,
      "epoch": 1.706951871657754,
      "loss": 1.51680588722229,
      "step": 399
    },
    {
      "barlow_loss": 0.1749984323978424,
      "epoch": 1.706951871657754,
      "loss": 1.524340271949768,
      "step": 399
    },
    {
      "barlow_loss": 0.14842191338539124,
      "epoch": 1.706951871657754,
      "loss": 1.1328927278518677,
      "step": 399
    },
    {
      "barlow_loss": 0.16389159858226776,
      "epoch": 1.706951871657754,
      "loss": 1.4428167343139648,
      "step": 399
    },
    {
      "epoch": 1.7112299465240641,
      "grad_norm": 1.0679799318313599,
      "learning_rate": 1.3503432118700711e-06,
      "loss": 1.3919,
      "step": 400
    },
    {
      "barlow_loss": 0.15939602255821228,
      "epoch": 1.7112299465240641,
      "loss": 1.5552315711975098,
      "step": 400
    },
    {
      "barlow_loss": 0.16922439634799957,
      "epoch": 1.7112299465240641,
      "loss": 1.2601591348648071,
      "step": 400
    },
    {
      "barlow_loss": 0.1732679009437561,
      "epoch": 1.7112299465240641,
      "loss": 1.206969141960144,
      "step": 400
    },
    {
      "barlow_loss": 0.1727072149515152,
      "epoch": 1.7112299465240641,
      "loss": 1.4327280521392822,
      "step": 400
    },
    {
      "barlow_loss": 0.1754273921251297,
      "epoch": 1.7155080213903742,
      "loss": 1.30107843875885,
      "step": 401
    },
    {
      "barlow_loss": 0.1683354675769806,
      "epoch": 1.7155080213903742,
      "loss": 1.4055986404418945,
      "step": 401
    },
    {
      "barlow_loss": 0.15949305891990662,
      "epoch": 1.7155080213903742,
      "loss": 1.5220601558685303,
      "step": 401
    },
    {
      "barlow_loss": 0.17098820209503174,
      "epoch": 1.7155080213903742,
      "loss": 1.3282110691070557,
      "step": 401
    },
    {
      "barlow_loss": 0.14831982553005219,
      "epoch": 1.7197860962566844,
      "loss": 1.500718355178833,
      "step": 402
    },
    {
      "barlow_loss": 0.15733696520328522,
      "epoch": 1.7197860962566844,
      "loss": 1.4870494604110718,
      "step": 402
    },
    {
      "barlow_loss": 0.16531358659267426,
      "epoch": 1.7197860962566844,
      "loss": 1.5199934244155884,
      "step": 402
    },
    {
      "barlow_loss": 0.1672108769416809,
      "epoch": 1.7197860962566844,
      "loss": 1.348727822303772,
      "step": 402
    },
    {
      "barlow_loss": 0.15615926682949066,
      "epoch": 1.7240641711229947,
      "loss": 1.257646083831787,
      "step": 403
    },
    {
      "barlow_loss": 0.16075405478477478,
      "epoch": 1.7240641711229947,
      "loss": 1.2393406629562378,
      "step": 403
    },
    {
      "barlow_loss": 0.1627914011478424,
      "epoch": 1.7240641711229947,
      "loss": 1.3047970533370972,
      "step": 403
    },
    {
      "barlow_loss": 0.16783668100833893,
      "epoch": 1.7240641711229947,
      "loss": 1.2933441400527954,
      "step": 403
    },
    {
      "epoch": 1.728342245989305,
      "grad_norm": 1.0024456977844238,
      "learning_rate": 1.1942054092926314e-06,
      "loss": 1.3727,
      "step": 404
    },
    {
      "barlow_loss": 0.16516007483005524,
      "epoch": 1.728342245989305,
      "loss": 1.5432723760604858,
      "step": 404
    },
    {
      "barlow_loss": 0.16395601630210876,
      "epoch": 1.728342245989305,
      "loss": 1.299622654914856,
      "step": 404
    },
    {
      "barlow_loss": 0.15898412466049194,
      "epoch": 1.728342245989305,
      "loss": 1.3342903852462769,
      "step": 404
    },
    {
      "barlow_loss": 0.16306167840957642,
      "epoch": 1.728342245989305,
      "loss": 1.4156981706619263,
      "step": 404
    },
    {
      "barlow_loss": 0.1641940325498581,
      "epoch": 1.732620320855615,
      "loss": 1.3492963314056396,
      "step": 405
    },
    {
      "barlow_loss": 0.17035555839538574,
      "epoch": 1.732620320855615,
      "loss": 1.4193763732910156,
      "step": 405
    },
    {
      "barlow_loss": 0.16826830804347992,
      "epoch": 1.732620320855615,
      "loss": 1.6344983577728271,
      "step": 405
    },
    {
      "barlow_loss": 0.14831574261188507,
      "epoch": 1.732620320855615,
      "loss": 1.4098659753799438,
      "step": 405
    },
    {
      "barlow_loss": 0.15615986287593842,
      "epoch": 1.736898395721925,
      "loss": 1.3630667924880981,
      "step": 406
    },
    {
      "barlow_loss": 0.15846866369247437,
      "epoch": 1.736898395721925,
      "loss": 1.3692694902420044,
      "step": 406
    },
    {
      "barlow_loss": 0.1694820672273636,
      "epoch": 1.736898395721925,
      "loss": 1.5564035177230835,
      "step": 406
    },
    {
      "barlow_loss": 0.15148766338825226,
      "epoch": 1.736898395721925,
      "loss": 1.3442217111587524,
      "step": 406
    },
    {
      "barlow_loss": 0.16400295495986938,
      "epoch": 1.7411764705882353,
      "loss": 1.4945766925811768,
      "step": 407
    },
    {
      "barlow_loss": 0.14815038442611694,
      "epoch": 1.7411764705882353,
      "loss": 1.2842788696289062,
      "step": 407
    },
    {
      "barlow_loss": 0.1695544570684433,
      "epoch": 1.7411764705882353,
      "loss": 1.440415382385254,
      "step": 407
    },
    {
      "barlow_loss": 0.16857239603996277,
      "epoch": 1.7411764705882353,
      "loss": 1.3451112508773804,
      "step": 407
    },
    {
      "epoch": 1.7454545454545456,
      "grad_norm": 1.03217613697052,
      "learning_rate": 1.0472055198712847e-06,
      "loss": 1.4127,
      "step": 408
    },
    {
      "barlow_loss": 0.17281591892242432,
      "epoch": 1.7454545454545456,
      "loss": 1.4039658308029175,
      "step": 408
    },
    {
      "barlow_loss": 0.16403216123580933,
      "epoch": 1.7454545454545456,
      "loss": 1.339292049407959,
      "step": 408
    },
    {
      "barlow_loss": 0.15488186478614807,
      "epoch": 1.7454545454545456,
      "loss": 1.5495725870132446,
      "step": 408
    },
    {
      "barlow_loss": 0.1655692458152771,
      "epoch": 1.7454545454545456,
      "loss": 1.3290252685546875,
      "step": 408
    },
    {
      "barlow_loss": 0.18152844905853271,
      "epoch": 1.7497326203208556,
      "loss": 1.489258885383606,
      "step": 409
    },
    {
      "barlow_loss": 0.16344943642616272,
      "epoch": 1.7497326203208556,
      "loss": 1.3796608448028564,
      "step": 409
    },
    {
      "barlow_loss": 0.1711360514163971,
      "epoch": 1.7497326203208556,
      "loss": 1.3534148931503296,
      "step": 409
    },
    {
      "barlow_loss": 0.15475502610206604,
      "epoch": 1.7497326203208556,
      "loss": 1.444381594657898,
      "step": 409
    },
    {
      "barlow_loss": 0.1553666591644287,
      "epoch": 1.7540106951871657,
      "loss": 1.2650096416473389,
      "step": 410
    },
    {
      "barlow_loss": 0.17246565222740173,
      "epoch": 1.7540106951871657,
      "loss": 1.1668823957443237,
      "step": 410
    },
    {
      "barlow_loss": 0.14898210763931274,
      "epoch": 1.7540106951871657,
      "loss": 1.2542661428451538,
      "step": 410
    },
    {
      "barlow_loss": 0.17206402122974396,
      "epoch": 1.7540106951871657,
      "loss": 1.5133916139602661,
      "step": 410
    },
    {
      "barlow_loss": 0.15988557040691376,
      "epoch": 1.758288770053476,
      "loss": 1.3240889310836792,
      "step": 411
    },
    {
      "barlow_loss": 0.15356232225894928,
      "epoch": 1.758288770053476,
      "loss": 1.2029759883880615,
      "step": 411
    },
    {
      "barlow_loss": 0.17232130467891693,
      "epoch": 1.758288770053476,
      "loss": 1.202345609664917,
      "step": 411
    },
    {
      "barlow_loss": 0.1556706726551056,
      "epoch": 1.758288770053476,
      "loss": 1.283949613571167,
      "step": 411
    },
    {
      "epoch": 1.7625668449197862,
      "grad_norm": 0.9563727974891663,
      "learning_rate": 9.094623563274826e-07,
      "loss": 1.3438,
      "step": 412
    },
    {
      "barlow_loss": 0.17084215581417084,
      "epoch": 1.7625668449197862,
      "loss": 1.5121862888336182,
      "step": 412
    },
    {
      "barlow_loss": 0.17319415509700775,
      "epoch": 1.7625668449197862,
      "loss": 1.3660292625427246,
      "step": 412
    },
    {
      "barlow_loss": 0.16012650728225708,
      "epoch": 1.7625668449197862,
      "loss": 1.2595776319503784,
      "step": 412
    },
    {
      "barlow_loss": 0.14574049413204193,
      "epoch": 1.7625668449197862,
      "loss": 1.284920573234558,
      "step": 412
    },
    {
      "barlow_loss": 0.15982407331466675,
      "epoch": 1.7668449197860963,
      "loss": 1.2304530143737793,
      "step": 413
    },
    {
      "barlow_loss": 0.15305015444755554,
      "epoch": 1.7668449197860963,
      "loss": 1.2373696565628052,
      "step": 413
    },
    {
      "barlow_loss": 0.14062891900539398,
      "epoch": 1.7668449197860963,
      "loss": 1.3538380861282349,
      "step": 413
    },
    {
      "barlow_loss": 0.16972672939300537,
      "epoch": 1.7668449197860963,
      "loss": 1.4071924686431885,
      "step": 413
    },
    {
      "barlow_loss": 0.16866710782051086,
      "epoch": 1.7711229946524063,
      "loss": 1.333011269569397,
      "step": 414
    },
    {
      "barlow_loss": 0.16293808817863464,
      "epoch": 1.7711229946524063,
      "loss": 1.56315279006958,
      "step": 414
    },
    {
      "barlow_loss": 0.15919475257396698,
      "epoch": 1.7711229946524063,
      "loss": 1.527635931968689,
      "step": 414
    },
    {
      "barlow_loss": 0.16585861146450043,
      "epoch": 1.7711229946524063,
      "loss": 1.4565227031707764,
      "step": 414
    },
    {
      "barlow_loss": 0.16983820497989655,
      "epoch": 1.7754010695187166,
      "loss": 1.3599557876586914,
      "step": 415
    },
    {
      "barlow_loss": 0.14274363219738007,
      "epoch": 1.7754010695187166,
      "loss": 1.301833987236023,
      "step": 415
    },
    {
      "barlow_loss": 0.16460129618644714,
      "epoch": 1.7754010695187166,
      "loss": 1.2684136629104614,
      "step": 415
    },
    {
      "barlow_loss": 0.15501190721988678,
      "epoch": 1.7754010695187166,
      "loss": 1.2475236654281616,
      "step": 415
    },
    {
      "epoch": 1.7796791443850268,
      "grad_norm": 1.0249156951904297,
      "learning_rate": 7.810872496301524e-07,
      "loss": 1.3569,
      "step": 416
    },
    {
      "barlow_loss": 0.15065720677375793,
      "epoch": 1.7796791443850268,
      "loss": 1.3945598602294922,
      "step": 416
    },
    {
      "barlow_loss": 0.15741419792175293,
      "epoch": 1.7796791443850268,
      "loss": 1.4255152940750122,
      "step": 416
    },
    {
      "barlow_loss": 0.155485600233078,
      "epoch": 1.7796791443850268,
      "loss": 1.1921899318695068,
      "step": 416
    },
    {
      "barlow_loss": 0.16558222472667694,
      "epoch": 1.7796791443850268,
      "loss": 1.3873965740203857,
      "step": 416
    },
    {
      "barlow_loss": 0.1676730215549469,
      "epoch": 1.7839572192513369,
      "loss": 1.2166274785995483,
      "step": 417
    },
    {
      "barlow_loss": 0.17496958374977112,
      "epoch": 1.7839572192513369,
      "loss": 1.3131225109100342,
      "step": 417
    },
    {
      "barlow_loss": 0.16831432282924652,
      "epoch": 1.7839572192513369,
      "loss": 1.4553680419921875,
      "step": 417
    },
    {
      "barlow_loss": 0.16818250715732574,
      "epoch": 1.7839572192513369,
      "loss": 1.3309426307678223,
      "step": 417
    },
    {
      "barlow_loss": 0.17825847864151,
      "epoch": 1.788235294117647,
      "loss": 1.5263365507125854,
      "step": 418
    },
    {
      "barlow_loss": 0.15922054648399353,
      "epoch": 1.788235294117647,
      "loss": 1.3866283893585205,
      "step": 418
    },
    {
      "barlow_loss": 0.16675850749015808,
      "epoch": 1.788235294117647,
      "loss": 1.3743480443954468,
      "step": 418
    },
    {
      "barlow_loss": 0.15700682997703552,
      "epoch": 1.788235294117647,
      "loss": 1.0873676538467407,
      "step": 418
    },
    {
      "barlow_loss": 0.17780181765556335,
      "epoch": 1.7925133689839572,
      "loss": 1.2911404371261597,
      "step": 419
    },
    {
      "barlow_loss": 0.14515253901481628,
      "epoch": 1.7925133689839572,
      "loss": 1.372534155845642,
      "step": 419
    },
    {
      "barlow_loss": 0.1662854105234146,
      "epoch": 1.7925133689839572,
      "loss": 1.405494213104248,
      "step": 419
    },
    {
      "barlow_loss": 0.14923623204231262,
      "epoch": 1.7925133689839572,
      "loss": 1.4424785375595093,
      "step": 419
    },
    {
      "epoch": 1.7967914438502675,
      "grad_norm": 1.30230712890625,
      "learning_rate": 6.621839590124038e-07,
      "loss": 1.3501,
      "step": 420
    },
    {
      "barlow_loss": 0.16960585117340088,
      "epoch": 1.7967914438502675,
      "loss": 1.3814178705215454,
      "step": 420
    },
    {
      "barlow_loss": 0.14774709939956665,
      "epoch": 1.7967914438502675,
      "loss": 1.37383234500885,
      "step": 420
    },
    {
      "barlow_loss": 0.16195060312747955,
      "epoch": 1.7967914438502675,
      "loss": 1.4655914306640625,
      "step": 420
    },
    {
      "barlow_loss": 0.14820007979869843,
      "epoch": 1.7967914438502675,
      "loss": 1.4735932350158691,
      "step": 420
    },
    {
      "barlow_loss": 0.16983260214328766,
      "epoch": 1.8010695187165775,
      "loss": 1.3141541481018066,
      "step": 421
    },
    {
      "barlow_loss": 0.15996326506137848,
      "epoch": 1.8010695187165775,
      "loss": 1.4228096008300781,
      "step": 421
    },
    {
      "barlow_loss": 0.1567988097667694,
      "epoch": 1.8010695187165775,
      "loss": 1.395566701889038,
      "step": 421
    },
    {
      "barlow_loss": 0.17252366244792938,
      "epoch": 1.8010695187165775,
      "loss": 1.1833546161651611,
      "step": 421
    },
    {
      "barlow_loss": 0.15606877207756042,
      "epoch": 1.8053475935828875,
      "loss": 1.3784117698669434,
      "step": 422
    },
    {
      "barlow_loss": 0.16858474910259247,
      "epoch": 1.8053475935828875,
      "loss": 1.210524320602417,
      "step": 422
    },
    {
      "barlow_loss": 0.17947521805763245,
      "epoch": 1.8053475935828875,
      "loss": 1.478119134902954,
      "step": 422
    },
    {
      "barlow_loss": 0.16237857937812805,
      "epoch": 1.8053475935828875,
      "loss": 1.528372883796692,
      "step": 422
    },
    {
      "barlow_loss": 0.17458601295948029,
      "epoch": 1.8096256684491978,
      "loss": 1.311232328414917,
      "step": 423
    },
    {
      "barlow_loss": 0.15704458951950073,
      "epoch": 1.8096256684491978,
      "loss": 1.4745362997055054,
      "step": 423
    },
    {
      "barlow_loss": 0.1592562347650528,
      "epoch": 1.8096256684491978,
      "loss": 1.2915983200073242,
      "step": 423
    },
    {
      "barlow_loss": 0.16271668672561646,
      "epoch": 1.8096256684491978,
      "loss": 1.5111236572265625,
      "step": 423
    },
    {
      "epoch": 1.813903743315508,
      "grad_norm": 1.1478102207183838,
      "learning_rate": 5.528485881080792e-07,
      "loss": 1.3871,
      "step": 424
    },
    {
      "barlow_loss": 0.15936847031116486,
      "epoch": 1.813903743315508,
      "loss": 1.5197679996490479,
      "step": 424
    },
    {
      "barlow_loss": 0.16570886969566345,
      "epoch": 1.813903743315508,
      "loss": 1.4400067329406738,
      "step": 424
    },
    {
      "barlow_loss": 0.15656672418117523,
      "epoch": 1.813903743315508,
      "loss": 1.4901524782180786,
      "step": 424
    },
    {
      "barlow_loss": 0.1636146754026413,
      "epoch": 1.813903743315508,
      "loss": 1.3913706541061401,
      "step": 424
    },
    {
      "barlow_loss": 0.1806619018316269,
      "epoch": 1.8181818181818183,
      "loss": 1.3617385625839233,
      "step": 425
    },
    {
      "barlow_loss": 0.18005633354187012,
      "epoch": 1.8181818181818183,
      "loss": 1.6238442659378052,
      "step": 425
    },
    {
      "barlow_loss": 0.1742164045572281,
      "epoch": 1.8181818181818183,
      "loss": 1.2659193277359009,
      "step": 425
    },
    {
      "barlow_loss": 0.17248517274856567,
      "epoch": 1.8181818181818183,
      "loss": 1.5544971227645874,
      "step": 425
    },
    {
      "barlow_loss": 0.17481553554534912,
      "epoch": 1.8224598930481284,
      "loss": 1.2903965711593628,
      "step": 426
    },
    {
      "barlow_loss": 0.1589963585138321,
      "epoch": 1.8224598930481284,
      "loss": 1.2440621852874756,
      "step": 426
    },
    {
      "barlow_loss": 0.17480415105819702,
      "epoch": 1.8224598930481284,
      "loss": 1.5345280170440674,
      "step": 426
    },
    {
      "barlow_loss": 0.17546191811561584,
      "epoch": 1.8224598930481284,
      "loss": 1.5880253314971924,
      "step": 426
    },
    {
      "barlow_loss": 0.14145585894584656,
      "epoch": 1.8267379679144384,
      "loss": 1.4113825559616089,
      "step": 427
    },
    {
      "barlow_loss": 0.15658918023109436,
      "epoch": 1.8267379679144384,
      "loss": 1.2842826843261719,
      "step": 427
    },
    {
      "barlow_loss": 0.16662566363811493,
      "epoch": 1.8267379679144384,
      "loss": 1.2669305801391602,
      "step": 427
    },
    {
      "barlow_loss": 0.1586907058954239,
      "epoch": 1.8267379679144384,
      "loss": 1.3836358785629272,
      "step": 427
    },
    {
      "epoch": 1.8310160427807487,
      "grad_norm": 1.0529863834381104,
      "learning_rate": 4.5316950727595093e-07,
      "loss": 1.4157,
      "step": 428
    },
    {
      "barlow_loss": 0.1720731407403946,
      "epoch": 1.8310160427807487,
      "loss": 1.441510558128357,
      "step": 428
    },
    {
      "barlow_loss": 0.16999275982379913,
      "epoch": 1.8310160427807487,
      "loss": 1.3116719722747803,
      "step": 428
    },
    {
      "barlow_loss": 0.16748104989528656,
      "epoch": 1.8310160427807487,
      "loss": 1.4294607639312744,
      "step": 428
    },
    {
      "barlow_loss": 0.15915772318840027,
      "epoch": 1.8310160427807487,
      "loss": 1.3410860300064087,
      "step": 428
    },
    {
      "barlow_loss": 0.17730817198753357,
      "epoch": 1.835294117647059,
      "loss": 1.4024600982666016,
      "step": 429
    },
    {
      "barlow_loss": 0.18273009359836578,
      "epoch": 1.835294117647059,
      "loss": 1.3510934114456177,
      "step": 429
    },
    {
      "barlow_loss": 0.15344373881816864,
      "epoch": 1.835294117647059,
      "loss": 1.239652395248413,
      "step": 429
    },
    {
      "barlow_loss": 0.1650843769311905,
      "epoch": 1.835294117647059,
      "loss": 1.5234166383743286,
      "step": 429
    },
    {
      "barlow_loss": 0.1706102192401886,
      "epoch": 1.839572192513369,
      "loss": 1.4199060201644897,
      "step": 430
    },
    {
      "barlow_loss": 0.16890546679496765,
      "epoch": 1.839572192513369,
      "loss": 1.3037376403808594,
      "step": 430
    },
    {
      "barlow_loss": 0.1856047511100769,
      "epoch": 1.839572192513369,
      "loss": 1.1983107328414917,
      "step": 430
    },
    {
      "barlow_loss": 0.17823375761508942,
      "epoch": 1.839572192513369,
      "loss": 1.3563992977142334,
      "step": 430
    },
    {
      "barlow_loss": 0.166975200176239,
      "epoch": 1.843850267379679,
      "loss": 1.3576501607894897,
      "step": 431
    },
    {
      "barlow_loss": 0.16448426246643066,
      "epoch": 1.843850267379679,
      "loss": 1.317081093788147,
      "step": 431
    },
    {
      "barlow_loss": 0.1719483584165573,
      "epoch": 1.843850267379679,
      "loss": 1.3385889530181885,
      "step": 431
    },
    {
      "barlow_loss": 0.17551544308662415,
      "epoch": 1.843850267379679,
      "loss": 1.398679494857788,
      "step": 431
    },
    {
      "epoch": 1.8481283422459893,
      "grad_norm": 1.0822070837020874,
      "learning_rate": 3.6322728217434986e-07,
      "loss": 1.3582,
      "step": 432
    },
    {
      "barlow_loss": 0.16266722977161407,
      "epoch": 1.8481283422459893,
      "loss": 1.321031093597412,
      "step": 432
    },
    {
      "barlow_loss": 0.17089787125587463,
      "epoch": 1.8481283422459893,
      "loss": 1.7040897607803345,
      "step": 432
    },
    {
      "barlow_loss": 0.1520431786775589,
      "epoch": 1.8481283422459893,
      "loss": 1.2001675367355347,
      "step": 432
    },
    {
      "barlow_loss": 0.1505565643310547,
      "epoch": 1.8481283422459893,
      "loss": 1.3719834089279175,
      "step": 432
    },
    {
      "barlow_loss": 0.172381192445755,
      "epoch": 1.8524064171122996,
      "loss": 1.4091147184371948,
      "step": 433
    },
    {
      "barlow_loss": 0.14783978462219238,
      "epoch": 1.8524064171122996,
      "loss": 1.4364012479782104,
      "step": 433
    },
    {
      "barlow_loss": 0.17159390449523926,
      "epoch": 1.8524064171122996,
      "loss": 1.3571823835372925,
      "step": 433
    },
    {
      "barlow_loss": 0.17819802463054657,
      "epoch": 1.8524064171122996,
      "loss": 1.4939842224121094,
      "step": 433
    },
    {
      "barlow_loss": 0.16560570895671844,
      "epoch": 1.8566844919786096,
      "loss": 1.2691946029663086,
      "step": 434
    },
    {
      "barlow_loss": 0.15496137738227844,
      "epoch": 1.8566844919786096,
      "loss": 1.31818425655365,
      "step": 434
    },
    {
      "barlow_loss": 0.15619927644729614,
      "epoch": 1.8566844919786096,
      "loss": 1.4125854969024658,
      "step": 434
    },
    {
      "barlow_loss": 0.18037351965904236,
      "epoch": 1.8566844919786096,
      "loss": 1.418508529663086,
      "step": 434
    },
    {
      "barlow_loss": 0.17507509887218475,
      "epoch": 1.8609625668449197,
      "loss": 1.5171958208084106,
      "step": 435
    },
    {
      "barlow_loss": 0.1650359332561493,
      "epoch": 1.8609625668449197,
      "loss": 1.404914140701294,
      "step": 435
    },
    {
      "barlow_loss": 0.16647624969482422,
      "epoch": 1.8609625668449197,
      "loss": 1.334757685661316,
      "step": 435
    },
    {
      "barlow_loss": 0.17874455451965332,
      "epoch": 1.8609625668449197,
      "loss": 1.2954760789871216,
      "step": 435
    },
    {
      "epoch": 1.86524064171123,
      "grad_norm": 1.115835428237915,
      "learning_rate": 2.8309460864393013e-07,
      "loss": 1.3915,
      "step": 436
    },
    {
      "barlow_loss": 0.1628563404083252,
      "epoch": 1.86524064171123,
      "loss": 1.3809983730316162,
      "step": 436
    },
    {
      "barlow_loss": 0.16664762794971466,
      "epoch": 1.86524064171123,
      "loss": 1.460471272468567,
      "step": 436
    },
    {
      "barlow_loss": 0.17857210338115692,
      "epoch": 1.86524064171123,
      "loss": 1.4650263786315918,
      "step": 436
    },
    {
      "barlow_loss": 0.16135229170322418,
      "epoch": 1.86524064171123,
      "loss": 1.3414744138717651,
      "step": 436
    },
    {
      "barlow_loss": 0.1739460974931717,
      "epoch": 1.8695187165775402,
      "loss": 1.2668609619140625,
      "step": 437
    },
    {
      "barlow_loss": 0.14082594215869904,
      "epoch": 1.8695187165775402,
      "loss": 1.1924322843551636,
      "step": 437
    },
    {
      "barlow_loss": 0.15618571639060974,
      "epoch": 1.8695187165775402,
      "loss": 1.4051692485809326,
      "step": 437
    },
    {
      "barlow_loss": 0.15521842241287231,
      "epoch": 1.8695187165775402,
      "loss": 1.4607276916503906,
      "step": 437
    },
    {
      "barlow_loss": 0.17388206720352173,
      "epoch": 1.8737967914438503,
      "loss": 1.2946914434432983,
      "step": 438
    },
    {
      "barlow_loss": 0.16483049094676971,
      "epoch": 1.8737967914438503,
      "loss": 1.4655728340148926,
      "step": 438
    },
    {
      "barlow_loss": 0.16544097661972046,
      "epoch": 1.8737967914438503,
      "loss": 1.4449812173843384,
      "step": 438
    },
    {
      "barlow_loss": 0.1563642919063568,
      "epoch": 1.8737967914438503,
      "loss": 1.4067662954330444,
      "step": 438
    },
    {
      "barlow_loss": 0.15755021572113037,
      "epoch": 1.8780748663101603,
      "loss": 1.3757147789001465,
      "step": 439
    },
    {
      "barlow_loss": 0.16755035519599915,
      "epoch": 1.8780748663101603,
      "loss": 1.4043265581130981,
      "step": 439
    },
    {
      "barlow_loss": 0.15041443705558777,
      "epoch": 1.8780748663101603,
      "loss": 1.2304072380065918,
      "step": 439
    },
    {
      "barlow_loss": 0.17428869009017944,
      "epoch": 1.8780748663101603,
      "loss": 1.2432490587234497,
      "step": 439
    },
    {
      "epoch": 1.8823529411764706,
      "grad_norm": 1.1069031953811646,
      "learning_rate": 2.128362539512277e-07,
      "loss": 1.3649,
      "step": 440
    },
    {
      "barlow_loss": 0.16325390338897705,
      "epoch": 1.8823529411764706,
      "loss": 1.2893980741500854,
      "step": 440
    },
    {
      "barlow_loss": 0.16411802172660828,
      "epoch": 1.8823529411764706,
      "loss": 1.3695212602615356,
      "step": 440
    },
    {
      "barlow_loss": 0.17526306211948395,
      "epoch": 1.8823529411764706,
      "loss": 1.5280025005340576,
      "step": 440
    },
    {
      "barlow_loss": 0.15608230233192444,
      "epoch": 1.8823529411764706,
      "loss": 1.310736060142517,
      "step": 440
    },
    {
      "barlow_loss": 0.16645421087741852,
      "epoch": 1.8866310160427808,
      "loss": 1.2182164192199707,
      "step": 441
    },
    {
      "barlow_loss": 0.1753074824810028,
      "epoch": 1.8866310160427808,
      "loss": 1.4205093383789062,
      "step": 441
    },
    {
      "barlow_loss": 0.16718527674674988,
      "epoch": 1.8866310160427808,
      "loss": 1.4247838258743286,
      "step": 441
    },
    {
      "barlow_loss": 0.17607681453227997,
      "epoch": 1.8866310160427808,
      "loss": 1.5066057443618774,
      "step": 441
    },
    {
      "barlow_loss": 0.16328537464141846,
      "epoch": 1.8909090909090909,
      "loss": 1.470837950706482,
      "step": 442
    },
    {
      "barlow_loss": 0.14898093044757843,
      "epoch": 1.8909090909090909,
      "loss": 1.2826038599014282,
      "step": 442
    },
    {
      "barlow_loss": 0.16837602853775024,
      "epoch": 1.8909090909090909,
      "loss": 1.2406951189041138,
      "step": 442
    },
    {
      "barlow_loss": 0.16086435317993164,
      "epoch": 1.8909090909090909,
      "loss": 1.695623517036438,
      "step": 442
    },
    {
      "barlow_loss": 0.1747979074716568,
      "epoch": 1.895187165775401,
      "loss": 1.4319769144058228,
      "step": 443
    },
    {
      "barlow_loss": 0.16076907515525818,
      "epoch": 1.895187165775401,
      "loss": 1.2816025018692017,
      "step": 443
    },
    {
      "barlow_loss": 0.15084120631217957,
      "epoch": 1.895187165775401,
      "loss": 1.3048993349075317,
      "step": 443
    },
    {
      "barlow_loss": 0.17147251963615417,
      "epoch": 1.895187165775401,
      "loss": 1.2905192375183105,
      "step": 443
    },
    {
      "epoch": 1.8994652406417112,
      "grad_norm": 1.0322579145431519,
      "learning_rate": 1.5250900444049866e-07,
      "loss": 1.3792,
      "step": 444
    },
    {
      "barlow_loss": 0.17447282373905182,
      "epoch": 1.8994652406417112,
      "loss": 1.5770343542099,
      "step": 444
    },
    {
      "barlow_loss": 0.16736336052417755,
      "epoch": 1.8994652406417112,
      "loss": 1.3721215724945068,
      "step": 444
    },
    {
      "barlow_loss": 0.1544693410396576,
      "epoch": 1.8994652406417112,
      "loss": 1.4035537242889404,
      "step": 444
    },
    {
      "barlow_loss": 0.15957629680633545,
      "epoch": 1.8994652406417112,
      "loss": 1.387835144996643,
      "step": 444
    },
    {
      "barlow_loss": 0.15478640794754028,
      "epoch": 1.9037433155080214,
      "loss": 1.3874608278274536,
      "step": 445
    },
    {
      "barlow_loss": 0.15238119661808014,
      "epoch": 1.9037433155080214,
      "loss": 1.3614667654037476,
      "step": 445
    },
    {
      "barlow_loss": 0.15329617261886597,
      "epoch": 1.9037433155080214,
      "loss": 1.3083593845367432,
      "step": 445
    },
    {
      "barlow_loss": 0.15421874821186066,
      "epoch": 1.9037433155080214,
      "loss": 1.339473843574524,
      "step": 445
    },
    {
      "barlow_loss": 0.17825180292129517,
      "epoch": 1.9080213903743317,
      "loss": 1.439695954322815,
      "step": 446
    },
    {
      "barlow_loss": 0.1673332154750824,
      "epoch": 1.9080213903743317,
      "loss": 1.349808692932129,
      "step": 446
    },
    {
      "barlow_loss": 0.1490553319454193,
      "epoch": 1.9080213903743317,
      "loss": 1.4288618564605713,
      "step": 446
    },
    {
      "barlow_loss": 0.17918585240840912,
      "epoch": 1.9080213903743317,
      "loss": 1.2857377529144287,
      "step": 446
    },
    {
      "barlow_loss": 0.1642368733882904,
      "epoch": 1.9122994652406418,
      "loss": 1.3211535215377808,
      "step": 447
    },
    {
      "barlow_loss": 0.1739424616098404,
      "epoch": 1.9122994652406418,
      "loss": 1.3765318393707275,
      "step": 447
    },
    {
      "barlow_loss": 0.17706362903118134,
      "epoch": 1.9122994652406418,
      "loss": 1.254636287689209,
      "step": 447
    },
    {
      "barlow_loss": 0.17940019071102142,
      "epoch": 1.9122994652406418,
      "loss": 1.3884583711624146,
      "step": 447
    },
    {
      "epoch": 1.9165775401069518,
      "grad_norm": 1.0510393381118774,
      "learning_rate": 1.0216161963611942e-07,
      "loss": 1.3739,
      "step": 448
    },
    {
      "barlow_loss": 0.1598195880651474,
      "epoch": 1.9165775401069518,
      "loss": 1.3731080293655396,
      "step": 448
    },
    {
      "barlow_loss": 0.1556139886379242,
      "epoch": 1.9165775401069518,
      "loss": 1.3278557062149048,
      "step": 448
    },
    {
      "barlow_loss": 0.16333264112472534,
      "epoch": 1.9165775401069518,
      "loss": 1.2913800477981567,
      "step": 448
    },
    {
      "barlow_loss": 0.1553560048341751,
      "epoch": 1.9165775401069518,
      "loss": 1.2532906532287598,
      "step": 448
    },
    {
      "barlow_loss": 0.16787652671337128,
      "epoch": 1.920855614973262,
      "loss": 1.382952332496643,
      "step": 449
    },
    {
      "barlow_loss": 0.15452200174331665,
      "epoch": 1.920855614973262,
      "loss": 1.3713843822479248,
      "step": 449
    },
    {
      "barlow_loss": 0.1457369327545166,
      "epoch": 1.920855614973262,
      "loss": 1.4568723440170288,
      "step": 449
    },
    {
      "barlow_loss": 0.16208995878696442,
      "epoch": 1.920855614973262,
      "loss": 1.2308082580566406,
      "step": 449
    },
    {
      "barlow_loss": 0.160675510764122,
      "epoch": 1.9251336898395723,
      "loss": 1.2753363847732544,
      "step": 450
    },
    {
      "barlow_loss": 0.15340952575206757,
      "epoch": 1.9251336898395723,
      "loss": 1.4157423973083496,
      "step": 450
    },
    {
      "barlow_loss": 0.16590896248817444,
      "epoch": 1.9251336898395723,
      "loss": 1.4253004789352417,
      "step": 450
    },
    {
      "barlow_loss": 0.16912615299224854,
      "epoch": 1.9251336898395723,
      "loss": 1.5063117742538452,
      "step": 450
    },
    {
      "barlow_loss": 0.1579963117837906,
      "epoch": 1.9294117647058824,
      "loss": 1.2711868286132812,
      "step": 451
    },
    {
      "barlow_loss": 0.15727496147155762,
      "epoch": 1.9294117647058824,
      "loss": 1.4025635719299316,
      "step": 451
    },
    {
      "barlow_loss": 0.15803742408752441,
      "epoch": 1.9294117647058824,
      "loss": 1.41140615940094,
      "step": 451
    },
    {
      "barlow_loss": 0.17218272387981415,
      "epoch": 1.9294117647058824,
      "loss": 1.5026233196258545,
      "step": 451
    },
    {
      "epoch": 1.9336898395721924,
      "grad_norm": 1.0778145790100098,
      "learning_rate": 6.183479283269344e-08,
      "loss": 1.3686,
      "step": 452
    },
    {
      "barlow_loss": 0.17787259817123413,
      "epoch": 1.9336898395721924,
      "loss": 1.4760316610336304,
      "step": 452
    },
    {
      "barlow_loss": 0.16632866859436035,
      "epoch": 1.9336898395721924,
      "loss": 1.2841585874557495,
      "step": 452
    },
    {
      "barlow_loss": 0.15575842559337616,
      "epoch": 1.9336898395721924,
      "loss": 1.3036056756973267,
      "step": 452
    },
    {
      "barlow_loss": 0.168816938996315,
      "epoch": 1.9336898395721924,
      "loss": 1.3758260011672974,
      "step": 452
    },
    {
      "barlow_loss": 0.1635483205318451,
      "epoch": 1.9379679144385027,
      "loss": 1.2977896928787231,
      "step": 453
    },
    {
      "barlow_loss": 0.1642785370349884,
      "epoch": 1.9379679144385027,
      "loss": 1.402327299118042,
      "step": 453
    },
    {
      "barlow_loss": 0.16880331933498383,
      "epoch": 1.9379679144385027,
      "loss": 1.451903223991394,
      "step": 453
    },
    {
      "barlow_loss": 0.16503280401229858,
      "epoch": 1.9379679144385027,
      "loss": 1.3030550479888916,
      "step": 453
    },
    {
      "barlow_loss": 0.16655945777893066,
      "epoch": 1.942245989304813,
      "loss": 1.4849272966384888,
      "step": 454
    },
    {
      "barlow_loss": 0.1675148755311966,
      "epoch": 1.942245989304813,
      "loss": 1.3848751783370972,
      "step": 454
    },
    {
      "barlow_loss": 0.1728467494249344,
      "epoch": 1.942245989304813,
      "loss": 1.1874336004257202,
      "step": 454
    },
    {
      "barlow_loss": 0.1713748723268509,
      "epoch": 1.942245989304813,
      "loss": 1.5108528137207031,
      "step": 454
    },
    {
      "barlow_loss": 0.16278263926506042,
      "epoch": 1.946524064171123,
      "loss": 1.3241770267486572,
      "step": 455
    },
    {
      "barlow_loss": 0.15170127153396606,
      "epoch": 1.946524064171123,
      "loss": 1.2668020725250244,
      "step": 455
    },
    {
      "barlow_loss": 0.16550955176353455,
      "epoch": 1.946524064171123,
      "loss": 1.2827467918395996,
      "step": 455
    },
    {
      "barlow_loss": 0.17232653498649597,
      "epoch": 1.946524064171123,
      "loss": 1.4023932218551636,
      "step": 455
    },
    {
      "epoch": 1.950802139037433,
      "grad_norm": 0.9615890383720398,
      "learning_rate": 3.156111820466906e-08,
      "loss": 1.3587,
      "step": 456
    },
    {
      "barlow_loss": 0.16652727127075195,
      "epoch": 1.950802139037433,
      "loss": 1.3746559619903564,
      "step": 456
    },
    {
      "barlow_loss": 0.16498856246471405,
      "epoch": 1.950802139037433,
      "loss": 1.3785923719406128,
      "step": 456
    },
    {
      "barlow_loss": 0.15632355213165283,
      "epoch": 1.950802139037433,
      "loss": 1.3031736612319946,
      "step": 456
    },
    {
      "barlow_loss": 0.16773854196071625,
      "epoch": 1.950802139037433,
      "loss": 1.1372143030166626,
      "step": 456
    },
    {
      "barlow_loss": 0.17078478634357452,
      "epoch": 1.9550802139037433,
      "loss": 1.4282546043395996,
      "step": 457
    },
    {
      "barlow_loss": 0.15967169404029846,
      "epoch": 1.9550802139037433,
      "loss": 1.3524423837661743,
      "step": 457
    },
    {
      "barlow_loss": 0.16949792206287384,
      "epoch": 1.9550802139037433,
      "loss": 1.30246102809906,
      "step": 457
    },
    {
      "barlow_loss": 0.1573553830385208,
      "epoch": 1.9550802139037433,
      "loss": 1.3749467134475708,
      "step": 457
    },
    {
      "barlow_loss": 0.164541095495224,
      "epoch": 1.9593582887700536,
      "loss": 1.3204598426818848,
      "step": 458
    },
    {
      "barlow_loss": 0.1590062528848648,
      "epoch": 1.9593582887700536,
      "loss": 1.2531229257583618,
      "step": 458
    },
    {
      "barlow_loss": 0.1786956489086151,
      "epoch": 1.9593582887700536,
      "loss": 1.4180567264556885,
      "step": 458
    },
    {
      "barlow_loss": 0.16394354403018951,
      "epoch": 1.9593582887700536,
      "loss": 1.5425219535827637,
      "step": 458
    },
    {
      "barlow_loss": 0.1496400237083435,
      "epoch": 1.9636363636363636,
      "loss": 1.4202296733856201,
      "step": 459
    },
    {
      "barlow_loss": 0.1757873147726059,
      "epoch": 1.9636363636363636,
      "loss": 1.3795939683914185,
      "step": 459
    },
    {
      "barlow_loss": 0.15386058390140533,
      "epoch": 1.9636363636363636,
      "loss": 1.3321892023086548,
      "step": 459
    },
    {
      "barlow_loss": 0.17398026585578918,
      "epoch": 1.9636363636363636,
      "loss": 1.3567699193954468,
      "step": 459
    },
    {
      "epoch": 1.9679144385026737,
      "grad_norm": 1.0365643501281738,
      "learning_rate": 1.1365064462093289e-08,
      "loss": 1.3547,
      "step": 460
    },
    {
      "barlow_loss": 0.16474869847297668,
      "epoch": 1.9679144385026737,
      "loss": 1.5494263172149658,
      "step": 460
    },
    {
      "barlow_loss": 0.17776434123516083,
      "epoch": 1.9679144385026737,
      "loss": 1.3601950407028198,
      "step": 460
    },
    {
      "barlow_loss": 0.159606471657753,
      "epoch": 1.9679144385026737,
      "loss": 1.4622505903244019,
      "step": 460
    },
    {
      "barlow_loss": 0.17251276969909668,
      "epoch": 1.9679144385026737,
      "loss": 1.4202593564987183,
      "step": 460
    },
    {
      "barlow_loss": 0.16664284467697144,
      "epoch": 1.972192513368984,
      "loss": 1.319115161895752,
      "step": 461
    },
    {
      "barlow_loss": 0.16155177354812622,
      "epoch": 1.972192513368984,
      "loss": 1.3710269927978516,
      "step": 461
    },
    {
      "barlow_loss": 0.163053959608078,
      "epoch": 1.972192513368984,
      "loss": 1.4260821342468262,
      "step": 461
    },
    {
      "barlow_loss": 0.15158171951770782,
      "epoch": 1.972192513368984,
      "loss": 1.3169053792953491,
      "step": 461
    },
    {
      "barlow_loss": 0.16841821372509003,
      "epoch": 1.9764705882352942,
      "loss": 1.5694676637649536,
      "step": 462
    },
    {
      "barlow_loss": 0.16407392919063568,
      "epoch": 1.9764705882352942,
      "loss": 1.3965853452682495,
      "step": 462
    },
    {
      "barlow_loss": 0.16299988329410553,
      "epoch": 1.9764705882352942,
      "loss": 1.3108279705047607,
      "step": 462
    },
    {
      "barlow_loss": 0.16152027249336243,
      "epoch": 1.9764705882352942,
      "loss": 1.5480862855911255,
      "step": 462
    },
    {
      "barlow_loss": 0.1681302934885025,
      "epoch": 1.9807486631016042,
      "loss": 1.4458439350128174,
      "step": 463
    },
    {
      "barlow_loss": 0.1611861288547516,
      "epoch": 1.9807486631016042,
      "loss": 1.4340038299560547,
      "step": 463
    },
    {
      "barlow_loss": 0.16632625460624695,
      "epoch": 1.9807486631016042,
      "loss": 1.3550571203231812,
      "step": 463
    },
    {
      "barlow_loss": 0.1565222591161728,
      "epoch": 1.9807486631016042,
      "loss": 1.367367148399353,
      "step": 463
    },
    {
      "epoch": 1.9850267379679143,
      "grad_norm": 1.103601336479187,
      "learning_rate": 1.2629550737580653e-09,
      "loss": 1.4158,
      "step": 464
    },
    {
      "barlow_loss": 0.17691823840141296,
      "epoch": 1.9850267379679143,
      "loss": 1.3824107646942139,
      "step": 464
    },
    {
      "barlow_loss": 0.17358267307281494,
      "epoch": 1.9850267379679143,
      "loss": 1.290208339691162,
      "step": 464
    },
    {
      "barlow_loss": 0.17283137142658234,
      "epoch": 1.9850267379679143,
      "loss": 1.3575440645217896,
      "step": 464
    },
    {
      "barlow_loss": 0.1647225320339203,
      "epoch": 1.9850267379679143,
      "loss": 1.3828794956207275,
      "step": 464
    },
    {
      "barlow_loss": 0.16073894500732422,
      "epoch": 1.9893048128342246,
      "loss": 1.2199201583862305,
      "step": 465
    },
    {
      "barlow_loss": 0.17165106534957886,
      "epoch": 1.9893048128342246,
      "loss": 1.4424973726272583,
      "step": 465
    },
    {
      "barlow_loss": 0.15313942730426788,
      "epoch": 1.9893048128342246,
      "loss": 1.5989676713943481,
      "step": 465
    },
    {
      "barlow_loss": 0.17264939844608307,
      "epoch": 1.9893048128342246,
      "loss": 1.4196101427078247,
      "step": 465
    },
    {
      "barlow_loss": 0.15844154357910156,
      "epoch": 1.9935828877005348,
      "loss": 1.55910325050354,
      "step": 466
    },
    {
      "barlow_loss": 0.16770020127296448,
      "epoch": 1.9935828877005348,
      "loss": 1.2988256216049194,
      "step": 466
    },
    {
      "barlow_loss": 0.17810124158859253,
      "epoch": 1.9935828877005348,
      "loss": 1.2494821548461914,
      "step": 466
    },
    {
      "barlow_loss": 0.16402503848075867,
      "epoch": 1.9935828877005348,
      "loss": 1.400961995124817,
      "step": 466
    },
    {
      "barlow_loss": 0.1773127168416977,
      "epoch": 1.9935828877005348,
      "loss": 1.1419506072998047,
      "step": 466
    },
    {
      "barlow_loss": 0.1751735657453537,
      "epoch": 1.9935828877005348,
      "loss": 1.5892081260681152,
      "step": 466
    },
    {
      "barlow_loss": 0.170654296875,
      "epoch": 1.9935828877005348,
      "loss": 1.2516310214996338,
      "step": 466
    },
    {
      "barlow_loss": 0.175925150513649,
      "epoch": 1.9935828877005348,
      "loss": 1.396144151687622,
      "step": 466
    },
    {
      "barlow_loss": 0.1643402874469757,
      "epoch": 1.9935828877005348,
      "loss": 1.2402821779251099,
      "step": 466
    },
    {
      "barlow_loss": 0.16132009029388428,
      "epoch": 1.9935828877005348,
      "loss": 1.2746901512145996,
      "step": 466
    },
    {
      "barlow_loss": 0.1817024052143097,
      "epoch": 1.9935828877005348,
      "loss": 1.6636234521865845,
      "step": 466
    },
    {
      "barlow_loss": 0.1896342635154724,
      "epoch": 1.9935828877005348,
      "loss": 1.2436567544937134,
      "step": 466
    },
    {
      "barlow_loss": 0.16263772547245026,
      "epoch": 1.9935828877005348,
      "loss": 1.3684221506118774,
      "step": 466
    },
    {
      "barlow_loss": 0.17033524811267853,
      "epoch": 1.9935828877005348,
      "loss": 1.3461748361587524,
      "step": 466
    },
    {
      "barlow_loss": 0.168716698884964,
      "epoch": 1.9935828877005348,
      "loss": 1.2770277261734009,
      "step": 466
    },
    {
      "barlow_loss": 0.16668103635311127,
      "epoch": 1.9935828877005348,
      "loss": 1.5350408554077148,
      "step": 466
    },
    {
      "barlow_loss": 0.16811503469944,
      "epoch": 1.9935828877005348,
      "loss": 1.5469186305999756,
      "step": 466
    },
    {
      "barlow_loss": 0.17147141695022583,
      "epoch": 1.9935828877005348,
      "loss": 1.2749117612838745,
      "step": 466
    },
    {
      "barlow_loss": 0.16228589415550232,
      "epoch": 1.9935828877005348,
      "loss": 1.3345612287521362,
      "step": 466
    },
    {
      "barlow_loss": 0.16380563378334045,
      "epoch": 1.9935828877005348,
      "loss": 1.4650218486785889,
      "step": 466
    },
    {
      "barlow_loss": 0.16926978528499603,
      "epoch": 1.9935828877005348,
      "loss": 1.651007890701294,
      "step": 466
    },
    {
      "barlow_loss": 0.16380149126052856,
      "epoch": 1.9935828877005348,
      "loss": 1.2788912057876587,
      "step": 466
    },
    {
      "barlow_loss": 0.15838877856731415,
      "epoch": 1.9935828877005348,
      "loss": 1.3225486278533936,
      "step": 466
    },
    {
      "barlow_loss": 0.19925221800804138,
      "epoch": 1.9935828877005348,
      "loss": 1.283705234527588,
      "step": 466
    },
    {
      "barlow_loss": 0.18066906929016113,
      "epoch": 1.9935828877005348,
      "loss": 1.5915716886520386,
      "step": 466
    },
    {
      "barlow_loss": 0.16650423407554626,
      "epoch": 1.9935828877005348,
      "loss": 1.4900611639022827,
      "step": 466
    },
    {
      "barlow_loss": 0.16220779716968536,
      "epoch": 1.9935828877005348,
      "loss": 1.107014775276184,
      "step": 466
    },
    {
      "barlow_loss": 0.16655230522155762,
      "epoch": 1.9935828877005348,
      "loss": 1.566054344177246,
      "step": 466
    },
    {
      "barlow_loss": 0.17069782316684723,
      "epoch": 1.9935828877005348,
      "loss": 1.1504822969436646,
      "step": 466
    },
    {
      "barlow_loss": 0.18254728615283966,
      "epoch": 1.9935828877005348,
      "loss": 1.8475462198257446,
      "step": 466
    },
    {
      "barlow_loss": 0.1738716959953308,
      "epoch": 1.9935828877005348,
      "loss": 1.1877106428146362,
      "step": 466
    },
    {
      "barlow_loss": 0.17999188601970673,
      "epoch": 1.9935828877005348,
      "loss": 1.3650647401809692,
      "step": 466
    },
    {
      "barlow_loss": 0.1680765300989151,
      "epoch": 1.9935828877005348,
      "loss": 1.2568556070327759,
      "step": 466
    },
    {
      "barlow_loss": 0.17953725159168243,
      "epoch": 1.9935828877005348,
      "loss": 1.4671002626419067,
      "step": 466
    },
    {
      "barlow_loss": 0.17863065004348755,
      "epoch": 1.9935828877005348,
      "loss": 1.533764362335205,
      "step": 466
    },
    {
      "barlow_loss": 0.18060606718063354,
      "epoch": 1.9935828877005348,
      "loss": 1.449137806892395,
      "step": 466
    },
    {
      "barlow_loss": 0.1712372601032257,
      "epoch": 1.9935828877005348,
      "loss": 1.471323013305664,
      "step": 466
    },
    {
      "barlow_loss": 0.16323336958885193,
      "epoch": 1.9935828877005348,
      "loss": 1.212869644165039,
      "step": 466
    },
    {
      "barlow_loss": 0.1566883623600006,
      "epoch": 1.9935828877005348,
      "loss": 1.5191620588302612,
      "step": 466
    },
    {
      "barlow_loss": 0.18010175228118896,
      "epoch": 1.9935828877005348,
      "loss": 1.5181121826171875,
      "step": 466
    },
    {
      "barlow_loss": 0.16774669289588928,
      "epoch": 1.9935828877005348,
      "loss": 1.439023733139038,
      "step": 466
    },
    {
      "barlow_loss": 0.16453202068805695,
      "epoch": 1.9935828877005348,
      "loss": 1.2920407056808472,
      "step": 466
    },
    {
      "barlow_loss": 0.17156818509101868,
      "epoch": 1.9935828877005348,
      "loss": 1.4171056747436523,
      "step": 466
    },
    {
      "barlow_loss": 0.1791706681251526,
      "epoch": 1.9935828877005348,
      "loss": 1.1545771360397339,
      "step": 466
    },
    {
      "barlow_loss": 0.17945721745491028,
      "epoch": 1.9935828877005348,
      "loss": 1.2536035776138306,
      "step": 466
    },
    {
      "barlow_loss": 0.17314821481704712,
      "epoch": 1.9935828877005348,
      "loss": 1.3135422468185425,
      "step": 466
    },
    {
      "barlow_loss": 0.17212918400764465,
      "epoch": 1.9935828877005348,
      "loss": 1.5149914026260376,
      "step": 466
    },
    {
      "barlow_loss": 0.1684998869895935,
      "epoch": 1.9935828877005348,
      "loss": 1.1249264478683472,
      "step": 466
    },
    {
      "barlow_loss": 0.17438971996307373,
      "epoch": 1.9935828877005348,
      "loss": 1.4019747972488403,
      "step": 466
    },
    {
      "barlow_loss": 0.17523859441280365,
      "epoch": 1.9935828877005348,
      "loss": 1.3924589157104492,
      "step": 466
    },
    {
      "barlow_loss": 0.17652277648448944,
      "epoch": 1.9935828877005348,
      "loss": 1.2268202304840088,
      "step": 466
    },
    {
      "barlow_loss": 0.18098025023937225,
      "epoch": 1.9935828877005348,
      "loss": 1.5129631757736206,
      "step": 466
    },
    {
      "barlow_loss": 0.18262089788913727,
      "epoch": 1.9935828877005348,
      "loss": 1.3203555345535278,
      "step": 466
    },
    {
      "barlow_loss": 0.16948983073234558,
      "epoch": 1.9935828877005348,
      "loss": 1.3738577365875244,
      "step": 466
    },
    {
      "barlow_loss": 0.17653630673885345,
      "epoch": 1.9935828877005348,
      "loss": 1.359853744506836,
      "step": 466
    },
    {
      "barlow_loss": 0.16925856471061707,
      "epoch": 1.9935828877005348,
      "loss": 1.3012702465057373,
      "step": 466
    },
    {
      "barlow_loss": 0.17322450876235962,
      "epoch": 1.9935828877005348,
      "loss": 1.3532484769821167,
      "step": 466
    },
    {
      "barlow_loss": 0.15670840442180634,
      "epoch": 1.9935828877005348,
      "loss": 1.2350512742996216,
      "step": 466
    },
    {
      "barlow_loss": 0.17538538575172424,
      "epoch": 1.9935828877005348,
      "loss": 1.413375735282898,
      "step": 466
    },
    {
      "barlow_loss": 0.17556416988372803,
      "epoch": 1.9935828877005348,
      "loss": 1.2145743370056152,
      "step": 466
    },
    {
      "barlow_loss": 0.17922672629356384,
      "epoch": 1.9935828877005348,
      "loss": 1.351898431777954,
      "step": 466
    },
    {
      "barlow_loss": 0.18351969122886658,
      "epoch": 1.9935828877005348,
      "loss": 1.2758305072784424,
      "step": 466
    },
    {
      "barlow_loss": 0.15488743782043457,
      "epoch": 1.9935828877005348,
      "loss": 1.3123431205749512,
      "step": 466
    },
    {
      "barlow_loss": 0.16788987815380096,
      "epoch": 1.9935828877005348,
      "loss": 1.4799431562423706,
      "step": 466
    },
    {
      "barlow_loss": 0.17906111478805542,
      "epoch": 1.9935828877005348,
      "loss": 1.337437629699707,
      "step": 466
    },
    {
      "barlow_loss": 0.17080137133598328,
      "epoch": 1.9935828877005348,
      "loss": 1.2312583923339844,
      "step": 466
    },
    {
      "barlow_loss": 0.17565694451332092,
      "epoch": 1.9935828877005348,
      "loss": 1.385770559310913,
      "step": 466
    },
    {
      "barlow_loss": 0.1829058825969696,
      "epoch": 1.9935828877005348,
      "loss": 1.4714958667755127,
      "step": 466
    },
    {
      "barlow_loss": 0.18700841069221497,
      "epoch": 1.9935828877005348,
      "loss": 1.457454800605774,
      "step": 466
    },
    {
      "barlow_loss": 0.16842924058437347,
      "epoch": 1.9935828877005348,
      "loss": 1.5164389610290527,
      "step": 466
    },
    {
      "barlow_loss": 0.17061719298362732,
      "epoch": 1.9935828877005348,
      "loss": 1.37839674949646,
      "step": 466
    },
    {
      "barlow_loss": 0.1712886393070221,
      "epoch": 1.9935828877005348,
      "loss": 1.468414306640625,
      "step": 466
    },
    {
      "barlow_loss": 0.18655060231685638,
      "epoch": 1.9935828877005348,
      "loss": 1.1738817691802979,
      "step": 466
    },
    {
      "barlow_loss": 0.1738956719636917,
      "epoch": 1.9935828877005348,
      "loss": 1.531315565109253,
      "step": 466
    },
    {
      "barlow_loss": 0.18385694921016693,
      "epoch": 1.9935828877005348,
      "loss": 1.3493192195892334,
      "step": 466
    },
    {
      "barlow_loss": 0.15932339429855347,
      "epoch": 1.9935828877005348,
      "loss": 1.298538327217102,
      "step": 466
    },
    {
      "barlow_loss": 0.1771521419286728,
      "epoch": 1.9935828877005348,
      "loss": 1.507441759109497,
      "step": 466
    },
    {
      "barlow_loss": 0.1582649201154709,
      "epoch": 1.9935828877005348,
      "loss": 1.2313752174377441,
      "step": 466
    },
    {
      "barlow_loss": 0.14637373387813568,
      "epoch": 1.9935828877005348,
      "loss": 1.1609364748001099,
      "step": 466
    },
    {
      "barlow_loss": 0.19506920874118805,
      "epoch": 1.9935828877005348,
      "loss": 1.4768441915512085,
      "step": 466
    },
    {
      "barlow_loss": 0.1601303666830063,
      "epoch": 1.9935828877005348,
      "loss": 1.5068750381469727,
      "step": 466
    },
    {
      "barlow_loss": 0.17695537209510803,
      "epoch": 1.9935828877005348,
      "loss": 1.3570284843444824,
      "step": 466
    },
    {
      "barlow_loss": 0.18718433380126953,
      "epoch": 1.9935828877005348,
      "loss": 1.4644578695297241,
      "step": 466
    },
    {
      "barlow_loss": 0.1575610637664795,
      "epoch": 1.9935828877005348,
      "loss": 1.1965312957763672,
      "step": 466
    },
    {
      "barlow_loss": 0.16989468038082123,
      "epoch": 1.9935828877005348,
      "loss": 1.4790054559707642,
      "step": 466
    },
    {
      "barlow_loss": 0.1674996167421341,
      "epoch": 1.9935828877005348,
      "loss": 1.2257190942764282,
      "step": 466
    },
    {
      "barlow_loss": 0.19357745349407196,
      "epoch": 1.9935828877005348,
      "loss": 2.016331911087036,
      "step": 466
    },
    {
      "barlow_loss": 0.17207174003124237,
      "epoch": 1.9935828877005348,
      "loss": 1.390500783920288,
      "step": 466
    },
    {
      "barlow_loss": 0.16580528020858765,
      "epoch": 1.9935828877005348,
      "loss": 1.2937597036361694,
      "step": 466
    },
    {
      "barlow_loss": 0.1767144501209259,
      "epoch": 1.9935828877005348,
      "loss": 1.4691588878631592,
      "step": 466
    },
    {
      "barlow_loss": 0.1746521294116974,
      "epoch": 1.9935828877005348,
      "loss": 1.6154279708862305,
      "step": 466
    },
    {
      "barlow_loss": 0.18751507997512817,
      "epoch": 1.9935828877005348,
      "loss": 1.3774564266204834,
      "step": 466
    },
    {
      "barlow_loss": 0.18239668011665344,
      "epoch": 1.9935828877005348,
      "loss": 1.2648578882217407,
      "step": 466
    },
    {
      "barlow_loss": 0.1724260449409485,
      "epoch": 1.9935828877005348,
      "loss": 1.308597445487976,
      "step": 466
    },
    {
      "barlow_loss": 0.18539850413799286,
      "epoch": 1.9935828877005348,
      "loss": 1.7207083702087402,
      "step": 466
    },
    {
      "barlow_loss": 0.17692317068576813,
      "epoch": 1.9935828877005348,
      "loss": 1.500016450881958,
      "step": 466
    },
    {
      "barlow_loss": 0.185557559132576,
      "epoch": 1.9935828877005348,
      "loss": 1.217378854751587,
      "step": 466
    },
    {
      "barlow_loss": 0.1579446941614151,
      "epoch": 1.9935828877005348,
      "loss": 1.5189402103424072,
      "step": 466
    },
    {
      "barlow_loss": 0.1851278692483902,
      "epoch": 1.9935828877005348,
      "loss": 1.4754736423492432,
      "step": 466
    },
    {
      "barlow_loss": 0.16298069059848785,
      "epoch": 1.9935828877005348,
      "loss": 1.497021198272705,
      "step": 466
    },
    {
      "barlow_loss": 0.16898798942565918,
      "epoch": 1.9935828877005348,
      "loss": 1.5624717473983765,
      "step": 466
    },
    {
      "barlow_loss": 0.16991345584392548,
      "epoch": 1.9935828877005348,
      "loss": 1.2096003293991089,
      "step": 466
    },
    {
      "barlow_loss": 0.1560637503862381,
      "epoch": 1.9935828877005348,
      "loss": 1.2085415124893188,
      "step": 466
    },
    {
      "barlow_loss": 0.17932955920696259,
      "epoch": 1.9935828877005348,
      "loss": 1.1927663087844849,
      "step": 466
    },
    {
      "barlow_loss": 0.17395584285259247,
      "epoch": 1.9935828877005348,
      "loss": 1.3505806922912598,
      "step": 466
    },
    {
      "barlow_loss": 0.16210278868675232,
      "epoch": 1.9935828877005348,
      "loss": 1.2156689167022705,
      "step": 466
    },
    {
      "barlow_loss": 0.14964108169078827,
      "epoch": 1.9935828877005348,
      "loss": 1.378498911857605,
      "step": 466
    },
    {
      "barlow_loss": 0.18945930898189545,
      "epoch": 1.9935828877005348,
      "loss": 1.5839200019836426,
      "step": 466
    },
    {
      "barlow_loss": 0.1681990623474121,
      "epoch": 1.9935828877005348,
      "loss": 1.4166477918624878,
      "step": 466
    },
    {
      "barlow_loss": 0.16918201744556427,
      "epoch": 1.9935828877005348,
      "loss": 1.3403116464614868,
      "step": 466
    },
    {
      "barlow_loss": 0.1668129861354828,
      "epoch": 1.9935828877005348,
      "loss": 1.342706322669983,
      "step": 466
    },
    {
      "barlow_loss": 0.17677642405033112,
      "epoch": 1.9935828877005348,
      "loss": 1.5487712621688843,
      "step": 466
    },
    {
      "barlow_loss": 0.1756037175655365,
      "epoch": 1.9935828877005348,
      "loss": 1.574649453163147,
      "step": 466
    },
    {
      "barlow_loss": 0.17342548072338104,
      "epoch": 1.9935828877005348,
      "loss": 1.200282335281372,
      "step": 466
    },
    {
      "barlow_loss": 0.18230947852134705,
      "epoch": 1.9935828877005348,
      "loss": 1.327332854270935,
      "step": 466
    },
    {
      "barlow_loss": 0.16873668134212494,
      "epoch": 1.9935828877005348,
      "loss": 1.6498056650161743,
      "step": 466
    },
    {
      "barlow_loss": 0.16085875034332275,
      "epoch": 1.9935828877005348,
      "loss": 1.4366804361343384,
      "step": 466
    },
    {
      "barlow_loss": 0.1654238998889923,
      "epoch": 1.9935828877005348,
      "loss": 1.5490342378616333,
      "step": 466
    },
    {
      "barlow_loss": 0.18397563695907593,
      "epoch": 1.9935828877005348,
      "loss": 1.1613163948059082,
      "step": 466
    },
    {
      "barlow_loss": 0.14540575444698334,
      "epoch": 1.9935828877005348,
      "loss": 1.1880359649658203,
      "step": 466
    },
    {
      "barlow_loss": 0.17743749916553497,
      "epoch": 1.9935828877005348,
      "loss": 1.4666626453399658,
      "step": 466
    },
    {
      "barlow_loss": 0.16301897168159485,
      "epoch": 1.9935828877005348,
      "loss": 1.816037893295288,
      "step": 466
    },
    {
      "barlow_loss": 0.1822887361049652,
      "epoch": 1.9935828877005348,
      "loss": 1.5219165086746216,
      "step": 466
    },
    {
      "barlow_loss": 0.17651303112506866,
      "epoch": 1.9935828877005348,
      "loss": 1.2563985586166382,
      "step": 466
    },
    {
      "barlow_loss": 0.17597182095050812,
      "epoch": 1.9935828877005348,
      "loss": 1.4903786182403564,
      "step": 466
    },
    {
      "barlow_loss": 0.1725389063358307,
      "epoch": 1.9935828877005348,
      "loss": 1.540999412536621,
      "step": 466
    },
    {
      "barlow_loss": 0.1832234263420105,
      "epoch": 1.9935828877005348,
      "loss": 1.5075268745422363,
      "step": 466
    },
    {
      "barlow_loss": 0.1634185016155243,
      "epoch": 1.9935828877005348,
      "loss": 1.4721430540084839,
      "step": 466
    },
    {
      "barlow_loss": 0.1735246777534485,
      "epoch": 1.9935828877005348,
      "loss": 1.510787844657898,
      "step": 466
    },
    {
      "barlow_loss": 0.19247442483901978,
      "epoch": 1.9935828877005348,
      "loss": 1.5134462118148804,
      "step": 466
    },
    {
      "barlow_loss": 0.1877429485321045,
      "epoch": 1.9935828877005348,
      "loss": 1.568515419960022,
      "step": 466
    },
    {
      "barlow_loss": 0.14885008335113525,
      "epoch": 1.9935828877005348,
      "loss": 1.3322550058364868,
      "step": 466
    },
    {
      "barlow_loss": 0.16932983696460724,
      "epoch": 1.9935828877005348,
      "loss": 1.405914545059204,
      "step": 466
    },
    {
      "barlow_loss": 0.1801140010356903,
      "epoch": 1.9935828877005348,
      "loss": 1.5794581174850464,
      "step": 466
    },
    {
      "barlow_loss": 0.1830083727836609,
      "epoch": 1.9935828877005348,
      "loss": 1.2387944459915161,
      "step": 466
    },
    {
      "barlow_loss": 0.17705340683460236,
      "epoch": 1.9935828877005348,
      "loss": 1.4681271314620972,
      "step": 466
    },
    {
      "barlow_loss": 0.1794569492340088,
      "epoch": 1.9935828877005348,
      "loss": 1.3456454277038574,
      "step": 466
    },
    {
      "barlow_loss": 0.1725936383008957,
      "epoch": 1.9935828877005348,
      "loss": 1.1463755369186401,
      "step": 466
    },
    {
      "barlow_loss": 0.16839714348316193,
      "epoch": 1.9935828877005348,
      "loss": 1.1726486682891846,
      "step": 466
    },
    {
      "barlow_loss": 0.17483559250831604,
      "epoch": 1.9935828877005348,
      "loss": 1.2432808876037598,
      "step": 466
    },
    {
      "barlow_loss": 0.16266396641731262,
      "epoch": 1.9935828877005348,
      "loss": 1.440904140472412,
      "step": 466
    },
    {
      "barlow_loss": 0.17286306619644165,
      "epoch": 1.9935828877005348,
      "loss": 1.2758362293243408,
      "step": 466
    },
    {
      "barlow_loss": 0.16279946267604828,
      "epoch": 1.9935828877005348,
      "loss": 1.454646348953247,
      "step": 466
    },
    {
      "barlow_loss": 0.16425016522407532,
      "epoch": 1.9935828877005348,
      "loss": 1.3948547840118408,
      "step": 466
    },
    {
      "barlow_loss": 0.17765623331069946,
      "epoch": 1.9935828877005348,
      "loss": 1.3331135511398315,
      "step": 466
    },
    {
      "barlow_loss": 0.14832012355327606,
      "epoch": 1.9935828877005348,
      "loss": 1.2223376035690308,
      "step": 466
    },
    {
      "barlow_loss": 0.17358006536960602,
      "epoch": 1.9935828877005348,
      "loss": 1.6152417659759521,
      "step": 466
    },
    {
      "barlow_loss": 0.16872984170913696,
      "epoch": 1.9935828877005348,
      "loss": 1.501960039138794,
      "step": 466
    },
    {
      "barlow_loss": 0.1848996877670288,
      "epoch": 1.9935828877005348,
      "loss": 1.4125694036483765,
      "step": 466
    },
    {
      "barlow_loss": 0.1723681390285492,
      "epoch": 1.9935828877005348,
      "loss": 1.5361709594726562,
      "step": 466
    },
    {
      "barlow_loss": 0.1727135181427002,
      "epoch": 1.9935828877005348,
      "loss": 1.3044015169143677,
      "step": 466
    },
    {
      "barlow_loss": 0.1725984513759613,
      "epoch": 1.9935828877005348,
      "loss": 1.3720279932022095,
      "step": 466
    },
    {
      "barlow_loss": 0.16655117273330688,
      "epoch": 1.9935828877005348,
      "loss": 1.3406398296356201,
      "step": 466
    },
    {
      "barlow_loss": 0.18508994579315186,
      "epoch": 1.9935828877005348,
      "loss": 1.4316011667251587,
      "step": 466
    },
    {
      "barlow_loss": 0.16515369713306427,
      "epoch": 1.9935828877005348,
      "loss": 1.3660259246826172,
      "step": 466
    },
    {
      "barlow_loss": 0.18106597661972046,
      "epoch": 1.9935828877005348,
      "loss": 1.2512245178222656,
      "step": 466
    },
    {
      "barlow_loss": 0.1673363745212555,
      "epoch": 1.9935828877005348,
      "loss": 1.17960524559021,
      "step": 466
    },
    {
      "barlow_loss": 0.17037677764892578,
      "epoch": 1.9935828877005348,
      "loss": 1.162317156791687,
      "step": 466
    },
    {
      "barlow_loss": 0.1754152923822403,
      "epoch": 1.9935828877005348,
      "loss": 1.3222100734710693,
      "step": 466
    },
    {
      "barlow_loss": 0.18430322408676147,
      "epoch": 1.9935828877005348,
      "loss": 1.240824580192566,
      "step": 466
    },
    {
      "barlow_loss": 0.15483328700065613,
      "epoch": 1.9935828877005348,
      "loss": 1.4760220050811768,
      "step": 466
    },
    {
      "barlow_loss": 0.18775050342082977,
      "epoch": 1.9935828877005348,
      "loss": 1.5993058681488037,
      "step": 466
    },
    {
      "barlow_loss": 0.15771479904651642,
      "epoch": 1.9935828877005348,
      "loss": 1.2792702913284302,
      "step": 466
    },
    {
      "barlow_loss": 0.19438736140727997,
      "epoch": 1.9935828877005348,
      "loss": 1.3508208990097046,
      "step": 466
    },
    {
      "barlow_loss": 0.17526111006736755,
      "epoch": 1.9935828877005348,
      "loss": 1.2000889778137207,
      "step": 466
    },
    {
      "barlow_loss": 0.18229027092456818,
      "epoch": 1.9935828877005348,
      "loss": 1.4704381227493286,
      "step": 466
    },
    {
      "barlow_loss": 0.1603279858827591,
      "epoch": 1.9935828877005348,
      "loss": 1.3369649648666382,
      "step": 466
    },
    {
      "barlow_loss": 0.16360899806022644,
      "epoch": 1.9935828877005348,
      "loss": 1.487348198890686,
      "step": 466
    },
    {
      "barlow_loss": 0.15046972036361694,
      "epoch": 1.9935828877005348,
      "loss": 1.2543339729309082,
      "step": 466
    },
    {
      "barlow_loss": 0.1574476808309555,
      "epoch": 1.9935828877005348,
      "loss": 1.4598435163497925,
      "step": 466
    },
    {
      "barlow_loss": 0.17253665626049042,
      "epoch": 1.9935828877005348,
      "loss": 1.366977572441101,
      "step": 466
    },
    {
      "barlow_loss": 0.188909113407135,
      "epoch": 1.9935828877005348,
      "loss": 1.4257837533950806,
      "step": 466
    },
    {
      "barlow_loss": 0.16792090237140656,
      "epoch": 1.9935828877005348,
      "loss": 1.468987226486206,
      "step": 466
    },
    {
      "barlow_loss": 0.1453402191400528,
      "epoch": 1.9935828877005348,
      "loss": 1.0486693382263184,
      "step": 466
    },
    {
      "barlow_loss": 0.1554126739501953,
      "epoch": 1.9935828877005348,
      "loss": 1.2284048795700073,
      "step": 466
    },
    {
      "barlow_loss": 0.19094380736351013,
      "epoch": 1.9935828877005348,
      "loss": 1.0909723043441772,
      "step": 466
    },
    {
      "barlow_loss": 0.15945947170257568,
      "epoch": 1.9935828877005348,
      "loss": 1.384800672531128,
      "step": 466
    },
    {
      "barlow_loss": 0.18514475226402283,
      "epoch": 1.9935828877005348,
      "loss": 1.4503248929977417,
      "step": 466
    },
    {
      "barlow_loss": 0.15888828039169312,
      "epoch": 1.9935828877005348,
      "loss": 1.2489458322525024,
      "step": 466
    },
    {
      "barlow_loss": 0.16816362738609314,
      "epoch": 1.9935828877005348,
      "loss": 1.1840752363204956,
      "step": 466
    },
    {
      "barlow_loss": 0.17768356204032898,
      "epoch": 1.9935828877005348,
      "loss": 1.5696381330490112,
      "step": 466
    },
    {
      "barlow_loss": 0.175256609916687,
      "epoch": 1.9935828877005348,
      "loss": 1.313697338104248,
      "step": 466
    },
    {
      "barlow_loss": 0.17695903778076172,
      "epoch": 1.9935828877005348,
      "loss": 1.0523239374160767,
      "step": 466
    },
    {
      "barlow_loss": 0.1803579479455948,
      "epoch": 1.9935828877005348,
      "loss": 1.4821176528930664,
      "step": 466
    },
    {
      "barlow_loss": 0.15915942192077637,
      "epoch": 1.9935828877005348,
      "loss": 1.2909481525421143,
      "step": 466
    },
    {
      "barlow_loss": 0.17414523661136627,
      "epoch": 1.9935828877005348,
      "loss": 1.393547534942627,
      "step": 466
    },
    {
      "barlow_loss": 0.15660086274147034,
      "epoch": 1.9935828877005348,
      "loss": 1.6387583017349243,
      "step": 466
    },
    {
      "barlow_loss": 0.16568544507026672,
      "epoch": 1.9935828877005348,
      "loss": 1.2893210649490356,
      "step": 466
    },
    {
      "barlow_loss": 0.19103427231311798,
      "epoch": 1.9935828877005348,
      "loss": 1.3796817064285278,
      "step": 466
    },
    {
      "barlow_loss": 0.18038715422153473,
      "epoch": 1.9935828877005348,
      "loss": 1.3332277536392212,
      "step": 466
    },
    {
      "barlow_loss": 0.16450375318527222,
      "epoch": 1.9935828877005348,
      "loss": 1.2298855781555176,
      "step": 466
    },
    {
      "barlow_loss": 0.17448288202285767,
      "epoch": 1.9935828877005348,
      "loss": 1.196679949760437,
      "step": 466
    },
    {
      "barlow_loss": 0.16295234858989716,
      "epoch": 1.9935828877005348,
      "loss": 1.6061650514602661,
      "step": 466
    },
    {
      "barlow_loss": 0.17740026116371155,
      "epoch": 1.9935828877005348,
      "loss": 1.5883142948150635,
      "step": 466
    },
    {
      "barlow_loss": 0.18239784240722656,
      "epoch": 1.9935828877005348,
      "loss": 1.3221598863601685,
      "step": 466
    },
    {
      "barlow_loss": 0.1811053454875946,
      "epoch": 1.9935828877005348,
      "loss": 1.5249290466308594,
      "step": 466
    },
    {
      "barlow_loss": 0.17564672231674194,
      "epoch": 1.9935828877005348,
      "loss": 1.468495488166809,
      "step": 466
    },
    {
      "barlow_loss": 0.17592740058898926,
      "epoch": 1.9935828877005348,
      "loss": 1.4284820556640625,
      "step": 466
    },
    {
      "barlow_loss": 0.17362937331199646,
      "epoch": 1.9935828877005348,
      "loss": 1.5089330673217773,
      "step": 466
    },
    {
      "barlow_loss": 0.1844925880432129,
      "epoch": 1.9935828877005348,
      "loss": 1.3950105905532837,
      "step": 466
    },
    {
      "barlow_loss": 0.17643298208713531,
      "epoch": 1.9935828877005348,
      "loss": 1.7563165426254272,
      "step": 466
    },
    {
      "barlow_loss": 0.18058863282203674,
      "epoch": 1.9935828877005348,
      "loss": 1.4626662731170654,
      "step": 466
    },
    {
      "barlow_loss": 0.18033641576766968,
      "epoch": 1.9935828877005348,
      "loss": 1.5148838758468628,
      "step": 466
    },
    {
      "barlow_loss": 0.15576805174350739,
      "epoch": 1.9935828877005348,
      "loss": 1.4766093492507935,
      "step": 466
    },
    {
      "barlow_loss": 0.17573505640029907,
      "epoch": 1.9935828877005348,
      "loss": 1.2420921325683594,
      "step": 466
    },
    {
      "barlow_loss": 0.18004754185676575,
      "epoch": 1.9935828877005348,
      "loss": 3.645977258682251,
      "step": 466
    },
    {
      "barlow_loss": 0.17536064982414246,
      "epoch": 1.9935828877005348,
      "loss": 1.572413444519043,
      "step": 466
    },
    {
      "barlow_loss": 0.16551648080348969,
      "epoch": 1.9935828877005348,
      "loss": 1.411842703819275,
      "step": 466
    },
    {
      "barlow_loss": 0.1829730123281479,
      "epoch": 1.9935828877005348,
      "loss": 1.6415388584136963,
      "step": 466
    },
    {
      "barlow_loss": 0.19005978107452393,
      "epoch": 1.9935828877005348,
      "loss": 1.3556408882141113,
      "step": 466
    },
    {
      "barlow_loss": 0.1810600608587265,
      "epoch": 1.9935828877005348,
      "loss": 1.443932294845581,
      "step": 466
    },
    {
      "barlow_loss": 0.17627084255218506,
      "epoch": 1.9935828877005348,
      "loss": 1.2821017503738403,
      "step": 466
    },
    {
      "barlow_loss": 0.18983179330825806,
      "epoch": 1.9935828877005348,
      "loss": 1.189368486404419,
      "step": 466
    },
    {
      "barlow_loss": 0.16388331353664398,
      "epoch": 1.9935828877005348,
      "loss": 1.5155243873596191,
      "step": 466
    },
    {
      "barlow_loss": 0.17063818871974945,
      "epoch": 1.9935828877005348,
      "loss": 1.5064830780029297,
      "step": 466
    },
    {
      "barlow_loss": 0.1762358546257019,
      "epoch": 1.9935828877005348,
      "loss": 1.2875370979309082,
      "step": 466
    },
    {
      "barlow_loss": 0.1819469928741455,
      "epoch": 1.9935828877005348,
      "loss": 1.1699312925338745,
      "step": 466
    },
    {
      "barlow_loss": 0.16541467607021332,
      "epoch": 1.9935828877005348,
      "loss": 1.3241097927093506,
      "step": 466
    },
    {
      "barlow_loss": 0.17423340678215027,
      "epoch": 1.9935828877005348,
      "loss": 1.2380781173706055,
      "step": 466
    },
    {
      "barlow_loss": 0.17548733949661255,
      "epoch": 1.9935828877005348,
      "loss": 1.4385565519332886,
      "step": 466
    },
    {
      "barlow_loss": 0.1713038831949234,
      "epoch": 1.9935828877005348,
      "loss": 1.391984224319458,
      "step": 466
    },
    {
      "barlow_loss": 0.1638113260269165,
      "epoch": 1.9935828877005348,
      "loss": 1.5076161623001099,
      "step": 466
    },
    {
      "barlow_loss": 0.1780470311641693,
      "epoch": 1.9935828877005348,
      "loss": 1.455837607383728,
      "step": 466
    },
    {
      "barlow_loss": 0.17769655585289001,
      "epoch": 1.9935828877005348,
      "loss": 1.512059211730957,
      "step": 466
    },
    {
      "barlow_loss": 0.1557215303182602,
      "epoch": 1.9935828877005348,
      "loss": 1.386579155921936,
      "step": 466
    },
    {
      "barlow_loss": 0.1666092574596405,
      "epoch": 1.9935828877005348,
      "loss": 1.6168757677078247,
      "step": 466
    },
    {
      "barlow_loss": 0.18102239072322845,
      "epoch": 1.9935828877005348,
      "loss": 1.3130548000335693,
      "step": 466
    },
    {
      "barlow_loss": 0.16451628506183624,
      "epoch": 1.9935828877005348,
      "loss": 1.4753905534744263,
      "step": 466
    },
    {
      "barlow_loss": 0.17467378079891205,
      "epoch": 1.9935828877005348,
      "loss": 1.5298902988433838,
      "step": 466
    },
    {
      "barlow_loss": 0.18277396261692047,
      "epoch": 1.9935828877005348,
      "loss": 1.4089123010635376,
      "step": 466
    },
    {
      "barlow_loss": 0.18383122980594635,
      "epoch": 1.9935828877005348,
      "loss": 1.2197163105010986,
      "step": 466
    },
    {
      "barlow_loss": 0.15126338601112366,
      "epoch": 1.9935828877005348,
      "loss": 2.073958396911621,
      "step": 466
    },
    {
      "barlow_loss": 0.17267611622810364,
      "epoch": 1.9935828877005348,
      "loss": 1.5542272329330444,
      "step": 466
    },
    {
      "barlow_loss": 0.18349412083625793,
      "epoch": 1.9935828877005348,
      "loss": 1.3263341188430786,
      "step": 466
    },
    {
      "barlow_loss": 0.18647751212120056,
      "epoch": 1.9935828877005348,
      "loss": 1.5081206560134888,
      "step": 466
    },
    {
      "barlow_loss": 0.1799541860818863,
      "epoch": 1.9935828877005348,
      "loss": 1.5841280221939087,
      "step": 466
    },
    {
      "barlow_loss": 0.17077137529850006,
      "epoch": 1.9935828877005348,
      "loss": 1.5930787324905396,
      "step": 466
    },
    {
      "barlow_loss": 0.17699061334133148,
      "epoch": 1.9935828877005348,
      "loss": 1.4763481616973877,
      "step": 466
    },
    {
      "barlow_loss": 0.17401528358459473,
      "epoch": 1.9935828877005348,
      "loss": 1.3280030488967896,
      "step": 466
    },
    {
      "barlow_loss": 0.1576175093650818,
      "epoch": 1.9935828877005348,
      "loss": 1.222225308418274,
      "step": 466
    },
    {
      "barlow_loss": 0.18143492937088013,
      "epoch": 1.9935828877005348,
      "loss": 1.3481756448745728,
      "step": 466
    },
    {
      "barlow_loss": 0.17406855523586273,
      "epoch": 1.9935828877005348,
      "loss": 1.5918314456939697,
      "step": 466
    },
    {
      "barlow_loss": 0.17081445455551147,
      "epoch": 1.9935828877005348,
      "loss": 1.4093456268310547,
      "step": 466
    },
    {
      "barlow_loss": 0.1710529923439026,
      "epoch": 1.9935828877005348,
      "loss": 1.5003910064697266,
      "step": 466
    },
    {
      "barlow_loss": 0.1529356837272644,
      "epoch": 1.9935828877005348,
      "loss": 1.5187318325042725,
      "step": 466
    },
    {
      "barlow_loss": 0.14876052737236023,
      "epoch": 1.9935828877005348,
      "loss": 1.3417909145355225,
      "step": 466
    },
    {
      "barlow_loss": 0.1729934960603714,
      "epoch": 1.9935828877005348,
      "loss": 1.2594441175460815,
      "step": 466
    },
    {
      "barlow_loss": 0.16185837984085083,
      "epoch": 1.9935828877005348,
      "loss": 1.4220584630966187,
      "step": 466
    },
    {
      "barlow_loss": 0.17860552668571472,
      "epoch": 1.9935828877005348,
      "loss": 1.468289852142334,
      "step": 466
    },
    {
      "barlow_loss": 0.18543601036071777,
      "epoch": 1.9935828877005348,
      "loss": 1.4418714046478271,
      "step": 466
    },
    {
      "barlow_loss": 0.17100955545902252,
      "epoch": 1.9935828877005348,
      "loss": 1.260468602180481,
      "step": 466
    },
    {
      "barlow_loss": 0.17995333671569824,
      "epoch": 1.9935828877005348,
      "loss": 1.3866759538650513,
      "step": 466
    },
    {
      "barlow_loss": 0.18510162830352783,
      "epoch": 1.9935828877005348,
      "loss": 1.2673366069793701,
      "step": 466
    },
    {
      "barlow_loss": 0.1642880141735077,
      "epoch": 1.9935828877005348,
      "loss": 1.4102625846862793,
      "step": 466
    },
    {
      "barlow_loss": 0.1486741304397583,
      "epoch": 1.9935828877005348,
      "loss": 1.2957640886306763,
      "step": 466
    },
    {
      "barlow_loss": 0.18301615118980408,
      "epoch": 1.9935828877005348,
      "loss": 1.2021743059158325,
      "step": 466
    },
    {
      "barlow_loss": 0.17411869764328003,
      "epoch": 1.9935828877005348,
      "loss": 1.3419735431671143,
      "step": 466
    },
    {
      "barlow_loss": 0.17106059193611145,
      "epoch": 1.9935828877005348,
      "loss": 1.3334590196609497,
      "step": 466
    },
    {
      "barlow_loss": 0.18885882198810577,
      "epoch": 1.9935828877005348,
      "loss": 1.4112632274627686,
      "step": 466
    },
    {
      "barlow_loss": 0.16862396895885468,
      "epoch": 1.9935828877005348,
      "loss": 1.3249468803405762,
      "step": 466
    },
    {
      "barlow_loss": 0.16274772584438324,
      "epoch": 1.9935828877005348,
      "loss": 1.165174126625061,
      "step": 466
    },
    {
      "barlow_loss": 0.16825073957443237,
      "epoch": 1.9935828877005348,
      "loss": 1.2498559951782227,
      "step": 466
    },
    {
      "barlow_loss": 0.18190699815750122,
      "epoch": 1.9935828877005348,
      "loss": 1.4099169969558716,
      "step": 466
    },
    {
      "barlow_loss": 0.16996069252490997,
      "epoch": 1.9935828877005348,
      "loss": 1.4674378633499146,
      "step": 466
    },
    {
      "barlow_loss": 0.1850348711013794,
      "epoch": 1.9935828877005348,
      "loss": 1.2833068370819092,
      "step": 466
    },
    {
      "barlow_loss": 0.1865023970603943,
      "epoch": 1.9935828877005348,
      "loss": 1.3194056749343872,
      "step": 466
    },
    {
      "barlow_loss": 0.1695876568555832,
      "epoch": 1.9935828877005348,
      "loss": 1.2918697595596313,
      "step": 466
    },
    {
      "barlow_loss": 0.16043734550476074,
      "epoch": 1.9935828877005348,
      "loss": 1.2610390186309814,
      "step": 466
    },
    {
      "barlow_loss": 0.1839526891708374,
      "epoch": 1.9935828877005348,
      "loss": 1.3474053144454956,
      "step": 466
    },
    {
      "barlow_loss": 0.1657014936208725,
      "epoch": 1.9935828877005348,
      "loss": 1.2069882154464722,
      "step": 466
    },
    {
      "barlow_loss": 0.1830369234085083,
      "epoch": 1.9935828877005348,
      "loss": 1.170039176940918,
      "step": 466
    },
    {
      "barlow_loss": 0.1584814339876175,
      "epoch": 1.9935828877005348,
      "loss": 1.4474549293518066,
      "step": 466
    },
    {
      "barlow_loss": 0.165033221244812,
      "epoch": 1.9935828877005348,
      "loss": 1.2429169416427612,
      "step": 466
    },
    {
      "barlow_loss": 0.17398658394813538,
      "epoch": 1.9935828877005348,
      "loss": 1.507822036743164,
      "step": 466
    },
    {
      "barlow_loss": 0.18312431871891022,
      "epoch": 1.9935828877005348,
      "loss": 1.2628060579299927,
      "step": 466
    },
    {
      "barlow_loss": 0.17563579976558685,
      "epoch": 1.9935828877005348,
      "loss": 1.1568626165390015,
      "step": 466
    },
    {
      "barlow_loss": 0.1603459268808365,
      "epoch": 1.9935828877005348,
      "loss": 1.343230128288269,
      "step": 466
    },
    {
      "barlow_loss": 0.1857345849275589,
      "epoch": 1.9935828877005348,
      "loss": 1.4762142896652222,
      "step": 466
    },
    {
      "barlow_loss": 0.17132066190242767,
      "epoch": 1.9935828877005348,
      "loss": 1.7056831121444702,
      "step": 466
    },
    {
      "barlow_loss": 0.17759303748607635,
      "epoch": 1.9935828877005348,
      "loss": 1.429875135421753,
      "step": 466
    },
    {
      "barlow_loss": 0.17947198450565338,
      "epoch": 1.9935828877005348,
      "loss": 1.4532474279403687,
      "step": 466
    },
    {
      "barlow_loss": 0.17627069354057312,
      "epoch": 1.9935828877005348,
      "loss": 1.6543322801589966,
      "step": 466
    },
    {
      "barlow_loss": 0.17406782507896423,
      "epoch": 1.9935828877005348,
      "loss": 1.544477939605713,
      "step": 466
    },
    {
      "barlow_loss": 0.1751706898212433,
      "epoch": 1.9935828877005348,
      "loss": 1.2461339235305786,
      "step": 466
    },
    {
      "barlow_loss": 0.17909161746501923,
      "epoch": 1.9935828877005348,
      "loss": 1.5271106958389282,
      "step": 466
    },
    {
      "barlow_loss": 0.1676226556301117,
      "epoch": 1.9935828877005348,
      "loss": 1.418002724647522,
      "step": 466
    },
    {
      "barlow_loss": 0.17159099876880646,
      "epoch": 1.9935828877005348,
      "loss": 1.3825570344924927,
      "step": 466
    },
    {
      "barlow_loss": 0.18266963958740234,
      "epoch": 1.9935828877005348,
      "loss": 1.525394320487976,
      "step": 466
    },
    {
      "barlow_loss": 0.1778537482023239,
      "epoch": 1.9935828877005348,
      "loss": 1.4828462600708008,
      "step": 466
    },
    {
      "barlow_loss": 0.1603221595287323,
      "epoch": 1.9935828877005348,
      "loss": 1.1737736463546753,
      "step": 466
    },
    {
      "barlow_loss": 0.1734057068824768,
      "epoch": 1.9935828877005348,
      "loss": 1.2096301317214966,
      "step": 466
    },
    {
      "barlow_loss": 0.19049862027168274,
      "epoch": 1.9935828877005348,
      "loss": 1.4184372425079346,
      "step": 466
    },
    {
      "barlow_loss": 0.18412604928016663,
      "epoch": 1.9935828877005348,
      "loss": 1.4090133905410767,
      "step": 466
    },
    {
      "barlow_loss": 0.1701204776763916,
      "epoch": 1.9935828877005348,
      "loss": 1.5502761602401733,
      "step": 466
    },
    {
      "barlow_loss": 0.18072034418582916,
      "epoch": 1.9935828877005348,
      "loss": 1.3437683582305908,
      "step": 466
    },
    {
      "barlow_loss": 0.1689855456352234,
      "epoch": 1.9935828877005348,
      "loss": 1.2409815788269043,
      "step": 466
    },
    {
      "barlow_loss": 0.17443789541721344,
      "epoch": 1.9935828877005348,
      "loss": 1.3501379489898682,
      "step": 466
    },
    {
      "barlow_loss": 0.1774132400751114,
      "epoch": 1.9935828877005348,
      "loss": 1.1791174411773682,
      "step": 466
    },
    {
      "barlow_loss": 0.18419328331947327,
      "epoch": 1.9935828877005348,
      "loss": 1.477718472480774,
      "step": 466
    },
    {
      "barlow_loss": 0.19975590705871582,
      "epoch": 1.9935828877005348,
      "loss": 1.5301809310913086,
      "step": 466
    },
    {
      "barlow_loss": 0.17084528505802155,
      "epoch": 1.9935828877005348,
      "loss": 1.3185813426971436,
      "step": 466
    },
    {
      "barlow_loss": 0.16849076747894287,
      "epoch": 1.9935828877005348,
      "loss": 1.5135830640792847,
      "step": 466
    },
    {
      "barlow_loss": 0.17077070474624634,
      "epoch": 1.9935828877005348,
      "loss": 1.373759388923645,
      "step": 466
    },
    {
      "barlow_loss": 0.1809321492910385,
      "epoch": 1.9935828877005348,
      "loss": 1.4293049573898315,
      "step": 466
    },
    {
      "barlow_loss": 0.17917965352535248,
      "epoch": 1.9935828877005348,
      "loss": 1.275863528251648,
      "step": 466
    },
    {
      "barlow_loss": 0.17290373146533966,
      "epoch": 1.9935828877005348,
      "loss": 1.433896541595459,
      "step": 466
    },
    {
      "barlow_loss": 0.16762232780456543,
      "epoch": 1.9935828877005348,
      "loss": 1.3317147493362427,
      "step": 466
    },
    {
      "barlow_loss": 0.16226115822792053,
      "epoch": 1.9935828877005348,
      "loss": 1.3213801383972168,
      "step": 466
    },
    {
      "barlow_loss": 0.17161943018436432,
      "epoch": 1.9935828877005348,
      "loss": 1.4754016399383545,
      "step": 466
    },
    {
      "barlow_loss": 0.1655489206314087,
      "epoch": 1.9935828877005348,
      "loss": 1.4737575054168701,
      "step": 466
    },
    {
      "barlow_loss": 0.18282447755336761,
      "epoch": 1.9935828877005348,
      "loss": 1.2608044147491455,
      "step": 466
    },
    {
      "barlow_loss": 0.1776142120361328,
      "epoch": 1.9935828877005348,
      "loss": 1.3675416707992554,
      "step": 466
    },
    {
      "barlow_loss": 0.18181908130645752,
      "epoch": 1.9935828877005348,
      "loss": 1.3989559412002563,
      "step": 466
    },
    {
      "barlow_loss": 0.1638319343328476,
      "epoch": 1.9935828877005348,
      "loss": 1.455249309539795,
      "step": 466
    },
    {
      "barlow_loss": 0.17995138466358185,
      "epoch": 1.9935828877005348,
      "loss": 1.4046584367752075,
      "step": 466
    },
    {
      "barlow_loss": 0.1627897173166275,
      "epoch": 1.9935828877005348,
      "loss": 1.330666422843933,
      "step": 466
    },
    {
      "barlow_loss": 0.1686641275882721,
      "epoch": 1.9935828877005348,
      "loss": 1.3251852989196777,
      "step": 466
    },
    {
      "barlow_loss": 0.17191673815250397,
      "epoch": 1.9935828877005348,
      "loss": 1.2827489376068115,
      "step": 466
    },
    {
      "barlow_loss": 0.17440743744373322,
      "epoch": 1.9935828877005348,
      "loss": 1.482214331626892,
      "step": 466
    },
    {
      "barlow_loss": 0.17655843496322632,
      "epoch": 1.9935828877005348,
      "loss": 1.3518717288970947,
      "step": 466
    },
    {
      "barlow_loss": 0.17045678198337555,
      "epoch": 1.9935828877005348,
      "loss": 1.313044786453247,
      "step": 466
    },
    {
      "barlow_loss": 0.1617840975522995,
      "epoch": 1.9935828877005348,
      "loss": 1.3876713514328003,
      "step": 466
    },
    {
      "barlow_loss": 0.18260642886161804,
      "epoch": 1.9935828877005348,
      "loss": 1.1963380575180054,
      "step": 466
    },
    {
      "barlow_loss": 0.1785857081413269,
      "epoch": 1.9935828877005348,
      "loss": 1.1049002408981323,
      "step": 466
    },
    {
      "barlow_loss": 0.1726684868335724,
      "epoch": 1.9935828877005348,
      "loss": 1.3362637758255005,
      "step": 466
    },
    {
      "barlow_loss": 0.1574617624282837,
      "epoch": 1.9935828877005348,
      "loss": 1.2811108827590942,
      "step": 466
    },
    {
      "barlow_loss": 0.18737158179283142,
      "epoch": 1.9935828877005348,
      "loss": 1.3237464427947998,
      "step": 466
    },
    {
      "barlow_loss": 0.1639525592327118,
      "epoch": 1.9935828877005348,
      "loss": 1.3033792972564697,
      "step": 466
    },
    {
      "barlow_loss": 0.17850393056869507,
      "epoch": 1.9935828877005348,
      "loss": 1.343427300453186,
      "step": 466
    },
    {
      "barlow_loss": 0.18801645934581757,
      "epoch": 1.9935828877005348,
      "loss": 1.4302083253860474,
      "step": 466
    },
    {
      "epoch": 1.9935828877005348,
      "eval_loss": 1.3928166627883911,
      "eval_runtime": 840.8293,
      "eval_samples_per_second": 1.569,
      "eval_steps_per_second": 0.392,
      "step": 466
    },
    {
      "epoch": 1.9935828877005348,
      "step": 466,
      "total_flos": 1.3898406524273295e+18,
      "train_loss": 1.4915099583981886,
      "train_runtime": 9407.7438,
      "train_samples_per_second": 1.589,
      "train_steps_per_second": 0.05
    }
  ],
  "logging_steps": 4,
  "max_steps": 466,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 1.3898406524273295e+18,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}