Skip to content

Commit f8ba640

Browse files
committed
update export_lite
1 parent 0d80449 commit f8ba640

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

deploy/lite/export_lite.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ def export_lite():
2121
opt = lite.Opt()
2222
model_file = os.path.join(FLAGS.model_dir, '__model__')
2323
params_file = os.path.join(FLAGS.model_dir, '__params__')
24-
opt.run_optimize("", model_file, params_file, FLAGS.place, FLAGS.save_file)
24+
opt.run_optimize("", model_file, params_file, 'naive_buffer', FLAGS.place,
25+
FLAGS.save_file)
2526

2627

2728
if __name__ == '__main__':

0 commit comments

Comments
 (0)