get attr

PaddlePaddle · Oct 10, 2022 · 290be20 · 290be20
1 parent 8b8c64a
commit 290be20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ppfleetx/core/engine/eager_engine.py b/ppfleetx/core/engine/eager_engine.py
@@ -153,12 +153,12 @@ def configure_optimizers(self):
             'sharding_degree']
         self._sharding_offload = self._dist_configs['sharding'][
             'sharding_offload']
-        self._reduce_overlap = self._dist_configs['sharding']['reduce_overlap']
+        self._reduce_overlap = getattr(self._dist_configs['sharding'], 'reduce_overlap', False)
         if self._sharding_degree > 1 and self._reduce_overlap:
             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 = getattr(self._dist_configs['sharding'], 'broadcast_overlap', False)
         if self._sharding_degree > 1 and self._broadcast_overlap:
             if self._sharding_stage == 3 or self._sharding_offload:
                 self._broadcast_overlap = False