diff --git a/tools/export_model.py b/tools/export_model.py
index d911856d8c32e0d9596de2d73ebf4a1bc7528c84..b3d2fe2faad61582756cae4062bc0e93f2e35da8 100644
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -55,7 +55,10 @@ def create_model(args, model, input, class_dim=1000):
 def main():
     args = parse_args()
 
-    model = architectures.__dict__[args.model]()
+    if "EfficientNet" in args.model:
+        model = architectures.__dict__[args.model](is_test=True)
+    else:
+        model = architectures.__dict__[args.model]()
 
     place = fluid.CPUPlace()
     exe = fluid.Executor(place)