Kaynağa Gözat

modify slim.py

sunyanfang01 5 yıl önce
ebeveyn
işleme
9083410dd9
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  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)