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

View File

@ -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"])