mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

fix inference_model api

Browse Source
This commit is contained in:
Xin Chen 2022-09-06 19:02:45 +08:00
parent a2f82874bb
commit 0aad6359e2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmdeploy/apis/inference.py
View File

@ -48,6 +48,6 @@ def inference_model(model_cfg: Union[str, mmengine.Config],
model_inputs, _ = task_processor.create_input(img, input_shape) model_inputs, _ = task_processor.create_input(img, input_shape)
with torch.no_grad(): with torch.no_grad():
result = model.test_step([model_inputs]) result = model.test_step(model_inputs)
return result return result