{ "best_metric": 0.7827788649706457, "best_model_checkpoint": "tiny-bert-sst2-distilled/run-29/checkpoint-192", "epoch": 1.0, "eval_steps": 500, "global_step": 192, "is_hyper_param_search": true, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "grad_norm": 3.586991786956787, "learning_rate": 3.692143561656546e-05, "loss": 0.6017, "step": 192 }, { "epoch": 1.0, "eval_accuracy": 0.7827788649706457, "eval_f1": 0.8069565217391305, "eval_loss": 0.4981837570667267, "eval_mcc": 0.5841818055857166, "eval_precision": 0.7261345852895149, "eval_recall": 0.9080234833659491, "eval_runtime": 66.722, "eval_samples_per_second": 15.317, "eval_steps_per_second": 0.48, "step": 192 } ], "logging_steps": 500, "max_steps": 384, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 500, "total_flos": 235695197280.0, "train_batch_size": 16, "trial_name": null, "trial_params": { "alpha": 0.9213628300882283, "learning_rate": 7.384287123313092e-05, "num_train_epochs": 2, "per_device_train_batch_size": 16, "temperature": 21 } }