minhah's picture
End of training
2abec99 verified
{
"best_metric": 0.49748743718592964,
"best_model_checkpoint": "videomae-base-finetuned-ElderReact-anger-balancedf1/checkpoint-75",
"epoch": 9.0625,
"eval_steps": 500,
"global_step": 240,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"grad_norm": 3.997068166732788,
"learning_rate": 2.0833333333333336e-05,
"loss": 0.7186,
"step": 10
},
{
"epoch": 0.08,
"grad_norm": 3.005845785140991,
"learning_rate": 4.166666666666667e-05,
"loss": 0.7339,
"step": 20
},
{
"epoch": 0.1,
"eval_f1": 0.4463276836158192,
"eval_loss": 0.6864726543426514,
"eval_runtime": 171.9816,
"eval_samples_per_second": 2.442,
"eval_steps_per_second": 0.157,
"step": 25
},
{
"epoch": 1.02,
"grad_norm": 6.235264301300049,
"learning_rate": 4.8611111111111115e-05,
"loss": 0.7253,
"step": 30
},
{
"epoch": 1.06,
"grad_norm": 3.536816358566284,
"learning_rate": 4.62962962962963e-05,
"loss": 0.7024,
"step": 40
},
{
"epoch": 1.1,
"grad_norm": 12.325151443481445,
"learning_rate": 4.3981481481481486e-05,
"loss": 0.7007,
"step": 50
},
{
"epoch": 1.1,
"eval_f1": 0.047619047619047616,
"eval_loss": 0.7107335925102234,
"eval_runtime": 176.4951,
"eval_samples_per_second": 2.38,
"eval_steps_per_second": 0.153,
"step": 50
},
{
"epoch": 2.04,
"grad_norm": 2.6023736000061035,
"learning_rate": 4.166666666666667e-05,
"loss": 0.6583,
"step": 60
},
{
"epoch": 2.08,
"grad_norm": 4.512753963470459,
"learning_rate": 3.935185185185186e-05,
"loss": 0.6854,
"step": 70
},
{
"epoch": 2.1,
"eval_f1": 0.49748743718592964,
"eval_loss": 0.6994316577911377,
"eval_runtime": 178.8944,
"eval_samples_per_second": 2.348,
"eval_steps_per_second": 0.151,
"step": 75
},
{
"epoch": 3.02,
"grad_norm": 5.125423908233643,
"learning_rate": 3.7037037037037037e-05,
"loss": 0.7021,
"step": 80
},
{
"epoch": 3.06,
"grad_norm": 3.5603673458099365,
"learning_rate": 3.472222222222222e-05,
"loss": 0.6419,
"step": 90
},
{
"epoch": 3.1,
"grad_norm": 21.498886108398438,
"learning_rate": 3.240740740740741e-05,
"loss": 0.715,
"step": 100
},
{
"epoch": 3.1,
"eval_f1": 0.13333333333333333,
"eval_loss": 0.7238408923149109,
"eval_runtime": 178.3055,
"eval_samples_per_second": 2.356,
"eval_steps_per_second": 0.151,
"step": 100
},
{
"epoch": 4.04,
"grad_norm": 3.3730030059814453,
"learning_rate": 3.0092592592592593e-05,
"loss": 0.6555,
"step": 110
},
{
"epoch": 4.08,
"grad_norm": 6.0418829917907715,
"learning_rate": 2.777777777777778e-05,
"loss": 0.6866,
"step": 120
},
{
"epoch": 4.1,
"eval_f1": 0.01951219512195122,
"eval_loss": 0.7563002109527588,
"eval_runtime": 176.1376,
"eval_samples_per_second": 2.384,
"eval_steps_per_second": 0.153,
"step": 125
},
{
"epoch": 5.02,
"grad_norm": 6.097543716430664,
"learning_rate": 2.5462962962962965e-05,
"loss": 0.7311,
"step": 130
},
{
"epoch": 5.06,
"grad_norm": 2.698549509048462,
"learning_rate": 2.314814814814815e-05,
"loss": 0.5913,
"step": 140
},
{
"epoch": 5.1,
"grad_norm": 17.406471252441406,
"learning_rate": 2.0833333333333336e-05,
"loss": 0.6564,
"step": 150
},
{
"epoch": 5.1,
"eval_f1": 0.15637860082304528,
"eval_loss": 0.7628450393676758,
"eval_runtime": 177.1888,
"eval_samples_per_second": 2.37,
"eval_steps_per_second": 0.152,
"step": 150
},
{
"epoch": 6.04,
"grad_norm": 4.0975470542907715,
"learning_rate": 1.8518518518518518e-05,
"loss": 0.598,
"step": 160
},
{
"epoch": 6.08,
"grad_norm": 4.26428747177124,
"learning_rate": 1.6203703703703704e-05,
"loss": 0.5809,
"step": 170
},
{
"epoch": 6.1,
"eval_f1": 0.38311688311688313,
"eval_loss": 0.7543146014213562,
"eval_runtime": 167.9465,
"eval_samples_per_second": 2.501,
"eval_steps_per_second": 0.161,
"step": 175
},
{
"epoch": 7.02,
"grad_norm": 2.8742892742156982,
"learning_rate": 1.388888888888889e-05,
"loss": 0.4884,
"step": 180
},
{
"epoch": 7.06,
"grad_norm": 4.310091972351074,
"learning_rate": 1.1574074074074075e-05,
"loss": 0.5761,
"step": 190
},
{
"epoch": 7.1,
"grad_norm": 3.7042274475097656,
"learning_rate": 9.259259259259259e-06,
"loss": 0.6035,
"step": 200
},
{
"epoch": 7.1,
"eval_f1": 0.21224489795918366,
"eval_loss": 0.7982825040817261,
"eval_runtime": 168.7315,
"eval_samples_per_second": 2.489,
"eval_steps_per_second": 0.16,
"step": 200
},
{
"epoch": 8.04,
"grad_norm": 4.604009628295898,
"learning_rate": 6.944444444444445e-06,
"loss": 0.5098,
"step": 210
},
{
"epoch": 8.08,
"grad_norm": 3.782595157623291,
"learning_rate": 4.6296296296296296e-06,
"loss": 0.6153,
"step": 220
},
{
"epoch": 8.1,
"eval_f1": 0.40634920634920635,
"eval_loss": 0.7542010545730591,
"eval_runtime": 176.9739,
"eval_samples_per_second": 2.373,
"eval_steps_per_second": 0.153,
"step": 225
},
{
"epoch": 9.02,
"grad_norm": 3.7313973903656006,
"learning_rate": 2.3148148148148148e-06,
"loss": 0.5072,
"step": 230
},
{
"epoch": 9.06,
"grad_norm": 2.534163475036621,
"learning_rate": 0.0,
"loss": 0.5099,
"step": 240
},
{
"epoch": 9.06,
"eval_f1": 0.4525993883792049,
"eval_loss": 0.7408385276794434,
"eval_runtime": 159.3385,
"eval_samples_per_second": 2.636,
"eval_steps_per_second": 0.169,
"step": 240
},
{
"epoch": 9.06,
"step": 240,
"total_flos": 4.616669303992812e+18,
"train_loss": 0.6372330923875172,
"train_runtime": 3422.175,
"train_samples_per_second": 1.122,
"train_steps_per_second": 0.07
},
{
"epoch": 9.06,
"eval_f1": 0.5797752808988764,
"eval_loss": 0.7431801557540894,
"eval_runtime": 150.1599,
"eval_samples_per_second": 2.537,
"eval_steps_per_second": 0.16,
"step": 240
},
{
"epoch": 9.06,
"eval_f1": 0.5797752808988764,
"eval_loss": 0.7424257397651672,
"eval_runtime": 135.4113,
"eval_samples_per_second": 2.814,
"eval_steps_per_second": 0.177,
"step": 240
}
],
"logging_steps": 10,
"max_steps": 240,
"num_input_tokens_seen": 0,
"num_train_epochs": 9223372036854775807,
"save_steps": 500,
"total_flos": 4.616669303992812e+18,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}