diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 12d183674d16684d528573d709fc5fc5730ada41..40496ab019702fa91706cdd58f9e23161b7420b4 100755 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -486,6 +486,7 @@ class Engine(object): @paddle.no_grad() def infer(self): assert self.mode == "infer" and self.eval_mode == "classification" + results = [] total_trainer = dist.get_world_size() local_rank = dist.get_rank() image_list = get_image_list(self.config["Infer"]["infer_imgs"])