{ "best_metric": 0.1501207798719406, "best_model_checkpoint": "/content/drive/MyDrive/RoBERTa/checkpoint-14545", "epoch": 5.0, "eval_steps": 100, "global_step": 14545, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "grad_norm": 4.497077941894531, "learning_rate": 4.5e-06, "loss": 0.2489, "step": 2909 }, { "epoch": 1.0, "eval_accuracy": 0.7945676465532061, "eval_f1": 0.7961558430656073, "eval_loss": 0.16617761552333832, "eval_precision": 0.801102553760894, "eval_recall": 0.7945676465532061, "eval_runtime": 46.2471, "eval_samples_per_second": 125.781, "eval_steps_per_second": 7.871, "step": 2909 }, { "epoch": 2.0, "grad_norm": 2.7504544258117676, "learning_rate": 4.000000000000001e-06, "loss": 0.165, "step": 5818 }, { "epoch": 2.0, "eval_accuracy": 0.8066013408973698, "eval_f1": 0.806334324115106, "eval_loss": 0.15410974621772766, "eval_precision": 0.807004318007734, "eval_recall": 0.8066013408973698, "eval_runtime": 46.1274, "eval_samples_per_second": 126.107, "eval_steps_per_second": 7.891, "step": 5818 }, { "epoch": 3.0, "grad_norm": 6.142695426940918, "learning_rate": 3.5e-06, "loss": 0.1479, "step": 8727 }, { "epoch": 3.0, "eval_accuracy": 0.8038507821901324, "eval_f1": 0.8008391832785562, "eval_loss": 0.15494388341903687, "eval_precision": 0.8086474488952677, "eval_recall": 0.8038507821901324, "eval_runtime": 46.076, "eval_samples_per_second": 126.248, "eval_steps_per_second": 7.9, "step": 8727 }, { "epoch": 4.0, "grad_norm": 16.692214965820312, "learning_rate": 3e-06, "loss": 0.1375, "step": 11636 }, { "epoch": 4.0, "eval_accuracy": 0.8102114492006188, "eval_f1": 0.8090868438898687, "eval_loss": 0.15274302661418915, "eval_precision": 0.8111386977694952, "eval_recall": 0.8102114492006188, "eval_runtime": 46.1151, "eval_samples_per_second": 126.141, "eval_steps_per_second": 7.893, "step": 11636 }, { "epoch": 5.0, "grad_norm": 24.896465301513672, "learning_rate": 2.5e-06, "loss": 0.1289, "step": 14545 }, { "epoch": 5.0, "eval_accuracy": 0.8096957194430119, "eval_f1": 0.8100808721825671, "eval_loss": 0.1501207798719406, "eval_precision": 0.8116662168004147, "eval_recall": 0.8096957194430119, "eval_runtime": 46.0976, "eval_samples_per_second": 126.189, "eval_steps_per_second": 7.896, "step": 14545 } ], "logging_steps": 100, "max_steps": 29090, "num_input_tokens_seen": 0, "num_train_epochs": 10, "save_steps": 100, "stateful_callbacks": { "EarlyStoppingCallback": { "args": { "early_stopping_patience": 2, "early_stopping_threshold": 0.0 }, "attributes": { "early_stopping_patience_counter": 0 } }, "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 6.122024785926144e+16, "train_batch_size": 16, "trial_name": null, "trial_params": null }