update paddle2onnx version (#1372)

paddleslim/auto_compression/compressor.py

@@ -846,9 +846,12 @@ class AutoCompression:
         assert os.path.exists(
             infer_model_path), 'Not found {}, please check it.'.format(
                 infer_model_path)
+        onnx_save_path = os.path.join(self.final_dir, 'ONNX')
+        if not os.path.exists(onnx_save_path):
+            os.makedirs(onnx_save_path)
         export_onnx(
             self.final_dir,
             model_filename=self.model_filename,
             params_filename=self.params_filename,
-            save_file_path=os.path.join(self.final_dir, model_name),
+            save_file_path=os.path.join(onnx_save_path, model_name),
             deploy_backend=deploy_backend)

paddleslim/common/load_model.py

@@ -207,10 +207,10 @@ def export_onnx(model_dir,
     if not params_filename:
         params_filename = 'model.pdiparams'
     try:
-        pkg.require('paddle2onnx')
+        os.system(' python -m pip install -U paddle2onnx==1.0.0rc4')
     except:
         from pip._internal import main
-        main(['install', 'paddle2onnx==1.0.0rc3'])
+        main(['install', 'paddle2onnx==1.0.0rc4'])
     import paddle2onnx
     paddle2onnx.command.c_paddle_to_onnx(
         model_file=os.path.join(model_dir, model_filename),
@@ -218,5 +218,9 @@ def export_onnx(model_dir,
         save_file=save_file_path,
         opset_version=opset_version,
         enable_onnx_checker=True,
-        deploy_backend=deploy_backend)
+        deploy_backend=deploy_backend,
+        scale_file=os.path.join(model_dir, 'calibration_table.txt'),
+        calibration_file=os.path.join(
+            save_file_path.rstrip(os.path.split(save_file_path)[-1]),
+            'calibration.cache'))
     _logger.info('Convert model to ONNX: {}'.format(save_file_path))