thenam's picture
End of training
e74473c verified
{
"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
}