Guardar y reiniciar los trabajos híbridos mediante puntos de control

Puede guardar las iteraciones intermedias de sus trabajos híbridos mediante puntos de control. En el ejemplo del script de algoritmos de la sección anterior, añadiría las siguientes líneas comentadas con #ADD para crear archivos de puntos de control.

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

Al ejecutar el trabajo híbrido, se crea el archivo -checkpoint-1.json <jobname>en los artefactos del trabajo híbrido del directorio de puntos de control con una ruta predeterminada. /opt/jobs/checkpoints El script del trabajo híbrido permanece inalterado a menos que desee cambiar esta ruta predeterminada.

Si desea cargar un trabajo híbrido desde un punto de control generado por un trabajo híbrido anterior, utilice from braket.jobs import load_job_checkpoint el script del algoritmo. La lógica que se debe cargar en el script de su algoritmo es la siguiente.

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

Tras cargar este punto de control, puede continuar con la lógica en función del contenido cargado en él. checkpoint-1

El guion de orquestación debe especificar lo del trabajo híbrido anterior con la línea job-arn comentada con #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
    )