From aa69732dd7b368a4677f0295d84d0e9f876d6450 Mon Sep 17 00:00:00 2001 From: zhang wenhui Date: Wed, 4 Mar 2020 17:52:44 +0800 Subject: [PATCH] fix ssr (#4370) --- PaddleRec/ssr/train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/PaddleRec/ssr/train.py b/PaddleRec/ssr/train.py index 40380531..f13b7e11 100644 --- a/PaddleRec/ssr/train.py +++ b/PaddleRec/ssr/train.py @@ -94,8 +94,9 @@ def train(args): feed_list=train_input_data, capacity=10000, iterable=True) loader.set_sample_list_generator(train_reader, places=place) if parallel: - train_exe = fluid.ParallelExecutor( - use_cuda=use_cuda, loss_name=avg_cost.name) + train_exe = exe + #train_exe = fluid.ParallelExecutor( + # use_cuda=use_cuda, loss_name=avg_cost.name) else: train_exe = exe -- GitLab