Menyimpan dan memulai kembali pekerjaan hibrida menggunakan pos pemeriksaan

Anda dapat menyimpan iterasi perantara dari pekerjaan hybrid Anda menggunakan pos pemeriksaan. Dalam contoh skrip algoritma dari bagian sebelumnya, Anda akan menambahkan baris berikut yang dikomentari dengan #ADD untuk membuat file pos pemeriksaan.

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!!!!!")

Saat Anda menjalankan pekerjaan hybrid, itu membuat file -checkpoint-1.json <jobname>di artefak pekerjaan hybrid Anda di direktori pos pemeriksaan dengan jalur default. /opt/jobs/checkpoints Skrip pekerjaan hybrid tetap tidak berubah kecuali Anda ingin mengubah jalur default ini.

Jika Anda ingin memuat pekerjaan hybrid dari pos pemeriksaan yang dihasilkan oleh pekerjaan hybrid sebelumnya, skrip algoritme menggunakanfrom braket.jobs import load_job_checkpoint. Logika untuk memuat dalam skrip algoritma Anda adalah sebagai berikut.

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

Setelah memuat pos pemeriksaan ini, Anda dapat melanjutkan logika berdasarkan konten yang dimuatcheckpoint-1.

Skrip orkestrasi Anda perlu menentukan job-arn dari pekerjaan hybrid sebelumnya dengan baris yang dikomentari dengan #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
    )