(cfg: DictConfig)
| 111 | |
| 112 | |
| 113 | def log_config(cfg: DictConfig): |
| 114 | print(om.to_yaml(cfg)) |
| 115 | if "wandb" in cfg.get("loggers", {}): |
| 116 | try: |
| 117 | import wandb |
| 118 | except ImportError as e: |
| 119 | raise e |
| 120 | if wandb.run: |
| 121 | wandb.config.update(om.to_container(cfg, resolve=True)) |
| 122 | |
| 123 | |
| 124 | def build_algorithm(name, kwargs): |