diff --git a/fastdeploy/runtime/backends/tensorrt/option.h b/fastdeploy/runtime/backends/tensorrt/option.h index 2825076b1..48b8f83fb 100755 --- a/fastdeploy/runtime/backends/tensorrt/option.h +++ b/fastdeploy/runtime/backends/tensorrt/option.h @@ -46,8 +46,8 @@ struct TrtBackendOption { */ void SetShape(const std::string& tensor_name, const std::vector& min, - const std::vector& opt, - const std::vector& max) { + const std::vector& opt = std::vector(), + const std::vector& max = std::vector()) { min_shape[tensor_name].clear(); max_shape[tensor_name].clear(); opt_shape[tensor_name].clear();