From 2140e7757076ea96346a2ccbb6c04767172f3c96 Mon Sep 17 00:00:00 2001
From: xulongteng <xulongteng@baidu.com>
Date: Tue, 29 Oct 2019 17:15:21 +0800
Subject: [PATCH] set enable_memory_optimization for bert

---
 demo-serving/conf/model_toolkit.prototxt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/demo-serving/conf/model_toolkit.prototxt b/demo-serving/conf/model_toolkit.prototxt
index c3f618d1..fae40a9d 100644
--- a/demo-serving/conf/model_toolkit.prototxt
+++ b/demo-serving/conf/model_toolkit.prototxt
@@ -45,4 +45,5 @@ engines {
   runtime_thread_num: 0
   batch_infer_size: 0
   enable_batch_align: 0
+  enable_memory_optimization: true
 }
-- 
GitLab