Skip to content

Commit

Permalink
Merge pull request #3751 from LDOUBLEV/fix_pgnet
Browse files Browse the repository at this point in the history
fix predict_e2e.py
  • Loading branch information
LDOUBLEV committed Aug 25, 2021
2 parents 45dff75 + 8587b40 commit 68e5225
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/infer/predict_e2e.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ def __init__(self, args):

self.preprocess_op = create_operators(pre_process_list)
self.postprocess_op = build_post_process(postprocess_params)
self.predictor, self.input_tensor, self.output_tensors = utility.create_predictor(
self.predictor, self.input_tensor, self.output_tensors, _ = utility.create_predictor(
args, 'e2e', logger) # paddle.jit.load(args.det_model_dir)
# self.predictor.eval()

Expand Down

0 comments on commit 68e5225

Please sign in to comment.