| { | |
| "best_metric": 0.31418120861053467, | |
| "best_model_checkpoint": "gal_ensp_mBERT/checkpoint-750", | |
| "epoch": 4.0, | |
| "global_step": 1000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.8990431795878312, | |
| "eval_f1": 0.6289113806690875, | |
| "eval_loss": 0.3602694869041443, | |
| "eval_precision": 0.676416819012797, | |
| "eval_recall": 0.5876407738954663, | |
| "eval_runtime": 5.3964, | |
| "eval_samples_per_second": 184.751, | |
| "eval_steps_per_second": 11.674, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "learning_rate": 1.9600000000000002e-05, | |
| "loss": 0.4515, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.9109421000981355, | |
| "eval_f1": 0.6887108521485796, | |
| "eval_loss": 0.3213185966014862, | |
| "eval_precision": 0.6948853615520282, | |
| "eval_recall": 0.6826451053999423, | |
| "eval_runtime": 5.4855, | |
| "eval_samples_per_second": 181.753, | |
| "eval_steps_per_second": 11.485, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.9141928361138371, | |
| "eval_f1": 0.6999709976798144, | |
| "eval_loss": 0.31418120861053467, | |
| "eval_precision": 0.703029420332071, | |
| "eval_recall": 0.6969390701703725, | |
| "eval_runtime": 5.502, | |
| "eval_samples_per_second": 181.208, | |
| "eval_steps_per_second": 11.45, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "learning_rate": 1.9200000000000003e-05, | |
| "loss": 0.2256, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.9180569185475956, | |
| "eval_f1": 0.7156714809226128, | |
| "eval_loss": 0.3158390522003174, | |
| "eval_precision": 0.7123444428550995, | |
| "eval_recall": 0.7190297429974011, | |
| "eval_runtime": 5.4912, | |
| "eval_samples_per_second": 181.562, | |
| "eval_steps_per_second": 11.473, | |
| "step": 1000 | |
| } | |
| ], | |
| "max_steps": 25000, | |
| "num_train_epochs": 100, | |
| "total_flos": 623488737039000.0, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |