diff --git a/ppcls/engine/evaluation/retrieval.py b/ppcls/engine/evaluation/retrieval.py index 3dfe6337c..b481efae1 100644 --- 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