Save the general/model hyperparameters in the Weights & Biases dashboard. Args: wandb_logger (any): Wandb logger. general_hyperparameters (dict): General hyperparameters. model_hyperparameters (dict): Model hyperparameters. Returns: None.
(
wandb_logger: Any,
general_hyperparameters: dict[str, Any],
model_hyperparameters: dict[str, Any],
)
| 362 | |
| 363 | |
| 364 | def wandb_hyperparameters_saving( |
| 365 | wandb_logger: Any, |
| 366 | general_hyperparameters: dict[str, Any], |
| 367 | model_hyperparameters: dict[str, Any], |
| 368 | ) -> None: |
| 369 | """ |
| 370 | Save the general/model hyperparameters in the Weights & Biases dashboard. |
| 371 | Args: |
| 372 | wandb_logger (any): Wandb logger. |
| 373 | general_hyperparameters (dict): General hyperparameters. |
| 374 | model_hyperparameters (dict): Model hyperparameters. |
| 375 | |
| 376 | Returns: |
| 377 | None. |
| 378 | """ |
| 379 | wbl = wandb_logger |
| 380 | for key in general_hyperparameters: |
| 381 | wbl.experiment.config[key] = general_hyperparameters[key] |
| 382 | for key in model_hyperparameters: |
| 383 | wbl.experiment.config[key] = model_hyperparameters[key] |
| 384 | |
| 385 | |
| 386 | def str2bool(v): |