File di configurazione - Amazon SageMaker

Le traduzioni sono generate tramite traduzione automatica. In caso di conflitto tra il contenuto di una traduzione e la versione originale in Inglese, quest'ultima prevarrà.

File di configurazione

Amazon SageMaker Python SDK supporta l'impostazione di valori predefiniti per i tipi primitivi AWS dell'infrastruttura. Dopo che gli amministratori hanno configurato queste impostazioni predefinite, vengono passate automaticamente quando sono supportate le chiamate Python SageMaker . SDK APIs Gli argomenti per la funzione decoratore possono essere inseriti all'interno dei file di configurazione. In questo modo è possibile separare le impostazioni relative all'infrastruttura dal codice base. Per ulteriori informazioni sui parametri e gli argomenti per la funzione e i metodi remoti, consulta Remote function classes and methods specification.

Puoi configurare le impostazioni dell'infrastruttura per la configurazione di rete, IAM i ruoli, la cartella Amazon S3 per i dati di input, output e i tag all'interno del file di configurazione. Il file di configurazione può essere utilizzato quando si richiama una funzione utilizzando il decoratore @remote o il. RemoteExecutor API

Segue un esempio di file di configurazione che definisce le dipendenze, le risorse e altri argomenti. Questo file di configurazione di esempio viene utilizzato per richiamare una funzione che viene avviata utilizzando il decoratore @remote o il. RemoteExecutor API

SchemaVersion: '1.0' SageMaker: PythonSDK: Modules: RemoteFunction: Dependencies: 'path/to/requirements.txt' EnableInterContainerTrafficEncryption: true EnvironmentVariables: {'EnvVarKey': 'EnvVarValue'} ImageUri: '366666666666.dkr.ecr.us-west-2.amazonaws.com/my-image:latest' IncludeLocalWorkDir: true CustomFileFilter: IgnoreNamePatterns: - "*.ipynb" - "data" InstanceType: 'ml.m5.large' JobCondaEnvironment: 'your_conda_env' PreExecutionCommands: - 'command_1' - 'command_2' PreExecutionScript: 'path/to/script.sh' RoleArn: 'arn:aws:iam::366666666666:role/MyRole' S3KmsKeyId: 'yourkmskeyid' S3RootUri: 's3://amzn-s3-demo-bucket/my-project' VpcConfig: SecurityGroupIds: - 'sg123' Subnets: - 'subnet-1234' Tags: [{'Key': 'yourTagKey', 'Value':'yourTagValue'}] VolumeKmsKeyId: 'yourkmskeyid'

Il decoratore @remote e RemoteExecutor cercheranno Dependencies nei seguenti file di configurazione:

  • Un file di configurazione definito dall'amministratore.

  • Un file di configurazione definito dall'utente.

Le posizioni predefinite per questi file di configurazione dipendono e sono relative all'ambiente. Il seguente esempio di codice restituisce la posizione predefinita dei file di configurazione dell'amministratore e dell'utente. Questi comandi devono essere eseguiti nello stesso ambiente in cui si utilizza SageMaker PythonSDK.

import os from platformdirs import site_config_dir, user_config_dir #Prints the location of the admin config file print(os.path.join(site_config_dir("sagemaker"), "config.yaml")) #Prints the location of the user config file print(os.path.join(user_config_dir("sagemaker"), "config.yaml"))

È possibile escludere le posizioni predefinite di questi file impostando rispettivamente le variabili di ambiente SAGEMAKER_ADMIN_CONFIG_OVERRIDE e SAGEMAKER_USER_CONFIG_OVERRIDE per i percorsi dei file di configurazione definiti dall'amministratore e dall'utente.

Se esiste una chiave sia nel file di configurazione definito dall'amministratore che in quello definito dall'utente, verrà utilizzato il valore nel file definito dall'utente.