| { | |
| "best_metric": 0.7865168539325843, | |
| "best_model_checkpoint": "distilhubert-finetuned-not-a-word2/run-12/checkpoint-240", | |
| "epoch": 5.0, | |
| "eval_steps": 500, | |
| "global_step": 240, | |
| "is_hyper_param_search": true, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 1.2605946063995361, | |
| "learning_rate": 5.868670588038625e-06, | |
| "loss": 0.7019, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 1.1554573774337769, | |
| "learning_rate": 1.173734117607725e-05, | |
| "loss": 0.6822, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "grad_norm": 1.2631292343139648, | |
| "learning_rate": 1.7606011764115876e-05, | |
| "loss": 0.6823, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "grad_norm": 0.8986643552780151, | |
| "learning_rate": 2.34746823521545e-05, | |
| "loss": 0.6573, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "grad_norm": 0.8833038210868835, | |
| "learning_rate": 2.934335294019313e-05, | |
| "loss": 0.6302, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "grad_norm": 0.6974115967750549, | |
| "learning_rate": 3.521202352823175e-05, | |
| "loss": 0.6181, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "grad_norm": 2.8611502647399902, | |
| "learning_rate": 4.1080694116270374e-05, | |
| "loss": 0.5431, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "grad_norm": 1.6229726076126099, | |
| "learning_rate": 4.6949364704309e-05, | |
| "loss": 0.6722, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "grad_norm": 3.6481218338012695, | |
| "learning_rate": 5.281803529234763e-05, | |
| "loss": 0.6342, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_f1": 0.7326732673267327, | |
| "eval_loss": 0.7596015930175781, | |
| "eval_runtime": 1.3819, | |
| "eval_samples_per_second": 46.313, | |
| "eval_steps_per_second": 5.789, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 1.04, | |
| "grad_norm": 4.841191291809082, | |
| "learning_rate": 5.607840784125798e-05, | |
| "loss": 0.7106, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "grad_norm": 0.9242434501647949, | |
| "learning_rate": 5.542633333147591e-05, | |
| "loss": 0.703, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 1.9776262044906616, | |
| "learning_rate": 5.4774258821693835e-05, | |
| "loss": 0.4098, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.35, | |
| "grad_norm": 0.7563148140907288, | |
| "learning_rate": 5.412218431191177e-05, | |
| "loss": 0.5599, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 1.46, | |
| "grad_norm": 1.44520103931427, | |
| "learning_rate": 5.34701098021297e-05, | |
| "loss": 0.4823, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.56, | |
| "grad_norm": Infinity, | |
| "learning_rate": 5.294845019430404e-05, | |
| "loss": 0.7782, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.67, | |
| "grad_norm": 2.0363686084747314, | |
| "learning_rate": 5.229637568452197e-05, | |
| "loss": 0.6186, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.77, | |
| "grad_norm": 1.6177257299423218, | |
| "learning_rate": 5.16443011747399e-05, | |
| "loss": 0.5907, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.88, | |
| "grad_norm": 1.8490890264511108, | |
| "learning_rate": 5.0992226664957835e-05, | |
| "loss": 0.5909, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.98, | |
| "grad_norm": 6.961443901062012, | |
| "learning_rate": 5.034015215517576e-05, | |
| "loss": 0.603, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_f1": 0.74, | |
| "eval_loss": 0.6603889465332031, | |
| "eval_runtime": 1.4385, | |
| "eval_samples_per_second": 44.492, | |
| "eval_steps_per_second": 5.561, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 2.08, | |
| "grad_norm": 1.232475757598877, | |
| "learning_rate": 4.981849254735011e-05, | |
| "loss": 0.572, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 2.19, | |
| "grad_norm": 1.0792688131332397, | |
| "learning_rate": 4.9166418037568046e-05, | |
| "loss": 0.4763, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 2.29, | |
| "grad_norm": 1.4945162534713745, | |
| "learning_rate": 4.8514343527785974e-05, | |
| "loss": 0.5616, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 2.4, | |
| "grad_norm": 1.323120355606079, | |
| "learning_rate": 4.78622690180039e-05, | |
| "loss": 0.5489, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "grad_norm": 5.674253463745117, | |
| "learning_rate": 4.721019450822183e-05, | |
| "loss": 0.5401, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 2.6, | |
| "grad_norm": 4.727626323699951, | |
| "learning_rate": 4.668853490039618e-05, | |
| "loss": 0.4278, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 2.71, | |
| "grad_norm": 8.496376037597656, | |
| "learning_rate": 4.6036460390614105e-05, | |
| "loss": 0.6216, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 2.81, | |
| "grad_norm": 2.796064615249634, | |
| "learning_rate": 4.538438588083204e-05, | |
| "loss": 0.3347, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 2.92, | |
| "grad_norm": 7.401052951812744, | |
| "learning_rate": 4.486272627300638e-05, | |
| "loss": 0.3109, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_f1": 0.72, | |
| "eval_loss": 0.8806066513061523, | |
| "eval_runtime": 1.3788, | |
| "eval_samples_per_second": 46.418, | |
| "eval_steps_per_second": 5.802, | |
| "step": 144 | |
| }, | |
| { | |
| "epoch": 3.02, | |
| "grad_norm": 3.1919455528259277, | |
| "learning_rate": 4.421065176322431e-05, | |
| "loss": 0.485, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 3.12, | |
| "grad_norm": 56.47049331665039, | |
| "learning_rate": 4.355857725344224e-05, | |
| "loss": 0.3607, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 3.23, | |
| "grad_norm": 4.917665481567383, | |
| "learning_rate": 4.290650274366017e-05, | |
| "loss": 0.2077, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "grad_norm": 8.88060188293457, | |
| "learning_rate": 4.238484313583451e-05, | |
| "loss": 0.6496, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 3.44, | |
| "grad_norm": 1.9647566080093384, | |
| "learning_rate": 4.1732768626052447e-05, | |
| "loss": 0.1394, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 3.54, | |
| "grad_norm": 50.96482849121094, | |
| "learning_rate": 4.1080694116270374e-05, | |
| "loss": 0.2834, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 3.65, | |
| "grad_norm": 132.5663299560547, | |
| "learning_rate": 4.042861960648831e-05, | |
| "loss": 0.485, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 3.75, | |
| "grad_norm": 3.1624388694763184, | |
| "learning_rate": 3.977654509670624e-05, | |
| "loss": 0.2353, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 3.85, | |
| "grad_norm": 0.5671400427818298, | |
| "learning_rate": 3.912447058692417e-05, | |
| "loss": 0.3058, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 3.96, | |
| "grad_norm": 23.86330223083496, | |
| "learning_rate": 3.84723960771421e-05, | |
| "loss": 0.3764, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_f1": 0.7500000000000001, | |
| "eval_loss": 1.1437296867370605, | |
| "eval_runtime": 1.4019, | |
| "eval_samples_per_second": 45.651, | |
| "eval_steps_per_second": 5.706, | |
| "step": 192 | |
| }, | |
| { | |
| "epoch": 4.06, | |
| "grad_norm": 0.443531334400177, | |
| "learning_rate": 3.782032156736003e-05, | |
| "loss": 0.2652, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 4.17, | |
| "grad_norm": 5.118373394012451, | |
| "learning_rate": 3.716824705757796e-05, | |
| "loss": 0.3585, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 4.27, | |
| "grad_norm": 0.7104140520095825, | |
| "learning_rate": 3.651617254779589e-05, | |
| "loss": 0.2533, | |
| "step": 205 | |
| }, | |
| { | |
| "epoch": 4.38, | |
| "grad_norm": Infinity, | |
| "learning_rate": 3.599451293997023e-05, | |
| "loss": 0.1963, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 4.48, | |
| "grad_norm": 1.0030689239501953, | |
| "learning_rate": 3.5342438430188165e-05, | |
| "loss": 0.0602, | |
| "step": 215 | |
| }, | |
| { | |
| "epoch": 4.58, | |
| "grad_norm": 193.33819580078125, | |
| "learning_rate": 3.469036392040609e-05, | |
| "loss": 0.2754, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 4.69, | |
| "grad_norm": 1.3048127889633179, | |
| "learning_rate": 3.403828941062403e-05, | |
| "loss": 0.1878, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 4.79, | |
| "grad_norm": 0.18795377016067505, | |
| "learning_rate": 3.3386214900841956e-05, | |
| "loss": 0.0245, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 4.9, | |
| "grad_norm": 0.7579631805419922, | |
| "learning_rate": 3.2864555293016304e-05, | |
| "loss": 0.3502, | |
| "step": 235 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "grad_norm": 0.39707404375076294, | |
| "learning_rate": 3.221248078323424e-05, | |
| "loss": 0.2378, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_f1": 0.7865168539325843, | |
| "eval_loss": 1.2380468845367432, | |
| "eval_runtime": 1.3627, | |
| "eval_samples_per_second": 46.967, | |
| "eval_steps_per_second": 5.871, | |
| "step": 240 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 480, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 10, | |
| "save_steps": 500, | |
| "total_flos": 3654362860415712.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": { | |
| "learning_rate": 5.6339237645170805e-05, | |
| "per_device_train_batch_size": 4 | |
| } | |
| } | |