{ "best_metric": 0.852670349907919, "best_model_checkpoint": "BERTForDetectingDepression-Twitter2015/run-8/checkpoint-736", "epoch": 5.0, "eval_steps": 500, "global_step": 920, "is_hyper_param_search": true, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "eval_accuracy": 0.8182934315531001, "eval_loss": 0.39697179198265076, "eval_runtime": 1.4648, "eval_samples_per_second": 1112.118, "eval_steps_per_second": 69.635, "step": 184 }, { "epoch": 2.0, "eval_accuracy": 0.8446899938612645, "eval_loss": 0.4469450116157532, "eval_runtime": 1.4697, "eval_samples_per_second": 1108.387, "eval_steps_per_second": 69.402, "step": 368 }, { "epoch": 2.717391304347826, "grad_norm": 8.623266220092773, "learning_rate": 3.7374530326680673e-05, "loss": 0.3328, "step": 500 }, { "epoch": 3.0, "eval_accuracy": 0.8496009821976673, "eval_loss": 0.7693418264389038, "eval_runtime": 1.4817, "eval_samples_per_second": 1099.419, "eval_steps_per_second": 68.84, "step": 552 }, { "epoch": 4.0, "eval_accuracy": 0.852670349907919, "eval_loss": 0.9155253767967224, "eval_runtime": 1.4898, "eval_samples_per_second": 1093.449, "eval_steps_per_second": 68.466, "step": 736 }, { "epoch": 5.0, "eval_accuracy": 0.8520564763658687, "eval_loss": 0.915794312953949, "eval_runtime": 1.5686, "eval_samples_per_second": 1038.509, "eval_steps_per_second": 65.026, "step": 920 } ], "logging_steps": 500, "max_steps": 920, "num_input_tokens_seen": 0, "num_train_epochs": 5, "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": 166182381453960.0, "train_batch_size": 8, "trial_name": null, "trial_params": { "learning_rate": 8.186801881082434e-05, "num_train_epochs": 5, "per_device_train_batch_size": 8, "seed": 29 } }