diff --git a/tinker_cookbook/utils/ml_log.py b/tinker_cookbook/utils/ml_log.py index e4928be5..9fd5b223 100644 --- a/tinker_cookbook/utils/ml_log.py +++ b/tinker_cookbook/utils/ml_log.py @@ -227,7 +227,7 @@ def __init__( def log_hparams(self, config: Any) -> None: """Log hyperparameters to wandb.""" if self.run and wandb is not None: - wandb.config.update(dump_config(config)) + wandb.config.update(dump_config(config), allow_val_change=True) def log_metrics(self, metrics: Dict[str, Any], step: int | None = None) -> None: """Log metrics to wandb."""