diff --git a/fluid/recommendation/gru4rec/README.md b/fluid/recommendation/gru4rec/README.md index 1092454c40d9bc6868967ffd09f5088643563578..c8e50dafebbaacefe529704f11b2f06c813df27c 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