{ "package_id": "Text2LocPP-SCAR-HT/PandaSetPose-50-25-25/bb0/frozen", "created": "2026-10-27", "inference": "(feats - mean) / std per column, then MLP 23->64->32->1 (ReLU, no output activation)", "pinned_rule": "columns with TRAIN std < 1e-3 pinned to std=1.0", "heads": [ { "method": "Text2Loc++ + SCAR(H-T)", "dataset_version": "PandaSetPose-50-25-25", "backbone_seed": 0, "head_seed": 0, "source_path": "", "bytes": 27780, "mean": [ 0.4608134329319, 0.5, 0.5343223214149475, 0.5, 0.5, 0.15052300691604614, 0.1505965143442154, 0.22912763059139252, 0.4999999701976776, -0.028604188933968544, -0.009919974021613598, 0.055696964263916016, 0.22624221444129944, 0.18046113848686218, 0.4999999701976776, 0.5000002980232239, 0.49999991059303284, 0.3710075318813324, 0.5338996648788452, 1.4515472650527954, 1.4602689743041992, 1.4773361682891846, 1.4770563840866089 ], "std": [ 0.116506427526474, 0.3004809617996216, 0.27012988924980164, 0.3004809617996216, 0.3004809617996216, 0.35746872425079346, 0.13275882601737976, 0.3479200005531311, 0.3004809617996216, 0.04183800518512726, 0.03888953477144241, 0.03282184898853302, 0.3440459370613098, 0.27901825308799744, 0.3004809617996216, 0.29166659712791443, 0.3004809617996216, 0.16501377522945404, 0.2703131437301636, 1.0025379657745361, 1.0008338689804077, 0.9802831411361694, 0.9835349321365356 ], "pinned": [ false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false ], "feature_names": [ "cos", "cos_rank", "sp", "p0_rank", "p2_rank", "is_tied", "tied_size", "sc_center", "sc_center_rank", "off_dx", "off_dy", "off_norm", "sc_fine", "sat_fine", "sc_fine_rank", "borda", "c_rank", "n_obj", "n_matched", "dist_nearest_matched", "dist_centroid_matched", "pred_dist_anchor", "center_dist_anchor" ], "config": { "epochs": 30, "lr": 0.001, "wd": 0.0001, "batch": 512, "hidden": [ 64, 32 ], "loss": "listwise_softmax_ce", "seed": 0, "k": 25, "dataset": "pandaset", "backbone": "text2locpp", "norms": { "pool_norm": 24, "rank_norm": 24, "tied_norm": 25 } }, "selection": { "rule": "max VAL cell R@1, tie max top1@5m, earlier epoch", "epoch": 30, "key": [ 0.4795, 0.4235 ], "val_cell": { "R@1": 0.4795, "R@2": 0.6285, "R@3": 0.7005, "R@5": 0.7765, "R@10": 0.863, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4235, "top3@5m": 0.611, "top5@5m": 0.676, "top10@5m": 0.753 } }, "history": [ { "epoch": 1, "loss": 3.2355407101319633, "train_top1": 0.284938553872535, "val_cell": { "R@1": 0.249, "R@2": 0.383, "R@3": 0.5005, "R@5": 0.6245, "R@10": 0.847, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.2205, "top3@5m": 0.44, "top5@5m": 0.534, "top10@5m": 0.7255 } }, { "epoch": 2, "loss": 3.0153459980679296, "train_top1": 0.4338382394969991, "val_cell": { "R@1": 0.3365, "R@2": 0.4975, "R@3": 0.6045, "R@5": 0.758, "R@10": 0.8945, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3005, "top3@5m": 0.535, "top5@5m": 0.6485, "top10@5m": 0.7655 } }, { "epoch": 3, "loss": 2.7865659389948294, "train_top1": 0.47870820234352673, "val_cell": { "R@1": 0.3725, "R@2": 0.523, "R@3": 0.6275, "R@5": 0.769, "R@10": 0.903, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.327, "top3@5m": 0.552, "top5@5m": 0.6575, "top10@5m": 0.7735 } }, { "epoch": 4, "loss": 2.5261723561162914, "train_top1": 0.49528436696198913, "val_cell": { "R@1": 0.3785, "R@2": 0.5405, "R@3": 0.6415, "R@5": 0.7705, "R@10": 0.909, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3345, "top3@5m": 0.561, "top5@5m": 0.6655, "top10@5m": 0.7795 } }, { "epoch": 5, "loss": 2.268064622369348, "train_top1": 0.5030008573878251, "val_cell": { "R@1": 0.384, "R@2": 0.5485, "R@3": 0.655, "R@5": 0.774, "R@10": 0.911, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.342, "top3@5m": 0.5705, "top5@5m": 0.664, "top10@5m": 0.781 } }, { "epoch": 6, "loss": 2.041821400483495, "train_top1": 0.5150042869391255, "val_cell": { "R@1": 0.4015, "R@2": 0.5725, "R@3": 0.6715, "R@5": 0.7975, "R@10": 0.9155, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3585, "top3@5m": 0.5825, "top5@5m": 0.6785, "top10@5m": 0.785 } }, { "epoch": 7, "loss": 1.8652910388309842, "train_top1": 0.5332952272077737, "val_cell": { "R@1": 0.4225, "R@2": 0.585, "R@3": 0.669, "R@5": 0.8055, "R@10": 0.92, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3735, "top3@5m": 0.583, "top5@5m": 0.686, "top10@5m": 0.7885 } }, { "epoch": 8, "loss": 1.7321244744649442, "train_top1": 0.5687339239782795, "val_cell": { "R@1": 0.4395, "R@2": 0.5925, "R@3": 0.666, "R@5": 0.8115, "R@10": 0.9165, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.381, "top3@5m": 0.5765, "top5@5m": 0.6905, "top10@5m": 0.784 } }, { "epoch": 9, "loss": 1.6265111728953308, "train_top1": 0.5827379251214633, "val_cell": { "R@1": 0.44, "R@2": 0.5895, "R@3": 0.669, "R@5": 0.806, "R@10": 0.9155, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.383, "top3@5m": 0.5815, "top5@5m": 0.6865, "top10@5m": 0.7835 } }, { "epoch": 10, "loss": 1.5514449462034798, "train_top1": 0.5904544155472993, "val_cell": { "R@1": 0.4505, "R@2": 0.5985, "R@3": 0.688, "R@5": 0.811, "R@10": 0.905, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.394, "top3@5m": 0.594, "top5@5m": 0.6925, "top10@5m": 0.772 } }, { "epoch": 11, "loss": 1.4944905900859806, "train_top1": 0.5904544155472993, "val_cell": { "R@1": 0.4535, "R@2": 0.602, "R@3": 0.6865, "R@5": 0.8025, "R@10": 0.899, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.397, "top3@5m": 0.593, "top5@5m": 0.6905, "top10@5m": 0.766 } }, { "epoch": 12, "loss": 1.4536659638859197, "train_top1": 0.5958845384395541, "val_cell": { "R@1": 0.452, "R@2": 0.6095, "R@3": 0.6815, "R@5": 0.7855, "R@10": 0.8865, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.395, "top3@5m": 0.5945, "top5@5m": 0.6775, "top10@5m": 0.757 } }, { "epoch": 13, "loss": 1.4212763514236642, "train_top1": 0.6018862532152044, "val_cell": { "R@1": 0.4605, "R@2": 0.6125, "R@3": 0.6865, "R@5": 0.7695, "R@10": 0.8775, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.403, "top3@5m": 0.599, "top5@5m": 0.6735, "top10@5m": 0.756 } }, { "epoch": 14, "loss": 1.3973085882323713, "train_top1": 0.6084595598742498, "val_cell": { "R@1": 0.464, "R@2": 0.6185, "R@3": 0.687, "R@5": 0.77, "R@10": 0.878, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4055, "top3@5m": 0.6, "top5@5m": 0.6735, "top10@5m": 0.7555 } }, { "epoch": 15, "loss": 1.3801768197438211, "train_top1": 0.6116033152329237, "val_cell": { "R@1": 0.4635, "R@2": 0.618, "R@3": 0.6805, "R@5": 0.7635, "R@10": 0.8765, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.406, "top3@5m": 0.593, "top5@5m": 0.6695, "top10@5m": 0.7545 } }, { "epoch": 16, "loss": 1.3660904567696157, "train_top1": 0.6164618462417834, "val_cell": { "R@1": 0.4675, "R@2": 0.6185, "R@3": 0.6815, "R@5": 0.76, "R@10": 0.869, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.412, "top3@5m": 0.593, "top5@5m": 0.6665, "top10@5m": 0.747 } }, { "epoch": 17, "loss": 1.354925928064059, "train_top1": 0.6207487853672478, "val_cell": { "R@1": 0.4675, "R@2": 0.6185, "R@3": 0.6815, "R@5": 0.759, "R@10": 0.86, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.413, "top3@5m": 0.5925, "top5@5m": 0.6675, "top10@5m": 0.7405 } }, { "epoch": 18, "loss": 1.344560150045366, "train_top1": 0.6198913975421549, "val_cell": { "R@1": 0.4655, "R@2": 0.6175, "R@3": 0.6835, "R@5": 0.7615, "R@10": 0.851, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4115, "top3@5m": 0.595, "top5@5m": 0.668, "top10@5m": 0.742 } }, { "epoch": 19, "loss": 1.3348511342288223, "train_top1": 0.6224635610174335, "val_cell": { "R@1": 0.468, "R@2": 0.618, "R@3": 0.6855, "R@5": 0.765, "R@10": 0.85, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.414, "top3@5m": 0.597, "top5@5m": 0.6705, "top10@5m": 0.743 } }, { "epoch": 20, "loss": 1.3263166988533066, "train_top1": 0.6273220920262932, "val_cell": { "R@1": 0.47, "R@2": 0.6165, "R@3": 0.6855, "R@5": 0.7675, "R@10": 0.853, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.416, "top3@5m": 0.5965, "top5@5m": 0.673, "top10@5m": 0.7455 } }, { "epoch": 21, "loss": 1.317855204163568, "train_top1": 0.6281794798513861, "val_cell": { "R@1": 0.469, "R@2": 0.623, "R@3": 0.689, "R@5": 0.769, "R@10": 0.8565, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4145, "top3@5m": 0.5995, "top5@5m": 0.675, "top10@5m": 0.747 } }, { "epoch": 22, "loss": 1.3099652240534583, "train_top1": 0.6330380108602458, "val_cell": { "R@1": 0.473, "R@2": 0.6265, "R@3": 0.688, "R@5": 0.769, "R@10": 0.856, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.417, "top3@5m": 0.5975, "top5@5m": 0.6745, "top10@5m": 0.7475 } }, { "epoch": 23, "loss": 1.3029229015103405, "train_top1": 0.6344669905687339, "val_cell": { "R@1": 0.4685, "R@2": 0.6245, "R@3": 0.692, "R@5": 0.772, "R@10": 0.8645, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4145, "top3@5m": 0.6015, "top5@5m": 0.6755, "top10@5m": 0.7545 } }, { "epoch": 24, "loss": 1.296140853184637, "train_top1": 0.6373249499857102, "val_cell": { "R@1": 0.4725, "R@2": 0.628, "R@3": 0.6935, "R@5": 0.7695, "R@10": 0.863, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4185, "top3@5m": 0.6035, "top5@5m": 0.6735, "top10@5m": 0.7535 } }, { "epoch": 25, "loss": 1.288698569268082, "train_top1": 0.6376107459274079, "val_cell": { "R@1": 0.4735, "R@2": 0.629, "R@3": 0.6955, "R@5": 0.7715, "R@10": 0.8665, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4175, "top3@5m": 0.6055, "top5@5m": 0.6745, "top10@5m": 0.757 } }, { "epoch": 26, "loss": 1.2834566818642323, "train_top1": 0.6378965418691055, "val_cell": { "R@1": 0.472, "R@2": 0.6265, "R@3": 0.696, "R@5": 0.773, "R@10": 0.8665, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.417, "top3@5m": 0.606, "top5@5m": 0.6755, "top10@5m": 0.757 } }, { "epoch": 27, "loss": 1.277862523684811, "train_top1": 0.6407545012860817, "val_cell": { "R@1": 0.475, "R@2": 0.628, "R@3": 0.698, "R@5": 0.7725, "R@10": 0.8605, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4195, "top3@5m": 0.608, "top5@5m": 0.673, "top10@5m": 0.7515 } }, { "epoch": 28, "loss": 1.2728685966795872, "train_top1": 0.6421834809945699, "val_cell": { "R@1": 0.475, "R@2": 0.6275, "R@3": 0.7005, "R@5": 0.7765, "R@10": 0.8685, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4205, "top3@5m": 0.6105, "top5@5m": 0.676, "top10@5m": 0.7585 } }, { "epoch": 29, "loss": 1.2686733768885052, "train_top1": 0.6433266647613604, "val_cell": { "R@1": 0.4755, "R@2": 0.628, "R@3": 0.7, "R@5": 0.7765, "R@10": 0.866, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4205, "top3@5m": 0.61, "top5@5m": 0.675, "top10@5m": 0.7565 } }, { "epoch": 30, "loss": 1.2646111042440262, "train_top1": 0.6476136038868248, "val_cell": { "R@1": 0.4795, "R@2": 0.6285, "R@3": 0.7005, "R@5": 0.7765, "R@10": 0.863, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4235, "top3@5m": 0.611, "top5@5m": 0.676, "top10@5m": 0.753 } } ], "n_train_total": 4000, "n_train_valid": 3499, "train_npz": "", "val_npz": "", "train_sha256": "78b73bcb1cf8d573a6df506efcbd377cf91f5c1afcb05369c1f75b5a2d1d02e5", "val_sha256": "f1fc30ea813798c12590376af9b843a120d90cc8b75acb660948a2d5c6726c7b", "state_dict_shapes": { "net.0.weight": [ 64, 23 ], "net.0.bias": [ 64 ], "net.2.weight": [ 32, 64 ], "net.2.bias": [ 32 ], "net.4.weight": [ 1, 32 ], "net.4.bias": [ 1 ] }, "file": "heads/head_seed0_ht23.pt", "sha256": "7f727592e2b7d4fbaef05463525129eda17f385838a5e34c13284c77fad30c16" }, { "method": "Text2Loc++ + SCAR(H-T)", "dataset_version": "PandaSetPose-50-25-25", "backbone_seed": 0, "head_seed": 1, "source_path": "", "bytes": 27780, "mean": [ 0.4608134329319, 0.5, 0.5343223214149475, 0.5, 0.5, 0.15052300691604614, 0.1505965143442154, 0.22912763059139252, 0.4999999701976776, -0.028604188933968544, -0.009919974021613598, 0.055696964263916016, 0.22624221444129944, 0.18046113848686218, 0.4999999701976776, 0.5000002980232239, 0.49999991059303284, 0.3710075318813324, 0.5338996648788452, 1.4515472650527954, 1.4602689743041992, 1.4773361682891846, 1.4770563840866089 ], "std": [ 0.116506427526474, 0.3004809617996216, 0.27012988924980164, 0.3004809617996216, 0.3004809617996216, 0.35746872425079346, 0.13275882601737976, 0.3479200005531311, 0.3004809617996216, 0.04183800518512726, 0.03888953477144241, 0.03282184898853302, 0.3440459370613098, 0.27901825308799744, 0.3004809617996216, 0.29166659712791443, 0.3004809617996216, 0.16501377522945404, 0.2703131437301636, 1.0025379657745361, 1.0008338689804077, 0.9802831411361694, 0.9835349321365356 ], "pinned": [ false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false ], "feature_names": [ "cos", "cos_rank", "sp", "p0_rank", "p2_rank", "is_tied", "tied_size", "sc_center", "sc_center_rank", "off_dx", "off_dy", "off_norm", "sc_fine", "sat_fine", "sc_fine_rank", "borda", "c_rank", "n_obj", "n_matched", "dist_nearest_matched", "dist_centroid_matched", "pred_dist_anchor", "center_dist_anchor" ], "config": { "epochs": 30, "lr": 0.001, "wd": 0.0001, "batch": 512, "hidden": [ 64, 32 ], "loss": "listwise_softmax_ce", "seed": 1, "k": 25, "dataset": "pandaset", "backbone": "text2locpp", "norms": { "pool_norm": 24, "rank_norm": 24, "tied_norm": 25 } }, "selection": { "rule": "max VAL cell R@1, tie max top1@5m, earlier epoch", "epoch": 29, "key": [ 0.4685, 0.411 ], "val_cell": { "R@1": 0.4685, "R@2": 0.6215, "R@3": 0.6955, "R@5": 0.77, "R@10": 0.86, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.411, "top3@5m": 0.609, "top5@5m": 0.6725, "top10@5m": 0.748 } }, "history": [ { "epoch": 1, "loss": 3.0708708398578577, "train_top1": 0.40068591026007433, "val_cell": { "R@1": 0.339, "R@2": 0.486, "R@3": 0.5665, "R@5": 0.719, "R@10": 0.904, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3065, "top3@5m": 0.5055, "top5@5m": 0.627, "top10@5m": 0.776 } }, { "epoch": 2, "loss": 2.8265085330722197, "train_top1": 0.41783366676193195, "val_cell": { "R@1": 0.3425, "R@2": 0.492, "R@3": 0.587, "R@5": 0.742, "R@10": 0.9075, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.311, "top3@5m": 0.515, "top5@5m": 0.643, "top10@5m": 0.779 } }, { "epoch": 3, "loss": 2.587840037878734, "train_top1": 0.4284081166047442, "val_cell": { "R@1": 0.3475, "R@2": 0.497, "R@3": 0.599, "R@5": 0.7525, "R@10": 0.911, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.316, "top3@5m": 0.5255, "top5@5m": 0.651, "top10@5m": 0.783 } }, { "epoch": 4, "loss": 2.341551933127766, "train_top1": 0.44641326093169476, "val_cell": { "R@1": 0.3565, "R@2": 0.516, "R@3": 0.6135, "R@5": 0.7605, "R@10": 0.916, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.326, "top3@5m": 0.5385, "top5@5m": 0.656, "top10@5m": 0.788 } }, { "epoch": 5, "loss": 2.1060318695745934, "train_top1": 0.4681337525007145, "val_cell": { "R@1": 0.3665, "R@2": 0.521, "R@3": 0.622, "R@5": 0.773, "R@10": 0.923, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.334, "top3@5m": 0.5445, "top5@5m": 0.6635, "top10@5m": 0.7935 } }, { "epoch": 6, "loss": 1.923926444352099, "train_top1": 0.4941411831951986, "val_cell": { "R@1": 0.3745, "R@2": 0.545, "R@3": 0.643, "R@5": 0.793, "R@10": 0.9225, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3415, "top3@5m": 0.559, "top5@5m": 0.674, "top10@5m": 0.7915 } }, { "epoch": 7, "loss": 1.7814958686316207, "train_top1": 0.523006573306659, "val_cell": { "R@1": 0.4155, "R@2": 0.573, "R@3": 0.661, "R@5": 0.793, "R@10": 0.9245, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3645, "top3@5m": 0.579, "top5@5m": 0.6745, "top10@5m": 0.7915 } }, { "epoch": 8, "loss": 1.6629273437983514, "train_top1": 0.552157759359817, "val_cell": { "R@1": 0.4235, "R@2": 0.5795, "R@3": 0.6625, "R@5": 0.799, "R@10": 0.9055, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3705, "top3@5m": 0.58, "top5@5m": 0.6805, "top10@5m": 0.773 } }, { "epoch": 9, "loss": 1.580421434977832, "train_top1": 0.5767362103458131, "val_cell": { "R@1": 0.44, "R@2": 0.5915, "R@3": 0.6805, "R@5": 0.8075, "R@10": 0.8945, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3835, "top3@5m": 0.5825, "top5@5m": 0.6865, "top10@5m": 0.7635 } }, { "epoch": 10, "loss": 1.5229166570340746, "train_top1": 0.589882823663904, "val_cell": { "R@1": 0.4495, "R@2": 0.589, "R@3": 0.688, "R@5": 0.788, "R@10": 0.887, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.392, "top3@5m": 0.5905, "top5@5m": 0.6775, "top10@5m": 0.7565 } }, { "epoch": 11, "loss": 1.4796220758500254, "train_top1": 0.5964561303229494, "val_cell": { "R@1": 0.449, "R@2": 0.5855, "R@3": 0.666, "R@5": 0.777, "R@10": 0.888, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3925, "top3@5m": 0.584, "top5@5m": 0.671, "top10@5m": 0.756 } }, { "epoch": 12, "loss": 1.4443719040908007, "train_top1": 0.6036010288653901, "val_cell": { "R@1": 0.447, "R@2": 0.594, "R@3": 0.6735, "R@5": 0.7795, "R@10": 0.88, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.391, "top3@5m": 0.5895, "top5@5m": 0.6695, "top10@5m": 0.755 } }, { "epoch": 13, "loss": 1.4197744185940202, "train_top1": 0.6116033152329237, "val_cell": { "R@1": 0.4465, "R@2": 0.606, "R@3": 0.678, "R@5": 0.7625, "R@10": 0.8765, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.391, "top3@5m": 0.594, "top5@5m": 0.6655, "top10@5m": 0.7545 } }, { "epoch": 14, "loss": 1.4003626746291875, "train_top1": 0.6144612746499, "val_cell": { "R@1": 0.451, "R@2": 0.6085, "R@3": 0.677, "R@5": 0.76, "R@10": 0.8645, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.395, "top3@5m": 0.5935, "top5@5m": 0.6625, "top10@5m": 0.743 } }, { "epoch": 15, "loss": 1.387775941963501, "train_top1": 0.616747642183481, "val_cell": { "R@1": 0.4525, "R@2": 0.6075, "R@3": 0.6805, "R@5": 0.755, "R@10": 0.86, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3965, "top3@5m": 0.597, "top5@5m": 0.659, "top10@5m": 0.7395 } }, { "epoch": 16, "loss": 1.3763365840598425, "train_top1": 0.6173192340668763, "val_cell": { "R@1": 0.448, "R@2": 0.611, "R@3": 0.6805, "R@5": 0.753, "R@10": 0.8555, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.393, "top3@5m": 0.596, "top5@5m": 0.658, "top10@5m": 0.736 } }, { "epoch": 17, "loss": 1.3663009792506677, "train_top1": 0.6181766218919691, "val_cell": { "R@1": 0.451, "R@2": 0.614, "R@3": 0.685, "R@5": 0.756, "R@10": 0.8585, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.397, "top3@5m": 0.6005, "top5@5m": 0.6625, "top10@5m": 0.739 } }, { "epoch": 18, "loss": 1.3563347701584827, "train_top1": 0.6207487853672478, "val_cell": { "R@1": 0.454, "R@2": 0.6165, "R@3": 0.683, "R@5": 0.756, "R@10": 0.8455, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4, "top3@5m": 0.598, "top5@5m": 0.662, "top10@5m": 0.7315 } }, { "epoch": 19, "loss": 1.3481616810343204, "train_top1": 0.6230351529008288, "val_cell": { "R@1": 0.4515, "R@2": 0.6175, "R@3": 0.684, "R@5": 0.7565, "R@10": 0.84, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.397, "top3@5m": 0.598, "top5@5m": 0.6635, "top10@5m": 0.7305 } }, { "epoch": 20, "loss": 1.339202657124764, "train_top1": 0.624464132609317, "val_cell": { "R@1": 0.4555, "R@2": 0.62, "R@3": 0.687, "R@5": 0.7575, "R@10": 0.84, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.401, "top3@5m": 0.601, "top5@5m": 0.6635, "top10@5m": 0.731 } }, { "epoch": 21, "loss": 1.3305975927833695, "train_top1": 0.6276078879679908, "val_cell": { "R@1": 0.454, "R@2": 0.6165, "R@3": 0.689, "R@5": 0.7585, "R@10": 0.839, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3995, "top3@5m": 0.6025, "top5@5m": 0.6645, "top10@5m": 0.7305 } }, { "epoch": 22, "loss": 1.321947655973519, "train_top1": 0.6296084595598742, "val_cell": { "R@1": 0.4545, "R@2": 0.6195, "R@3": 0.6905, "R@5": 0.762, "R@10": 0.8425, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4, "top3@5m": 0.6035, "top5@5m": 0.6675, "top10@5m": 0.732 } }, { "epoch": 23, "loss": 1.3141709268689326, "train_top1": 0.6296084595598742, "val_cell": { "R@1": 0.454, "R@2": 0.6175, "R@3": 0.6905, "R@5": 0.763, "R@10": 0.839, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.399, "top3@5m": 0.603, "top5@5m": 0.6685, "top10@5m": 0.7315 } }, { "epoch": 24, "loss": 1.3062227800867496, "train_top1": 0.6304658473849671, "val_cell": { "R@1": 0.4555, "R@2": 0.618, "R@3": 0.69, "R@5": 0.7625, "R@10": 0.85, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4005, "top3@5m": 0.603, "top5@5m": 0.667, "top10@5m": 0.7345 } }, { "epoch": 25, "loss": 1.2981911414553349, "train_top1": 0.6330380108602458, "val_cell": { "R@1": 0.454, "R@2": 0.6205, "R@3": 0.6915, "R@5": 0.764, "R@10": 0.847, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4, "top3@5m": 0.604, "top5@5m": 0.669, "top10@5m": 0.7385 } }, { "epoch": 26, "loss": 1.2920679686988685, "train_top1": 0.6353243783938268, "val_cell": { "R@1": 0.458, "R@2": 0.6195, "R@3": 0.691, "R@5": 0.766, "R@10": 0.8535, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4035, "top3@5m": 0.604, "top5@5m": 0.671, "top10@5m": 0.745 } }, { "epoch": 27, "loss": 1.2836464354909465, "train_top1": 0.6376107459274079, "val_cell": { "R@1": 0.459, "R@2": 0.6205, "R@3": 0.6915, "R@5": 0.7655, "R@10": 0.856, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4035, "top3@5m": 0.6045, "top5@5m": 0.67, "top10@5m": 0.7475 } }, { "epoch": 28, "loss": 1.27686165867277, "train_top1": 0.6378965418691055, "val_cell": { "R@1": 0.466, "R@2": 0.622, "R@3": 0.6935, "R@5": 0.7675, "R@10": 0.8585, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.41, "top3@5m": 0.607, "top5@5m": 0.671, "top10@5m": 0.7475 } }, { "epoch": 29, "loss": 1.270808222601161, "train_top1": 0.6404687053443842, "val_cell": { "R@1": 0.4685, "R@2": 0.6215, "R@3": 0.6955, "R@5": 0.77, "R@10": 0.86, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.411, "top3@5m": 0.609, "top5@5m": 0.6725, "top10@5m": 0.748 } }, { "epoch": 30, "loss": 1.265663771126467, "train_top1": 0.6410402972277793, "val_cell": { "R@1": 0.4675, "R@2": 0.618, "R@3": 0.7, "R@5": 0.774, "R@10": 0.864, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4105, "top3@5m": 0.6115, "top5@5m": 0.676, "top10@5m": 0.752 } } ], "n_train_total": 4000, "n_train_valid": 3499, "train_npz": "", "val_npz": "", "train_sha256": "78b73bcb1cf8d573a6df506efcbd377cf91f5c1afcb05369c1f75b5a2d1d02e5", "val_sha256": "f1fc30ea813798c12590376af9b843a120d90cc8b75acb660948a2d5c6726c7b", "state_dict_shapes": { "net.0.weight": [ 64, 23 ], "net.0.bias": [ 64 ], "net.2.weight": [ 32, 64 ], "net.2.bias": [ 32 ], "net.4.weight": [ 1, 32 ], "net.4.bias": [ 1 ] }, "file": "heads/head_seed1_ht23.pt", "sha256": "3b6f036a1e6b72334c8e9ec8f23d1b7ac324febd21d0691ae61d7e31a578ddcf" }, { "method": "Text2Loc++ + SCAR(H-T)", "dataset_version": "PandaSetPose-50-25-25", "backbone_seed": 0, "head_seed": 2, "source_path": "", "bytes": 27780, "mean": [ 0.4608134329319, 0.5, 0.5343223214149475, 0.5, 0.5, 0.15052300691604614, 0.1505965143442154, 0.22912763059139252, 0.4999999701976776, -0.028604188933968544, -0.009919974021613598, 0.055696964263916016, 0.22624221444129944, 0.18046113848686218, 0.4999999701976776, 0.5000002980232239, 0.49999991059303284, 0.3710075318813324, 0.5338996648788452, 1.4515472650527954, 1.4602689743041992, 1.4773361682891846, 1.4770563840866089 ], "std": [ 0.116506427526474, 0.3004809617996216, 0.27012988924980164, 0.3004809617996216, 0.3004809617996216, 0.35746872425079346, 0.13275882601737976, 0.3479200005531311, 0.3004809617996216, 0.04183800518512726, 0.03888953477144241, 0.03282184898853302, 0.3440459370613098, 0.27901825308799744, 0.3004809617996216, 0.29166659712791443, 0.3004809617996216, 0.16501377522945404, 0.2703131437301636, 1.0025379657745361, 1.0008338689804077, 0.9802831411361694, 0.9835349321365356 ], "pinned": [ false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false, false ], "feature_names": [ "cos", "cos_rank", "sp", "p0_rank", "p2_rank", "is_tied", "tied_size", "sc_center", "sc_center_rank", "off_dx", "off_dy", "off_norm", "sc_fine", "sat_fine", "sc_fine_rank", "borda", "c_rank", "n_obj", "n_matched", "dist_nearest_matched", "dist_centroid_matched", "pred_dist_anchor", "center_dist_anchor" ], "config": { "epochs": 30, "lr": 0.001, "wd": 0.0001, "batch": 512, "hidden": [ 64, 32 ], "loss": "listwise_softmax_ce", "seed": 2, "k": 25, "dataset": "pandaset", "backbone": "text2locpp", "norms": { "pool_norm": 24, "rank_norm": 24, "tied_norm": 25 } }, "selection": { "rule": "max VAL cell R@1, tie max top1@5m, earlier epoch", "epoch": 25, "key": [ 0.474, 0.4165 ], "val_cell": { "R@1": 0.474, "R@2": 0.6265, "R@3": 0.6995, "R@5": 0.778, "R@10": 0.869, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4165, "top3@5m": 0.6095, "top5@5m": 0.679, "top10@5m": 0.76 } }, "history": [ { "epoch": 1, "loss": 3.1237227094279048, "train_top1": 0.4764218348099457, "val_cell": { "R@1": 0.3605, "R@2": 0.5245, "R@3": 0.625, "R@5": 0.724, "R@10": 0.855, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.329, "top3@5m": 0.553, "top5@5m": 0.6405, "top10@5m": 0.7495 } }, { "epoch": 2, "loss": 2.9047028108745616, "train_top1": 0.49814232637896544, "val_cell": { "R@1": 0.393, "R@2": 0.549, "R@3": 0.6415, "R@5": 0.7425, "R@10": 0.905, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3565, "top3@5m": 0.566, "top5@5m": 0.6545, "top10@5m": 0.7785 } }, { "epoch": 3, "loss": 2.6387013262017995, "train_top1": 0.5027150614461274, "val_cell": { "R@1": 0.4085, "R@2": 0.5545, "R@3": 0.649, "R@5": 0.755, "R@10": 0.91, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3645, "top3@5m": 0.5715, "top5@5m": 0.6575, "top10@5m": 0.7815 } }, { "epoch": 4, "loss": 2.330591563259952, "train_top1": 0.5035724492712204, "val_cell": { "R@1": 0.402, "R@2": 0.5625, "R@3": 0.6525, "R@5": 0.773, "R@10": 0.917, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3615, "top3@5m": 0.5705, "top5@5m": 0.668, "top10@5m": 0.7865 } }, { "epoch": 5, "loss": 2.026866092856321, "train_top1": 0.5110031437553587, "val_cell": { "R@1": 0.41, "R@2": 0.5675, "R@3": 0.6635, "R@5": 0.791, "R@10": 0.92, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3655, "top3@5m": 0.58, "top5@5m": 0.6775, "top10@5m": 0.789 } }, { "epoch": 6, "loss": 1.8082004524565112, "train_top1": 0.5310088596741926, "val_cell": { "R@1": 0.418, "R@2": 0.5875, "R@3": 0.6645, "R@5": 0.7965, "R@10": 0.9215, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.367, "top3@5m": 0.5815, "top5@5m": 0.68, "top10@5m": 0.791 } }, { "epoch": 7, "loss": 1.6725132783912666, "train_top1": 0.5621606173192341, "val_cell": { "R@1": 0.427, "R@2": 0.5915, "R@3": 0.673, "R@5": 0.814, "R@10": 0.9285, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.375, "top3@5m": 0.5875, "top5@5m": 0.6915, "top10@5m": 0.7955 } }, { "epoch": 8, "loss": 1.5723317702792992, "train_top1": 0.5878822520720206, "val_cell": { "R@1": 0.4395, "R@2": 0.5935, "R@3": 0.688, "R@5": 0.811, "R@10": 0.92, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.384, "top3@5m": 0.5895, "top5@5m": 0.6905, "top10@5m": 0.787 } }, { "epoch": 9, "loss": 1.5141780878619215, "train_top1": 0.5884538439554158, "val_cell": { "R@1": 0.4505, "R@2": 0.5965, "R@3": 0.6805, "R@5": 0.805, "R@10": 0.91, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.393, "top3@5m": 0.583, "top5@5m": 0.6905, "top10@5m": 0.7755 } }, { "epoch": 10, "loss": 1.4721462495601596, "train_top1": 0.5987424978565304, "val_cell": { "R@1": 0.4475, "R@2": 0.601, "R@3": 0.674, "R@5": 0.787, "R@10": 0.9065, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3915, "top3@5m": 0.587, "top5@5m": 0.679, "top10@5m": 0.773 } }, { "epoch": 11, "loss": 1.43728081452298, "train_top1": 0.5961703343812518, "val_cell": { "R@1": 0.4495, "R@2": 0.6085, "R@3": 0.68, "R@5": 0.7795, "R@10": 0.8875, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.395, "top3@5m": 0.5925, "top5@5m": 0.6785, "top10@5m": 0.761 } }, { "epoch": 12, "loss": 1.4098762401549194, "train_top1": 0.6053158045155759, "val_cell": { "R@1": 0.4495, "R@2": 0.613, "R@3": 0.682, "R@5": 0.7765, "R@10": 0.8875, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.3945, "top3@5m": 0.595, "top5@5m": 0.6795, "top10@5m": 0.7615 } }, { "epoch": 13, "loss": 1.3897593776714328, "train_top1": 0.6116033152329237, "val_cell": { "R@1": 0.4505, "R@2": 0.613, "R@3": 0.6825, "R@5": 0.775, "R@10": 0.8725, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.395, "top3@5m": 0.5955, "top5@5m": 0.6775, "top10@5m": 0.7495 } }, { "epoch": 14, "loss": 1.3749668383537001, "train_top1": 0.6161760503000857, "val_cell": { "R@1": 0.4555, "R@2": 0.6085, "R@3": 0.681, "R@5": 0.7755, "R@10": 0.865, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4025, "top3@5m": 0.5935, "top5@5m": 0.6765, "top10@5m": 0.745 } }, { "epoch": 15, "loss": 1.3620876526280654, "train_top1": 0.6173192340668763, "val_cell": { "R@1": 0.4565, "R@2": 0.611, "R@3": 0.684, "R@5": 0.773, "R@10": 0.8475, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4025, "top3@5m": 0.595, "top5@5m": 0.676, "top10@5m": 0.737 } }, { "epoch": 16, "loss": 1.3511149162358984, "train_top1": 0.616747642183481, "val_cell": { "R@1": 0.4595, "R@2": 0.6105, "R@3": 0.6855, "R@5": 0.7665, "R@10": 0.8425, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.406, "top3@5m": 0.5965, "top5@5m": 0.671, "top10@5m": 0.7335 } }, { "epoch": 17, "loss": 1.34176909303488, "train_top1": 0.6181766218919691, "val_cell": { "R@1": 0.46, "R@2": 0.6145, "R@3": 0.6855, "R@5": 0.767, "R@10": 0.844, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4055, "top3@5m": 0.5965, "top5@5m": 0.6715, "top10@5m": 0.735 } }, { "epoch": 18, "loss": 1.3320877331261909, "train_top1": 0.6218919691340383, "val_cell": { "R@1": 0.463, "R@2": 0.615, "R@3": 0.688, "R@5": 0.7655, "R@10": 0.844, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4085, "top3@5m": 0.5985, "top5@5m": 0.6695, "top10@5m": 0.7345 } }, { "epoch": 19, "loss": 1.3232767944780204, "train_top1": 0.6227493569591311, "val_cell": { "R@1": 0.463, "R@2": 0.6225, "R@3": 0.688, "R@5": 0.769, "R@10": 0.852, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4075, "top3@5m": 0.599, "top5@5m": 0.6735, "top10@5m": 0.742 } }, { "epoch": 20, "loss": 1.313859792667377, "train_top1": 0.629036867676479, "val_cell": { "R@1": 0.468, "R@2": 0.6205, "R@3": 0.693, "R@5": 0.7685, "R@10": 0.852, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.412, "top3@5m": 0.6035, "top5@5m": 0.6715, "top10@5m": 0.742 } }, { "epoch": 21, "loss": 1.305471752670432, "train_top1": 0.6296084595598742, "val_cell": { "R@1": 0.466, "R@2": 0.624, "R@3": 0.6945, "R@5": 0.77, "R@10": 0.8585, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4095, "top3@5m": 0.605, "top5@5m": 0.673, "top10@5m": 0.748 } }, { "epoch": 22, "loss": 1.29751972093961, "train_top1": 0.6333238068019434, "val_cell": { "R@1": 0.4695, "R@2": 0.6265, "R@3": 0.6965, "R@5": 0.7715, "R@10": 0.859, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4125, "top3@5m": 0.607, "top5@5m": 0.6745, "top10@5m": 0.7485 } }, { "epoch": 23, "loss": 1.2902718309539833, "train_top1": 0.6367533581023149, "val_cell": { "R@1": 0.4725, "R@2": 0.6275, "R@3": 0.6985, "R@5": 0.776, "R@10": 0.865, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4155, "top3@5m": 0.6085, "top5@5m": 0.6785, "top10@5m": 0.7545 } }, { "epoch": 24, "loss": 1.283441310202813, "train_top1": 0.6384681337525007, "val_cell": { "R@1": 0.471, "R@2": 0.626, "R@3": 0.699, "R@5": 0.7775, "R@10": 0.868, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4135, "top3@5m": 0.6085, "top5@5m": 0.6785, "top10@5m": 0.759 } }, { "epoch": 25, "loss": 1.2773756506648122, "train_top1": 0.6427550728779652, "val_cell": { "R@1": 0.474, "R@2": 0.6265, "R@3": 0.6995, "R@5": 0.778, "R@10": 0.869, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4165, "top3@5m": 0.6095, "top5@5m": 0.679, "top10@5m": 0.76 } }, { "epoch": 26, "loss": 1.2718939817302668, "train_top1": 0.6441840525864533, "val_cell": { "R@1": 0.4735, "R@2": 0.6275, "R@3": 0.7, "R@5": 0.7785, "R@10": 0.871, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4155, "top3@5m": 0.6095, "top5@5m": 0.68, "top10@5m": 0.7625 } }, { "epoch": 27, "loss": 1.2672604250887456, "train_top1": 0.6441840525864533, "val_cell": { "R@1": 0.472, "R@2": 0.626, "R@3": 0.703, "R@5": 0.7775, "R@10": 0.8695, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4145, "top3@5m": 0.6125, "top5@5m": 0.6795, "top10@5m": 0.76 } }, { "epoch": 28, "loss": 1.2629213777396568, "train_top1": 0.645898828236639, "val_cell": { "R@1": 0.4705, "R@2": 0.6265, "R@3": 0.6985, "R@5": 0.782, "R@10": 0.872, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4135, "top3@5m": 0.6075, "top5@5m": 0.683, "top10@5m": 0.7635 } }, { "epoch": 29, "loss": 1.2585725883784789, "train_top1": 0.6444698485281509, "val_cell": { "R@1": 0.472, "R@2": 0.625, "R@3": 0.6995, "R@5": 0.78, "R@10": 0.8725, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4145, "top3@5m": 0.608, "top5@5m": 0.681, "top10@5m": 0.7635 } }, { "epoch": 30, "loss": 1.2538649950684326, "train_top1": 0.6427550728779652, "val_cell": { "R@1": 0.47, "R@2": 0.6255, "R@3": 0.7045, "R@5": 0.7835, "R@10": 0.873, "R@25": 0.956 }, "val_loc": { "top1@5m": 0.4125, "top3@5m": 0.6125, "top5@5m": 0.684, "top10@5m": 0.7635 } } ], "n_train_total": 4000, "n_train_valid": 3499, "train_npz": "", "val_npz": "", "train_sha256": "78b73bcb1cf8d573a6df506efcbd377cf91f5c1afcb05369c1f75b5a2d1d02e5", "val_sha256": "f1fc30ea813798c12590376af9b843a120d90cc8b75acb660948a2d5c6726c7b", "state_dict_shapes": { "net.0.weight": [ 64, 23 ], "net.0.bias": [ 64 ], "net.2.weight": [ 32, 64 ], "net.2.bias": [ 32 ], "net.4.weight": [ 1, 32 ], "net.4.bias": [ 1 ] }, "file": "heads/head_seed2_ht23.pt", "sha256": "168791d0fe29bd2aaeab5614afefa8a1438283288b114db2018d2dabef63b9cd" } ], "ablation_heads": [] }