From 52f16cc85d221fd303e1a2af046cb5cbde5cd93e Mon Sep 17 00:00:00 2001 From: parap1uie-s Date: Fri, 17 Mar 2023 17:49:11 +0800 Subject: [PATCH] Update engine.py --- ppcls/engine/engine.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 12d18367..40496ab0 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"]) -- GitLab