detr-resnet50-cd45rb-subdataset / trainer_state.json
polejowska's picture
End of training
425a064
{
"best_metric": 2.7267191410064697,
"best_model_checkpoint": "detr-resnet50-cd45rb-subdataset/checkpoint-2420",
"epoch": 10.0,
"global_step": 2420,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"learning_rate": 9e-05,
"loss": 3.774,
"step": 242
},
{
"epoch": 1.0,
"eval_loss": 2.962869644165039,
"eval_runtime": 11.2116,
"eval_samples_per_second": 8.919,
"eval_steps_per_second": 1.16,
"step": 242
},
{
"epoch": 2.0,
"learning_rate": 8e-05,
"loss": 3.6768,
"step": 484
},
{
"epoch": 2.0,
"eval_loss": 2.8228960037231445,
"eval_runtime": 11.2793,
"eval_samples_per_second": 8.866,
"eval_steps_per_second": 1.153,
"step": 484
},
{
"epoch": 3.0,
"learning_rate": 7e-05,
"loss": 3.6554,
"step": 726
},
{
"epoch": 3.0,
"eval_loss": 2.8254785537719727,
"eval_runtime": 11.3653,
"eval_samples_per_second": 8.799,
"eval_steps_per_second": 1.144,
"step": 726
},
{
"epoch": 4.0,
"learning_rate": 6e-05,
"loss": 3.5964,
"step": 968
},
{
"epoch": 4.0,
"eval_loss": 2.8284406661987305,
"eval_runtime": 11.2727,
"eval_samples_per_second": 8.871,
"eval_steps_per_second": 1.153,
"step": 968
},
{
"epoch": 5.0,
"learning_rate": 5e-05,
"loss": 3.5668,
"step": 1210
},
{
"epoch": 5.0,
"eval_loss": 2.750312089920044,
"eval_runtime": 20.6727,
"eval_samples_per_second": 4.837,
"eval_steps_per_second": 0.629,
"step": 1210
},
{
"epoch": 6.0,
"learning_rate": 4e-05,
"loss": 3.5733,
"step": 1452
},
{
"epoch": 6.0,
"eval_loss": 2.755269765853882,
"eval_runtime": 26.9669,
"eval_samples_per_second": 3.708,
"eval_steps_per_second": 0.482,
"step": 1452
},
{
"epoch": 7.0,
"learning_rate": 3e-05,
"loss": 3.5197,
"step": 1694
},
{
"epoch": 7.0,
"eval_loss": 2.7466766834259033,
"eval_runtime": 11.2293,
"eval_samples_per_second": 8.905,
"eval_steps_per_second": 1.158,
"step": 1694
},
{
"epoch": 8.0,
"learning_rate": 2e-05,
"loss": 3.5313,
"step": 1936
},
{
"epoch": 8.0,
"eval_loss": 2.752133846282959,
"eval_runtime": 12.0479,
"eval_samples_per_second": 8.3,
"eval_steps_per_second": 1.079,
"step": 1936
},
{
"epoch": 9.0,
"learning_rate": 1e-05,
"loss": 3.4862,
"step": 2178
},
{
"epoch": 9.0,
"eval_loss": 2.7274692058563232,
"eval_runtime": 12.3145,
"eval_samples_per_second": 8.121,
"eval_steps_per_second": 1.056,
"step": 2178
},
{
"epoch": 10.0,
"learning_rate": 0.0,
"loss": 3.503,
"step": 2420
},
{
"epoch": 10.0,
"eval_loss": 2.7267191410064697,
"eval_runtime": 11.8778,
"eval_samples_per_second": 8.419,
"eval_steps_per_second": 1.094,
"step": 2420
},
{
"epoch": 10.0,
"step": 2420,
"total_flos": 4.6164408869376e+18,
"train_loss": 3.588286495996901,
"train_runtime": 26239.9252,
"train_samples_per_second": 0.368,
"train_steps_per_second": 0.092
}
],
"max_steps": 2420,
"num_train_epochs": 10,
"total_flos": 4.6164408869376e+18,
"trial_name": null,
"trial_params": null
}