| { |
| "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": 1.9140732288360596, |
| "learning_rate": 0.000498046875, |
| "loss": 6.9310712814331055, |
| "step": 256 |
| }, |
| { |
| "epoch": 0.023647868458731697, |
| "grad_norm": 0.966057538986206, |
| "learning_rate": 0.000998046875, |
| "loss": 1.017655372619629, |
| "step": 512 |
| }, |
| { |
| "epoch": 0.03547180268809755, |
| "grad_norm": 0.6036155223846436, |
| "learning_rate": 0.000999640996023194, |
| "loss": 0.5299859046936035, |
| "step": 768 |
| }, |
| { |
| "epoch": 0.047295736917463395, |
| "grad_norm": 0.6692569851875305, |
| "learning_rate": 0.0009985588674043958, |
| "loss": 0.43251049518585205, |
| "step": 1024 |
| }, |
| { |
| "epoch": 0.047295736917463395, |
| "eval_bleu": 0.6267380536600159, |
| "eval_ce_loss": 1.563563620935292, |
| "eval_cos_loss": 0.11390362637432198, |
| "eval_loss": 0.40128198402113024, |
| "eval_mse_loss": 0.2176408769061032, |
| "eval_rec_loss": 0.015894381031734215, |
| "flow/cos_sim": 0.8860963983078526, |
| "flow/improvement_ratio": 0.9765386980146034, |
| "flow/mag_ratio_mean": 0.8838755380889597, |
| "flow/mag_ratio_std": 0.1646749794006892, |
| "step": 1024 |
| }, |
| { |
| "epoch": 0.047295736917463395, |
| "eval_bleu": 0.6267380536600159, |
| "eval_ce_loss": 1.563563620935292, |
| "eval_cos_loss": 0.11390362637432198, |
| "eval_loss": 0.40128198402113024, |
| "eval_mse_loss": 0.2176408769061032, |
| "eval_rec_loss": 0.015894381031734215, |
| "eval_runtime": 145.0392, |
| "eval_samples_per_second": 193.003, |
| "eval_steps_per_second": 3.02, |
| "flow/cos_sim": 0.8860963983078526, |
| "flow/improvement_ratio": 0.9765386980146034, |
| "flow/mag_ratio_mean": 0.8838755380889597, |
| "flow/mag_ratio_std": 0.1646749794006892, |
| "step": 1024 |
| }, |
| { |
| "epoch": 0.05911967114682925, |
| "grad_norm": 0.46750983595848083, |
| "learning_rate": 0.0009967551747861387, |
| "loss": 0.3926387429237366, |
| "step": 1280 |
| }, |
| { |
| "epoch": 0.0709436053761951, |
| "grad_norm": 0.5673767328262329, |
| "learning_rate": 0.000994232528651847, |
| "loss": 0.3741423189640045, |
| "step": 1536 |
| }, |
| { |
| "epoch": 0.08276753960556095, |
| "grad_norm": 0.5806387066841125, |
| "learning_rate": 0.0009909945800260092, |
| "loss": 0.35307449102401733, |
| "step": 1792 |
| }, |
| { |
| "epoch": 0.09459147383492679, |
| "grad_norm": 0.6838217973709106, |
| "learning_rate": 0.0009870460151900522, |
| "loss": 0.3399304151535034, |
| "step": 2048 |
| }, |
| { |
| "epoch": 0.09459147383492679, |
| "eval_bleu": 0.6478349738418474, |
| "eval_ce_loss": 1.3434572242982856, |
| "eval_cos_loss": 0.09738318450306649, |
| "eval_loss": 0.3373560005943525, |
| "eval_mse_loss": 0.18800275102597938, |
| "eval_rec_loss": 0.005269206022433341, |
| "flow/cos_sim": 0.9026168323542973, |
| "flow/improvement_ratio": 0.9768092384621433, |
| "flow/mag_ratio_mean": 0.9078535956062682, |
| "flow/mag_ratio_std": 0.12002966954419601, |
| "step": 2048 |
| }, |
| { |
| "epoch": 0.09459147383492679, |
| "eval_bleu": 0.6478349738418474, |
| "eval_ce_loss": 1.3434572242982856, |
| "eval_cos_loss": 0.09738318450306649, |
| "eval_loss": 0.3373560005943525, |
| "eval_mse_loss": 0.18800275102597938, |
| "eval_rec_loss": 0.005269206022433341, |
| "eval_runtime": 143.5996, |
| "eval_samples_per_second": 194.938, |
| "eval_steps_per_second": 3.05, |
| "flow/cos_sim": 0.9026168323542973, |
| "flow/improvement_ratio": 0.9768092384621433, |
| "flow/mag_ratio_mean": 0.9078535956062682, |
| "flow/mag_ratio_std": 0.12002966954419601, |
| "step": 2048 |
| }, |
| { |
| "epoch": 0.10641540806429264, |
| "grad_norm": 0.6462650299072266, |
| "learning_rate": 0.0009823925488998885, |
| "loss": 0.3380688428878784, |
| "step": 2304 |
| }, |
| { |
| "epoch": 0.1182393422936585, |
| "grad_norm": 0.6545562744140625, |
| "learning_rate": 0.0009770409161149525, |
| "loss": 0.3214532732963562, |
| "step": 2560 |
| }, |
| { |
| "epoch": 0.13006327652302435, |
| "grad_norm": 0.6667947173118591, |
| "learning_rate": 0.0009709988622506973, |
| "loss": 0.31182003021240234, |
| "step": 2816 |
| }, |
| { |
| "epoch": 0.1418872107523902, |
| "grad_norm": 0.600439727306366, |
| "learning_rate": 0.000964275131968659, |
| "loss": 0.3082171380519867, |
| "step": 3072 |
| }, |
| { |
| "epoch": 0.1418872107523902, |
| "eval_bleu": 0.6794402302458825, |
| "eval_ce_loss": 1.18399681810919, |
| "eval_cos_loss": 0.08983433964453875, |
| "eval_loss": 0.3044066195414491, |
| "eval_mse_loss": 0.17457320809908652, |
| "eval_rec_loss": 0.0024502936500277726, |
| "flow/cos_sim": 0.9101656942062726, |
| "flow/improvement_ratio": 0.9734727941691603, |
| "flow/mag_ratio_mean": 0.9100713347463303, |
| "flow/mag_ratio_std": 0.09228039255629392, |
| "step": 3072 |
| }, |
| { |
| "epoch": 0.1418872107523902, |
| "eval_bleu": 0.6794402302458825, |
| "eval_ce_loss": 1.18399681810919, |
| "eval_cos_loss": 0.08983433964453875, |
| "eval_loss": 0.3044066195414491, |
| "eval_mse_loss": 0.17457320809908652, |
| "eval_rec_loss": 0.0024502936500277726, |
| "eval_runtime": 142.616, |
| "eval_samples_per_second": 196.282, |
| "eval_steps_per_second": 3.071, |
| "flow/cos_sim": 0.9101656942062726, |
| "flow/improvement_ratio": 0.9734727941691603, |
| "flow/mag_ratio_mean": 0.9100713347463303, |
| "flow/mag_ratio_std": 0.09228039255629392, |
| "step": 3072 |
| }, |
| { |
| "epoch": 0.15371114498175603, |
| "grad_norm": 0.5188813209533691, |
| "learning_rate": 0.0009568794565203123, |
| "loss": 0.3006798326969147, |
| "step": 3328 |
| }, |
| { |
| "epoch": 0.1655350792111219, |
| "grad_norm": 0.3702296316623688, |
| "learning_rate": 0.0009488225396630347, |
| "loss": 0.29647037386894226, |
| "step": 3584 |
| }, |
| { |
| "epoch": 0.17735901344048774, |
| "grad_norm": 0.49310481548309326, |
| "learning_rate": 0.0009401160421685646, |
| "loss": 0.3130263388156891, |
| "step": 3840 |
| }, |
| { |
| "epoch": 0.18918294766985358, |
| "grad_norm": 0.5644649267196655, |
| "learning_rate": 0.0009307725649463714, |
| "loss": 0.2926834523677826, |
| "step": 4096 |
| }, |
| { |
| "epoch": 0.18918294766985358, |
| "eval_bleu": 0.6839715035857402, |
| "eval_ce_loss": 1.1360810732460458, |
| "eval_cos_loss": 0.08503438343536364, |
| "eval_loss": 0.2889118402788084, |
| "eval_mse_loss": 0.16517186124030858, |
| "eval_rec_loss": 0.0016284317239192974, |
| "flow/cos_sim": 0.9149656362457362, |
| "flow/improvement_ratio": 0.9750959768415042, |
| "flow/mag_ratio_mean": 0.9128544130553938, |
| "flow/mag_ratio_std": 0.09354445787325297, |
| "step": 4096 |
| }, |
| { |
| "epoch": 0.18918294766985358, |
| "eval_bleu": 0.6839715035857402, |
| "eval_ce_loss": 1.1360810732460458, |
| "eval_cos_loss": 0.08503438343536364, |
| "eval_loss": 0.2889118402788084, |
| "eval_mse_loss": 0.16517186124030858, |
| "eval_rec_loss": 0.0016284317239192974, |
| "eval_runtime": 143.3522, |
| "eval_samples_per_second": 195.274, |
| "eval_steps_per_second": 3.055, |
| "flow/cos_sim": 0.9149656362457362, |
| "flow/improvement_ratio": 0.9750959768415042, |
| "flow/mag_ratio_mean": 0.9128544130553938, |
| "flow/mag_ratio_std": 0.09354445787325297, |
| "step": 4096 |
| }, |
| { |
| "epoch": 0.20100688189921945, |
| "grad_norm": 0.3648587763309479, |
| "learning_rate": 0.0009208056308063659, |
| "loss": 0.28915518522262573, |
| "step": 4352 |
| }, |
| { |
| "epoch": 0.2128308161285853, |
| "grad_norm": 0.5004331469535828, |
| "learning_rate": 0.0009102296648873445, |
| "loss": 0.28424444794654846, |
| "step": 4608 |
| }, |
| { |
| "epoch": 0.22465475035795113, |
| "grad_norm": 0.27222171425819397, |
| "learning_rate": 0.0008990599737794927, |
| "loss": 0.2838270366191864, |
| "step": 4864 |
| }, |
| { |
| "epoch": 0.236478684587317, |
| "grad_norm": 0.39166298508644104, |
| "learning_rate": 0.0008873127233711644, |
| "loss": 0.2802566885948181, |
| "step": 5120 |
| }, |
| { |
| "epoch": 0.236478684587317, |
| "eval_bleu": 0.6966429437264656, |
| "eval_ce_loss": 1.0860335096784923, |
| "eval_cos_loss": 0.08069509963534738, |
| "eval_loss": 0.27587767567944854, |
| "eval_mse_loss": 0.15802378180228413, |
| "eval_rec_loss": 0.0011810324893420196, |
| "flow/cos_sim": 0.9193049165756191, |
| "flow/improvement_ratio": 0.9735453040360316, |
| "flow/mag_ratio_mean": 0.9165048223652251, |
| "flow/mag_ratio_std": 0.08227625193252955, |
| "step": 5120 |
| }, |
| { |
| "epoch": 0.236478684587317, |
| "eval_bleu": 0.6966429437264656, |
| "eval_ce_loss": 1.0860335096784923, |
| "eval_cos_loss": 0.08069509963534738, |
| "eval_loss": 0.27587767567944854, |
| "eval_mse_loss": 0.15802378180228413, |
| "eval_rec_loss": 0.0011810324893420196, |
| "eval_runtime": 144.8851, |
| "eval_samples_per_second": 193.208, |
| "eval_steps_per_second": 3.023, |
| "flow/cos_sim": 0.9193049165756191, |
| "flow/improvement_ratio": 0.9735453040360316, |
| "flow/mag_ratio_mean": 0.9165048223652251, |
| "flow/mag_ratio_std": 0.08227625193252955, |
| "step": 5120 |
| }, |
| { |
| "epoch": 0.24830261881668284, |
| "grad_norm": 0.4552461802959442, |
| "learning_rate": 0.0008750049154520011, |
| "loss": 0.27747780084609985, |
| "step": 5376 |
| }, |
| { |
| "epoch": 0.2601265530460487, |
| "grad_norm": 0.3430848717689514, |
| "learning_rate": 0.0008621543631062487, |
| "loss": 0.2891384959220886, |
| "step": 5632 |
| }, |
| { |
| "epoch": 0.27195048727541454, |
| "grad_norm": 0.4337022304534912, |
| "learning_rate": 0.0008487796649318904, |
| "loss": 0.274080365896225, |
| "step": 5888 |
| }, |
| { |
| "epoch": 0.2837744215047804, |
| "grad_norm": 0.40342026948928833, |
| "learning_rate": 0.0008349001781229053, |
| "loss": 0.2737179100513458, |
| "step": 6144 |
| }, |
| { |
| "epoch": 0.2837744215047804, |
| "eval_bleu": 0.7005510372485733, |
| "eval_ce_loss": 1.067978764981984, |
| "eval_cos_loss": 0.0791517910994079, |
| "eval_loss": 0.2710464134744313, |
| "eval_mse_loss": 0.15530247285485813, |
| "eval_rec_loss": 0.00103088276002016, |
| "flow/cos_sim": 0.9208482295682986, |
| "flow/improvement_ratio": 0.9731569932476026, |
| "flow/mag_ratio_mean": 0.9284612283314744, |
| "flow/mag_ratio_std": 0.07813496222574962, |
| "step": 6144 |
| }, |
| { |
| "epoch": 0.2837744215047804, |
| "eval_bleu": 0.7005510372485733, |
| "eval_ce_loss": 1.067978764981984, |
| "eval_cos_loss": 0.0791517910994079, |
| "eval_loss": 0.2710464134744313, |
| "eval_mse_loss": 0.15530247285485813, |
| "eval_rec_loss": 0.00103088276002016, |
| "eval_runtime": 142.9519, |
| "eval_samples_per_second": 195.821, |
| "eval_steps_per_second": 3.064, |
| "flow/cos_sim": 0.9208482295682986, |
| "flow/improvement_ratio": 0.9731569932476026, |
| "flow/mag_ratio_mean": 0.9284612283314744, |
| "flow/mag_ratio_std": 0.07813496222574962, |
| "step": 6144 |
| }, |
| { |
| "epoch": 0.2955983557341462, |
| "grad_norm": 0.36669906973838806, |
| "learning_rate": 0.0008205359904536107, |
| "loss": 0.2686825394630432, |
| "step": 6400 |
| }, |
| { |
| "epoch": 0.30742228996351206, |
| "grad_norm": 0.2844366729259491, |
| "learning_rate": 0.0008057078912056363, |
| "loss": 0.26841098070144653, |
| "step": 6656 |
| }, |
| { |
| "epoch": 0.3192462241928779, |
| "grad_norm": 0.341339647769928, |
| "learning_rate": 0.0007904373410796086, |
| "loss": 0.2661970555782318, |
| "step": 6912 |
| }, |
| { |
| "epoch": 0.3310701584222438, |
| "grad_norm": 0.35041218996047974, |
| "learning_rate": 0.0007747464411350876, |
| "loss": 0.26556745171546936, |
| "step": 7168 |
| }, |
| { |
| "epoch": 0.3310701584222438, |
| "eval_bleu": 0.710653856755971, |
| "eval_ce_loss": 1.0037036097213015, |
| "eval_cos_loss": 0.0774884378733156, |
| "eval_loss": 0.2617515635817018, |
| "eval_mse_loss": 0.15282062886784611, |
| "eval_rec_loss": 0.0008117282769037305, |
| "flow/cos_sim": 0.9225115811443765, |
| "flow/improvement_ratio": 0.9752856880834658, |
| "flow/mag_ratio_mean": 0.9207794453999768, |
| "flow/mag_ratio_std": 0.08132727078447059, |
| "step": 7168 |
| }, |
| { |
| "epoch": 0.3310701584222438, |
| "eval_bleu": 0.710653856755971, |
| "eval_ce_loss": 1.0037036097213015, |
| "eval_cos_loss": 0.0774884378733156, |
| "eval_loss": 0.2617515635817018, |
| "eval_mse_loss": 0.15282062886784611, |
| "eval_rec_loss": 0.0008117282769037305, |
| "eval_runtime": 142.6116, |
| "eval_samples_per_second": 196.288, |
| "eval_steps_per_second": 3.071, |
| "flow/cos_sim": 0.9225115811443765, |
| "flow/improvement_ratio": 0.9752856880834658, |
| "flow/mag_ratio_mean": 0.9207794453999768, |
| "flow/mag_ratio_std": 0.08132727078447059, |
| "step": 7168 |
| }, |
| { |
| "epoch": 0.34289409265160964, |
| "grad_norm": 0.328535258769989, |
| "learning_rate": 0.000758657900803716, |
| "loss": 0.27869582176208496, |
| "step": 7424 |
| }, |
| { |
| "epoch": 0.3547180268809755, |
| "grad_norm": 0.27767524123191833, |
| "learning_rate": 0.000742195005021869, |
| "loss": 0.26837044954299927, |
| "step": 7680 |
| }, |
| { |
| "epoch": 0.3665419611103413, |
| "grad_norm": 0.40343427658081055, |
| "learning_rate": 0.0007253815805303786, |
| "loss": 0.26287224888801575, |
| "step": 7936 |
| }, |
| { |
| "epoch": 0.37836589533970716, |
| "grad_norm": 0.3494727611541748, |
| "learning_rate": 0.0007082419613901028, |
| "loss": 0.2608710527420044, |
| "step": 8192 |
| }, |
| { |
| "epoch": 0.37836589533970716, |
| "eval_bleu": 0.7258561394806508, |
| "eval_ce_loss": 0.954990051584701, |
| "eval_cos_loss": 0.07798272708099183, |
| "eval_loss": 0.2584831827239359, |
| "eval_mse_loss": 0.15443320324023566, |
| "eval_rec_loss": 0.0007526992644982298, |
| "flow/cos_sim": 0.9220172950666244, |
| "flow/improvement_ratio": 0.9747015749482804, |
| "flow/mag_ratio_mean": 0.9233354187175019, |
| "flow/mag_ratio_std": 0.07788482179108276, |
| "step": 8192 |
| }, |
| { |
| "epoch": 0.37836589533970716, |
| "eval_bleu": 0.7258561394806508, |
| "eval_ce_loss": 0.954990051584701, |
| "eval_cos_loss": 0.07798272708099183, |
| "eval_loss": 0.2584831827239359, |
| "eval_mse_loss": 0.15443320324023566, |
| "eval_rec_loss": 0.0007526992644982298, |
| "eval_runtime": 144.1923, |
| "eval_samples_per_second": 194.137, |
| "eval_steps_per_second": 3.038, |
| "flow/cos_sim": 0.9220172950666244, |
| "flow/improvement_ratio": 0.9747015749482804, |
| "flow/mag_ratio_mean": 0.9233354187175019, |
| "flow/mag_ratio_std": 0.07788482179108276, |
| "step": 8192 |
| }, |
| { |
| "epoch": 0.390189829569073, |
| "grad_norm": 0.4279082417488098, |
| "learning_rate": 0.0006908009537632514, |
| "loss": 0.2599097788333893, |
| "step": 8448 |
| }, |
| { |
| "epoch": 0.4020137637984389, |
| "grad_norm": 0.43606042861938477, |
| "learning_rate": 0.0006730838000114403, |
| "loss": 0.26111704111099243, |
| "step": 8704 |
| }, |
| { |
| "epoch": 0.41383769802780473, |
| "grad_norm": 0.21560809016227722, |
| "learning_rate": 0.0006551161421624341, |
| "loss": 0.268413245677948, |
| "step": 8960 |
| }, |
| { |
| "epoch": 0.4256616322571706, |
| "grad_norm": 0.5756855607032776, |
| "learning_rate": 0.0006369239847984517, |
| "loss": 0.2562841475009918, |
| "step": 9216 |
| }, |
| { |
| "epoch": 0.4256616322571706, |
| "eval_bleu": 0.7095258647762687, |
| "eval_ce_loss": 1.0158189947474492, |
| "eval_cos_loss": 0.0742807937743457, |
| "eval_loss": 0.256784121705789, |
| "eval_mse_loss": 0.14713338680871546, |
| "eval_rec_loss": 0.0006407552403591884, |
| "flow/cos_sim": 0.9257192251072627, |
| "flow/improvement_ratio": 0.9743275895510635, |
| "flow/mag_ratio_mean": 0.9202466026016566, |
| "flow/mag_ratio_std": 0.0764387545588354, |
| "step": 9216 |
| }, |
| { |
| "epoch": 0.4256616322571706, |
| "eval_bleu": 0.7095258647762687, |
| "eval_ce_loss": 1.0158189947474492, |
| "eval_cos_loss": 0.0742807937743457, |
| "eval_loss": 0.256784121705789, |
| "eval_mse_loss": 0.14713338680871546, |
| "eval_rec_loss": 0.0006407552403591884, |
| "eval_runtime": 145.076, |
| "eval_samples_per_second": 192.954, |
| "eval_steps_per_second": 3.019, |
| "flow/cos_sim": 0.9257192251072627, |
| "flow/improvement_ratio": 0.9743275895510635, |
| "flow/mag_ratio_mean": 0.9202466026016566, |
| "flow/mag_ratio_std": 0.0764387545588354, |
| "step": 9216 |
| }, |
| { |
| "epoch": 0.4374855664865364, |
| "grad_norm": 0.376579225063324, |
| "learning_rate": 0.0006185336574197479, |
| "loss": 0.25477761030197144, |
| "step": 9472 |
| }, |
| { |
| "epoch": 0.44930950071590225, |
| "grad_norm": 0.4893060624599457, |
| "learning_rate": 0.0005999717763379407, |
| "loss": 0.2554967701435089, |
| "step": 9728 |
| }, |
| { |
| "epoch": 0.4611334349452681, |
| "grad_norm": 0.5277368426322937, |
| "learning_rate": 0.0005812652061542363, |
| "loss": 0.2537921667098999, |
| "step": 9984 |
| }, |
| { |
| "epoch": 0.472957369174634, |
| "grad_norm": 0.3493780791759491, |
| "learning_rate": 0.0005624410208783071, |
| "loss": 0.2518165707588196, |
| "step": 10240 |
| }, |
| { |
| "epoch": 0.472957369174634, |
| "eval_bleu": 0.7175600139987302, |
| "eval_ce_loss": 0.9684542848231041, |
| "eval_cos_loss": 0.07330035061902923, |
| "eval_loss": 0.25065779049782994, |
| "eval_mse_loss": 0.1458930650292194, |
| "eval_rec_loss": 0.000589260474122746, |
| "flow/cos_sim": 0.9266996725236989, |
| "flow/improvement_ratio": 0.9765301047394809, |
| "flow/mag_ratio_mean": 0.9194769399351181, |
| "flow/mag_ratio_std": 0.07280441308946914, |
| "step": 10240 |
| }, |
| { |
| "epoch": 0.472957369174634, |
| "eval_bleu": 0.7175600139987302, |
| "eval_ce_loss": 0.9684542848231041, |
| "eval_cos_loss": 0.07330035061902923, |
| "eval_loss": 0.25065779049782994, |
| "eval_mse_loss": 0.1458930650292194, |
| "eval_rec_loss": 0.000589260474122746, |
| "eval_runtime": 144.907, |
| "eval_samples_per_second": 193.179, |
| "eval_steps_per_second": 3.023, |
| "flow/cos_sim": 0.9266996725236989, |
| "flow/improvement_ratio": 0.9765301047394809, |
| "flow/mag_ratio_mean": 0.9194769399351181, |
| "flow/mag_ratio_std": 0.07280441308946914, |
| "step": 10240 |
| }, |
| { |
| "epoch": 0.48478130340399983, |
| "grad_norm": 0.5341387391090393, |
| "learning_rate": 0.0005435264647440881, |
| "loss": 0.2537766695022583, |
| "step": 10496 |
| }, |
| { |
| "epoch": 0.49660523763336567, |
| "grad_norm": 0.37015679478645325, |
| "learning_rate": 0.000524548912779213, |
| "loss": 0.2499093860387802, |
| "step": 10752 |
| }, |
| { |
| "epoch": 0.5084291718627315, |
| "grad_norm": 0.6591759920120239, |
| "learning_rate": 0.0005055358311851499, |
| "loss": 0.25186970829963684, |
| "step": 11008 |
| }, |
| { |
| "epoch": 0.5202531060920974, |
| "grad_norm": 0.31277748942375183, |
| "learning_rate": 0.0004865147375853812, |
| "loss": 0.25062766671180725, |
| "step": 11264 |
| }, |
| { |
| "epoch": 0.5202531060920974, |
| "eval_bleu": 0.7260803484693739, |
| "eval_ce_loss": 0.9400546782092961, |
| "eval_cos_loss": 0.07318596699228298, |
| "eval_loss": 0.2483225218169221, |
| "eval_mse_loss": 0.14649179712210073, |
| "eval_rec_loss": 0.0005066586955975142, |
| "flow/cos_sim": 0.9268140568308634, |
| "flow/improvement_ratio": 0.9762515691045213, |
| "flow/mag_ratio_mean": 0.9157658136326429, |
| "flow/mag_ratio_std": 0.07639177793373256, |
| "step": 11264 |
| }, |
| { |
| "epoch": 0.5202531060920974, |
| "eval_bleu": 0.7260803484693739, |
| "eval_ce_loss": 0.9400546782092961, |
| "eval_cos_loss": 0.07318596699228298, |
| "eval_loss": 0.2483225218169221, |
| "eval_mse_loss": 0.14649179712210073, |
| "eval_rec_loss": 0.0005066586955975142, |
| "eval_runtime": 143.4087, |
| "eval_samples_per_second": 195.197, |
| "eval_steps_per_second": 3.054, |
| "flow/cos_sim": 0.9268140568308634, |
| "flow/improvement_ratio": 0.9762515691045213, |
| "flow/mag_ratio_mean": 0.9157658136326429, |
| "flow/mag_ratio_std": 0.07639177793373256, |
| "step": 11264 |
| }, |
| { |
| "epoch": 0.5320770403214632, |
| "grad_norm": 0.7305059432983398, |
| "learning_rate": 0.0004675131611991607, |
| "loss": 0.2499980330467224, |
| "step": 11520 |
| }, |
| { |
| "epoch": 0.5439009745508291, |
| "grad_norm": 0.2771352231502533, |
| "learning_rate": 0.0004485586029984899, |
| "loss": 0.24800576269626617, |
| "step": 11776 |
| }, |
| { |
| "epoch": 0.5557249087801949, |
| "grad_norm": 0.2686781883239746, |
| "learning_rate": 0.00042967849590597266, |
| "loss": 0.24677802622318268, |
| "step": 12032 |
| }, |
| { |
| "epoch": 0.5675488430095608, |
| "grad_norm": 0.37717506289482117, |
| "learning_rate": 0.0004109001650911621, |
| "loss": 0.24666211009025574, |
| "step": 12288 |
| }, |
| { |
| "epoch": 0.5675488430095608, |
| "eval_bleu": 0.724567852062454, |
| "eval_ce_loss": 0.9509265820185343, |
| "eval_cos_loss": 0.07338333341663014, |
| "eval_loss": 0.24947158049911125, |
| "eval_mse_loss": 0.1465663725518745, |
| "eval_rec_loss": 0.00047421499268288965, |
| "flow/cos_sim": 0.9266166863920482, |
| "flow/improvement_ratio": 0.9748387396607769, |
| "flow/mag_ratio_mean": 0.9328112143631939, |
| "flow/mag_ratio_std": 0.07196474395113993, |
| "step": 12288 |
| }, |
| { |
| "epoch": 0.5675488430095608, |
| "eval_bleu": 0.724567852062454, |
| "eval_ce_loss": 0.9509265820185343, |
| "eval_cos_loss": 0.07338333341663014, |
| "eval_loss": 0.24947158049911125, |
| "eval_mse_loss": 0.1465663725518745, |
| "eval_rec_loss": 0.00047421499268288965, |
| "eval_runtime": 144.4458, |
| "eval_samples_per_second": 193.796, |
| "eval_steps_per_second": 3.032, |
| "flow/cos_sim": 0.9266166863920482, |
| "flow/improvement_ratio": 0.9748387396607769, |
| "flow/mag_ratio_mean": 0.9328112143631939, |
| "flow/mag_ratio_std": 0.07196474395113993, |
| "step": 12288 |
| }, |
| { |
| "epoch": 0.5793727772389267, |
| "grad_norm": 0.7573108673095703, |
| "learning_rate": 0.0003922507884228551, |
| "loss": 0.24700023233890533, |
| "step": 12544 |
| }, |
| { |
| "epoch": 0.5911967114682924, |
| "grad_norm": 0.2122378647327423, |
| "learning_rate": 0.00037375735713457723, |
| "loss": 0.2433316856622696, |
| "step": 12800 |
| }, |
| { |
| "epoch": 0.6030206456976583, |
| "grad_norm": 0.22209292650222778, |
| "learning_rate": 0.00035544663676018276, |
| "loss": 0.246763676404953, |
| "step": 13056 |
| }, |
| { |
| "epoch": 0.6148445799270241, |
| "grad_norm": 0.309989333152771, |
| "learning_rate": 0.00033734512839611255, |
| "loss": 0.24262374639511108, |
| "step": 13312 |
| }, |
| { |
| "epoch": 0.6148445799270241, |
| "eval_bleu": 0.7300156788258321, |
| "eval_ce_loss": 0.933146977737614, |
| "eval_cos_loss": 0.07192521648626077, |
| "eval_loss": 0.24540512995224564, |
| "eval_mse_loss": 0.14446957482526837, |
| "eval_rec_loss": 0.00042833433703537476, |
| "flow/cos_sim": 0.9280747922315989, |
| "flow/improvement_ratio": 0.9743193072271129, |
| "flow/mag_ratio_mean": 0.9250430587219866, |
| "flow/mag_ratio_std": 0.07064938510355612, |
| "step": 13312 |
| }, |
| { |
| "epoch": 0.6148445799270241, |
| "eval_bleu": 0.7300156788258321, |
| "eval_ce_loss": 0.933146977737614, |
| "eval_cos_loss": 0.07192521648626077, |
| "eval_loss": 0.24540512995224564, |
| "eval_mse_loss": 0.14446957482526837, |
| "eval_rec_loss": 0.00042833433703537476, |
| "eval_runtime": 144.2347, |
| "eval_samples_per_second": 194.079, |
| "eval_steps_per_second": 3.037, |
| "flow/cos_sim": 0.9280747922315989, |
| "flow/improvement_ratio": 0.9743193072271129, |
| "flow/mag_ratio_mean": 0.9250430587219866, |
| "flow/mag_ratio_std": 0.07064938510355612, |
| "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 |
| } |
|
|
