{ "best_metric": 1.3029065132141113, "best_model_checkpoint": "/content/pubmedbert-abstract-cord19/checkpoint-100000", "epoch": 3.0, "global_step": 102084, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.15, "learning_rate": 2.496e-05, "loss": 1.3774, "step": 5000 }, { "epoch": 0.15, "eval_loss": 1.3211560249328613, "eval_runtime": 169.3169, "eval_samples_per_second": 86.258, "eval_steps_per_second": 10.785, "step": 5000 }, { "epoch": 0.29, "learning_rate": 4.9945000000000004e-05, "loss": 1.3937, "step": 10000 }, { "epoch": 0.29, "eval_loss": 1.4059454202651978, "eval_runtime": 169.3069, "eval_samples_per_second": 86.263, "eval_steps_per_second": 10.785, "step": 10000 }, { "epoch": 0.44, "learning_rate": 4.7291603318709e-05, "loss": 1.6812, "step": 15000 }, { "epoch": 0.44, "eval_loss": 1.6174367666244507, "eval_runtime": 168.9751, "eval_samples_per_second": 86.433, "eval_steps_per_second": 10.806, "step": 15000 }, { "epoch": 0.59, "learning_rate": 4.457723382998132e-05, "loss": 1.4712, "step": 20000 }, { "epoch": 0.59, "eval_loss": 1.4382678270339966, "eval_runtime": 168.9508, "eval_samples_per_second": 86.445, "eval_steps_per_second": 10.808, "step": 20000 }, { "epoch": 0.73, "learning_rate": 4.186340732374788e-05, "loss": 1.4293, "step": 25000 }, { "epoch": 0.73, "eval_loss": 1.4356446266174316, "eval_runtime": 168.8828, "eval_samples_per_second": 86.48, "eval_steps_per_second": 10.812, "step": 25000 }, { "epoch": 0.88, "learning_rate": 3.914849485252596e-05, "loss": 1.4155, "step": 30000 }, { "epoch": 0.88, "eval_loss": 1.4283361434936523, "eval_runtime": 169.0831, "eval_samples_per_second": 86.378, "eval_steps_per_second": 10.799, "step": 30000 }, { "epoch": 1.03, "learning_rate": 3.6435754311281005e-05, "loss": 1.3963, "step": 35000 }, { "epoch": 1.03, "eval_loss": 1.4134665727615356, "eval_runtime": 169.088, "eval_samples_per_second": 86.375, "eval_steps_per_second": 10.799, "step": 35000 }, { "epoch": 1.18, "learning_rate": 3.372138482255332e-05, "loss": 1.3718, "step": 40000 }, { "epoch": 1.18, "eval_loss": 1.3948187828063965, "eval_runtime": 169.0271, "eval_samples_per_second": 86.406, "eval_steps_per_second": 10.803, "step": 40000 }, { "epoch": 1.32, "learning_rate": 3.1006472351331396e-05, "loss": 1.369, "step": 45000 }, { "epoch": 1.32, "eval_loss": 1.3961154222488403, "eval_runtime": 169.0308, "eval_samples_per_second": 86.404, "eval_steps_per_second": 10.803, "step": 45000 }, { "epoch": 1.47, "learning_rate": 2.8292645845097955e-05, "loss": 1.354, "step": 50000 }, { "epoch": 1.47, "eval_loss": 1.378829836845398, "eval_runtime": 169.1643, "eval_samples_per_second": 86.336, "eval_steps_per_second": 10.794, "step": 50000 }, { "epoch": 1.62, "learning_rate": 2.5579362321358764e-05, "loss": 1.3399, "step": 55000 }, { "epoch": 1.62, "eval_loss": 1.3865987062454224, "eval_runtime": 169.0688, "eval_samples_per_second": 86.385, "eval_steps_per_second": 10.8, "step": 55000 }, { "epoch": 1.76, "learning_rate": 2.2866078797619567e-05, "loss": 1.3289, "step": 60000 }, { "epoch": 1.76, "eval_loss": 1.362976312637329, "eval_runtime": 169.1136, "eval_samples_per_second": 86.362, "eval_steps_per_second": 10.797, "step": 60000 }, { "epoch": 1.91, "learning_rate": 2.0151709308891882e-05, "loss": 1.3155, "step": 65000 }, { "epoch": 1.91, "eval_loss": 1.3609150648117065, "eval_runtime": 169.0556, "eval_samples_per_second": 86.392, "eval_steps_per_second": 10.801, "step": 65000 }, { "epoch": 2.06, "learning_rate": 1.743788280265844e-05, "loss": 1.2976, "step": 70000 }, { "epoch": 2.06, "eval_loss": 1.3489034175872803, "eval_runtime": 169.1114, "eval_samples_per_second": 86.363, "eval_steps_per_second": 10.798, "step": 70000 }, { "epoch": 2.2, "learning_rate": 1.4724599278919249e-05, "loss": 1.2783, "step": 75000 }, { "epoch": 2.2, "eval_loss": 1.3333380222320557, "eval_runtime": 169.0576, "eval_samples_per_second": 86.391, "eval_steps_per_second": 10.801, "step": 75000 }, { "epoch": 2.35, "learning_rate": 1.2009686807697321e-05, "loss": 1.2696, "step": 80000 }, { "epoch": 2.35, "eval_loss": 1.3259785175323486, "eval_runtime": 168.9798, "eval_samples_per_second": 86.43, "eval_steps_per_second": 10.806, "step": 80000 }, { "epoch": 2.5, "learning_rate": 9.29586030146388e-06, "loss": 1.2607, "step": 85000 }, { "epoch": 2.5, "eval_loss": 1.3232313394546509, "eval_runtime": 168.7484, "eval_samples_per_second": 86.549, "eval_steps_per_second": 10.821, "step": 85000 }, { "epoch": 2.64, "learning_rate": 6.582033795230443e-06, "loss": 1.2547, "step": 90000 }, { "epoch": 2.64, "eval_loss": 1.3033902645111084, "eval_runtime": 168.7446, "eval_samples_per_second": 86.551, "eval_steps_per_second": 10.821, "step": 90000 }, { "epoch": 2.79, "learning_rate": 3.867664306502758e-06, "loss": 1.2495, "step": 95000 }, { "epoch": 2.79, "eval_loss": 1.3035175800323486, "eval_runtime": 168.7573, "eval_samples_per_second": 86.544, "eval_steps_per_second": 10.82, "step": 95000 }, { "epoch": 2.94, "learning_rate": 1.1538378002693194e-06, "loss": 1.2404, "step": 100000 }, { "epoch": 2.94, "eval_loss": 1.3029065132141113, "eval_runtime": 168.8016, "eval_samples_per_second": 86.522, "eval_steps_per_second": 10.817, "step": 100000 }, { "epoch": 3.0, "step": 102084, "total_flos": 1.04954332801725e+17, "train_loss": 1.35220570928797, "train_runtime": 32542.0864, "train_samples_per_second": 25.095, "train_steps_per_second": 3.137 } ], "max_steps": 102084, "num_train_epochs": 3, "total_flos": 1.04954332801725e+17, "trial_name": null, "trial_params": null }