From d01c05f27e001efb8139f4c9f011b6b32e7b2fb2 Mon Sep 17 00:00:00 2001
From: frankwhzhang <frankwhzhang@126.com>
Date: Thu, 17 Sep 2020 15:27:38 +0800
Subject: [PATCH] fix mmoe

---
 models/multitask/mmoe/config.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/multitask/mmoe/config.yaml b/models/multitask/mmoe/config.yaml
index 354bd218..18f9a182 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
-- 
GitLab