提交 cfc41ff8 编写于 作者: Y Yulong Ao 提交者: GitHub

Revert "[Auto Parallel] Add the micro-bathsize config (#52912)"

This reverts commit 94afa5ab.
上级 94afa5ab
...@@ -107,11 +107,6 @@ class Parallelizer: ...@@ -107,11 +107,6 @@ class Parallelizer:
) )
# Do reshard process # Do reshard process
time0 = time.time() time0 = time.time()
micro_bsz = (
1
if not self._strategy.pipeline.enable
else self._strategy.pipeline.micro_batch_size
)
set_grad_var_shape(dist_main_prog, self._dist_context) set_grad_var_shape(dist_main_prog, self._dist_context)
resharder = Resharder( resharder = Resharder(
dist_main_prog, dist_main_prog,
...@@ -119,7 +114,6 @@ class Parallelizer: ...@@ -119,7 +114,6 @@ class Parallelizer:
rank, rank,
self._dist_context, self._dist_context,
dist_params_grads, dist_params_grads,
micro_bsz,
) )
resharder.reshard() resharder.reshard()
self._logger.debug( self._logger.debug(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册