OneNorth's picture
End of training
1d1ecf2 verified
{
"best_metric": 0.35714285714285715,
"best_model_checkpoint": "videomae-base-finetuned-ucf101-subset/checkpoint-75",
"epoch": 1.0,
"eval_steps": 500,
"global_step": 75,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.13,
"grad_norm": 9.760223388671875,
"learning_rate": 4.850746268656717e-05,
"loss": 2.2528,
"step": 10
},
{
"epoch": 0.27,
"grad_norm": 10.098638534545898,
"learning_rate": 4.104477611940299e-05,
"loss": 2.2017,
"step": 20
},
{
"epoch": 0.4,
"grad_norm": 9.566693305969238,
"learning_rate": 3.358208955223881e-05,
"loss": 2.106,
"step": 30
},
{
"epoch": 0.53,
"grad_norm": 9.71388053894043,
"learning_rate": 2.6119402985074626e-05,
"loss": 2.0158,
"step": 40
},
{
"epoch": 0.67,
"grad_norm": 13.246223449707031,
"learning_rate": 1.865671641791045e-05,
"loss": 2.0417,
"step": 50
},
{
"epoch": 0.8,
"grad_norm": 10.271443367004395,
"learning_rate": 1.119402985074627e-05,
"loss": 1.8307,
"step": 60
},
{
"epoch": 0.93,
"grad_norm": 16.004535675048828,
"learning_rate": 3.7313432835820893e-06,
"loss": 1.9649,
"step": 70
},
{
"epoch": 1.0,
"eval_accuracy": 0.35714285714285715,
"eval_loss": 1.7548341751098633,
"eval_runtime": 12.6786,
"eval_samples_per_second": 5.521,
"eval_steps_per_second": 1.42,
"step": 75
},
{
"epoch": 1.0,
"step": 75,
"total_flos": 3.73846047031296e+17,
"train_loss": 2.0365064112345377,
"train_runtime": 139.8497,
"train_samples_per_second": 2.145,
"train_steps_per_second": 0.536
},
{
"epoch": 1.0,
"eval_accuracy": 0.27741935483870966,
"eval_loss": 1.9401741027832031,
"eval_runtime": 29.4337,
"eval_samples_per_second": 5.266,
"eval_steps_per_second": 1.325,
"step": 75
},
{
"epoch": 1.0,
"eval_accuracy": 0.27741935483870966,
"eval_loss": 1.9401745796203613,
"eval_runtime": 28.1588,
"eval_samples_per_second": 5.505,
"eval_steps_per_second": 1.385,
"step": 75
}
],
"logging_steps": 10,
"max_steps": 75,
"num_input_tokens_seen": 0,
"num_train_epochs": 9223372036854775807,
"save_steps": 500,
"total_flos": 3.73846047031296e+17,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}