mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

dbg

Browse Source
This commit is contained in:
weishengyu 2021-06-04 23:12:07 +08:00
parent 9af5f0cf09
commit b48d2c7a97
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/engine/trainer.py
View File

@ -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)