I1 - authentication, connection and jobs
from azure.ai.ml import MLClient, command
from azure.identity import DefaultAzureCredential
# authenticate
credential = DefaultAzureCredential()
# connect to AML workspace
ml_client = MLClient(
credential=credential,
subscription_id="<SUBSCRIPTION_ID>",
resource_group_name="<RESOURCE_GROUP>",
workspace_name="<WORKSPACE_NAME>",
)
# define job
job = command(
code="./src",
command="python train.py",
environment="azureml:train-env:4",
compute="cpu-cluster",
)
# submit job
returned_job = ml_client.jobs.create_or_update(
job,
experiment_name="detector-training",
)
# stream logs/station until completion
ml_client.jobs.stream(returned_job.name)