detr-finetuned-cppe-5-10k-steps / trainer_state.json
Rajerswari's picture
End of training
9d54790 verified
{
"best_global_step": 2130,
"best_metric": 0.1619,
"best_model_checkpoint": "detr-finetuned-cppe-5-10k-steps\\checkpoint-2130",
"epoch": 10.0,
"eval_steps": 500,
"global_step": 2130,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 75.64894104003906,
"learning_rate": 2.5821596244131456e-05,
"loss": 2.5499,
"step": 213
},
{
"epoch": 1.0,
"eval_loss": 2.3247692584991455,
"eval_map": 0.0367,
"eval_map_50": 0.0798,
"eval_map_75": 0.0325,
"eval_map_Coverall": 0.1697,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0078,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0058,
"eval_map_large": 0.0417,
"eval_map_medium": 0.0114,
"eval_map_small": 0.0018,
"eval_mar_1": 0.0555,
"eval_mar_10": 0.1278,
"eval_mar_100": 0.1648,
"eval_mar_100_Coverall": 0.5104,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.1813,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.1324,
"eval_mar_large": 0.2026,
"eval_mar_medium": 0.1044,
"eval_mar_small": 0.0416,
"eval_runtime": 58.8392,
"eval_samples_per_second": 2.549,
"eval_steps_per_second": 0.323,
"step": 213
},
{
"epoch": 2.0,
"grad_norm": 47.80084228515625,
"learning_rate": 9.389671361502348e-07,
"loss": 2.1119,
"step": 426
},
{
"epoch": 2.0,
"eval_loss": 2.0866501331329346,
"eval_map": 0.0493,
"eval_map_50": 0.1064,
"eval_map_75": 0.0379,
"eval_map_Coverall": 0.1978,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0253,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0236,
"eval_map_large": 0.0574,
"eval_map_medium": 0.0304,
"eval_map_small": 0.0072,
"eval_mar_1": 0.0805,
"eval_mar_10": 0.1667,
"eval_mar_100": 0.2069,
"eval_mar_100_Coverall": 0.5874,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.204,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2431,
"eval_mar_large": 0.2325,
"eval_mar_medium": 0.1572,
"eval_mar_small": 0.073,
"eval_runtime": 61.6528,
"eval_samples_per_second": 2.433,
"eval_steps_per_second": 0.308,
"step": 426
},
{
"epoch": 3.0,
"grad_norm": 48.019927978515625,
"learning_rate": 3.5187793427230045e-05,
"loss": 2.0052,
"step": 639
},
{
"epoch": 3.0,
"eval_loss": 2.168910026550293,
"eval_map": 0.0563,
"eval_map_50": 0.1279,
"eval_map_75": 0.0441,
"eval_map_Coverall": 0.2039,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0158,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0618,
"eval_map_large": 0.0688,
"eval_map_medium": 0.0414,
"eval_map_small": 0.0154,
"eval_mar_1": 0.0818,
"eval_mar_10": 0.1611,
"eval_mar_100": 0.1793,
"eval_mar_100_Coverall": 0.4757,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.1942,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2267,
"eval_mar_large": 0.2199,
"eval_mar_medium": 0.131,
"eval_mar_small": 0.0616,
"eval_runtime": 55.4656,
"eval_samples_per_second": 2.704,
"eval_steps_per_second": 0.343,
"step": 639
},
{
"epoch": 4.0,
"grad_norm": 124.8763427734375,
"learning_rate": 3.018779342723005e-05,
"loss": 1.9373,
"step": 852
},
{
"epoch": 4.0,
"eval_loss": 1.9264371395111084,
"eval_map": 0.0813,
"eval_map_50": 0.1755,
"eval_map_75": 0.0679,
"eval_map_Coverall": 0.3464,
"eval_map_Face_Shield": 0.0,
"eval_map_Gloves": 0.0245,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0357,
"eval_map_large": 0.0953,
"eval_map_medium": 0.056,
"eval_map_small": 0.0125,
"eval_mar_1": 0.0916,
"eval_mar_10": 0.1816,
"eval_mar_100": 0.206,
"eval_mar_100_Coverall": 0.6302,
"eval_mar_100_Face_Shield": 0.0,
"eval_mar_100_Gloves": 0.2004,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.1996,
"eval_mar_large": 0.2549,
"eval_mar_medium": 0.1446,
"eval_mar_small": 0.0662,
"eval_runtime": 63.4064,
"eval_samples_per_second": 2.366,
"eval_steps_per_second": 0.3,
"step": 852
},
{
"epoch": 5.0,
"grad_norm": 49.01365661621094,
"learning_rate": 2.518779342723005e-05,
"loss": 1.8396,
"step": 1065
},
{
"epoch": 5.0,
"eval_loss": 1.841763973236084,
"eval_map": 0.0958,
"eval_map_50": 0.208,
"eval_map_75": 0.0731,
"eval_map_Coverall": 0.3278,
"eval_map_Face_Shield": 0.0228,
"eval_map_Gloves": 0.031,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.0975,
"eval_map_large": 0.1163,
"eval_map_medium": 0.0751,
"eval_map_small": 0.0194,
"eval_mar_1": 0.116,
"eval_mar_10": 0.2195,
"eval_mar_100": 0.2399,
"eval_mar_100_Coverall": 0.6054,
"eval_mar_100_Face_Shield": 0.1127,
"eval_mar_100_Gloves": 0.2071,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2742,
"eval_mar_large": 0.2939,
"eval_mar_medium": 0.1802,
"eval_mar_small": 0.095,
"eval_runtime": 62.8825,
"eval_samples_per_second": 2.385,
"eval_steps_per_second": 0.302,
"step": 1065
},
{
"epoch": 6.0,
"grad_norm": 49.57404327392578,
"learning_rate": 2.0187793427230047e-05,
"loss": 1.7659,
"step": 1278
},
{
"epoch": 6.0,
"eval_loss": 1.8737202882766724,
"eval_map": 0.1004,
"eval_map_50": 0.2399,
"eval_map_75": 0.0791,
"eval_map_Coverall": 0.2906,
"eval_map_Face_Shield": 0.0439,
"eval_map_Gloves": 0.0497,
"eval_map_Goggles": 0.0,
"eval_map_Mask": 0.1178,
"eval_map_large": 0.1207,
"eval_map_medium": 0.0913,
"eval_map_small": 0.0377,
"eval_mar_1": 0.1226,
"eval_mar_10": 0.2204,
"eval_mar_100": 0.2382,
"eval_mar_100_Coverall": 0.5856,
"eval_mar_100_Face_Shield": 0.1165,
"eval_mar_100_Gloves": 0.2281,
"eval_mar_100_Goggles": 0.0,
"eval_mar_100_Mask": 0.2609,
"eval_mar_large": 0.295,
"eval_mar_medium": 0.1912,
"eval_mar_small": 0.0808,
"eval_runtime": 56.4295,
"eval_samples_per_second": 2.658,
"eval_steps_per_second": 0.337,
"step": 1278
},
{
"epoch": 7.0,
"grad_norm": 79.07571411132812,
"learning_rate": 1.5187793427230049e-05,
"loss": 1.6415,
"step": 1491
},
{
"epoch": 7.0,
"eval_loss": 1.720004677772522,
"eval_map": 0.1305,
"eval_map_50": 0.2822,
"eval_map_75": 0.105,
"eval_map_Coverall": 0.3916,
"eval_map_Face_Shield": 0.0516,
"eval_map_Gloves": 0.0526,
"eval_map_Goggles": 0.0059,
"eval_map_Mask": 0.1508,
"eval_map_large": 0.172,
"eval_map_medium": 0.1026,
"eval_map_small": 0.044,
"eval_mar_1": 0.1505,
"eval_mar_10": 0.2689,
"eval_mar_100": 0.2891,
"eval_mar_100_Coverall": 0.6257,
"eval_mar_100_Face_Shield": 0.1962,
"eval_mar_100_Gloves": 0.2554,
"eval_mar_100_Goggles": 0.0523,
"eval_mar_100_Mask": 0.316,
"eval_mar_large": 0.386,
"eval_mar_medium": 0.2239,
"eval_mar_small": 0.1047,
"eval_runtime": 63.9942,
"eval_samples_per_second": 2.344,
"eval_steps_per_second": 0.297,
"step": 1491
},
{
"epoch": 8.0,
"grad_norm": 68.87022399902344,
"learning_rate": 1.0187793427230047e-05,
"loss": 1.6405,
"step": 1704
},
{
"epoch": 8.0,
"eval_loss": 1.6820415258407593,
"eval_map": 0.1411,
"eval_map_50": 0.2866,
"eval_map_75": 0.1303,
"eval_map_Coverall": 0.4147,
"eval_map_Face_Shield": 0.0549,
"eval_map_Gloves": 0.0592,
"eval_map_Goggles": 0.0053,
"eval_map_Mask": 0.1713,
"eval_map_large": 0.1854,
"eval_map_medium": 0.1162,
"eval_map_small": 0.0542,
"eval_mar_1": 0.1572,
"eval_mar_10": 0.2867,
"eval_mar_100": 0.3088,
"eval_mar_100_Coverall": 0.6347,
"eval_mar_100_Face_Shield": 0.2468,
"eval_mar_100_Gloves": 0.271,
"eval_mar_100_Goggles": 0.0631,
"eval_mar_100_Mask": 0.3284,
"eval_mar_large": 0.4042,
"eval_mar_medium": 0.2432,
"eval_mar_small": 0.1204,
"eval_runtime": 61.79,
"eval_samples_per_second": 2.428,
"eval_steps_per_second": 0.307,
"step": 1704
},
{
"epoch": 9.0,
"grad_norm": 28.72616195678711,
"learning_rate": 5.211267605633803e-06,
"loss": 1.5513,
"step": 1917
},
{
"epoch": 9.0,
"eval_loss": 1.6379631757736206,
"eval_map": 0.1546,
"eval_map_50": 0.3144,
"eval_map_75": 0.1307,
"eval_map_Coverall": 0.4352,
"eval_map_Face_Shield": 0.0804,
"eval_map_Gloves": 0.0657,
"eval_map_Goggles": 0.0062,
"eval_map_Mask": 0.1853,
"eval_map_large": 0.2131,
"eval_map_medium": 0.1215,
"eval_map_small": 0.0569,
"eval_mar_1": 0.17,
"eval_mar_10": 0.3014,
"eval_mar_100": 0.323,
"eval_mar_100_Coverall": 0.6414,
"eval_mar_100_Face_Shield": 0.2544,
"eval_mar_100_Gloves": 0.2871,
"eval_mar_100_Goggles": 0.0846,
"eval_mar_100_Mask": 0.3476,
"eval_mar_large": 0.4431,
"eval_mar_medium": 0.2563,
"eval_mar_small": 0.112,
"eval_runtime": 61.6355,
"eval_samples_per_second": 2.434,
"eval_steps_per_second": 0.308,
"step": 1917
},
{
"epoch": 10.0,
"grad_norm": 53.006046295166016,
"learning_rate": 2.112676056338028e-07,
"loss": 1.5564,
"step": 2130
},
{
"epoch": 10.0,
"eval_loss": 1.623809814453125,
"eval_map": 0.1619,
"eval_map_50": 0.3226,
"eval_map_75": 0.1429,
"eval_map_Coverall": 0.4278,
"eval_map_Face_Shield": 0.1078,
"eval_map_Gloves": 0.0679,
"eval_map_Goggles": 0.0102,
"eval_map_Mask": 0.1959,
"eval_map_large": 0.227,
"eval_map_medium": 0.129,
"eval_map_small": 0.0501,
"eval_mar_1": 0.1773,
"eval_mar_10": 0.3167,
"eval_mar_100": 0.3392,
"eval_mar_100_Coverall": 0.6532,
"eval_mar_100_Face_Shield": 0.2937,
"eval_mar_100_Gloves": 0.2991,
"eval_mar_100_Goggles": 0.0985,
"eval_mar_100_Mask": 0.3516,
"eval_mar_large": 0.4711,
"eval_mar_medium": 0.2626,
"eval_mar_small": 0.128,
"eval_runtime": 65.5535,
"eval_samples_per_second": 2.288,
"eval_steps_per_second": 0.29,
"step": 2130
},
{
"epoch": 10.0,
"step": 2130,
"total_flos": 2.2764936536496e+18,
"train_loss": 1.393765215806558,
"train_runtime": 1396.8073,
"train_samples_per_second": 6.085,
"train_steps_per_second": 1.525
}
],
"logging_steps": 500,
"max_steps": 2130,
"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.2764936536496e+18,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}