update

PaddlePaddle · Sep 30, 2022 · e0a0f2f · e0a0f2f
1 parent cc2bb7d
commit e0a0f2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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