Log scheduler and optimizer ocnfigs

This commit is contained in:
mbsantiago 2026-05-05 00:42:49 +01:00
parent 7a46fa021b
commit 2d0b810ed3
2 changed files with 13 additions and 0 deletions

View File

@ -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)

View File

@ -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)