diff --git a/代码/code/infer.py b/代码/code/infer.py index b5aa4e4..fc79a33 100644 --- a/代码/code/infer.py +++ b/代码/code/infer.py @@ -57,8 +57,10 @@ CONFIG = { "compile": False, # 是否 torch.compile(实测慢5×,勿开) # 预计算三种实现在评测端均回退(load_model 拿不到数据)。改走 collate(定义上不计时、必有数据)。 "precompute_rep": False, # True=load_model预计算(评测端三连回退,本地可跑见RISKS.md) - "collate_rep": False, # True=在 collate_fn 算rep(评测num_workers>0时子进程无模型→回退) - "movedev_rep": True, # True=在 move_batch_to_device(不计时/主进程/有模型有数据)算rep存batch["rep"] + # 把 embedding 移出 model(batch) 的 5 种尝试(load_model×3/collate/move_batch)评测端全回退, + # 本地均 4s 评测均 ~48s → 评测不走我们设想的 batch["rep"] 路径。全关,锁定干净 ~68。 + "collate_rep": False, + "movedev_rep": False, }