| { |
| "best_metric": 0.8023715415019763, |
| "best_model_checkpoint": "./toxicity_c_202201181030/checkpoint-3000", |
| "epoch": 1.2414895617829602, |
| "global_step": 3300, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.04, |
| "learning_rate": 2.94e-05, |
| "loss": 0.2336, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.04, |
| "eval_accuracy": 0.9445713034502174, |
| "eval_f1": 0.6696495152870993, |
| "eval_loss": 0.1531878262758255, |
| "eval_precision": 0.8655421686746988, |
| "eval_recall": 0.5460626330191548, |
| "eval_runtime": 209.3715, |
| "eval_samples_per_second": 152.69, |
| "eval_steps_per_second": 19.09, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.08, |
| "learning_rate": 2.88e-05, |
| "loss": 0.1549, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.08, |
| "eval_accuracy": 0.9456973943507773, |
| "eval_f1": 0.6756352765321375, |
| "eval_loss": 0.1629406213760376, |
| "eval_precision": 0.8763936015511391, |
| "eval_recall": 0.5497111584068106, |
| "eval_runtime": 209.809, |
| "eval_samples_per_second": 152.372, |
| "eval_steps_per_second": 19.051, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.11, |
| "learning_rate": 2.8199999999999998e-05, |
| "loss": 0.158, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.11, |
| "eval_accuracy": 0.949920235227877, |
| "eval_f1": 0.7275820997107366, |
| "eval_loss": 0.13407668471336365, |
| "eval_precision": 0.8261205564142194, |
| "eval_recall": 0.6500456065673457, |
| "eval_runtime": 210.4551, |
| "eval_samples_per_second": 151.904, |
| "eval_steps_per_second": 18.992, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.15, |
| "learning_rate": 2.7600000000000003e-05, |
| "loss": 0.1321, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.15, |
| "eval_accuracy": 0.9515468109731302, |
| "eval_f1": 0.7355301348813386, |
| "eval_loss": 0.1435689479112625, |
| "eval_precision": 0.8387850467289719, |
| "eval_recall": 0.6549103070842202, |
| "eval_runtime": 209.6366, |
| "eval_samples_per_second": 152.497, |
| "eval_steps_per_second": 19.066, |
| "step": 400 |
| }, |
| { |
| "epoch": 0.19, |
| "learning_rate": 2.7000000000000002e-05, |
| "loss": 0.1408, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.19, |
| "eval_accuracy": 0.9473865307016172, |
| "eval_f1": 0.7431276725717777, |
| "eval_loss": 0.13682132959365845, |
| "eval_precision": 0.7465480208652961, |
| "eval_recall": 0.739738522347218, |
| "eval_runtime": 210.8965, |
| "eval_samples_per_second": 151.586, |
| "eval_steps_per_second": 18.952, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.23, |
| "learning_rate": 2.64e-05, |
| "loss": 0.1385, |
| "step": 600 |
| }, |
| { |
| "epoch": 0.23, |
| "eval_accuracy": 0.9507960837060903, |
| "eval_f1": 0.7527118377613583, |
| "eval_loss": 0.12854768335819244, |
| "eval_precision": 0.779296875, |
| "eval_recall": 0.7278808148373366, |
| "eval_runtime": 210.2245, |
| "eval_samples_per_second": 152.071, |
| "eval_steps_per_second": 19.013, |
| "step": 600 |
| }, |
| { |
| "epoch": 0.26, |
| "learning_rate": 2.58e-05, |
| "loss": 0.1313, |
| "step": 700 |
| }, |
| { |
| "epoch": 0.26, |
| "eval_accuracy": 0.9529857049016235, |
| "eval_f1": 0.7370078740157481, |
| "eval_loss": 0.14518482983112335, |
| "eval_precision": 0.8680956306677658, |
| "eval_recall": 0.6403162055335968, |
| "eval_runtime": 210.6173, |
| "eval_samples_per_second": 151.787, |
| "eval_steps_per_second": 18.978, |
| "step": 700 |
| }, |
| { |
| "epoch": 0.3, |
| "learning_rate": 2.52e-05, |
| "loss": 0.1399, |
| "step": 800 |
| }, |
| { |
| "epoch": 0.3, |
| "eval_accuracy": 0.949826394319497, |
| "eval_f1": 0.7610250297973777, |
| "eval_loss": 0.13412030041217804, |
| "eval_precision": 0.7461291264972246, |
| "eval_recall": 0.7765278200060809, |
| "eval_runtime": 209.9319, |
| "eval_samples_per_second": 152.283, |
| "eval_steps_per_second": 19.04, |
| "step": 800 |
| }, |
| { |
| "epoch": 0.34, |
| "learning_rate": 2.4599999999999998e-05, |
| "loss": 0.1272, |
| "step": 900 |
| }, |
| { |
| "epoch": 0.34, |
| "eval_accuracy": 0.9536425912602834, |
| "eval_f1": 0.7640878701050622, |
| "eval_loss": 0.12946395576000214, |
| "eval_precision": 0.8018710324089542, |
| "eval_recall": 0.7297050775311645, |
| "eval_runtime": 210.1105, |
| "eval_samples_per_second": 152.153, |
| "eval_steps_per_second": 19.023, |
| "step": 900 |
| }, |
| { |
| "epoch": 0.38, |
| "learning_rate": 2.4e-05, |
| "loss": 0.1324, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.38, |
| "eval_accuracy": 0.9478244549407239, |
| "eval_f1": 0.7562116340251388, |
| "eval_loss": 0.14241644740104675, |
| "eval_precision": 0.7281170841542358, |
| "eval_recall": 0.7865612648221344, |
| "eval_runtime": 210.5453, |
| "eval_samples_per_second": 151.839, |
| "eval_steps_per_second": 18.984, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.41, |
| "learning_rate": 2.3400000000000003e-05, |
| "loss": 0.1198, |
| "step": 1100 |
| }, |
| { |
| "epoch": 0.41, |
| "eval_accuracy": 0.9530482655072101, |
| "eval_f1": 0.7681137030743087, |
| "eval_loss": 0.1265845000743866, |
| "eval_precision": 0.7807788944723618, |
| "eval_recall": 0.7558528428093646, |
| "eval_runtime": 210.173, |
| "eval_samples_per_second": 152.108, |
| "eval_steps_per_second": 19.018, |
| "step": 1100 |
| }, |
| { |
| "epoch": 0.45, |
| "learning_rate": 2.2800000000000002e-05, |
| "loss": 0.1232, |
| "step": 1200 |
| }, |
| { |
| "epoch": 0.45, |
| "eval_accuracy": 0.9558634927586099, |
| "eval_f1": 0.7611308616895208, |
| "eval_loss": 0.11988817900419235, |
| "eval_precision": 0.8586707410236822, |
| "eval_recall": 0.6834904226208574, |
| "eval_runtime": 209.5961, |
| "eval_samples_per_second": 152.527, |
| "eval_steps_per_second": 19.07, |
| "step": 1200 |
| }, |
| { |
| "epoch": 0.49, |
| "learning_rate": 2.22e-05, |
| "loss": 0.1239, |
| "step": 1300 |
| }, |
| { |
| "epoch": 0.49, |
| "eval_accuracy": 0.9564578185116832, |
| "eval_f1": 0.7667560321715818, |
| "eval_loss": 0.12424681335687637, |
| "eval_precision": 0.8540500186636805, |
| "eval_recall": 0.6956521739130435, |
| "eval_runtime": 210.1306, |
| "eval_samples_per_second": 152.139, |
| "eval_steps_per_second": 19.022, |
| "step": 1300 |
| }, |
| { |
| "epoch": 0.53, |
| "learning_rate": 2.16e-05, |
| "loss": 0.1236, |
| "step": 1400 |
| }, |
| { |
| "epoch": 0.53, |
| "eval_accuracy": 0.9570834245675498, |
| "eval_f1": 0.7704918032786885, |
| "eval_loss": 0.1350642293691635, |
| "eval_precision": 0.8564522127184827, |
| "eval_recall": 0.7002128306476133, |
| "eval_runtime": 210.8014, |
| "eval_samples_per_second": 151.655, |
| "eval_steps_per_second": 18.961, |
| "step": 1400 |
| }, |
| { |
| "epoch": 0.56, |
| "learning_rate": 2.1e-05, |
| "loss": 0.1254, |
| "step": 1500 |
| }, |
| { |
| "epoch": 0.56, |
| "eval_accuracy": 0.9575526291094498, |
| "eval_f1": 0.7775774463202754, |
| "eval_loss": 0.11828587204217911, |
| "eval_precision": 0.8435277382645804, |
| "eval_recall": 0.7211918516266342, |
| "eval_runtime": 209.8683, |
| "eval_samples_per_second": 152.329, |
| "eval_steps_per_second": 19.045, |
| "step": 1500 |
| }, |
| { |
| "epoch": 0.6, |
| "learning_rate": 2.04e-05, |
| "loss": 0.1252, |
| "step": 1600 |
| }, |
| { |
| "epoch": 0.6, |
| "eval_accuracy": 0.9568644624479965, |
| "eval_f1": 0.7638294228463779, |
| "eval_loss": 0.12166079878807068, |
| "eval_precision": 0.8745098039215686, |
| "eval_recall": 0.6780176345393737, |
| "eval_runtime": 210.147, |
| "eval_samples_per_second": 152.127, |
| "eval_steps_per_second": 19.02, |
| "step": 1600 |
| }, |
| { |
| "epoch": 0.64, |
| "learning_rate": 1.98e-05, |
| "loss": 0.1256, |
| "step": 1700 |
| }, |
| { |
| "epoch": 0.64, |
| "eval_accuracy": 0.9574275078982765, |
| "eval_f1": 0.7709910819451455, |
| "eval_loss": 0.11569029092788696, |
| "eval_precision": 0.8632253202712886, |
| "eval_recall": 0.6965643052599574, |
| "eval_runtime": 210.5963, |
| "eval_samples_per_second": 151.802, |
| "eval_steps_per_second": 18.979, |
| "step": 1700 |
| }, |
| { |
| "epoch": 0.68, |
| "learning_rate": 1.9200000000000003e-05, |
| "loss": 0.1324, |
| "step": 1800 |
| }, |
| { |
| "epoch": 0.68, |
| "eval_accuracy": 0.95426819731615, |
| "eval_f1": 0.7747303543913714, |
| "eval_loss": 0.12124760448932648, |
| "eval_precision": 0.7853795688847235, |
| "eval_recall": 0.7643660687138948, |
| "eval_runtime": 210.0339, |
| "eval_samples_per_second": 152.209, |
| "eval_steps_per_second": 19.03, |
| "step": 1800 |
| }, |
| { |
| "epoch": 0.71, |
| "learning_rate": 1.86e-05, |
| "loss": 0.1192, |
| "step": 1900 |
| }, |
| { |
| "epoch": 0.71, |
| "eval_accuracy": 0.9576777503206231, |
| "eval_f1": 0.7719534805326143, |
| "eval_loss": 0.12099746614694595, |
| "eval_precision": 0.8661119515885023, |
| "eval_recall": 0.6962602614776527, |
| "eval_runtime": 209.8273, |
| "eval_samples_per_second": 152.359, |
| "eval_steps_per_second": 19.049, |
| "step": 1900 |
| }, |
| { |
| "epoch": 0.75, |
| "learning_rate": 1.8e-05, |
| "loss": 0.1234, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.75, |
| "eval_accuracy": 0.9573023866871031, |
| "eval_f1": 0.781074578989575, |
| "eval_loss": 0.11586691439151764, |
| "eval_precision": 0.8265444670739986, |
| "eval_recall": 0.7403466099118273, |
| "eval_runtime": 209.7848, |
| "eval_samples_per_second": 152.39, |
| "eval_steps_per_second": 19.053, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.79, |
| "learning_rate": 1.74e-05, |
| "loss": 0.1129, |
| "step": 2100 |
| }, |
| { |
| "epoch": 0.79, |
| "eval_accuracy": 0.9564578185116832, |
| "eval_f1": 0.7782733354571519, |
| "eval_loss": 0.12270316481590271, |
| "eval_precision": 0.8173302107728337, |
| "eval_recall": 0.7427789601702646, |
| "eval_runtime": 209.6077, |
| "eval_samples_per_second": 152.518, |
| "eval_steps_per_second": 19.069, |
| "step": 2100 |
| }, |
| { |
| "epoch": 0.83, |
| "learning_rate": 1.6800000000000002e-05, |
| "loss": 0.1244, |
| "step": 2200 |
| }, |
| { |
| "epoch": 0.83, |
| "eval_accuracy": 0.9576777503206231, |
| "eval_f1": 0.776769509981851, |
| "eval_loss": 0.11879286915063858, |
| "eval_precision": 0.8492063492063492, |
| "eval_recall": 0.7157190635451505, |
| "eval_runtime": 209.8694, |
| "eval_samples_per_second": 152.328, |
| "eval_steps_per_second": 19.045, |
| "step": 2200 |
| }, |
| { |
| "epoch": 0.87, |
| "learning_rate": 1.62e-05, |
| "loss": 0.1198, |
| "step": 2300 |
| }, |
| { |
| "epoch": 0.87, |
| "eval_accuracy": 0.9573336669898964, |
| "eval_f1": 0.7855345911949686, |
| "eval_loss": 0.12149158120155334, |
| "eval_precision": 0.8134158254640182, |
| "eval_recall": 0.7595013681970204, |
| "eval_runtime": 209.8421, |
| "eval_samples_per_second": 152.348, |
| "eval_steps_per_second": 19.048, |
| "step": 2300 |
| }, |
| { |
| "epoch": 0.9, |
| "learning_rate": 1.56e-05, |
| "loss": 0.1167, |
| "step": 2400 |
| }, |
| { |
| "epoch": 0.9, |
| "eval_accuracy": 0.9582095154681097, |
| "eval_f1": 0.7829054273643159, |
| "eval_loss": 0.11379627883434296, |
| "eval_precision": 0.8408376963350785, |
| "eval_recall": 0.7324414715719063, |
| "eval_runtime": 210.0964, |
| "eval_samples_per_second": 152.163, |
| "eval_steps_per_second": 19.025, |
| "step": 2400 |
| }, |
| { |
| "epoch": 0.94, |
| "learning_rate": 1.5e-05, |
| "loss": 0.1162, |
| "step": 2500 |
| }, |
| { |
| "epoch": 0.94, |
| "eval_accuracy": 0.9579592730457631, |
| "eval_f1": 0.7870722433460076, |
| "eval_loss": 0.11324156820774078, |
| "eval_precision": 0.8217002977174992, |
| "eval_recall": 0.7552447552447552, |
| "eval_runtime": 210.2617, |
| "eval_samples_per_second": 152.044, |
| "eval_steps_per_second": 19.01, |
| "step": 2500 |
| }, |
| { |
| "epoch": 0.98, |
| "learning_rate": 1.44e-05, |
| "loss": 0.1202, |
| "step": 2600 |
| }, |
| { |
| "epoch": 0.98, |
| "eval_accuracy": 0.9577403109262098, |
| "eval_f1": 0.785385226370135, |
| "eval_loss": 0.11547254770994186, |
| "eval_precision": 0.8223552894211577, |
| "eval_recall": 0.7515962298570994, |
| "eval_runtime": 209.2873, |
| "eval_samples_per_second": 152.752, |
| "eval_steps_per_second": 19.098, |
| "step": 2600 |
| }, |
| { |
| "epoch": 1.02, |
| "learning_rate": 1.3800000000000002e-05, |
| "loss": 0.1069, |
| "step": 2700 |
| }, |
| { |
| "epoch": 1.02, |
| "eval_accuracy": 0.9589289624323564, |
| "eval_f1": 0.7883282282766404, |
| "eval_loss": 0.11415638774633408, |
| "eval_precision": 0.8390528483184626, |
| "eval_recall": 0.7433870477348739, |
| "eval_runtime": 209.3049, |
| "eval_samples_per_second": 152.739, |
| "eval_steps_per_second": 19.097, |
| "step": 2700 |
| }, |
| { |
| "epoch": 1.05, |
| "learning_rate": 1.32e-05, |
| "loss": 0.0866, |
| "step": 2800 |
| }, |
| { |
| "epoch": 1.05, |
| "eval_accuracy": 0.9558947730614032, |
| "eval_f1": 0.7875866224766495, |
| "eval_loss": 0.12708334624767303, |
| "eval_precision": 0.7805315019408778, |
| "eval_recall": 0.7947704469443599, |
| "eval_runtime": 208.8539, |
| "eval_samples_per_second": 153.069, |
| "eval_steps_per_second": 19.138, |
| "step": 2800 |
| }, |
| { |
| "epoch": 1.09, |
| "learning_rate": 1.26e-05, |
| "loss": 0.0913, |
| "step": 2900 |
| }, |
| { |
| "epoch": 1.09, |
| "eval_accuracy": 0.9557696518502299, |
| "eval_f1": 0.7855626326963907, |
| "eval_loss": 0.12602387368679047, |
| "eval_precision": 0.783661119515885, |
| "eval_recall": 0.7874733961690483, |
| "eval_runtime": 209.1412, |
| "eval_samples_per_second": 152.858, |
| "eval_steps_per_second": 19.111, |
| "step": 2900 |
| }, |
| { |
| "epoch": 1.13, |
| "learning_rate": 1.2e-05, |
| "loss": 0.087, |
| "step": 3000 |
| }, |
| { |
| "epoch": 1.13, |
| "eval_accuracy": 0.95548812912509, |
| "eval_f1": 0.7876436352783167, |
| "eval_loss": 0.12803590297698975, |
| "eval_precision": 0.7734466588511137, |
| "eval_recall": 0.8023715415019763, |
| "eval_runtime": 209.0203, |
| "eval_samples_per_second": 152.947, |
| "eval_steps_per_second": 19.123, |
| "step": 3000 |
| }, |
| { |
| "epoch": 1.17, |
| "learning_rate": 1.1400000000000001e-05, |
| "loss": 0.0839, |
| "step": 3100 |
| }, |
| { |
| "epoch": 1.17, |
| "eval_accuracy": 0.958991523037943, |
| "eval_f1": 0.7922016167379933, |
| "eval_loss": 0.12513290345668793, |
| "eval_precision": 0.8274834437086093, |
| "eval_recall": 0.759805411979325, |
| "eval_runtime": 209.1866, |
| "eval_samples_per_second": 152.825, |
| "eval_steps_per_second": 19.107, |
| "step": 3100 |
| }, |
| { |
| "epoch": 1.2, |
| "learning_rate": 1.08e-05, |
| "loss": 0.0889, |
| "step": 3200 |
| }, |
| { |
| "epoch": 1.2, |
| "eval_accuracy": 0.9583033563764898, |
| "eval_f1": 0.7890489001424276, |
| "eval_loss": 0.12521126866340637, |
| "eval_precision": 0.8227722772277227, |
| "eval_recall": 0.7579811492854971, |
| "eval_runtime": 209.1839, |
| "eval_samples_per_second": 152.827, |
| "eval_steps_per_second": 19.108, |
| "step": 3200 |
| }, |
| { |
| "epoch": 1.24, |
| "learning_rate": 1.02e-05, |
| "loss": 0.0887, |
| "step": 3300 |
| }, |
| { |
| "epoch": 1.24, |
| "eval_accuracy": 0.9562701366949232, |
| "eval_f1": 0.7879890809827116, |
| "eval_loss": 0.12457986176013947, |
| "eval_precision": 0.7860816944024206, |
| "eval_recall": 0.7899057464274856, |
| "eval_runtime": 209.6152, |
| "eval_samples_per_second": 152.513, |
| "eval_steps_per_second": 19.068, |
| "step": 3300 |
| } |
| ], |
| "max_steps": 5000, |
| "num_train_epochs": 2, |
| "total_flos": 2.279247411490296e+16, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
