diff --git a/ppcls/engine/evaluation/retrieval.py b/ppcls/engine/evaluation/retrieval.py index cd8830a55ef0a1bc896e47b38387128d831285d6..49d9626f639510eb98947eacbe280c2564a3e554 100644 --- a/ppcls/engine/evaluation/retrieval.py +++ b/ppcls/engine/evaluation/retrieval.py @@ -126,10 +126,10 @@ def cal_feature(evaler, name='gallery'): batch_feas = paddle.divide(batch_feas, feas_norm) # do binarize - if self.config["Global"].get("feature_binarize") == "round": + if evaler.config["Global"].get("feature_binarize") == "round": batch_feas = paddle.round(batch_feas).astype("float32") * 2.0 - 1.0 - if self.config["Global"].get("feature_binarize") == "sign": + if evaler.config["Global"].get("feature_binarize") == "sign": batch_feas = paddle.sign(batch_feas).astype("float32") if all_feas is None: