onnx_diagnostic.helpers.config_helper

onnx_diagnostic.helpers.config_helper.build_diff_config(config0, config1)

Returns all the modified values between two configuration

onnx_diagnostic.helpers.config_helper.check_hasattr(config: Any, *args: str | Tuple[Any, ...])

Checks the confiugation has all the attributes in args. Raises an exception otherwise.

onnx_diagnostic.helpers.config_helper.config_class_from_architecture(arch: str, exc: bool = False) → type | None

Retrieves the configuration class for a given architecture.

Parameters:

• arch – architecture (clas name)

• exc – raise an exception if not found

Returns:

type

onnx_diagnostic.helpers.config_helper.default_num_hidden_layers()

Returns the default number of layers. It is lower when the unit tests are running when UNITTEST_GOING=1.

onnx_diagnostic.helpers.config_helper.pick(config, name: str, default_value: Any) → Any

Returns the value of a attribute if config has it otherwise the default value.

onnx_diagnostic.helpers.config_helper.update_config(config: Any, mkwargs: Dict[str, Any])

Updates a configuration with different values.