Jelajahi Sumber

modify slim.py

sunyanfang01 5 tahun lalu
induk
melakukan
9083410dd9
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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)