From 7c877a169575a76d6764476f78316dd81e53d066 Mon Sep 17 00:00:00 2001 From: ioangatop Date: Mon, 4 Dec 2023 16:51:17 +0100 Subject: [PATCH] Call `configure_model` from LightningCLI --- src/lightning/pytorch/cli.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/lightning/pytorch/cli.py b/src/lightning/pytorch/cli.py index f2ecc8d12b08f..9a68e0d9abb33 100644 --- a/src/lightning/pytorch/cli.py +++ b/src/lightning/pytorch/cli.py @@ -656,6 +656,7 @@ def get_automatic( f"`{self.__class__.__name__}.configure_optimizers`." ) + self.model.configure_model() optimizer = instantiate_class(self.model.parameters(), optimizer_init) lr_scheduler = instantiate_class(optimizer, lr_scheduler_init) if lr_scheduler_init else None fn = partial(self.configure_optimizers, optimizer=optimizer, lr_scheduler=lr_scheduler)