提交 06b326b6 编写于 作者: T typhoonzero

follow comments

上级 c24da0d3
......@@ -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'
]
......
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册