Config

class braket.jobs.config.CheckpointConfig(localPath='/opt/jobs/checkpoints', s3Uri=None)[source]

Bases: object

Configuration that specifies the location where checkpoint data is stored.

Parameters:
  • localPath (str)

  • s3Uri (str | None)

localPath: str = '/opt/jobs/checkpoints'
s3Uri: str | None = None
class braket.jobs.config.InstanceConfig(instanceType='ml.m5.large', volumeSizeInGb=30, instanceCount=1)[source]

Bases: object

Configuration of the instance(s) used to run the hybrid job.

Parameters:
  • instanceType (str)

  • volumeSizeInGb (int)

  • instanceCount (int)

instanceType: str = 'ml.m5.large'
volumeSizeInGb: int = 30
instanceCount: int = 1
class braket.jobs.config.OutputDataConfig(s3Path=None, kmsKeyId=None)[source]

Bases: object

Configuration that specifies the location for the output of the hybrid job.

Parameters:
  • s3Path (str | None)

  • kmsKeyId (str | None)

s3Path: str | None = None
kmsKeyId: str | None = None
class braket.jobs.config.StoppingCondition(maxRuntimeInSeconds=432000)[source]

Bases: object

Conditions that specify when the hybrid job should be forcefully stopped.

Parameters:

maxRuntimeInSeconds (int)

maxRuntimeInSeconds: int = 432000
class braket.jobs.config.DeviceConfig(device)[source]

Bases: object

Parameters:

device (str)

device: str
class braket.jobs.config.S3DataSourceConfig(s3_data, content_type=None)[source]

Bases: object

Data source for data that lives on S3.

config

config passed to the Braket API

Type:

dict[str, dict]

Parameters:
  • s3_data (str)

  • content_type (str | None)

Create a definition for input data used by a Braket Hybrid job.

Parameters:
  • s3_data (str) – Defines the location of s3 data to train on.

  • content_type (str | None) – MIME type of the input data (default: None).