체크포인트를 사용하여 하이브리드 작업 저장 및 재시작

체크포인트를 사용하여 하이브리드 작업의 중간 반복을 저장할 수 있습니다. 이전 섹션의 알고리즘 스크립트 예제에서 체크포인트 파일을 생성하기 위해 #ADD으로 주석이 지정된 다음 줄을 추가합니다.

from braket.aws import AwsDevice
from braket.circuits import Circuit
from braket.jobs import save_job_checkpoint  #ADD
import os

def start_here():

    print("Test job starts!!!!!")

    device = AwsDevice(os.environ["AMZN_BRAKET_DEVICE_ARN"])

    #ADD the following code
    job_name = os.environ["AMZN_BRAKET_JOB_NAME"]
     save_job_checkpoint(
     checkpoint_data={"data": f"data for checkpoint from {job_name}"},
     checkpoint_file_suffix="checkpoint-1",
     ) #End of ADD

    bell = Circuit().h(0).cnot(0, 1)
    for count in range(5):
        task = device.run(bell, shots=100)
        print(task.result().measurement_counts)

    print("Test hybrid job completed!!!!!")

하이브리드 작업을 실행하면 기본 /opt/jobs/checkpoints 경로가 있는 체크포인트 디렉터리의 하이브리드 작업 아티팩트에 <jobname>-checkpoint-1.json 파일이 생성됩니다. 하이브리드 작업 스크립트는 이 기본 경로를 변경하지 않는 한 변경되지 않습니다.

이전 하이브리드 작업에서 생성된 체크포인트에서 하이브리드 작업을 로드하려는 경우 알고리즘 스크립트는 를 사용합니다from braket.jobs import load_job_checkpoint. 알고리즘 스크립트에 로드할 로직은 다음과 같습니다.

checkpoint_1 = load_job_checkpoint(
    "previous_job_name",
    checkpoint_file_suffix="checkpoint-1",
)

이 체크포인트를 로드한 후 에 로드된 콘텐츠를 기반으로 로직을 계속할 수 있습니다checkpoint-1.

참고

checkpoint_file_suffix는 체크포인트를 생성할 때 이전에 지정된 접미사와 일치해야 합니다.

오케스트레이션 스크립트는 #로 주석이 지정된 줄을 사용하여 이전 하이브리드 작업job-arn의 를 지정해야 합니다ADD.

job = AwsQuantumJob.create(
    source_module="source_dir",
    entry_point="source_dir.algorithm_script:start_here",
    device_arn="arn:aws:braket:::device/quantum-simulator/amazon/sv1",
    copy_checkpoints_from_job="<previous-job-ARN>", #ADD
    )