vit-huge-augmented-ph2-patch-14 / trainer_state.json
ahishamm's picture
Uploaded Model
053a596
{
"best_metric": 0.5004622936248779,
"best_model_checkpoint": "./vit-huge-augmented-ph2-patch-14/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.4185,
"step": 10
},
{
"epoch": 0.14,
"learning_rate": 0.0001927536231884058,
"loss": 0.1809,
"step": 20
},
{
"epoch": 0.22,
"learning_rate": 0.0001891304347826087,
"loss": 0.1261,
"step": 30
},
{
"epoch": 0.29,
"learning_rate": 0.0001855072463768116,
"loss": 0.0504,
"step": 40
},
{
"epoch": 0.36,
"learning_rate": 0.0001818840579710145,
"loss": 0.0835,
"step": 50
},
{
"epoch": 0.36,
"eval_accuracy": 0.8581196581196581,
"eval_f1": 0.8581196581196581,
"eval_loss": 0.5004622936248779,
"eval_precision": 0.8581196581196581,
"eval_recall": 0.8581196581196581,
"eval_runtime": 12.5476,
"eval_samples_per_second": 46.623,
"eval_steps_per_second": 5.898,
"step": 50
},
{
"epoch": 0.43,
"learning_rate": 0.0001782608695652174,
"loss": 0.0372,
"step": 60
},
{
"epoch": 0.51,
"learning_rate": 0.0001746376811594203,
"loss": 0.0741,
"step": 70
},
{
"epoch": 0.58,
"learning_rate": 0.0001710144927536232,
"loss": 0.0406,
"step": 80
},
{
"epoch": 0.65,
"learning_rate": 0.0001673913043478261,
"loss": 0.0156,
"step": 90
},
{
"epoch": 0.72,
"learning_rate": 0.000163768115942029,
"loss": 0.0168,
"step": 100
},
{
"epoch": 0.72,
"eval_accuracy": 0.8564102564102564,
"eval_f1": 0.8564102564102564,
"eval_loss": 0.5818132162094116,
"eval_precision": 0.8564102564102564,
"eval_recall": 0.8564102564102564,
"eval_runtime": 12.6681,
"eval_samples_per_second": 46.179,
"eval_steps_per_second": 5.841,
"step": 100
},
{
"epoch": 0.8,
"learning_rate": 0.00016014492753623188,
"loss": 0.039,
"step": 110
},
{
"epoch": 0.87,
"learning_rate": 0.0001565217391304348,
"loss": 0.0577,
"step": 120
},
{
"epoch": 0.94,
"learning_rate": 0.00015289855072463769,
"loss": 0.0093,
"step": 130
},
{
"epoch": 1.01,
"learning_rate": 0.00014927536231884058,
"loss": 0.0343,
"step": 140
},
{
"epoch": 1.09,
"learning_rate": 0.0001456521739130435,
"loss": 0.0205,
"step": 150
},
{
"epoch": 1.09,
"eval_accuracy": 0.8478632478632478,
"eval_f1": 0.8478632478632478,
"eval_loss": 0.5219599604606628,
"eval_precision": 0.8478632478632478,
"eval_recall": 0.8478632478632478,
"eval_runtime": 12.5714,
"eval_samples_per_second": 46.534,
"eval_steps_per_second": 5.886,
"step": 150
},
{
"epoch": 1.16,
"learning_rate": 0.00014202898550724638,
"loss": 0.0124,
"step": 160
},
{
"epoch": 1.23,
"learning_rate": 0.00013840579710144927,
"loss": 0.0048,
"step": 170
},
{
"epoch": 1.3,
"learning_rate": 0.0001347826086956522,
"loss": 0.0042,
"step": 180
},
{
"epoch": 1.38,
"learning_rate": 0.00013115942028985508,
"loss": 0.0278,
"step": 190
},
{
"epoch": 1.45,
"learning_rate": 0.00012753623188405797,
"loss": 0.0125,
"step": 200
},
{
"epoch": 1.45,
"eval_accuracy": 0.8512820512820513,
"eval_f1": 0.8512820512820513,
"eval_loss": 0.6188096404075623,
"eval_precision": 0.8512820512820513,
"eval_recall": 0.8512820512820513,
"eval_runtime": 12.5876,
"eval_samples_per_second": 46.474,
"eval_steps_per_second": 5.879,
"step": 200
},
{
"epoch": 1.52,
"learning_rate": 0.00012391304347826086,
"loss": 0.0034,
"step": 210
},
{
"epoch": 1.59,
"learning_rate": 0.00012028985507246378,
"loss": 0.0034,
"step": 220
},
{
"epoch": 1.67,
"learning_rate": 0.00011666666666666668,
"loss": 0.0033,
"step": 230
},
{
"epoch": 1.74,
"learning_rate": 0.00011304347826086956,
"loss": 0.0029,
"step": 240
},
{
"epoch": 1.81,
"learning_rate": 0.00010942028985507247,
"loss": 0.0029,
"step": 250
},
{
"epoch": 1.81,
"eval_accuracy": 0.8769230769230769,
"eval_f1": 0.8769230769230769,
"eval_loss": 0.6046304106712341,
"eval_precision": 0.8769230769230769,
"eval_recall": 0.8769230769230769,
"eval_runtime": 12.6151,
"eval_samples_per_second": 46.373,
"eval_steps_per_second": 5.866,
"step": 250
},
{
"epoch": 1.88,
"learning_rate": 0.00010579710144927538,
"loss": 0.0027,
"step": 260
},
{
"epoch": 1.96,
"learning_rate": 0.00010217391304347828,
"loss": 0.0024,
"step": 270
},
{
"epoch": 2.03,
"learning_rate": 9.855072463768117e-05,
"loss": 0.0024,
"step": 280
},
{
"epoch": 2.1,
"learning_rate": 9.492753623188406e-05,
"loss": 0.0022,
"step": 290
},
{
"epoch": 2.17,
"learning_rate": 9.130434782608696e-05,
"loss": 0.0021,
"step": 300
},
{
"epoch": 2.17,
"eval_accuracy": 0.8803418803418803,
"eval_f1": 0.8803418803418802,
"eval_loss": 0.6407693028450012,
"eval_precision": 0.8803418803418803,
"eval_recall": 0.8803418803418803,
"eval_runtime": 12.5666,
"eval_samples_per_second": 46.552,
"eval_steps_per_second": 5.889,
"step": 300
},
{
"epoch": 2.25,
"learning_rate": 8.768115942028987e-05,
"loss": 0.0021,
"step": 310
},
{
"epoch": 2.32,
"learning_rate": 8.405797101449276e-05,
"loss": 0.002,
"step": 320
},
{
"epoch": 2.39,
"learning_rate": 8.043478260869566e-05,
"loss": 0.0019,
"step": 330
},
{
"epoch": 2.46,
"learning_rate": 7.681159420289855e-05,
"loss": 0.0018,
"step": 340
},
{
"epoch": 2.54,
"learning_rate": 7.318840579710145e-05,
"loss": 0.0018,
"step": 350
},
{
"epoch": 2.54,
"eval_accuracy": 0.8803418803418803,
"eval_f1": 0.8803418803418802,
"eval_loss": 0.6587864756584167,
"eval_precision": 0.8803418803418803,
"eval_recall": 0.8803418803418803,
"eval_runtime": 12.6386,
"eval_samples_per_second": 46.287,
"eval_steps_per_second": 5.855,
"step": 350
},
{
"epoch": 2.61,
"learning_rate": 6.956521739130436e-05,
"loss": 0.0017,
"step": 360
},
{
"epoch": 2.68,
"learning_rate": 6.594202898550726e-05,
"loss": 0.0017,
"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.0016,
"step": 390
},
{
"epoch": 2.9,
"learning_rate": 5.507246376811594e-05,
"loss": 0.0015,
"step": 400
},
{
"epoch": 2.9,
"eval_accuracy": 0.8803418803418803,
"eval_f1": 0.8803418803418802,
"eval_loss": 0.6720498204231262,
"eval_precision": 0.8803418803418803,
"eval_recall": 0.8803418803418803,
"eval_runtime": 12.6206,
"eval_samples_per_second": 46.353,
"eval_steps_per_second": 5.863,
"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.0015,
"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.0014,
"step": 450
},
{
"epoch": 3.26,
"eval_accuracy": 0.8803418803418803,
"eval_f1": 0.8803418803418802,
"eval_loss": 0.6803039312362671,
"eval_precision": 0.8803418803418803,
"eval_recall": 0.8803418803418803,
"eval_runtime": 12.6282,
"eval_samples_per_second": 46.325,
"eval_steps_per_second": 5.86,
"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.0014,
"step": 470
},
{
"epoch": 3.48,
"learning_rate": 2.608695652173913e-05,
"loss": 0.0014,
"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.8803418803418803,
"eval_f1": 0.8803418803418802,
"eval_loss": 0.6861467361450195,
"eval_precision": 0.8803418803418803,
"eval_recall": 0.8803418803418803,
"eval_runtime": 12.7312,
"eval_samples_per_second": 45.95,
"eval_steps_per_second": 5.812,
"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.0013,
"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.8803418803418803,
"eval_f1": 0.8803418803418802,
"eval_loss": 0.6879416108131409,
"eval_precision": 0.8803418803418803,
"eval_recall": 0.8803418803418803,
"eval_runtime": 12.5938,
"eval_samples_per_second": 46.451,
"eval_steps_per_second": 5.876,
"step": 550
},
{
"epoch": 4.0,
"step": 552,
"total_flos": 5.013262291864781e+18,
"train_loss": 0.024155851723491283,
"train_runtime": 1143.7087,
"train_samples_per_second": 7.694,
"train_steps_per_second": 0.483
}
],
"max_steps": 552,
"num_train_epochs": 4,
"total_flos": 5.013262291864781e+18,
"trial_name": null,
"trial_params": null
}