(devices: Union[str, int])
| 495 | |
| 496 | |
| 497 | def parse_devices(devices: Union[str, int]) -> int: |
| 498 | if devices in (-1, 'auto'): |
| 499 | return torch.cuda.device_count() or 1 |
| 500 | if isinstance(devices, int) and devices > 0: |
| 501 | return devices |
| 502 | raise ValueError(f"Devices must be 'auto' or a positive integer, got: {devices!r}") |
| 503 | |
| 504 | |
| 505 | def choose_logger( |