diff --git a/ppfleetx/core/engine/eager_engine.py b/ppfleetx/core/engine/eager_engine.py index 514cab30e..d91a4e5ac 100644 --- a/ppfleetx/core/engine/eager_engine.py +++ b/ppfleetx/core/engine/eager_engine.py @@ -158,7 +158,7 @@ def configure_optimizers(self): if self._sharding_stage == 3 or self._sharding_offload: self._reduce_overlap = False logger.warning("reduce overlap only valid for sharding stage 2 without offload") - self._broadcast_overlap = self._dist_configs['sharding']['_broadcast_overlap'] + self._broadcast_overlap = self._dist_configs['sharding']['broadcast_overlap'] if self._sharding_degree > 1 and self._broadcast_overlap: if self._sharding_stage == 3 or self._sharding_offload: self._broadcast_overlap = False