create_strategy

deepinv.distributed.strategies.create_strategy(strategy_name, img_size, tiling_dims=None, **kwargs)

Create a distributed signal strategy by name.

Parameters:

• strategy_name (str) – name of the strategy ('basic', 'overlap_tiling').

• img_size (Sequence[int]) – full shape of the signal tensor, including batch and channel dimensions (e.g., (B, C, H, W)).

• tiling_dims (tuple[int, ...] | None) – dimensions to tile. If None, defaults to last N dimensions.

Returns:

the created strategy instance.

Return type:

DistributedSignalStrategy