vit-base-augmented-ph2-patch-16 / trainer_state.json
ahishamm's picture
Uploaded Model
73c5c60
{
"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
}