AcceleratorPartitionConfig
¶
Source code in docs_src/capo_sagemaker/types/accelerator_partition_config.py
count
instance-attribute
¶
count: NotRequired[Integer]
The number of accelerator partitions to allocate with the specified partition type. If you don't specify a value for vCPU and MemoryInGiB, SageMaker AI automatically allocates ratio-based values for those parameters based on the accelerator partition count you provide.
type
instance-attribute
¶
type: NotRequired[MIGProfileType]
The Multi-Instance GPU (MIG) profile type that defines the partition configuration. The profile specifies the compute and memory allocation for each partition instance. The available profile types depend on the instance type specified in the compute quota configuration.