ConvNeXt-FaceMask-Finetuned / trainer_state.json
AkshatSurolia's picture
Upload trainer_state.json
3d40c2f
{
"best_metric": 0.015090291388332844,
"best_model_checkpoint": "./convnext-tiny-mask-finetuned/checkpoint-500",
"epoch": 3.5398230088495577,
"global_step": 800,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"learning_rate": 0.00019778761061946905,
"loss": 0.0126,
"step": 10
},
{
"epoch": 0.09,
"learning_rate": 0.00019557522123893806,
"loss": 0.0237,
"step": 20
},
{
"epoch": 0.13,
"learning_rate": 0.0001933628318584071,
"loss": 0.0026,
"step": 30
},
{
"epoch": 0.18,
"learning_rate": 0.00019115044247787613,
"loss": 0.0213,
"step": 40
},
{
"epoch": 0.22,
"learning_rate": 0.00018893805309734514,
"loss": 0.01,
"step": 50
},
{
"epoch": 0.27,
"learning_rate": 0.00018672566371681417,
"loss": 0.0325,
"step": 60
},
{
"epoch": 0.31,
"learning_rate": 0.0001845132743362832,
"loss": 0.0212,
"step": 70
},
{
"epoch": 0.35,
"learning_rate": 0.00018230088495575222,
"loss": 0.0138,
"step": 80
},
{
"epoch": 0.4,
"learning_rate": 0.00018008849557522126,
"loss": 0.0175,
"step": 90
},
{
"epoch": 0.44,
"learning_rate": 0.0001778761061946903,
"loss": 0.013,
"step": 100
},
{
"epoch": 0.44,
"eval_accuracy": 0.9944382647385984,
"eval_loss": 0.023499028757214546,
"eval_runtime": 80.0912,
"eval_samples_per_second": 44.899,
"eval_steps_per_second": 5.619,
"step": 100
},
{
"epoch": 0.49,
"learning_rate": 0.0001756637168141593,
"loss": 0.0188,
"step": 110
},
{
"epoch": 0.53,
"learning_rate": 0.00017345132743362834,
"loss": 0.0157,
"step": 120
},
{
"epoch": 0.58,
"learning_rate": 0.00017123893805309737,
"loss": 0.0108,
"step": 130
},
{
"epoch": 0.62,
"learning_rate": 0.00016902654867256638,
"loss": 0.0125,
"step": 140
},
{
"epoch": 0.66,
"learning_rate": 0.00016681415929203542,
"loss": 0.0165,
"step": 150
},
{
"epoch": 0.71,
"learning_rate": 0.00016460176991150443,
"loss": 0.0226,
"step": 160
},
{
"epoch": 0.75,
"learning_rate": 0.00016238938053097346,
"loss": 0.0083,
"step": 170
},
{
"epoch": 0.8,
"learning_rate": 0.0001601769911504425,
"loss": 0.0231,
"step": 180
},
{
"epoch": 0.84,
"learning_rate": 0.0001579646017699115,
"loss": 0.0151,
"step": 190
},
{
"epoch": 0.88,
"learning_rate": 0.00015575221238938055,
"loss": 0.014,
"step": 200
},
{
"epoch": 0.88,
"eval_accuracy": 0.9936040044493882,
"eval_loss": 0.01512084435671568,
"eval_runtime": 80.2971,
"eval_samples_per_second": 44.784,
"eval_steps_per_second": 5.604,
"step": 200
},
{
"epoch": 0.93,
"learning_rate": 0.00015353982300884958,
"loss": 0.0195,
"step": 210
},
{
"epoch": 0.97,
"learning_rate": 0.0001513274336283186,
"loss": 0.0328,
"step": 220
},
{
"epoch": 1.02,
"learning_rate": 0.00014911504424778763,
"loss": 0.0053,
"step": 230
},
{
"epoch": 1.06,
"learning_rate": 0.00014690265486725664,
"loss": 0.004,
"step": 240
},
{
"epoch": 1.11,
"learning_rate": 0.00014469026548672567,
"loss": 0.021,
"step": 250
},
{
"epoch": 1.15,
"learning_rate": 0.00014247787610619468,
"loss": 0.0078,
"step": 260
},
{
"epoch": 1.19,
"learning_rate": 0.00014026548672566372,
"loss": 0.0149,
"step": 270
},
{
"epoch": 1.24,
"learning_rate": 0.00013805309734513276,
"loss": 0.0149,
"step": 280
},
{
"epoch": 1.28,
"learning_rate": 0.00013584070796460177,
"loss": 0.0052,
"step": 290
},
{
"epoch": 1.33,
"learning_rate": 0.0001336283185840708,
"loss": 0.0022,
"step": 300
},
{
"epoch": 1.33,
"eval_accuracy": 0.9947163515016685,
"eval_loss": 0.017131101340055466,
"eval_runtime": 80.9561,
"eval_samples_per_second": 44.419,
"eval_steps_per_second": 5.559,
"step": 300
},
{
"epoch": 1.37,
"learning_rate": 0.0001314159292035398,
"loss": 0.0056,
"step": 310
},
{
"epoch": 1.42,
"learning_rate": 0.00012920353982300885,
"loss": 0.0176,
"step": 320
},
{
"epoch": 1.46,
"learning_rate": 0.00012699115044247788,
"loss": 0.0066,
"step": 330
},
{
"epoch": 1.5,
"learning_rate": 0.0001247787610619469,
"loss": 0.0139,
"step": 340
},
{
"epoch": 1.55,
"learning_rate": 0.00012256637168141593,
"loss": 0.0056,
"step": 350
},
{
"epoch": 1.59,
"learning_rate": 0.00012035398230088497,
"loss": 0.0009,
"step": 360
},
{
"epoch": 1.64,
"learning_rate": 0.00011814159292035399,
"loss": 0.0004,
"step": 370
},
{
"epoch": 1.68,
"learning_rate": 0.00011592920353982301,
"loss": 0.0094,
"step": 380
},
{
"epoch": 1.73,
"learning_rate": 0.00011371681415929205,
"loss": 0.0037,
"step": 390
},
{
"epoch": 1.77,
"learning_rate": 0.00011150442477876106,
"loss": 0.0022,
"step": 400
},
{
"epoch": 1.77,
"eval_accuracy": 0.9958286985539488,
"eval_loss": 0.01814076118171215,
"eval_runtime": 80.2281,
"eval_samples_per_second": 44.822,
"eval_steps_per_second": 5.609,
"step": 400
},
{
"epoch": 1.81,
"learning_rate": 0.00010929203539823009,
"loss": 0.01,
"step": 410
},
{
"epoch": 1.86,
"learning_rate": 0.00010707964601769913,
"loss": 0.0117,
"step": 420
},
{
"epoch": 1.9,
"learning_rate": 0.00010486725663716814,
"loss": 0.0055,
"step": 430
},
{
"epoch": 1.95,
"learning_rate": 0.00010265486725663717,
"loss": 0.0069,
"step": 440
},
{
"epoch": 1.99,
"learning_rate": 0.00010044247787610618,
"loss": 0.0102,
"step": 450
},
{
"epoch": 2.04,
"learning_rate": 9.823008849557522e-05,
"loss": 0.0018,
"step": 460
},
{
"epoch": 2.08,
"learning_rate": 9.601769911504426e-05,
"loss": 0.0027,
"step": 470
},
{
"epoch": 2.12,
"learning_rate": 9.380530973451328e-05,
"loss": 0.0115,
"step": 480
},
{
"epoch": 2.17,
"learning_rate": 9.15929203539823e-05,
"loss": 0.0098,
"step": 490
},
{
"epoch": 2.21,
"learning_rate": 8.938053097345133e-05,
"loss": 0.0037,
"step": 500
},
{
"epoch": 2.21,
"eval_accuracy": 0.9961067853170189,
"eval_loss": 0.015090291388332844,
"eval_runtime": 80.1059,
"eval_samples_per_second": 44.891,
"eval_steps_per_second": 5.618,
"step": 500
},
{
"epoch": 2.26,
"learning_rate": 8.716814159292036e-05,
"loss": 0.0031,
"step": 510
},
{
"epoch": 2.3,
"learning_rate": 8.495575221238938e-05,
"loss": 0.0033,
"step": 520
},
{
"epoch": 2.35,
"learning_rate": 8.274336283185841e-05,
"loss": 0.0005,
"step": 530
},
{
"epoch": 2.39,
"learning_rate": 8.053097345132744e-05,
"loss": 0.0008,
"step": 540
},
{
"epoch": 2.43,
"learning_rate": 7.831858407079647e-05,
"loss": 0.0008,
"step": 550
},
{
"epoch": 2.48,
"learning_rate": 7.610619469026549e-05,
"loss": 0.0009,
"step": 560
},
{
"epoch": 2.52,
"learning_rate": 7.389380530973453e-05,
"loss": 0.0001,
"step": 570
},
{
"epoch": 2.57,
"learning_rate": 7.168141592920355e-05,
"loss": 0.0004,
"step": 580
},
{
"epoch": 2.61,
"learning_rate": 6.946902654867257e-05,
"loss": 0.0084,
"step": 590
},
{
"epoch": 2.65,
"learning_rate": 6.725663716814161e-05,
"loss": 0.0102,
"step": 600
},
{
"epoch": 2.65,
"eval_accuracy": 0.9949944382647385,
"eval_loss": 0.016862528398633003,
"eval_runtime": 79.8413,
"eval_samples_per_second": 45.039,
"eval_steps_per_second": 5.636,
"step": 600
},
{
"epoch": 2.7,
"learning_rate": 6.504424778761063e-05,
"loss": 0.0002,
"step": 610
},
{
"epoch": 2.74,
"learning_rate": 6.283185840707965e-05,
"loss": 0.0035,
"step": 620
},
{
"epoch": 2.79,
"learning_rate": 6.061946902654868e-05,
"loss": 0.0017,
"step": 630
},
{
"epoch": 2.83,
"learning_rate": 5.8407079646017705e-05,
"loss": 0.0004,
"step": 640
},
{
"epoch": 2.88,
"learning_rate": 5.619469026548673e-05,
"loss": 0.0001,
"step": 650
},
{
"epoch": 2.92,
"learning_rate": 5.398230088495575e-05,
"loss": 0.0,
"step": 660
},
{
"epoch": 2.96,
"learning_rate": 5.176991150442479e-05,
"loss": 0.0003,
"step": 670
},
{
"epoch": 3.01,
"learning_rate": 4.955752212389381e-05,
"loss": 0.0005,
"step": 680
},
{
"epoch": 3.05,
"learning_rate": 4.734513274336283e-05,
"loss": 0.0001,
"step": 690
},
{
"epoch": 3.1,
"learning_rate": 4.5132743362831855e-05,
"loss": 0.0003,
"step": 700
},
{
"epoch": 3.1,
"eval_accuracy": 0.9961067853170189,
"eval_loss": 0.01964888535439968,
"eval_runtime": 79.8617,
"eval_samples_per_second": 45.028,
"eval_steps_per_second": 5.635,
"step": 700
},
{
"epoch": 3.14,
"learning_rate": 4.2920353982300885e-05,
"loss": 0.0001,
"step": 710
},
{
"epoch": 3.19,
"learning_rate": 4.0707964601769914e-05,
"loss": 0.0007,
"step": 720
},
{
"epoch": 3.23,
"learning_rate": 3.849557522123894e-05,
"loss": 0.0001,
"step": 730
},
{
"epoch": 3.27,
"learning_rate": 3.628318584070797e-05,
"loss": 0.0017,
"step": 740
},
{
"epoch": 3.32,
"learning_rate": 3.407079646017699e-05,
"loss": 0.0006,
"step": 750
},
{
"epoch": 3.36,
"learning_rate": 3.185840707964602e-05,
"loss": 0.0003,
"step": 760
},
{
"epoch": 3.41,
"learning_rate": 2.964601769911505e-05,
"loss": 0.0001,
"step": 770
},
{
"epoch": 3.45,
"learning_rate": 2.743362831858407e-05,
"loss": 0.0001,
"step": 780
},
{
"epoch": 3.5,
"learning_rate": 2.5221238938053098e-05,
"loss": 0.0008,
"step": 790
},
{
"epoch": 3.54,
"learning_rate": 2.3008849557522124e-05,
"loss": 0.0002,
"step": 800
},
{
"epoch": 3.54,
"eval_accuracy": 0.9955506117908788,
"eval_loss": 0.017847377806901932,
"eval_runtime": 81.2034,
"eval_samples_per_second": 44.284,
"eval_steps_per_second": 5.542,
"step": 800
},
{
"epoch": 3.54,
"step": 800,
"total_flos": 1.2838213030685737e+18,
"train_loss": 0.007916164267808199,
"train_runtime": 4100.2539,
"train_samples_per_second": 14.075,
"train_steps_per_second": 0.22
}
],
"max_steps": 904,
"num_train_epochs": 4,
"total_flos": 1.2838213030685737e+18,
"trial_name": null,
"trial_params": null
}