diff --git a/python/paddle/distributed/auto_parallel/parallelizer_v2.py b/python/paddle/distributed/auto_parallel/parallelizer_v2.py index 6807016c34f172485cbf3b1c71d80f48545ac1a3..c4ef623b172605855001fcb00060b8312803977d 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(