提交 f141e3fd 编写于 作者: Y Yibing Liu

Code cleanup in profiling

上级 c2b239e8
...@@ -52,7 +52,7 @@ def parse_args(): ...@@ -52,7 +52,7 @@ def parse_args():
"--num_iters", "--num_iters",
type=int, type=int,
default=10, default=10,
help="The number of iterations profiling over.") help="The maximum number of iterations profiling over.")
parser.add_argument( parser.add_argument(
"--pool_size", "--pool_size",
type=int, type=int,
...@@ -114,8 +114,6 @@ def train_loop(exe, train_progm, init, num_iters, train_data, dev_count, ...@@ -114,8 +114,6 @@ def train_loop(exe, train_progm, init, num_iters, train_data, dev_count,
-1] + label_data_input_fields -1] + label_data_input_fields
util_input_names = encoder_util_input_fields + decoder_util_input_fields util_input_names = encoder_util_input_fields + decoder_util_input_fields
#for pass_id in xrange(TrainTaskConfig.pass_num):
start_time = time.time() start_time = time.time()
exec_time = 0.0 exec_time = 0.0
for batch_id, data in enumerate(train_data()): for batch_id, data in enumerate(train_data()):
...@@ -175,7 +173,7 @@ def profile(args): ...@@ -175,7 +173,7 @@ def profile(args):
if not TrainTaskConfig.use_gpu: if not TrainTaskConfig.use_gpu:
place = fluid.CPUPlace() place = fluid.CPUPlace()
dev_count = int(os.environ.get('CPU_NUM', multiprocessing.cpu_count())) dev_count = multiprocessing.cpu_count()
else: else:
place = fluid.CUDAPlace(0) place = fluid.CUDAPlace(0)
dev_count = fluid.core.get_cuda_device_count() dev_count = fluid.core.get_cuda_device_count()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册