未验证 提交 a6c45e92 编写于 作者: Q Qiyang Min 提交者: GitHub

Merge pull request #1295 from velconia/fix_image_classification_syntax

Port image classification dist train to python3
...@@ -22,6 +22,7 @@ import numpy as np ...@@ -22,6 +22,7 @@ import numpy as np
import paddle import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.core as core import paddle.fluid.core as core
import six
import sys import sys
sys.path.append("..") sys.path.append("..")
import models import models
...@@ -172,7 +173,7 @@ def dist_transpile(trainer_id, args, train_prog, startup_prog): ...@@ -172,7 +173,7 @@ def dist_transpile(trainer_id, args, train_prog, startup_prog):
def test_parallel(exe, test_args, args, test_prog, feeder): def test_parallel(exe, test_args, args, test_prog, feeder):
acc_evaluators = [] acc_evaluators = []
for i in xrange(len(test_args[2])): for i in six.moves.xrange(len(test_args[2])):
acc_evaluators.append(fluid.metrics.Accuracy()) acc_evaluators.append(fluid.metrics.Accuracy())
to_fetch = [v.name for v in test_args[2]] to_fetch = [v.name for v in test_args[2]]
...@@ -291,7 +292,7 @@ def train_parallel(train_args, test_args, args, train_prog, test_prog, ...@@ -291,7 +292,7 @@ def train_parallel(train_args, test_args, args, train_prog, test_prog,
def print_arguments(args): def print_arguments(args):
print('----------- Configuration Arguments -----------') print('----------- Configuration Arguments -----------')
for arg, value in sorted(vars(args).iteritems()): for arg, value in sorted(six.iteritems(vars(args))):
print('%s: %s' % (arg, value)) print('%s: %s' % (arg, value))
print('------------------------------------------------') print('------------------------------------------------')
...@@ -307,7 +308,7 @@ def print_paddle_envs(): ...@@ -307,7 +308,7 @@ def print_paddle_envs():
print('----------- Configuration envs -----------') print('----------- Configuration envs -----------')
for k in os.environ: for k in os.environ:
if "PADDLE_" in k: if "PADDLE_" in k:
print "ENV %s:%s" % (k, os.environ[k]) print("ENV %s:%s" % (k, os.environ[k]))
print('------------------------------------------------') print('------------------------------------------------')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册