Add extra guards

ashleve · Oct 17, 2023 · 473cfbf · 473cfbf
1 parent bffa428
commit 473cfbf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/eval.py b/src/eval.py
@@ -68,20 +68,20 @@ def evaluate(cfg: DictConfig) -> Tuple[Dict[str, Any], Dict[str, Any]]:
     if "_target_" in cfg.strategy:
         log.info(f"Instantiating strategy <{cfg.strategy._target_}>")
         strategy: Strategy = hydra.utils.instantiate(cfg.strategy)
-        if "mixed_precision" in strategy.__dict__ and strategy.mixed_precision is not None:
+        if "mixed_precision" in strategy.__dict__ and getattr(strategy, "mixed_precision", None) is not None:
             strategy.mixed_precision.param_dtype = (
                 resolve_omegaconf_variable(cfg.strategy.mixed_precision.param_dtype)
-                if cfg.strategy.mixed_precision.param_dtype is not None
+                if getattr(cfg.strategy.mixed_precision, "param_dtype", None) is not None
                 else None
             )
             strategy.mixed_precision.reduce_dtype = (
                 resolve_omegaconf_variable(cfg.strategy.mixed_precision.reduce_dtype)
-                if cfg.strategy.mixed_precision.reduce_dtype is not None
+                if getattr(cfg.strategy.mixed_precision, "reduce_dtype", None) is not None
                 else None
             )
             strategy.mixed_precision.buffer_dtype = (
                 resolve_omegaconf_variable(cfg.strategy.mixed_precision.buffer_dtype)
-                if cfg.strategy.mixed_precision.buffer_dtype is not None
+                if getattr(cfg.strategy.mixed_precision, "buffer_dtype", None) is not None
                 else None
             )