vit-base-augmented-ph2-patch-32 / trainer_state.json
ahishamm's picture
Uploaded Model
ee47ee9
raw
history blame
11.3 kB
{
"best_metric": 0.3903774321079254,
"best_model_checkpoint": "./vit-base-augmented-ph2-patch-32/checkpoint-50",
"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.4337,
"step": 10
},
{
"epoch": 0.14,
"learning_rate": 0.0001927536231884058,
"loss": 0.1958,
"step": 20
},
{
"epoch": 0.22,
"learning_rate": 0.0001891304347826087,
"loss": 0.2358,
"step": 30
},
{
"epoch": 0.29,
"learning_rate": 0.0001855072463768116,
"loss": 0.1231,
"step": 40
},
{
"epoch": 0.36,
"learning_rate": 0.0001818840579710145,
"loss": 0.1087,
"step": 50
},
{
"epoch": 0.36,
"eval_accuracy": 0.8683760683760684,
"eval_f1": 0.8683760683760683,
"eval_loss": 0.3903774321079254,
"eval_precision": 0.8683760683760684,
"eval_recall": 0.8683760683760684,
"eval_runtime": 10.4647,
"eval_samples_per_second": 55.902,
"eval_steps_per_second": 7.071,
"step": 50
},
{
"epoch": 0.43,
"learning_rate": 0.0001782608695652174,
"loss": 0.0318,
"step": 60
},
{
"epoch": 0.51,
"learning_rate": 0.0001746376811594203,
"loss": 0.0115,
"step": 70
},
{
"epoch": 0.58,
"learning_rate": 0.0001710144927536232,
"loss": 0.0412,
"step": 80
},
{
"epoch": 0.65,
"learning_rate": 0.0001673913043478261,
"loss": 0.009,
"step": 90
},
{
"epoch": 0.72,
"learning_rate": 0.000163768115942029,
"loss": 0.066,
"step": 100
},
{
"epoch": 0.72,
"eval_accuracy": 0.8273504273504273,
"eval_f1": 0.8273504273504273,
"eval_loss": 0.7072790265083313,
"eval_precision": 0.8273504273504273,
"eval_recall": 0.8273504273504273,
"eval_runtime": 10.5596,
"eval_samples_per_second": 55.4,
"eval_steps_per_second": 7.008,
"step": 100
},
{
"epoch": 0.8,
"learning_rate": 0.00016014492753623188,
"loss": 0.0898,
"step": 110
},
{
"epoch": 0.87,
"learning_rate": 0.0001565217391304348,
"loss": 0.0228,
"step": 120
},
{
"epoch": 0.94,
"learning_rate": 0.00015289855072463769,
"loss": 0.035,
"step": 130
},
{
"epoch": 1.01,
"learning_rate": 0.00014927536231884058,
"loss": 0.0211,
"step": 140
},
{
"epoch": 1.09,
"learning_rate": 0.0001456521739130435,
"loss": 0.0092,
"step": 150
},
{
"epoch": 1.09,
"eval_accuracy": 0.8153846153846154,
"eval_f1": 0.8153846153846154,
"eval_loss": 0.663502037525177,
"eval_precision": 0.8153846153846154,
"eval_recall": 0.8153846153846154,
"eval_runtime": 10.5172,
"eval_samples_per_second": 55.623,
"eval_steps_per_second": 7.036,
"step": 150
},
{
"epoch": 1.16,
"learning_rate": 0.00014202898550724638,
"loss": 0.006,
"step": 160
},
{
"epoch": 1.23,
"learning_rate": 0.00013840579710144927,
"loss": 0.0686,
"step": 170
},
{
"epoch": 1.3,
"learning_rate": 0.0001347826086956522,
"loss": 0.004,
"step": 180
},
{
"epoch": 1.38,
"learning_rate": 0.00013115942028985508,
"loss": 0.0654,
"step": 190
},
{
"epoch": 1.45,
"learning_rate": 0.00012753623188405797,
"loss": 0.0716,
"step": 200
},
{
"epoch": 1.45,
"eval_accuracy": 0.8341880341880342,
"eval_f1": 0.8341880341880342,
"eval_loss": 0.7824386954307556,
"eval_precision": 0.8341880341880342,
"eval_recall": 0.8341880341880342,
"eval_runtime": 10.4823,
"eval_samples_per_second": 55.808,
"eval_steps_per_second": 7.06,
"step": 200
},
{
"epoch": 1.52,
"learning_rate": 0.00012391304347826086,
"loss": 0.1392,
"step": 210
},
{
"epoch": 1.59,
"learning_rate": 0.00012028985507246378,
"loss": 0.0547,
"step": 220
},
{
"epoch": 1.67,
"learning_rate": 0.00011666666666666668,
"loss": 0.0113,
"step": 230
},
{
"epoch": 1.74,
"learning_rate": 0.00011304347826086956,
"loss": 0.004,
"step": 240
},
{
"epoch": 1.81,
"learning_rate": 0.00010942028985507247,
"loss": 0.0056,
"step": 250
},
{
"epoch": 1.81,
"eval_accuracy": 0.8957264957264958,
"eval_f1": 0.8957264957264957,
"eval_loss": 0.5070696473121643,
"eval_precision": 0.8957264957264958,
"eval_recall": 0.8957264957264958,
"eval_runtime": 10.5158,
"eval_samples_per_second": 55.63,
"eval_steps_per_second": 7.037,
"step": 250
},
{
"epoch": 1.88,
"learning_rate": 0.00010579710144927538,
"loss": 0.0037,
"step": 260
},
{
"epoch": 1.96,
"learning_rate": 0.00010217391304347828,
"loss": 0.0027,
"step": 270
},
{
"epoch": 2.03,
"learning_rate": 9.855072463768117e-05,
"loss": 0.0025,
"step": 280
},
{
"epoch": 2.1,
"learning_rate": 9.492753623188406e-05,
"loss": 0.0024,
"step": 290
},
{
"epoch": 2.17,
"learning_rate": 9.130434782608696e-05,
"loss": 0.0023,
"step": 300
},
{
"epoch": 2.17,
"eval_accuracy": 0.8854700854700854,
"eval_f1": 0.8854700854700854,
"eval_loss": 0.5977781414985657,
"eval_precision": 0.8854700854700854,
"eval_recall": 0.8854700854700854,
"eval_runtime": 10.6417,
"eval_samples_per_second": 54.973,
"eval_steps_per_second": 6.954,
"step": 300
},
{
"epoch": 2.25,
"learning_rate": 8.768115942028987e-05,
"loss": 0.0022,
"step": 310
},
{
"epoch": 2.32,
"learning_rate": 8.405797101449276e-05,
"loss": 0.0021,
"step": 320
},
{
"epoch": 2.39,
"learning_rate": 8.043478260869566e-05,
"loss": 0.002,
"step": 330
},
{
"epoch": 2.46,
"learning_rate": 7.681159420289855e-05,
"loss": 0.0019,
"step": 340
},
{
"epoch": 2.54,
"learning_rate": 7.318840579710145e-05,
"loss": 0.0019,
"step": 350
},
{
"epoch": 2.54,
"eval_accuracy": 0.8854700854700854,
"eval_f1": 0.8854700854700854,
"eval_loss": 0.614289402961731,
"eval_precision": 0.8854700854700854,
"eval_recall": 0.8854700854700854,
"eval_runtime": 10.5592,
"eval_samples_per_second": 55.402,
"eval_steps_per_second": 7.008,
"step": 350
},
{
"epoch": 2.61,
"learning_rate": 6.956521739130436e-05,
"loss": 0.0018,
"step": 360
},
{
"epoch": 2.68,
"learning_rate": 6.594202898550726e-05,
"loss": 0.0018,
"step": 370
},
{
"epoch": 2.75,
"learning_rate": 6.231884057971015e-05,
"loss": 0.0017,
"step": 380
},
{
"epoch": 2.83,
"learning_rate": 5.869565217391305e-05,
"loss": 0.0017,
"step": 390
},
{
"epoch": 2.9,
"learning_rate": 5.507246376811594e-05,
"loss": 0.0016,
"step": 400
},
{
"epoch": 2.9,
"eval_accuracy": 0.8888888888888888,
"eval_f1": 0.8888888888888888,
"eval_loss": 0.6227333545684814,
"eval_precision": 0.8888888888888888,
"eval_recall": 0.8888888888888888,
"eval_runtime": 10.6134,
"eval_samples_per_second": 55.119,
"eval_steps_per_second": 6.972,
"step": 400
},
{
"epoch": 2.97,
"learning_rate": 5.144927536231884e-05,
"loss": 0.0016,
"step": 410
},
{
"epoch": 3.04,
"learning_rate": 4.782608695652174e-05,
"loss": 0.0016,
"step": 420
},
{
"epoch": 3.12,
"learning_rate": 4.4202898550724645e-05,
"loss": 0.0015,
"step": 430
},
{
"epoch": 3.19,
"learning_rate": 4.057971014492754e-05,
"loss": 0.0016,
"step": 440
},
{
"epoch": 3.26,
"learning_rate": 3.695652173913043e-05,
"loss": 0.0015,
"step": 450
},
{
"epoch": 3.26,
"eval_accuracy": 0.8888888888888888,
"eval_f1": 0.8888888888888888,
"eval_loss": 0.6293916702270508,
"eval_precision": 0.8888888888888888,
"eval_recall": 0.8888888888888888,
"eval_runtime": 10.4957,
"eval_samples_per_second": 55.737,
"eval_steps_per_second": 7.05,
"step": 450
},
{
"epoch": 3.33,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0015,
"step": 460
},
{
"epoch": 3.41,
"learning_rate": 2.971014492753623e-05,
"loss": 0.0015,
"step": 470
},
{
"epoch": 3.48,
"learning_rate": 2.608695652173913e-05,
"loss": 0.0015,
"step": 480
},
{
"epoch": 3.55,
"learning_rate": 2.246376811594203e-05,
"loss": 0.0014,
"step": 490
},
{
"epoch": 3.62,
"learning_rate": 1.8840579710144928e-05,
"loss": 0.0014,
"step": 500
},
{
"epoch": 3.62,
"eval_accuracy": 0.8888888888888888,
"eval_f1": 0.8888888888888888,
"eval_loss": 0.6337538957595825,
"eval_precision": 0.8888888888888888,
"eval_recall": 0.8888888888888888,
"eval_runtime": 10.5742,
"eval_samples_per_second": 55.323,
"eval_steps_per_second": 6.998,
"step": 500
},
{
"epoch": 3.7,
"learning_rate": 1.5217391304347828e-05,
"loss": 0.0014,
"step": 510
},
{
"epoch": 3.77,
"learning_rate": 1.1594202898550725e-05,
"loss": 0.0014,
"step": 520
},
{
"epoch": 3.84,
"learning_rate": 7.971014492753623e-06,
"loss": 0.0014,
"step": 530
},
{
"epoch": 3.91,
"learning_rate": 4.347826086956522e-06,
"loss": 0.0014,
"step": 540
},
{
"epoch": 3.99,
"learning_rate": 7.246376811594203e-07,
"loss": 0.0014,
"step": 550
},
{
"epoch": 3.99,
"eval_accuracy": 0.8888888888888888,
"eval_f1": 0.8888888888888888,
"eval_loss": 0.6350913643836975,
"eval_precision": 0.8888888888888888,
"eval_recall": 0.8888888888888888,
"eval_runtime": 10.4521,
"eval_samples_per_second": 55.97,
"eval_steps_per_second": 7.08,
"step": 550
},
{
"epoch": 4.0,
"step": 552,
"total_flos": 6.95095773216768e+17,
"train_loss": 0.034777708027673805,
"train_runtime": 468.7352,
"train_samples_per_second": 18.774,
"train_steps_per_second": 1.178
}
],
"max_steps": 552,
"num_train_epochs": 4,
"total_flos": 6.95095773216768e+17,
"trial_name": null,
"trial_params": null
}