diff --git a/tools/infer/infer.py b/tools/infer/infer.py
index ec768fbabf6cefca0d5c82ff7eb80e07b4e2162a..ea2befe5066a25f80b7ddfff7ea3e62c91b19657 100644
--- a/tools/infer/infer.py
+++ b/tools/infer/infer.py
@@ -53,7 +53,10 @@ def create_predictor(args):
             out = fluid.layers.softmax(out)
         return out
 
-    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.CUDAPlace(0) if args.use_gpu else fluid.CPUPlace()
     exe = fluid.Executor(place)