syne_tune.backend.sagemaker_backend.instance_info module

class syne_tune.backend.sagemaker_backend.instance_info.InstanceInfo(name, num_cpu, num_gpu, cost_per_hour)

Bases: object

name: str

num_cpu: int

num_gpu: int

cost_per_hour: float

class syne_tune.backend.sagemaker_backend.instance_info.InstanceInfos

Bases: object

Utility to get information of an instance type (num cpu/gpu, cost per hour).

syne_tune.backend.sagemaker_backend.instance_info.select_instance_type(min_gpu=0, max_gpu=16, min_cost_per_hour=None, max_cost_per_hour=None)

Parameters:

• min_gpu (int) –

• max_gpu (int) –

• min_cost_per_hour (Optional[float]) –

• max_cost_per_hour (Optional[float]) –

Return type:

List[str]

Returns:

a list of instance type that met the required constrain on minimum/maximum number of GPU and

minimum/maximum cost per hour.