mirror of
https://github.com/macaodha/batdetect2.git
synced 2025-06-29 14:41:58 +02:00
Add manual logging of hyperparams
This commit is contained in:
parent
2ac968d65b
commit
4b6acd5e6e
@ -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)
|
||||
|
||||
|
@ -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"}),
|
||||
|
Loading…
Reference in New Issue
Block a user