{ "best_metric": null, "best_model_checkpoint": null, "epoch": 20.0, "eval_steps": 500, "global_step": 2440, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "grad_norm": 5.043847560882568, "learning_rate": 4.75e-05, "loss": 0.5619, "step": 122 }, { "epoch": 1.0, "eval_accuracy": 0.7167919799498746, "eval_f1": 0.6483759017352311, "eval_loss": 0.5126742124557495, "eval_precision": 0.653631914482581, "eval_recall": 0.6446172031278414, "eval_runtime": 5.1315, "eval_samples_per_second": 77.755, "eval_steps_per_second": 9.744, "step": 122 }, { "epoch": 2.0, "grad_norm": 3.8154773712158203, "learning_rate": 4.5e-05, "loss": 0.5059, "step": 244 }, { "epoch": 2.0, "eval_accuracy": 0.7343358395989975, "eval_f1": 0.7022388059701492, "eval_loss": 0.4967426359653473, "eval_precision": 0.695631275368511, "eval_recall": 0.7220403709765412, "eval_runtime": 5.0502, "eval_samples_per_second": 79.006, "eval_steps_per_second": 9.901, "step": 244 }, { "epoch": 3.0, "grad_norm": 4.306382656097412, "learning_rate": 4.25e-05, "loss": 0.4822, "step": 366 }, { "epoch": 3.0, "eval_accuracy": 0.7468671679197995, "eval_f1": 0.7064830258490717, "eval_loss": 0.45057693123817444, "eval_precision": 0.7006054632497888, "eval_recall": 0.7159028914348063, "eval_runtime": 5.0702, "eval_samples_per_second": 78.696, "eval_steps_per_second": 9.862, "step": 366 }, { "epoch": 4.0, "grad_norm": 3.528663158416748, "learning_rate": 4e-05, "loss": 0.4402, "step": 488 }, { "epoch": 4.0, "eval_accuracy": 0.8195488721804511, "eval_f1": 0.7727934198038595, "eval_loss": 0.39838677644729614, "eval_precision": 0.7875606352581567, "eval_recall": 0.76232042189489, "eval_runtime": 5.0966, "eval_samples_per_second": 78.287, "eval_steps_per_second": 9.81, "step": 488 }, { "epoch": 5.0, "grad_norm": 0.9896986484527588, "learning_rate": 3.7500000000000003e-05, "loss": 0.4068, "step": 610 }, { "epoch": 5.0, "eval_accuracy": 0.7869674185463659, "eval_f1": 0.7560721246862345, "eval_loss": 0.4135726988315582, "eval_precision": 0.7472972972972973, "eval_recall": 0.7717766866703037, "eval_runtime": 5.0746, "eval_samples_per_second": 78.627, "eval_steps_per_second": 9.853, "step": 610 }, { "epoch": 6.0, "grad_norm": 2.2119059562683105, "learning_rate": 3.5e-05, "loss": 0.3791, "step": 732 }, { "epoch": 6.0, "eval_accuracy": 0.8320802005012531, "eval_f1": 0.7979441442122369, "eval_loss": 0.3770710527896881, "eval_precision": 0.7972133421798662, "eval_recall": 0.7986906710310966, "eval_runtime": 5.0775, "eval_samples_per_second": 78.582, "eval_steps_per_second": 9.847, "step": 732 }, { "epoch": 7.0, "grad_norm": 1.8872367143630981, "learning_rate": 3.2500000000000004e-05, "loss": 0.3635, "step": 854 }, { "epoch": 7.0, "eval_accuracy": 0.8195488721804511, "eval_f1": 0.7912087912087913, "eval_loss": 0.39156973361968994, "eval_precision": 0.7821969696969697, "eval_recall": 0.8048281505728314, "eval_runtime": 5.0874, "eval_samples_per_second": 78.429, "eval_steps_per_second": 9.828, "step": 854 }, { "epoch": 8.0, "grad_norm": 5.425744533538818, "learning_rate": 3e-05, "loss": 0.3433, "step": 976 }, { "epoch": 8.0, "eval_accuracy": 0.8295739348370927, "eval_f1": 0.7974014336917563, "eval_loss": 0.37988749146461487, "eval_precision": 0.7934340756451043, "eval_recall": 0.801918530641935, "eval_runtime": 5.0731, "eval_samples_per_second": 78.649, "eval_steps_per_second": 9.856, "step": 976 }, { "epoch": 9.0, "grad_norm": 4.274303436279297, "learning_rate": 2.7500000000000004e-05, "loss": 0.3379, "step": 1098 }, { "epoch": 9.0, "eval_accuracy": 0.8270676691729323, "eval_f1": 0.7958625291958625, "eval_loss": 0.3713899850845337, "eval_precision": 0.7902930402930403, "eval_recall": 0.802645935624659, "eval_runtime": 5.0561, "eval_samples_per_second": 78.914, "eval_steps_per_second": 9.889, "step": 1098 }, { "epoch": 10.0, "grad_norm": 6.189886569976807, "learning_rate": 2.5e-05, "loss": 0.3296, "step": 1220 }, { "epoch": 10.0, "eval_accuracy": 0.8370927318295739, "eval_f1": 0.8039756622954537, "eval_loss": 0.3635365664958954, "eval_precision": 0.8032299897460643, "eval_recall": 0.8047372249499909, "eval_runtime": 5.0588, "eval_samples_per_second": 78.873, "eval_steps_per_second": 9.884, "step": 1220 }, { "epoch": 11.0, "grad_norm": 4.368456840515137, "learning_rate": 2.25e-05, "loss": 0.3105, "step": 1342 }, { "epoch": 11.0, "eval_accuracy": 0.8295739348370927, "eval_f1": 0.7983530530947143, "eval_loss": 0.36521923542022705, "eval_precision": 0.7932554744525547, "eval_recall": 0.8044189852700492, "eval_runtime": 5.1131, "eval_samples_per_second": 78.035, "eval_steps_per_second": 9.779, "step": 1342 }, { "epoch": 12.0, "grad_norm": 11.303323745727539, "learning_rate": 2e-05, "loss": 0.3024, "step": 1464 }, { "epoch": 12.0, "eval_accuracy": 0.8345864661654135, "eval_f1": 0.8069354838709677, "eval_loss": 0.3701755404472351, "eval_precision": 0.798820781588242, "eval_recall": 0.8179669030732861, "eval_runtime": 5.0725, "eval_samples_per_second": 78.659, "eval_steps_per_second": 9.857, "step": 1464 }, { "epoch": 13.0, "grad_norm": 1.7624099254608154, "learning_rate": 1.75e-05, "loss": 0.309, "step": 1586 }, { "epoch": 13.0, "eval_accuracy": 0.8370927318295739, "eval_f1": 0.8039756622954537, "eval_loss": 0.35115164518356323, "eval_precision": 0.8032299897460643, "eval_recall": 0.8047372249499909, "eval_runtime": 5.0924, "eval_samples_per_second": 78.352, "eval_steps_per_second": 9.818, "step": 1586 }, { "epoch": 14.0, "grad_norm": 20.14969825744629, "learning_rate": 1.5e-05, "loss": 0.3021, "step": 1708 }, { "epoch": 14.0, "eval_accuracy": 0.8395989974937343, "eval_f1": 0.8074672778816575, "eval_loss": 0.35049960017204285, "eval_precision": 0.8059873949579832, "eval_recall": 0.8090107292234952, "eval_runtime": 5.1067, "eval_samples_per_second": 78.132, "eval_steps_per_second": 9.791, "step": 1708 }, { "epoch": 15.0, "grad_norm": 1.1024099588394165, "learning_rate": 1.25e-05, "loss": 0.2903, "step": 1830 }, { "epoch": 15.0, "eval_accuracy": 0.8421052631578947, "eval_f1": 0.8136136136136136, "eval_loss": 0.3553332984447479, "eval_precision": 0.8076923076923077, "eval_recall": 0.820785597381342, "eval_runtime": 5.1172, "eval_samples_per_second": 77.972, "eval_steps_per_second": 9.771, "step": 1830 }, { "epoch": 16.0, "grad_norm": 7.029820442199707, "learning_rate": 1e-05, "loss": 0.2834, "step": 1952 }, { "epoch": 16.0, "eval_accuracy": 0.8395989974937343, "eval_f1": 0.8119476846942383, "eval_loss": 0.3529706597328186, "eval_precision": 0.8045650301464256, "eval_recall": 0.8215130023640662, "eval_runtime": 5.0639, "eval_samples_per_second": 78.793, "eval_steps_per_second": 9.874, "step": 1952 }, { "epoch": 17.0, "grad_norm": 4.537665367126465, "learning_rate": 7.5e-06, "loss": 0.2811, "step": 2074 }, { "epoch": 17.0, "eval_accuracy": 0.8446115288220551, "eval_f1": 0.8134839254478557, "eval_loss": 0.34711357951164246, "eval_precision": 0.8119747899159664, "eval_recall": 0.8150572831423895, "eval_runtime": 5.0778, "eval_samples_per_second": 78.577, "eval_steps_per_second": 9.847, "step": 2074 }, { "epoch": 18.0, "grad_norm": 3.4649994373321533, "learning_rate": 5e-06, "loss": 0.288, "step": 2196 }, { "epoch": 18.0, "eval_accuracy": 0.8446115288220551, "eval_f1": 0.8161454307628278, "eval_loss": 0.350491464138031, "eval_precision": 0.8107299270072992, "eval_recall": 0.8225586470267321, "eval_runtime": 5.0501, "eval_samples_per_second": 79.009, "eval_steps_per_second": 9.901, "step": 2196 }, { "epoch": 19.0, "grad_norm": 4.690791130065918, "learning_rate": 2.5e-06, "loss": 0.277, "step": 2318 }, { "epoch": 19.0, "eval_accuracy": 0.8395989974937343, "eval_f1": 0.8084033613445378, "eval_loss": 0.3479045629501343, "eval_precision": 0.8055472976990309, "eval_recall": 0.8115111838516094, "eval_runtime": 5.059, "eval_samples_per_second": 78.87, "eval_steps_per_second": 9.883, "step": 2318 }, { "epoch": 20.0, "grad_norm": 7.5570807456970215, "learning_rate": 0.0, "loss": 0.2775, "step": 2440 }, { "epoch": 20.0, "eval_accuracy": 0.8395989974937343, "eval_f1": 0.8084033613445378, "eval_loss": 0.3485976755619049, "eval_precision": 0.8055472976990309, "eval_recall": 0.8115111838516094, "eval_runtime": 5.0614, "eval_samples_per_second": 78.831, "eval_steps_per_second": 9.879, "step": 2440 }, { "epoch": 20.0, "step": 2440, "total_flos": 7597037114448000.0, "train_loss": 0.35358311856379276, "train_runtime": 1955.2774, "train_samples_per_second": 37.212, "train_steps_per_second": 1.248 } ], "logging_steps": 500, "max_steps": 2440, "num_input_tokens_seen": 0, "num_train_epochs": 20, "save_steps": 500, "total_flos": 7597037114448000.0, "train_batch_size": 30, "trial_name": null, "trial_params": null }