(config: 'Config', checkpoint_dir: Path)
| 489 | |
| 490 | |
| 491 | def save_config(config: 'Config', checkpoint_dir: Path) -> None: |
| 492 | config_dict = asdict(config) |
| 493 | with open(checkpoint_dir / 'model_config.yaml', 'w') as fp: |
| 494 | yaml.dump(config_dict, fp) |
| 495 | |
| 496 | |
| 497 | def parse_devices(devices: Union[str, int]) -> int: |
nothing calls this directly
no outgoing calls
no test coverage detected