| { |
| "best_metric": 0.040135130286216736, |
| "best_model_checkpoint": "/content/qlora_vuln_model/checkpoint-500", |
| "epoch": 0.9995294117647059, |
| "eval_steps": 100, |
| "global_step": 531, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.018823529411764704, |
| "grad_norm": 10.599871635437012, |
| "learning_rate": 3.7037037037037037e-05, |
| "loss": 24.749, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.03764705882352941, |
| "grad_norm": 12.27408218383789, |
| "learning_rate": 7.037037037037038e-05, |
| "loss": 21.6719, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.05647058823529412, |
| "grad_norm": 8.119482040405273, |
| "learning_rate": 9.999611462404875e-05, |
| "loss": 15.8453, |
| "step": 30 |
| }, |
| { |
| "epoch": 0.07529411764705882, |
| "grad_norm": 6.152805805206299, |
| "learning_rate": 9.986018985905901e-05, |
| "loss": 8.853, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.09411764705882353, |
| "grad_norm": 3.1945641040802, |
| "learning_rate": 9.953059972065265e-05, |
| "loss": 5.7261, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.11294117647058824, |
| "grad_norm": 4.127964496612549, |
| "learning_rate": 9.900862439242719e-05, |
| "loss": 4.4831, |
| "step": 60 |
| }, |
| { |
| "epoch": 0.13176470588235295, |
| "grad_norm": 4.9141082763671875, |
| "learning_rate": 9.829629131445342e-05, |
| "loss": 3.5453, |
| "step": 70 |
| }, |
| { |
| "epoch": 0.15058823529411763, |
| "grad_norm": 5.234217166900635, |
| "learning_rate": 9.73963673083566e-05, |
| "loss": 2.6558, |
| "step": 80 |
| }, |
| { |
| "epoch": 0.16941176470588235, |
| "grad_norm": 5.785156726837158, |
| "learning_rate": 9.631234783051544e-05, |
| "loss": 2.1081, |
| "step": 90 |
| }, |
| { |
| "epoch": 0.18823529411764706, |
| "grad_norm": 4.8869309425354, |
| "learning_rate": 9.504844339512095e-05, |
| "loss": 1.6306, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.18823529411764706, |
| "eval_loss": 0.09413356333971024, |
| "eval_runtime": 1659.6813, |
| "eval_samples_per_second": 0.904, |
| "eval_steps_per_second": 0.904, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.20705882352941177, |
| "grad_norm": 5.362945556640625, |
| "learning_rate": 9.360956321983028e-05, |
| "loss": 1.3676, |
| "step": 110 |
| }, |
| { |
| "epoch": 0.22588235294117648, |
| "grad_norm": 3.2882444858551025, |
| "learning_rate": 9.200129615753859e-05, |
| "loss": 1.2626, |
| "step": 120 |
| }, |
| { |
| "epoch": 0.2447058823529412, |
| "grad_norm": 4.0295610427856445, |
| "learning_rate": 9.022988898833342e-05, |
| "loss": 1.1255, |
| "step": 130 |
| }, |
| { |
| "epoch": 0.2635294117647059, |
| "grad_norm": 3.485719680786133, |
| "learning_rate": 8.83022221559489e-05, |
| "loss": 1.003, |
| "step": 140 |
| }, |
| { |
| "epoch": 0.2823529411764706, |
| "grad_norm": 5.177315711975098, |
| "learning_rate": 8.622578304296364e-05, |
| "loss": 0.9306, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.30117647058823527, |
| "grad_norm": 3.201077938079834, |
| "learning_rate": 8.400863688854597e-05, |
| "loss": 0.9209, |
| "step": 160 |
| }, |
| { |
| "epoch": 0.32, |
| "grad_norm": 3.020800828933716, |
| "learning_rate": 8.1659395461707e-05, |
| "loss": 0.8847, |
| "step": 170 |
| }, |
| { |
| "epoch": 0.3388235294117647, |
| "grad_norm": 2.2553532123565674, |
| "learning_rate": 7.91871836117395e-05, |
| "loss": 0.8919, |
| "step": 180 |
| }, |
| { |
| "epoch": 0.35764705882352943, |
| "grad_norm": 2.698606014251709, |
| "learning_rate": 7.660160382576683e-05, |
| "loss": 0.7767, |
| "step": 190 |
| }, |
| { |
| "epoch": 0.3764705882352941, |
| "grad_norm": 3.7030203342437744, |
| "learning_rate": 7.391269893106592e-05, |
| "loss": 0.7816, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.3764705882352941, |
| "eval_loss": 0.049124155193567276, |
| "eval_runtime": 1662.6885, |
| "eval_samples_per_second": 0.902, |
| "eval_steps_per_second": 0.902, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.3952941176470588, |
| "grad_norm": 2.39546275138855, |
| "learning_rate": 7.113091308703498e-05, |
| "loss": 0.7294, |
| "step": 210 |
| }, |
| { |
| "epoch": 0.41411764705882353, |
| "grad_norm": 1.8437654972076416, |
| "learning_rate": 6.826705121831976e-05, |
| "loss": 0.7932, |
| "step": 220 |
| }, |
| { |
| "epoch": 0.4329411764705882, |
| "grad_norm": 3.2156364917755127, |
| "learning_rate": 6.533223704666672e-05, |
| "loss": 0.7796, |
| "step": 230 |
| }, |
| { |
| "epoch": 0.45176470588235296, |
| "grad_norm": 3.9197657108306885, |
| "learning_rate": 6.233786988451468e-05, |
| "loss": 0.7135, |
| "step": 240 |
| }, |
| { |
| "epoch": 0.47058823529411764, |
| "grad_norm": 1.7182209491729736, |
| "learning_rate": 5.9295580358145744e-05, |
| "loss": 0.772, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.4894117647058824, |
| "grad_norm": 1.3930495977401733, |
| "learning_rate": 5.621718523237427e-05, |
| "loss": 0.7166, |
| "step": 260 |
| }, |
| { |
| "epoch": 0.508235294117647, |
| "grad_norm": 2.4380180835723877, |
| "learning_rate": 5.311464151224261e-05, |
| "loss": 0.7207, |
| "step": 270 |
| }, |
| { |
| "epoch": 0.5270588235294118, |
| "grad_norm": 2.6220102310180664, |
| "learning_rate": 5e-05, |
| "loss": 0.711, |
| "step": 280 |
| }, |
| { |
| "epoch": 0.5458823529411765, |
| "grad_norm": 1.6312707662582397, |
| "learning_rate": 4.68853584877574e-05, |
| "loss": 0.705, |
| "step": 290 |
| }, |
| { |
| "epoch": 0.5647058823529412, |
| "grad_norm": 2.2927358150482178, |
| "learning_rate": 4.378281476762576e-05, |
| "loss": 0.68, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.5647058823529412, |
| "eval_loss": 0.043063607066869736, |
| "eval_runtime": 1659.8453, |
| "eval_samples_per_second": 0.904, |
| "eval_steps_per_second": 0.904, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.5835294117647059, |
| "grad_norm": 2.068061590194702, |
| "learning_rate": 4.0704419641854274e-05, |
| "loss": 0.7047, |
| "step": 310 |
| }, |
| { |
| "epoch": 0.6023529411764705, |
| "grad_norm": 1.5260778665542603, |
| "learning_rate": 3.7662130115485314e-05, |
| "loss": 0.6668, |
| "step": 320 |
| }, |
| { |
| "epoch": 0.6211764705882353, |
| "grad_norm": 1.9088001251220703, |
| "learning_rate": 3.4667762953333295e-05, |
| "loss": 0.6768, |
| "step": 330 |
| }, |
| { |
| "epoch": 0.64, |
| "grad_norm": 1.6153029203414917, |
| "learning_rate": 3.173294878168025e-05, |
| "loss": 0.6679, |
| "step": 340 |
| }, |
| { |
| "epoch": 0.6588235294117647, |
| "grad_norm": 1.6346434354782104, |
| "learning_rate": 2.886908691296504e-05, |
| "loss": 0.6432, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.6776470588235294, |
| "grad_norm": 2.301887273788452, |
| "learning_rate": 2.6087301068934106e-05, |
| "loss": 0.6596, |
| "step": 360 |
| }, |
| { |
| "epoch": 0.6964705882352941, |
| "grad_norm": 1.4241043329238892, |
| "learning_rate": 2.3398396174233178e-05, |
| "loss": 0.646, |
| "step": 370 |
| }, |
| { |
| "epoch": 0.7152941176470589, |
| "grad_norm": 1.6611512899398804, |
| "learning_rate": 2.0812816388260518e-05, |
| "loss": 0.646, |
| "step": 380 |
| }, |
| { |
| "epoch": 0.7341176470588235, |
| "grad_norm": 1.8180968761444092, |
| "learning_rate": 1.8340604538293015e-05, |
| "loss": 0.6794, |
| "step": 390 |
| }, |
| { |
| "epoch": 0.7529411764705882, |
| "grad_norm": 1.842568039894104, |
| "learning_rate": 1.599136311145402e-05, |
| "loss": 0.6595, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.7529411764705882, |
| "eval_loss": 0.0410185307264328, |
| "eval_runtime": 1654.6683, |
| "eval_samples_per_second": 0.907, |
| "eval_steps_per_second": 0.907, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.7717647058823529, |
| "grad_norm": 1.68463134765625, |
| "learning_rate": 1.3774216957036367e-05, |
| "loss": 0.6562, |
| "step": 410 |
| }, |
| { |
| "epoch": 0.7905882352941176, |
| "grad_norm": 1.7055809497833252, |
| "learning_rate": 1.1697777844051105e-05, |
| "loss": 0.667, |
| "step": 420 |
| }, |
| { |
| "epoch": 0.8094117647058824, |
| "grad_norm": 1.723899245262146, |
| "learning_rate": 9.770111011666583e-06, |
| "loss": 0.6825, |
| "step": 430 |
| }, |
| { |
| "epoch": 0.8282352941176471, |
| "grad_norm": 1.6462485790252686, |
| "learning_rate": 7.998703842461431e-06, |
| "loss": 0.6474, |
| "step": 440 |
| }, |
| { |
| "epoch": 0.8470588235294118, |
| "grad_norm": 2.414433717727661, |
| "learning_rate": 6.390436780169734e-06, |
| "loss": 0.6615, |
| "step": 450 |
| }, |
| { |
| "epoch": 0.8658823529411764, |
| "grad_norm": 1.415352463722229, |
| "learning_rate": 4.951556604879048e-06, |
| "loss": 0.6418, |
| "step": 460 |
| }, |
| { |
| "epoch": 0.8847058823529412, |
| "grad_norm": 1.53042733669281, |
| "learning_rate": 3.687652169484568e-06, |
| "loss": 0.6491, |
| "step": 470 |
| }, |
| { |
| "epoch": 0.9035294117647059, |
| "grad_norm": 2.4654157161712646, |
| "learning_rate": 2.603632691643415e-06, |
| "loss": 0.6648, |
| "step": 480 |
| }, |
| { |
| "epoch": 0.9223529411764706, |
| "grad_norm": 1.743988037109375, |
| "learning_rate": 1.70370868554659e-06, |
| "loss": 0.6416, |
| "step": 490 |
| }, |
| { |
| "epoch": 0.9411764705882353, |
| "grad_norm": 1.802857756614685, |
| "learning_rate": 9.913756075728087e-07, |
| "loss": 0.6335, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.9411764705882353, |
| "eval_loss": 0.040135130286216736, |
| "eval_runtime": 1659.3157, |
| "eval_samples_per_second": 0.904, |
| "eval_steps_per_second": 0.904, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.96, |
| "grad_norm": 1.6982253789901733, |
| "learning_rate": 4.6940027934735954e-07, |
| "loss": 0.628, |
| "step": 510 |
| }, |
| { |
| "epoch": 0.9788235294117648, |
| "grad_norm": 1.3386036157608032, |
| "learning_rate": 1.3981014094099353e-07, |
| "loss": 0.6857, |
| "step": 520 |
| }, |
| { |
| "epoch": 0.9976470588235294, |
| "grad_norm": 2.097684621810913, |
| "learning_rate": 3.885375951256931e-09, |
| "loss": 0.6498, |
| "step": 530 |
| } |
| ], |
| "logging_steps": 10, |
| "max_steps": 531, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 1, |
| "save_steps": 100, |
| "stateful_callbacks": { |
| "EarlyStoppingCallback": { |
| "args": { |
| "early_stopping_patience": 3, |
| "early_stopping_threshold": 0.0 |
| }, |
| "attributes": { |
| "early_stopping_patience_counter": 0 |
| } |
| }, |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": true |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 1.3002607472723558e+17, |
| "train_batch_size": 1, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
