fix: dbg

2022-04-26 03:26:37 +00:00 · 2022-04-26 03:26:37 +00:00 · 8765dadfbb
parent d2bfa3a3ce
commit 8765dadfbb
1 changed files with 1 additions and 0 deletions
--- a/ppcls/engine/evaluation/retrieval.py
+++ b/ppcls/engine/evaluation/retrieval.py
@ -90,6 +90,7 @@ def retrieval_eval(engine, epoch_id=0):

 def cal_feature(engine, name='gallery'):
    has_unique_id = False
+    all_unique_id = None

    if name == 'gallery':
        dataloader = engine.gallery_dataloader