From ad9ff0b4e300aff10be8d74accd2ff8a4d2715de Mon Sep 17 00:00:00 2001 From: frankwhzhang Date: Wed, 24 Oct 2018 14:52:19 +0800 Subject: [PATCH] fix Readme & train.py --- fluid/recommendation/gru4rec/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/recommendation/gru4rec/README.md b/fluid/recommendation/gru4rec/README.md index d4489bfb..62214113 100644 --- a/fluid/recommendation/gru4rec/README.md +++ b/fluid/recommendation/gru4rec/README.md @@ -74,7 +74,7 @@ python convert_format.py ``` ## 训练 -'--use_cuda 1' 表示使用gpu --parallel 1 表示多卡 +'--use_cuda 1' 表示使用gpu, 缺省表示使用cpu '--parallel 1' 表示使用多卡,缺省表示使用单卡 GPU 环境 运行命令 `CUDA_VISIBLE_DEVICES=0 python train.py train_file test_file --use_cuda 1` 开始训练模型。 -- GitLab