diff --git a/models/rank/fibinet/config.yaml b/models/rank/fibinet/config.yaml index 520d709d16cbd24fe619e568d81e93f289328815..44f47b136bda53e8de026db348f91b250c501c9d 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: