From 9d284e3fad4c137d6b69adc319f4a7d612a832da Mon Sep 17 00:00:00 2001 From: frankwhzhang Date: Wed, 24 Oct 2018 11:43:07 +0800 Subject: [PATCH] fix README & train --- fluid/recommendation/gru4rec/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/fluid/recommendation/gru4rec/README.md b/fluid/recommendation/gru4rec/README.md index c8e50daf..cb8fff93 100644 --- a/fluid/recommendation/gru4rec/README.md +++ b/fluid/recommendation/gru4rec/README.md @@ -74,18 +74,18 @@ python convert_format.py ``` ## 训练 ---use_cuda 1 表示使用gpu --parallel 1 表示多卡 0 表示单卡 +--use_cuda 表示使用gpu --parallel 表示多卡 GPU 环境 运行命令 `CUDA_VISIBLE_DEVICES=0 python train.py train_file test_file ` 开始训练模型。 ```python -CUDA_VISIBLE_DEVICES=0 python train.py small_train.txt small_test.txt --use_cuda 1 --parallel 0 +CUDA_VISIBLE_DEVICES=0 python train.py small_train.txt small_test.txt --use_cuda ``` CPU 环境 运行命令 `python train.py train_file test_file` 开始训练模型。 ```python -python train.py small_train.txt small_test.txt --use_cuda 0 --parallel 0 +python train.py small_train.txt small_test.txt ``` 当前支持的参数可参见[train.py](./train.py) `train_net` 函数 -- GitLab