| { | |
| "best_metric": 0.9142857142857143, | |
| "best_model_checkpoint": "videomae-base-finetuned-ucf101-subset/checkpoint-600", | |
| "epoch": 3.25, | |
| "eval_steps": 500, | |
| "global_step": 600, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.016666666666666666, | |
| "grad_norm": 13.878811836242676, | |
| "learning_rate": 8.333333333333334e-06, | |
| "loss": 2.3864, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.03333333333333333, | |
| "grad_norm": 12.958414077758789, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": 2.2972, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 13.340024948120117, | |
| "learning_rate": 2.5e-05, | |
| "loss": 2.3691, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.06666666666666667, | |
| "grad_norm": 14.514385223388672, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 2.2344, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.08333333333333333, | |
| "grad_norm": 14.681234359741211, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 2.1617, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 15.572694778442383, | |
| "learning_rate": 5e-05, | |
| "loss": 2.2188, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.11666666666666667, | |
| "grad_norm": 16.609718322753906, | |
| "learning_rate": 4.9074074074074075e-05, | |
| "loss": 2.0842, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.13333333333333333, | |
| "grad_norm": 16.88509750366211, | |
| "learning_rate": 4.814814814814815e-05, | |
| "loss": 1.8851, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 16.43447494506836, | |
| "learning_rate": 4.722222222222222e-05, | |
| "loss": 2.1666, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.16666666666666666, | |
| "grad_norm": 16.14389419555664, | |
| "learning_rate": 4.62962962962963e-05, | |
| "loss": 2.0483, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.18333333333333332, | |
| "grad_norm": 12.657491683959961, | |
| "learning_rate": 4.5370370370370374e-05, | |
| "loss": 1.9418, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 17.41259002685547, | |
| "learning_rate": 4.4444444444444447e-05, | |
| "loss": 1.869, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.21666666666666667, | |
| "grad_norm": 28.190662384033203, | |
| "learning_rate": 4.351851851851852e-05, | |
| "loss": 1.6983, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.23333333333333334, | |
| "grad_norm": 15.854883193969727, | |
| "learning_rate": 4.259259259259259e-05, | |
| "loss": 1.9582, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 12.764759063720703, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 1.3516, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "eval_accuracy": 0.5571428571428572, | |
| "eval_loss": 1.3335753679275513, | |
| "eval_runtime": 21.7132, | |
| "eval_samples_per_second": 3.224, | |
| "eval_steps_per_second": 1.612, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.0166666666666666, | |
| "grad_norm": 17.67679786682129, | |
| "learning_rate": 4.074074074074074e-05, | |
| "loss": 1.083, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.0333333333333334, | |
| "grad_norm": 28.23631477355957, | |
| "learning_rate": 3.981481481481482e-05, | |
| "loss": 1.5288, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.05, | |
| "grad_norm": 18.444087982177734, | |
| "learning_rate": 3.888888888888889e-05, | |
| "loss": 1.2424, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.0666666666666667, | |
| "grad_norm": 7.792266368865967, | |
| "learning_rate": 3.7962962962962964e-05, | |
| "loss": 1.1327, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.0833333333333333, | |
| "grad_norm": 35.91878890991211, | |
| "learning_rate": 3.7037037037037037e-05, | |
| "loss": 1.3238, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "grad_norm": 14.137838363647461, | |
| "learning_rate": 3.611111111111111e-05, | |
| "loss": 0.8591, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.1166666666666667, | |
| "grad_norm": 14.379003524780273, | |
| "learning_rate": 3.518518518518519e-05, | |
| "loss": 0.8904, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.1333333333333333, | |
| "grad_norm": 35.3349723815918, | |
| "learning_rate": 3.425925925925926e-05, | |
| "loss": 1.0365, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "grad_norm": 4.670235633850098, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 1.0926, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.1666666666666667, | |
| "grad_norm": 24.737224578857422, | |
| "learning_rate": 3.240740740740741e-05, | |
| "loss": 0.6525, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.1833333333333333, | |
| "grad_norm": 7.743138313293457, | |
| "learning_rate": 3.148148148148148e-05, | |
| "loss": 0.6751, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "grad_norm": 19.620027542114258, | |
| "learning_rate": 3.055555555555556e-05, | |
| "loss": 0.9504, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 1.2166666666666668, | |
| "grad_norm": 11.127130508422852, | |
| "learning_rate": 2.962962962962963e-05, | |
| "loss": 0.3769, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 1.2333333333333334, | |
| "grad_norm": 15.655902862548828, | |
| "learning_rate": 2.8703703703703706e-05, | |
| "loss": 1.0, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 3.454685926437378, | |
| "learning_rate": 2.777777777777778e-05, | |
| "loss": 1.0445, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "eval_accuracy": 0.7714285714285715, | |
| "eval_loss": 0.804114580154419, | |
| "eval_runtime": 21.0006, | |
| "eval_samples_per_second": 3.333, | |
| "eval_steps_per_second": 1.667, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.0166666666666666, | |
| "grad_norm": 10.995410919189453, | |
| "learning_rate": 2.6851851851851855e-05, | |
| "loss": 0.4714, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.033333333333333, | |
| "grad_norm": 1.7594047784805298, | |
| "learning_rate": 2.5925925925925925e-05, | |
| "loss": 0.1752, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.05, | |
| "grad_norm": 13.059496879577637, | |
| "learning_rate": 2.5e-05, | |
| "loss": 0.4913, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 2.066666666666667, | |
| "grad_norm": 12.100168228149414, | |
| "learning_rate": 2.4074074074074074e-05, | |
| "loss": 0.2387, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.0833333333333335, | |
| "grad_norm": 14.437403678894043, | |
| "learning_rate": 2.314814814814815e-05, | |
| "loss": 0.6051, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "grad_norm": 18.535968780517578, | |
| "learning_rate": 2.2222222222222223e-05, | |
| "loss": 0.5252, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 2.1166666666666667, | |
| "grad_norm": 27.02277183532715, | |
| "learning_rate": 2.1296296296296296e-05, | |
| "loss": 0.5674, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 2.1333333333333333, | |
| "grad_norm": 2.0069992542266846, | |
| "learning_rate": 2.037037037037037e-05, | |
| "loss": 0.1657, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 2.15, | |
| "grad_norm": 0.19569337368011475, | |
| "learning_rate": 1.9444444444444445e-05, | |
| "loss": 0.461, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 2.1666666666666665, | |
| "grad_norm": 0.5586942434310913, | |
| "learning_rate": 1.8518518518518518e-05, | |
| "loss": 0.4876, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.183333333333333, | |
| "grad_norm": 1.0278728008270264, | |
| "learning_rate": 1.7592592592592595e-05, | |
| "loss": 0.5508, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 2.2, | |
| "grad_norm": 1.1786044836044312, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": 0.1829, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 2.216666666666667, | |
| "grad_norm": 5.859825611114502, | |
| "learning_rate": 1.574074074074074e-05, | |
| "loss": 0.2612, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 2.2333333333333334, | |
| "grad_norm": 0.49297359585762024, | |
| "learning_rate": 1.4814814814814815e-05, | |
| "loss": 0.1112, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "grad_norm": 29.458885192871094, | |
| "learning_rate": 1.388888888888889e-05, | |
| "loss": 0.1752, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "eval_accuracy": 0.7285714285714285, | |
| "eval_loss": 0.7214392423629761, | |
| "eval_runtime": 21.636, | |
| "eval_samples_per_second": 3.235, | |
| "eval_steps_per_second": 1.618, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 3.0166666666666666, | |
| "grad_norm": 0.2373313158750534, | |
| "learning_rate": 1.2962962962962962e-05, | |
| "loss": 0.0994, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 3.033333333333333, | |
| "grad_norm": 3.266122341156006, | |
| "learning_rate": 1.2037037037037037e-05, | |
| "loss": 0.1565, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 3.05, | |
| "grad_norm": 0.10638546943664551, | |
| "learning_rate": 1.1111111111111112e-05, | |
| "loss": 0.0355, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 3.066666666666667, | |
| "grad_norm": 0.10515712946653366, | |
| "learning_rate": 1.0185185185185185e-05, | |
| "loss": 0.066, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 3.0833333333333335, | |
| "grad_norm": 0.4172205328941345, | |
| "learning_rate": 9.259259259259259e-06, | |
| "loss": 0.0395, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.1, | |
| "grad_norm": 38.50326156616211, | |
| "learning_rate": 8.333333333333334e-06, | |
| "loss": 0.3863, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 3.1166666666666667, | |
| "grad_norm": 0.11968211829662323, | |
| "learning_rate": 7.4074074074074075e-06, | |
| "loss": 0.1334, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 3.1333333333333333, | |
| "grad_norm": 7.747541427612305, | |
| "learning_rate": 6.481481481481481e-06, | |
| "loss": 0.0893, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 3.15, | |
| "grad_norm": 0.860350489616394, | |
| "learning_rate": 5.555555555555556e-06, | |
| "loss": 0.3064, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 3.1666666666666665, | |
| "grad_norm": 1.3779197931289673, | |
| "learning_rate": 4.6296296296296296e-06, | |
| "loss": 0.2478, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 3.183333333333333, | |
| "grad_norm": 0.1085130125284195, | |
| "learning_rate": 3.7037037037037037e-06, | |
| "loss": 0.302, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 3.2, | |
| "grad_norm": 0.2623406648635864, | |
| "learning_rate": 2.777777777777778e-06, | |
| "loss": 0.0376, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 3.216666666666667, | |
| "grad_norm": 0.7496708631515503, | |
| "learning_rate": 1.8518518518518519e-06, | |
| "loss": 0.0716, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 3.2333333333333334, | |
| "grad_norm": 0.5140882134437561, | |
| "learning_rate": 9.259259259259259e-07, | |
| "loss": 0.4109, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "grad_norm": 0.12395908683538437, | |
| "learning_rate": 0.0, | |
| "loss": 0.1347, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "eval_accuracy": 0.9142857142857143, | |
| "eval_loss": 0.28625765442848206, | |
| "eval_runtime": 22.6931, | |
| "eval_samples_per_second": 3.085, | |
| "eval_steps_per_second": 1.542, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "step": 600, | |
| "total_flos": 1.495384188125184e+18, | |
| "train_loss": 0.8924319915970167, | |
| "train_runtime": 896.196, | |
| "train_samples_per_second": 1.339, | |
| "train_steps_per_second": 0.669 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "eval_accuracy": 0.9032258064516129, | |
| "eval_loss": 0.3167099952697754, | |
| "eval_runtime": 48.193, | |
| "eval_samples_per_second": 3.216, | |
| "eval_steps_per_second": 1.618, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "eval_accuracy": 0.9032258064516129, | |
| "eval_loss": 0.3167099952697754, | |
| "eval_runtime": 46.9133, | |
| "eval_samples_per_second": 3.304, | |
| "eval_steps_per_second": 1.663, | |
| "step": 600 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 600, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 9223372036854775807, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.495384188125184e+18, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |