detr-r50-cd45rb-8ah-12l-corrected / trainer_state.json
polejowska's picture
End of training
e0ca93c
{
"best_metric": 1.7297718524932861,
"best_model_checkpoint": "detr-r50-cd45rb-8ah-12l-corrected\\checkpoint-46060",
"epoch": 10.0,
"global_step": 46060,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"learning_rate": 9.00238818931828e-06,
"loss": 3.1161,
"step": 4606
},
{
"epoch": 1.0,
"eval_loss": 2.238567590713501,
"eval_runtime": 206.6437,
"eval_samples_per_second": 8.619,
"eval_steps_per_second": 1.079,
"step": 4606
},
{
"epoch": 2.0,
"learning_rate": 8.002388189318281e-06,
"loss": 2.7777,
"step": 9212
},
{
"epoch": 2.0,
"eval_loss": 2.0665035247802734,
"eval_runtime": 205.9971,
"eval_samples_per_second": 8.646,
"eval_steps_per_second": 1.083,
"step": 9212
},
{
"epoch": 3.0,
"learning_rate": 7.0028224055579685e-06,
"loss": 2.6042,
"step": 13818
},
{
"epoch": 3.0,
"eval_loss": 1.9953538179397583,
"eval_runtime": 206.4087,
"eval_samples_per_second": 8.629,
"eval_steps_per_second": 1.08,
"step": 13818
},
{
"epoch": 4.0,
"learning_rate": 6.003256621797656e-06,
"loss": 2.5082,
"step": 18424
},
{
"epoch": 4.0,
"eval_loss": 1.8990634679794312,
"eval_runtime": 205.9514,
"eval_samples_per_second": 8.648,
"eval_steps_per_second": 1.083,
"step": 18424
},
{
"epoch": 5.0,
"learning_rate": 5.003907946157187e-06,
"loss": 2.4529,
"step": 23030
},
{
"epoch": 5.0,
"eval_loss": 1.9227577447891235,
"eval_runtime": 206.4991,
"eval_samples_per_second": 8.625,
"eval_steps_per_second": 1.08,
"step": 23030
},
{
"epoch": 6.0,
"learning_rate": 4.0041250542770305e-06,
"loss": 2.3944,
"step": 27636
},
{
"epoch": 6.0,
"eval_loss": 1.8829206228256226,
"eval_runtime": 205.9122,
"eval_samples_per_second": 8.649,
"eval_steps_per_second": 1.083,
"step": 27636
},
{
"epoch": 7.0,
"learning_rate": 3.004342162396874e-06,
"loss": 2.3405,
"step": 32242
},
{
"epoch": 7.0,
"eval_loss": 1.813354253768921,
"eval_runtime": 207.4583,
"eval_samples_per_second": 8.585,
"eval_steps_per_second": 1.075,
"step": 32242
},
{
"epoch": 8.0,
"learning_rate": 2.0045592705167173e-06,
"loss": 2.3082,
"step": 36848
},
{
"epoch": 8.0,
"eval_loss": 1.7850828170776367,
"eval_runtime": 205.9799,
"eval_samples_per_second": 8.646,
"eval_steps_per_second": 1.083,
"step": 36848
},
{
"epoch": 9.0,
"learning_rate": 1.0049934867564047e-06,
"loss": 2.2684,
"step": 41454
},
{
"epoch": 9.0,
"eval_loss": 1.7471296787261963,
"eval_runtime": 206.2096,
"eval_samples_per_second": 8.637,
"eval_steps_per_second": 1.081,
"step": 41454
},
{
"epoch": 10.0,
"learning_rate": 5.644811115935737e-09,
"loss": 2.2422,
"step": 46060
},
{
"epoch": 10.0,
"eval_loss": 1.7297718524932861,
"eval_runtime": 207.5274,
"eval_samples_per_second": 8.582,
"eval_steps_per_second": 1.075,
"step": 46060
},
{
"epoch": 10.0,
"step": 46060,
"total_flos": 1.2487845435287042e+20,
"train_loss": 2.5012789025184543,
"train_runtime": 32905.3974,
"train_samples_per_second": 5.598,
"train_steps_per_second": 1.4
}
],
"max_steps": 46060,
"num_train_epochs": 10,
"total_flos": 1.2487845435287042e+20,
"trial_name": null,
"trial_params": null
}