| { | |
| "best_metric": 0.8571428571428571, | |
| "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": 14.557563781738281, | |
| "learning_rate": 8.333333333333334e-06, | |
| "loss": 2.2978, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.03333333333333333, | |
| "grad_norm": 13.211992263793945, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": 2.2533, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 13.925783157348633, | |
| "learning_rate": 2.5e-05, | |
| "loss": 2.2775, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.06666666666666667, | |
| "grad_norm": 14.46908950805664, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 2.1786, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.08333333333333333, | |
| "grad_norm": 14.618884086608887, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 2.1506, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 13.801518440246582, | |
| "learning_rate": 5e-05, | |
| "loss": 2.2209, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.11666666666666667, | |
| "grad_norm": 17.560274124145508, | |
| "learning_rate": 4.9074074074074075e-05, | |
| "loss": 2.0931, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.13333333333333333, | |
| "grad_norm": 16.784074783325195, | |
| "learning_rate": 4.814814814814815e-05, | |
| "loss": 1.8886, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 14.860028266906738, | |
| "learning_rate": 4.722222222222222e-05, | |
| "loss": 1.8185, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.16666666666666666, | |
| "grad_norm": 19.40709114074707, | |
| "learning_rate": 4.62962962962963e-05, | |
| "loss": 1.8849, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.18333333333333332, | |
| "grad_norm": 15.577845573425293, | |
| "learning_rate": 4.5370370370370374e-05, | |
| "loss": 1.5716, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 10.431912422180176, | |
| "learning_rate": 4.4444444444444447e-05, | |
| "loss": 1.7213, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.21666666666666667, | |
| "grad_norm": 12.733132362365723, | |
| "learning_rate": 4.351851851851852e-05, | |
| "loss": 1.5038, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.23333333333333334, | |
| "grad_norm": 36.75209045410156, | |
| "learning_rate": 4.259259259259259e-05, | |
| "loss": 1.7559, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 19.06173324584961, | |
| "learning_rate": 4.166666666666667e-05, | |
| "loss": 1.2581, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "eval_accuracy": 0.6857142857142857, | |
| "eval_loss": 1.0654476881027222, | |
| "eval_runtime": 9.7584, | |
| "eval_samples_per_second": 7.173, | |
| "eval_steps_per_second": 3.587, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.0166666666666666, | |
| "grad_norm": 18.379487991333008, | |
| "learning_rate": 4.074074074074074e-05, | |
| "loss": 1.0356, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.0333333333333334, | |
| "grad_norm": 18.300090789794922, | |
| "learning_rate": 3.981481481481482e-05, | |
| "loss": 1.1754, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.05, | |
| "grad_norm": 19.084692001342773, | |
| "learning_rate": 3.888888888888889e-05, | |
| "loss": 1.0393, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.0666666666666667, | |
| "grad_norm": 7.776337146759033, | |
| "learning_rate": 3.7962962962962964e-05, | |
| "loss": 0.8038, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.0833333333333333, | |
| "grad_norm": 26.543712615966797, | |
| "learning_rate": 3.7037037037037037e-05, | |
| "loss": 1.1244, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "grad_norm": 18.047842025756836, | |
| "learning_rate": 3.611111111111111e-05, | |
| "loss": 0.7, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.1166666666666667, | |
| "grad_norm": 12.758904457092285, | |
| "learning_rate": 3.518518518518519e-05, | |
| "loss": 0.9777, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.1333333333333333, | |
| "grad_norm": 26.797882080078125, | |
| "learning_rate": 3.425925925925926e-05, | |
| "loss": 0.8592, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "grad_norm": 3.118074655532837, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": 0.6271, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.1666666666666667, | |
| "grad_norm": 41.11469650268555, | |
| "learning_rate": 3.240740740740741e-05, | |
| "loss": 0.8166, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.1833333333333333, | |
| "grad_norm": 15.812960624694824, | |
| "learning_rate": 3.148148148148148e-05, | |
| "loss": 0.7833, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "grad_norm": 27.69546890258789, | |
| "learning_rate": 3.055555555555556e-05, | |
| "loss": 0.6339, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 1.2166666666666668, | |
| "grad_norm": 22.310779571533203, | |
| "learning_rate": 2.962962962962963e-05, | |
| "loss": 0.1696, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 1.2333333333333334, | |
| "grad_norm": 11.531342506408691, | |
| "learning_rate": 2.8703703703703706e-05, | |
| "loss": 1.1017, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 1.763854742050171, | |
| "learning_rate": 2.777777777777778e-05, | |
| "loss": 0.8167, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "eval_accuracy": 0.6571428571428571, | |
| "eval_loss": 0.7648990154266357, | |
| "eval_runtime": 9.8899, | |
| "eval_samples_per_second": 7.078, | |
| "eval_steps_per_second": 3.539, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.0166666666666666, | |
| "grad_norm": 4.815032958984375, | |
| "learning_rate": 2.6851851851851855e-05, | |
| "loss": 0.4077, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.033333333333333, | |
| "grad_norm": 1.701940894126892, | |
| "learning_rate": 2.5925925925925925e-05, | |
| "loss": 0.2598, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.05, | |
| "grad_norm": 10.027705192565918, | |
| "learning_rate": 2.5e-05, | |
| "loss": 0.3591, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 2.066666666666667, | |
| "grad_norm": 46.323997497558594, | |
| "learning_rate": 2.4074074074074074e-05, | |
| "loss": 0.3167, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.0833333333333335, | |
| "grad_norm": 4.542367935180664, | |
| "learning_rate": 2.314814814814815e-05, | |
| "loss": 0.3251, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "grad_norm": 6.4903883934021, | |
| "learning_rate": 2.2222222222222223e-05, | |
| "loss": 0.3563, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 2.1166666666666667, | |
| "grad_norm": 19.011329650878906, | |
| "learning_rate": 2.1296296296296296e-05, | |
| "loss": 0.5848, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 2.1333333333333333, | |
| "grad_norm": 4.820107936859131, | |
| "learning_rate": 2.037037037037037e-05, | |
| "loss": 0.1668, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 2.15, | |
| "grad_norm": 0.16449818015098572, | |
| "learning_rate": 1.9444444444444445e-05, | |
| "loss": 0.299, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 2.1666666666666665, | |
| "grad_norm": 2.329817533493042, | |
| "learning_rate": 1.8518518518518518e-05, | |
| "loss": 0.3169, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 2.183333333333333, | |
| "grad_norm": 2.8288075923919678, | |
| "learning_rate": 1.7592592592592595e-05, | |
| "loss": 0.0671, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 2.2, | |
| "grad_norm": 0.8677788972854614, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": 0.215, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 2.216666666666667, | |
| "grad_norm": 2.6000938415527344, | |
| "learning_rate": 1.574074074074074e-05, | |
| "loss": 0.6017, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 2.2333333333333334, | |
| "grad_norm": 0.7087007761001587, | |
| "learning_rate": 1.4814814814814815e-05, | |
| "loss": 0.0532, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "grad_norm": 19.552764892578125, | |
| "learning_rate": 1.388888888888889e-05, | |
| "loss": 0.1969, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "eval_accuracy": 0.8142857142857143, | |
| "eval_loss": 0.35542502999305725, | |
| "eval_runtime": 9.9338, | |
| "eval_samples_per_second": 7.047, | |
| "eval_steps_per_second": 3.523, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 3.0166666666666666, | |
| "grad_norm": 0.1819472312927246, | |
| "learning_rate": 1.2962962962962962e-05, | |
| "loss": 0.0727, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 3.033333333333333, | |
| "grad_norm": 1.34665048122406, | |
| "learning_rate": 1.2037037037037037e-05, | |
| "loss": 0.0592, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 3.05, | |
| "grad_norm": 0.10649258643388748, | |
| "learning_rate": 1.1111111111111112e-05, | |
| "loss": 0.0577, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 3.066666666666667, | |
| "grad_norm": 0.1528460830450058, | |
| "learning_rate": 1.0185185185185185e-05, | |
| "loss": 0.0405, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 3.0833333333333335, | |
| "grad_norm": 0.16135525703430176, | |
| "learning_rate": 9.259259259259259e-06, | |
| "loss": 0.0678, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.1, | |
| "grad_norm": 10.064668655395508, | |
| "learning_rate": 8.333333333333334e-06, | |
| "loss": 0.1043, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 3.1166666666666667, | |
| "grad_norm": 0.1387307345867157, | |
| "learning_rate": 7.4074074074074075e-06, | |
| "loss": 0.281, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 3.1333333333333333, | |
| "grad_norm": 28.900283813476562, | |
| "learning_rate": 6.481481481481481e-06, | |
| "loss": 0.0535, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 3.15, | |
| "grad_norm": 7.2274556159973145, | |
| "learning_rate": 5.555555555555556e-06, | |
| "loss": 0.1113, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 3.1666666666666665, | |
| "grad_norm": 0.45279788970947266, | |
| "learning_rate": 4.6296296296296296e-06, | |
| "loss": 0.2006, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 3.183333333333333, | |
| "grad_norm": 0.09546133875846863, | |
| "learning_rate": 3.7037037037037037e-06, | |
| "loss": 0.1717, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 3.2, | |
| "grad_norm": 0.3003263473510742, | |
| "learning_rate": 2.777777777777778e-06, | |
| "loss": 0.0947, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 3.216666666666667, | |
| "grad_norm": 0.5243334770202637, | |
| "learning_rate": 1.8518518518518519e-06, | |
| "loss": 0.0705, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 3.2333333333333334, | |
| "grad_norm": 0.14263661205768585, | |
| "learning_rate": 9.259259259259259e-07, | |
| "loss": 0.2079, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "grad_norm": 0.10210184007883072, | |
| "learning_rate": 0.0, | |
| "loss": 0.3976, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "eval_accuracy": 0.8571428571428571, | |
| "eval_loss": 0.30957481265068054, | |
| "eval_runtime": 9.7659, | |
| "eval_samples_per_second": 7.168, | |
| "eval_steps_per_second": 3.584, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "step": 600, | |
| "total_flos": 1.495384188125184e+18, | |
| "train_loss": 0.8009350387255351, | |
| "train_runtime": 409.9163, | |
| "train_samples_per_second": 2.927, | |
| "train_steps_per_second": 1.464 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "eval_accuracy": 0.8903225806451613, | |
| "eval_loss": 0.37351304292678833, | |
| "eval_runtime": 19.9071, | |
| "eval_samples_per_second": 7.786, | |
| "eval_steps_per_second": 3.918, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 3.25, | |
| "eval_accuracy": 0.8903225806451613, | |
| "eval_loss": 0.37351304292678833, | |
| "eval_runtime": 21.1792, | |
| "eval_samples_per_second": 7.318, | |
| "eval_steps_per_second": 3.683, | |
| "step": 600 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 600, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 9223372036854775807, | |
| "save_steps": 500, | |
| "total_flos": 1.495384188125184e+18, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |