From 6604e9b168d493652b9e737f4daead5f06a8e9de Mon Sep 17 00:00:00 2001
From: overlordmax <515704170@qq.com>
Date: Tue, 23 Jun 2020 20:58:38 +0800
Subject: [PATCH] fix bug

---
 models/rank/fibinet/config.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/rank/fibinet/config.yaml b/models/rank/fibinet/config.yaml
index 520d709d..44f47b13 100644
--- a/models/rank/fibinet/config.yaml
+++ b/models/rank/fibinet/config.yaml
@@ -47,7 +47,7 @@ hyper_parameters:
   dropout_rate: 0.5
 
 # select runner by name
-mode: [single_gpu_train, single_gpu_infer]
+mode: [single_cpu_train, single_cpu_infer]
 # config of each runner.
 # runner is a kind of paddle training class, which wraps the train/infer process.
 runner:
-- 
GitLab