detr-finetuned-cppe-5-10k-steps / trainer_state.json
niujiahui's picture
End of training
f494f6e verified
{
"best_metric": 0.1289,
"best_model_checkpoint": "detr-finetuned-cppe-5-10k-steps\\checkpoint-1070",
"epoch": 10.0,
"eval_steps": 500,
"global_step": 1070,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 53.4410400390625,
"learning_rate": 4.523364485981308e-05,
"loss": 2.6856,
"step": 107
},
{
"epoch": 1.0,
"eval_loss": 2.4603869915008545,
"eval_map": 0.0152,
"eval_map_50": 0.0396,
"eval_map_75": 0.0096,
"eval_map_Coverall": 0.0622,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0045,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0092,
"eval_map_large": 0.0234,
"eval_map_medium": 0.0048,
"eval_map_small": 0.0051,
"eval_mar_1": 0.0535,
"eval_mar_10": 0.1109,
"eval_mar_100": 0.1286,
"eval_mar_100_Coverall": 0.3509,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.1174,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.1747,
"eval_mar_large": 0.1752,
"eval_mar_medium": 0.089,
"eval_mar_small": 0.0401,
"eval_runtime": 26.4065,
"eval_samples_per_second": 5.68,
"eval_steps_per_second": 0.72,
"step": 107
},
{
"epoch": 2.0,
"grad_norm": 68.27020263671875,
"learning_rate": 4.023364485981308e-05,
"loss": 2.1242,
"step": 214
},
{
"epoch": 2.0,
"eval_loss": 2.1711459159851074,
"eval_map": 0.0464,
"eval_map_50": 0.1115,
"eval_map_75": 0.033,
"eval_map_Coverall": 0.1861,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0139,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0317,
"eval_map_large": 0.0621,
"eval_map_medium": 0.0466,
"eval_map_small": 0.01,
"eval_mar_1": 0.0776,
"eval_mar_10": 0.1595,
"eval_mar_100": 0.1801,
"eval_mar_100_Coverall": 0.5176,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.1437,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2391,
"eval_mar_large": 0.2105,
"eval_mar_medium": 0.1406,
"eval_mar_small": 0.0647,
"eval_runtime": 25.1578,
"eval_samples_per_second": 5.962,
"eval_steps_per_second": 0.755,
"step": 214
},
{
"epoch": 3.0,
"grad_norm": 103.06488800048828,
"learning_rate": 3.523364485981308e-05,
"loss": 1.9759,
"step": 321
},
{
"epoch": 3.0,
"eval_loss": 2.067054271697998,
"eval_map": 0.0662,
"eval_map_50": 0.1477,
"eval_map_75": 0.053,
"eval_map_Coverall": 0.2207,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0222,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0881,
"eval_map_large": 0.0846,
"eval_map_medium": 0.0709,
"eval_map_small": 0.0129,
"eval_mar_1": 0.0964,
"eval_mar_10": 0.1814,
"eval_mar_100": 0.199,
"eval_mar_100_Coverall": 0.5608,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.1688,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2653,
"eval_mar_large": 0.2279,
"eval_mar_medium": 0.1639,
"eval_mar_small": 0.057,
"eval_runtime": 25.5448,
"eval_samples_per_second": 5.872,
"eval_steps_per_second": 0.744,
"step": 321
},
{
"epoch": 4.0,
"grad_norm": 34.24587631225586,
"learning_rate": 3.0233644859813087e-05,
"loss": 1.8435,
"step": 428
},
{
"epoch": 4.0,
"eval_loss": 1.992311954498291,
"eval_map": 0.084,
"eval_map_50": 0.1759,
"eval_map_75": 0.0717,
"eval_map_Coverall": 0.2915,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0302,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0982,
"eval_map_large": 0.1068,
"eval_map_medium": 0.0673,
"eval_map_small": 0.0156,
"eval_mar_1": 0.1017,
"eval_mar_10": 0.1929,
"eval_mar_100": 0.2119,
"eval_mar_100_Coverall": 0.5856,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.1835,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2907,
"eval_mar_large": 0.2621,
"eval_mar_medium": 0.1646,
"eval_mar_small": 0.0717,
"eval_runtime": 25.7096,
"eval_samples_per_second": 5.834,
"eval_steps_per_second": 0.739,
"step": 428
},
{
"epoch": 5.0,
"grad_norm": 37.823734283447266,
"learning_rate": 2.5233644859813084e-05,
"loss": 1.7693,
"step": 535
},
{
"epoch": 5.0,
"eval_loss": 1.9163419008255005,
"eval_map": 0.0851,
"eval_map_50": 0.181,
"eval_map_75": 0.0718,
"eval_map_Coverall": 0.2895,
"eval_map_Face_Shield": 0.004,
"eval_map_Gloves": 0.0256,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.1064,
"eval_map_large": 0.1072,
"eval_map_medium": 0.0721,
"eval_map_small": 0.0201,
"eval_mar_1": 0.1005,
"eval_mar_10": 0.1956,
"eval_mar_100": 0.2137,
"eval_mar_100_Coverall": 0.5559,
"eval_mar_100_Face_Shield": 0.0063,
"eval_mar_100_Gloves": 0.1808,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.3253,
"eval_mar_large": 0.2556,
"eval_mar_medium": 0.1684,
"eval_mar_small": 0.0924,
"eval_runtime": 25.8375,
"eval_samples_per_second": 5.806,
"eval_steps_per_second": 0.735,
"step": 535
},
{
"epoch": 6.0,
"grad_norm": 65.30372619628906,
"learning_rate": 2.0233644859813085e-05,
"loss": 1.6961,
"step": 642
},
{
"epoch": 6.0,
"eval_loss": 1.8519624471664429,
"eval_map": 0.1045,
"eval_map_50": 0.2193,
"eval_map_75": 0.0909,
"eval_map_Coverall": 0.3181,
"eval_map_Face_Shield": 0.0062,
"eval_map_Gloves": 0.0432,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.1549,
"eval_map_large": 0.133,
"eval_map_medium": 0.0938,
"eval_map_small": 0.0529,
"eval_mar_1": 0.1183,
"eval_mar_10": 0.2171,
"eval_mar_100": 0.2416,
"eval_mar_100_Coverall": 0.5748,
"eval_mar_100_Face_Shield": 0.0329,
"eval_mar_100_Gloves": 0.2598,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.3404,
"eval_mar_large": 0.2975,
"eval_mar_medium": 0.2093,
"eval_mar_small": 0.1111,
"eval_runtime": 27.2155,
"eval_samples_per_second": 5.512,
"eval_steps_per_second": 0.698,
"step": 642
},
{
"epoch": 7.0,
"grad_norm": 49.685272216796875,
"learning_rate": 1.5233644859813085e-05,
"loss": 1.6116,
"step": 749
},
{
"epoch": 7.0,
"eval_loss": 1.7835921049118042,
"eval_map": 0.1118,
"eval_map_50": 0.2368,
"eval_map_75": 0.089,
"eval_map_Coverall": 0.3489,
"eval_map_Face_Shield": 0.0081,
"eval_map_Gloves": 0.0517,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.1503,
"eval_map_large": 0.1543,
"eval_map_medium": 0.0935,
"eval_map_small": 0.0334,
"eval_mar_1": 0.1308,
"eval_mar_10": 0.2439,
"eval_mar_100": 0.2684,
"eval_mar_100_Coverall": 0.6059,
"eval_mar_100_Face_Shield": 0.1165,
"eval_mar_100_Gloves": 0.2888,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.3307,
"eval_mar_large": 0.3409,
"eval_mar_medium": 0.2151,
"eval_mar_small": 0.1294,
"eval_runtime": 25.3036,
"eval_samples_per_second": 5.928,
"eval_steps_per_second": 0.751,
"step": 749
},
{
"epoch": 8.0,
"grad_norm": 28.98670768737793,
"learning_rate": 1.0233644859813085e-05,
"loss": 1.5518,
"step": 856
},
{
"epoch": 8.0,
"eval_loss": 1.722257137298584,
"eval_map": 0.1235,
"eval_map_50": 0.2558,
"eval_map_75": 0.1096,
"eval_map_Coverall": 0.3847,
"eval_map_Face_Shield": 0.0237,
"eval_map_Gloves": 0.0556,
"eval_map_Goggles": 0.0001,
"eval_map_Mask": 0.1537,
"eval_map_large": 0.1553,
"eval_map_medium": 0.1039,
"eval_map_small": 0.0336,
"eval_mar_1": 0.135,
"eval_mar_10": 0.2555,
"eval_mar_100": 0.2765,
"eval_mar_100_Coverall": 0.6203,
"eval_mar_100_Face_Shield": 0.1494,
"eval_mar_100_Gloves": 0.2661,
"eval_mar_100_Goggles": 0.0015,
"eval_mar_100_Mask": 0.3453,
"eval_mar_large": 0.3411,
"eval_mar_medium": 0.2311,
"eval_mar_small": 0.1211,
"eval_runtime": 24.7724,
"eval_samples_per_second": 6.055,
"eval_steps_per_second": 0.767,
"step": 856
},
{
"epoch": 9.0,
"grad_norm": 27.121028900146484,
"learning_rate": 5.233644859813085e-06,
"loss": 1.5112,
"step": 963
},
{
"epoch": 9.0,
"eval_loss": 1.6986401081085205,
"eval_map": 0.1268,
"eval_map_50": 0.2639,
"eval_map_75": 0.1029,
"eval_map_Coverall": 0.3928,
"eval_map_Face_Shield": 0.0274,
"eval_map_Gloves": 0.0536,
"eval_map_Goggles": 0.0012,
"eval_map_Mask": 0.1591,
"eval_map_large": 0.1676,
"eval_map_medium": 0.1025,
"eval_map_small": 0.0318,
"eval_mar_1": 0.1444,
"eval_mar_10": 0.2626,
"eval_mar_100": 0.2864,
"eval_mar_100_Coverall": 0.6392,
"eval_mar_100_Face_Shield": 0.157,
"eval_mar_100_Gloves": 0.2741,
"eval_mar_100_Goggles": 0.0062,
"eval_mar_100_Mask": 0.3556,
"eval_mar_large": 0.3611,
"eval_mar_medium": 0.2328,
"eval_mar_small": 0.1204,
"eval_runtime": 24.1622,
"eval_samples_per_second": 6.208,
"eval_steps_per_second": 0.786,
"step": 963
},
{
"epoch": 10.0,
"grad_norm": 38.41913986206055,
"learning_rate": 2.803738317757009e-07,
"loss": 1.4924,
"step": 1070
},
{
"epoch": 10.0,
"eval_loss": 1.693162202835083,
"eval_map": 0.1289,
"eval_map_50": 0.2606,
"eval_map_75": 0.1098,
"eval_map_Coverall": 0.399,
"eval_map_Face_Shield": 0.0257,
"eval_map_Gloves": 0.0535,
"eval_map_Goggles": 0.0002,
"eval_map_Mask": 0.166,
"eval_map_large": 0.1676,
"eval_map_medium": 0.1082,
"eval_map_small": 0.0355,
"eval_mar_1": 0.1429,
"eval_mar_10": 0.2628,
"eval_mar_100": 0.2869,
"eval_mar_100_Coverall": 0.6383,
"eval_mar_100_Face_Shield": 0.1557,
"eval_mar_100_Gloves": 0.2772,
"eval_mar_100_Goggles": 0.0031,
"eval_mar_100_Mask": 0.36,
"eval_mar_large": 0.3635,
"eval_mar_medium": 0.2299,
"eval_mar_small": 0.1256,
"eval_runtime": 25.4951,
"eval_samples_per_second": 5.883,
"eval_steps_per_second": 0.745,
"step": 1070
},
{
"epoch": 10.0,
"step": 1070,
"total_flos": 2.28455679528e+18,
"train_loss": 1.8261713616201811,
"train_runtime": 1263.216,
"train_samples_per_second": 6.729,
"train_steps_per_second": 0.847
}
],
"logging_steps": 500,
"max_steps": 1070,
"num_input_tokens_seen": 0,
"num_train_epochs": 10,
"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": 2.28455679528e+18,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}