{ "best_metric": 0.5419731140136719, "best_model_checkpoint": "./vit-base-augmented-ph2-patch-16/checkpoint-100", "epoch": 4.0, "global_step": 552, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.07, "learning_rate": 0.00019637681159420293, "loss": 0.4672, "step": 10 }, { "epoch": 0.14, "learning_rate": 0.0001927536231884058, "loss": 0.1827, "step": 20 }, { "epoch": 0.22, "learning_rate": 0.0001891304347826087, "loss": 0.1752, "step": 30 }, { "epoch": 0.29, "learning_rate": 0.0001855072463768116, "loss": 0.0594, "step": 40 }, { "epoch": 0.36, "learning_rate": 0.0001818840579710145, "loss": 0.0592, "step": 50 }, { "epoch": 0.36, "eval_accuracy": 0.8068376068376069, "eval_f1": 0.8068376068376067, "eval_loss": 0.7161407470703125, "eval_precision": 0.8068376068376069, "eval_recall": 0.8068376068376069, "eval_runtime": 10.6744, "eval_samples_per_second": 54.804, "eval_steps_per_second": 6.932, "step": 50 }, { "epoch": 0.43, "learning_rate": 0.0001782608695652174, "loss": 0.111, "step": 60 }, { "epoch": 0.51, "learning_rate": 0.0001746376811594203, "loss": 0.0664, "step": 70 }, { "epoch": 0.58, "learning_rate": 0.0001710144927536232, "loss": 0.084, "step": 80 }, { "epoch": 0.65, "learning_rate": 0.0001673913043478261, "loss": 0.0786, "step": 90 }, { "epoch": 0.72, "learning_rate": 0.000163768115942029, "loss": 0.0703, "step": 100 }, { "epoch": 0.72, "eval_accuracy": 0.8444444444444444, "eval_f1": 0.8444444444444444, "eval_loss": 0.5419731140136719, "eval_precision": 0.8444444444444444, "eval_recall": 0.8444444444444444, "eval_runtime": 10.53, "eval_samples_per_second": 55.555, "eval_steps_per_second": 7.028, "step": 100 }, { "epoch": 0.8, "learning_rate": 0.00016014492753623188, "loss": 0.0376, "step": 110 }, { "epoch": 0.87, "learning_rate": 0.0001565217391304348, "loss": 0.0069, "step": 120 }, { "epoch": 0.94, "learning_rate": 0.00015289855072463769, "loss": 0.006, "step": 130 }, { "epoch": 1.01, "learning_rate": 0.00014927536231884058, "loss": 0.0048, "step": 140 }, { "epoch": 1.09, "learning_rate": 0.0001456521739130435, "loss": 0.0042, "step": 150 }, { "epoch": 1.09, "eval_accuracy": 0.882051282051282, "eval_f1": 0.882051282051282, "eval_loss": 0.555678129196167, "eval_precision": 0.882051282051282, "eval_recall": 0.882051282051282, "eval_runtime": 10.4446, "eval_samples_per_second": 56.01, "eval_steps_per_second": 7.085, "step": 150 }, { "epoch": 1.16, "learning_rate": 0.00014202898550724638, "loss": 0.0302, "step": 160 }, { "epoch": 1.23, "learning_rate": 0.00013840579710144927, "loss": 0.0757, "step": 170 }, { "epoch": 1.3, "learning_rate": 0.0001347826086956522, "loss": 0.0042, "step": 180 }, { "epoch": 1.38, "learning_rate": 0.00013115942028985508, "loss": 0.0036, "step": 190 }, { "epoch": 1.45, "learning_rate": 0.00012753623188405797, "loss": 0.0034, "step": 200 }, { "epoch": 1.45, "eval_accuracy": 0.8700854700854701, "eval_f1": 0.8700854700854701, "eval_loss": 0.6464130878448486, "eval_precision": 0.8700854700854701, "eval_recall": 0.8700854700854701, "eval_runtime": 10.5631, "eval_samples_per_second": 55.382, "eval_steps_per_second": 7.006, "step": 200 }, { "epoch": 1.52, "learning_rate": 0.00012391304347826086, "loss": 0.0029, "step": 210 }, { "epoch": 1.59, "learning_rate": 0.00012028985507246378, "loss": 0.003, "step": 220 }, { "epoch": 1.67, "learning_rate": 0.00011666666666666668, "loss": 0.0026, "step": 230 }, { "epoch": 1.74, "learning_rate": 0.00011304347826086956, "loss": 0.0025, "step": 240 }, { "epoch": 1.81, "learning_rate": 0.00010942028985507247, "loss": 0.0023, "step": 250 }, { "epoch": 1.81, "eval_accuracy": 0.841025641025641, "eval_f1": 0.841025641025641, "eval_loss": 0.7943088412284851, "eval_precision": 0.841025641025641, "eval_recall": 0.841025641025641, "eval_runtime": 10.5827, "eval_samples_per_second": 55.279, "eval_steps_per_second": 6.993, "step": 250 }, { "epoch": 1.88, "learning_rate": 0.00010579710144927538, "loss": 0.0022, "step": 260 }, { "epoch": 1.96, "learning_rate": 0.00010217391304347828, "loss": 0.002, "step": 270 }, { "epoch": 2.03, "learning_rate": 9.855072463768117e-05, "loss": 0.002, "step": 280 }, { "epoch": 2.1, "learning_rate": 9.492753623188406e-05, "loss": 0.0019, "step": 290 }, { "epoch": 2.17, "learning_rate": 9.130434782608696e-05, "loss": 0.0018, "step": 300 }, { "epoch": 2.17, "eval_accuracy": 0.8598290598290599, "eval_f1": 0.8598290598290599, "eval_loss": 0.7109088897705078, "eval_precision": 0.8598290598290599, "eval_recall": 0.8598290598290599, "eval_runtime": 10.6322, "eval_samples_per_second": 55.022, "eval_steps_per_second": 6.96, "step": 300 }, { "epoch": 2.25, "learning_rate": 8.768115942028987e-05, "loss": 0.0018, "step": 310 }, { "epoch": 2.32, "learning_rate": 8.405797101449276e-05, "loss": 0.0017, "step": 320 }, { "epoch": 2.39, "learning_rate": 8.043478260869566e-05, "loss": 0.0017, "step": 330 }, { "epoch": 2.46, "learning_rate": 7.681159420289855e-05, "loss": 0.0016, "step": 340 }, { "epoch": 2.54, "learning_rate": 7.318840579710145e-05, "loss": 0.0015, "step": 350 }, { "epoch": 2.54, "eval_accuracy": 0.8598290598290599, "eval_f1": 0.8598290598290599, "eval_loss": 0.7254208922386169, "eval_precision": 0.8598290598290599, "eval_recall": 0.8598290598290599, "eval_runtime": 10.6498, "eval_samples_per_second": 54.93, "eval_steps_per_second": 6.948, "step": 350 }, { "epoch": 2.61, "learning_rate": 6.956521739130436e-05, "loss": 0.0015, "step": 360 }, { "epoch": 2.68, "learning_rate": 6.594202898550726e-05, "loss": 0.0015, "step": 370 }, { "epoch": 2.75, "learning_rate": 6.231884057971015e-05, "loss": 0.0015, "step": 380 }, { "epoch": 2.83, "learning_rate": 5.869565217391305e-05, "loss": 0.0014, "step": 390 }, { "epoch": 2.9, "learning_rate": 5.507246376811594e-05, "loss": 0.0013, "step": 400 }, { "epoch": 2.9, "eval_accuracy": 0.8598290598290599, "eval_f1": 0.8598290598290599, "eval_loss": 0.7364383339881897, "eval_precision": 0.8598290598290599, "eval_recall": 0.8598290598290599, "eval_runtime": 10.5336, "eval_samples_per_second": 55.537, "eval_steps_per_second": 7.025, "step": 400 }, { "epoch": 2.97, "learning_rate": 5.144927536231884e-05, "loss": 0.0014, "step": 410 }, { "epoch": 3.04, "learning_rate": 4.782608695652174e-05, "loss": 0.0013, "step": 420 }, { "epoch": 3.12, "learning_rate": 4.4202898550724645e-05, "loss": 0.0013, "step": 430 }, { "epoch": 3.19, "learning_rate": 4.057971014492754e-05, "loss": 0.0013, "step": 440 }, { "epoch": 3.26, "learning_rate": 3.695652173913043e-05, "loss": 0.0013, "step": 450 }, { "epoch": 3.26, "eval_accuracy": 0.8615384615384616, "eval_f1": 0.8615384615384615, "eval_loss": 0.7438333034515381, "eval_precision": 0.8615384615384616, "eval_recall": 0.8615384615384616, "eval_runtime": 10.477, "eval_samples_per_second": 55.837, "eval_steps_per_second": 7.063, "step": 450 }, { "epoch": 3.33, "learning_rate": 3.3333333333333335e-05, "loss": 0.0013, "step": 460 }, { "epoch": 3.41, "learning_rate": 2.971014492753623e-05, "loss": 0.0013, "step": 470 }, { "epoch": 3.48, "learning_rate": 2.608695652173913e-05, "loss": 0.0012, "step": 480 }, { "epoch": 3.55, "learning_rate": 2.246376811594203e-05, "loss": 0.0012, "step": 490 }, { "epoch": 3.62, "learning_rate": 1.8840579710144928e-05, "loss": 0.0012, "step": 500 }, { "epoch": 3.62, "eval_accuracy": 0.8615384615384616, "eval_f1": 0.8615384615384615, "eval_loss": 0.7488855719566345, "eval_precision": 0.8615384615384616, "eval_recall": 0.8615384615384616, "eval_runtime": 10.4515, "eval_samples_per_second": 55.973, "eval_steps_per_second": 7.08, "step": 500 }, { "epoch": 3.7, "learning_rate": 1.5217391304347828e-05, "loss": 0.0012, "step": 510 }, { "epoch": 3.77, "learning_rate": 1.1594202898550725e-05, "loss": 0.0012, "step": 520 }, { "epoch": 3.84, "learning_rate": 7.971014492753623e-06, "loss": 0.0012, "step": 530 }, { "epoch": 3.91, "learning_rate": 4.347826086956522e-06, "loss": 0.0012, "step": 540 }, { "epoch": 3.99, "learning_rate": 7.246376811594203e-07, "loss": 0.0012, "step": 550 }, { "epoch": 3.99, "eval_accuracy": 0.8615384615384616, "eval_f1": 0.8615384615384615, "eval_loss": 0.7506275773048401, "eval_precision": 0.8615384615384616, "eval_recall": 0.8615384615384616, "eval_runtime": 10.4592, "eval_samples_per_second": 55.932, "eval_steps_per_second": 7.075, "step": 550 }, { "epoch": 4.0, "step": 552, "total_flos": 6.819295086084096e+17, "train_loss": 0.028775763252507084, "train_runtime": 487.8679, "train_samples_per_second": 18.038, "train_steps_per_second": 1.131 } ], "max_steps": 552, "num_train_epochs": 4, "total_flos": 6.819295086084096e+17, "trial_name": null, "trial_params": null }