mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update engine.py

pull/2751/head
parap1uie-s 2023-03-17 17:49:11 +08:00 committed by Walter
parent 6e6586f59b
commit 52f16cc85d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ppcls/engine/engine.py
View File

@ -486,6 +486,7 @@ class Engine(object):
@paddle.no_grad() @paddle.no_grad()
def infer(self): def infer(self):
assert self.mode == "infer" and self.eval_mode == "classification" assert self.mode == "infer" and self.eval_mode == "classification"
results = []
total_trainer = dist.get_world_size() total_trainer = dist.get_world_size()
local_rank = dist.get_rank() local_rank = dist.get_rank()
image_list = get_image_list(self.config["Infer"]["infer_imgs"]) image_list = get_image_list(self.config["Infer"]["infer_imgs"])