diff --git a/configs/EfficientNet/EfficientNetB0.yaml b/configs/EfficientNet/EfficientNetB0.yaml
index 80e4771cc..01932d43a 100644
--- a/configs/EfficientNet/EfficientNetB0.yaml
+++ b/configs/EfficientNet/EfficientNetB0.yaml
@@ -1,8 +1,12 @@
 mode: 'train'
 ARCHITECTURE:
     name: "EfficientNetB0"
-    drop_connect_rate: 0.1
-    padding_type : "SAME"
+    params:
+        is_test: False
+        padding_type : "SAME"
+        override_params:
+            drop_connect_rate: 0.1
+
 pretrained_model: ""
 model_save_dir: "./output/"
 classes_num: 1000
diff --git a/tools/program.py b/tools/program.py
index 78f362b47..f70cb2dc8 100644
--- a/tools/program.py
+++ b/tools/program.py
@@ -103,7 +103,8 @@ def create_model(architecture, image, classes_num, is_train):
     """
     name = architecture["name"]
     params = architecture.get("params", {})
-    params['is_test'] = not is_train
+    if "is_test" in params:
+        params['is_test'] = not is_train
     model = architectures.__dict__[name](**params)
     out = model.net(input=image, class_dim=classes_num)
     return out
@@ -418,7 +419,6 @@ def run(dataloader,
 
     Returns:
     """
-    print(fetchs)
     fetch_list = [f[0] for f in fetchs.values()]
     metric_list = [f[1] for f in fetchs.values()]
     for m in metric_list: