From 3319d9b0d95fd5c4b8255027bb4c9199e4505d5d Mon Sep 17 00:00:00 2001 From: frankwhzhang Date: Wed, 24 Oct 2018 10:54:41 +0800 Subject: [PATCH] fix README, change train.py input format 2.0 --- fluid/recommendation/gru4rec/README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/recommendation/gru4rec/README.md b/fluid/recommendation/gru4rec/README.md index 1092454c..c8e50daf 100644 --- a/fluid/recommendation/gru4rec/README.md +++ b/fluid/recommendation/gru4rec/README.md @@ -103,8 +103,8 @@ python train.py small_train.txt small_test.txt --use_cuda 0 --parallel 0 base_lr=0.01, # base learning rate batch_size=batch_size, pass_num=10, # the number of passed for training - use_cuda=True, # whether to use GPU card - parallel=False, # whether to be parallel + use_cuda=use_cuda, # whether to use GPU card + parallel=parallel, # whether to be parallel model_dir="model_recall20", # directory to save model init_low_bound=-0.1, # uniform parameter initialization lower bound init_high_bound=0.1) # uniform parameter initialization upper bound -- GitLab