浏览代码

modify slim.py

sunyanfang01 5 年之前
父节点
当前提交
9083410dd9
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/slim.py

+ 1 - 1
paddlex/slim.py

@@ -31,4 +31,4 @@ def export_quant_model(model,
         batch_size=batch_size,
         batch_num=batch_num,
         save_dir=save_dir,
-        cache_dir='./temp')
+        cache_dir=cache_dir)