From cfc41ff8e68967408c74a10841435f91e18a9caf Mon Sep 17 00:00:00 2001 From: Yulong Ao Date: Mon, 17 Apr 2023 21:00:53 +0800 Subject: [PATCH] Revert "[Auto Parallel] Add the micro-bathsize config (#52912)" This reverts commit 94afa5abde2f6770ef01ddc21ea9ccec049ee024. --- python/paddle/distributed/auto_parallel/parallelizer_v2.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/python/paddle/distributed/auto_parallel/parallelizer_v2.py b/python/paddle/distributed/auto_parallel/parallelizer_v2.py index 6807016c34f..c4ef623b172 100644 --- a/python/paddle/distributed/auto_parallel/parallelizer_v2.py +++ b/python/paddle/distributed/auto_parallel/parallelizer_v2.py @@ -107,11 +107,6 @@ class Parallelizer: ) # Do reshard process 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) resharder = Resharder( dist_main_prog, @@ -119,7 +114,6 @@ class Parallelizer: rank, self._dist_context, dist_params_grads, - micro_bsz, ) resharder.reshard() self._logger.debug( -- GitLab