From 06b326b681e2d9a831e02a7f2f0a1e8c18c1bdeb Mon Sep 17 00:00:00 2001 From: typhoonzero Date: Fri, 12 Jan 2018 17:07:11 +0800 Subject: [PATCH] follow comments --- python/paddle/v2/fluid/__init__.py | 3 ++- python/paddle/v2/fluid/distribute_transpiler_simple.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/python/paddle/v2/fluid/__init__.py b/python/paddle/v2/fluid/__init__.py index 5e01b871980..a14422ee92f 100644 --- a/python/paddle/v2/fluid/__init__.py +++ b/python/paddle/v2/fluid/__init__.py @@ -18,13 +18,14 @@ from param_attr import ParamAttr from data_feeder import DataFeeder from core import LoDTensor, CPUPlace, CUDAPlace from distribute_transpiler import DistributeTranspiler +from distribute_transpiler_simple import SimpleDistributeTranspiler import clip Tensor = LoDTensor __all__ = framework.__all__ + executor.__all__ + [ 'io', 'initializer', 'layers', 'nets', 'optimizer', 'backward', 'regularizer', 'LoDTensor', 'CPUPlace', 'CUDAPlace', 'Tensor', 'ParamAttr' - 'DataFeeder', 'clip', 'DistributeTranspiler' + 'DataFeeder', 'clip', 'SimpleDistributeTranspiler', 'DistributeTranspiler' ] diff --git a/python/paddle/v2/fluid/distribute_transpiler_simple.py b/python/paddle/v2/fluid/distribute_transpiler_simple.py index 49ece7b725e..32db3df9aa2 100644 --- a/python/paddle/v2/fluid/distribute_transpiler_simple.py +++ b/python/paddle/v2/fluid/distribute_transpiler_simple.py @@ -48,7 +48,7 @@ def round_robin(params_grads, pserver_endpoints): return param_grad_map -class DistributeTranspiler: +class SimpleDistributeTranspiler: def transpile(self, optimize_ops, params_grads, -- GitLab