Scale TrajMixer hidden width with head count

2026-07-22 16:14:52 +08:00
parent 978c88a4ed
commit 68a6a3df88
8 changed files with 29 additions and 43 deletions
--- a/evaluate_auc_v2.py
+++ b/evaluate_auc_v2.py
@@ -205,7 +205,6 @@ def build_model_from_dataset(args: argparse.Namespace, cfg: Dict[str, Any], data
        time_mode=str(cfg_get(args, cfg, "time_mode", "relative")),
        dist_mode=str(cfg_get(args, cfg, "dist_mode", "exponential")),
        dropout=float(cfg_get(args, cfg, "dropout", 0.0)),
-        hidden_group=int(cfg_get(args, cfg, "hidden_group", 20)),
    )