ViTForImageClassification / trainer_state.json
Andron00e's picture
End of training
8697aff
{
"best_metric": 0.11452386528253555,
"best_model_checkpoint": "./vitforimcls/checkpoint-1500",
"epoch": 4.0,
"eval_steps": 100,
"global_step": 1500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.03,
"learning_rate": 0.00019866666666666668,
"loss": 1.975,
"step": 10
},
{
"epoch": 0.05,
"learning_rate": 0.00019733333333333335,
"loss": 1.1804,
"step": 20
},
{
"epoch": 0.08,
"learning_rate": 0.000196,
"loss": 0.7784,
"step": 30
},
{
"epoch": 0.11,
"learning_rate": 0.0001946666666666667,
"loss": 0.526,
"step": 40
},
{
"epoch": 0.13,
"learning_rate": 0.00019333333333333333,
"loss": 0.4219,
"step": 50
},
{
"epoch": 0.16,
"learning_rate": 0.000192,
"loss": 0.3939,
"step": 60
},
{
"epoch": 0.19,
"learning_rate": 0.00019066666666666668,
"loss": 0.4119,
"step": 70
},
{
"epoch": 0.21,
"learning_rate": 0.00018933333333333335,
"loss": 0.3594,
"step": 80
},
{
"epoch": 0.24,
"learning_rate": 0.000188,
"loss": 0.3315,
"step": 90
},
{
"epoch": 0.27,
"learning_rate": 0.0001866666666666667,
"loss": 0.2995,
"step": 100
},
{
"epoch": 0.27,
"eval_accuracy": 0.9108333333333334,
"eval_loss": 0.3419191539287567,
"eval_runtime": 221.995,
"eval_samples_per_second": 27.028,
"eval_steps_per_second": 0.423,
"step": 100
},
{
"epoch": 0.29,
"learning_rate": 0.00018533333333333333,
"loss": 0.2941,
"step": 110
},
{
"epoch": 0.32,
"learning_rate": 0.00018400000000000003,
"loss": 0.294,
"step": 120
},
{
"epoch": 0.35,
"learning_rate": 0.00018266666666666667,
"loss": 0.2903,
"step": 130
},
{
"epoch": 0.37,
"learning_rate": 0.00018133333333333334,
"loss": 0.2592,
"step": 140
},
{
"epoch": 0.4,
"learning_rate": 0.00018,
"loss": 0.2496,
"step": 150
},
{
"epoch": 0.43,
"learning_rate": 0.00017866666666666668,
"loss": 0.265,
"step": 160
},
{
"epoch": 0.45,
"learning_rate": 0.00017733333333333335,
"loss": 0.245,
"step": 170
},
{
"epoch": 0.48,
"learning_rate": 0.00017600000000000002,
"loss": 0.2321,
"step": 180
},
{
"epoch": 0.51,
"learning_rate": 0.00017466666666666667,
"loss": 0.2008,
"step": 190
},
{
"epoch": 0.53,
"learning_rate": 0.00017333333333333334,
"loss": 0.2289,
"step": 200
},
{
"epoch": 0.53,
"eval_accuracy": 0.9288333333333333,
"eval_loss": 0.24824613332748413,
"eval_runtime": 222.622,
"eval_samples_per_second": 26.952,
"eval_steps_per_second": 0.422,
"step": 200
},
{
"epoch": 0.56,
"learning_rate": 0.000172,
"loss": 0.2231,
"step": 210
},
{
"epoch": 0.59,
"learning_rate": 0.00017066666666666668,
"loss": 0.2198,
"step": 220
},
{
"epoch": 0.61,
"learning_rate": 0.00016933333333333335,
"loss": 0.2189,
"step": 230
},
{
"epoch": 0.64,
"learning_rate": 0.000168,
"loss": 0.2681,
"step": 240
},
{
"epoch": 0.67,
"learning_rate": 0.0001666666666666667,
"loss": 0.1843,
"step": 250
},
{
"epoch": 0.69,
"learning_rate": 0.00016533333333333333,
"loss": 0.219,
"step": 260
},
{
"epoch": 0.72,
"learning_rate": 0.000164,
"loss": 0.1871,
"step": 270
},
{
"epoch": 0.75,
"learning_rate": 0.00016266666666666667,
"loss": 0.2011,
"step": 280
},
{
"epoch": 0.77,
"learning_rate": 0.00016133333333333334,
"loss": 0.2038,
"step": 290
},
{
"epoch": 0.8,
"learning_rate": 0.00016,
"loss": 0.1811,
"step": 300
},
{
"epoch": 0.8,
"eval_accuracy": 0.9356666666666666,
"eval_loss": 0.2139243632555008,
"eval_runtime": 223.6566,
"eval_samples_per_second": 26.827,
"eval_steps_per_second": 0.42,
"step": 300
},
{
"epoch": 0.83,
"learning_rate": 0.00015866666666666668,
"loss": 0.2091,
"step": 310
},
{
"epoch": 0.85,
"learning_rate": 0.00015733333333333333,
"loss": 0.2127,
"step": 320
},
{
"epoch": 0.88,
"learning_rate": 0.00015600000000000002,
"loss": 0.1931,
"step": 330
},
{
"epoch": 0.91,
"learning_rate": 0.00015466666666666667,
"loss": 0.1942,
"step": 340
},
{
"epoch": 0.93,
"learning_rate": 0.00015333333333333334,
"loss": 0.1704,
"step": 350
},
{
"epoch": 0.96,
"learning_rate": 0.000152,
"loss": 0.1733,
"step": 360
},
{
"epoch": 0.99,
"learning_rate": 0.00015066666666666668,
"loss": 0.1759,
"step": 370
},
{
"epoch": 1.01,
"learning_rate": 0.00014933333333333335,
"loss": 0.1403,
"step": 380
},
{
"epoch": 1.04,
"learning_rate": 0.000148,
"loss": 0.0968,
"step": 390
},
{
"epoch": 1.07,
"learning_rate": 0.00014666666666666666,
"loss": 0.0797,
"step": 400
},
{
"epoch": 1.07,
"eval_accuracy": 0.946,
"eval_loss": 0.18132811784744263,
"eval_runtime": 198.5359,
"eval_samples_per_second": 30.221,
"eval_steps_per_second": 0.473,
"step": 400
},
{
"epoch": 1.09,
"learning_rate": 0.00014533333333333333,
"loss": 0.099,
"step": 410
},
{
"epoch": 1.12,
"learning_rate": 0.000144,
"loss": 0.1239,
"step": 420
},
{
"epoch": 1.15,
"learning_rate": 0.00014266666666666667,
"loss": 0.1084,
"step": 430
},
{
"epoch": 1.17,
"learning_rate": 0.00014133333333333334,
"loss": 0.1008,
"step": 440
},
{
"epoch": 1.2,
"learning_rate": 0.00014,
"loss": 0.1114,
"step": 450
},
{
"epoch": 1.23,
"learning_rate": 0.00013866666666666669,
"loss": 0.1102,
"step": 460
},
{
"epoch": 1.25,
"learning_rate": 0.00013733333333333333,
"loss": 0.0892,
"step": 470
},
{
"epoch": 1.28,
"learning_rate": 0.00013600000000000003,
"loss": 0.081,
"step": 480
},
{
"epoch": 1.31,
"learning_rate": 0.00013466666666666667,
"loss": 0.071,
"step": 490
},
{
"epoch": 1.33,
"learning_rate": 0.00013333333333333334,
"loss": 0.1128,
"step": 500
},
{
"epoch": 1.33,
"eval_accuracy": 0.9451666666666667,
"eval_loss": 0.1741384118795395,
"eval_runtime": 167.1262,
"eval_samples_per_second": 35.901,
"eval_steps_per_second": 0.562,
"step": 500
},
{
"epoch": 1.36,
"learning_rate": 0.000132,
"loss": 0.0966,
"step": 510
},
{
"epoch": 1.39,
"learning_rate": 0.00013066666666666668,
"loss": 0.0998,
"step": 520
},
{
"epoch": 1.41,
"learning_rate": 0.00012933333333333332,
"loss": 0.0951,
"step": 530
},
{
"epoch": 1.44,
"learning_rate": 0.00012800000000000002,
"loss": 0.0792,
"step": 540
},
{
"epoch": 1.47,
"learning_rate": 0.00012666666666666666,
"loss": 0.0928,
"step": 550
},
{
"epoch": 1.49,
"learning_rate": 0.00012533333333333334,
"loss": 0.0893,
"step": 560
},
{
"epoch": 1.52,
"learning_rate": 0.000124,
"loss": 0.103,
"step": 570
},
{
"epoch": 1.55,
"learning_rate": 0.00012266666666666668,
"loss": 0.1357,
"step": 580
},
{
"epoch": 1.57,
"learning_rate": 0.00012133333333333335,
"loss": 0.0883,
"step": 590
},
{
"epoch": 1.6,
"learning_rate": 0.00012,
"loss": 0.086,
"step": 600
},
{
"epoch": 1.6,
"eval_accuracy": 0.9513333333333334,
"eval_loss": 0.16589362919330597,
"eval_runtime": 175.0929,
"eval_samples_per_second": 34.268,
"eval_steps_per_second": 0.537,
"step": 600
},
{
"epoch": 1.63,
"learning_rate": 0.00011866666666666669,
"loss": 0.091,
"step": 610
},
{
"epoch": 1.65,
"learning_rate": 0.00011733333333333334,
"loss": 0.0701,
"step": 620
},
{
"epoch": 1.68,
"learning_rate": 0.000116,
"loss": 0.0834,
"step": 630
},
{
"epoch": 1.71,
"learning_rate": 0.00011466666666666667,
"loss": 0.0891,
"step": 640
},
{
"epoch": 1.73,
"learning_rate": 0.00011333333333333334,
"loss": 0.0896,
"step": 650
},
{
"epoch": 1.76,
"learning_rate": 0.00011200000000000001,
"loss": 0.0853,
"step": 660
},
{
"epoch": 1.79,
"learning_rate": 0.00011066666666666667,
"loss": 0.0799,
"step": 670
},
{
"epoch": 1.81,
"learning_rate": 0.00010933333333333333,
"loss": 0.0742,
"step": 680
},
{
"epoch": 1.84,
"learning_rate": 0.00010800000000000001,
"loss": 0.093,
"step": 690
},
{
"epoch": 1.87,
"learning_rate": 0.00010666666666666667,
"loss": 0.0815,
"step": 700
},
{
"epoch": 1.87,
"eval_accuracy": 0.9546666666666667,
"eval_loss": 0.14684614539146423,
"eval_runtime": 174.5528,
"eval_samples_per_second": 34.374,
"eval_steps_per_second": 0.539,
"step": 700
},
{
"epoch": 1.89,
"learning_rate": 0.00010533333333333332,
"loss": 0.0912,
"step": 710
},
{
"epoch": 1.92,
"learning_rate": 0.00010400000000000001,
"loss": 0.0677,
"step": 720
},
{
"epoch": 1.95,
"learning_rate": 0.00010266666666666666,
"loss": 0.0806,
"step": 730
},
{
"epoch": 1.97,
"learning_rate": 0.00010133333333333335,
"loss": 0.0762,
"step": 740
},
{
"epoch": 2.0,
"learning_rate": 0.0001,
"loss": 0.0876,
"step": 750
},
{
"epoch": 2.03,
"learning_rate": 9.866666666666668e-05,
"loss": 0.0425,
"step": 760
},
{
"epoch": 2.05,
"learning_rate": 9.733333333333335e-05,
"loss": 0.054,
"step": 770
},
{
"epoch": 2.08,
"learning_rate": 9.6e-05,
"loss": 0.0384,
"step": 780
},
{
"epoch": 2.11,
"learning_rate": 9.466666666666667e-05,
"loss": 0.0506,
"step": 790
},
{
"epoch": 2.13,
"learning_rate": 9.333333333333334e-05,
"loss": 0.048,
"step": 800
},
{
"epoch": 2.13,
"eval_accuracy": 0.9591666666666666,
"eval_loss": 0.1392858624458313,
"eval_runtime": 165.6771,
"eval_samples_per_second": 36.215,
"eval_steps_per_second": 0.567,
"step": 800
},
{
"epoch": 2.16,
"learning_rate": 9.200000000000001e-05,
"loss": 0.0346,
"step": 810
},
{
"epoch": 2.19,
"learning_rate": 9.066666666666667e-05,
"loss": 0.0258,
"step": 820
},
{
"epoch": 2.21,
"learning_rate": 8.933333333333334e-05,
"loss": 0.0385,
"step": 830
},
{
"epoch": 2.24,
"learning_rate": 8.800000000000001e-05,
"loss": 0.0239,
"step": 840
},
{
"epoch": 2.27,
"learning_rate": 8.666666666666667e-05,
"loss": 0.0327,
"step": 850
},
{
"epoch": 2.29,
"learning_rate": 8.533333333333334e-05,
"loss": 0.0334,
"step": 860
},
{
"epoch": 2.32,
"learning_rate": 8.4e-05,
"loss": 0.0318,
"step": 870
},
{
"epoch": 2.35,
"learning_rate": 8.266666666666667e-05,
"loss": 0.0307,
"step": 880
},
{
"epoch": 2.37,
"learning_rate": 8.133333333333334e-05,
"loss": 0.0267,
"step": 890
},
{
"epoch": 2.4,
"learning_rate": 8e-05,
"loss": 0.021,
"step": 900
},
{
"epoch": 2.4,
"eval_accuracy": 0.9603333333333334,
"eval_loss": 0.13989999890327454,
"eval_runtime": 169.1114,
"eval_samples_per_second": 35.48,
"eval_steps_per_second": 0.556,
"step": 900
},
{
"epoch": 2.43,
"learning_rate": 7.866666666666666e-05,
"loss": 0.0247,
"step": 910
},
{
"epoch": 2.45,
"learning_rate": 7.733333333333333e-05,
"loss": 0.0401,
"step": 920
},
{
"epoch": 2.48,
"learning_rate": 7.6e-05,
"loss": 0.0202,
"step": 930
},
{
"epoch": 2.51,
"learning_rate": 7.466666666666667e-05,
"loss": 0.0289,
"step": 940
},
{
"epoch": 2.53,
"learning_rate": 7.333333333333333e-05,
"loss": 0.0293,
"step": 950
},
{
"epoch": 2.56,
"learning_rate": 7.2e-05,
"loss": 0.025,
"step": 960
},
{
"epoch": 2.59,
"learning_rate": 7.066666666666667e-05,
"loss": 0.016,
"step": 970
},
{
"epoch": 2.61,
"learning_rate": 6.933333333333334e-05,
"loss": 0.0315,
"step": 980
},
{
"epoch": 2.64,
"learning_rate": 6.800000000000001e-05,
"loss": 0.0309,
"step": 990
},
{
"epoch": 2.67,
"learning_rate": 6.666666666666667e-05,
"loss": 0.0271,
"step": 1000
},
{
"epoch": 2.67,
"eval_accuracy": 0.9641666666666666,
"eval_loss": 0.13339994847774506,
"eval_runtime": 163.886,
"eval_samples_per_second": 36.611,
"eval_steps_per_second": 0.574,
"step": 1000
},
{
"epoch": 2.69,
"learning_rate": 6.533333333333334e-05,
"loss": 0.017,
"step": 1010
},
{
"epoch": 2.72,
"learning_rate": 6.400000000000001e-05,
"loss": 0.0231,
"step": 1020
},
{
"epoch": 2.75,
"learning_rate": 6.266666666666667e-05,
"loss": 0.0256,
"step": 1030
},
{
"epoch": 2.77,
"learning_rate": 6.133333333333334e-05,
"loss": 0.0166,
"step": 1040
},
{
"epoch": 2.8,
"learning_rate": 6e-05,
"loss": 0.027,
"step": 1050
},
{
"epoch": 2.83,
"learning_rate": 5.866666666666667e-05,
"loss": 0.0202,
"step": 1060
},
{
"epoch": 2.85,
"learning_rate": 5.7333333333333336e-05,
"loss": 0.0235,
"step": 1070
},
{
"epoch": 2.88,
"learning_rate": 5.6000000000000006e-05,
"loss": 0.0185,
"step": 1080
},
{
"epoch": 2.91,
"learning_rate": 5.466666666666666e-05,
"loss": 0.0167,
"step": 1090
},
{
"epoch": 2.93,
"learning_rate": 5.333333333333333e-05,
"loss": 0.0231,
"step": 1100
},
{
"epoch": 2.93,
"eval_accuracy": 0.9658333333333333,
"eval_loss": 0.12280993908643723,
"eval_runtime": 172.4826,
"eval_samples_per_second": 34.786,
"eval_steps_per_second": 0.545,
"step": 1100
},
{
"epoch": 2.96,
"learning_rate": 5.2000000000000004e-05,
"loss": 0.031,
"step": 1110
},
{
"epoch": 2.99,
"learning_rate": 5.0666666666666674e-05,
"loss": 0.0332,
"step": 1120
},
{
"epoch": 3.01,
"learning_rate": 4.933333333333334e-05,
"loss": 0.0226,
"step": 1130
},
{
"epoch": 3.04,
"learning_rate": 4.8e-05,
"loss": 0.0094,
"step": 1140
},
{
"epoch": 3.07,
"learning_rate": 4.666666666666667e-05,
"loss": 0.0095,
"step": 1150
},
{
"epoch": 3.09,
"learning_rate": 4.5333333333333335e-05,
"loss": 0.0075,
"step": 1160
},
{
"epoch": 3.12,
"learning_rate": 4.4000000000000006e-05,
"loss": 0.0069,
"step": 1170
},
{
"epoch": 3.15,
"learning_rate": 4.266666666666667e-05,
"loss": 0.0072,
"step": 1180
},
{
"epoch": 3.17,
"learning_rate": 4.133333333333333e-05,
"loss": 0.0048,
"step": 1190
},
{
"epoch": 3.2,
"learning_rate": 4e-05,
"loss": 0.0101,
"step": 1200
},
{
"epoch": 3.2,
"eval_accuracy": 0.9673333333333334,
"eval_loss": 0.12290121614933014,
"eval_runtime": 169.3314,
"eval_samples_per_second": 35.433,
"eval_steps_per_second": 0.555,
"step": 1200
},
{
"epoch": 3.23,
"learning_rate": 3.866666666666667e-05,
"loss": 0.0085,
"step": 1210
},
{
"epoch": 3.25,
"learning_rate": 3.733333333333334e-05,
"loss": 0.01,
"step": 1220
},
{
"epoch": 3.28,
"learning_rate": 3.6e-05,
"loss": 0.0091,
"step": 1230
},
{
"epoch": 3.31,
"learning_rate": 3.466666666666667e-05,
"loss": 0.0045,
"step": 1240
},
{
"epoch": 3.33,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0089,
"step": 1250
},
{
"epoch": 3.36,
"learning_rate": 3.2000000000000005e-05,
"loss": 0.0115,
"step": 1260
},
{
"epoch": 3.39,
"learning_rate": 3.066666666666667e-05,
"loss": 0.0051,
"step": 1270
},
{
"epoch": 3.41,
"learning_rate": 2.9333333333333336e-05,
"loss": 0.011,
"step": 1280
},
{
"epoch": 3.44,
"learning_rate": 2.8000000000000003e-05,
"loss": 0.0112,
"step": 1290
},
{
"epoch": 3.47,
"learning_rate": 2.6666666666666667e-05,
"loss": 0.0041,
"step": 1300
},
{
"epoch": 3.47,
"eval_accuracy": 0.9675,
"eval_loss": 0.11890539526939392,
"eval_runtime": 173.0615,
"eval_samples_per_second": 34.67,
"eval_steps_per_second": 0.543,
"step": 1300
},
{
"epoch": 3.49,
"learning_rate": 2.5333333333333337e-05,
"loss": 0.0062,
"step": 1310
},
{
"epoch": 3.52,
"learning_rate": 2.4e-05,
"loss": 0.0055,
"step": 1320
},
{
"epoch": 3.55,
"learning_rate": 2.2666666666666668e-05,
"loss": 0.007,
"step": 1330
},
{
"epoch": 3.57,
"learning_rate": 2.1333333333333335e-05,
"loss": 0.0039,
"step": 1340
},
{
"epoch": 3.6,
"learning_rate": 2e-05,
"loss": 0.0062,
"step": 1350
},
{
"epoch": 3.63,
"learning_rate": 1.866666666666667e-05,
"loss": 0.0042,
"step": 1360
},
{
"epoch": 3.65,
"learning_rate": 1.7333333333333336e-05,
"loss": 0.0077,
"step": 1370
},
{
"epoch": 3.68,
"learning_rate": 1.6000000000000003e-05,
"loss": 0.0065,
"step": 1380
},
{
"epoch": 3.71,
"learning_rate": 1.4666666666666668e-05,
"loss": 0.0144,
"step": 1390
},
{
"epoch": 3.73,
"learning_rate": 1.3333333333333333e-05,
"loss": 0.0043,
"step": 1400
},
{
"epoch": 3.73,
"eval_accuracy": 0.9683333333333334,
"eval_loss": 0.11652529239654541,
"eval_runtime": 173.2794,
"eval_samples_per_second": 34.626,
"eval_steps_per_second": 0.542,
"step": 1400
},
{
"epoch": 3.76,
"learning_rate": 1.2e-05,
"loss": 0.006,
"step": 1410
},
{
"epoch": 3.79,
"learning_rate": 1.0666666666666667e-05,
"loss": 0.0059,
"step": 1420
},
{
"epoch": 3.81,
"learning_rate": 9.333333333333334e-06,
"loss": 0.0093,
"step": 1430
},
{
"epoch": 3.84,
"learning_rate": 8.000000000000001e-06,
"loss": 0.0063,
"step": 1440
},
{
"epoch": 3.87,
"learning_rate": 6.666666666666667e-06,
"loss": 0.0044,
"step": 1450
},
{
"epoch": 3.89,
"learning_rate": 5.333333333333334e-06,
"loss": 0.0051,
"step": 1460
},
{
"epoch": 3.92,
"learning_rate": 4.000000000000001e-06,
"loss": 0.004,
"step": 1470
},
{
"epoch": 3.95,
"learning_rate": 2.666666666666667e-06,
"loss": 0.0037,
"step": 1480
},
{
"epoch": 3.97,
"learning_rate": 1.3333333333333334e-06,
"loss": 0.0039,
"step": 1490
},
{
"epoch": 4.0,
"learning_rate": 0.0,
"loss": 0.0067,
"step": 1500
},
{
"epoch": 4.0,
"eval_accuracy": 0.9696666666666667,
"eval_loss": 0.11452386528253555,
"eval_runtime": 166.2179,
"eval_samples_per_second": 36.097,
"eval_steps_per_second": 0.566,
"step": 1500
},
{
"epoch": 4.0,
"step": 1500,
"total_flos": 1.4879528813592576e+19,
"train_loss": 0.11711712082227071,
"train_runtime": 8681.751,
"train_samples_per_second": 22.115,
"train_steps_per_second": 0.173
}
],
"logging_steps": 10,
"max_steps": 1500,
"num_train_epochs": 4,
"save_steps": 100,
"total_flos": 1.4879528813592576e+19,
"trial_name": null,
"trial_params": null
}