{ "best_metric": 0.5736537575721741, "best_model_checkpoint": "./vit-large-augmented-ph2-patch-32/checkpoint-150", "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.4272, "step": 10 }, { "epoch": 0.14, "learning_rate": 0.0001927536231884058, "loss": 0.1337, "step": 20 }, { "epoch": 0.22, "learning_rate": 0.0001891304347826087, "loss": 0.1272, "step": 30 }, { "epoch": 0.29, "learning_rate": 0.0001855072463768116, "loss": 0.0933, "step": 40 }, { "epoch": 0.36, "learning_rate": 0.0001818840579710145, "loss": 0.0405, "step": 50 }, { "epoch": 0.36, "eval_accuracy": 0.8341880341880342, "eval_f1": 0.8341880341880342, "eval_loss": 0.6853269338607788, "eval_precision": 0.8341880341880342, "eval_recall": 0.8341880341880342, "eval_runtime": 11.7272, "eval_samples_per_second": 49.884, "eval_steps_per_second": 6.31, "step": 50 }, { "epoch": 0.43, "learning_rate": 0.0001782608695652174, "loss": 0.1968, "step": 60 }, { "epoch": 0.51, "learning_rate": 0.0001746376811594203, "loss": 0.0372, "step": 70 }, { "epoch": 0.58, "learning_rate": 0.0001710144927536232, "loss": 0.0156, "step": 80 }, { "epoch": 0.65, "learning_rate": 0.0001673913043478261, "loss": 0.0547, "step": 90 }, { "epoch": 0.72, "learning_rate": 0.000163768115942029, "loss": 0.0107, "step": 100 }, { "epoch": 0.72, "eval_accuracy": 0.8256410256410256, "eval_f1": 0.8256410256410256, "eval_loss": 0.8199310302734375, "eval_precision": 0.8256410256410256, "eval_recall": 0.8256410256410256, "eval_runtime": 11.7833, "eval_samples_per_second": 49.646, "eval_steps_per_second": 6.28, "step": 100 }, { "epoch": 0.8, "learning_rate": 0.00016014492753623188, "loss": 0.2263, "step": 110 }, { "epoch": 0.87, "learning_rate": 0.0001565217391304348, "loss": 0.0332, "step": 120 }, { "epoch": 0.94, "learning_rate": 0.00015289855072463769, "loss": 0.048, "step": 130 }, { "epoch": 1.01, "learning_rate": 0.00014927536231884058, "loss": 0.0626, "step": 140 }, { "epoch": 1.09, "learning_rate": 0.0001456521739130435, "loss": 0.0338, "step": 150 }, { "epoch": 1.09, "eval_accuracy": 0.8700854700854701, "eval_f1": 0.8700854700854701, "eval_loss": 0.5736537575721741, "eval_precision": 0.8700854700854701, "eval_recall": 0.8700854700854701, "eval_runtime": 12.1612, "eval_samples_per_second": 48.104, "eval_steps_per_second": 6.085, "step": 150 }, { "epoch": 1.16, "learning_rate": 0.00014202898550724638, "loss": 0.0042, "step": 160 }, { "epoch": 1.23, "learning_rate": 0.00013840579710144927, "loss": 0.0047, "step": 170 }, { "epoch": 1.3, "learning_rate": 0.0001347826086956522, "loss": 0.003, "step": 180 }, { "epoch": 1.38, "learning_rate": 0.00013115942028985508, "loss": 0.003, "step": 190 }, { "epoch": 1.45, "learning_rate": 0.00012753623188405797, "loss": 0.0026, "step": 200 }, { "epoch": 1.45, "eval_accuracy": 0.8683760683760684, "eval_f1": 0.8683760683760683, "eval_loss": 0.6007649302482605, "eval_precision": 0.8683760683760684, "eval_recall": 0.8683760683760684, "eval_runtime": 11.7792, "eval_samples_per_second": 49.664, "eval_steps_per_second": 6.282, "step": 200 }, { "epoch": 1.52, "learning_rate": 0.00012391304347826086, "loss": 0.0024, "step": 210 }, { "epoch": 1.59, "learning_rate": 0.00012028985507246378, "loss": 0.0022, "step": 220 }, { "epoch": 1.67, "learning_rate": 0.00011666666666666668, "loss": 0.0022, "step": 230 }, { "epoch": 1.74, "learning_rate": 0.00011304347826086956, "loss": 0.002, "step": 240 }, { "epoch": 1.81, "learning_rate": 0.00010942028985507247, "loss": 0.0019, "step": 250 }, { "epoch": 1.81, "eval_accuracy": 0.8735042735042735, "eval_f1": 0.8735042735042735, "eval_loss": 0.6275166273117065, "eval_precision": 0.8735042735042735, "eval_recall": 0.8735042735042735, "eval_runtime": 11.7834, "eval_samples_per_second": 49.646, "eval_steps_per_second": 6.28, "step": 250 }, { "epoch": 1.88, "learning_rate": 0.00010579710144927538, "loss": 0.0019, "step": 260 }, { "epoch": 1.96, "learning_rate": 0.00010217391304347828, "loss": 0.0017, "step": 270 }, { "epoch": 2.03, "learning_rate": 9.855072463768117e-05, "loss": 0.0017, "step": 280 }, { "epoch": 2.1, "learning_rate": 9.492753623188406e-05, "loss": 0.0016, "step": 290 }, { "epoch": 2.17, "learning_rate": 9.130434782608696e-05, "loss": 0.0016, "step": 300 }, { "epoch": 2.17, "eval_accuracy": 0.8735042735042735, "eval_f1": 0.8735042735042735, "eval_loss": 0.6487508416175842, "eval_precision": 0.8735042735042735, "eval_recall": 0.8735042735042735, "eval_runtime": 11.9506, "eval_samples_per_second": 48.951, "eval_steps_per_second": 6.192, "step": 300 }, { "epoch": 2.25, "learning_rate": 8.768115942028987e-05, "loss": 0.0015, "step": 310 }, { "epoch": 2.32, "learning_rate": 8.405797101449276e-05, "loss": 0.0014, "step": 320 }, { "epoch": 2.39, "learning_rate": 8.043478260869566e-05, "loss": 0.0014, "step": 330 }, { "epoch": 2.46, "learning_rate": 7.681159420289855e-05, "loss": 0.0014, "step": 340 }, { "epoch": 2.54, "learning_rate": 7.318840579710145e-05, "loss": 0.0013, "step": 350 }, { "epoch": 2.54, "eval_accuracy": 0.8752136752136752, "eval_f1": 0.8752136752136753, "eval_loss": 0.6639007925987244, "eval_precision": 0.8752136752136752, "eval_recall": 0.8752136752136752, "eval_runtime": 12.3145, "eval_samples_per_second": 47.505, "eval_steps_per_second": 6.009, "step": 350 }, { "epoch": 2.61, "learning_rate": 6.956521739130436e-05, "loss": 0.0013, "step": 360 }, { "epoch": 2.68, "learning_rate": 6.594202898550726e-05, "loss": 0.0013, "step": 370 }, { "epoch": 2.75, "learning_rate": 6.231884057971015e-05, "loss": 0.0013, "step": 380 }, { "epoch": 2.83, "learning_rate": 5.869565217391305e-05, "loss": 0.0012, "step": 390 }, { "epoch": 2.9, "learning_rate": 5.507246376811594e-05, "loss": 0.0012, "step": 400 }, { "epoch": 2.9, "eval_accuracy": 0.8752136752136752, "eval_f1": 0.8752136752136753, "eval_loss": 0.6756650805473328, "eval_precision": 0.8752136752136752, "eval_recall": 0.8752136752136752, "eval_runtime": 11.7227, "eval_samples_per_second": 49.903, "eval_steps_per_second": 6.313, "step": 400 }, { "epoch": 2.97, "learning_rate": 5.144927536231884e-05, "loss": 0.0012, "step": 410 }, { "epoch": 3.04, "learning_rate": 4.782608695652174e-05, "loss": 0.0012, "step": 420 }, { "epoch": 3.12, "learning_rate": 4.4202898550724645e-05, "loss": 0.0011, "step": 430 }, { "epoch": 3.19, "learning_rate": 4.057971014492754e-05, "loss": 0.0012, "step": 440 }, { "epoch": 3.26, "learning_rate": 3.695652173913043e-05, "loss": 0.0011, "step": 450 }, { "epoch": 3.26, "eval_accuracy": 0.8735042735042735, "eval_f1": 0.8735042735042735, "eval_loss": 0.6844302415847778, "eval_precision": 0.8735042735042735, "eval_recall": 0.8735042735042735, "eval_runtime": 11.7458, "eval_samples_per_second": 49.805, "eval_steps_per_second": 6.3, "step": 450 }, { "epoch": 3.33, "learning_rate": 3.3333333333333335e-05, "loss": 0.0011, "step": 460 }, { "epoch": 3.41, "learning_rate": 2.971014492753623e-05, "loss": 0.0011, "step": 470 }, { "epoch": 3.48, "learning_rate": 2.608695652173913e-05, "loss": 0.0011, "step": 480 }, { "epoch": 3.55, "learning_rate": 2.246376811594203e-05, "loss": 0.0011, "step": 490 }, { "epoch": 3.62, "learning_rate": 1.8840579710144928e-05, "loss": 0.001, "step": 500 }, { "epoch": 3.62, "eval_accuracy": 0.8735042735042735, "eval_f1": 0.8735042735042735, "eval_loss": 0.6895482540130615, "eval_precision": 0.8735042735042735, "eval_recall": 0.8735042735042735, "eval_runtime": 11.5982, "eval_samples_per_second": 50.439, "eval_steps_per_second": 6.38, "step": 500 }, { "epoch": 3.7, "learning_rate": 1.5217391304347828e-05, "loss": 0.001, "step": 510 }, { "epoch": 3.77, "learning_rate": 1.1594202898550725e-05, "loss": 0.001, "step": 520 }, { "epoch": 3.84, "learning_rate": 7.971014492753623e-06, "loss": 0.001, "step": 530 }, { "epoch": 3.91, "learning_rate": 4.347826086956522e-06, "loss": 0.001, "step": 540 }, { "epoch": 3.99, "learning_rate": 7.246376811594203e-07, "loss": 0.001, "step": 550 }, { "epoch": 3.99, "eval_accuracy": 0.8735042735042735, "eval_f1": 0.8735042735042735, "eval_loss": 0.6913025379180908, "eval_precision": 0.8735042735042735, "eval_recall": 0.8735042735042735, "eval_runtime": 12.0914, "eval_samples_per_second": 48.382, "eval_steps_per_second": 6.12, "step": 550 }, { "epoch": 4.0, "step": 552, "total_flos": 2.42817580228608e+18, "train_loss": 0.02911639397127041, "train_runtime": 866.2055, "train_samples_per_second": 10.159, "train_steps_per_second": 0.637 } ], "max_steps": 552, "num_train_epochs": 4, "total_flos": 2.42817580228608e+18, "trial_name": null, "trial_params": null }