From 28aa02df393775fb6f3e2826f25a55f7a6779251 Mon Sep 17 00:00:00 2001 From: zhangwenhui03 Date: Thu, 10 Jan 2019 20:01:41 +0800 Subject: [PATCH] fix style --- fluid/PaddleRec/gru4rec/README.md | 1 - fluid/PaddleRec/gru4rec/infer_sample_neg.py | 1 - 2 files changed, 2 deletions(-) diff --git a/fluid/PaddleRec/gru4rec/README.md b/fluid/PaddleRec/gru4rec/README.md index e94aea66..1271612b 100644 --- a/fluid/PaddleRec/gru4rec/README.md +++ b/fluid/PaddleRec/gru4rec/README.md @@ -136,7 +136,6 @@ bayesian pairwise ranking loss(bpr loss) 训练 CUDA_VISIBLE_DEVICES=0 python train_sample_neg.py --loss bpr --use_cuda 1 ``` - 请注意CPU环境下运行单机多卡任务(--parallel 1)时,batch_size应大于cpu核数。 ## 自定义网络结构 diff --git a/fluid/PaddleRec/gru4rec/infer_sample_neg.py b/fluid/PaddleRec/gru4rec/infer_sample_neg.py index 26c7d8ee..0915fe18 100644 --- a/fluid/PaddleRec/gru4rec/infer_sample_neg.py +++ b/fluid/PaddleRec/gru4rec/infer_sample_neg.py @@ -9,7 +9,6 @@ import six import paddle.fluid as fluid import paddle import net - import utils -- GitLab