update export_lite

f8ba640a · Channingss · 0d804498 · f8ba640a
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

deploy/lite/export_lite.py deploy/lite/export_lite.py +2 -1

未找到文件。
--- a/deploy/lite/export_lite.py
+++ b/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__':