From d33274ea613baf1fdfc4876229bdb1edb6c79147 Mon Sep 17 00:00:00 2001
From: yinhaofeng <1841837261@qq.com>
Date: Tue, 22 Sep 2020 14:05:54 +0000
Subject: [PATCH] fix

---
 models/rank/fm/config.yaml | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/models/rank/fm/config.yaml b/models/rank/fm/config.yaml
index cfaef626..8b72ab5d 100644
--- a/models/rank/fm/config.yaml
+++ b/models/rank/fm/config.yaml
@@ -48,7 +48,7 @@ mode: [train_runner,infer_runner]
 runner:
   - name: train_runner
     class: train
-    epochs: 2
+    epochs: 1
     device: cpu
     init_model_path: ""
     save_checkpoint_interval: 1
@@ -56,11 +56,14 @@ runner:
     save_checkpoint_path: "increment"
     save_inference_path: "inference"
     print_interval: 1
+    phases: phase1
+
   - name: infer_runner
     class: infer
     device: cpu
-    init_model_path: "increment"
+    init_model_path: "increment/0"
     print_interval: 1
+    phases: infer_phase
 
 phase:
 - name: phase1
-- 
GitLab