Python class
ProfilingConfig
ProfilingConfig
class max.pipelines.ProfilingConfig(*, config_file=None, section_name=None, gpu_profiling='off')
Bases: ConfigFileModel
Configuration for GPU profiling of pipeline models.
-
Parameters:
gpu_profiling
gpu_profiling: GPUProfilingMode
Whether to enable GPU profiling of the model.
model_config
model_config: ClassVar[ConfigDict] = {'extra': 'forbid', 'strict': False}
Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].
model_post_init()
model_post_init(context, /)
This function is meant to behave like a BaseModel method to initialise private attributes.
It takes context as an argument since that’s what pydantic-core passes when calling it.
-
Parameters:
-
- self (BaseModel) – The BaseModel instance.
- context (Any) – The context.
-
Return type:
-
None
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!