braintools.trainer.get_strategy# braintools.trainer.get_strategy(strategy='auto', **kwargs)[source]# Get a distributed training strategy by name. Parameters: strategy (str | Strategy | None) – Strategy name or instance. Options: ‘auto’, ‘single’, ‘ddp’, ‘dp’, ‘fsdp’, ‘sdp’ **kwargs – Additional arguments passed to the strategy constructor. Returns: The requested strategy instance. Return type: Strategy Examples >>> strategy = get_strategy('ddp') >>> strategy = get_strategy('fsdp', prefer_fsdp=True)
