detr-r50-cd45rb-1ah-6l / trainer_state.json
polejowska's picture
End of training
9d5ee91
raw
history blame contribute delete
No virus
2.26 kB
{
"best_metric": 2.4488213062286377,
"best_model_checkpoint": "detr-r50-cd45rb-1ah-6l\\checkpoint-23030",
"epoch": 5.0,
"global_step": 23030,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"learning_rate": 8.003907946157187e-06,
"loss": 3.8594,
"step": 4606
},
{
"epoch": 1.0,
"eval_loss": 2.9210634231567383,
"eval_runtime": 204.6437,
"eval_samples_per_second": 8.703,
"eval_steps_per_second": 1.09,
"step": 4606
},
{
"epoch": 2.0,
"learning_rate": 6.004776378636562e-06,
"loss": 3.4922,
"step": 9212
},
{
"epoch": 2.0,
"eval_loss": 2.6744484901428223,
"eval_runtime": 229.0649,
"eval_samples_per_second": 7.775,
"eval_steps_per_second": 0.974,
"step": 9212
},
{
"epoch": 3.0,
"learning_rate": 4.004776378636561e-06,
"loss": 3.2948,
"step": 13818
},
{
"epoch": 3.0,
"eval_loss": 2.548419713973999,
"eval_runtime": 326.7658,
"eval_samples_per_second": 5.45,
"eval_steps_per_second": 0.682,
"step": 13818
},
{
"epoch": 4.0,
"learning_rate": 2.005644811115936e-06,
"loss": 3.2029,
"step": 18424
},
{
"epoch": 4.0,
"eval_loss": 2.4830880165100098,
"eval_runtime": 325.5074,
"eval_samples_per_second": 5.471,
"eval_steps_per_second": 0.685,
"step": 18424
},
{
"epoch": 5.0,
"learning_rate": 6.079027355623101e-09,
"loss": 3.1514,
"step": 23030
},
{
"epoch": 5.0,
"eval_loss": 2.4488213062286377,
"eval_runtime": 338.901,
"eval_samples_per_second": 5.255,
"eval_steps_per_second": 0.658,
"step": 23030
},
{
"epoch": 5.0,
"step": 23030,
"total_flos": 4.401655508016e+19,
"train_loss": 3.4001299680443986,
"train_runtime": 20053.0547,
"train_samples_per_second": 4.593,
"train_steps_per_second": 1.148
}
],
"max_steps": 23030,
"num_train_epochs": 5,
"total_flos": 4.401655508016e+19,
"trial_name": null,
"trial_params": null
}