vit-large-0.0 / trainer_state.json
abhijitgayen's picture
Upload folder using huggingface_hub
76f95c4
{
"best_metric": 0.08173014223575592,
"best_model_checkpoint": "./vit-lage-beans-demo-v5/checkpoint-1200",
"epoch": 4.0,
"eval_steps": 100,
"global_step": 1440,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.03,
"learning_rate": 0.00019861111111111113,
"loss": 1.4551,
"step": 10
},
{
"epoch": 0.06,
"learning_rate": 0.00019722222222222225,
"loss": 1.148,
"step": 20
},
{
"epoch": 0.08,
"learning_rate": 0.00019583333333333334,
"loss": 0.9667,
"step": 30
},
{
"epoch": 0.11,
"learning_rate": 0.00019444444444444446,
"loss": 0.7212,
"step": 40
},
{
"epoch": 0.14,
"learning_rate": 0.00019305555555555558,
"loss": 1.0176,
"step": 50
},
{
"epoch": 0.17,
"learning_rate": 0.00019166666666666667,
"loss": 0.6942,
"step": 60
},
{
"epoch": 0.19,
"learning_rate": 0.00019027777777777776,
"loss": 0.8307,
"step": 70
},
{
"epoch": 0.22,
"learning_rate": 0.00018888888888888888,
"loss": 0.5401,
"step": 80
},
{
"epoch": 0.25,
"learning_rate": 0.0001875,
"loss": 0.7918,
"step": 90
},
{
"epoch": 0.28,
"learning_rate": 0.00018611111111111112,
"loss": 0.7861,
"step": 100
},
{
"epoch": 0.28,
"eval_accuracy": 0.725,
"eval_loss": 0.7609096765518188,
"eval_runtime": 45.8495,
"eval_samples_per_second": 15.704,
"eval_steps_per_second": 1.963,
"step": 100
},
{
"epoch": 0.31,
"learning_rate": 0.00018472222222222224,
"loss": 0.6622,
"step": 110
},
{
"epoch": 0.33,
"learning_rate": 0.00018333333333333334,
"loss": 0.8922,
"step": 120
},
{
"epoch": 0.36,
"learning_rate": 0.00018194444444444445,
"loss": 0.7098,
"step": 130
},
{
"epoch": 0.39,
"learning_rate": 0.00018055555555555557,
"loss": 0.6424,
"step": 140
},
{
"epoch": 0.42,
"learning_rate": 0.0001791666666666667,
"loss": 0.6945,
"step": 150
},
{
"epoch": 0.44,
"learning_rate": 0.00017777777777777779,
"loss": 0.634,
"step": 160
},
{
"epoch": 0.47,
"learning_rate": 0.0001763888888888889,
"loss": 0.5476,
"step": 170
},
{
"epoch": 0.5,
"learning_rate": 0.000175,
"loss": 0.6335,
"step": 180
},
{
"epoch": 0.53,
"learning_rate": 0.00017361111111111112,
"loss": 0.4134,
"step": 190
},
{
"epoch": 0.56,
"learning_rate": 0.00017222222222222224,
"loss": 0.6276,
"step": 200
},
{
"epoch": 0.56,
"eval_accuracy": 0.7763888888888889,
"eval_loss": 0.6835257411003113,
"eval_runtime": 46.4022,
"eval_samples_per_second": 15.517,
"eval_steps_per_second": 1.94,
"step": 200
},
{
"epoch": 0.58,
"learning_rate": 0.00017083333333333333,
"loss": 0.6095,
"step": 210
},
{
"epoch": 0.61,
"learning_rate": 0.00016944444444444445,
"loss": 0.5328,
"step": 220
},
{
"epoch": 0.64,
"learning_rate": 0.00016805555555555557,
"loss": 0.4397,
"step": 230
},
{
"epoch": 0.67,
"learning_rate": 0.0001666666666666667,
"loss": 0.5338,
"step": 240
},
{
"epoch": 0.69,
"learning_rate": 0.00016527777777777778,
"loss": 0.5808,
"step": 250
},
{
"epoch": 0.72,
"learning_rate": 0.0001638888888888889,
"loss": 0.601,
"step": 260
},
{
"epoch": 0.75,
"learning_rate": 0.00016250000000000002,
"loss": 0.445,
"step": 270
},
{
"epoch": 0.78,
"learning_rate": 0.0001611111111111111,
"loss": 0.3964,
"step": 280
},
{
"epoch": 0.81,
"learning_rate": 0.00015972222222222223,
"loss": 0.5958,
"step": 290
},
{
"epoch": 0.83,
"learning_rate": 0.00015833333333333332,
"loss": 0.6089,
"step": 300
},
{
"epoch": 0.83,
"eval_accuracy": 0.7930555555555555,
"eval_loss": 0.5567876100540161,
"eval_runtime": 46.1606,
"eval_samples_per_second": 15.598,
"eval_steps_per_second": 1.95,
"step": 300
},
{
"epoch": 0.86,
"learning_rate": 0.00015694444444444444,
"loss": 0.518,
"step": 310
},
{
"epoch": 0.89,
"learning_rate": 0.00015555555555555556,
"loss": 0.4721,
"step": 320
},
{
"epoch": 0.92,
"learning_rate": 0.00015416666666666668,
"loss": 0.4247,
"step": 330
},
{
"epoch": 0.94,
"learning_rate": 0.00015277777777777777,
"loss": 0.4503,
"step": 340
},
{
"epoch": 0.97,
"learning_rate": 0.0001513888888888889,
"loss": 0.344,
"step": 350
},
{
"epoch": 1.0,
"learning_rate": 0.00015000000000000001,
"loss": 0.3077,
"step": 360
},
{
"epoch": 1.03,
"learning_rate": 0.00014861111111111113,
"loss": 0.2311,
"step": 370
},
{
"epoch": 1.06,
"learning_rate": 0.00014722222222222223,
"loss": 0.2271,
"step": 380
},
{
"epoch": 1.08,
"learning_rate": 0.00014583333333333335,
"loss": 0.2595,
"step": 390
},
{
"epoch": 1.11,
"learning_rate": 0.00014444444444444444,
"loss": 0.302,
"step": 400
},
{
"epoch": 1.11,
"eval_accuracy": 0.8194444444444444,
"eval_loss": 0.5886520147323608,
"eval_runtime": 46.0306,
"eval_samples_per_second": 15.642,
"eval_steps_per_second": 1.955,
"step": 400
},
{
"epoch": 1.14,
"learning_rate": 0.00014305555555555556,
"loss": 0.2532,
"step": 410
},
{
"epoch": 1.17,
"learning_rate": 0.00014166666666666668,
"loss": 0.3509,
"step": 420
},
{
"epoch": 1.19,
"learning_rate": 0.00014027777777777777,
"loss": 0.2317,
"step": 430
},
{
"epoch": 1.22,
"learning_rate": 0.0001388888888888889,
"loss": 0.2371,
"step": 440
},
{
"epoch": 1.25,
"learning_rate": 0.0001375,
"loss": 0.2115,
"step": 450
},
{
"epoch": 1.28,
"learning_rate": 0.00013611111111111113,
"loss": 0.1403,
"step": 460
},
{
"epoch": 1.31,
"learning_rate": 0.00013472222222222225,
"loss": 0.163,
"step": 470
},
{
"epoch": 1.33,
"learning_rate": 0.00013333333333333334,
"loss": 0.4951,
"step": 480
},
{
"epoch": 1.36,
"learning_rate": 0.00013194444444444446,
"loss": 0.274,
"step": 490
},
{
"epoch": 1.39,
"learning_rate": 0.00013055555555555555,
"loss": 0.1723,
"step": 500
},
{
"epoch": 1.39,
"eval_accuracy": 0.8902777777777777,
"eval_loss": 0.298811137676239,
"eval_runtime": 45.7665,
"eval_samples_per_second": 15.732,
"eval_steps_per_second": 1.967,
"step": 500
},
{
"epoch": 1.42,
"learning_rate": 0.00012916666666666667,
"loss": 0.291,
"step": 510
},
{
"epoch": 1.44,
"learning_rate": 0.00012777777777777776,
"loss": 0.2423,
"step": 520
},
{
"epoch": 1.47,
"learning_rate": 0.00012638888888888888,
"loss": 0.1181,
"step": 530
},
{
"epoch": 1.5,
"learning_rate": 0.000125,
"loss": 0.2404,
"step": 540
},
{
"epoch": 1.53,
"learning_rate": 0.00012361111111111112,
"loss": 0.1568,
"step": 550
},
{
"epoch": 1.56,
"learning_rate": 0.00012222222222222224,
"loss": 0.1433,
"step": 560
},
{
"epoch": 1.58,
"learning_rate": 0.00012083333333333333,
"loss": 0.1673,
"step": 570
},
{
"epoch": 1.61,
"learning_rate": 0.00011944444444444445,
"loss": 0.1575,
"step": 580
},
{
"epoch": 1.64,
"learning_rate": 0.00011805555555555556,
"loss": 0.1429,
"step": 590
},
{
"epoch": 1.67,
"learning_rate": 0.00011666666666666668,
"loss": 0.2522,
"step": 600
},
{
"epoch": 1.67,
"eval_accuracy": 0.9166666666666666,
"eval_loss": 0.25636595487594604,
"eval_runtime": 45.8795,
"eval_samples_per_second": 15.693,
"eval_steps_per_second": 1.962,
"step": 600
},
{
"epoch": 1.69,
"learning_rate": 0.00011527777777777777,
"loss": 0.11,
"step": 610
},
{
"epoch": 1.72,
"learning_rate": 0.00011388888888888889,
"loss": 0.1725,
"step": 620
},
{
"epoch": 1.75,
"learning_rate": 0.00011250000000000001,
"loss": 0.159,
"step": 630
},
{
"epoch": 1.78,
"learning_rate": 0.00011111111111111112,
"loss": 0.0854,
"step": 640
},
{
"epoch": 1.81,
"learning_rate": 0.00010972222222222224,
"loss": 0.0999,
"step": 650
},
{
"epoch": 1.83,
"learning_rate": 0.00010833333333333333,
"loss": 0.2217,
"step": 660
},
{
"epoch": 1.86,
"learning_rate": 0.00010694444444444445,
"loss": 0.1712,
"step": 670
},
{
"epoch": 1.89,
"learning_rate": 0.00010555555555555557,
"loss": 0.1299,
"step": 680
},
{
"epoch": 1.92,
"learning_rate": 0.00010416666666666667,
"loss": 0.1376,
"step": 690
},
{
"epoch": 1.94,
"learning_rate": 0.00010277777777777778,
"loss": 0.1115,
"step": 700
},
{
"epoch": 1.94,
"eval_accuracy": 0.9472222222222222,
"eval_loss": 0.16800908744335175,
"eval_runtime": 46.034,
"eval_samples_per_second": 15.641,
"eval_steps_per_second": 1.955,
"step": 700
},
{
"epoch": 1.97,
"learning_rate": 0.00010138888888888889,
"loss": 0.1504,
"step": 710
},
{
"epoch": 2.0,
"learning_rate": 0.0001,
"loss": 0.0608,
"step": 720
},
{
"epoch": 2.03,
"learning_rate": 9.861111111111112e-05,
"loss": 0.0344,
"step": 730
},
{
"epoch": 2.06,
"learning_rate": 9.722222222222223e-05,
"loss": 0.0326,
"step": 740
},
{
"epoch": 2.08,
"learning_rate": 9.583333333333334e-05,
"loss": 0.2028,
"step": 750
},
{
"epoch": 2.11,
"learning_rate": 9.444444444444444e-05,
"loss": 0.0335,
"step": 760
},
{
"epoch": 2.14,
"learning_rate": 9.305555555555556e-05,
"loss": 0.0554,
"step": 770
},
{
"epoch": 2.17,
"learning_rate": 9.166666666666667e-05,
"loss": 0.0525,
"step": 780
},
{
"epoch": 2.19,
"learning_rate": 9.027777777777779e-05,
"loss": 0.0358,
"step": 790
},
{
"epoch": 2.22,
"learning_rate": 8.888888888888889e-05,
"loss": 0.1445,
"step": 800
},
{
"epoch": 2.22,
"eval_accuracy": 0.9402777777777778,
"eval_loss": 0.2064695954322815,
"eval_runtime": 45.6748,
"eval_samples_per_second": 15.764,
"eval_steps_per_second": 1.97,
"step": 800
},
{
"epoch": 2.25,
"learning_rate": 8.75e-05,
"loss": 0.0905,
"step": 810
},
{
"epoch": 2.28,
"learning_rate": 8.611111111111112e-05,
"loss": 0.006,
"step": 820
},
{
"epoch": 2.31,
"learning_rate": 8.472222222222222e-05,
"loss": 0.0438,
"step": 830
},
{
"epoch": 2.33,
"learning_rate": 8.333333333333334e-05,
"loss": 0.0172,
"step": 840
},
{
"epoch": 2.36,
"learning_rate": 8.194444444444445e-05,
"loss": 0.021,
"step": 850
},
{
"epoch": 2.39,
"learning_rate": 8.055555555555556e-05,
"loss": 0.0362,
"step": 860
},
{
"epoch": 2.42,
"learning_rate": 7.916666666666666e-05,
"loss": 0.0007,
"step": 870
},
{
"epoch": 2.44,
"learning_rate": 7.777777777777778e-05,
"loss": 0.0898,
"step": 880
},
{
"epoch": 2.47,
"learning_rate": 7.638888888888889e-05,
"loss": 0.0117,
"step": 890
},
{
"epoch": 2.5,
"learning_rate": 7.500000000000001e-05,
"loss": 0.0648,
"step": 900
},
{
"epoch": 2.5,
"eval_accuracy": 0.9569444444444445,
"eval_loss": 0.1673145890235901,
"eval_runtime": 44.888,
"eval_samples_per_second": 16.04,
"eval_steps_per_second": 2.005,
"step": 900
},
{
"epoch": 2.53,
"learning_rate": 7.361111111111111e-05,
"loss": 0.0248,
"step": 910
},
{
"epoch": 2.56,
"learning_rate": 7.222222222222222e-05,
"loss": 0.0785,
"step": 920
},
{
"epoch": 2.58,
"learning_rate": 7.083333333333334e-05,
"loss": 0.0009,
"step": 930
},
{
"epoch": 2.61,
"learning_rate": 6.944444444444444e-05,
"loss": 0.0249,
"step": 940
},
{
"epoch": 2.64,
"learning_rate": 6.805555555555556e-05,
"loss": 0.018,
"step": 950
},
{
"epoch": 2.67,
"learning_rate": 6.666666666666667e-05,
"loss": 0.0016,
"step": 960
},
{
"epoch": 2.69,
"learning_rate": 6.527777777777778e-05,
"loss": 0.0616,
"step": 970
},
{
"epoch": 2.72,
"learning_rate": 6.388888888888888e-05,
"loss": 0.0318,
"step": 980
},
{
"epoch": 2.75,
"learning_rate": 6.25e-05,
"loss": 0.0253,
"step": 990
},
{
"epoch": 2.78,
"learning_rate": 6.111111111111112e-05,
"loss": 0.0209,
"step": 1000
},
{
"epoch": 2.78,
"eval_accuracy": 0.9569444444444445,
"eval_loss": 0.16363362967967987,
"eval_runtime": 45.529,
"eval_samples_per_second": 15.814,
"eval_steps_per_second": 1.977,
"step": 1000
},
{
"epoch": 2.81,
"learning_rate": 5.972222222222223e-05,
"loss": 0.0209,
"step": 1010
},
{
"epoch": 2.83,
"learning_rate": 5.833333333333334e-05,
"loss": 0.0082,
"step": 1020
},
{
"epoch": 2.86,
"learning_rate": 5.6944444444444445e-05,
"loss": 0.051,
"step": 1030
},
{
"epoch": 2.89,
"learning_rate": 5.555555555555556e-05,
"loss": 0.0023,
"step": 1040
},
{
"epoch": 2.92,
"learning_rate": 5.4166666666666664e-05,
"loss": 0.0072,
"step": 1050
},
{
"epoch": 2.94,
"learning_rate": 5.2777777777777784e-05,
"loss": 0.0777,
"step": 1060
},
{
"epoch": 2.97,
"learning_rate": 5.138888888888889e-05,
"loss": 0.0186,
"step": 1070
},
{
"epoch": 3.0,
"learning_rate": 5e-05,
"loss": 0.0652,
"step": 1080
},
{
"epoch": 3.03,
"learning_rate": 4.8611111111111115e-05,
"loss": 0.0041,
"step": 1090
},
{
"epoch": 3.06,
"learning_rate": 4.722222222222222e-05,
"loss": 0.0003,
"step": 1100
},
{
"epoch": 3.06,
"eval_accuracy": 0.9694444444444444,
"eval_loss": 0.12931323051452637,
"eval_runtime": 45.4456,
"eval_samples_per_second": 15.843,
"eval_steps_per_second": 1.98,
"step": 1100
},
{
"epoch": 3.08,
"learning_rate": 4.5833333333333334e-05,
"loss": 0.0004,
"step": 1110
},
{
"epoch": 3.11,
"learning_rate": 4.4444444444444447e-05,
"loss": 0.0121,
"step": 1120
},
{
"epoch": 3.14,
"learning_rate": 4.305555555555556e-05,
"loss": 0.002,
"step": 1130
},
{
"epoch": 3.17,
"learning_rate": 4.166666666666667e-05,
"loss": 0.0035,
"step": 1140
},
{
"epoch": 3.19,
"learning_rate": 4.027777777777778e-05,
"loss": 0.0057,
"step": 1150
},
{
"epoch": 3.22,
"learning_rate": 3.888888888888889e-05,
"loss": 0.0002,
"step": 1160
},
{
"epoch": 3.25,
"learning_rate": 3.7500000000000003e-05,
"loss": 0.0005,
"step": 1170
},
{
"epoch": 3.28,
"learning_rate": 3.611111111111111e-05,
"loss": 0.0008,
"step": 1180
},
{
"epoch": 3.31,
"learning_rate": 3.472222222222222e-05,
"loss": 0.017,
"step": 1190
},
{
"epoch": 3.33,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0034,
"step": 1200
},
{
"epoch": 3.33,
"eval_accuracy": 0.9791666666666666,
"eval_loss": 0.08173014223575592,
"eval_runtime": 44.7405,
"eval_samples_per_second": 16.093,
"eval_steps_per_second": 2.012,
"step": 1200
},
{
"epoch": 3.36,
"learning_rate": 3.194444444444444e-05,
"loss": 0.0008,
"step": 1210
},
{
"epoch": 3.39,
"learning_rate": 3.055555555555556e-05,
"loss": 0.0003,
"step": 1220
},
{
"epoch": 3.42,
"learning_rate": 2.916666666666667e-05,
"loss": 0.0201,
"step": 1230
},
{
"epoch": 3.44,
"learning_rate": 2.777777777777778e-05,
"loss": 0.0003,
"step": 1240
},
{
"epoch": 3.47,
"learning_rate": 2.6388888888888892e-05,
"loss": 0.0001,
"step": 1250
},
{
"epoch": 3.5,
"learning_rate": 2.5e-05,
"loss": 0.0002,
"step": 1260
},
{
"epoch": 3.53,
"learning_rate": 2.361111111111111e-05,
"loss": 0.0001,
"step": 1270
},
{
"epoch": 3.56,
"learning_rate": 2.2222222222222223e-05,
"loss": 0.0022,
"step": 1280
},
{
"epoch": 3.58,
"learning_rate": 2.0833333333333336e-05,
"loss": 0.0099,
"step": 1290
},
{
"epoch": 3.61,
"learning_rate": 1.9444444444444445e-05,
"loss": 0.0006,
"step": 1300
},
{
"epoch": 3.61,
"eval_accuracy": 0.9833333333333333,
"eval_loss": 0.08741980046033859,
"eval_runtime": 44.9726,
"eval_samples_per_second": 16.01,
"eval_steps_per_second": 2.001,
"step": 1300
},
{
"epoch": 3.64,
"learning_rate": 1.8055555555555555e-05,
"loss": 0.0003,
"step": 1310
},
{
"epoch": 3.67,
"learning_rate": 1.6666666666666667e-05,
"loss": 0.0003,
"step": 1320
},
{
"epoch": 3.69,
"learning_rate": 1.527777777777778e-05,
"loss": 0.0016,
"step": 1330
},
{
"epoch": 3.72,
"learning_rate": 1.388888888888889e-05,
"loss": 0.001,
"step": 1340
},
{
"epoch": 3.75,
"learning_rate": 1.25e-05,
"loss": 0.0001,
"step": 1350
},
{
"epoch": 3.78,
"learning_rate": 1.1111111111111112e-05,
"loss": 0.0001,
"step": 1360
},
{
"epoch": 3.81,
"learning_rate": 9.722222222222223e-06,
"loss": 0.0005,
"step": 1370
},
{
"epoch": 3.83,
"learning_rate": 8.333333333333334e-06,
"loss": 0.0001,
"step": 1380
},
{
"epoch": 3.86,
"learning_rate": 6.944444444444445e-06,
"loss": 0.0002,
"step": 1390
},
{
"epoch": 3.89,
"learning_rate": 5.555555555555556e-06,
"loss": 0.0023,
"step": 1400
},
{
"epoch": 3.89,
"eval_accuracy": 0.9777777777777777,
"eval_loss": 0.10755519568920135,
"eval_runtime": 51.7521,
"eval_samples_per_second": 13.912,
"eval_steps_per_second": 1.739,
"step": 1400
},
{
"epoch": 3.92,
"learning_rate": 4.166666666666667e-06,
"loss": 0.0266,
"step": 1410
},
{
"epoch": 3.94,
"learning_rate": 2.777777777777778e-06,
"loss": 0.0128,
"step": 1420
},
{
"epoch": 3.97,
"learning_rate": 1.388888888888889e-06,
"loss": 0.0001,
"step": 1430
},
{
"epoch": 4.0,
"learning_rate": 0.0,
"loss": 0.0002,
"step": 1440
},
{
"epoch": 4.0,
"step": 1440,
"total_flos": 6.311528820941783e+18,
"train_loss": 0.2221095090903772,
"train_runtime": 4823.2045,
"train_samples_per_second": 4.777,
"train_steps_per_second": 0.299
},
{
"epoch": 4.0,
"eval_accuracy": 0.9791666666666666,
"eval_loss": 0.10265839099884033,
"eval_runtime": 48.2725,
"eval_samples_per_second": 14.915,
"eval_steps_per_second": 1.864,
"step": 1440
}
],
"logging_steps": 10,
"max_steps": 1440,
"num_train_epochs": 4,
"save_steps": 100,
"total_flos": 6.311528820941783e+18,
"trial_name": null,
"trial_params": null
}