{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.6148445799270241,
  "eval_steps": 1024,
  "global_step": 13312,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.011823934229365849,
      "grad_norm": 2.166717529296875,
      "learning_rate": 0.000498046875,
      "loss": 6.863234996795654,
      "step": 256
    },
    {
      "epoch": 0.023647868458731697,
      "grad_norm": 0.9976925849914551,
      "learning_rate": 0.000998046875,
      "loss": 1.0293713808059692,
      "step": 512
    },
    {
      "epoch": 0.03547180268809755,
      "grad_norm": 0.9951309561729431,
      "learning_rate": 0.000999640996023194,
      "loss": 0.5387133359909058,
      "step": 768
    },
    {
      "epoch": 0.047295736917463395,
      "grad_norm": 1.1936287879943848,
      "learning_rate": 0.0009985588674043958,
      "loss": 0.4395764172077179,
      "step": 1024
    },
    {
      "epoch": 0.047295736917463395,
      "eval_bleu": 0.630805532752935,
      "eval_ce_loss": 1.5585694417833738,
      "eval_cos_loss": 0.12051772464015713,
      "eval_loss": 0.41065708449169924,
      "eval_mse_loss": 0.22680107500727317,
      "eval_rec_loss": 0.015947288232876763,
      "flow/cos_sim": 0.8794823028997744,
      "flow/improvement_ratio": 0.9763019112147153,
      "flow/mag_ratio_mean": 0.8738831932141901,
      "flow/mag_ratio_std": 0.17382950611310463,
      "step": 1024
    },
    {
      "epoch": 0.047295736917463395,
      "eval_bleu": 0.630805532752935,
      "eval_ce_loss": 1.5585694417833738,
      "eval_cos_loss": 0.12051772464015713,
      "eval_loss": 0.41065708449169924,
      "eval_mse_loss": 0.22680107500727317,
      "eval_rec_loss": 0.015947288232876763,
      "eval_runtime": 149.5036,
      "eval_samples_per_second": 187.24,
      "eval_steps_per_second": 2.93,
      "flow/cos_sim": 0.8794823028997744,
      "flow/improvement_ratio": 0.9763019112147153,
      "flow/mag_ratio_mean": 0.8738831932141901,
      "flow/mag_ratio_std": 0.17382950611310463,
      "step": 1024
    },
    {
      "epoch": 0.05911967114682925,
      "grad_norm": 1.413312315940857,
      "learning_rate": 0.0009967551747861387,
      "loss": 0.40079358220100403,
      "step": 1280
    },
    {
      "epoch": 0.0709436053761951,
      "grad_norm": 0.4399195909500122,
      "learning_rate": 0.000994232528651847,
      "loss": 0.38004180788993835,
      "step": 1536
    },
    {
      "epoch": 0.08276753960556095,
      "grad_norm": 1.061546802520752,
      "learning_rate": 0.0009909945800260092,
      "loss": 0.3571281135082245,
      "step": 1792
    },
    {
      "epoch": 0.09459147383492679,
      "grad_norm": 0.4095996916294098,
      "learning_rate": 0.0009870460151900522,
      "loss": 0.34413549304008484,
      "step": 2048
    },
    {
      "epoch": 0.09459147383492679,
      "eval_bleu": 0.6370628963183373,
      "eval_ce_loss": 1.4389138337411838,
      "eval_cos_loss": 0.09142084970866164,
      "eval_loss": 0.33612148960431415,
      "eval_mse_loss": 0.17856517835566987,
      "eval_rec_loss": 0.004522840211489004,
      "flow/cos_sim": 0.9085791546460156,
      "flow/improvement_ratio": 0.9776561622205935,
      "flow/mag_ratio_mean": 0.9092166116248527,
      "flow/mag_ratio_std": 0.14505240416417928,
      "step": 2048
    },
    {
      "epoch": 0.09459147383492679,
      "eval_bleu": 0.6370628963183373,
      "eval_ce_loss": 1.4389138337411838,
      "eval_cos_loss": 0.09142084970866164,
      "eval_loss": 0.33612148960431415,
      "eval_mse_loss": 0.17856517835566987,
      "eval_rec_loss": 0.004522840211489004,
      "eval_runtime": 144.0023,
      "eval_samples_per_second": 194.393,
      "eval_steps_per_second": 3.042,
      "flow/cos_sim": 0.9085791546460156,
      "flow/improvement_ratio": 0.9776561622205935,
      "flow/mag_ratio_mean": 0.9092166116248527,
      "flow/mag_ratio_std": 0.14505240416417928,
      "step": 2048
    },
    {
      "epoch": 0.10641540806429264,
      "grad_norm": 1.3680005073547363,
      "learning_rate": 0.0009823925488998885,
      "loss": 0.3346008360385895,
      "step": 2304
    },
    {
      "epoch": 0.1182393422936585,
      "grad_norm": 0.9834803938865662,
      "learning_rate": 0.0009770409161149525,
      "loss": 0.32358497381210327,
      "step": 2560
    },
    {
      "epoch": 0.13006327652302435,
      "grad_norm": 0.7012990117073059,
      "learning_rate": 0.0009709988622506973,
      "loss": 0.31395941972732544,
      "step": 2816
    },
    {
      "epoch": 0.1418872107523902,
      "grad_norm": 0.4390225112438202,
      "learning_rate": 0.000964275131968659,
      "loss": 0.30854108929634094,
      "step": 3072
    },
    {
      "epoch": 0.1418872107523902,
      "eval_bleu": 0.6707740050213097,
      "eval_ce_loss": 1.2450400457534616,
      "eval_cos_loss": 0.08531694868622033,
      "eval_loss": 0.3036313052523082,
      "eval_mse_loss": 0.16824311817481638,
      "eval_rec_loss": 0.0023524861730428916,
      "flow/cos_sim": 0.9146830740312463,
      "flow/improvement_ratio": 0.9760218352487643,
      "flow/mag_ratio_mean": 0.9162033348595171,
      "flow/mag_ratio_std": 0.12317379108166586,
      "step": 3072
    },
    {
      "epoch": 0.1418872107523902,
      "eval_bleu": 0.6707740050213097,
      "eval_ce_loss": 1.2450400457534616,
      "eval_cos_loss": 0.08531694868622033,
      "eval_loss": 0.3036313052523082,
      "eval_mse_loss": 0.16824311817481638,
      "eval_rec_loss": 0.0023524861730428916,
      "eval_runtime": 145.6816,
      "eval_samples_per_second": 192.152,
      "eval_steps_per_second": 3.007,
      "flow/cos_sim": 0.9146830740312463,
      "flow/improvement_ratio": 0.9760218352487643,
      "flow/mag_ratio_mean": 0.9162033348595171,
      "flow/mag_ratio_std": 0.12317379108166586,
      "step": 3072
    },
    {
      "epoch": 0.15371114498175603,
      "grad_norm": 0.6193675398826599,
      "learning_rate": 0.0009568794565203123,
      "loss": 0.30133724212646484,
      "step": 3328
    },
    {
      "epoch": 0.1655350792111219,
      "grad_norm": 0.5207021236419678,
      "learning_rate": 0.0009488225396630347,
      "loss": 0.29369795322418213,
      "step": 3584
    },
    {
      "epoch": 0.17735901344048774,
      "grad_norm": 0.5521863102912903,
      "learning_rate": 0.0009401160421685646,
      "loss": 0.2897779941558838,
      "step": 3840
    },
    {
      "epoch": 0.18918294766985358,
      "grad_norm": 0.4839329421520233,
      "learning_rate": 0.0009307725649463714,
      "loss": 0.2864342927932739,
      "step": 4096
    },
    {
      "epoch": 0.18918294766985358,
      "eval_bleu": 0.6704121655421459,
      "eval_ce_loss": 1.2046921730858007,
      "eval_cos_loss": 0.07965970783575213,
      "eval_loss": 0.2842630836857508,
      "eval_mse_loss": 0.1541883000534147,
      "eval_rec_loss": 0.0016395924527162992,
      "flow/cos_sim": 0.9203403159363629,
      "flow/improvement_ratio": 0.9744544083669305,
      "flow/mag_ratio_mean": 0.9212693351588838,
      "flow/mag_ratio_std": 0.10148854444832563,
      "step": 4096
    },
    {
      "epoch": 0.18918294766985358,
      "eval_bleu": 0.6704121655421459,
      "eval_ce_loss": 1.2046921730858007,
      "eval_cos_loss": 0.07965970783575213,
      "eval_loss": 0.2842630836857508,
      "eval_mse_loss": 0.1541883000534147,
      "eval_rec_loss": 0.0016395924527162992,
      "eval_runtime": 144.2359,
      "eval_samples_per_second": 194.078,
      "eval_steps_per_second": 3.037,
      "flow/cos_sim": 0.9203403159363629,
      "flow/improvement_ratio": 0.9744544083669305,
      "flow/mag_ratio_mean": 0.9212693351588838,
      "flow/mag_ratio_std": 0.10148854444832563,
      "step": 4096
    },
    {
      "epoch": 0.20100688189921945,
      "grad_norm": 0.9023747444152832,
      "learning_rate": 0.0009208056308063659,
      "loss": 0.30037108063697815,
      "step": 4352
    },
    {
      "epoch": 0.2128308161285853,
      "grad_norm": 0.8910952806472778,
      "learning_rate": 0.0009102296648873445,
      "loss": 0.28084370493888855,
      "step": 4608
    },
    {
      "epoch": 0.22465475035795113,
      "grad_norm": 0.43350881338119507,
      "learning_rate": 0.0008990599737794927,
      "loss": 0.276326060295105,
      "step": 4864
    },
    {
      "epoch": 0.236478684587317,
      "grad_norm": 0.4241744577884674,
      "learning_rate": 0.0008873127233711644,
      "loss": 0.2720130980014801,
      "step": 5120
    },
    {
      "epoch": 0.236478684587317,
      "eval_bleu": 0.6884395396484843,
      "eval_ce_loss": 1.1380460211947627,
      "eval_cos_loss": 0.0743916420410621,
      "eval_loss": 0.26738106996909666,
      "eval_mse_loss": 0.14492133019925796,
      "eval_rec_loss": 0.0012159729793058606,
      "flow/cos_sim": 0.9256083738041795,
      "flow/improvement_ratio": 0.9722109245110865,
      "flow/mag_ratio_mean": 0.9361066815515632,
      "flow/mag_ratio_std": 0.08903367672812994,
      "step": 5120
    },
    {
      "epoch": 0.236478684587317,
      "eval_bleu": 0.6884395396484843,
      "eval_ce_loss": 1.1380460211947627,
      "eval_cos_loss": 0.0743916420410621,
      "eval_loss": 0.26738106996909666,
      "eval_mse_loss": 0.14492133019925796,
      "eval_rec_loss": 0.0012159729793058606,
      "eval_runtime": 145.1812,
      "eval_samples_per_second": 192.814,
      "eval_steps_per_second": 3.017,
      "flow/cos_sim": 0.9256083738041795,
      "flow/improvement_ratio": 0.9722109245110865,
      "flow/mag_ratio_mean": 0.9361066815515632,
      "flow/mag_ratio_std": 0.08903367672812994,
      "step": 5120
    },
    {
      "epoch": 0.24830261881668284,
      "grad_norm": 0.7429539561271667,
      "learning_rate": 0.0008750049154520011,
      "loss": 0.2687661945819855,
      "step": 5376
    },
    {
      "epoch": 0.2601265530460487,
      "grad_norm": 0.4944668114185333,
      "learning_rate": 0.0008621543631062487,
      "loss": 0.26752030849456787,
      "step": 5632
    },
    {
      "epoch": 0.27195048727541454,
      "grad_norm": 0.4112600088119507,
      "learning_rate": 0.0008487796649318904,
      "loss": 0.26332396268844604,
      "step": 5888
    },
    {
      "epoch": 0.2837744215047804,
      "grad_norm": 0.5803443789482117,
      "learning_rate": 0.0008349001781229053,
      "loss": 0.2650764286518097,
      "step": 6144
    },
    {
      "epoch": 0.2837744215047804,
      "eval_bleu": 0.6901447099765771,
      "eval_ce_loss": 1.1182073426845411,
      "eval_cos_loss": 0.07293294577701995,
      "eval_loss": 0.2625204250216484,
      "eval_mse_loss": 0.14227469792785166,
      "eval_rec_loss": 0.0011316965778382733,
      "flow/cos_sim": 0.9270670652117359,
      "flow/improvement_ratio": 0.972748759402532,
      "flow/mag_ratio_mean": 0.9329337213681713,
      "flow/mag_ratio_std": 0.08477971859310315,
      "step": 6144
    },
    {
      "epoch": 0.2837744215047804,
      "eval_bleu": 0.6901447099765771,
      "eval_ce_loss": 1.1182073426845411,
      "eval_cos_loss": 0.07293294577701995,
      "eval_loss": 0.2625204250216484,
      "eval_mse_loss": 0.14227469792785166,
      "eval_rec_loss": 0.0011316965778382733,
      "eval_runtime": 143.6154,
      "eval_samples_per_second": 194.916,
      "eval_steps_per_second": 3.05,
      "flow/cos_sim": 0.9270670652117359,
      "flow/improvement_ratio": 0.972748759402532,
      "flow/mag_ratio_mean": 0.9329337213681713,
      "flow/mag_ratio_std": 0.08477971859310315,
      "step": 6144
    },
    {
      "epoch": 0.2955983557341462,
      "grad_norm": 0.5521038770675659,
      "learning_rate": 0.0008205359904536107,
      "loss": 0.2589625120162964,
      "step": 6400
    },
    {
      "epoch": 0.30742228996351206,
      "grad_norm": 0.5204539895057678,
      "learning_rate": 0.0008057078912056363,
      "loss": 0.2580261528491974,
      "step": 6656
    },
    {
      "epoch": 0.3192462241928779,
      "grad_norm": 0.5325313806533813,
      "learning_rate": 0.0007904373410796086,
      "loss": 0.25551143288612366,
      "step": 6912
    },
    {
      "epoch": 0.3310701584222438,
      "grad_norm": 0.3639472424983978,
      "learning_rate": 0.0007747464411350876,
      "loss": 0.25460946559906006,
      "step": 7168
    },
    {
      "epoch": 0.3310701584222438,
      "eval_bleu": 0.7071091482680488,
      "eval_ce_loss": 1.0378137462215336,
      "eval_cos_loss": 0.07100786366521224,
      "eval_loss": 0.2512795734283042,
      "eval_mse_loss": 0.13953843853993503,
      "eval_rec_loss": 0.0008589719193517103,
      "flow/cos_sim": 0.9289921618759904,
      "flow/improvement_ratio": 0.9735576084487514,
      "flow/mag_ratio_mean": 0.9378764381691745,
      "flow/mag_ratio_std": 0.08363016963549401,
      "step": 7168
    },
    {
      "epoch": 0.3310701584222438,
      "eval_bleu": 0.7071091482680488,
      "eval_ce_loss": 1.0378137462215336,
      "eval_cos_loss": 0.07100786366521224,
      "eval_loss": 0.2512795734283042,
      "eval_mse_loss": 0.13953843853993503,
      "eval_rec_loss": 0.0008589719193517103,
      "eval_runtime": 145.8938,
      "eval_samples_per_second": 191.872,
      "eval_steps_per_second": 3.002,
      "flow/cos_sim": 0.9289921618759904,
      "flow/improvement_ratio": 0.9735576084487514,
      "flow/mag_ratio_mean": 0.9378764381691745,
      "flow/mag_ratio_std": 0.08363016963549401,
      "step": 7168
    },
    {
      "epoch": 0.34289409265160964,
      "grad_norm": 0.26728010177612305,
      "learning_rate": 0.000758657900803716,
      "loss": 0.25558242201805115,
      "step": 7424
    },
    {
      "epoch": 0.3547180268809755,
      "grad_norm": 0.3367791175842285,
      "learning_rate": 0.000742195005021869,
      "loss": 0.2547113597393036,
      "step": 7680
    },
    {
      "epoch": 0.3665419611103413,
      "grad_norm": 0.6501409411430359,
      "learning_rate": 0.0007253815805303786,
      "loss": 0.2516462504863739,
      "step": 7936
    },
    {
      "epoch": 0.37836589533970716,
      "grad_norm": 0.41470277309417725,
      "learning_rate": 0.0007082419613901028,
      "loss": 0.25756070017814636,
      "step": 8192
    },
    {
      "epoch": 0.37836589533970716,
      "eval_bleu": 0.7183146382916741,
      "eval_ce_loss": 1.0048845314408001,
      "eval_cos_loss": 0.07200061048478841,
      "eval_loss": 0.2508213480245577,
      "eval_mse_loss": 0.14229333698409333,
      "eval_rec_loss": 0.0008394962100133504,
      "flow/cos_sim": 0.9279994152180137,
      "flow/improvement_ratio": 0.9741122624645494,
      "flow/mag_ratio_mean": 0.9306073022759669,
      "flow/mag_ratio_std": 0.08248136017352478,
      "step": 8192
    },
    {
      "epoch": 0.37836589533970716,
      "eval_bleu": 0.7183146382916741,
      "eval_ce_loss": 1.0048845314408001,
      "eval_cos_loss": 0.07200061048478841,
      "eval_loss": 0.2508213480245577,
      "eval_mse_loss": 0.14229333698409333,
      "eval_rec_loss": 0.0008394962100133504,
      "eval_runtime": 144.5073,
      "eval_samples_per_second": 193.713,
      "eval_steps_per_second": 3.031,
      "flow/cos_sim": 0.9279994152180137,
      "flow/improvement_ratio": 0.9741122624645494,
      "flow/mag_ratio_mean": 0.9306073022759669,
      "flow/mag_ratio_std": 0.08248136017352478,
      "step": 8192
    },
    {
      "epoch": 0.390189829569073,
      "grad_norm": 0.5503519773483276,
      "learning_rate": 0.0006908009537632514,
      "loss": 0.24980439245700836,
      "step": 8448
    },
    {
      "epoch": 0.4020137637984389,
      "grad_norm": 0.5455402135848999,
      "learning_rate": 0.0006730838000114403,
      "loss": 0.25091099739074707,
      "step": 8704
    },
    {
      "epoch": 0.41383769802780473,
      "grad_norm": 0.29903262853622437,
      "learning_rate": 0.0006551161421624341,
      "loss": 0.24810341000556946,
      "step": 8960
    },
    {
      "epoch": 0.4256616322571706,
      "grad_norm": 0.5439685583114624,
      "learning_rate": 0.0006369239847984517,
      "loss": 0.24486252665519714,
      "step": 9216
    },
    {
      "epoch": 0.4256616322571706,
      "eval_bleu": 0.7059557795131848,
      "eval_ce_loss": 1.0441748459045201,
      "eval_cos_loss": 0.06742798249539175,
      "eval_loss": 0.24551006137780404,
      "eval_mse_loss": 0.13372647651730607,
      "eval_rec_loss": 0.0006233008009167367,
      "flow/cos_sim": 0.9325720338516583,
      "flow/improvement_ratio": 0.9731077969618583,
      "flow/mag_ratio_mean": 0.934824386688128,
      "flow/mag_ratio_std": 0.079195149838244,
      "step": 9216
    },
    {
      "epoch": 0.4256616322571706,
      "eval_bleu": 0.7059557795131848,
      "eval_ce_loss": 1.0441748459045201,
      "eval_cos_loss": 0.06742798249539175,
      "eval_loss": 0.24551006137780404,
      "eval_mse_loss": 0.13372647651730607,
      "eval_rec_loss": 0.0006233008009167367,
      "eval_runtime": 143.7536,
      "eval_samples_per_second": 194.729,
      "eval_steps_per_second": 3.047,
      "flow/cos_sim": 0.9325720338516583,
      "flow/improvement_ratio": 0.9731077969618583,
      "flow/mag_ratio_mean": 0.934824386688128,
      "flow/mag_ratio_std": 0.079195149838244,
      "step": 9216
    },
    {
      "epoch": 0.4374855664865364,
      "grad_norm": 0.6247985363006592,
      "learning_rate": 0.0006185336574197479,
      "loss": 0.24821333587169647,
      "step": 9472
    },
    {
      "epoch": 0.44930950071590225,
      "grad_norm": 0.5992924571037292,
      "learning_rate": 0.0005999717763379407,
      "loss": 0.2463277131319046,
      "step": 9728
    },
    {
      "epoch": 0.4611334349452681,
      "grad_norm": 0.5051174759864807,
      "learning_rate": 0.0005812652061542363,
      "loss": 0.24300755560398102,
      "step": 9984
    },
    {
      "epoch": 0.472957369174634,
      "grad_norm": 0.40211889147758484,
      "learning_rate": 0.0005624410208783071,
      "loss": 0.24105887115001678,
      "step": 10240
    },
    {
      "epoch": 0.472957369174634,
      "eval_bleu": 0.7120039792514781,
      "eval_ce_loss": 1.005759668472695,
      "eval_cos_loss": 0.06631913622118295,
      "eval_loss": 0.2399323729619588,
      "eval_mse_loss": 0.1321649712820848,
      "eval_rec_loss": 0.0005595201312656808,
      "flow/cos_sim": 0.9336808790627136,
      "flow/improvement_ratio": 0.9762771404497156,
      "flow/mag_ratio_mean": 0.926664409572131,
      "flow/mag_ratio_std": 0.0785302072763443,
      "step": 10240
    },
    {
      "epoch": 0.472957369174634,
      "eval_bleu": 0.7120039792514781,
      "eval_ce_loss": 1.005759668472695,
      "eval_cos_loss": 0.06631913622118295,
      "eval_loss": 0.2399323729619588,
      "eval_mse_loss": 0.1321649712820848,
      "eval_rec_loss": 0.0005595201312656808,
      "eval_runtime": 145.1926,
      "eval_samples_per_second": 192.799,
      "eval_steps_per_second": 3.017,
      "flow/cos_sim": 0.9336808790627136,
      "flow/improvement_ratio": 0.9762771404497156,
      "flow/mag_ratio_mean": 0.926664409572131,
      "flow/mag_ratio_std": 0.0785302072763443,
      "step": 10240
    },
    {
      "epoch": 0.48478130340399983,
      "grad_norm": 0.5163204073905945,
      "learning_rate": 0.0005435264647440881,
      "loss": 0.2431734949350357,
      "step": 10496
    },
    {
      "epoch": 0.49660523763336567,
      "grad_norm": 0.3941446840763092,
      "learning_rate": 0.000524548912779213,
      "loss": 0.23893995583057404,
      "step": 10752
    },
    {
      "epoch": 0.5084291718627315,
      "grad_norm": 0.600309431552887,
      "learning_rate": 0.0005055358311851499,
      "loss": 0.24141335487365723,
      "step": 11008
    },
    {
      "epoch": 0.5202531060920974,
      "grad_norm": 0.2635146975517273,
      "learning_rate": 0.0004865147375853812,
      "loss": 0.23994745314121246,
      "step": 11264
    },
    {
      "epoch": 0.5202531060920974,
      "eval_bleu": 0.721584420569106,
      "eval_ce_loss": 0.9708276380142665,
      "eval_cos_loss": 0.06598166635761794,
      "eval_loss": 0.236370724664159,
      "eval_mse_loss": 0.13219797124737473,
      "eval_rec_loss": 0.0004918215776982103,
      "flow/cos_sim": 0.9340183556624199,
      "flow/improvement_ratio": 0.9756744954411842,
      "flow/mag_ratio_mean": 0.9309794351662675,
      "flow/mag_ratio_std": 0.07776250285372886,
      "step": 11264
    },
    {
      "epoch": 0.5202531060920974,
      "eval_bleu": 0.721584420569106,
      "eval_ce_loss": 0.9708276380142665,
      "eval_cos_loss": 0.06598166635761794,
      "eval_loss": 0.236370724664159,
      "eval_mse_loss": 0.13219797124737473,
      "eval_rec_loss": 0.0004918215776982103,
      "eval_runtime": 145.3935,
      "eval_samples_per_second": 192.533,
      "eval_steps_per_second": 3.013,
      "flow/cos_sim": 0.9340183556624199,
      "flow/improvement_ratio": 0.9756744954411842,
      "flow/mag_ratio_mean": 0.9309794351662675,
      "flow/mag_ratio_std": 0.07776250285372886,
      "step": 11264
    },
    {
      "epoch": 0.5320770403214632,
      "grad_norm": 0.7631673216819763,
      "learning_rate": 0.0004675131611991607,
      "loss": 0.2390274852514267,
      "step": 11520
    },
    {
      "epoch": 0.5439009745508291,
      "grad_norm": 0.3735812306404114,
      "learning_rate": 0.0004485586029984899,
      "loss": 0.23720039427280426,
      "step": 11776
    },
    {
      "epoch": 0.5557249087801949,
      "grad_norm": 0.31787776947021484,
      "learning_rate": 0.00042967849590597266,
      "loss": 0.2358924150466919,
      "step": 12032
    },
    {
      "epoch": 0.5675488430095608,
      "grad_norm": 0.4151494801044464,
      "learning_rate": 0.0004109001650911621,
      "loss": 0.23592068254947662,
      "step": 12288
    },
    {
      "epoch": 0.5675488430095608,
      "eval_bleu": 0.7193234778931977,
      "eval_ce_loss": 0.9832807419370843,
      "eval_cos_loss": 0.06677046464968762,
      "eval_loss": 0.2394936037118032,
      "eval_mse_loss": 0.1340124544147487,
      "eval_rec_loss": 0.00047602696994042604,
      "flow/cos_sim": 0.933229555278064,
      "flow/improvement_ratio": 0.9741054156871691,
      "flow/mag_ratio_mean": 0.9458617875565133,
      "flow/mag_ratio_std": 0.07636498908201854,
      "step": 12288
    },
    {
      "epoch": 0.5675488430095608,
      "eval_bleu": 0.7193234778931977,
      "eval_ce_loss": 0.9832807419370843,
      "eval_cos_loss": 0.06677046464968762,
      "eval_loss": 0.2394936037118032,
      "eval_mse_loss": 0.1340124544147487,
      "eval_rec_loss": 0.00047602696994042604,
      "eval_runtime": 146.0267,
      "eval_samples_per_second": 191.698,
      "eval_steps_per_second": 2.999,
      "flow/cos_sim": 0.933229555278064,
      "flow/improvement_ratio": 0.9741054156871691,
      "flow/mag_ratio_mean": 0.9458617875565133,
      "flow/mag_ratio_std": 0.07636498908201854,
      "step": 12288
    },
    {
      "epoch": 0.5793727772389267,
      "grad_norm": 0.5829018950462341,
      "learning_rate": 0.0003922507884228551,
      "loss": 0.23718097805976868,
      "step": 12544
    },
    {
      "epoch": 0.5911967114682924,
      "grad_norm": 0.24561040103435516,
      "learning_rate": 0.00037375735713457723,
      "loss": 0.23286746442317963,
      "step": 12800
    },
    {
      "epoch": 0.6030206456976583,
      "grad_norm": 0.2852168381214142,
      "learning_rate": 0.00035544663676018276,
      "loss": 0.23593547940254211,
      "step": 13056
    },
    {
      "epoch": 0.6148445799270241,
      "grad_norm": 0.387464314699173,
      "learning_rate": 0.00033734512839611255,
      "loss": 0.23200258612632751,
      "step": 13312
    },
    {
      "epoch": 0.6148445799270241,
      "eval_bleu": 0.7243775501478836,
      "eval_ce_loss": 0.9625263297802782,
      "eval_cos_loss": 0.06494614632468518,
      "eval_loss": 0.2341489744581044,
      "eval_mse_loss": 0.13099264763465754,
      "eval_rec_loss": 0.00040907774940659357,
      "flow/cos_sim": 0.9350538734159514,
      "flow/improvement_ratio": 0.9740996707792151,
      "flow/mag_ratio_mean": 0.9335339481699957,
      "flow/mag_ratio_std": 0.07727514897987722,
      "step": 13312
    },
    {
      "epoch": 0.6148445799270241,
      "eval_bleu": 0.7243775501478836,
      "eval_ce_loss": 0.9625263297802782,
      "eval_cos_loss": 0.06494614632468518,
      "eval_loss": 0.2341489744581044,
      "eval_mse_loss": 0.13099264763465754,
      "eval_rec_loss": 0.00040907774940659357,
      "eval_runtime": 144.1869,
      "eval_samples_per_second": 194.144,
      "eval_steps_per_second": 3.038,
      "flow/cos_sim": 0.9350538734159514,
      "flow/improvement_ratio": 0.9740996707792151,
      "flow/mag_ratio_mean": 0.9335339481699957,
      "flow/mag_ratio_std": 0.07727514897987722,
      "step": 13312
    }
  ],
  "logging_steps": 256,
  "max_steps": 21651,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 1024,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 64,
  "trial_name": null,
  "trial_params": null
}