(self)
| 46 | offload_policy: CPUOffloadPolicy | None = None |
| 47 | |
| 48 | def __post_init__(self) -> None: |
| 49 | if self.use_ddp and self.use_fsdp: |
| 50 | raise ValueError("DistributedConfig cannot enable both DDP and FSDP; choose only one.") |
| 51 | |
| 52 | def get_mixed_precision_policy(self) -> MixedPrecisionPolicy | None: |
| 53 | if self.fsdp_mixed_precision is None: |
nothing calls this directly
no outgoing calls
no test coverage detected