| { | |
| "best_metric": 0.8165137614678899, | |
| "best_model_checkpoint": "swiftformer-xs-dmae-va-U-40/checkpoint-232", | |
| "epoch": 36.12903225806452, | |
| "eval_steps": 500, | |
| "global_step": 280, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.9, | |
| "eval_accuracy": 0.3211009174311927, | |
| "eval_loss": 1.3882640600204468, | |
| "eval_runtime": 0.5018, | |
| "eval_samples_per_second": 217.232, | |
| "eval_steps_per_second": 7.972, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 1.29, | |
| "learning_rate": 1.785714285714286e-05, | |
| "loss": 1.4011, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 1.94, | |
| "eval_accuracy": 0.3577981651376147, | |
| "eval_loss": 1.3383492231369019, | |
| "eval_runtime": 0.5683, | |
| "eval_samples_per_second": 191.814, | |
| "eval_steps_per_second": 7.039, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 2.58, | |
| "learning_rate": 3.571428571428572e-05, | |
| "loss": 1.3646, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "eval_accuracy": 0.44036697247706424, | |
| "eval_loss": 1.280238151550293, | |
| "eval_runtime": 0.4561, | |
| "eval_samples_per_second": 238.958, | |
| "eval_steps_per_second": 8.769, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 3.87, | |
| "learning_rate": 4.960317460317461e-05, | |
| "loss": 1.315, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.44954128440366975, | |
| "eval_loss": 1.2193504571914673, | |
| "eval_runtime": 0.8508, | |
| "eval_samples_per_second": 128.117, | |
| "eval_steps_per_second": 4.702, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 4.9, | |
| "eval_accuracy": 0.5229357798165137, | |
| "eval_loss": 1.1717596054077148, | |
| "eval_runtime": 0.4255, | |
| "eval_samples_per_second": 256.17, | |
| "eval_steps_per_second": 9.401, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 5.16, | |
| "learning_rate": 4.761904761904762e-05, | |
| "loss": 1.2634, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 5.94, | |
| "eval_accuracy": 0.5045871559633027, | |
| "eval_loss": 1.127877950668335, | |
| "eval_runtime": 0.4475, | |
| "eval_samples_per_second": 243.601, | |
| "eval_steps_per_second": 8.939, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 6.45, | |
| "learning_rate": 4.563492063492064e-05, | |
| "loss": 1.1949, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 6.97, | |
| "eval_accuracy": 0.5871559633027523, | |
| "eval_loss": 1.0761113166809082, | |
| "eval_runtime": 0.5058, | |
| "eval_samples_per_second": 215.497, | |
| "eval_steps_per_second": 7.908, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 7.74, | |
| "learning_rate": 4.3650793650793655e-05, | |
| "loss": 1.1136, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.6330275229357798, | |
| "eval_loss": 1.0223767757415771, | |
| "eval_runtime": 0.4226, | |
| "eval_samples_per_second": 257.914, | |
| "eval_steps_per_second": 9.465, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 8.9, | |
| "eval_accuracy": 0.6238532110091743, | |
| "eval_loss": 0.9975973963737488, | |
| "eval_runtime": 0.556, | |
| "eval_samples_per_second": 196.055, | |
| "eval_steps_per_second": 7.195, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 9.03, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 1.0824, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 9.94, | |
| "eval_accuracy": 0.6605504587155964, | |
| "eval_loss": 0.9517724514007568, | |
| "eval_runtime": 0.4569, | |
| "eval_samples_per_second": 238.566, | |
| "eval_steps_per_second": 8.755, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 10.32, | |
| "learning_rate": 3.968253968253968e-05, | |
| "loss": 1.0212, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 10.97, | |
| "eval_accuracy": 0.6697247706422018, | |
| "eval_loss": 0.9116750359535217, | |
| "eval_runtime": 0.4487, | |
| "eval_samples_per_second": 242.931, | |
| "eval_steps_per_second": 8.915, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 11.61, | |
| "learning_rate": 3.76984126984127e-05, | |
| "loss": 0.9566, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "eval_accuracy": 0.6880733944954128, | |
| "eval_loss": 0.8973050117492676, | |
| "eval_runtime": 0.448, | |
| "eval_samples_per_second": 243.303, | |
| "eval_steps_per_second": 8.929, | |
| "step": 93 | |
| }, | |
| { | |
| "epoch": 12.9, | |
| "learning_rate": 3.571428571428572e-05, | |
| "loss": 0.935, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 12.9, | |
| "eval_accuracy": 0.7064220183486238, | |
| "eval_loss": 0.8704947233200073, | |
| "eval_runtime": 0.4696, | |
| "eval_samples_per_second": 232.132, | |
| "eval_steps_per_second": 8.519, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 13.94, | |
| "eval_accuracy": 0.7155963302752294, | |
| "eval_loss": 0.8559499382972717, | |
| "eval_runtime": 0.5932, | |
| "eval_samples_per_second": 183.74, | |
| "eval_steps_per_second": 6.743, | |
| "step": 108 | |
| }, | |
| { | |
| "epoch": 14.19, | |
| "learning_rate": 3.3730158730158734e-05, | |
| "loss": 0.8826, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 14.97, | |
| "eval_accuracy": 0.7155963302752294, | |
| "eval_loss": 0.8371049165725708, | |
| "eval_runtime": 0.4298, | |
| "eval_samples_per_second": 253.624, | |
| "eval_steps_per_second": 9.307, | |
| "step": 116 | |
| }, | |
| { | |
| "epoch": 15.48, | |
| "learning_rate": 3.1746031746031745e-05, | |
| "loss": 0.8688, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 16.0, | |
| "eval_accuracy": 0.7155963302752294, | |
| "eval_loss": 0.8251588940620422, | |
| "eval_runtime": 0.4656, | |
| "eval_samples_per_second": 234.09, | |
| "eval_steps_per_second": 8.59, | |
| "step": 124 | |
| }, | |
| { | |
| "epoch": 16.77, | |
| "learning_rate": 2.9761904761904762e-05, | |
| "loss": 0.8436, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 16.9, | |
| "eval_accuracy": 0.6972477064220184, | |
| "eval_loss": 0.8211333155632019, | |
| "eval_runtime": 0.4255, | |
| "eval_samples_per_second": 256.176, | |
| "eval_steps_per_second": 9.401, | |
| "step": 131 | |
| }, | |
| { | |
| "epoch": 17.94, | |
| "eval_accuracy": 0.7339449541284404, | |
| "eval_loss": 0.8039615750312805, | |
| "eval_runtime": 0.4375, | |
| "eval_samples_per_second": 249.151, | |
| "eval_steps_per_second": 9.143, | |
| "step": 139 | |
| }, | |
| { | |
| "epoch": 18.06, | |
| "learning_rate": 2.777777777777778e-05, | |
| "loss": 0.8155, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 18.97, | |
| "eval_accuracy": 0.7431192660550459, | |
| "eval_loss": 0.7625071406364441, | |
| "eval_runtime": 0.435, | |
| "eval_samples_per_second": 250.55, | |
| "eval_steps_per_second": 9.195, | |
| "step": 147 | |
| }, | |
| { | |
| "epoch": 19.35, | |
| "learning_rate": 2.5793650793650796e-05, | |
| "loss": 0.7831, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "eval_accuracy": 0.7431192660550459, | |
| "eval_loss": 0.7451765537261963, | |
| "eval_runtime": 0.4363, | |
| "eval_samples_per_second": 249.828, | |
| "eval_steps_per_second": 9.168, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 20.65, | |
| "learning_rate": 2.380952380952381e-05, | |
| "loss": 0.7826, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 20.9, | |
| "eval_accuracy": 0.7431192660550459, | |
| "eval_loss": 0.7278565764427185, | |
| "eval_runtime": 0.5078, | |
| "eval_samples_per_second": 214.643, | |
| "eval_steps_per_second": 7.877, | |
| "step": 162 | |
| }, | |
| { | |
| "epoch": 21.94, | |
| "learning_rate": 2.1825396825396827e-05, | |
| "loss": 0.7499, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 21.94, | |
| "eval_accuracy": 0.7431192660550459, | |
| "eval_loss": 0.714833676815033, | |
| "eval_runtime": 0.4399, | |
| "eval_samples_per_second": 247.796, | |
| "eval_steps_per_second": 9.093, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 22.97, | |
| "eval_accuracy": 0.7522935779816514, | |
| "eval_loss": 0.7061465382575989, | |
| "eval_runtime": 0.5513, | |
| "eval_samples_per_second": 197.698, | |
| "eval_steps_per_second": 7.255, | |
| "step": 178 | |
| }, | |
| { | |
| "epoch": 23.23, | |
| "learning_rate": 1.984126984126984e-05, | |
| "loss": 0.7539, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 24.0, | |
| "eval_accuracy": 0.7522935779816514, | |
| "eval_loss": 0.7026045918464661, | |
| "eval_runtime": 0.4348, | |
| "eval_samples_per_second": 250.669, | |
| "eval_steps_per_second": 9.199, | |
| "step": 186 | |
| }, | |
| { | |
| "epoch": 24.52, | |
| "learning_rate": 1.785714285714286e-05, | |
| "loss": 0.7453, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 24.9, | |
| "eval_accuracy": 0.7889908256880734, | |
| "eval_loss": 0.6818734407424927, | |
| "eval_runtime": 0.4308, | |
| "eval_samples_per_second": 253.004, | |
| "eval_steps_per_second": 9.285, | |
| "step": 193 | |
| }, | |
| { | |
| "epoch": 25.81, | |
| "learning_rate": 1.5873015873015872e-05, | |
| "loss": 0.7174, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 25.94, | |
| "eval_accuracy": 0.7706422018348624, | |
| "eval_loss": 0.6837214231491089, | |
| "eval_runtime": 0.4357, | |
| "eval_samples_per_second": 250.193, | |
| "eval_steps_per_second": 9.181, | |
| "step": 201 | |
| }, | |
| { | |
| "epoch": 26.97, | |
| "eval_accuracy": 0.7798165137614679, | |
| "eval_loss": 0.6743292212486267, | |
| "eval_runtime": 0.4526, | |
| "eval_samples_per_second": 240.854, | |
| "eval_steps_per_second": 8.839, | |
| "step": 209 | |
| }, | |
| { | |
| "epoch": 27.1, | |
| "learning_rate": 1.388888888888889e-05, | |
| "loss": 0.7083, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 28.0, | |
| "eval_accuracy": 0.7798165137614679, | |
| "eval_loss": 0.6706274151802063, | |
| "eval_runtime": 0.4549, | |
| "eval_samples_per_second": 239.637, | |
| "eval_steps_per_second": 8.794, | |
| "step": 217 | |
| }, | |
| { | |
| "epoch": 28.39, | |
| "learning_rate": 1.1904761904761905e-05, | |
| "loss": 0.6813, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 28.9, | |
| "eval_accuracy": 0.8073394495412844, | |
| "eval_loss": 0.6643755435943604, | |
| "eval_runtime": 0.4318, | |
| "eval_samples_per_second": 252.45, | |
| "eval_steps_per_second": 9.264, | |
| "step": 224 | |
| }, | |
| { | |
| "epoch": 29.68, | |
| "learning_rate": 9.92063492063492e-06, | |
| "loss": 0.7107, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 29.94, | |
| "eval_accuracy": 0.8165137614678899, | |
| "eval_loss": 0.6422722935676575, | |
| "eval_runtime": 0.5633, | |
| "eval_samples_per_second": 193.507, | |
| "eval_steps_per_second": 7.101, | |
| "step": 232 | |
| }, | |
| { | |
| "epoch": 30.97, | |
| "learning_rate": 7.936507936507936e-06, | |
| "loss": 0.6912, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 30.97, | |
| "eval_accuracy": 0.7889908256880734, | |
| "eval_loss": 0.6418679356575012, | |
| "eval_runtime": 0.4378, | |
| "eval_samples_per_second": 248.984, | |
| "eval_steps_per_second": 9.137, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 32.0, | |
| "eval_accuracy": 0.7889908256880734, | |
| "eval_loss": 0.6465409994125366, | |
| "eval_runtime": 0.4513, | |
| "eval_samples_per_second": 241.531, | |
| "eval_steps_per_second": 8.864, | |
| "step": 248 | |
| }, | |
| { | |
| "epoch": 32.26, | |
| "learning_rate": 5.9523809523809525e-06, | |
| "loss": 0.7031, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 32.9, | |
| "eval_accuracy": 0.8073394495412844, | |
| "eval_loss": 0.6346263289451599, | |
| "eval_runtime": 0.4298, | |
| "eval_samples_per_second": 253.596, | |
| "eval_steps_per_second": 9.306, | |
| "step": 255 | |
| }, | |
| { | |
| "epoch": 33.55, | |
| "learning_rate": 3.968253968253968e-06, | |
| "loss": 0.6647, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 33.94, | |
| "eval_accuracy": 0.8073394495412844, | |
| "eval_loss": 0.6346942186355591, | |
| "eval_runtime": 0.4204, | |
| "eval_samples_per_second": 259.297, | |
| "eval_steps_per_second": 9.515, | |
| "step": 263 | |
| }, | |
| { | |
| "epoch": 34.84, | |
| "learning_rate": 1.984126984126984e-06, | |
| "loss": 0.6799, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 34.97, | |
| "eval_accuracy": 0.7981651376146789, | |
| "eval_loss": 0.6475719213485718, | |
| "eval_runtime": 0.5807, | |
| "eval_samples_per_second": 187.701, | |
| "eval_steps_per_second": 6.888, | |
| "step": 271 | |
| }, | |
| { | |
| "epoch": 36.0, | |
| "eval_accuracy": 0.7981651376146789, | |
| "eval_loss": 0.6428852081298828, | |
| "eval_runtime": 0.4339, | |
| "eval_samples_per_second": 251.222, | |
| "eval_steps_per_second": 9.219, | |
| "step": 279 | |
| }, | |
| { | |
| "epoch": 36.13, | |
| "learning_rate": 0.0, | |
| "loss": 0.6774, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 36.13, | |
| "eval_accuracy": 0.7889908256880734, | |
| "eval_loss": 0.6517751812934875, | |
| "eval_runtime": 0.4506, | |
| "eval_samples_per_second": 241.9, | |
| "eval_steps_per_second": 8.877, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 36.13, | |
| "step": 280, | |
| "total_flos": 9.686412043576934e+16, | |
| "train_loss": 0.8966831156185695, | |
| "train_runtime": 256.9851, | |
| "train_samples_per_second": 152.227, | |
| "train_steps_per_second": 1.09 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 280, | |
| "num_train_epochs": 40, | |
| "save_steps": 500, | |
| "total_flos": 9.686412043576934e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |