vit-base-patch16-224-type / trainer_state.json
barten's picture
End of training
3e1cb09
{
"best_metric": 0.7593896713615024,
"best_model_checkpoint": "vit-base-patch16-224-type/checkpoint-560",
"epoch": 9.959839357429718,
"eval_steps": 500,
"global_step": 620,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.16,
"learning_rate": 8.064516129032258e-06,
"loss": 2.3882,
"step": 10
},
{
"epoch": 0.32,
"learning_rate": 1.6129032258064517e-05,
"loss": 2.2523,
"step": 20
},
{
"epoch": 0.48,
"learning_rate": 2.4193548387096777e-05,
"loss": 2.0603,
"step": 30
},
{
"epoch": 0.64,
"learning_rate": 3.2258064516129034e-05,
"loss": 1.8355,
"step": 40
},
{
"epoch": 0.8,
"learning_rate": 4.032258064516129e-05,
"loss": 1.5709,
"step": 50
},
{
"epoch": 0.96,
"learning_rate": 4.8387096774193554e-05,
"loss": 1.3494,
"step": 60
},
{
"epoch": 1.0,
"eval_accuracy": 0.6349765258215962,
"eval_loss": 1.1001132726669312,
"eval_runtime": 22.8557,
"eval_samples_per_second": 74.555,
"eval_steps_per_second": 2.363,
"step": 62
},
{
"epoch": 1.12,
"learning_rate": 4.92831541218638e-05,
"loss": 1.1904,
"step": 70
},
{
"epoch": 1.29,
"learning_rate": 4.8387096774193554e-05,
"loss": 1.139,
"step": 80
},
{
"epoch": 1.45,
"learning_rate": 4.74910394265233e-05,
"loss": 1.0743,
"step": 90
},
{
"epoch": 1.61,
"learning_rate": 4.659498207885305e-05,
"loss": 1.1071,
"step": 100
},
{
"epoch": 1.77,
"learning_rate": 4.56989247311828e-05,
"loss": 1.0113,
"step": 110
},
{
"epoch": 1.93,
"learning_rate": 4.4802867383512545e-05,
"loss": 0.9612,
"step": 120
},
{
"epoch": 1.99,
"eval_accuracy": 0.6772300469483568,
"eval_loss": 0.8708141446113586,
"eval_runtime": 22.7556,
"eval_samples_per_second": 74.883,
"eval_steps_per_second": 2.373,
"step": 124
},
{
"epoch": 2.09,
"learning_rate": 4.390681003584229e-05,
"loss": 0.9621,
"step": 130
},
{
"epoch": 2.25,
"learning_rate": 4.301075268817205e-05,
"loss": 0.9291,
"step": 140
},
{
"epoch": 2.41,
"learning_rate": 4.2114695340501795e-05,
"loss": 0.8999,
"step": 150
},
{
"epoch": 2.57,
"learning_rate": 4.121863799283154e-05,
"loss": 0.9236,
"step": 160
},
{
"epoch": 2.73,
"learning_rate": 4.032258064516129e-05,
"loss": 0.9358,
"step": 170
},
{
"epoch": 2.89,
"learning_rate": 3.9426523297491045e-05,
"loss": 0.8817,
"step": 180
},
{
"epoch": 2.99,
"eval_accuracy": 0.7265258215962441,
"eval_loss": 0.7897992730140686,
"eval_runtime": 23.0974,
"eval_samples_per_second": 73.774,
"eval_steps_per_second": 2.338,
"step": 186
},
{
"epoch": 3.05,
"learning_rate": 3.8530465949820786e-05,
"loss": 0.9056,
"step": 190
},
{
"epoch": 3.21,
"learning_rate": 3.763440860215054e-05,
"loss": 0.8122,
"step": 200
},
{
"epoch": 3.37,
"learning_rate": 3.673835125448029e-05,
"loss": 0.7853,
"step": 210
},
{
"epoch": 3.53,
"learning_rate": 3.5842293906810036e-05,
"loss": 0.8539,
"step": 220
},
{
"epoch": 3.69,
"learning_rate": 3.494623655913979e-05,
"loss": 0.7822,
"step": 230
},
{
"epoch": 3.86,
"learning_rate": 3.405017921146954e-05,
"loss": 0.8362,
"step": 240
},
{
"epoch": 4.0,
"eval_accuracy": 0.7276995305164319,
"eval_loss": 0.7642938494682312,
"eval_runtime": 22.8716,
"eval_samples_per_second": 74.503,
"eval_steps_per_second": 2.361,
"step": 249
},
{
"epoch": 4.02,
"learning_rate": 3.3154121863799286e-05,
"loss": 0.806,
"step": 250
},
{
"epoch": 4.18,
"learning_rate": 3.2258064516129034e-05,
"loss": 0.7131,
"step": 260
},
{
"epoch": 4.34,
"learning_rate": 3.136200716845878e-05,
"loss": 0.7659,
"step": 270
},
{
"epoch": 4.5,
"learning_rate": 3.046594982078853e-05,
"loss": 0.7285,
"step": 280
},
{
"epoch": 4.66,
"learning_rate": 2.9569892473118284e-05,
"loss": 0.7295,
"step": 290
},
{
"epoch": 4.82,
"learning_rate": 2.8673835125448028e-05,
"loss": 0.776,
"step": 300
},
{
"epoch": 4.98,
"learning_rate": 2.777777777777778e-05,
"loss": 0.7959,
"step": 310
},
{
"epoch": 5.0,
"eval_accuracy": 0.7382629107981221,
"eval_loss": 0.731025218963623,
"eval_runtime": 22.8792,
"eval_samples_per_second": 74.478,
"eval_steps_per_second": 2.36,
"step": 311
},
{
"epoch": 5.14,
"learning_rate": 2.6881720430107527e-05,
"loss": 0.7269,
"step": 320
},
{
"epoch": 5.3,
"learning_rate": 2.5985663082437278e-05,
"loss": 0.6598,
"step": 330
},
{
"epoch": 5.46,
"learning_rate": 2.5089605734767026e-05,
"loss": 0.6914,
"step": 340
},
{
"epoch": 5.62,
"learning_rate": 2.4193548387096777e-05,
"loss": 0.6695,
"step": 350
},
{
"epoch": 5.78,
"learning_rate": 2.3297491039426525e-05,
"loss": 0.6746,
"step": 360
},
{
"epoch": 5.94,
"learning_rate": 2.2401433691756272e-05,
"loss": 0.6765,
"step": 370
},
{
"epoch": 5.99,
"eval_accuracy": 0.7470657276995305,
"eval_loss": 0.7246695756912231,
"eval_runtime": 22.9912,
"eval_samples_per_second": 74.115,
"eval_steps_per_second": 2.349,
"step": 373
},
{
"epoch": 6.1,
"learning_rate": 2.1505376344086024e-05,
"loss": 0.6417,
"step": 380
},
{
"epoch": 6.27,
"learning_rate": 2.060931899641577e-05,
"loss": 0.633,
"step": 390
},
{
"epoch": 6.43,
"learning_rate": 1.9713261648745522e-05,
"loss": 0.6314,
"step": 400
},
{
"epoch": 6.59,
"learning_rate": 1.881720430107527e-05,
"loss": 0.6661,
"step": 410
},
{
"epoch": 6.75,
"learning_rate": 1.7921146953405018e-05,
"loss": 0.6203,
"step": 420
},
{
"epoch": 6.91,
"learning_rate": 1.702508960573477e-05,
"loss": 0.6504,
"step": 430
},
{
"epoch": 6.99,
"eval_accuracy": 0.7576291079812206,
"eval_loss": 0.6938906311988831,
"eval_runtime": 22.8151,
"eval_samples_per_second": 74.687,
"eval_steps_per_second": 2.367,
"step": 435
},
{
"epoch": 7.07,
"learning_rate": 1.6129032258064517e-05,
"loss": 0.5945,
"step": 440
},
{
"epoch": 7.23,
"learning_rate": 1.5232974910394265e-05,
"loss": 0.5912,
"step": 450
},
{
"epoch": 7.39,
"learning_rate": 1.4336917562724014e-05,
"loss": 0.5645,
"step": 460
},
{
"epoch": 7.55,
"learning_rate": 1.3440860215053763e-05,
"loss": 0.5811,
"step": 470
},
{
"epoch": 7.71,
"learning_rate": 1.2544802867383513e-05,
"loss": 0.6433,
"step": 480
},
{
"epoch": 7.87,
"learning_rate": 1.1648745519713262e-05,
"loss": 0.5846,
"step": 490
},
{
"epoch": 8.0,
"eval_accuracy": 0.7576291079812206,
"eval_loss": 0.6983180046081543,
"eval_runtime": 22.9755,
"eval_samples_per_second": 74.166,
"eval_steps_per_second": 2.35,
"step": 498
},
{
"epoch": 8.03,
"learning_rate": 1.0752688172043012e-05,
"loss": 0.6193,
"step": 500
},
{
"epoch": 8.19,
"learning_rate": 9.856630824372761e-06,
"loss": 0.6121,
"step": 510
},
{
"epoch": 8.35,
"learning_rate": 8.960573476702509e-06,
"loss": 0.5985,
"step": 520
},
{
"epoch": 8.51,
"learning_rate": 8.064516129032258e-06,
"loss": 0.554,
"step": 530
},
{
"epoch": 8.67,
"learning_rate": 7.168458781362007e-06,
"loss": 0.5987,
"step": 540
},
{
"epoch": 8.84,
"learning_rate": 6.2724014336917564e-06,
"loss": 0.5881,
"step": 550
},
{
"epoch": 9.0,
"learning_rate": 5.376344086021506e-06,
"loss": 0.5774,
"step": 560
},
{
"epoch": 9.0,
"eval_accuracy": 0.7593896713615024,
"eval_loss": 0.6935292482376099,
"eval_runtime": 23.2143,
"eval_samples_per_second": 73.403,
"eval_steps_per_second": 2.326,
"step": 560
},
{
"epoch": 9.16,
"learning_rate": 4.4802867383512545e-06,
"loss": 0.5731,
"step": 570
},
{
"epoch": 9.32,
"learning_rate": 3.5842293906810035e-06,
"loss": 0.5458,
"step": 580
},
{
"epoch": 9.48,
"learning_rate": 2.688172043010753e-06,
"loss": 0.5373,
"step": 590
},
{
"epoch": 9.64,
"learning_rate": 1.7921146953405017e-06,
"loss": 0.5995,
"step": 600
},
{
"epoch": 9.8,
"learning_rate": 8.960573476702509e-07,
"loss": 0.5283,
"step": 610
},
{
"epoch": 9.96,
"learning_rate": 0.0,
"loss": 0.5749,
"step": 620
},
{
"epoch": 9.96,
"eval_accuracy": 0.7570422535211268,
"eval_loss": 0.6881020665168762,
"eval_runtime": 22.8182,
"eval_samples_per_second": 74.677,
"eval_steps_per_second": 2.367,
"step": 620
},
{
"epoch": 9.96,
"step": 620,
"total_flos": 6.145965294585532e+18,
"train_loss": 0.8511337226436985,
"train_runtime": 3069.2061,
"train_samples_per_second": 25.942,
"train_steps_per_second": 0.202
}
],
"logging_steps": 10,
"max_steps": 620,
"num_train_epochs": 10,
"save_steps": 500,
"total_flos": 6.145965294585532e+18,
"trial_name": null,
"trial_params": null
}