Hyper-parameters class Store hyper-parameters in training / infer / ... scripts.
| 15 | |
| 16 | |
| 17 | class HParams(dict): |
| 18 | """ Hyper-parameters class |
| 19 | |
| 20 | Store hyper-parameters in training / infer / ... scripts. |
| 21 | """ |
| 22 | |
| 23 | def __getattr__(self, name): |
| 24 | if name in self.keys(): |
| 25 | return self[name] |
| 26 | for v in self.values(): |
| 27 | if isinstance(v, HParams): |
| 28 | if name in v: |
| 29 | return v[name] |
| 30 | raise AttributeError(f"'HParams' object has no attribute '{name}'") |
| 31 | |
| 32 | def __setattr__(self, name, value): |
| 33 | self[name] = value |
| 34 | |
| 35 | def save(self, filename): |
| 36 | with open(filename, 'w', encoding='utf-8') as fp: |
| 37 | json.dump(self, fp, ensure_ascii=False, indent=4, sort_keys=False) |
| 38 | |
| 39 | def load(self, filename): |
| 40 | with open(filename, 'r', encoding='utf-8') as fp: |
| 41 | params_dict = json.load(fp) |
| 42 | for k, v in params_dict.items(): |
| 43 | if isinstance(v, dict): |
| 44 | self[k].update(HParams(v)) |
| 45 | else: |
| 46 | self[k] = v |
| 47 | |
| 48 | |
| 49 | def parse_args(parser): |
no outgoing calls
no test coverage detected
searching dependent graphs…