Update engine.py
parent
6e6586f59b
commit
52f16cc85d
|
@ -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"])
|
||||
|
|
Loading…
Reference in New Issue