Load configuration from YAML file.
(self, path: Path)
| 49 | yaml.dump(self, f) |
| 50 | |
| 51 | def load(self, path: Path): |
| 52 | """Load configuration from YAML file.""" |
| 53 | data = yaml.load(path.read_text(), Loader=yaml.Loader) |
| 54 | if isinstance(data, dict): # for training |
| 55 | self.load_dict(data) |
| 56 | elif isinstance(data, self.__class__): |
| 57 | self = data |
| 58 | else: |
| 59 | raise ValueError(f"Invalid config file: {path}") |
| 60 | # config = cls(**config) # if yaml.dump(self.__dict__, ...) is used |
| 61 | return self |
| 62 | |
| 63 | def load_dict(self, data: dict): |
| 64 | if "model" in data: |