fix dist eval (#364)

2020-11-04 11:06:23 +08:00 · 2020-11-04 11:06:23 +08:00 · 82e7a90bfe
parent 081ca857ce
commit 82e7a90bfe
1 changed files with 7 additions and 2 deletions
--- a/tools/eval.py
+++ b/tools/eval.py
@ -59,9 +59,14 @@ def main(args, return_dict={}):

    paddle.disable_static(place)

-    strategy = paddle.distributed.init_parallel_env()
+    use_data_parallel = int(os.getenv("PADDLE_TRAINERS_NUM", 1)) != 1
+    config["use_data_parallel"] = use_data_parallel
+
    net = program.create_model(config.ARCHITECTURE, config.classes_num)
+    if config["use_data_parallel"]:
+        strategy = paddle.distributed.init_parallel_env()
        net = paddle.DataParallel(net, strategy)
+
    init_model(config, net, optimizer=None)
    valid_dataloader = Reader(config, 'valid', places=place)()
    net.eval()