Sharing models before instantiating classes

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Dec 30, 2024 · 628b060 · 628b060
1 parent 30dfdf1
commit 628b060
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/terratorch/cli_tools.py b/terratorch/cli_tools.py
@@ -370,7 +370,6 @@ def add_arguments_to_parser(self, parser: LightningArgumentParser) -> None:
 
     def instantiate_classes(self) -> None:
 
-        super().instantiate_classes()
         # get the predict_output_dir. Depending on the value of run, it may be in the subcommand
         try:
             config = self.config.predict
@@ -398,6 +397,7 @@ def instantiate_classes(self) -> None:
             custom_modules_path = os.getenv("TERRATORCH_CUSTOM_MODULE_PATH", None)
 
         import_custom_modules(custom_modules_path)
+        super().instantiate_classes()
 
 def build_lightning_cli(
     args: ArgsType = None,