From 628b06091acc2e1fbf3661c68e9ec224c97668fe Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Lucas=20de=20Sousa=20Almeida?= Date: Mon, 30 Dec 2024 11:59:03 -0300 Subject: [PATCH] Sharing models before instantiating classes MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: João Lucas de Sousa Almeida --- terratorch/cli_tools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/terratorch/cli_tools.py b/terratorch/cli_tools.py index a00f2ad3..e8dceae1 100644 --- 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,