diff --git a/models/multitask/mmoe/config.yaml b/models/multitask/mmoe/config.yaml
index 354bd218a9e63eeaa7657b2d77c94d9507a3e8bc..18f9a18256fb9d8785789d42be06b4748e721022 100644
--- a/models/multitask/mmoe/config.yaml
+++ b/models/multitask/mmoe/config.yaml
@@ -49,10 +49,12 @@ runner:
   save_checkpoint_path: "increment"
   save_inference_path: "inference"
   print_interval: 1
+  phases: [train]
 - name: infer_runner
   class: infer
   init_model_path: "increment/1"
   device: cpu
+  phases: [infer]
 
 phase:
 - name: train