diff --git a/models/rank/wide_deep/config.yaml b/models/rank/wide_deep/config.yaml
index 2b3bf5ecdaefc50f216723dc97e5550f12747c7f..f4ec27f82ff99f703e2dbb2851337c030afe2730 100755
--- a/models/rank/wide_deep/config.yaml
+++ b/models/rank/wide_deep/config.yaml
@@ -40,7 +40,7 @@ hyper_parameters:
     hidden3_units: 25
 
 
-mode: train_runner 
+mode: [train_runner , infer_runner]
 # if infer, change mode to "infer_runner" and change phase to "infer_phase"
 
 runner: