mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-05-22 22:32:18 +02:00
Log scheduler and optimizer ocnfigs
This commit is contained in:
parent
7a46fa021b
commit
2d0b810ed3
@ -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)
|
||||
|
||||
@ -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)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user