utd-pstnet-pde-fromscratch / trainer_state.json
Zhuoxu Huang
init
ea88836
{
"best_metric": 0.958139534883721,
"best_model_checkpoint": "UTD_PSTNet_0.05_0.1_32_20_one_stage-256-8-16-4-0.1-0.1/checkpoint-1950",
"epoch": 13.0,
"global_step": 1950,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.07,
"learning_rate": 0.0016666666666666668,
"loss": 3.7699,
"step": 10
},
{
"epoch": 0.13,
"learning_rate": 0.0033333333333333335,
"loss": 3.742,
"step": 20
},
{
"epoch": 0.2,
"learning_rate": 0.005000000000000001,
"loss": 3.7013,
"step": 30
},
{
"epoch": 0.27,
"learning_rate": 0.006666666666666667,
"loss": 3.6404,
"step": 40
},
{
"epoch": 0.33,
"learning_rate": 0.008333333333333333,
"loss": 3.5677,
"step": 50
},
{
"epoch": 0.4,
"learning_rate": 0.010000000000000002,
"loss": 3.3664,
"step": 60
},
{
"epoch": 0.47,
"learning_rate": 0.011666666666666667,
"loss": 3.2695,
"step": 70
},
{
"epoch": 0.53,
"learning_rate": 0.013333333333333334,
"loss": 3.2016,
"step": 80
},
{
"epoch": 0.6,
"learning_rate": 0.015,
"loss": 3.1609,
"step": 90
},
{
"epoch": 0.67,
"learning_rate": 0.016666666666666666,
"loss": 2.9526,
"step": 100
},
{
"epoch": 0.73,
"learning_rate": 0.018333333333333333,
"loss": 2.7886,
"step": 110
},
{
"epoch": 0.8,
"learning_rate": 0.020000000000000004,
"loss": 2.6072,
"step": 120
},
{
"epoch": 0.87,
"learning_rate": 0.021666666666666667,
"loss": 2.317,
"step": 130
},
{
"epoch": 0.93,
"learning_rate": 0.023333333333333334,
"loss": 2.2291,
"step": 140
},
{
"epoch": 1.0,
"learning_rate": 0.025,
"loss": 2.0348,
"step": 150
},
{
"epoch": 1.0,
"eval_acc": 0.42221289103506193,
"eval_class_acc": [
0.4074074074074074,
0.0,
0.0,
0.0,
0.0,
0.3125,
0.2,
0.0,
1.0,
0.0,
1.0,
0.6875,
0.0,
0.4375,
0.0,
0.0,
0.875,
0.0,
0.0,
0.0,
1.0,
0.0,
1.0,
0.0,
0.0,
1.0,
0.9333333333333333
],
"eval_loss": 2.4160144329071045,
"eval_video_acc": 0.33488372093023255,
"step": 150
},
{
"epoch": 1.07,
"learning_rate": 0.02666666666666667,
"loss": 1.7569,
"step": 160
},
{
"epoch": 1.13,
"learning_rate": 0.028333333333333335,
"loss": 1.7345,
"step": 170
},
{
"epoch": 1.2,
"learning_rate": 0.03,
"loss": 1.4386,
"step": 180
},
{
"epoch": 1.27,
"learning_rate": 0.03166666666666667,
"loss": 1.2771,
"step": 190
},
{
"epoch": 1.33,
"learning_rate": 0.03333333333333333,
"loss": 1.2684,
"step": 200
},
{
"epoch": 1.4,
"learning_rate": 0.034999999999999996,
"loss": 1.0345,
"step": 210
},
{
"epoch": 1.47,
"learning_rate": 0.03666666666666667,
"loss": 1.0647,
"step": 220
},
{
"epoch": 1.53,
"learning_rate": 0.03833333333333334,
"loss": 0.936,
"step": 230
},
{
"epoch": 1.6,
"learning_rate": 0.04000000000000001,
"loss": 0.8767,
"step": 240
},
{
"epoch": 1.67,
"learning_rate": 0.04166666666666667,
"loss": 0.7804,
"step": 250
},
{
"epoch": 1.73,
"learning_rate": 0.043333333333333335,
"loss": 0.6099,
"step": 260
},
{
"epoch": 1.8,
"learning_rate": 0.045000000000000005,
"loss": 0.6498,
"step": 270
},
{
"epoch": 1.87,
"learning_rate": 0.04666666666666667,
"loss": 0.597,
"step": 280
},
{
"epoch": 1.93,
"learning_rate": 0.04833333333333334,
"loss": 0.5412,
"step": 290
},
{
"epoch": 2.0,
"learning_rate": 0.05,
"loss": 0.4441,
"step": 300
},
{
"epoch": 2.0,
"eval_acc": 0.7711526348939743,
"eval_class_acc": [
0.8888888888888888,
0.875,
0.375,
0.4375,
0.8,
0.9375,
1.0,
0.9375,
1.0,
1.0,
1.0,
0.9375,
1.0,
0.8125,
0.5714285714285714,
0.23076923076923078,
1.0,
0.75,
0.75,
0.6153846153846154,
1.0,
1.0,
0.6666666666666666,
0.375,
0.8666666666666667,
1.0,
1.0
],
"eval_loss": 1.0841315984725952,
"eval_video_acc": 0.8162790697674419,
"step": 300
},
{
"epoch": 2.07,
"learning_rate": 0.04981481481481482,
"loss": 0.3756,
"step": 310
},
{
"epoch": 2.13,
"learning_rate": 0.049629629629629635,
"loss": 0.3214,
"step": 320
},
{
"epoch": 2.2,
"learning_rate": 0.04944444444444445,
"loss": 0.3505,
"step": 330
},
{
"epoch": 2.27,
"learning_rate": 0.04925925925925926,
"loss": 0.3809,
"step": 340
},
{
"epoch": 2.33,
"learning_rate": 0.049074074074074076,
"loss": 0.3881,
"step": 350
},
{
"epoch": 2.4,
"learning_rate": 0.04888888888888889,
"loss": 0.2558,
"step": 360
},
{
"epoch": 2.47,
"learning_rate": 0.04870370370370371,
"loss": 0.2281,
"step": 370
},
{
"epoch": 2.53,
"learning_rate": 0.048518518518518516,
"loss": 0.2088,
"step": 380
},
{
"epoch": 2.6,
"learning_rate": 0.04833333333333334,
"loss": 0.2067,
"step": 390
},
{
"epoch": 2.67,
"learning_rate": 0.04814814814814815,
"loss": 0.2293,
"step": 400
},
{
"epoch": 2.73,
"learning_rate": 0.047962962962962964,
"loss": 0.2194,
"step": 410
},
{
"epoch": 2.8,
"learning_rate": 0.04777777777777778,
"loss": 0.1643,
"step": 420
},
{
"epoch": 2.87,
"learning_rate": 0.047592592592592596,
"loss": 0.1991,
"step": 430
},
{
"epoch": 2.93,
"learning_rate": 0.04740740740740741,
"loss": 0.198,
"step": 440
},
{
"epoch": 3.0,
"learning_rate": 0.04722222222222222,
"loss": 0.159,
"step": 450
},
{
"epoch": 3.0,
"eval_acc": 0.805374763804325,
"eval_class_acc": [
0.8888888888888888,
0.3125,
0.5625,
0.75,
0.6666666666666666,
0.625,
1.0,
0.9375,
1.0,
1.0,
0.75,
0.9375,
1.0,
0.9375,
0.7142857142857143,
1.0,
0.8125,
1.0,
1.0,
0.6153846153846154,
1.0,
0.625,
0.9333333333333333,
0.875,
0.9333333333333333,
1.0,
1.0
],
"eval_loss": 1.1171455383300781,
"eval_video_acc": 0.8488372093023255,
"step": 450
},
{
"epoch": 3.07,
"learning_rate": 0.047037037037037044,
"loss": 0.1375,
"step": 460
},
{
"epoch": 3.13,
"learning_rate": 0.04685185185185185,
"loss": 0.1349,
"step": 470
},
{
"epoch": 3.2,
"learning_rate": 0.04666666666666667,
"loss": 0.1553,
"step": 480
},
{
"epoch": 3.27,
"learning_rate": 0.046481481481481485,
"loss": 0.1162,
"step": 490
},
{
"epoch": 3.33,
"learning_rate": 0.0462962962962963,
"loss": 0.1296,
"step": 500
},
{
"epoch": 3.4,
"learning_rate": 0.04611111111111112,
"loss": 0.1581,
"step": 510
},
{
"epoch": 3.47,
"learning_rate": 0.045925925925925926,
"loss": 0.0973,
"step": 520
},
{
"epoch": 3.53,
"learning_rate": 0.04574074074074075,
"loss": 0.0849,
"step": 530
},
{
"epoch": 3.6,
"learning_rate": 0.04555555555555556,
"loss": 0.0759,
"step": 540
},
{
"epoch": 3.67,
"learning_rate": 0.04537037037037037,
"loss": 0.0867,
"step": 550
},
{
"epoch": 3.73,
"learning_rate": 0.04518518518518519,
"loss": 0.13,
"step": 560
},
{
"epoch": 3.8,
"learning_rate": 0.045000000000000005,
"loss": 0.0903,
"step": 570
},
{
"epoch": 3.87,
"learning_rate": 0.044814814814814814,
"loss": 0.1622,
"step": 580
},
{
"epoch": 3.93,
"learning_rate": 0.04462962962962963,
"loss": 0.1265,
"step": 590
},
{
"epoch": 4.0,
"learning_rate": 0.044444444444444446,
"loss": 0.0714,
"step": 600
},
{
"epoch": 4.0,
"eval_acc": 0.8584925467142557,
"eval_class_acc": [
1.0,
0.5,
0.5625,
1.0,
0.9333333333333333,
0.5625,
1.0,
0.875,
1.0,
0.9375,
0.9375,
1.0,
1.0,
0.6875,
0.9285714285714286,
1.0,
0.8125,
1.0,
0.9375,
0.6153846153846154,
1.0,
0.8125,
0.9333333333333333,
0.875,
1.0,
1.0,
1.0
],
"eval_loss": 0.923019289970398,
"eval_video_acc": 0.8883720930232558,
"step": 600
},
{
"epoch": 4.07,
"learning_rate": 0.04425925925925926,
"loss": 0.1165,
"step": 610
},
{
"epoch": 4.13,
"learning_rate": 0.04407407407407408,
"loss": 0.0729,
"step": 620
},
{
"epoch": 4.2,
"learning_rate": 0.043888888888888894,
"loss": 0.1219,
"step": 630
},
{
"epoch": 4.27,
"learning_rate": 0.04370370370370371,
"loss": 0.0795,
"step": 640
},
{
"epoch": 4.33,
"learning_rate": 0.04351851851851852,
"loss": 0.0644,
"step": 650
},
{
"epoch": 4.4,
"learning_rate": 0.043333333333333335,
"loss": 0.0773,
"step": 660
},
{
"epoch": 4.47,
"learning_rate": 0.04314814814814815,
"loss": 0.0748,
"step": 670
},
{
"epoch": 4.53,
"learning_rate": 0.04296296296296297,
"loss": 0.0527,
"step": 680
},
{
"epoch": 4.6,
"learning_rate": 0.042777777777777776,
"loss": 0.068,
"step": 690
},
{
"epoch": 4.67,
"learning_rate": 0.0425925925925926,
"loss": 0.0654,
"step": 700
},
{
"epoch": 4.73,
"learning_rate": 0.04240740740740741,
"loss": 0.1152,
"step": 710
},
{
"epoch": 4.8,
"learning_rate": 0.042222222222222223,
"loss": 0.071,
"step": 720
},
{
"epoch": 4.87,
"learning_rate": 0.04203703703703704,
"loss": 0.0397,
"step": 730
},
{
"epoch": 4.93,
"learning_rate": 0.041851851851851855,
"loss": 0.0437,
"step": 740
},
{
"epoch": 5.0,
"learning_rate": 0.04166666666666667,
"loss": 0.0631,
"step": 750
},
{
"epoch": 5.0,
"eval_acc": 0.8988032752466932,
"eval_class_acc": [
0.9629629629629629,
0.9375,
0.6875,
0.9375,
1.0,
1.0,
1.0,
1.0,
0.9375,
1.0,
0.9375,
1.0,
0.9375,
0.6875,
0.7857142857142857,
0.8461538461538461,
0.75,
1.0,
0.9375,
0.8461538461538461,
1.0,
0.8125,
0.9333333333333333,
1.0,
1.0,
1.0,
1.0
],
"eval_loss": 0.6772840023040771,
"eval_video_acc": 0.9255813953488372,
"step": 750
},
{
"epoch": 5.07,
"learning_rate": 0.04148148148148148,
"loss": 0.0717,
"step": 760
},
{
"epoch": 5.13,
"learning_rate": 0.0412962962962963,
"loss": 0.0423,
"step": 770
},
{
"epoch": 5.2,
"learning_rate": 0.04111111111111111,
"loss": 0.042,
"step": 780
},
{
"epoch": 5.27,
"learning_rate": 0.04092592592592593,
"loss": 0.0396,
"step": 790
},
{
"epoch": 5.33,
"learning_rate": 0.040740740740740744,
"loss": 0.0414,
"step": 800
},
{
"epoch": 5.4,
"learning_rate": 0.04055555555555556,
"loss": 0.046,
"step": 810
},
{
"epoch": 5.47,
"learning_rate": 0.040370370370370376,
"loss": 0.045,
"step": 820
},
{
"epoch": 5.53,
"learning_rate": 0.040185185185185185,
"loss": 0.1175,
"step": 830
},
{
"epoch": 5.6,
"learning_rate": 0.04000000000000001,
"loss": 0.052,
"step": 840
},
{
"epoch": 5.67,
"learning_rate": 0.03981481481481482,
"loss": 0.0706,
"step": 850
},
{
"epoch": 5.73,
"learning_rate": 0.03962962962962963,
"loss": 0.0359,
"step": 860
},
{
"epoch": 5.8,
"learning_rate": 0.03944444444444445,
"loss": 0.0315,
"step": 870
},
{
"epoch": 5.87,
"learning_rate": 0.039259259259259265,
"loss": 0.0371,
"step": 880
},
{
"epoch": 5.93,
"learning_rate": 0.039074074074074074,
"loss": 0.051,
"step": 890
},
{
"epoch": 6.0,
"learning_rate": 0.03888888888888889,
"loss": 0.0524,
"step": 900
},
{
"epoch": 6.0,
"eval_acc": 0.8803275246693261,
"eval_class_acc": [
0.7777777777777778,
0.25,
0.5,
1.0,
0.8666666666666667,
0.9375,
0.9333333333333333,
0.9375,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.5714285714285714,
1.0,
1.0,
1.0,
1.0,
0.8461538461538461,
1.0,
1.0,
0.8,
1.0,
1.0,
1.0,
1.0
],
"eval_loss": 0.8187194466590881,
"eval_video_acc": 0.9023255813953488,
"step": 900
},
{
"epoch": 6.07,
"learning_rate": 0.038703703703703705,
"loss": 0.0745,
"step": 910
},
{
"epoch": 6.13,
"learning_rate": 0.03851851851851852,
"loss": 0.0283,
"step": 920
},
{
"epoch": 6.2,
"learning_rate": 0.03833333333333334,
"loss": 0.035,
"step": 930
},
{
"epoch": 6.27,
"learning_rate": 0.03814814814814815,
"loss": 0.0377,
"step": 940
},
{
"epoch": 6.33,
"learning_rate": 0.03796296296296297,
"loss": 0.0573,
"step": 950
},
{
"epoch": 6.4,
"learning_rate": 0.03777777777777778,
"loss": 0.0276,
"step": 960
},
{
"epoch": 6.47,
"learning_rate": 0.037592592592592594,
"loss": 0.0269,
"step": 970
},
{
"epoch": 6.53,
"learning_rate": 0.03740740740740741,
"loss": 0.0215,
"step": 980
},
{
"epoch": 6.6,
"learning_rate": 0.037222222222222226,
"loss": 0.0197,
"step": 990
},
{
"epoch": 6.67,
"learning_rate": 0.037037037037037035,
"loss": 0.0213,
"step": 1000
},
{
"epoch": 6.73,
"learning_rate": 0.03685185185185185,
"loss": 0.0203,
"step": 1010
},
{
"epoch": 6.8,
"learning_rate": 0.03666666666666667,
"loss": 0.0309,
"step": 1020
},
{
"epoch": 6.87,
"learning_rate": 0.03648148148148148,
"loss": 0.0204,
"step": 1030
},
{
"epoch": 6.93,
"learning_rate": 0.0362962962962963,
"loss": 0.0173,
"step": 1040
},
{
"epoch": 7.0,
"learning_rate": 0.036111111111111115,
"loss": 0.0243,
"step": 1050
},
{
"epoch": 7.0,
"eval_acc": 0.897753516691161,
"eval_class_acc": [
0.8888888888888888,
0.9375,
0.5625,
0.9375,
0.8666666666666667,
0.8125,
1.0,
1.0,
1.0,
0.9375,
1.0,
1.0,
1.0,
0.8125,
0.7857142857142857,
1.0,
1.0,
1.0,
1.0,
0.7692307692307693,
0.9375,
0.9375,
0.9333333333333333,
1.0,
1.0,
1.0,
1.0
],
"eval_loss": 0.8365077972412109,
"eval_video_acc": 0.9302325581395349,
"step": 1050
},
{
"epoch": 7.07,
"learning_rate": 0.03592592592592593,
"loss": 0.022,
"step": 1060
},
{
"epoch": 7.13,
"learning_rate": 0.03574074074074074,
"loss": 0.022,
"step": 1070
},
{
"epoch": 7.2,
"learning_rate": 0.035555555555555556,
"loss": 0.0172,
"step": 1080
},
{
"epoch": 7.27,
"learning_rate": 0.03537037037037037,
"loss": 0.0163,
"step": 1090
},
{
"epoch": 7.33,
"learning_rate": 0.03518518518518519,
"loss": 0.0149,
"step": 1100
},
{
"epoch": 7.4,
"learning_rate": 0.034999999999999996,
"loss": 0.0175,
"step": 1110
},
{
"epoch": 7.47,
"learning_rate": 0.03481481481481482,
"loss": 0.0177,
"step": 1120
},
{
"epoch": 7.53,
"learning_rate": 0.03462962962962963,
"loss": 0.0149,
"step": 1130
},
{
"epoch": 7.6,
"learning_rate": 0.034444444444444444,
"loss": 0.0142,
"step": 1140
},
{
"epoch": 7.67,
"learning_rate": 0.03425925925925926,
"loss": 0.0133,
"step": 1150
},
{
"epoch": 7.73,
"learning_rate": 0.034074074074074076,
"loss": 0.0124,
"step": 1160
},
{
"epoch": 7.8,
"learning_rate": 0.03388888888888889,
"loss": 0.0124,
"step": 1170
},
{
"epoch": 7.87,
"learning_rate": 0.0337037037037037,
"loss": 0.0121,
"step": 1180
},
{
"epoch": 7.93,
"learning_rate": 0.033518518518518524,
"loss": 0.0115,
"step": 1190
},
{
"epoch": 8.0,
"learning_rate": 0.03333333333333333,
"loss": 0.0123,
"step": 1200
},
{
"epoch": 8.0,
"eval_acc": 0.9118202813352929,
"eval_class_acc": [
0.8888888888888888,
0.875,
0.5625,
1.0,
0.8,
0.875,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.8125,
0.5714285714285714,
1.0,
1.0,
1.0,
1.0,
0.7692307692307693,
1.0,
1.0,
0.9333333333333333,
1.0,
1.0,
1.0,
1.0
],
"eval_loss": 0.751693844795227,
"eval_video_acc": 0.9302325581395349,
"step": 1200
},
{
"epoch": 8.07,
"learning_rate": 0.03314814814814815,
"loss": 0.0113,
"step": 1210
},
{
"epoch": 8.13,
"learning_rate": 0.032962962962962965,
"loss": 0.0128,
"step": 1220
},
{
"epoch": 8.2,
"learning_rate": 0.03277777777777778,
"loss": 0.012,
"step": 1230
},
{
"epoch": 8.27,
"learning_rate": 0.0325925925925926,
"loss": 0.0101,
"step": 1240
},
{
"epoch": 8.33,
"learning_rate": 0.032407407407407406,
"loss": 0.0121,
"step": 1250
},
{
"epoch": 8.4,
"learning_rate": 0.03222222222222223,
"loss": 0.0107,
"step": 1260
},
{
"epoch": 8.47,
"learning_rate": 0.03203703703703704,
"loss": 0.012,
"step": 1270
},
{
"epoch": 8.53,
"learning_rate": 0.03185185185185185,
"loss": 0.0139,
"step": 1280
},
{
"epoch": 8.6,
"learning_rate": 0.03166666666666667,
"loss": 0.0196,
"step": 1290
},
{
"epoch": 8.67,
"learning_rate": 0.031481481481481485,
"loss": 0.0106,
"step": 1300
},
{
"epoch": 8.73,
"learning_rate": 0.031296296296296294,
"loss": 0.0174,
"step": 1310
},
{
"epoch": 8.8,
"learning_rate": 0.031111111111111114,
"loss": 0.0198,
"step": 1320
},
{
"epoch": 8.87,
"learning_rate": 0.030925925925925926,
"loss": 0.011,
"step": 1330
},
{
"epoch": 8.93,
"learning_rate": 0.030740740740740742,
"loss": 0.0112,
"step": 1340
},
{
"epoch": 9.0,
"learning_rate": 0.030555555555555558,
"loss": 0.0114,
"step": 1350
},
{
"epoch": 9.0,
"eval_acc": 0.8935544824690321,
"eval_class_acc": [
0.9629629629629629,
0.875,
0.625,
1.0,
0.6,
0.75,
0.8666666666666667,
1.0,
1.0,
0.9375,
1.0,
1.0,
1.0,
0.8125,
0.7857142857142857,
1.0,
1.0,
1.0,
1.0,
0.8461538461538461,
1.0,
0.9375,
1.0,
1.0,
1.0,
1.0,
1.0
],
"eval_loss": 0.7627997398376465,
"eval_video_acc": 0.9279069767441861,
"step": 1350
},
{
"epoch": 9.07,
"learning_rate": 0.03037037037037037,
"loss": 0.0105,
"step": 1360
},
{
"epoch": 9.13,
"learning_rate": 0.03018518518518519,
"loss": 0.0099,
"step": 1370
},
{
"epoch": 9.2,
"learning_rate": 0.03,
"loss": 0.0097,
"step": 1380
},
{
"epoch": 9.27,
"learning_rate": 0.02981481481481482,
"loss": 0.0102,
"step": 1390
},
{
"epoch": 9.33,
"learning_rate": 0.02962962962962963,
"loss": 0.01,
"step": 1400
},
{
"epoch": 9.4,
"learning_rate": 0.029444444444444447,
"loss": 0.0114,
"step": 1410
},
{
"epoch": 9.47,
"learning_rate": 0.02925925925925926,
"loss": 0.0116,
"step": 1420
},
{
"epoch": 9.53,
"learning_rate": 0.029074074074074075,
"loss": 0.0119,
"step": 1430
},
{
"epoch": 9.6,
"learning_rate": 0.028888888888888888,
"loss": 0.0101,
"step": 1440
},
{
"epoch": 9.67,
"learning_rate": 0.028703703703703703,
"loss": 0.0109,
"step": 1450
},
{
"epoch": 9.73,
"learning_rate": 0.028518518518518523,
"loss": 0.0103,
"step": 1460
},
{
"epoch": 9.8,
"learning_rate": 0.028333333333333335,
"loss": 0.0114,
"step": 1470
},
{
"epoch": 9.87,
"learning_rate": 0.02814814814814815,
"loss": 0.0138,
"step": 1480
},
{
"epoch": 9.93,
"learning_rate": 0.027962962962962964,
"loss": 0.01,
"step": 1490
},
{
"epoch": 10.0,
"learning_rate": 0.02777777777777778,
"loss": 0.0091,
"step": 1500
},
{
"epoch": 10.0,
"eval_acc": 0.9065714885576317,
"eval_class_acc": [
0.9629629629629629,
0.875,
0.625,
1.0,
0.6,
0.875,
1.0,
1.0,
1.0,
1.0,
0.9375,
1.0,
0.875,
0.875,
0.35714285714285715,
1.0,
1.0,
1.0,
1.0,
0.8461538461538461,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0
],
"eval_loss": 0.7203760147094727,
"eval_video_acc": 0.9232558139534883,
"step": 1500
},
{
"epoch": 10.07,
"learning_rate": 0.027592592592592592,
"loss": 0.0207,
"step": 1510
},
{
"epoch": 10.13,
"learning_rate": 0.027407407407407408,
"loss": 0.0096,
"step": 1520
},
{
"epoch": 10.2,
"learning_rate": 0.02722222222222222,
"loss": 0.0096,
"step": 1530
},
{
"epoch": 10.27,
"learning_rate": 0.02703703703703704,
"loss": 0.0089,
"step": 1540
},
{
"epoch": 10.33,
"learning_rate": 0.026851851851851856,
"loss": 0.0091,
"step": 1550
},
{
"epoch": 10.4,
"learning_rate": 0.02666666666666667,
"loss": 0.0088,
"step": 1560
},
{
"epoch": 10.47,
"learning_rate": 0.026481481481481484,
"loss": 0.0092,
"step": 1570
},
{
"epoch": 10.53,
"learning_rate": 0.026296296296296297,
"loss": 0.0101,
"step": 1580
},
{
"epoch": 10.6,
"learning_rate": 0.026111111111111113,
"loss": 0.0159,
"step": 1590
},
{
"epoch": 10.67,
"learning_rate": 0.025925925925925925,
"loss": 0.0118,
"step": 1600
},
{
"epoch": 10.73,
"learning_rate": 0.025740740740740745,
"loss": 0.0101,
"step": 1610
},
{
"epoch": 10.8,
"learning_rate": 0.025555555555555554,
"loss": 0.0091,
"step": 1620
},
{
"epoch": 10.87,
"learning_rate": 0.025370370370370373,
"loss": 0.0102,
"step": 1630
},
{
"epoch": 10.93,
"learning_rate": 0.025185185185185185,
"loss": 0.0084,
"step": 1640
},
{
"epoch": 11.0,
"learning_rate": 0.025,
"loss": 0.0081,
"step": 1650
},
{
"epoch": 11.0,
"eval_acc": 0.9202183497795507,
"eval_class_acc": [
0.9259259259259259,
0.875,
0.6875,
1.0,
0.6666666666666666,
1.0,
0.9333333333333333,
1.0,
1.0,
0.9375,
1.0,
1.0,
1.0,
0.875,
0.8571428571428571,
1.0,
1.0,
1.0,
1.0,
0.8461538461538461,
0.9375,
1.0,
0.9333333333333333,
0.9375,
1.0,
1.0,
1.0
],
"eval_loss": 0.5837204456329346,
"eval_video_acc": 0.9418604651162791,
"step": 1650
},
{
"epoch": 11.07,
"learning_rate": 0.024814814814814817,
"loss": 0.0082,
"step": 1660
},
{
"epoch": 11.13,
"learning_rate": 0.02462962962962963,
"loss": 0.0081,
"step": 1670
},
{
"epoch": 11.2,
"learning_rate": 0.024444444444444446,
"loss": 0.0102,
"step": 1680
},
{
"epoch": 11.27,
"learning_rate": 0.024259259259259258,
"loss": 0.015,
"step": 1690
},
{
"epoch": 11.33,
"learning_rate": 0.024074074074074074,
"loss": 0.0081,
"step": 1700
},
{
"epoch": 11.4,
"learning_rate": 0.02388888888888889,
"loss": 0.0069,
"step": 1710
},
{
"epoch": 11.47,
"learning_rate": 0.023703703703703706,
"loss": 0.0075,
"step": 1720
},
{
"epoch": 11.53,
"learning_rate": 0.023518518518518522,
"loss": 0.0087,
"step": 1730
},
{
"epoch": 11.6,
"learning_rate": 0.023333333333333334,
"loss": 0.0074,
"step": 1740
},
{
"epoch": 11.67,
"learning_rate": 0.02314814814814815,
"loss": 0.0075,
"step": 1750
},
{
"epoch": 11.73,
"learning_rate": 0.022962962962962963,
"loss": 0.0091,
"step": 1760
},
{
"epoch": 11.8,
"learning_rate": 0.02277777777777778,
"loss": 0.0072,
"step": 1770
},
{
"epoch": 11.87,
"learning_rate": 0.022592592592592595,
"loss": 0.0079,
"step": 1780
},
{
"epoch": 11.93,
"learning_rate": 0.022407407407407407,
"loss": 0.0076,
"step": 1790
},
{
"epoch": 12.0,
"learning_rate": 0.022222222222222223,
"loss": 0.0074,
"step": 1800
},
{
"epoch": 12.0,
"eval_acc": 0.9128700398908252,
"eval_class_acc": [
0.9259259259259259,
0.9375,
0.6875,
1.0,
0.8,
1.0,
0.9333333333333333,
1.0,
1.0,
0.9375,
1.0,
1.0,
1.0,
0.6875,
0.8571428571428571,
1.0,
1.0,
1.0,
1.0,
0.8461538461538461,
1.0,
0.9375,
1.0,
1.0,
0.9333333333333333,
1.0,
1.0
],
"eval_loss": 0.6411592960357666,
"eval_video_acc": 0.9441860465116279,
"step": 1800
},
{
"epoch": 12.07,
"learning_rate": 0.02203703703703704,
"loss": 0.007,
"step": 1810
},
{
"epoch": 12.13,
"learning_rate": 0.021851851851851855,
"loss": 0.0071,
"step": 1820
},
{
"epoch": 12.2,
"learning_rate": 0.021666666666666667,
"loss": 0.0075,
"step": 1830
},
{
"epoch": 12.27,
"learning_rate": 0.021481481481481483,
"loss": 0.0074,
"step": 1840
},
{
"epoch": 12.33,
"learning_rate": 0.0212962962962963,
"loss": 0.0074,
"step": 1850
},
{
"epoch": 12.4,
"learning_rate": 0.021111111111111112,
"loss": 0.0063,
"step": 1860
},
{
"epoch": 12.47,
"learning_rate": 0.020925925925925928,
"loss": 0.0077,
"step": 1870
},
{
"epoch": 12.53,
"learning_rate": 0.02074074074074074,
"loss": 0.0139,
"step": 1880
},
{
"epoch": 12.6,
"learning_rate": 0.020555555555555556,
"loss": 0.0072,
"step": 1890
},
{
"epoch": 12.67,
"learning_rate": 0.020370370370370372,
"loss": 0.0075,
"step": 1900
},
{
"epoch": 12.73,
"learning_rate": 0.020185185185185188,
"loss": 0.0072,
"step": 1910
},
{
"epoch": 12.8,
"learning_rate": 0.020000000000000004,
"loss": 0.009,
"step": 1920
},
{
"epoch": 12.87,
"learning_rate": 0.019814814814814816,
"loss": 0.0074,
"step": 1930
},
{
"epoch": 12.93,
"learning_rate": 0.019629629629629632,
"loss": 0.0075,
"step": 1940
},
{
"epoch": 13.0,
"learning_rate": 0.019444444444444445,
"loss": 0.0068,
"step": 1950
},
{
"epoch": 13.0,
"eval_acc": 0.926936804534957,
"eval_class_acc": [
0.9629629629629629,
0.9375,
0.6875,
1.0,
0.8666666666666667,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
1.0,
0.9375,
0.875,
0.8571428571428571,
1.0,
1.0,
1.0,
1.0,
0.9230769230769231,
0.9375,
1.0,
0.9333333333333333,
1.0,
0.9333333333333333,
1.0,
1.0
],
"eval_loss": 0.5733085870742798,
"eval_video_acc": 0.958139534883721,
"step": 1950
}
],
"max_steps": 3000,
"num_train_epochs": 20,
"total_flos": 4.96065378739028e+17,
"trial_name": null,
"trial_params": null
}