bridge.training.utils.config_utils#

Module Contents#

Functions#

create_ddp_config

Create a finalized Bridge DDP config for external model construction.

apply_run_config_backward_compat

Apply backward compatibility transformations to run config.

Data#

logger

API#

bridge.training.utils.config_utils.logger#

‘getLogger(…)’

bridge.training.utils.config_utils.create_ddp_config(
wrap_with_ddp: bool = True,
use_distributed_optimizer: bool = True,
use_megatron_fsdp: bool = False,
overrides: Mapping[str, object] | None = None,
finalize: bool = True,
) → object | None#

Create a finalized Bridge DDP config for external model construction.

bridge.training.utils.config_utils.apply_run_config_backward_compat(
config_dict: dict[str, Any],
) → dict[str, Any]#

Apply backward compatibility transformations to run config.

This function handles dataclass config fields that should not be passed to the constructor when loading older checkpoints. It automatically detects init=False fields by inspecting the target class.

The entire config is sanitized recursively to handle init=False fields in any part of the configuration hierarchy.

Parameters:

config_dict – The full run configuration dictionary.

Returns:

The config dictionary with backward compatibility fixes applied.