diff --git a/PaddleRec/ssr/train.py b/PaddleRec/ssr/train.py index 40380531d2307d4ba0179b2a21bdfd33cc904e54..f13b7e1158342037c1caf1375f5f4cffb00dcf0b 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