unilab.training.dp_config.apply_dp_rank_config

unilab.training.dp_config.apply_dp_rank_config(cfg, rank)[source]

Apply the per-rank seed and return this rank’s visibility-local device.

Rank 0 keeps the configured seed; rank i>0 trains with seed + i until init broadcast lands in a later stage. The resolved rank-local device is returned instead of being written back into a synthetic training.device key; with rank-local CUDA visibility this is always cuda:0.

Parameters:
Return type:

str | None