提交 85cfc51e 编写于 作者: littletomatodonkey's avatar littletomatodonkey

fix

上级 c425ba9b
...@@ -283,9 +283,6 @@ def create_predictor(args, mode, logger): ...@@ -283,9 +283,6 @@ def create_predictor(args, mode, logger):
else: else:
use_dynamic_shape = False use_dynamic_shape = False
if use_dynamic_shape: if use_dynamic_shape:
print("min_input_shape: ", min_input_shape)
print("max_input_shape: ", max_input_shape)
print("opt_input_shape: ", opt_input_shape)
config.set_trt_dynamic_shape_info( config.set_trt_dynamic_shape_info(
min_input_shape, max_input_shape, opt_input_shape) min_input_shape, max_input_shape, opt_input_shape)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册