diff --git a/fluid/PaddleRec/gru4rec/README.md b/fluid/PaddleRec/gru4rec/README.md index e94aea661737c8cba696ae96ccdd54fd2f974f1a..1271612b1b34274c455090c8ba2d5d540a9241e8 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 26c7d8eeb44d8da35e5c99bc4018090fa1698db8..0915fe18d571ba459930960d7a39735dc075c930 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