|
{ |
|
"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": 14.668457984924316, |
|
"learning_rate": 8.333333333333334e-06, |
|
"loss": 2.3218, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.03333333333333333, |
|
"grad_norm": 12.71061897277832, |
|
"learning_rate": 1.6666666666666667e-05, |
|
"loss": 2.3311, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 14.201922416687012, |
|
"learning_rate": 2.5e-05, |
|
"loss": 2.3554, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.06666666666666667, |
|
"grad_norm": 14.50876522064209, |
|
"learning_rate": 3.3333333333333335e-05, |
|
"loss": 2.229, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.08333333333333333, |
|
"grad_norm": 17.770288467407227, |
|
"learning_rate": 4.166666666666667e-05, |
|
"loss": 1.9869, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 16.49291229248047, |
|
"learning_rate": 5e-05, |
|
"loss": 2.1513, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.11666666666666667, |
|
"grad_norm": 19.02190399169922, |
|
"learning_rate": 4.9074074074074075e-05, |
|
"loss": 2.1328, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.13333333333333333, |
|
"grad_norm": 20.942832946777344, |
|
"learning_rate": 4.814814814814815e-05, |
|
"loss": 1.8189, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 14.845840454101562, |
|
"learning_rate": 4.722222222222222e-05, |
|
"loss": 1.7776, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.16666666666666666, |
|
"grad_norm": 23.292200088500977, |
|
"learning_rate": 4.62962962962963e-05, |
|
"loss": 1.6319, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.18333333333333332, |
|
"grad_norm": 13.430813789367676, |
|
"learning_rate": 4.5370370370370374e-05, |
|
"loss": 1.4878, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 34.19646453857422, |
|
"learning_rate": 4.4444444444444447e-05, |
|
"loss": 1.5586, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.21666666666666667, |
|
"grad_norm": 6.934573650360107, |
|
"learning_rate": 4.351851851851852e-05, |
|
"loss": 1.3921, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.23333333333333334, |
|
"grad_norm": 53.162132263183594, |
|
"learning_rate": 4.259259259259259e-05, |
|
"loss": 1.1011, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 15.927412033081055, |
|
"learning_rate": 4.166666666666667e-05, |
|
"loss": 1.4636, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"eval_accuracy": 0.7285714285714285, |
|
"eval_loss": 0.9626212120056152, |
|
"eval_runtime": 19.2283, |
|
"eval_samples_per_second": 3.64, |
|
"eval_steps_per_second": 1.82, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.0166666666666666, |
|
"grad_norm": 25.266735076904297, |
|
"learning_rate": 4.074074074074074e-05, |
|
"loss": 0.9829, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.0333333333333334, |
|
"grad_norm": 21.198898315429688, |
|
"learning_rate": 3.981481481481482e-05, |
|
"loss": 1.1155, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.05, |
|
"grad_norm": 18.009357452392578, |
|
"learning_rate": 3.888888888888889e-05, |
|
"loss": 1.2407, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.0666666666666667, |
|
"grad_norm": 33.55907440185547, |
|
"learning_rate": 3.7962962962962964e-05, |
|
"loss": 1.0741, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 1.0833333333333333, |
|
"grad_norm": 22.596759796142578, |
|
"learning_rate": 3.7037037037037037e-05, |
|
"loss": 0.9267, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"grad_norm": 18.239469528198242, |
|
"learning_rate": 3.611111111111111e-05, |
|
"loss": 0.7076, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 1.1166666666666667, |
|
"grad_norm": 8.70384693145752, |
|
"learning_rate": 3.518518518518519e-05, |
|
"loss": 0.5935, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 1.1333333333333333, |
|
"grad_norm": 24.94647789001465, |
|
"learning_rate": 3.425925925925926e-05, |
|
"loss": 0.8396, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 1.15, |
|
"grad_norm": 2.4496705532073975, |
|
"learning_rate": 3.3333333333333335e-05, |
|
"loss": 0.6741, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 1.1666666666666667, |
|
"grad_norm": 27.570804595947266, |
|
"learning_rate": 3.240740740740741e-05, |
|
"loss": 0.5976, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 1.1833333333333333, |
|
"grad_norm": 3.083174467086792, |
|
"learning_rate": 3.148148148148148e-05, |
|
"loss": 0.3042, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"grad_norm": 31.844913482666016, |
|
"learning_rate": 3.055555555555556e-05, |
|
"loss": 0.671, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 1.2166666666666668, |
|
"grad_norm": 33.209102630615234, |
|
"learning_rate": 2.962962962962963e-05, |
|
"loss": 0.1118, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 1.2333333333333334, |
|
"grad_norm": 13.752717018127441, |
|
"learning_rate": 2.8703703703703706e-05, |
|
"loss": 0.9374, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"grad_norm": 2.6056244373321533, |
|
"learning_rate": 2.777777777777778e-05, |
|
"loss": 1.3288, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 1.25, |
|
"eval_accuracy": 0.8, |
|
"eval_loss": 0.6001584529876709, |
|
"eval_runtime": 18.9652, |
|
"eval_samples_per_second": 3.691, |
|
"eval_steps_per_second": 1.845, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 2.0166666666666666, |
|
"grad_norm": 3.0927469730377197, |
|
"learning_rate": 2.6851851851851855e-05, |
|
"loss": 0.3046, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 2.033333333333333, |
|
"grad_norm": 0.6525083184242249, |
|
"learning_rate": 2.5925925925925925e-05, |
|
"loss": 0.2637, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 2.05, |
|
"grad_norm": 19.304582595825195, |
|
"learning_rate": 2.5e-05, |
|
"loss": 0.2921, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 2.066666666666667, |
|
"grad_norm": 66.67961883544922, |
|
"learning_rate": 2.4074074074074074e-05, |
|
"loss": 0.4924, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 2.0833333333333335, |
|
"grad_norm": 45.11677551269531, |
|
"learning_rate": 2.314814814814815e-05, |
|
"loss": 0.4357, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 2.1, |
|
"grad_norm": 1.1428983211517334, |
|
"learning_rate": 2.2222222222222223e-05, |
|
"loss": 0.2407, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 2.1166666666666667, |
|
"grad_norm": 2.3014516830444336, |
|
"learning_rate": 2.1296296296296296e-05, |
|
"loss": 0.3134, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 2.1333333333333333, |
|
"grad_norm": 0.6744370460510254, |
|
"learning_rate": 2.037037037037037e-05, |
|
"loss": 0.2942, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 2.15, |
|
"grad_norm": 0.36875030398368835, |
|
"learning_rate": 1.9444444444444445e-05, |
|
"loss": 0.2949, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 2.1666666666666665, |
|
"grad_norm": 1.6662416458129883, |
|
"learning_rate": 1.8518518518518518e-05, |
|
"loss": 0.2714, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 2.183333333333333, |
|
"grad_norm": 1.8638778924942017, |
|
"learning_rate": 1.7592592592592595e-05, |
|
"loss": 0.0853, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"grad_norm": 24.525318145751953, |
|
"learning_rate": 1.6666666666666667e-05, |
|
"loss": 0.0878, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 2.216666666666667, |
|
"grad_norm": 4.855034351348877, |
|
"learning_rate": 1.574074074074074e-05, |
|
"loss": 0.3186, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 2.2333333333333334, |
|
"grad_norm": 3.9246416091918945, |
|
"learning_rate": 1.4814814814814815e-05, |
|
"loss": 0.0599, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 2.25, |
|
"grad_norm": 32.75083541870117, |
|
"learning_rate": 1.388888888888889e-05, |
|
"loss": 0.1638, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 2.25, |
|
"eval_accuracy": 0.8142857142857143, |
|
"eval_loss": 0.44258585572242737, |
|
"eval_runtime": 18.9666, |
|
"eval_samples_per_second": 3.691, |
|
"eval_steps_per_second": 1.845, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 3.0166666666666666, |
|
"grad_norm": 0.29513224959373474, |
|
"learning_rate": 1.2962962962962962e-05, |
|
"loss": 0.0522, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 3.033333333333333, |
|
"grad_norm": 1.6883537769317627, |
|
"learning_rate": 1.2037037037037037e-05, |
|
"loss": 0.0449, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 3.05, |
|
"grad_norm": 0.08803384006023407, |
|
"learning_rate": 1.1111111111111112e-05, |
|
"loss": 0.0224, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 3.066666666666667, |
|
"grad_norm": 0.23137041926383972, |
|
"learning_rate": 1.0185185185185185e-05, |
|
"loss": 0.0324, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 3.0833333333333335, |
|
"grad_norm": 0.16817709803581238, |
|
"learning_rate": 9.259259259259259e-06, |
|
"loss": 0.0591, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 3.1, |
|
"grad_norm": 9.277715682983398, |
|
"learning_rate": 8.333333333333334e-06, |
|
"loss": 0.0852, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 3.1166666666666667, |
|
"grad_norm": 0.08057785779237747, |
|
"learning_rate": 7.4074074074074075e-06, |
|
"loss": 0.0407, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 3.1333333333333333, |
|
"grad_norm": 26.872711181640625, |
|
"learning_rate": 6.481481481481481e-06, |
|
"loss": 0.0517, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 3.15, |
|
"grad_norm": 0.36958783864974976, |
|
"learning_rate": 5.555555555555556e-06, |
|
"loss": 0.0991, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 3.1666666666666665, |
|
"grad_norm": 0.3635166883468628, |
|
"learning_rate": 4.6296296296296296e-06, |
|
"loss": 0.2209, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 3.183333333333333, |
|
"grad_norm": 0.0713394284248352, |
|
"learning_rate": 3.7037037037037037e-06, |
|
"loss": 0.2704, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 3.2, |
|
"grad_norm": 0.17939713597297668, |
|
"learning_rate": 2.777777777777778e-06, |
|
"loss": 0.0767, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 3.216666666666667, |
|
"grad_norm": 0.5136755108833313, |
|
"learning_rate": 1.8518518518518519e-06, |
|
"loss": 0.1464, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 3.2333333333333334, |
|
"grad_norm": 0.3038472831249237, |
|
"learning_rate": 9.259259259259259e-07, |
|
"loss": 0.3934, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"grad_norm": 0.18430446088314056, |
|
"learning_rate": 0.0, |
|
"loss": 0.3692, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"eval_accuracy": 0.9142857142857143, |
|
"eval_loss": 0.23388925194740295, |
|
"eval_runtime": 20.4156, |
|
"eval_samples_per_second": 3.429, |
|
"eval_steps_per_second": 1.714, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"step": 600, |
|
"total_flos": 1.495384188125184e+18, |
|
"train_loss": 0.7621474319696426, |
|
"train_runtime": 906.4456, |
|
"train_samples_per_second": 1.324, |
|
"train_steps_per_second": 0.662 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"eval_accuracy": 0.9225806451612903, |
|
"eval_loss": 0.26695409417152405, |
|
"eval_runtime": 47.7519, |
|
"eval_samples_per_second": 3.246, |
|
"eval_steps_per_second": 1.633, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 3.25, |
|
"eval_accuracy": 0.9225806451612903, |
|
"eval_loss": 0.26695412397384644, |
|
"eval_runtime": 43.0893, |
|
"eval_samples_per_second": 3.597, |
|
"eval_steps_per_second": 1.81, |
|
"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 |
|
} |
|
|