| { | |
| "best_metric": 0.9571428571428572, | |
| "best_model_checkpoint": "videomae-base-finetuned-ucf101-subset/checkpoint-525", | |
| "epoch": 9.1, | |
| "eval_steps": 500, | |
| "global_step": 750, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.013333333333333334, | |
| "grad_norm": 8.795476913452148, | |
| "learning_rate": 6.666666666666667e-06, | |
| "loss": 2.4054, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.02666666666666667, | |
| "grad_norm": 10.026263236999512, | |
| "learning_rate": 1.3333333333333333e-05, | |
| "loss": 2.4288, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 9.563196182250977, | |
| "learning_rate": 2e-05, | |
| "loss": 2.2903, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.05333333333333334, | |
| "grad_norm": 9.015666961669922, | |
| "learning_rate": 2.6666666666666667e-05, | |
| "loss": 2.3069, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.06666666666666667, | |
| "grad_norm": 11.751035690307617, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 2.2176, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 10.45116138458252, | |
| "learning_rate": 4e-05, | |
| "loss": 2.0852, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.09333333333333334, | |
| "grad_norm": 12.927900314331055, | |
| "learning_rate": 4.666666666666667e-05, | |
| "loss": 2.1525, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "eval_accuracy": 0.5428571428571428, | |
| "eval_loss": 1.722894549369812, | |
| "eval_runtime": 19.7611, | |
| "eval_samples_per_second": 3.542, | |
| "eval_steps_per_second": 0.911, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.0066666666666666, | |
| "grad_norm": 11.718605041503906, | |
| "learning_rate": 4.962962962962963e-05, | |
| "loss": 1.7414, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.02, | |
| "grad_norm": 14.952427864074707, | |
| "learning_rate": 4.888888888888889e-05, | |
| "loss": 1.4744, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.0333333333333334, | |
| "grad_norm": 13.099126815795898, | |
| "learning_rate": 4.814814814814815e-05, | |
| "loss": 1.3828, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.0466666666666666, | |
| "grad_norm": 7.654531478881836, | |
| "learning_rate": 4.740740740740741e-05, | |
| "loss": 1.0222, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 1.06, | |
| "grad_norm": 4.138798236846924, | |
| "learning_rate": 4.666666666666667e-05, | |
| "loss": 0.9468, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.0733333333333333, | |
| "grad_norm": 23.42486572265625, | |
| "learning_rate": 4.592592592592593e-05, | |
| "loss": 0.6556, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.0866666666666667, | |
| "grad_norm": 9.991631507873535, | |
| "learning_rate": 4.518518518518519e-05, | |
| "loss": 0.515, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "grad_norm": 10.264222145080566, | |
| "learning_rate": 4.4444444444444447e-05, | |
| "loss": 1.028, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "eval_accuracy": 0.7571428571428571, | |
| "eval_loss": 0.718971848487854, | |
| "eval_runtime": 20.4493, | |
| "eval_samples_per_second": 3.423, | |
| "eval_steps_per_second": 0.88, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 2.013333333333333, | |
| "grad_norm": 1.5132960081100464, | |
| "learning_rate": 4.3703703703703705e-05, | |
| "loss": 0.393, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 2.026666666666667, | |
| "grad_norm": 28.45006561279297, | |
| "learning_rate": 4.296296296296296e-05, | |
| "loss": 0.3865, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 2.04, | |
| "grad_norm": 3.5434415340423584, | |
| "learning_rate": 4.222222222222222e-05, | |
| "loss": 0.4954, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 2.0533333333333332, | |
| "grad_norm": 20.566238403320312, | |
| "learning_rate": 4.148148148148148e-05, | |
| "loss": 0.5306, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 2.066666666666667, | |
| "grad_norm": 4.603804111480713, | |
| "learning_rate": 4.074074074074074e-05, | |
| "loss": 0.5211, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 2.08, | |
| "grad_norm": 67.62308502197266, | |
| "learning_rate": 4e-05, | |
| "loss": 0.4294, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 2.0933333333333333, | |
| "grad_norm": 4.679758548736572, | |
| "learning_rate": 3.925925925925926e-05, | |
| "loss": 0.3541, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "eval_accuracy": 0.8714285714285714, | |
| "eval_loss": 0.30722978711128235, | |
| "eval_runtime": 20.3903, | |
| "eval_samples_per_second": 3.433, | |
| "eval_steps_per_second": 0.883, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 3.006666666666667, | |
| "grad_norm": 1.4587314128875732, | |
| "learning_rate": 3.851851851851852e-05, | |
| "loss": 0.1373, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 3.02, | |
| "grad_norm": 0.4367457628250122, | |
| "learning_rate": 3.777777777777778e-05, | |
| "loss": 0.0492, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 3.033333333333333, | |
| "grad_norm": 65.70573425292969, | |
| "learning_rate": 3.7037037037037037e-05, | |
| "loss": 0.1173, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 3.046666666666667, | |
| "grad_norm": 0.18782749772071838, | |
| "learning_rate": 3.62962962962963e-05, | |
| "loss": 0.3055, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 3.06, | |
| "grad_norm": 15.650634765625, | |
| "learning_rate": 3.555555555555556e-05, | |
| "loss": 0.3789, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 3.0733333333333333, | |
| "grad_norm": 0.7454498410224915, | |
| "learning_rate": 3.481481481481482e-05, | |
| "loss": 0.483, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 3.086666666666667, | |
| "grad_norm": 0.5320239067077637, | |
| "learning_rate": 3.4074074074074077e-05, | |
| "loss": 0.3415, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 3.1, | |
| "grad_norm": 45.54346466064453, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 0.6501, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 3.1, | |
| "eval_accuracy": 0.7857142857142857, | |
| "eval_loss": 0.5646315217018127, | |
| "eval_runtime": 19.9887, | |
| "eval_samples_per_second": 3.502, | |
| "eval_steps_per_second": 0.901, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 4.013333333333334, | |
| "grad_norm": 0.2744936943054199, | |
| "learning_rate": 3.25925925925926e-05, | |
| "loss": 0.1115, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 4.026666666666666, | |
| "grad_norm": 52.5593376159668, | |
| "learning_rate": 3.185185185185185e-05, | |
| "loss": 0.4402, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 4.04, | |
| "grad_norm": 0.3685360252857208, | |
| "learning_rate": 3.111111111111111e-05, | |
| "loss": 0.1409, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 4.053333333333334, | |
| "grad_norm": 0.13833816349506378, | |
| "learning_rate": 3.037037037037037e-05, | |
| "loss": 0.2763, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 4.066666666666666, | |
| "grad_norm": 0.11666622757911682, | |
| "learning_rate": 2.962962962962963e-05, | |
| "loss": 0.0503, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 4.08, | |
| "grad_norm": 0.06857863068580627, | |
| "learning_rate": 2.8888888888888888e-05, | |
| "loss": 0.1914, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 4.093333333333334, | |
| "grad_norm": 3.4672675132751465, | |
| "learning_rate": 2.814814814814815e-05, | |
| "loss": 0.0448, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 4.1, | |
| "eval_accuracy": 0.8571428571428571, | |
| "eval_loss": 0.44399380683898926, | |
| "eval_runtime": 17.9423, | |
| "eval_samples_per_second": 3.901, | |
| "eval_steps_per_second": 1.003, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 5.006666666666667, | |
| "grad_norm": 1.6939857006072998, | |
| "learning_rate": 2.7407407407407408e-05, | |
| "loss": 0.0308, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 5.02, | |
| "grad_norm": 1.3869863748550415, | |
| "learning_rate": 2.6666666666666667e-05, | |
| "loss": 0.1281, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 5.033333333333333, | |
| "grad_norm": 0.26335254311561584, | |
| "learning_rate": 2.5925925925925925e-05, | |
| "loss": 0.0254, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 5.046666666666667, | |
| "grad_norm": 0.060862887650728226, | |
| "learning_rate": 2.5185185185185183e-05, | |
| "loss": 0.0119, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 5.06, | |
| "grad_norm": 0.22715356945991516, | |
| "learning_rate": 2.4444444444444445e-05, | |
| "loss": 0.1588, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 5.073333333333333, | |
| "grad_norm": 61.806556701660156, | |
| "learning_rate": 2.3703703703703707e-05, | |
| "loss": 0.0864, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 5.086666666666667, | |
| "grad_norm": 0.05195135250687599, | |
| "learning_rate": 2.2962962962962965e-05, | |
| "loss": 0.073, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 5.1, | |
| "grad_norm": 43.471065521240234, | |
| "learning_rate": 2.2222222222222223e-05, | |
| "loss": 0.0785, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 5.1, | |
| "eval_accuracy": 0.9, | |
| "eval_loss": 0.3218167722225189, | |
| "eval_runtime": 19.9327, | |
| "eval_samples_per_second": 3.512, | |
| "eval_steps_per_second": 0.903, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 6.013333333333334, | |
| "grad_norm": 0.06781550496816635, | |
| "learning_rate": 2.148148148148148e-05, | |
| "loss": 0.1588, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 6.026666666666666, | |
| "grad_norm": 0.09493797272443771, | |
| "learning_rate": 2.074074074074074e-05, | |
| "loss": 0.0833, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 6.04, | |
| "grad_norm": 1.557625412940979, | |
| "learning_rate": 2e-05, | |
| "loss": 0.038, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 6.053333333333334, | |
| "grad_norm": 0.041507575660943985, | |
| "learning_rate": 1.925925925925926e-05, | |
| "loss": 0.0782, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 6.066666666666666, | |
| "grad_norm": 0.13706688582897186, | |
| "learning_rate": 1.8518518518518518e-05, | |
| "loss": 0.0061, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 6.08, | |
| "grad_norm": 0.04828852787613869, | |
| "learning_rate": 1.777777777777778e-05, | |
| "loss": 0.005, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 6.093333333333334, | |
| "grad_norm": 0.047101154923439026, | |
| "learning_rate": 1.7037037037037038e-05, | |
| "loss": 0.0214, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 6.1, | |
| "eval_accuracy": 0.9571428571428572, | |
| "eval_loss": 0.1372106671333313, | |
| "eval_runtime": 19.7597, | |
| "eval_samples_per_second": 3.543, | |
| "eval_steps_per_second": 0.911, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 7.006666666666667, | |
| "grad_norm": 0.18555110692977905, | |
| "learning_rate": 1.62962962962963e-05, | |
| "loss": 0.0045, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 7.02, | |
| "grad_norm": 0.11104493588209152, | |
| "learning_rate": 1.5555555555555555e-05, | |
| "loss": 0.0273, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 7.033333333333333, | |
| "grad_norm": 0.04018757492303848, | |
| "learning_rate": 1.4814814814814815e-05, | |
| "loss": 0.1058, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 7.046666666666667, | |
| "grad_norm": 0.04107681289315224, | |
| "learning_rate": 1.4074074074074075e-05, | |
| "loss": 0.0064, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 7.06, | |
| "grad_norm": 0.05029289796948433, | |
| "learning_rate": 1.3333333333333333e-05, | |
| "loss": 0.0708, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 7.073333333333333, | |
| "grad_norm": 0.046671733260154724, | |
| "learning_rate": 1.2592592592592592e-05, | |
| "loss": 0.0056, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 7.086666666666667, | |
| "grad_norm": 0.05804697796702385, | |
| "learning_rate": 1.1851851851851853e-05, | |
| "loss": 0.012, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 7.1, | |
| "grad_norm": 0.04299278184771538, | |
| "learning_rate": 1.1111111111111112e-05, | |
| "loss": 0.0039, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 7.1, | |
| "eval_accuracy": 0.9571428571428572, | |
| "eval_loss": 0.10343696922063828, | |
| "eval_runtime": 21.2334, | |
| "eval_samples_per_second": 3.297, | |
| "eval_steps_per_second": 0.848, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 8.013333333333334, | |
| "grad_norm": 0.03256387636065483, | |
| "learning_rate": 1.037037037037037e-05, | |
| "loss": 0.0064, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 8.026666666666667, | |
| "grad_norm": 0.0486493855714798, | |
| "learning_rate": 9.62962962962963e-06, | |
| "loss": 0.0037, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 8.04, | |
| "grad_norm": 0.03539065271615982, | |
| "learning_rate": 8.88888888888889e-06, | |
| "loss": 0.1036, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 8.053333333333333, | |
| "grad_norm": 0.030678970739245415, | |
| "learning_rate": 8.14814814814815e-06, | |
| "loss": 0.0034, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 8.066666666666666, | |
| "grad_norm": 0.03648629039525986, | |
| "learning_rate": 7.4074074074074075e-06, | |
| "loss": 0.0557, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 8.08, | |
| "grad_norm": 0.036471571773290634, | |
| "learning_rate": 6.666666666666667e-06, | |
| "loss": 0.0984, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 8.093333333333334, | |
| "grad_norm": 0.05204283446073532, | |
| "learning_rate": 5.925925925925927e-06, | |
| "loss": 0.0048, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 8.1, | |
| "eval_accuracy": 0.9142857142857143, | |
| "eval_loss": 0.15946045517921448, | |
| "eval_runtime": 18.3528, | |
| "eval_samples_per_second": 3.814, | |
| "eval_steps_per_second": 0.981, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 9.006666666666666, | |
| "grad_norm": 0.0372748076915741, | |
| "learning_rate": 5.185185185185185e-06, | |
| "loss": 0.0034, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 9.02, | |
| "grad_norm": 0.043480608612298965, | |
| "learning_rate": 4.444444444444445e-06, | |
| "loss": 0.0033, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 9.033333333333333, | |
| "grad_norm": 0.034943871200084686, | |
| "learning_rate": 3.7037037037037037e-06, | |
| "loss": 0.0034, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 9.046666666666667, | |
| "grad_norm": 0.05171409249305725, | |
| "learning_rate": 2.9629629629629633e-06, | |
| "loss": 0.0036, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 9.06, | |
| "grad_norm": 0.035231392830610275, | |
| "learning_rate": 2.2222222222222225e-06, | |
| "loss": 0.0031, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 9.073333333333334, | |
| "grad_norm": 0.06839217245578766, | |
| "learning_rate": 1.4814814814814817e-06, | |
| "loss": 0.0111, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 9.086666666666666, | |
| "grad_norm": 0.032866183668375015, | |
| "learning_rate": 7.407407407407408e-07, | |
| "loss": 0.0041, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 9.1, | |
| "grad_norm": 0.06153057515621185, | |
| "learning_rate": 0.0, | |
| "loss": 0.0052, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 9.1, | |
| "eval_accuracy": 0.9571428571428572, | |
| "eval_loss": 0.10069848597049713, | |
| "eval_runtime": 20.7858, | |
| "eval_samples_per_second": 3.368, | |
| "eval_steps_per_second": 0.866, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 9.1, | |
| "step": 750, | |
| "total_flos": 3.73846047031296e+18, | |
| "train_loss": 0.4401957253267368, | |
| "train_runtime": 1973.9568, | |
| "train_samples_per_second": 1.52, | |
| "train_steps_per_second": 0.38 | |
| }, | |
| { | |
| "epoch": 9.1, | |
| "eval_accuracy": 0.9225806451612903, | |
| "eval_loss": 0.2622383236885071, | |
| "eval_runtime": 45.5213, | |
| "eval_samples_per_second": 3.405, | |
| "eval_steps_per_second": 0.857, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 9.1, | |
| "eval_accuracy": 0.9225806451612903, | |
| "eval_loss": 0.2622382938861847, | |
| "eval_runtime": 44.9977, | |
| "eval_samples_per_second": 3.445, | |
| "eval_steps_per_second": 0.867, | |
| "step": 750 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 750, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 9223372036854775807, | |
| "save_steps": 500, | |
| "total_flos": 3.73846047031296e+18, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |