detr-resnet-50-base-coco / trainer_state.json
amyeroberts's picture
amyeroberts HF staff
End of training
3f563dc
raw
history blame
9.04 kB
{
"best_metric": 6719.021484375,
"best_model_checkpoint": "./coco_outputs/checkpoint-640",
"epoch": 5.0,
"global_step": 640,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.08,
"learning_rate": 1.96875e-05,
"loss": 6346.0008,
"step": 10
},
{
"epoch": 0.16,
"learning_rate": 1.9375e-05,
"loss": 6400.0195,
"step": 20
},
{
"epoch": 0.23,
"learning_rate": 1.9062500000000003e-05,
"loss": 6295.0742,
"step": 30
},
{
"epoch": 0.31,
"learning_rate": 1.8750000000000002e-05,
"loss": 6280.2168,
"step": 40
},
{
"epoch": 0.39,
"learning_rate": 1.84375e-05,
"loss": 6138.5656,
"step": 50
},
{
"epoch": 0.47,
"learning_rate": 1.8125e-05,
"loss": 6088.3699,
"step": 60
},
{
"epoch": 0.55,
"learning_rate": 1.7812500000000003e-05,
"loss": 6224.8137,
"step": 70
},
{
"epoch": 0.62,
"learning_rate": 1.7500000000000002e-05,
"loss": 6147.6773,
"step": 80
},
{
"epoch": 0.7,
"learning_rate": 1.71875e-05,
"loss": 6426.423,
"step": 90
},
{
"epoch": 0.78,
"learning_rate": 1.6875e-05,
"loss": 6195.2883,
"step": 100
},
{
"epoch": 0.86,
"learning_rate": 1.6562500000000003e-05,
"loss": 6409.0055,
"step": 110
},
{
"epoch": 0.94,
"learning_rate": 1.6250000000000002e-05,
"loss": 6457.3832,
"step": 120
},
{
"epoch": 1.0,
"eval_loss": 6720.3701171875,
"eval_runtime": 16.7008,
"eval_samples_per_second": 15.329,
"eval_steps_per_second": 1.916,
"step": 128
},
{
"epoch": 1.02,
"learning_rate": 1.59375e-05,
"loss": 6196.2672,
"step": 130
},
{
"epoch": 1.09,
"learning_rate": 1.5625e-05,
"loss": 6187.7188,
"step": 140
},
{
"epoch": 1.17,
"learning_rate": 1.5312500000000003e-05,
"loss": 6357.3324,
"step": 150
},
{
"epoch": 1.25,
"learning_rate": 1.5000000000000002e-05,
"loss": 6108.043,
"step": 160
},
{
"epoch": 1.33,
"learning_rate": 1.4687500000000001e-05,
"loss": 6499.2922,
"step": 170
},
{
"epoch": 1.41,
"learning_rate": 1.4375e-05,
"loss": 6227.1672,
"step": 180
},
{
"epoch": 1.48,
"learning_rate": 1.4062500000000001e-05,
"loss": 6416.4754,
"step": 190
},
{
"epoch": 1.56,
"learning_rate": 1.375e-05,
"loss": 6024.552,
"step": 200
},
{
"epoch": 1.64,
"learning_rate": 1.3437500000000001e-05,
"loss": 5964.7684,
"step": 210
},
{
"epoch": 1.72,
"learning_rate": 1.3125e-05,
"loss": 5876.4055,
"step": 220
},
{
"epoch": 1.8,
"learning_rate": 1.2812500000000001e-05,
"loss": 6149.1523,
"step": 230
},
{
"epoch": 1.88,
"learning_rate": 1.25e-05,
"loss": 6330.3543,
"step": 240
},
{
"epoch": 1.95,
"learning_rate": 1.2187500000000001e-05,
"loss": 6234.0246,
"step": 250
},
{
"epoch": 2.0,
"eval_loss": 6719.166015625,
"eval_runtime": 16.7578,
"eval_samples_per_second": 15.276,
"eval_steps_per_second": 1.91,
"step": 256
},
{
"epoch": 2.03,
"learning_rate": 1.1875e-05,
"loss": 6279.7289,
"step": 260
},
{
"epoch": 2.11,
"learning_rate": 1.1562500000000002e-05,
"loss": 6253.7887,
"step": 270
},
{
"epoch": 2.19,
"learning_rate": 1.125e-05,
"loss": 6329.3629,
"step": 280
},
{
"epoch": 2.27,
"learning_rate": 1.0937500000000002e-05,
"loss": 6185.052,
"step": 290
},
{
"epoch": 2.34,
"learning_rate": 1.0625e-05,
"loss": 5918.475,
"step": 300
},
{
"epoch": 2.42,
"learning_rate": 1.0312500000000002e-05,
"loss": 6114.2844,
"step": 310
},
{
"epoch": 2.5,
"learning_rate": 1e-05,
"loss": 6149.1941,
"step": 320
},
{
"epoch": 2.58,
"learning_rate": 9.6875e-06,
"loss": 6154.8313,
"step": 330
},
{
"epoch": 2.66,
"learning_rate": 9.375000000000001e-06,
"loss": 6146.2449,
"step": 340
},
{
"epoch": 2.73,
"learning_rate": 9.0625e-06,
"loss": 6099.1227,
"step": 350
},
{
"epoch": 2.81,
"learning_rate": 8.750000000000001e-06,
"loss": 6083.9359,
"step": 360
},
{
"epoch": 2.89,
"learning_rate": 8.4375e-06,
"loss": 6283.482,
"step": 370
},
{
"epoch": 2.97,
"learning_rate": 8.125000000000001e-06,
"loss": 6335.5293,
"step": 380
},
{
"epoch": 3.0,
"eval_loss": 6719.03076171875,
"eval_runtime": 16.7226,
"eval_samples_per_second": 15.309,
"eval_steps_per_second": 1.914,
"step": 384
},
{
"epoch": 3.05,
"learning_rate": 7.8125e-06,
"loss": 6466.4867,
"step": 390
},
{
"epoch": 3.12,
"learning_rate": 7.500000000000001e-06,
"loss": 6327.5625,
"step": 400
},
{
"epoch": 3.2,
"learning_rate": 7.1875e-06,
"loss": 6213.9516,
"step": 410
},
{
"epoch": 3.28,
"learning_rate": 6.875e-06,
"loss": 6337.8066,
"step": 420
},
{
"epoch": 3.36,
"learning_rate": 6.5625e-06,
"loss": 6188.782,
"step": 430
},
{
"epoch": 3.44,
"learning_rate": 6.25e-06,
"loss": 6432.0715,
"step": 440
},
{
"epoch": 3.52,
"learning_rate": 5.9375e-06,
"loss": 6178.6059,
"step": 450
},
{
"epoch": 3.59,
"learning_rate": 5.625e-06,
"loss": 6356.5617,
"step": 460
},
{
"epoch": 3.67,
"learning_rate": 5.3125e-06,
"loss": 6572.6281,
"step": 470
},
{
"epoch": 3.75,
"learning_rate": 5e-06,
"loss": 6376.5836,
"step": 480
},
{
"epoch": 3.83,
"learning_rate": 4.6875000000000004e-06,
"loss": 6186.7969,
"step": 490
},
{
"epoch": 3.91,
"learning_rate": 4.3750000000000005e-06,
"loss": 6402.0125,
"step": 500
},
{
"epoch": 3.98,
"learning_rate": 4.0625000000000005e-06,
"loss": 6520.8047,
"step": 510
},
{
"epoch": 4.0,
"eval_loss": 6719.060546875,
"eval_runtime": 16.8687,
"eval_samples_per_second": 15.176,
"eval_steps_per_second": 1.897,
"step": 512
},
{
"epoch": 4.06,
"learning_rate": 3.7500000000000005e-06,
"loss": 6016.2836,
"step": 520
},
{
"epoch": 4.14,
"learning_rate": 3.4375e-06,
"loss": 5934.718,
"step": 530
},
{
"epoch": 4.22,
"learning_rate": 3.125e-06,
"loss": 6232.1816,
"step": 540
},
{
"epoch": 4.3,
"learning_rate": 2.8125e-06,
"loss": 6584.9633,
"step": 550
},
{
"epoch": 4.38,
"learning_rate": 2.5e-06,
"loss": 6533.6297,
"step": 560
},
{
"epoch": 4.45,
"learning_rate": 2.1875000000000002e-06,
"loss": 6380.1676,
"step": 570
},
{
"epoch": 4.53,
"learning_rate": 1.8750000000000003e-06,
"loss": 6350.5738,
"step": 580
},
{
"epoch": 4.61,
"learning_rate": 1.5625e-06,
"loss": 5988.5664,
"step": 590
},
{
"epoch": 4.69,
"learning_rate": 1.25e-06,
"loss": 6169.0441,
"step": 600
},
{
"epoch": 4.77,
"learning_rate": 9.375000000000001e-07,
"loss": 6024.5156,
"step": 610
},
{
"epoch": 4.84,
"learning_rate": 6.25e-07,
"loss": 6515.2469,
"step": 620
},
{
"epoch": 4.92,
"learning_rate": 3.125e-07,
"loss": 6284.4926,
"step": 630
},
{
"epoch": 5.0,
"learning_rate": 0.0,
"loss": 5738.273,
"step": 640
},
{
"epoch": 5.0,
"eval_loss": 6719.021484375,
"eval_runtime": 17.1732,
"eval_samples_per_second": 14.907,
"eval_steps_per_second": 1.863,
"step": 640
},
{
"epoch": 5.0,
"step": 640,
"total_flos": 2.4480067682304e+18,
"train_loss": 6244.167712402344,
"train_runtime": 818.3257,
"train_samples_per_second": 6.257,
"train_steps_per_second": 0.782
}
],
"max_steps": 640,
"num_train_epochs": 5,
"total_flos": 2.4480067682304e+18,
"trial_name": null,
"trial_params": null
}