Explorar o código

modify slim.py

sunyanfang01 %!s(int64=5) %!d(string=hai) anos
pai
achega
9083410dd9
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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)