Skip to content

Commit 683c635

Browse files
committed
Tensorrt mobilenet and resnet56 support
1 parent 6858363 commit 683c635

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/inference/analysis/analyzer.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424

2525
namespace paddle {
2626

27-
DEFINE_bool(inference_analysis_enable_tensorrt_subgraph_engine, true,
27+
DEFINE_bool(inference_analysis_enable_tensorrt_subgraph_engine, false,
2828
"Enable subgraph to TensorRT engine for acceleration");
2929

3030
DEFINE_string(inference_analysis_graphviz_log_root, "./",

paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ class TensorRTSubgraphPredictor : public NativePaddlePredictor {
3232
: NativePaddlePredictor(config), config_(config) {}
3333

3434
bool Init(const std::shared_ptr<framework::Scope>& parent_scope) {
35-
FLAGS_IA_enable_tensorrt_subgraph_engine = true;
35+
FLAGS_inference_analysis_enable_tensorrt_subgraph_engine = true;
3636
VLOG(3) << "Predictor::init()";
3737
FLAGS_tensorrt_max_batch_size = config_.max_batch_size;
3838
FLAGS_tensorrt_workspace_size = config_.workspace_size;

0 commit comments

Comments
 (0)