vit-large-augmented-ph2-patch-32 / trainer_state.json
ahishamm's picture
Uploaded Model
260df1b
{
"best_metric": 0.5736537575721741,
"best_model_checkpoint": "./vit-large-augmented-ph2-patch-32/checkpoint-150",
"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.4272,
"step": 10
},
{
"epoch": 0.14,
"learning_rate": 0.0001927536231884058,
"loss": 0.1337,
"step": 20
},
{
"epoch": 0.22,
"learning_rate": 0.0001891304347826087,
"loss": 0.1272,
"step": 30
},
{
"epoch": 0.29,
"learning_rate": 0.0001855072463768116,
"loss": 0.0933,
"step": 40
},
{
"epoch": 0.36,
"learning_rate": 0.0001818840579710145,
"loss": 0.0405,
"step": 50
},
{
"epoch": 0.36,
"eval_accuracy": 0.8341880341880342,
"eval_f1": 0.8341880341880342,
"eval_loss": 0.6853269338607788,
"eval_precision": 0.8341880341880342,
"eval_recall": 0.8341880341880342,
"eval_runtime": 11.7272,
"eval_samples_per_second": 49.884,
"eval_steps_per_second": 6.31,
"step": 50
},
{
"epoch": 0.43,
"learning_rate": 0.0001782608695652174,
"loss": 0.1968,
"step": 60
},
{
"epoch": 0.51,
"learning_rate": 0.0001746376811594203,
"loss": 0.0372,
"step": 70
},
{
"epoch": 0.58,
"learning_rate": 0.0001710144927536232,
"loss": 0.0156,
"step": 80
},
{
"epoch": 0.65,
"learning_rate": 0.0001673913043478261,
"loss": 0.0547,
"step": 90
},
{
"epoch": 0.72,
"learning_rate": 0.000163768115942029,
"loss": 0.0107,
"step": 100
},
{
"epoch": 0.72,
"eval_accuracy": 0.8256410256410256,
"eval_f1": 0.8256410256410256,
"eval_loss": 0.8199310302734375,
"eval_precision": 0.8256410256410256,
"eval_recall": 0.8256410256410256,
"eval_runtime": 11.7833,
"eval_samples_per_second": 49.646,
"eval_steps_per_second": 6.28,
"step": 100
},
{
"epoch": 0.8,
"learning_rate": 0.00016014492753623188,
"loss": 0.2263,
"step": 110
},
{
"epoch": 0.87,
"learning_rate": 0.0001565217391304348,
"loss": 0.0332,
"step": 120
},
{
"epoch": 0.94,
"learning_rate": 0.00015289855072463769,
"loss": 0.048,
"step": 130
},
{
"epoch": 1.01,
"learning_rate": 0.00014927536231884058,
"loss": 0.0626,
"step": 140
},
{
"epoch": 1.09,
"learning_rate": 0.0001456521739130435,
"loss": 0.0338,
"step": 150
},
{
"epoch": 1.09,
"eval_accuracy": 0.8700854700854701,
"eval_f1": 0.8700854700854701,
"eval_loss": 0.5736537575721741,
"eval_precision": 0.8700854700854701,
"eval_recall": 0.8700854700854701,
"eval_runtime": 12.1612,
"eval_samples_per_second": 48.104,
"eval_steps_per_second": 6.085,
"step": 150
},
{
"epoch": 1.16,
"learning_rate": 0.00014202898550724638,
"loss": 0.0042,
"step": 160
},
{
"epoch": 1.23,
"learning_rate": 0.00013840579710144927,
"loss": 0.0047,
"step": 170
},
{
"epoch": 1.3,
"learning_rate": 0.0001347826086956522,
"loss": 0.003,
"step": 180
},
{
"epoch": 1.38,
"learning_rate": 0.00013115942028985508,
"loss": 0.003,
"step": 190
},
{
"epoch": 1.45,
"learning_rate": 0.00012753623188405797,
"loss": 0.0026,
"step": 200
},
{
"epoch": 1.45,
"eval_accuracy": 0.8683760683760684,
"eval_f1": 0.8683760683760683,
"eval_loss": 0.6007649302482605,
"eval_precision": 0.8683760683760684,
"eval_recall": 0.8683760683760684,
"eval_runtime": 11.7792,
"eval_samples_per_second": 49.664,
"eval_steps_per_second": 6.282,
"step": 200
},
{
"epoch": 1.52,
"learning_rate": 0.00012391304347826086,
"loss": 0.0024,
"step": 210
},
{
"epoch": 1.59,
"learning_rate": 0.00012028985507246378,
"loss": 0.0022,
"step": 220
},
{
"epoch": 1.67,
"learning_rate": 0.00011666666666666668,
"loss": 0.0022,
"step": 230
},
{
"epoch": 1.74,
"learning_rate": 0.00011304347826086956,
"loss": 0.002,
"step": 240
},
{
"epoch": 1.81,
"learning_rate": 0.00010942028985507247,
"loss": 0.0019,
"step": 250
},
{
"epoch": 1.81,
"eval_accuracy": 0.8735042735042735,
"eval_f1": 0.8735042735042735,
"eval_loss": 0.6275166273117065,
"eval_precision": 0.8735042735042735,
"eval_recall": 0.8735042735042735,
"eval_runtime": 11.7834,
"eval_samples_per_second": 49.646,
"eval_steps_per_second": 6.28,
"step": 250
},
{
"epoch": 1.88,
"learning_rate": 0.00010579710144927538,
"loss": 0.0019,
"step": 260
},
{
"epoch": 1.96,
"learning_rate": 0.00010217391304347828,
"loss": 0.0017,
"step": 270
},
{
"epoch": 2.03,
"learning_rate": 9.855072463768117e-05,
"loss": 0.0017,
"step": 280
},
{
"epoch": 2.1,
"learning_rate": 9.492753623188406e-05,
"loss": 0.0016,
"step": 290
},
{
"epoch": 2.17,
"learning_rate": 9.130434782608696e-05,
"loss": 0.0016,
"step": 300
},
{
"epoch": 2.17,
"eval_accuracy": 0.8735042735042735,
"eval_f1": 0.8735042735042735,
"eval_loss": 0.6487508416175842,
"eval_precision": 0.8735042735042735,
"eval_recall": 0.8735042735042735,
"eval_runtime": 11.9506,
"eval_samples_per_second": 48.951,
"eval_steps_per_second": 6.192,
"step": 300
},
{
"epoch": 2.25,
"learning_rate": 8.768115942028987e-05,
"loss": 0.0015,
"step": 310
},
{
"epoch": 2.32,
"learning_rate": 8.405797101449276e-05,
"loss": 0.0014,
"step": 320
},
{
"epoch": 2.39,
"learning_rate": 8.043478260869566e-05,
"loss": 0.0014,
"step": 330
},
{
"epoch": 2.46,
"learning_rate": 7.681159420289855e-05,
"loss": 0.0014,
"step": 340
},
{
"epoch": 2.54,
"learning_rate": 7.318840579710145e-05,
"loss": 0.0013,
"step": 350
},
{
"epoch": 2.54,
"eval_accuracy": 0.8752136752136752,
"eval_f1": 0.8752136752136753,
"eval_loss": 0.6639007925987244,
"eval_precision": 0.8752136752136752,
"eval_recall": 0.8752136752136752,
"eval_runtime": 12.3145,
"eval_samples_per_second": 47.505,
"eval_steps_per_second": 6.009,
"step": 350
},
{
"epoch": 2.61,
"learning_rate": 6.956521739130436e-05,
"loss": 0.0013,
"step": 360
},
{
"epoch": 2.68,
"learning_rate": 6.594202898550726e-05,
"loss": 0.0013,
"step": 370
},
{
"epoch": 2.75,
"learning_rate": 6.231884057971015e-05,
"loss": 0.0013,
"step": 380
},
{
"epoch": 2.83,
"learning_rate": 5.869565217391305e-05,
"loss": 0.0012,
"step": 390
},
{
"epoch": 2.9,
"learning_rate": 5.507246376811594e-05,
"loss": 0.0012,
"step": 400
},
{
"epoch": 2.9,
"eval_accuracy": 0.8752136752136752,
"eval_f1": 0.8752136752136753,
"eval_loss": 0.6756650805473328,
"eval_precision": 0.8752136752136752,
"eval_recall": 0.8752136752136752,
"eval_runtime": 11.7227,
"eval_samples_per_second": 49.903,
"eval_steps_per_second": 6.313,
"step": 400
},
{
"epoch": 2.97,
"learning_rate": 5.144927536231884e-05,
"loss": 0.0012,
"step": 410
},
{
"epoch": 3.04,
"learning_rate": 4.782608695652174e-05,
"loss": 0.0012,
"step": 420
},
{
"epoch": 3.12,
"learning_rate": 4.4202898550724645e-05,
"loss": 0.0011,
"step": 430
},
{
"epoch": 3.19,
"learning_rate": 4.057971014492754e-05,
"loss": 0.0012,
"step": 440
},
{
"epoch": 3.26,
"learning_rate": 3.695652173913043e-05,
"loss": 0.0011,
"step": 450
},
{
"epoch": 3.26,
"eval_accuracy": 0.8735042735042735,
"eval_f1": 0.8735042735042735,
"eval_loss": 0.6844302415847778,
"eval_precision": 0.8735042735042735,
"eval_recall": 0.8735042735042735,
"eval_runtime": 11.7458,
"eval_samples_per_second": 49.805,
"eval_steps_per_second": 6.3,
"step": 450
},
{
"epoch": 3.33,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0011,
"step": 460
},
{
"epoch": 3.41,
"learning_rate": 2.971014492753623e-05,
"loss": 0.0011,
"step": 470
},
{
"epoch": 3.48,
"learning_rate": 2.608695652173913e-05,
"loss": 0.0011,
"step": 480
},
{
"epoch": 3.55,
"learning_rate": 2.246376811594203e-05,
"loss": 0.0011,
"step": 490
},
{
"epoch": 3.62,
"learning_rate": 1.8840579710144928e-05,
"loss": 0.001,
"step": 500
},
{
"epoch": 3.62,
"eval_accuracy": 0.8735042735042735,
"eval_f1": 0.8735042735042735,
"eval_loss": 0.6895482540130615,
"eval_precision": 0.8735042735042735,
"eval_recall": 0.8735042735042735,
"eval_runtime": 11.5982,
"eval_samples_per_second": 50.439,
"eval_steps_per_second": 6.38,
"step": 500
},
{
"epoch": 3.7,
"learning_rate": 1.5217391304347828e-05,
"loss": 0.001,
"step": 510
},
{
"epoch": 3.77,
"learning_rate": 1.1594202898550725e-05,
"loss": 0.001,
"step": 520
},
{
"epoch": 3.84,
"learning_rate": 7.971014492753623e-06,
"loss": 0.001,
"step": 530
},
{
"epoch": 3.91,
"learning_rate": 4.347826086956522e-06,
"loss": 0.001,
"step": 540
},
{
"epoch": 3.99,
"learning_rate": 7.246376811594203e-07,
"loss": 0.001,
"step": 550
},
{
"epoch": 3.99,
"eval_accuracy": 0.8735042735042735,
"eval_f1": 0.8735042735042735,
"eval_loss": 0.6913025379180908,
"eval_precision": 0.8735042735042735,
"eval_recall": 0.8735042735042735,
"eval_runtime": 12.0914,
"eval_samples_per_second": 48.382,
"eval_steps_per_second": 6.12,
"step": 550
},
{
"epoch": 4.0,
"step": 552,
"total_flos": 2.42817580228608e+18,
"train_loss": 0.02911639397127041,
"train_runtime": 866.2055,
"train_samples_per_second": 10.159,
"train_steps_per_second": 0.637
}
],
"max_steps": 552,
"num_train_epochs": 4,
"total_flos": 2.42817580228608e+18,
"trial_name": null,
"trial_params": null
}