|
{
|
|
"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
|
|
}
|
|
|