diff --git a/src/batdetect2/train/optimizers.py b/src/batdetect2/train/optimizers.py index c5b79cc..092e57a 100644 --- a/src/batdetect2/train/optimizers.py +++ b/src/batdetect2/train/optimizers.py @@ -3,6 +3,7 @@ from collections.abc import Iterable from typing import Annotated, Literal +from loguru import logger from pydantic import Field from torch import nn from torch.optim import Adam, Optimizer @@ -84,4 +85,10 @@ def build_optimizer( Optimizer configuration. Defaults to ``AdamOptimizerConfig``. """ config = config or AdamOptimizerConfig() + + logger.opt(lazy=True).debug( + "Building optimizer with config: \n{}", + lambda: config.to_yaml_string(), + ) + return optimizer_registry.build(config, parameters) diff --git a/src/batdetect2/train/schedulers.py b/src/batdetect2/train/schedulers.py index e60e7df..73ebd09 100644 --- a/src/batdetect2/train/schedulers.py +++ b/src/batdetect2/train/schedulers.py @@ -2,6 +2,7 @@ from typing import Annotated, Literal +from loguru import logger from pydantic import Field from torch.optim import Optimizer from torch.optim.lr_scheduler import CosineAnnealingLR, LRScheduler @@ -78,4 +79,9 @@ def build_scheduler( """Build a scheduler from configuration.""" config = config or CosineAnnealingSchedulerConfig() + logger.opt(lazy=True).debug( + "Building scheduler with config: \n{}", + lambda: config.to_yaml_string(), + ) + return scheduler_registry.build(config, optimizer)