未验证 提交 7852d060 编写于 作者: Q qingqing01 提交者: GitHub

Fix bug in some Python env. (#1457)

* Fix bug in some Python env.
上级 2c4a77ed
......@@ -5,6 +5,7 @@ import argparse
import functools
import shutil
import math
import multiprocessing
import paddle
import paddle.fluid as fluid
......@@ -127,7 +128,7 @@ def train(args,
if parallel:
train_exe = fluid.ParallelExecutor(main_program=train_prog,
use_cuda=use_gpu, loss_name=loss.name)
use_cuda=True if use_gpu else False, loss_name=loss.name)
train_reader = reader.train(data_args,
train_file_list,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册