提交 b48d2c7a 编写于 作者: W weishengyu

dbg

上级 9af5f0cf
......@@ -246,11 +246,11 @@ class Trainer(object):
elif self.eval_mode == "retrieval":
if self.gallery_dataloader is None:
self.gallery_dataloader = build_dataloader(
self.config["DataLoader"], "Gallery", self.device)
self.config["DataLoader"]["Eval"], "Gallery", self.device)
if self.query_dataloader is None:
self.query_dataloader = build_dataloader(
self.config["DataLoader"], "Query", self.device)
self.config["DataLoader"]["Eval"], "Query", self.device)
# build metric info
if self.eval_metric_func is None:
metric_config = self.config.get("Metric", None)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册