Browse Source

update export_lite

Channingss 5 years ago
parent
commit
f8ba640a97
1 changed files with 2 additions and 1 deletions
  1. 2 1
      deploy/lite/export_lite.py

+ 2 - 1
deploy/lite/export_lite.py

@@ -21,7 +21,8 @@ def export_lite():
     opt = lite.Opt()
     model_file = os.path.join(FLAGS.model_dir, '__model__')
     params_file = os.path.join(FLAGS.model_dir, '__params__')
-    opt.run_optimize("", model_file, params_file, FLAGS.place, FLAGS.save_file)
+    opt.run_optimize("", model_file, params_file, 'naive_buffer', FLAGS.place,
+                     FLAGS.save_file)
 
 
 if __name__ == '__main__':