{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.2934030165497639, "eval_steps": 200, "global_step": 3200, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.02, "eval_bertscore": 0.7401605248451233, "eval_loss": 1.9530484676361084, "eval_rouge1": 0.6562857460474375, "eval_rouge2": 0.3640670727106235, "eval_rougeL": 0.5655212336424695, "eval_rougeLsum": 0.6414840198810386, "eval_runtime": 21.7196, "eval_samples_per_second": 1.381, "eval_steps_per_second": 0.691, "step": 200 }, { "epoch": 0.02, "grad_norm": 0.25105270743370056, "learning_rate": 0.00019771674842969145, "loss": 1.7353, "step": 250 }, { "epoch": 0.04, "eval_bertscore": 0.7432050108909607, "eval_loss": 1.9583823680877686, "eval_rouge1": 0.6554226269617707, "eval_rouge2": 0.36661086995296877, "eval_rougeL": 0.5637448790342183, "eval_rougeLsum": 0.6419796784912521, "eval_runtime": 21.9623, "eval_samples_per_second": 1.366, "eval_steps_per_second": 0.683, "step": 400 }, { "epoch": 0.05, "grad_norm": 0.26550447940826416, "learning_rate": 0.00019542432717436156, "loss": 1.7786, "step": 500 }, { "epoch": 0.06, "eval_bertscore": 0.7469045519828796, "eval_loss": 1.9245686531066895, "eval_rouge1": 0.6662431635890791, "eval_rouge2": 0.3735263724826765, "eval_rougeL": 0.5755071616151013, "eval_rougeLsum": 0.6538383087686117, "eval_runtime": 21.5302, "eval_samples_per_second": 1.393, "eval_steps_per_second": 0.697, "step": 600 }, { "epoch": 0.07, "grad_norm": 0.1538015753030777, "learning_rate": 0.0001931319059190317, "loss": 1.8851, "step": 750 }, { "epoch": 0.07, "eval_bertscore": 0.7442477941513062, "eval_loss": 1.9187489748001099, "eval_rouge1": 0.6606221897489035, "eval_rouge2": 0.368654563659435, "eval_rougeL": 0.5731546210408094, "eval_rougeLsum": 0.6470590823125606, "eval_runtime": 21.9831, "eval_samples_per_second": 1.365, "eval_steps_per_second": 0.682, "step": 800 }, { "epoch": 0.09, "grad_norm": 0.1681252270936966, "learning_rate": 0.0001908394846637018, "loss": 1.8919, "step": 1000 }, { "epoch": 0.09, "eval_bertscore": 0.7458053231239319, "eval_loss": 1.9159075021743774, "eval_rouge1": 0.6621259186456026, "eval_rouge2": 0.372024043683234, "eval_rougeL": 0.5743354509339939, "eval_rougeLsum": 0.6491550893780276, "eval_runtime": 21.7159, "eval_samples_per_second": 1.381, "eval_steps_per_second": 0.691, "step": 1000 }, { "epoch": 0.11, "eval_bertscore": 0.7468854784965515, "eval_loss": 1.9140182733535767, "eval_rouge1": 0.6626581781149132, "eval_rouge2": 0.37318557504782157, "eval_rougeL": 0.5759264203594217, "eval_rougeLsum": 0.6490702446275723, "eval_runtime": 21.6486, "eval_samples_per_second": 1.386, "eval_steps_per_second": 0.693, "step": 1200 }, { "epoch": 0.11, "grad_norm": 0.1552441120147705, "learning_rate": 0.00018854706340837193, "loss": 1.9052, "step": 1250 }, { "epoch": 0.13, "eval_bertscore": 0.7475314736366272, "eval_loss": 1.913794755935669, "eval_rouge1": 0.6648687174353192, "eval_rouge2": 0.3760379232448734, "eval_rougeL": 0.5784915488164926, "eval_rougeLsum": 0.6513864520108938, "eval_runtime": 21.664, "eval_samples_per_second": 1.385, "eval_steps_per_second": 0.692, "step": 1400 }, { "epoch": 0.14, "grad_norm": 0.14638397097587585, "learning_rate": 0.00018625464215304204, "loss": 1.8843, "step": 1500 }, { "epoch": 0.15, "eval_bertscore": 0.747238039970398, "eval_loss": 1.9117029905319214, "eval_rouge1": 0.6638085237198453, "eval_rouge2": 0.3742779818055127, "eval_rougeL": 0.5754209460423059, "eval_rougeLsum": 0.6506476155592722, "eval_runtime": 21.9308, "eval_samples_per_second": 1.368, "eval_steps_per_second": 0.684, "step": 1600 }, { "epoch": 0.16, "grad_norm": 0.15738993883132935, "learning_rate": 0.00018396222089771218, "loss": 1.8964, "step": 1750 }, { "epoch": 0.17, "eval_bertscore": 0.7473016381263733, "eval_loss": 1.9117563962936401, "eval_rouge1": 0.6620053151663765, "eval_rouge2": 0.37406692119411245, "eval_rougeL": 0.5758911607323577, "eval_rougeLsum": 0.6494070575604445, "eval_runtime": 21.6727, "eval_samples_per_second": 1.384, "eval_steps_per_second": 0.692, "step": 1800 }, { "epoch": 0.18, "grad_norm": 0.1588907092809677, "learning_rate": 0.00018166979964238228, "loss": 1.8827, "step": 2000 }, { "epoch": 0.18, "eval_bertscore": 0.7485987544059753, "eval_loss": 1.9126006364822388, "eval_rouge1": 0.6641836156334741, "eval_rouge2": 0.37320215574735827, "eval_rougeL": 0.5783015040447993, "eval_rougeLsum": 0.6522235940423647, "eval_runtime": 21.9759, "eval_samples_per_second": 1.365, "eval_steps_per_second": 0.683, "step": 2000 }, { "epoch": 0.2, "eval_bertscore": 0.7482583522796631, "eval_loss": 1.9075205326080322, "eval_rouge1": 0.6658219484766166, "eval_rouge2": 0.37723364952258465, "eval_rougeL": 0.5769040785174693, "eval_rougeLsum": 0.6511328888044219, "eval_runtime": 21.5892, "eval_samples_per_second": 1.39, "eval_steps_per_second": 0.695, "step": 2200 }, { "epoch": 0.21, "grad_norm": 0.15247465670108795, "learning_rate": 0.00017937737838705242, "loss": 1.8831, "step": 2250 }, { "epoch": 0.22, "eval_bertscore": 0.7460805177688599, "eval_loss": 1.9088668823242188, "eval_rouge1": 0.6627321043292516, "eval_rouge2": 0.3696581195003696, "eval_rougeL": 0.5740988544467178, "eval_rougeLsum": 0.6478729042661874, "eval_runtime": 21.9221, "eval_samples_per_second": 1.368, "eval_steps_per_second": 0.684, "step": 2400 }, { "epoch": 0.23, "grad_norm": 0.1587379276752472, "learning_rate": 0.00017708495713172253, "loss": 1.8829, "step": 2500 }, { "epoch": 0.24, "eval_bertscore": 0.7472203373908997, "eval_loss": 1.906219482421875, "eval_rouge1": 0.6637415370426804, "eval_rouge2": 0.37565276875837994, "eval_rougeL": 0.5773879369079004, "eval_rougeLsum": 0.6488719947518645, "eval_runtime": 21.8112, "eval_samples_per_second": 1.375, "eval_steps_per_second": 0.688, "step": 2600 }, { "epoch": 0.25, "grad_norm": 0.1558646410703659, "learning_rate": 0.00017479253587639266, "loss": 1.8978, "step": 2750 }, { "epoch": 0.26, "eval_bertscore": 0.7466126680374146, "eval_loss": 1.9045982360839844, "eval_rouge1": 0.6616225540296956, "eval_rouge2": 0.37370762164745913, "eval_rougeL": 0.5759418528371097, "eval_rougeLsum": 0.6479977636906877, "eval_runtime": 21.8772, "eval_samples_per_second": 1.371, "eval_steps_per_second": 0.686, "step": 2800 }, { "epoch": 0.28, "grad_norm": 0.14783035218715668, "learning_rate": 0.00017250011462106277, "loss": 1.8978, "step": 3000 }, { "epoch": 0.28, "eval_bertscore": 0.7485571503639221, "eval_loss": 1.9035439491271973, "eval_rouge1": 0.6664050030501707, "eval_rouge2": 0.379492440917784, "eval_rougeL": 0.5806973731221475, "eval_rougeLsum": 0.6524346156604702, "eval_runtime": 21.9217, "eval_samples_per_second": 1.369, "eval_steps_per_second": 0.684, "step": 3000 }, { "epoch": 0.29, "eval_bertscore": 0.7483461499214172, "eval_loss": 1.9022458791732788, "eval_rouge1": 0.6618989733136488, "eval_rouge2": 0.37377379177271053, "eval_rougeL": 0.5780989082173933, "eval_rougeLsum": 0.6490379362631586, "eval_runtime": 21.7847, "eval_samples_per_second": 1.377, "eval_steps_per_second": 0.689, "step": 3200 } ], "logging_steps": 250, "max_steps": 21812, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 800, "total_flos": 2.15717014142976e+17, "train_batch_size": 2, "trial_name": null, "trial_params": null }