|
{ |
|
"best_metric": 0.9, |
|
"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.084315299987793, |
|
"learning_rate": 8.333333333333334e-06, |
|
"loss": 2.2625, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.03333333333333333, |
|
"grad_norm": 12.500144958496094, |
|
"learning_rate": 1.6666666666666667e-05, |
|
"loss": 2.3353, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 13.111361503601074, |
|
"learning_rate": 2.5e-05, |
|
"loss": 2.2527, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.06666666666666667, |
|
"grad_norm": 14.977873802185059, |
|
"learning_rate": 3.3333333333333335e-05, |
|
"loss": 2.2332, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.08333333333333333, |
|
"grad_norm": 18.667680740356445, |
|
"learning_rate": 4.166666666666667e-05, |
|
"loss": 2.1483, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 14.930619239807129, |
|
"learning_rate": 5e-05, |
|
"loss": 2.2316, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.11666666666666667, |
|
"grad_norm": 17.389162063598633, |
|
"learning_rate": 4.9074074074074075e-05, |
|
"loss": 2.0506, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.13333333333333333, |
|
"grad_norm": 19.799802780151367, |
|
"learning_rate": 4.814814814814815e-05, |
|
"loss": 1.7776, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 15.30734920501709, |
|
"learning_rate": 4.722222222222222e-05, |
|
"loss": 1.817, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.16666666666666666, |
|
"grad_norm": 17.61277198791504, |
|
"learning_rate": 4.62962962962963e-05, |
|
"loss": 1.7994, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.18333333333333332, |
|
"grad_norm": 9.471660614013672, |
|
"learning_rate": 4.5370370370370374e-05, |
|
"loss": 1.3306, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 4.7816081047058105, |
|
"learning_rate": 4.4444444444444447e-05, |
|
"loss": 1.6416, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.21666666666666667, |
|
"grad_norm": 21.287919998168945, |
|
"learning_rate": 4.351851851851852e-05, |
|
"loss": 1.6549, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.23333333333333334, |
|
"grad_norm": 16.014236450195312, |
|
"learning_rate": 4.259259259259259e-05, |
|
"loss": 1.5492, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 14.71437931060791, |
|
"learning_rate": 4.166666666666667e-05, |
|
"loss": 1.3234, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"eval_accuracy": 0.6142857142857143, |
|
"eval_loss": 1.1605435609817505, |
|
"eval_runtime": 18.4031, |
|
"eval_samples_per_second": 3.804, |
|
"eval_steps_per_second": 1.902, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.0166666666666666, |
|
"grad_norm": 14.739291191101074, |
|
"learning_rate": 4.074074074074074e-05, |
|
"loss": 0.8406, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.0333333333333334, |
|
"grad_norm": 20.024723052978516, |
|
"learning_rate": 3.981481481481482e-05, |
|
"loss": 1.1478, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.05, |
|
"grad_norm": 64.58138275146484, |
|
"learning_rate": 3.888888888888889e-05, |
|
"loss": 1.0488, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.0666666666666667, |
|
"grad_norm": 12.84802532196045, |
|
"learning_rate": 3.7962962962962964e-05, |
|
"loss": 0.8007, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 1.0833333333333333, |
|
"grad_norm": 27.170969009399414, |
|
"learning_rate": 3.7037037037037037e-05, |
|
"loss": 1.2599, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"grad_norm": 14.179140090942383, |
|
"learning_rate": 3.611111111111111e-05, |
|
"loss": 0.6757, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 1.1166666666666667, |
|
"grad_norm": 26.276153564453125, |
|
"learning_rate": 3.518518518518519e-05, |
|
"loss": 0.8593, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 1.1333333333333333, |
|
"grad_norm": 3.5043022632598877, |
|
"learning_rate": 3.425925925925926e-05, |
|
"loss": 1.0581, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 1.15, |
|
"grad_norm": 12.9857177734375, |
|
"learning_rate": 3.3333333333333335e-05, |
|
"loss": 0.8675, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 1.1666666666666667, |
|
"grad_norm": 23.625690460205078, |
|
"learning_rate": 3.240740740740741e-05, |
|
"loss": 0.4225, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 1.1833333333333333, |
|
"grad_norm": 33.589900970458984, |
|
"learning_rate": 3.148148148148148e-05, |
|
"loss": 0.6679, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"grad_norm": 77.1363296508789, |
|
"learning_rate": 3.055555555555556e-05, |
|
"loss": 0.871, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 1.2166666666666668, |
|
"grad_norm": 26.5102481842041, |
|
"learning_rate": 2.962962962962963e-05, |
|
"loss": 0.345, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 1.2333333333333334, |
|
"grad_norm": 97.94902801513672, |
|
"learning_rate": 2.8703703703703706e-05, |
|
"loss": 0.9062, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"grad_norm": 4.820228576660156, |
|
"learning_rate": 2.777777777777778e-05, |
|
"loss": 1.0245, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"eval_accuracy": 0.6714285714285714, |
|
"eval_loss": 0.776366114616394, |
|
"eval_runtime": 18.2334, |
|
"eval_samples_per_second": 3.839, |
|
"eval_steps_per_second": 1.92, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 2.0166666666666666, |
|
"grad_norm": 5.84061861038208, |
|
"learning_rate": 2.6851851851851855e-05, |
|
"loss": 0.3563, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 2.033333333333333, |
|
"grad_norm": 7.712353229522705, |
|
"learning_rate": 2.5925925925925925e-05, |
|
"loss": 0.2456, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 2.05, |
|
"grad_norm": 10.726452827453613, |
|
"learning_rate": 2.5e-05, |
|
"loss": 0.4232, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 2.066666666666667, |
|
"grad_norm": 9.719975471496582, |
|
"learning_rate": 2.4074074074074074e-05, |
|
"loss": 0.1975, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 2.0833333333333335, |
|
"grad_norm": 3.219958543777466, |
|
"learning_rate": 2.314814814814815e-05, |
|
"loss": 0.2796, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 2.1, |
|
"grad_norm": 4.559834957122803, |
|
"learning_rate": 2.2222222222222223e-05, |
|
"loss": 0.3343, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 2.1166666666666667, |
|
"grad_norm": 5.237907409667969, |
|
"learning_rate": 2.1296296296296296e-05, |
|
"loss": 0.2744, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 2.1333333333333333, |
|
"grad_norm": 1.3404731750488281, |
|
"learning_rate": 2.037037037037037e-05, |
|
"loss": 0.2313, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 2.15, |
|
"grad_norm": 0.18039122223854065, |
|
"learning_rate": 1.9444444444444445e-05, |
|
"loss": 0.4182, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 2.1666666666666665, |
|
"grad_norm": 0.4312891960144043, |
|
"learning_rate": 1.8518518518518518e-05, |
|
"loss": 0.2587, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 2.183333333333333, |
|
"grad_norm": 2.762604236602783, |
|
"learning_rate": 1.7592592592592595e-05, |
|
"loss": 0.0869, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"grad_norm": 16.173503875732422, |
|
"learning_rate": 1.6666666666666667e-05, |
|
"loss": 0.0854, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 2.216666666666667, |
|
"grad_norm": 4.518730640411377, |
|
"learning_rate": 1.574074074074074e-05, |
|
"loss": 0.2236, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 2.2333333333333334, |
|
"grad_norm": 0.27387773990631104, |
|
"learning_rate": 1.4814814814814815e-05, |
|
"loss": 0.0327, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 2.25, |
|
"grad_norm": 25.623882293701172, |
|
"learning_rate": 1.388888888888889e-05, |
|
"loss": 0.1441, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 2.25, |
|
"eval_accuracy": 0.8571428571428571, |
|
"eval_loss": 0.4293404221534729, |
|
"eval_runtime": 18.5539, |
|
"eval_samples_per_second": 3.773, |
|
"eval_steps_per_second": 1.886, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 3.0166666666666666, |
|
"grad_norm": 0.1813192516565323, |
|
"learning_rate": 1.2962962962962962e-05, |
|
"loss": 0.0688, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 3.033333333333333, |
|
"grad_norm": 1.2744430303573608, |
|
"learning_rate": 1.2037037037037037e-05, |
|
"loss": 0.0496, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"grad_norm": 0.08318241685628891, |
|
"learning_rate": 1.1111111111111112e-05, |
|
"loss": 0.0262, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 3.066666666666667, |
|
"grad_norm": 0.11013840138912201, |
|
"learning_rate": 1.0185185185185185e-05, |
|
"loss": 0.0512, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 3.0833333333333335, |
|
"grad_norm": 0.11020367592573166, |
|
"learning_rate": 9.259259259259259e-06, |
|
"loss": 0.076, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 3.1, |
|
"grad_norm": 4.4319586753845215, |
|
"learning_rate": 8.333333333333334e-06, |
|
"loss": 0.2574, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 3.1166666666666667, |
|
"grad_norm": 0.10417469590902328, |
|
"learning_rate": 7.4074074074074075e-06, |
|
"loss": 0.0439, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 3.1333333333333333, |
|
"grad_norm": 53.32948684692383, |
|
"learning_rate": 6.481481481481481e-06, |
|
"loss": 0.1041, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 3.15, |
|
"grad_norm": 0.8091713786125183, |
|
"learning_rate": 5.555555555555556e-06, |
|
"loss": 0.1931, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 3.1666666666666665, |
|
"grad_norm": 0.8358808159828186, |
|
"learning_rate": 4.6296296296296296e-06, |
|
"loss": 0.2557, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 3.183333333333333, |
|
"grad_norm": 0.0853860154747963, |
|
"learning_rate": 3.7037037037037037e-06, |
|
"loss": 0.1618, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 3.2, |
|
"grad_norm": 0.3539259135723114, |
|
"learning_rate": 2.777777777777778e-06, |
|
"loss": 0.3177, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 3.216666666666667, |
|
"grad_norm": 1.287772297859192, |
|
"learning_rate": 1.8518518518518519e-06, |
|
"loss": 0.124, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 3.2333333333333334, |
|
"grad_norm": 0.10092153400182724, |
|
"learning_rate": 9.259259259259259e-07, |
|
"loss": 0.2376, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"grad_norm": 0.5576777458190918, |
|
"learning_rate": 0.0, |
|
"loss": 0.4346, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"eval_accuracy": 0.9, |
|
"eval_loss": 0.22466129064559937, |
|
"eval_runtime": 18.3475, |
|
"eval_samples_per_second": 3.815, |
|
"eval_steps_per_second": 1.908, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"step": 600, |
|
"total_flos": 1.495384188125184e+18, |
|
"train_loss": 0.7866153143843015, |
|
"train_runtime": 807.4229, |
|
"train_samples_per_second": 1.486, |
|
"train_steps_per_second": 0.743 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"eval_accuracy": 0.9161290322580645, |
|
"eval_loss": 0.2788711488246918, |
|
"eval_runtime": 41.526, |
|
"eval_samples_per_second": 3.733, |
|
"eval_steps_per_second": 1.878, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"eval_accuracy": 0.9161290322580645, |
|
"eval_loss": 0.2788711488246918, |
|
"eval_runtime": 41.4461, |
|
"eval_samples_per_second": 3.74, |
|
"eval_steps_per_second": 1.882, |
|
"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 |
|
} |
|
|