onnx_diagnostic.helpers.config_helper¶
- onnx_diagnostic.helpers.config_helper.build_diff_config(config0, config1)[source][source]¶
Returns all the modified values between two configuration
- onnx_diagnostic.helpers.config_helper.check_hasattr(config: Any, *args: str | Tuple[Any, ...])[source][source]¶
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 [source][source]¶
Retrieves the configuration class for a given architecture.
- Parameters:
arch – architecture (clas name)
exc – raise an exception if not found
- Returns:
type
Returns the default number of layers. It is lower when the unit tests are running when
UNITTEST_GOING=1
.