| { | |
| "best_metric": 0.49937015771865845, | |
| "best_model_checkpoint": "sequence_classification/checkpoint-9794", | |
| "epoch": 2.0, | |
| "eval_steps": 500, | |
| "global_step": 9794, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.10210332856851133, | |
| "grad_norm": 65.69414520263672, | |
| "learning_rate": 9.489483357157444e-05, | |
| "loss": 0.7807, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.20420665713702266, | |
| "grad_norm": 63.07129669189453, | |
| "learning_rate": 8.978966714314887e-05, | |
| "loss": 0.6422, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.306309985705534, | |
| "grad_norm": 25.590089797973633, | |
| "learning_rate": 8.46845007147233e-05, | |
| "loss": 0.6025, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.4084133142740453, | |
| "grad_norm": 15.698567390441895, | |
| "learning_rate": 7.957933428629773e-05, | |
| "loss": 0.579, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.5105166428425567, | |
| "grad_norm": 21.827106475830078, | |
| "learning_rate": 7.447416785787217e-05, | |
| "loss": 0.5758, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.612619971411068, | |
| "grad_norm": 16.527406692504883, | |
| "learning_rate": 6.93690014294466e-05, | |
| "loss": 0.5614, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.7147232999795793, | |
| "grad_norm": 38.88615417480469, | |
| "learning_rate": 6.426383500102104e-05, | |
| "loss": 0.5579, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.8168266285480906, | |
| "grad_norm": 30.469078063964844, | |
| "learning_rate": 5.9158668572595464e-05, | |
| "loss": 0.5451, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.918929957116602, | |
| "grad_norm": 31.0489444732666, | |
| "learning_rate": 5.40535021441699e-05, | |
| "loss": 0.5408, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.7511115271212617, | |
| "eval_f1-score": 0.7514023934181002, | |
| "eval_loss": 0.5151644945144653, | |
| "eval_precision": 0.7523871934094739, | |
| "eval_recall": 0.7504201680672269, | |
| "eval_runtime": 6409.5702, | |
| "eval_samples_per_second": 3.334, | |
| "eval_steps_per_second": 0.417, | |
| "step": 4897 | |
| }, | |
| { | |
| "epoch": 1.0210332856851134, | |
| "grad_norm": 30.953594207763672, | |
| "learning_rate": 4.8948335715744334e-05, | |
| "loss": 0.5478, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 1.1231366142536248, | |
| "grad_norm": 22.760286331176758, | |
| "learning_rate": 4.384316928731877e-05, | |
| "loss": 0.54, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 1.225239942822136, | |
| "grad_norm": 63.70338821411133, | |
| "learning_rate": 3.8738002858893204e-05, | |
| "loss": 0.5308, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 1.3273432713906472, | |
| "grad_norm": 15.89914321899414, | |
| "learning_rate": 3.363283643046763e-05, | |
| "loss": 0.5258, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 1.4294465999591588, | |
| "grad_norm": 13.948808670043945, | |
| "learning_rate": 2.8527670002042067e-05, | |
| "loss": 0.5244, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 1.53154992852767, | |
| "grad_norm": 43.452964782714844, | |
| "learning_rate": 2.34225035736165e-05, | |
| "loss": 0.5135, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 1.6336532570961815, | |
| "grad_norm": 12.346921920776367, | |
| "learning_rate": 1.8317337145190937e-05, | |
| "loss": 0.5244, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 1.7357565856646926, | |
| "grad_norm": 11.3629150390625, | |
| "learning_rate": 1.3212170716765368e-05, | |
| "loss": 0.4938, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 1.8378599142332042, | |
| "grad_norm": 30.596006393432617, | |
| "learning_rate": 8.1070042883398e-06, | |
| "loss": 0.4958, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 1.9399632428017153, | |
| "grad_norm": 21.20801544189453, | |
| "learning_rate": 3.001837859914233e-06, | |
| "loss": 0.5157, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.7600037440913558, | |
| "eval_f1-score": 0.7677746580925641, | |
| "eval_loss": 0.49937015771865845, | |
| "eval_precision": 0.7454273654590221, | |
| "eval_recall": 0.7915032679738562, | |
| "eval_runtime": 8190.7633, | |
| "eval_samples_per_second": 2.609, | |
| "eval_steps_per_second": 0.326, | |
| "step": 9794 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 9794, | |
| "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": 4.589981822188339e+17, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |