{ "best_metric": 0.9559725730783111, "best_model_checkpoint": "violation-classification-bantai-vit-v80ep/checkpoint-8000", "epoch": 79.99753086419753, "global_step": 8080, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 4.95, "learning_rate": 3.094059405940594e-05, "loss": 0.797, "step": 500 }, { "epoch": 4.95, "eval_accuracy": 0.8715265247203176, "eval_loss": 0.392585426568985, "eval_runtime": 63.7974, "eval_samples_per_second": 86.869, "eval_steps_per_second": 2.727, "step": 500 }, { "epoch": 9.9, "learning_rate": 4.867986798679868e-05, "loss": 0.3095, "step": 1000 }, { "epoch": 9.9, "eval_accuracy": 0.9106820642367376, "eval_loss": 0.2596624493598938, "eval_runtime": 63.8682, "eval_samples_per_second": 86.772, "eval_steps_per_second": 2.724, "step": 1000 }, { "epoch": 14.85, "learning_rate": 4.5242024202420245e-05, "loss": 0.1726, "step": 1500 }, { "epoch": 14.85, "eval_accuracy": 0.9252977264525442, "eval_loss": 0.2156512439250946, "eval_runtime": 63.7151, "eval_samples_per_second": 86.981, "eval_steps_per_second": 2.731, "step": 1500 }, { "epoch": 19.8, "learning_rate": 4.18041804180418e-05, "loss": 0.1259, "step": 2000 }, { "epoch": 19.8, "eval_accuracy": 0.939191627571274, "eval_loss": 0.1869734823703766, "eval_runtime": 63.965, "eval_samples_per_second": 86.641, "eval_steps_per_second": 2.72, "step": 2000 }, { "epoch": 24.75, "learning_rate": 3.8366336633663367e-05, "loss": 0.0959, "step": 2500 }, { "epoch": 24.75, "eval_accuracy": 0.9444243955250812, "eval_loss": 0.17974421381950378, "eval_runtime": 63.8189, "eval_samples_per_second": 86.84, "eval_steps_per_second": 2.726, "step": 2500 }, { "epoch": 29.7, "learning_rate": 3.492849284928493e-05, "loss": 0.0835, "step": 3000 }, { "epoch": 29.7, "eval_accuracy": 0.9354023818116204, "eval_loss": 0.22931958734989166, "eval_runtime": 63.8202, "eval_samples_per_second": 86.838, "eval_steps_per_second": 2.726, "step": 3000 }, { "epoch": 34.65, "learning_rate": 3.149064906490649e-05, "loss": 0.0722, "step": 3500 }, { "epoch": 34.65, "eval_accuracy": 0.9440635149765427, "eval_loss": 0.19207924604415894, "eval_runtime": 63.813, "eval_samples_per_second": 86.848, "eval_steps_per_second": 2.727, "step": 3500 }, { "epoch": 39.6, "learning_rate": 2.8052805280528056e-05, "loss": 0.0628, "step": 4000 }, { "epoch": 39.6, "eval_accuracy": 0.9491158426560808, "eval_loss": 0.18968293070793152, "eval_runtime": 63.6858, "eval_samples_per_second": 87.021, "eval_steps_per_second": 2.732, "step": 4000 }, { "epoch": 44.55, "learning_rate": 2.4614961496149617e-05, "loss": 0.059, "step": 4500 }, { "epoch": 44.55, "eval_accuracy": 0.9520028870443883, "eval_loss": 0.1719067543745041, "eval_runtime": 63.8275, "eval_samples_per_second": 86.828, "eval_steps_per_second": 2.726, "step": 4500 }, { "epoch": 49.5, "learning_rate": 2.1177117711771178e-05, "loss": 0.0531, "step": 5000 }, { "epoch": 49.5, "eval_accuracy": 0.9512811259473114, "eval_loss": 0.19867293536663055, "eval_runtime": 63.7873, "eval_samples_per_second": 86.882, "eval_steps_per_second": 2.728, "step": 5000 }, { "epoch": 54.45, "learning_rate": 1.7739273927392743e-05, "loss": 0.046, "step": 5500 }, { "epoch": 54.45, "eval_accuracy": 0.9556116925297726, "eval_loss": 0.17134784162044525, "eval_runtime": 63.8123, "eval_samples_per_second": 86.848, "eval_steps_per_second": 2.727, "step": 5500 }, { "epoch": 59.4, "learning_rate": 1.4301430143014302e-05, "loss": 0.0444, "step": 6000 }, { "epoch": 59.4, "eval_accuracy": 0.952544207867196, "eval_loss": 0.2016247808933258, "eval_runtime": 63.9341, "eval_samples_per_second": 86.683, "eval_steps_per_second": 2.722, "step": 6000 }, { "epoch": 64.36, "learning_rate": 1.0863586358635865e-05, "loss": 0.042, "step": 6500 }, { "epoch": 64.36, "eval_accuracy": 0.952544207867196, "eval_loss": 0.19500622153282166, "eval_runtime": 63.6967, "eval_samples_per_second": 87.006, "eval_steps_per_second": 2.732, "step": 6500 }, { "epoch": 69.31, "learning_rate": 7.4257425742574256e-06, "loss": 0.0363, "step": 7000 }, { "epoch": 69.31, "eval_accuracy": 0.9548899314326957, "eval_loss": 0.20167508721351624, "eval_runtime": 63.6587, "eval_samples_per_second": 87.058, "eval_steps_per_second": 2.733, "step": 7000 }, { "epoch": 74.26, "learning_rate": 3.987898789878988e-06, "loss": 0.037, "step": 7500 }, { "epoch": 74.26, "eval_accuracy": 0.955070371706965, "eval_loss": 0.1943303495645523, "eval_runtime": 63.8065, "eval_samples_per_second": 86.856, "eval_steps_per_second": 2.727, "step": 7500 }, { "epoch": 79.21, "learning_rate": 5.500550055005501e-07, "loss": 0.0343, "step": 8000 }, { "epoch": 79.21, "eval_accuracy": 0.9559725730783111, "eval_loss": 0.19736222922801971, "eval_runtime": 63.8534, "eval_samples_per_second": 86.793, "eval_steps_per_second": 2.725, "step": 8000 }, { "epoch": 80.0, "step": 8080, "total_flos": 8.01589959270506e+19, "train_loss": 0.12853888580114534, "train_runtime": 24615.3911, "train_samples_per_second": 42.022, "train_steps_per_second": 0.328 } ], "max_steps": 8080, "num_train_epochs": 80, "total_flos": 8.01589959270506e+19, "trial_name": null, "trial_params": null }