{ "best_metric": null, "best_model_checkpoint": null, "epoch": 5.361930294906166, "eval_steps": 500, "global_step": 4000, "is_hyper_param_search": true, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.6702412868632708, "grad_norm": 9.020133018493652, "learning_rate": 1.4299394279479472e-06, "loss": 0.6244, "step": 500 }, { "epoch": 1.0, "eval_accuracy": 0.7267851233482361, "eval_loss": 0.5142917633056641, "eval_runtime": 9.4466, "eval_samples_per_second": 315.773, "eval_steps_per_second": 19.795, "step": 746 }, { "epoch": 1.3404825737265416, "grad_norm": 10.139765739440918, "learning_rate": 1.3272138943309972e-06, "loss": 0.5438, "step": 1000 }, { "epoch": 2.0, "eval_accuracy": 0.7405296564102173, "eval_loss": 0.49428224563598633, "eval_runtime": 9.4794, "eval_samples_per_second": 314.682, "eval_steps_per_second": 19.727, "step": 1492 }, { "epoch": 2.0107238605898123, "grad_norm": 7.425673961639404, "learning_rate": 1.2244883607140467e-06, "loss": 0.5108, "step": 1500 }, { "epoch": 2.680965147453083, "grad_norm": 8.044295310974121, "learning_rate": 1.1217628270970966e-06, "loss": 0.4787, "step": 2000 }, { "epoch": 3.0, "eval_accuracy": 0.7468990683555603, "eval_loss": 0.5006520748138428, "eval_runtime": 9.5789, "eval_samples_per_second": 311.415, "eval_steps_per_second": 19.522, "step": 2238 }, { "epoch": 3.351206434316354, "grad_norm": 19.568927764892578, "learning_rate": 1.0190372934801463e-06, "loss": 0.458, "step": 2500 }, { "epoch": 4.0, "eval_accuracy": 0.7522628307342529, "eval_loss": 0.49020281434059143, "eval_runtime": 9.4558, "eval_samples_per_second": 315.467, "eval_steps_per_second": 19.776, "step": 2984 }, { "epoch": 4.021447721179625, "grad_norm": 5.546102046966553, "learning_rate": 9.16311759863196e-07, "loss": 0.4303, "step": 3000 }, { "epoch": 4.6916890080428955, "grad_norm": 15.734197616577148, "learning_rate": 8.135862262462459e-07, "loss": 0.4115, "step": 3500 }, { "epoch": 5.0, "eval_accuracy": 0.7536037564277649, "eval_loss": 0.5176616907119751, "eval_runtime": 9.5583, "eval_samples_per_second": 312.084, "eval_steps_per_second": 19.564, "step": 3730 }, { "epoch": 5.361930294906166, "grad_norm": 15.1699857711792, "learning_rate": 7.108606926292956e-07, "loss": 0.401, "step": 4000 } ], "logging_steps": 500, "max_steps": 7460, "num_input_tokens_seen": 0, "num_train_epochs": 10, "save_steps": 500, "total_flos": 1.0420730286237396e+16, "train_batch_size": 16, "trial_name": null, "trial_params": { "learning_rate": 1.5326649615648975e-06, "per_device_train_batch_size": 16 } }