emvisee's picture
End of training
77b9315 verified
{
"best_metric": 0.6264,
"best_model_checkpoint": "rtdetr-finetuned/checkpoint-1752",
"epoch": 10.0,
"eval_steps": 500,
"global_step": 4380,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 51.22775650024414,
"learning_rate": 2.9657534246575345e-05,
"loss": 23.0537,
"step": 438
},
{
"epoch": 1.0,
"eval_loss": 9.24314022064209,
"eval_map": 0.6183,
"eval_map_50": 0.7399,
"eval_map_75": 0.7053,
"eval_map_baseball-bat": 0.6135,
"eval_map_basketball": 0.4784,
"eval_map_car": -1.0,
"eval_map_football": 0.5092,
"eval_map_human": 0.7931,
"eval_map_large": 0.6277,
"eval_map_luggage": 0.7264,
"eval_map_mattress": 0.4151,
"eval_map_medium": 0.6975,
"eval_map_motorcycle": 0.7746,
"eval_map_skis": 0.7873,
"eval_map_small": 0.4862,
"eval_map_snowboard": 0.7978,
"eval_map_soccer-ball": 0.3779,
"eval_map_stop-sign": 0.8549,
"eval_map_tennis-racket": 0.6769,
"eval_map_umbrella": 0.377,
"eval_map_volleyball": 0.4746,
"eval_mar_1": 0.698,
"eval_mar_10": 0.8445,
"eval_mar_100": 0.863,
"eval_mar_100_baseball-bat": 0.8496,
"eval_mar_100_basketball": 0.7922,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.7406,
"eval_mar_100_human": 0.9349,
"eval_mar_100_luggage": 0.8453,
"eval_mar_100_mattress": 0.9649,
"eval_mar_100_motorcycle": 0.8995,
"eval_mar_100_skis": 0.9153,
"eval_mar_100_snowboard": 0.9122,
"eval_mar_100_soccer-ball": 0.6828,
"eval_mar_100_stop-sign": 0.957,
"eval_mar_100_tennis-racket": 0.885,
"eval_mar_100_umbrella": 0.9438,
"eval_mar_100_volleyball": 0.7593,
"eval_mar_large": 0.9333,
"eval_mar_medium": 0.8914,
"eval_mar_small": 0.727,
"eval_runtime": 93.4684,
"eval_samples_per_second": 16.048,
"eval_steps_per_second": 1.006,
"step": 438
},
{
"epoch": 2.0,
"grad_norm": 37.76764678955078,
"learning_rate": 3e-05,
"loss": 9.6444,
"step": 876
},
{
"epoch": 2.0,
"eval_loss": 8.435422897338867,
"eval_map": 0.5654,
"eval_map_50": 0.6543,
"eval_map_75": 0.6454,
"eval_map_baseball-bat": 0.6536,
"eval_map_basketball": 0.4753,
"eval_map_car": -1.0,
"eval_map_football": 0.4716,
"eval_map_human": 0.7632,
"eval_map_large": 0.6517,
"eval_map_luggage": 0.7675,
"eval_map_mattress": 0.1446,
"eval_map_medium": 0.6759,
"eval_map_motorcycle": 0.7513,
"eval_map_skis": 0.8092,
"eval_map_small": 0.4377,
"eval_map_snowboard": 0.796,
"eval_map_soccer-ball": 0.1811,
"eval_map_stop-sign": 0.6988,
"eval_map_tennis-racket": 0.734,
"eval_map_umbrella": 0.2025,
"eval_map_volleyball": 0.4669,
"eval_mar_1": 0.6178,
"eval_mar_10": 0.8051,
"eval_mar_100": 0.8567,
"eval_mar_100_baseball-bat": 0.8175,
"eval_mar_100_basketball": 0.7836,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.7781,
"eval_mar_100_human": 0.925,
"eval_mar_100_luggage": 0.8975,
"eval_mar_100_mattress": 0.8849,
"eval_mar_100_motorcycle": 0.884,
"eval_mar_100_skis": 0.8936,
"eval_mar_100_snowboard": 0.9045,
"eval_mar_100_soccer-ball": 0.8046,
"eval_mar_100_stop-sign": 0.9162,
"eval_mar_100_tennis-racket": 0.8762,
"eval_mar_100_umbrella": 0.8361,
"eval_mar_100_volleyball": 0.7924,
"eval_mar_large": 0.9001,
"eval_mar_medium": 0.8748,
"eval_mar_small": 0.7616,
"eval_runtime": 78.3885,
"eval_samples_per_second": 19.135,
"eval_steps_per_second": 1.199,
"step": 876
},
{
"epoch": 3.0,
"grad_norm": 37.99458312988281,
"learning_rate": 3e-05,
"loss": 7.9736,
"step": 1314
},
{
"epoch": 3.0,
"eval_loss": 8.123346328735352,
"eval_map": 0.6032,
"eval_map_50": 0.7082,
"eval_map_75": 0.6974,
"eval_map_baseball-bat": 0.6745,
"eval_map_basketball": 0.5687,
"eval_map_car": -1.0,
"eval_map_football": 0.5098,
"eval_map_human": 0.8075,
"eval_map_large": 0.7013,
"eval_map_luggage": 0.7803,
"eval_map_mattress": 0.1936,
"eval_map_medium": 0.7302,
"eval_map_motorcycle": 0.8528,
"eval_map_skis": 0.8126,
"eval_map_small": 0.4632,
"eval_map_snowboard": 0.8344,
"eval_map_soccer-ball": 0.2384,
"eval_map_stop-sign": 0.8343,
"eval_map_tennis-racket": 0.7266,
"eval_map_umbrella": 0.2001,
"eval_map_volleyball": 0.4116,
"eval_mar_1": 0.6284,
"eval_mar_10": 0.8059,
"eval_mar_100": 0.8518,
"eval_mar_100_baseball-bat": 0.7985,
"eval_mar_100_basketball": 0.8135,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.7643,
"eval_mar_100_human": 0.917,
"eval_mar_100_luggage": 0.8888,
"eval_mar_100_mattress": 0.8645,
"eval_mar_100_motorcycle": 0.9092,
"eval_mar_100_skis": 0.903,
"eval_mar_100_snowboard": 0.8903,
"eval_mar_100_soccer-ball": 0.8224,
"eval_mar_100_stop-sign": 0.9179,
"eval_mar_100_tennis-racket": 0.8549,
"eval_mar_100_umbrella": 0.7809,
"eval_mar_100_volleyball": 0.8002,
"eval_mar_large": 0.8919,
"eval_mar_medium": 0.8713,
"eval_mar_small": 0.7726,
"eval_runtime": 79.4809,
"eval_samples_per_second": 18.872,
"eval_steps_per_second": 1.183,
"step": 1314
},
{
"epoch": 4.0,
"grad_norm": 48.02970504760742,
"learning_rate": 3e-05,
"loss": 7.5623,
"step": 1752
},
{
"epoch": 4.0,
"eval_loss": 8.071721076965332,
"eval_map": 0.6264,
"eval_map_50": 0.7342,
"eval_map_75": 0.7215,
"eval_map_baseball-bat": 0.7056,
"eval_map_basketball": 0.6522,
"eval_map_car": -1.0,
"eval_map_football": 0.5832,
"eval_map_human": 0.7739,
"eval_map_large": 0.7239,
"eval_map_luggage": 0.808,
"eval_map_mattress": 0.0914,
"eval_map_medium": 0.734,
"eval_map_motorcycle": 0.8597,
"eval_map_skis": 0.8828,
"eval_map_small": 0.5182,
"eval_map_snowboard": 0.8274,
"eval_map_soccer-ball": 0.2734,
"eval_map_stop-sign": 0.8612,
"eval_map_tennis-racket": 0.7139,
"eval_map_umbrella": 0.2524,
"eval_map_volleyball": 0.4843,
"eval_mar_1": 0.651,
"eval_mar_10": 0.8219,
"eval_mar_100": 0.8665,
"eval_mar_100_baseball-bat": 0.8205,
"eval_mar_100_basketball": 0.8386,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.8142,
"eval_mar_100_human": 0.9234,
"eval_mar_100_luggage": 0.8942,
"eval_mar_100_mattress": 0.8634,
"eval_mar_100_motorcycle": 0.9093,
"eval_mar_100_skis": 0.9376,
"eval_mar_100_snowboard": 0.8924,
"eval_mar_100_soccer-ball": 0.8501,
"eval_mar_100_stop-sign": 0.9209,
"eval_mar_100_tennis-racket": 0.8124,
"eval_mar_100_umbrella": 0.8567,
"eval_mar_100_volleyball": 0.7968,
"eval_mar_large": 0.9228,
"eval_mar_medium": 0.8706,
"eval_mar_small": 0.7837,
"eval_runtime": 81.9891,
"eval_samples_per_second": 18.295,
"eval_steps_per_second": 1.146,
"step": 1752
},
{
"epoch": 5.0,
"grad_norm": 90.77857208251953,
"learning_rate": 3e-05,
"loss": 7.2751,
"step": 2190
},
{
"epoch": 5.0,
"eval_loss": 8.76298713684082,
"eval_map": 0.5761,
"eval_map_50": 0.6678,
"eval_map_75": 0.6549,
"eval_map_baseball-bat": 0.5765,
"eval_map_basketball": 0.5702,
"eval_map_car": -1.0,
"eval_map_football": 0.4783,
"eval_map_human": 0.7352,
"eval_map_large": 0.6059,
"eval_map_luggage": 0.7497,
"eval_map_mattress": 0.0284,
"eval_map_medium": 0.6658,
"eval_map_motorcycle": 0.7861,
"eval_map_skis": 0.8352,
"eval_map_small": 0.4548,
"eval_map_snowboard": 0.833,
"eval_map_soccer-ball": 0.4065,
"eval_map_stop-sign": 0.7123,
"eval_map_tennis-racket": 0.655,
"eval_map_umbrella": 0.2883,
"eval_map_volleyball": 0.4105,
"eval_mar_1": 0.6224,
"eval_mar_10": 0.7803,
"eval_mar_100": 0.8305,
"eval_mar_100_baseball-bat": 0.7835,
"eval_mar_100_basketball": 0.7898,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.7281,
"eval_mar_100_human": 0.9097,
"eval_mar_100_luggage": 0.8839,
"eval_mar_100_mattress": 0.8625,
"eval_mar_100_motorcycle": 0.8506,
"eval_mar_100_skis": 0.9277,
"eval_mar_100_snowboard": 0.8627,
"eval_mar_100_soccer-ball": 0.7625,
"eval_mar_100_stop-sign": 0.9137,
"eval_mar_100_tennis-racket": 0.8026,
"eval_mar_100_umbrella": 0.8388,
"eval_mar_100_volleyball": 0.7105,
"eval_mar_large": 0.9175,
"eval_mar_medium": 0.8521,
"eval_mar_small": 0.7163,
"eval_runtime": 80.9984,
"eval_samples_per_second": 18.519,
"eval_steps_per_second": 1.161,
"step": 2190
},
{
"epoch": 6.0,
"grad_norm": 52.22979736328125,
"learning_rate": 3e-05,
"loss": 7.2919,
"step": 2628
},
{
"epoch": 6.0,
"eval_loss": 8.339943885803223,
"eval_map": 0.6044,
"eval_map_50": 0.7342,
"eval_map_75": 0.6966,
"eval_map_baseball-bat": 0.6153,
"eval_map_basketball": 0.5137,
"eval_map_car": -1.0,
"eval_map_football": 0.5038,
"eval_map_human": 0.8326,
"eval_map_large": 0.7245,
"eval_map_luggage": 0.7848,
"eval_map_mattress": 0.0252,
"eval_map_medium": 0.6987,
"eval_map_motorcycle": 0.803,
"eval_map_skis": 0.8442,
"eval_map_small": 0.4756,
"eval_map_snowboard": 0.8247,
"eval_map_soccer-ball": 0.4423,
"eval_map_stop-sign": 0.8512,
"eval_map_tennis-racket": 0.6717,
"eval_map_umbrella": 0.3123,
"eval_map_volleyball": 0.4365,
"eval_mar_1": 0.6285,
"eval_mar_10": 0.7863,
"eval_mar_100": 0.8232,
"eval_mar_100_baseball-bat": 0.7723,
"eval_mar_100_basketball": 0.7211,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.7329,
"eval_mar_100_human": 0.9302,
"eval_mar_100_luggage": 0.8806,
"eval_mar_100_mattress": 0.7449,
"eval_mar_100_motorcycle": 0.8632,
"eval_mar_100_skis": 0.9158,
"eval_mar_100_snowboard": 0.8912,
"eval_mar_100_soccer-ball": 0.7519,
"eval_mar_100_stop-sign": 0.9458,
"eval_mar_100_tennis-racket": 0.813,
"eval_mar_100_umbrella": 0.8643,
"eval_mar_100_volleyball": 0.6975,
"eval_mar_large": 0.8962,
"eval_mar_medium": 0.8579,
"eval_mar_small": 0.6783,
"eval_runtime": 83.0051,
"eval_samples_per_second": 18.071,
"eval_steps_per_second": 1.132,
"step": 2628
},
{
"epoch": 7.0,
"grad_norm": 94.34876251220703,
"learning_rate": 3e-05,
"loss": 6.9699,
"step": 3066
},
{
"epoch": 7.0,
"eval_loss": 8.577847480773926,
"eval_map": 0.5806,
"eval_map_50": 0.7101,
"eval_map_75": 0.6569,
"eval_map_baseball-bat": 0.6135,
"eval_map_basketball": 0.5259,
"eval_map_car": -1.0,
"eval_map_football": 0.3524,
"eval_map_human": 0.8118,
"eval_map_large": 0.7185,
"eval_map_luggage": 0.7602,
"eval_map_mattress": 0.1264,
"eval_map_medium": 0.6901,
"eval_map_motorcycle": 0.7603,
"eval_map_skis": 0.8279,
"eval_map_small": 0.4371,
"eval_map_snowboard": 0.8645,
"eval_map_soccer-ball": 0.3614,
"eval_map_stop-sign": 0.711,
"eval_map_tennis-racket": 0.663,
"eval_map_umbrella": 0.3093,
"eval_map_volleyball": 0.4402,
"eval_mar_1": 0.6187,
"eval_mar_10": 0.7898,
"eval_mar_100": 0.8279,
"eval_mar_100_baseball-bat": 0.7665,
"eval_mar_100_basketball": 0.7135,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.6591,
"eval_mar_100_human": 0.9329,
"eval_mar_100_luggage": 0.8796,
"eval_mar_100_mattress": 0.8605,
"eval_mar_100_motorcycle": 0.8343,
"eval_mar_100_skis": 0.9312,
"eval_mar_100_snowboard": 0.9105,
"eval_mar_100_soccer-ball": 0.7684,
"eval_mar_100_stop-sign": 0.9377,
"eval_mar_100_tennis-racket": 0.8212,
"eval_mar_100_umbrella": 0.8638,
"eval_mar_100_volleyball": 0.7118,
"eval_mar_large": 0.9156,
"eval_mar_medium": 0.865,
"eval_mar_small": 0.6748,
"eval_runtime": 80.6096,
"eval_samples_per_second": 18.608,
"eval_steps_per_second": 1.166,
"step": 3066
},
{
"epoch": 8.0,
"grad_norm": 72.42890167236328,
"learning_rate": 3e-05,
"loss": 6.7247,
"step": 3504
},
{
"epoch": 8.0,
"eval_loss": 8.50732135772705,
"eval_map": 0.5993,
"eval_map_50": 0.7448,
"eval_map_75": 0.6827,
"eval_map_baseball-bat": 0.6037,
"eval_map_basketball": 0.6004,
"eval_map_car": -1.0,
"eval_map_football": 0.4472,
"eval_map_human": 0.8139,
"eval_map_large": 0.7777,
"eval_map_luggage": 0.7039,
"eval_map_mattress": 0.2481,
"eval_map_medium": 0.6121,
"eval_map_motorcycle": 0.8006,
"eval_map_skis": 0.885,
"eval_map_small": 0.4588,
"eval_map_snowboard": 0.8342,
"eval_map_soccer-ball": 0.398,
"eval_map_stop-sign": 0.5666,
"eval_map_tennis-racket": 0.5825,
"eval_map_umbrella": 0.4441,
"eval_map_volleyball": 0.4621,
"eval_mar_1": 0.6438,
"eval_mar_10": 0.8042,
"eval_mar_100": 0.8321,
"eval_mar_100_baseball-bat": 0.7583,
"eval_mar_100_basketball": 0.719,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.7137,
"eval_mar_100_human": 0.9333,
"eval_mar_100_luggage": 0.8694,
"eval_mar_100_mattress": 0.9102,
"eval_mar_100_motorcycle": 0.845,
"eval_mar_100_skis": 0.9599,
"eval_mar_100_snowboard": 0.8815,
"eval_mar_100_soccer-ball": 0.7242,
"eval_mar_100_stop-sign": 0.9377,
"eval_mar_100_tennis-racket": 0.8135,
"eval_mar_100_umbrella": 0.8946,
"eval_mar_100_volleyball": 0.689,
"eval_mar_large": 0.9399,
"eval_mar_medium": 0.8377,
"eval_mar_small": 0.6614,
"eval_runtime": 82.1105,
"eval_samples_per_second": 18.268,
"eval_steps_per_second": 1.145,
"step": 3504
},
{
"epoch": 9.0,
"grad_norm": 28.80988311767578,
"learning_rate": 3e-05,
"loss": 6.5664,
"step": 3942
},
{
"epoch": 9.0,
"eval_loss": 9.097264289855957,
"eval_map": 0.4861,
"eval_map_50": 0.6532,
"eval_map_75": 0.5236,
"eval_map_baseball-bat": 0.442,
"eval_map_basketball": 0.4655,
"eval_map_car": -1.0,
"eval_map_football": 0.3113,
"eval_map_human": 0.7863,
"eval_map_large": 0.6237,
"eval_map_luggage": 0.6849,
"eval_map_mattress": 0.0783,
"eval_map_medium": 0.536,
"eval_map_motorcycle": 0.6747,
"eval_map_skis": 0.8484,
"eval_map_small": 0.3138,
"eval_map_snowboard": 0.7503,
"eval_map_soccer-ball": 0.2637,
"eval_map_stop-sign": 0.4794,
"eval_map_tennis-racket": 0.4634,
"eval_map_umbrella": 0.2248,
"eval_map_volleyball": 0.3317,
"eval_mar_1": 0.5459,
"eval_mar_10": 0.7169,
"eval_mar_100": 0.7528,
"eval_mar_100_baseball-bat": 0.6763,
"eval_mar_100_basketball": 0.6182,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.5423,
"eval_mar_100_human": 0.9349,
"eval_mar_100_luggage": 0.8274,
"eval_mar_100_mattress": 0.7612,
"eval_mar_100_motorcycle": 0.7367,
"eval_mar_100_skis": 0.9386,
"eval_mar_100_snowboard": 0.8272,
"eval_mar_100_soccer-ball": 0.644,
"eval_mar_100_stop-sign": 0.9464,
"eval_mar_100_tennis-racket": 0.7295,
"eval_mar_100_umbrella": 0.744,
"eval_mar_100_volleyball": 0.6123,
"eval_mar_large": 0.8686,
"eval_mar_medium": 0.7868,
"eval_mar_small": 0.5389,
"eval_runtime": 81.6702,
"eval_samples_per_second": 18.367,
"eval_steps_per_second": 1.151,
"step": 3942
},
{
"epoch": 10.0,
"grad_norm": 74.36066436767578,
"learning_rate": 3e-05,
"loss": 6.2978,
"step": 4380
},
{
"epoch": 10.0,
"eval_loss": 9.184855461120605,
"eval_map": 0.4811,
"eval_map_50": 0.6742,
"eval_map_75": 0.5153,
"eval_map_baseball-bat": 0.49,
"eval_map_basketball": 0.561,
"eval_map_car": -1.0,
"eval_map_football": 0.3393,
"eval_map_human": 0.7301,
"eval_map_large": 0.5739,
"eval_map_luggage": 0.6005,
"eval_map_mattress": 0.048,
"eval_map_medium": 0.5129,
"eval_map_motorcycle": 0.5833,
"eval_map_skis": 0.8396,
"eval_map_small": 0.385,
"eval_map_snowboard": 0.6757,
"eval_map_soccer-ball": 0.3701,
"eval_map_stop-sign": 0.4419,
"eval_map_tennis-racket": 0.4382,
"eval_map_umbrella": 0.2355,
"eval_map_volleyball": 0.382,
"eval_mar_1": 0.5409,
"eval_mar_10": 0.7299,
"eval_mar_100": 0.7598,
"eval_mar_100_baseball-bat": 0.6707,
"eval_mar_100_basketball": 0.6806,
"eval_mar_100_car": -1.0,
"eval_mar_100_football": 0.6131,
"eval_mar_100_human": 0.9417,
"eval_mar_100_luggage": 0.8216,
"eval_mar_100_mattress": 0.6785,
"eval_mar_100_motorcycle": 0.6382,
"eval_mar_100_skis": 0.9198,
"eval_mar_100_snowboard": 0.8016,
"eval_mar_100_soccer-ball": 0.7694,
"eval_mar_100_stop-sign": 0.9539,
"eval_mar_100_tennis-racket": 0.7518,
"eval_mar_100_umbrella": 0.7241,
"eval_mar_100_volleyball": 0.6728,
"eval_mar_large": 0.8269,
"eval_mar_medium": 0.7825,
"eval_mar_small": 0.6142,
"eval_runtime": 80.9774,
"eval_samples_per_second": 18.524,
"eval_steps_per_second": 1.161,
"step": 4380
},
{
"epoch": 10.0,
"step": 4380,
"total_flos": 2.2065014071296e+19,
"train_loss": 8.935971166880707,
"train_runtime": 3440.1755,
"train_samples_per_second": 20.348,
"train_steps_per_second": 1.273
}
],
"logging_steps": 500,
"max_steps": 4380,
"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.2065014071296e+19,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}