{ "best_metric": 0.8033268101761253, "best_model_checkpoint": "tiny-bert-sst2-distilled/run-42/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.1318306922912598, "learning_rate": 0.0001222965835274321, "loss": 0.5547, "step": 192 }, { "epoch": 1.0, "eval_accuracy": 0.8033268101761253, "eval_f1": 0.8076555023923445, "eval_loss": 0.4663587808609009, "eval_mcc": 0.607269059552277, "eval_precision": 0.7902621722846442, "eval_recall": 0.8258317025440313, "eval_runtime": 66.9367, "eval_samples_per_second": 15.268, "eval_steps_per_second": 0.478, "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.946682979258311, "learning_rate": 0.0002445931670548642, "num_train_epochs": 2, "per_device_train_batch_size": 16, "temperature": 14 } }