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

Browse Source
This commit is contained in:
RunningLeon 2023-06-08 16:33:35 +08:00
parent 1d10af990d
commit 2273a00ff5
3 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_apis/test_onnx2tensorrt.py
View File

@ -34,7 +34,7 @@ def get_deploy_cfg():
backend_config=dict( backend_config=dict(
type='tensorrt', type='tensorrt',
common_config=dict( common_config=dict(
fp16_mode=False, max_workspace_size=1 << 30), fp16_mode=False, max_workspace_size=1 << 20),
model_inputs=[ model_inputs=[
dict( dict(
input_shapes=dict( input_shapes=dict(

4
tests/test_codebase/test_mmocr/test_mmocr_models.py
View File

@ -403,7 +403,7 @@ def test_forward_of_fpnc(backend: Backend):
dict( dict(
backend_config=dict( backend_config=dict(
type=backend.value, type=backend.value,
common_config=dict(max_workspace_size=1 << 30), common_config=dict(max_workspace_size=1 << 20),
model_inputs=[ model_inputs=[
dict( dict(
input_shapes=dict( input_shapes=dict(
@ -550,7 +550,7 @@ def test_mmdet_wrapper__forward(backend):
dict( dict(
backend_config=dict( backend_config=dict(
type=backend.value, type=backend.value,
common_config=dict(max_workspace_size=1 << 30)), common_config=dict(max_workspace_size=1 << 20)),
onnx_config=dict( onnx_config=dict(
input_shape=None, input_shape=None,
input_names=['inputs'], input_names=['inputs'],

5
tests/test_ops/utils.py
View File

@ -119,7 +119,7 @@ class TestTensorRTExporter:
backend_config=dict( backend_config=dict(
type='tensorrt', type='tensorrt',
common_config=dict( common_config=dict(
fp16_mode=False, max_workspace_size=1 << 28), fp16_mode=False, max_workspace_size=1 << 20),
model_inputs=[ model_inputs=[
dict( dict(
input_shapes=dict( input_shapes=dict(
@ -132,14 +132,13 @@ class TestTensorRTExporter:
]))) ])))
]))) ])))
onnx_model = onnx.load(onnx_file_path)
work_dir, filename = os.path.split(trt_file_path) work_dir, filename = os.path.split(trt_file_path)
trt_apis.onnx2tensorrt( trt_apis.onnx2tensorrt(
work_dir, work_dir,
filename, filename,
0, 0,
deploy_cfg=deploy_cfg, deploy_cfg=deploy_cfg,
onnx_model=onnx_model) onnx_model=onnx_file_path)
if expected_result is None and not isinstance( if expected_result is None and not isinstance(
model, onnx.onnx_ml_pb2.ModelProto): model, onnx.onnx_ml_pb2.ModelProto):
with torch.no_grad(): with torch.no_grad():