| { | |
| "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 | |
| } | |