Load training configuration.
(config_path: str = "configs/base.yaml")
| 23 | |
| 24 | |
| 25 | def load_config(config_path: str = "configs/base.yaml") -> dict: |
| 26 | """Load training configuration.""" |
| 27 | with open(config_path) as f: |
| 28 | return yaml.safe_load(f) |
| 29 | |
| 30 | |
| 31 | def build_optimizer( |