{
  "best_metric": 0.6790379881858826,
  "best_model_checkpoint": "petsona-cat/checkpoint-1835",
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 1835,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.04904632152588556,
      "grad_norm": 13.090154647827148,
      "learning_rate": 4.891304347826087e-06,
      "loss": 2.2711,
      "step": 18
    },
    {
      "epoch": 0.09809264305177112,
      "grad_norm": 10.665365219116211,
      "learning_rate": 9.782608695652175e-06,
      "loss": 2.2911,
      "step": 36
    },
    {
      "epoch": 0.14713896457765668,
      "grad_norm": 9.563919067382812,
      "learning_rate": 1.4673913043478263e-05,
      "loss": 2.059,
      "step": 54
    },
    {
      "epoch": 0.19618528610354224,
      "grad_norm": 11.644034385681152,
      "learning_rate": 1.956521739130435e-05,
      "loss": 1.9083,
      "step": 72
    },
    {
      "epoch": 0.2452316076294278,
      "grad_norm": 9.686708450317383,
      "learning_rate": 2.4456521739130436e-05,
      "loss": 1.7345,
      "step": 90
    },
    {
      "epoch": 0.29427792915531337,
      "grad_norm": 8.462738037109375,
      "learning_rate": 2.9347826086956526e-05,
      "loss": 1.5031,
      "step": 108
    },
    {
      "epoch": 0.34332425068119893,
      "grad_norm": 10.496886253356934,
      "learning_rate": 3.423913043478261e-05,
      "loss": 1.3148,
      "step": 126
    },
    {
      "epoch": 0.3923705722070845,
      "grad_norm": 10.159104347229004,
      "learning_rate": 3.91304347826087e-05,
      "loss": 1.2048,
      "step": 144
    },
    {
      "epoch": 0.44141689373297005,
      "grad_norm": 7.318648338317871,
      "learning_rate": 4.4021739130434786e-05,
      "loss": 1.2868,
      "step": 162
    },
    {
      "epoch": 0.4904632152588556,
      "grad_norm": 11.507461547851562,
      "learning_rate": 4.891304347826087e-05,
      "loss": 1.2105,
      "step": 180
    },
    {
      "epoch": 0.5395095367847411,
      "grad_norm": 5.680470943450928,
      "learning_rate": 4.957601453664446e-05,
      "loss": 1.0724,
      "step": 198
    },
    {
      "epoch": 0.5885558583106267,
      "grad_norm": 8.683788299560547,
      "learning_rate": 4.9030890369473045e-05,
      "loss": 1.1899,
      "step": 216
    },
    {
      "epoch": 0.6376021798365122,
      "grad_norm": 7.334160804748535,
      "learning_rate": 4.848576620230164e-05,
      "loss": 1.0224,
      "step": 234
    },
    {
      "epoch": 0.6866485013623979,
      "grad_norm": 8.717308044433594,
      "learning_rate": 4.794064203513023e-05,
      "loss": 0.9976,
      "step": 252
    },
    {
      "epoch": 0.7356948228882834,
      "grad_norm": 12.613802909851074,
      "learning_rate": 4.739551786795882e-05,
      "loss": 0.9753,
      "step": 270
    },
    {
      "epoch": 0.784741144414169,
      "grad_norm": 7.462376594543457,
      "learning_rate": 4.6850393700787405e-05,
      "loss": 1.1063,
      "step": 288
    },
    {
      "epoch": 0.8337874659400545,
      "grad_norm": 8.719904899597168,
      "learning_rate": 4.630526953361599e-05,
      "loss": 1.0311,
      "step": 306
    },
    {
      "epoch": 0.8828337874659401,
      "grad_norm": 9.132880210876465,
      "learning_rate": 4.576014536644458e-05,
      "loss": 0.9735,
      "step": 324
    },
    {
      "epoch": 0.9318801089918256,
      "grad_norm": 8.101548194885254,
      "learning_rate": 4.521502119927317e-05,
      "loss": 1.0608,
      "step": 342
    },
    {
      "epoch": 0.9809264305177112,
      "grad_norm": 9.457178115844727,
      "learning_rate": 4.466989703210176e-05,
      "loss": 1.0543,
      "step": 360
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.7394270122783083,
      "eval_f1_macro": 0.6775647712813081,
      "eval_f1_micro": 0.7394270122783083,
      "eval_f1_weighted": 0.7229253713683983,
      "eval_loss": 0.7820252180099487,
      "eval_precision_macro": 0.6673572429051899,
      "eval_precision_micro": 0.7394270122783083,
      "eval_precision_weighted": 0.7129395935578096,
      "eval_recall_macro": 0.6932339656729901,
      "eval_recall_micro": 0.7394270122783083,
      "eval_recall_weighted": 0.7394270122783083,
      "eval_runtime": 501.3115,
      "eval_samples_per_second": 1.462,
      "eval_steps_per_second": 0.092,
      "step": 367
    },
    {
      "epoch": 1.0299727520435966,
      "grad_norm": 6.45409631729126,
      "learning_rate": 4.4124772864930345e-05,
      "loss": 0.8511,
      "step": 378
    },
    {
      "epoch": 1.0790190735694822,
      "grad_norm": 4.8151164054870605,
      "learning_rate": 4.357964869775893e-05,
      "loss": 0.7645,
      "step": 396
    },
    {
      "epoch": 1.1280653950953679,
      "grad_norm": 6.06305456161499,
      "learning_rate": 4.3034524530587525e-05,
      "loss": 0.7095,
      "step": 414
    },
    {
      "epoch": 1.1771117166212535,
      "grad_norm": 8.338302612304688,
      "learning_rate": 4.248940036341612e-05,
      "loss": 0.7537,
      "step": 432
    },
    {
      "epoch": 1.226158038147139,
      "grad_norm": 8.91988754272461,
      "learning_rate": 4.1944276196244705e-05,
      "loss": 1.0268,
      "step": 450
    },
    {
      "epoch": 1.2752043596730245,
      "grad_norm": 8.219625473022461,
      "learning_rate": 4.139915202907329e-05,
      "loss": 0.7882,
      "step": 468
    },
    {
      "epoch": 1.32425068119891,
      "grad_norm": 6.649393081665039,
      "learning_rate": 4.085402786190188e-05,
      "loss": 0.9177,
      "step": 486
    },
    {
      "epoch": 1.3732970027247957,
      "grad_norm": 10.92117977142334,
      "learning_rate": 4.030890369473047e-05,
      "loss": 0.8006,
      "step": 504
    },
    {
      "epoch": 1.422343324250681,
      "grad_norm": 7.338125705718994,
      "learning_rate": 3.976377952755906e-05,
      "loss": 0.7833,
      "step": 522
    },
    {
      "epoch": 1.4713896457765667,
      "grad_norm": 7.391980171203613,
      "learning_rate": 3.9218655360387645e-05,
      "loss": 0.794,
      "step": 540
    },
    {
      "epoch": 1.5204359673024523,
      "grad_norm": 7.255425930023193,
      "learning_rate": 3.867353119321623e-05,
      "loss": 0.6818,
      "step": 558
    },
    {
      "epoch": 1.569482288828338,
      "grad_norm": 9.407766342163086,
      "learning_rate": 3.812840702604482e-05,
      "loss": 0.7966,
      "step": 576
    },
    {
      "epoch": 1.6185286103542236,
      "grad_norm": 8.508045196533203,
      "learning_rate": 3.758328285887341e-05,
      "loss": 0.7673,
      "step": 594
    },
    {
      "epoch": 1.667574931880109,
      "grad_norm": 10.17292308807373,
      "learning_rate": 3.7038158691702005e-05,
      "loss": 0.8005,
      "step": 612
    },
    {
      "epoch": 1.7166212534059946,
      "grad_norm": 7.042759895324707,
      "learning_rate": 3.649303452453059e-05,
      "loss": 0.7716,
      "step": 630
    },
    {
      "epoch": 1.76566757493188,
      "grad_norm": 10.45976448059082,
      "learning_rate": 3.594791035735918e-05,
      "loss": 0.722,
      "step": 648
    },
    {
      "epoch": 1.8147138964577656,
      "grad_norm": 10.023469924926758,
      "learning_rate": 3.5402786190187765e-05,
      "loss": 0.6968,
      "step": 666
    },
    {
      "epoch": 1.8637602179836512,
      "grad_norm": 8.609762191772461,
      "learning_rate": 3.485766202301636e-05,
      "loss": 0.8295,
      "step": 684
    },
    {
      "epoch": 1.9128065395095368,
      "grad_norm": 5.7018842697143555,
      "learning_rate": 3.4312537855844944e-05,
      "loss": 0.7729,
      "step": 702
    },
    {
      "epoch": 1.9618528610354224,
      "grad_norm": 9.336709022521973,
      "learning_rate": 3.376741368867353e-05,
      "loss": 0.8679,
      "step": 720
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7708049113233287,
      "eval_f1_macro": 0.744396838344231,
      "eval_f1_micro": 0.7708049113233287,
      "eval_f1_weighted": 0.769490062085142,
      "eval_loss": 0.7198029160499573,
      "eval_precision_macro": 0.7471497356802382,
      "eval_precision_micro": 0.7708049113233287,
      "eval_precision_weighted": 0.7700316353146535,
      "eval_recall_macro": 0.7438821138211381,
      "eval_recall_micro": 0.7708049113233287,
      "eval_recall_weighted": 0.7708049113233287,
      "eval_runtime": 526.4938,
      "eval_samples_per_second": 1.392,
      "eval_steps_per_second": 0.087,
      "step": 734
    },
    {
      "epoch": 2.010899182561308,
      "grad_norm": 3.749079942703247,
      "learning_rate": 3.322228952150212e-05,
      "loss": 0.7638,
      "step": 738
    },
    {
      "epoch": 2.0599455040871932,
      "grad_norm": 5.066211223602295,
      "learning_rate": 3.2677165354330704e-05,
      "loss": 0.7479,
      "step": 756
    },
    {
      "epoch": 2.108991825613079,
      "grad_norm": 5.5539116859436035,
      "learning_rate": 3.21320411871593e-05,
      "loss": 0.6329,
      "step": 774
    },
    {
      "epoch": 2.1580381471389645,
      "grad_norm": 7.377701282501221,
      "learning_rate": 3.158691701998789e-05,
      "loss": 0.5928,
      "step": 792
    },
    {
      "epoch": 2.20708446866485,
      "grad_norm": 1.7499698400497437,
      "learning_rate": 3.104179285281648e-05,
      "loss": 0.5315,
      "step": 810
    },
    {
      "epoch": 2.2561307901907357,
      "grad_norm": 12.155342102050781,
      "learning_rate": 3.0496668685645064e-05,
      "loss": 0.6642,
      "step": 828
    },
    {
      "epoch": 2.3051771117166213,
      "grad_norm": 7.592235088348389,
      "learning_rate": 2.9951544518473658e-05,
      "loss": 0.5364,
      "step": 846
    },
    {
      "epoch": 2.354223433242507,
      "grad_norm": 8.232931137084961,
      "learning_rate": 2.9406420351302244e-05,
      "loss": 0.5136,
      "step": 864
    },
    {
      "epoch": 2.4032697547683926,
      "grad_norm": 7.107213497161865,
      "learning_rate": 2.886129618413083e-05,
      "loss": 0.5777,
      "step": 882
    },
    {
      "epoch": 2.452316076294278,
      "grad_norm": 4.498622894287109,
      "learning_rate": 2.8316172016959417e-05,
      "loss": 0.6736,
      "step": 900
    },
    {
      "epoch": 2.5013623978201633,
      "grad_norm": 13.404940605163574,
      "learning_rate": 2.7771047849788007e-05,
      "loss": 0.6628,
      "step": 918
    },
    {
      "epoch": 2.550408719346049,
      "grad_norm": 10.440755844116211,
      "learning_rate": 2.72259236826166e-05,
      "loss": 0.5867,
      "step": 936
    },
    {
      "epoch": 2.5994550408719346,
      "grad_norm": 5.166727066040039,
      "learning_rate": 2.6680799515445187e-05,
      "loss": 0.6759,
      "step": 954
    },
    {
      "epoch": 2.64850136239782,
      "grad_norm": 6.91835355758667,
      "learning_rate": 2.6135675348273774e-05,
      "loss": 0.8065,
      "step": 972
    },
    {
      "epoch": 2.697547683923706,
      "grad_norm": 5.450559139251709,
      "learning_rate": 2.5590551181102364e-05,
      "loss": 0.5886,
      "step": 990
    },
    {
      "epoch": 2.7465940054495914,
      "grad_norm": 11.601627349853516,
      "learning_rate": 2.504542701393095e-05,
      "loss": 0.8291,
      "step": 1008
    },
    {
      "epoch": 2.7956403269754766,
      "grad_norm": 6.433958530426025,
      "learning_rate": 2.450030284675954e-05,
      "loss": 0.6658,
      "step": 1026
    },
    {
      "epoch": 2.844686648501362,
      "grad_norm": 8.672781944274902,
      "learning_rate": 2.395517867958813e-05,
      "loss": 0.577,
      "step": 1044
    },
    {
      "epoch": 2.893732970027248,
      "grad_norm": 5.865307807922363,
      "learning_rate": 2.3410054512416717e-05,
      "loss": 0.6015,
      "step": 1062
    },
    {
      "epoch": 2.9427792915531334,
      "grad_norm": 7.018499374389648,
      "learning_rate": 2.2864930345245307e-05,
      "loss": 0.5292,
      "step": 1080
    },
    {
      "epoch": 2.991825613079019,
      "grad_norm": 8.684929847717285,
      "learning_rate": 2.2319806178073897e-05,
      "loss": 0.6743,
      "step": 1098
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7762619372442019,
      "eval_f1_macro": 0.7352337273015235,
      "eval_f1_micro": 0.7762619372442019,
      "eval_f1_weighted": 0.7700415467128165,
      "eval_loss": 0.7334316372871399,
      "eval_precision_macro": 0.7602211726711597,
      "eval_precision_micro": 0.7762619372442019,
      "eval_precision_weighted": 0.7759104610468709,
      "eval_recall_macro": 0.7309406052393856,
      "eval_recall_micro": 0.7762619372442019,
      "eval_recall_weighted": 0.7762619372442019,
      "eval_runtime": 468.0039,
      "eval_samples_per_second": 1.566,
      "eval_steps_per_second": 0.098,
      "step": 1101
    },
    {
      "epoch": 3.0408719346049047,
      "grad_norm": 5.529566287994385,
      "learning_rate": 2.1774682010902484e-05,
      "loss": 0.5119,
      "step": 1116
    },
    {
      "epoch": 3.0899182561307903,
      "grad_norm": 8.672910690307617,
      "learning_rate": 2.1229557843731074e-05,
      "loss": 0.3846,
      "step": 1134
    },
    {
      "epoch": 3.138964577656676,
      "grad_norm": 6.369070529937744,
      "learning_rate": 2.068443367655966e-05,
      "loss": 0.4705,
      "step": 1152
    },
    {
      "epoch": 3.1880108991825615,
      "grad_norm": 6.632961273193359,
      "learning_rate": 2.013930950938825e-05,
      "loss": 0.5078,
      "step": 1170
    },
    {
      "epoch": 3.2370572207084467,
      "grad_norm": 13.185933113098145,
      "learning_rate": 1.959418534221684e-05,
      "loss": 0.4132,
      "step": 1188
    },
    {
      "epoch": 3.2861035422343323,
      "grad_norm": 2.94797682762146,
      "learning_rate": 1.9049061175045427e-05,
      "loss": 0.5002,
      "step": 1206
    },
    {
      "epoch": 3.335149863760218,
      "grad_norm": 8.657658576965332,
      "learning_rate": 1.8503937007874017e-05,
      "loss": 0.7647,
      "step": 1224
    },
    {
      "epoch": 3.3841961852861036,
      "grad_norm": 7.174124717712402,
      "learning_rate": 1.7958812840702604e-05,
      "loss": 0.5301,
      "step": 1242
    },
    {
      "epoch": 3.433242506811989,
      "grad_norm": 8.120543479919434,
      "learning_rate": 1.7413688673531194e-05,
      "loss": 0.4428,
      "step": 1260
    },
    {
      "epoch": 3.482288828337875,
      "grad_norm": 10.651689529418945,
      "learning_rate": 1.6868564506359784e-05,
      "loss": 0.6033,
      "step": 1278
    },
    {
      "epoch": 3.53133514986376,
      "grad_norm": 14.153423309326172,
      "learning_rate": 1.6323440339188374e-05,
      "loss": 0.5152,
      "step": 1296
    },
    {
      "epoch": 3.5803814713896456,
      "grad_norm": 8.533557891845703,
      "learning_rate": 1.577831617201696e-05,
      "loss": 0.6689,
      "step": 1314
    },
    {
      "epoch": 3.629427792915531,
      "grad_norm": 12.795997619628906,
      "learning_rate": 1.5233192004845548e-05,
      "loss": 0.4701,
      "step": 1332
    },
    {
      "epoch": 3.678474114441417,
      "grad_norm": 8.181031227111816,
      "learning_rate": 1.4688067837674138e-05,
      "loss": 0.4875,
      "step": 1350
    },
    {
      "epoch": 3.7275204359673024,
      "grad_norm": 8.397069931030273,
      "learning_rate": 1.4142943670502725e-05,
      "loss": 0.4234,
      "step": 1368
    },
    {
      "epoch": 3.776566757493188,
      "grad_norm": 6.6668548583984375,
      "learning_rate": 1.3597819503331317e-05,
      "loss": 0.4799,
      "step": 1386
    },
    {
      "epoch": 3.8256130790190737,
      "grad_norm": 3.7310755252838135,
      "learning_rate": 1.3052695336159903e-05,
      "loss": 0.4241,
      "step": 1404
    },
    {
      "epoch": 3.8746594005449593,
      "grad_norm": 7.294261455535889,
      "learning_rate": 1.2507571168988492e-05,
      "loss": 0.4828,
      "step": 1422
    },
    {
      "epoch": 3.923705722070845,
      "grad_norm": 11.431266784667969,
      "learning_rate": 1.1962447001817082e-05,
      "loss": 0.4837,
      "step": 1440
    },
    {
      "epoch": 3.9727520435967305,
      "grad_norm": 10.630217552185059,
      "learning_rate": 1.141732283464567e-05,
      "loss": 0.4148,
      "step": 1458
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.7926330150068213,
      "eval_f1_macro": 0.7482822112088491,
      "eval_f1_micro": 0.7926330150068213,
      "eval_f1_weighted": 0.7859610744604464,
      "eval_loss": 0.680126428604126,
      "eval_precision_macro": 0.7591045700614258,
      "eval_precision_micro": 0.7926330150068213,
      "eval_precision_weighted": 0.7865124628151742,
      "eval_recall_macro": 0.7507125112917796,
      "eval_recall_micro": 0.7926330150068213,
      "eval_recall_weighted": 0.7926330150068213,
      "eval_runtime": 446.0494,
      "eval_samples_per_second": 1.643,
      "eval_steps_per_second": 0.103,
      "step": 1468
    },
    {
      "epoch": 4.021798365122616,
      "grad_norm": 8.096855163574219,
      "learning_rate": 1.0872198667474258e-05,
      "loss": 0.4162,
      "step": 1476
    },
    {
      "epoch": 4.070844686648502,
      "grad_norm": 4.436280250549316,
      "learning_rate": 1.0327074500302846e-05,
      "loss": 0.3999,
      "step": 1494
    },
    {
      "epoch": 4.1198910081743865,
      "grad_norm": 3.2468502521514893,
      "learning_rate": 9.781950333131436e-06,
      "loss": 0.269,
      "step": 1512
    },
    {
      "epoch": 4.168937329700272,
      "grad_norm": 5.88806676864624,
      "learning_rate": 9.236826165960025e-06,
      "loss": 0.3749,
      "step": 1530
    },
    {
      "epoch": 4.217983651226158,
      "grad_norm": 1.9926618337631226,
      "learning_rate": 8.691701998788613e-06,
      "loss": 0.3893,
      "step": 1548
    },
    {
      "epoch": 4.267029972752043,
      "grad_norm": 7.686309814453125,
      "learning_rate": 8.146577831617203e-06,
      "loss": 0.3134,
      "step": 1566
    },
    {
      "epoch": 4.316076294277929,
      "grad_norm": 3.9294235706329346,
      "learning_rate": 7.6014536644457905e-06,
      "loss": 0.3415,
      "step": 1584
    },
    {
      "epoch": 4.3651226158038146,
      "grad_norm": 4.247501850128174,
      "learning_rate": 7.056329497274379e-06,
      "loss": 0.4167,
      "step": 1602
    },
    {
      "epoch": 4.4141689373297,
      "grad_norm": 8.502470970153809,
      "learning_rate": 6.511205330102968e-06,
      "loss": 0.5261,
      "step": 1620
    },
    {
      "epoch": 4.463215258855586,
      "grad_norm": 5.880737781524658,
      "learning_rate": 5.966081162931557e-06,
      "loss": 0.5806,
      "step": 1638
    },
    {
      "epoch": 4.512261580381471,
      "grad_norm": 10.550583839416504,
      "learning_rate": 5.420956995760146e-06,
      "loss": 0.3798,
      "step": 1656
    },
    {
      "epoch": 4.561307901907357,
      "grad_norm": 6.919607639312744,
      "learning_rate": 4.875832828588734e-06,
      "loss": 0.4235,
      "step": 1674
    },
    {
      "epoch": 4.610354223433243,
      "grad_norm": 11.466157913208008,
      "learning_rate": 4.330708661417323e-06,
      "loss": 0.4009,
      "step": 1692
    },
    {
      "epoch": 4.659400544959128,
      "grad_norm": 7.428606033325195,
      "learning_rate": 3.785584494245912e-06,
      "loss": 0.3749,
      "step": 1710
    },
    {
      "epoch": 4.708446866485014,
      "grad_norm": 14.596545219421387,
      "learning_rate": 3.2404603270745003e-06,
      "loss": 0.3822,
      "step": 1728
    },
    {
      "epoch": 4.7574931880108995,
      "grad_norm": 1.3340034484863281,
      "learning_rate": 2.695336159903089e-06,
      "loss": 0.402,
      "step": 1746
    },
    {
      "epoch": 4.806539509536785,
      "grad_norm": 12.544507026672363,
      "learning_rate": 2.150211992731678e-06,
      "loss": 0.3577,
      "step": 1764
    },
    {
      "epoch": 4.855585831062671,
      "grad_norm": 7.779797077178955,
      "learning_rate": 1.6050878255602666e-06,
      "loss": 0.5092,
      "step": 1782
    },
    {
      "epoch": 4.904632152588556,
      "grad_norm": 4.97097110748291,
      "learning_rate": 1.0599636583888554e-06,
      "loss": 0.3382,
      "step": 1800
    },
    {
      "epoch": 4.953678474114441,
      "grad_norm": 2.9842166900634766,
      "learning_rate": 5.14839491217444e-07,
      "loss": 0.2765,
      "step": 1818
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.7953615279672579,
      "eval_f1_macro": 0.7441167259748385,
      "eval_f1_micro": 0.7953615279672579,
      "eval_f1_weighted": 0.7877909517913628,
      "eval_loss": 0.6790379881858826,
      "eval_precision_macro": 0.7576756562000598,
      "eval_precision_micro": 0.7953615279672579,
      "eval_precision_weighted": 0.7895482398510546,
      "eval_recall_macro": 0.7464961607949413,
      "eval_recall_micro": 0.7953615279672579,
      "eval_recall_weighted": 0.7953615279672579,
      "eval_runtime": 505.8176,
      "eval_samples_per_second": 1.449,
      "eval_steps_per_second": 0.091,
      "step": 1835
    }
  ],
  "logging_steps": 18,
  "max_steps": 1835,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "stateful_callbacks": {
    "EarlyStoppingCallback": {
      "args": {
        "early_stopping_patience": 5,
        "early_stopping_threshold": 0.01
      },
      "attributes": {
        "early_stopping_patience_counter": 0
      }
    },
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 1.1887708574999347e+19,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}