diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py
index 9080feea74..7c3abb3086 100755
--- a/deploy/slim/quantization/export_model.py
+++ b/deploy/slim/quantization/export_model.py
@@ -120,8 +120,7 @@ def main():
     for k, v in metric.items():
         logger.info('{}:{}'.format(k, v))
 
-    infer_shape = [3, 32, 100] if config['Architecture'][
-        'model_type'] != "det" else [3, 640, 640]
+    infer_shape = [3, 32, 100] if model_type != "det" else [3, 640, 640]
 
     save_path = config["Global"]["save_inference_dir"]