Make loading of certain components optional

capjason · Feb 10, 2023 · 9170ab5 · 9170ab5
1 parent c47f155
commit 9170ab5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/gdl/models/DECA.py b/gdl/models/DECA.py
@@ -234,7 +234,7 @@ def _init_lipread_loss(self):
         """
         Initialize the au perceptual loss (not currently used in EMOCA)
         """
-        if 'lipread_loss' in self.deca.config.keys():
+        if 'lipread_loss' in self.deca.config.keys() and self.deca.config.lipread_loss.get('load', True):
             if self.lipread_loss is not None:
                 force_override = True if 'force_override' in self.deca.config.lipread_loss.keys() \
                                          and self.deca.config.lipread_loss.force_override else False

diff --git a/gdl/models/EmoDECA.py b/gdl/models/EmoDECA.py
@@ -54,9 +54,12 @@ def __init__(self, config):
             "stage_name": "testing",
         }
         # instantiate the face net
-        self.deca = instantiate_deca(config.model.deca_cfg, deca_stage , "test", deca_checkpoint, deca_checkpoint_kwargs)
-        self.deca.inout_params.full_run_dir = config.inout.full_run_dir
-        self._setup_deca(False)
+        if bool(deca_checkpoint):
+            self.deca = instantiate_deca(config.model.deca_cfg, deca_stage , "test", deca_checkpoint, deca_checkpoint_kwargs)
+            self.deca.inout_params.full_run_dir = config.inout.full_run_dir
+            self._setup_deca(False)
+        else: 
+            self.deca = None
 
         # which latent codes are being used
         in_size = 0