lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-05-22 22:32:18 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Log scheduler and optimizer ocnfigs

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/batdetect2/train/optimizers.py
View File

@ -3,6 +3,7 @@
from collections.abc import Iterable from collections.abc import Iterable
from typing import Annotated, Literal from typing import Annotated, Literal
from loguru import logger
from pydantic import Field from pydantic import Field
from torch import nn from torch import nn
from torch.optim import Adam, Optimizer from torch.optim import Adam, Optimizer
@ -84,4 +85,10 @@ def build_optimizer(
Optimizer configuration. Defaults to ``AdamOptimizerConfig``. Optimizer configuration. Defaults to ``AdamOptimizerConfig``.
""" """
config = config or 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) return optimizer_registry.build(config, parameters)

6
src/batdetect2/train/schedulers.py
View File

@ -2,6 +2,7 @@
from typing import Annotated, Literal from typing import Annotated, Literal
from loguru import logger
from pydantic import Field from pydantic import Field
from torch.optim import Optimizer from torch.optim import Optimizer
from torch.optim.lr_scheduler import CosineAnnealingLR, LRScheduler from torch.optim.lr_scheduler import CosineAnnealingLR, LRScheduler
@ -78,4 +79,9 @@ def build_scheduler(
"""Build a scheduler from configuration.""" """Build a scheduler from configuration."""
config = config or CosineAnnealingSchedulerConfig() 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) return scheduler_registry.build(config, optimizer)