AykeeSalazar's picture
End of training
1a2c9af
{
"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
}