ahishamm's picture
Uploaded Model
59892fe
{
"best_metric": 0.0012013022787868977,
"best_model_checkpoint": "./vit-huge-modified-augmented-ph2-patch-14/checkpoint-650",
"epoch": 4.0,
"global_step": 680,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.06,
"learning_rate": 0.00019705882352941177,
"loss": 0.4297,
"step": 10
},
{
"epoch": 0.12,
"learning_rate": 0.00019411764705882354,
"loss": 0.2305,
"step": 20
},
{
"epoch": 0.18,
"learning_rate": 0.0001911764705882353,
"loss": 0.191,
"step": 30
},
{
"epoch": 0.24,
"learning_rate": 0.00018823529411764707,
"loss": 0.0725,
"step": 40
},
{
"epoch": 0.29,
"learning_rate": 0.00018529411764705883,
"loss": 0.0996,
"step": 50
},
{
"epoch": 0.29,
"eval_accuracy": 0.9366438356164384,
"eval_f1": 0.9366438356164383,
"eval_loss": 0.137769415974617,
"eval_precision": 0.9366438356164384,
"eval_recall": 0.9366438356164384,
"eval_runtime": 12.0421,
"eval_samples_per_second": 48.497,
"eval_steps_per_second": 6.062,
"step": 50
},
{
"epoch": 0.35,
"learning_rate": 0.0001823529411764706,
"loss": 0.0622,
"step": 60
},
{
"epoch": 0.41,
"learning_rate": 0.00017941176470588236,
"loss": 0.0802,
"step": 70
},
{
"epoch": 0.47,
"learning_rate": 0.00017647058823529413,
"loss": 0.0313,
"step": 80
},
{
"epoch": 0.53,
"learning_rate": 0.0001735294117647059,
"loss": 0.0278,
"step": 90
},
{
"epoch": 0.59,
"learning_rate": 0.00017058823529411766,
"loss": 0.0096,
"step": 100
},
{
"epoch": 0.59,
"eval_accuracy": 0.9743150684931506,
"eval_f1": 0.9743150684931506,
"eval_loss": 0.05090475082397461,
"eval_precision": 0.9743150684931506,
"eval_recall": 0.9743150684931506,
"eval_runtime": 12.3686,
"eval_samples_per_second": 47.217,
"eval_steps_per_second": 5.902,
"step": 100
},
{
"epoch": 0.65,
"learning_rate": 0.00016764705882352942,
"loss": 0.0087,
"step": 110
},
{
"epoch": 0.71,
"learning_rate": 0.0001647058823529412,
"loss": 0.0274,
"step": 120
},
{
"epoch": 0.76,
"learning_rate": 0.00016176470588235295,
"loss": 0.0053,
"step": 130
},
{
"epoch": 0.82,
"learning_rate": 0.0001588235294117647,
"loss": 0.0048,
"step": 140
},
{
"epoch": 0.88,
"learning_rate": 0.00015588235294117648,
"loss": 0.0049,
"step": 150
},
{
"epoch": 0.88,
"eval_accuracy": 0.9982876712328768,
"eval_f1": 0.9982876712328768,
"eval_loss": 0.008507030084729195,
"eval_precision": 0.9982876712328768,
"eval_recall": 0.9982876712328768,
"eval_runtime": 12.0996,
"eval_samples_per_second": 48.266,
"eval_steps_per_second": 6.033,
"step": 150
},
{
"epoch": 0.94,
"learning_rate": 0.00015294117647058822,
"loss": 0.0045,
"step": 160
},
{
"epoch": 1.0,
"learning_rate": 0.00015000000000000001,
"loss": 0.0037,
"step": 170
},
{
"epoch": 1.06,
"learning_rate": 0.00014705882352941178,
"loss": 0.0034,
"step": 180
},
{
"epoch": 1.12,
"learning_rate": 0.00014411764705882354,
"loss": 0.0032,
"step": 190
},
{
"epoch": 1.18,
"learning_rate": 0.0001411764705882353,
"loss": 0.0029,
"step": 200
},
{
"epoch": 1.18,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.00365791586227715,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.0585,
"eval_samples_per_second": 48.431,
"eval_steps_per_second": 6.054,
"step": 200
},
{
"epoch": 1.24,
"learning_rate": 0.00013823529411764707,
"loss": 0.0028,
"step": 210
},
{
"epoch": 1.29,
"learning_rate": 0.00013529411764705884,
"loss": 0.0026,
"step": 220
},
{
"epoch": 1.35,
"learning_rate": 0.0001323529411764706,
"loss": 0.0024,
"step": 230
},
{
"epoch": 1.41,
"learning_rate": 0.00012941176470588237,
"loss": 0.0023,
"step": 240
},
{
"epoch": 1.47,
"learning_rate": 0.0001264705882352941,
"loss": 0.0022,
"step": 250
},
{
"epoch": 1.47,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.002802133560180664,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.043,
"eval_samples_per_second": 48.493,
"eval_steps_per_second": 6.062,
"step": 250
},
{
"epoch": 1.53,
"learning_rate": 0.0001235294117647059,
"loss": 0.0022,
"step": 260
},
{
"epoch": 1.59,
"learning_rate": 0.00012058823529411765,
"loss": 0.002,
"step": 270
},
{
"epoch": 1.65,
"learning_rate": 0.00011764705882352942,
"loss": 0.002,
"step": 280
},
{
"epoch": 1.71,
"learning_rate": 0.00011470588235294118,
"loss": 0.0018,
"step": 290
},
{
"epoch": 1.76,
"learning_rate": 0.00011176470588235294,
"loss": 0.0018,
"step": 300
},
{
"epoch": 1.76,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.002224571071565151,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.0863,
"eval_samples_per_second": 48.319,
"eval_steps_per_second": 6.04,
"step": 300
},
{
"epoch": 1.82,
"learning_rate": 0.0001088235294117647,
"loss": 0.0017,
"step": 310
},
{
"epoch": 1.88,
"learning_rate": 0.00010588235294117647,
"loss": 0.0017,
"step": 320
},
{
"epoch": 1.94,
"learning_rate": 0.00010294117647058823,
"loss": 0.0016,
"step": 330
},
{
"epoch": 2.0,
"learning_rate": 0.0001,
"loss": 0.0015,
"step": 340
},
{
"epoch": 2.06,
"learning_rate": 9.705882352941177e-05,
"loss": 0.0015,
"step": 350
},
{
"epoch": 2.06,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.0020704269409179688,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.1137,
"eval_samples_per_second": 48.21,
"eval_steps_per_second": 6.026,
"step": 350
},
{
"epoch": 2.12,
"learning_rate": 9.411764705882353e-05,
"loss": 0.0015,
"step": 360
},
{
"epoch": 2.18,
"learning_rate": 9.11764705882353e-05,
"loss": 0.0014,
"step": 370
},
{
"epoch": 2.24,
"learning_rate": 8.823529411764706e-05,
"loss": 0.0014,
"step": 380
},
{
"epoch": 2.29,
"learning_rate": 8.529411764705883e-05,
"loss": 0.0013,
"step": 390
},
{
"epoch": 2.35,
"learning_rate": 8.23529411764706e-05,
"loss": 0.0013,
"step": 400
},
{
"epoch": 2.35,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.001730100717395544,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.202,
"eval_samples_per_second": 47.861,
"eval_steps_per_second": 5.983,
"step": 400
},
{
"epoch": 2.41,
"learning_rate": 7.941176470588235e-05,
"loss": 0.0013,
"step": 410
},
{
"epoch": 2.47,
"learning_rate": 7.647058823529411e-05,
"loss": 0.0012,
"step": 420
},
{
"epoch": 2.53,
"learning_rate": 7.352941176470589e-05,
"loss": 0.0012,
"step": 430
},
{
"epoch": 2.59,
"learning_rate": 7.058823529411765e-05,
"loss": 0.0012,
"step": 440
},
{
"epoch": 2.65,
"learning_rate": 6.764705882352942e-05,
"loss": 0.0011,
"step": 450
},
{
"epoch": 2.65,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.0015468703350052238,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 11.9628,
"eval_samples_per_second": 48.818,
"eval_steps_per_second": 6.102,
"step": 450
},
{
"epoch": 2.71,
"learning_rate": 6.470588235294118e-05,
"loss": 0.0011,
"step": 460
},
{
"epoch": 2.76,
"learning_rate": 6.176470588235295e-05,
"loss": 0.0012,
"step": 470
},
{
"epoch": 2.82,
"learning_rate": 5.882352941176471e-05,
"loss": 0.0011,
"step": 480
},
{
"epoch": 2.88,
"learning_rate": 5.588235294117647e-05,
"loss": 0.0011,
"step": 490
},
{
"epoch": 2.94,
"learning_rate": 5.294117647058824e-05,
"loss": 0.0011,
"step": 500
},
{
"epoch": 2.94,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.0013941257493570447,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.2299,
"eval_samples_per_second": 47.752,
"eval_steps_per_second": 5.969,
"step": 500
},
{
"epoch": 3.0,
"learning_rate": 5e-05,
"loss": 0.001,
"step": 510
},
{
"epoch": 3.06,
"learning_rate": 4.705882352941177e-05,
"loss": 0.001,
"step": 520
},
{
"epoch": 3.12,
"learning_rate": 4.411764705882353e-05,
"loss": 0.0011,
"step": 530
},
{
"epoch": 3.18,
"learning_rate": 4.11764705882353e-05,
"loss": 0.001,
"step": 540
},
{
"epoch": 3.24,
"learning_rate": 3.8235294117647055e-05,
"loss": 0.001,
"step": 550
},
{
"epoch": 3.24,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.001295336289331317,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.262,
"eval_samples_per_second": 47.627,
"eval_steps_per_second": 5.953,
"step": 550
},
{
"epoch": 3.29,
"learning_rate": 3.529411764705883e-05,
"loss": 0.001,
"step": 560
},
{
"epoch": 3.35,
"learning_rate": 3.235294117647059e-05,
"loss": 0.001,
"step": 570
},
{
"epoch": 3.41,
"learning_rate": 2.9411764705882354e-05,
"loss": 0.001,
"step": 580
},
{
"epoch": 3.47,
"learning_rate": 2.647058823529412e-05,
"loss": 0.0009,
"step": 590
},
{
"epoch": 3.53,
"learning_rate": 2.3529411764705884e-05,
"loss": 0.0009,
"step": 600
},
{
"epoch": 3.53,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.0012326836585998535,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.2022,
"eval_samples_per_second": 47.86,
"eval_steps_per_second": 5.983,
"step": 600
},
{
"epoch": 3.59,
"learning_rate": 2.058823529411765e-05,
"loss": 0.0009,
"step": 610
},
{
"epoch": 3.65,
"learning_rate": 1.7647058823529414e-05,
"loss": 0.0009,
"step": 620
},
{
"epoch": 3.71,
"learning_rate": 1.4705882352941177e-05,
"loss": 0.0009,
"step": 630
},
{
"epoch": 3.76,
"learning_rate": 1.1764705882352942e-05,
"loss": 0.0009,
"step": 640
},
{
"epoch": 3.82,
"learning_rate": 8.823529411764707e-06,
"loss": 0.0009,
"step": 650
},
{
"epoch": 3.82,
"eval_accuracy": 1.0,
"eval_f1": 1.0,
"eval_loss": 0.0012013022787868977,
"eval_precision": 1.0,
"eval_recall": 1.0,
"eval_runtime": 12.3868,
"eval_samples_per_second": 47.147,
"eval_steps_per_second": 5.893,
"step": 650
},
{
"epoch": 3.88,
"learning_rate": 5.882352941176471e-06,
"loss": 0.0009,
"step": 660
},
{
"epoch": 3.94,
"learning_rate": 2.9411764705882355e-06,
"loss": 0.0009,
"step": 670
},
{
"epoch": 4.0,
"learning_rate": 0.0,
"loss": 0.001,
"step": 680
},
{
"epoch": 4.0,
"step": 680,
"total_flos": 6.18454266369137e+18,
"train_loss": 0.020127322464523947,
"train_runtime": 978.8347,
"train_samples_per_second": 11.091,
"train_steps_per_second": 0.695
}
],
"max_steps": 680,
"num_train_epochs": 4,
"total_flos": 6.18454266369137e+18,
"trial_name": null,
"trial_params": null
}