Add manual logging of hyperparams

This commit is contained in:
mbsantiago 2025-06-26 11:59:33 -06:00
parent 2ac968d65b
commit 4b6acd5e6e
2 changed files with 2 additions and 2 deletions

View File

@ -40,8 +40,6 @@ class TrainingModule(L.LightningModule):
self.learning_rate = learning_rate
self.t_max = t_max
self.save_hyperparameters()
def forward(self, spec: torch.Tensor) -> ModelOutput:
return self.detector(spec)

View File

@ -84,6 +84,8 @@ def train(
)
logger = build_logger(config.logger)
if logger and hasattr(logger, 'log_hyperparams'):
logger.log_hyperparams(config.model_dump(exclude_none=True))
trainer = Trainer(
**config.trainer.model_dump(exclude_none=True, exclude={"logger"}),