Skip to content

Commit 886c715

Browse files
Feat/fix eval bug (#2203)
1 parent 06d2e99 commit 886c715

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

boxmot/engine/tuner.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,9 @@ def yaml_to_search_space(config: dict, tune) -> dict:
4646
elif t == "choice":
4747
space[param] = tune.choice(details["options"])
4848
elif t == "grid_search":
49-
space[param] = tune.grid_search(details["values"])
49+
# Optuna doesn't support grid_search directly in the same way as basic Tune
50+
# Mapping grid_search to choice for Optuna compatibility
51+
space[param] = tune.choice(details["values"])
5052
elif t == "loguniform":
5153
space[param] = tune.loguniform(*details["range"])
5254
return space

boxmot/utils/dataloaders/MOT17.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ def process_sequences_lazily(dataset: MOT17DetEmbDataset) -> None:
217217

218218
if __name__ == "__main__":
219219
dataset = MOT17DetEmbDataset(
220-
mot_root="./tracking/TrackEval/data/MOT17-ablation/train",
220+
mot_root="./tracking/TrackEval/MOT17-ablation/train",
221221
det_emb_root="./runs/dets_n_embs",
222222
model_name="yolox_x_ablation",
223223
reid_name="lmbn_n_duke",

0 commit comments

Comments
 (0)