fix dist train (#4940)

7a951e87 · littletomatodonkey · GitHub · 787f1aaf · 7a951e87
隐藏空白更改
内联并排

Showing with 0 addition and 4 deletion

dygraph/resnet/train.py dygraph/resnet/train.py +0 -4

未找到文件。
--- a/dygraph/resnet/train.py
+++ b/dygraph/resnet/train.py
@@ -411,10 +411,6 @@ def train_resnet():
                batch_size=batch_size,
                drop_last=True)

-        if args.use_data_parallel:
-            train_reader = fluid.contrib.reader.distributed_batch_reader(
-                train_reader)
-
        if args.use_imagenet_data:
            test_reader = imagenet_reader.val(settings=args)
        else: