m21041's picture
Model save
13965d3 verified
{
"best_global_step": 19,
"best_metric": 1.0,
"best_model_checkpoint": "VideoMAEv2-Base-finetuned-ucf101-subset/checkpoint-19",
"epoch": 3.2083333333333335,
"eval_steps": 500,
"global_step": 72,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.1388888888888889,
"grad_norm": 9.568665504455566,
"learning_rate": 4.921875e-05,
"loss": 2.1288,
"step": 10
},
{
"epoch": 0.2638888888888889,
"eval_accuracy": 1.0,
"eval_loss": 0.4425038993358612,
"eval_runtime": 22.424,
"eval_samples_per_second": 3.122,
"eval_steps_per_second": 0.223,
"step": 19
},
{
"epoch": 1.0138888888888888,
"grad_norm": 6.739668846130371,
"learning_rate": 4.140625e-05,
"loss": 0.949,
"step": 20
},
{
"epoch": 1.1527777777777777,
"grad_norm": 7.086241722106934,
"learning_rate": 3.359375e-05,
"loss": 0.2366,
"step": 30
},
{
"epoch": 1.2638888888888888,
"eval_accuracy": 1.0,
"eval_loss": 0.06767228245735168,
"eval_runtime": 22.4196,
"eval_samples_per_second": 3.122,
"eval_steps_per_second": 0.223,
"step": 38
},
{
"epoch": 2.0277777777777777,
"grad_norm": 0.7161458730697632,
"learning_rate": 2.578125e-05,
"loss": 0.0963,
"step": 40
},
{
"epoch": 2.1666666666666665,
"grad_norm": 0.4962587356567383,
"learning_rate": 1.796875e-05,
"loss": 0.0428,
"step": 50
},
{
"epoch": 2.263888888888889,
"eval_accuracy": 1.0,
"eval_loss": 0.03534603491425514,
"eval_runtime": 21.9482,
"eval_samples_per_second": 3.189,
"eval_steps_per_second": 0.228,
"step": 57
},
{
"epoch": 3.0416666666666665,
"grad_norm": 0.2468690574169159,
"learning_rate": 1.0156250000000001e-05,
"loss": 0.0201,
"step": 60
},
{
"epoch": 3.1805555555555554,
"grad_norm": 0.20965361595153809,
"learning_rate": 2.3437500000000002e-06,
"loss": 0.0239,
"step": 70
},
{
"epoch": 3.2083333333333335,
"eval_accuracy": 1.0,
"eval_loss": 0.028124911710619926,
"eval_runtime": 22.0601,
"eval_samples_per_second": 3.173,
"eval_steps_per_second": 0.227,
"step": 72
},
{
"epoch": 3.2083333333333335,
"step": 72,
"total_flos": 0.0,
"train_loss": 0.48614118706124526,
"train_runtime": 889.8242,
"train_samples_per_second": 1.295,
"train_steps_per_second": 0.081
},
{
"epoch": 3.2083333333333335,
"eval_accuracy": 1.0,
"eval_loss": 0.4690203368663788,
"eval_runtime": 53.9134,
"eval_samples_per_second": 2.875,
"eval_steps_per_second": 0.185,
"step": 72
}
],
"logging_steps": 10,
"max_steps": 72,
"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": 0.0,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}