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

Fix bug in some Python env. (#1457)

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