View a markdown version of this page

Algoritmos e frameworks compatíveis - SageMaker IA da Amazon

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Algoritmos e frameworks compatíveis

nota

O Amazon SageMaker Debugger não está mais aberto a novos clientes. Os clientes existentes podem continuar usando o serviço normalmente. AWS continua investindo em melhorias de segurança e disponibilidade para o Debugger, mas não planejamos introduzir novos recursos. Para obter mais informações, consulte Alteração na disponibilidade do depurador.

A tabela a seguir mostra as estruturas e algoritmos de aprendizado de máquina de SageMaker IA suportados pelo Debugger.

SageMaker AI-supported frameworks and algorithms Debugging output tensors

TensorFlow

AWS TensorFlow contêineres de aprendizado profundo 1.15.4 ou posterior

PyTorch

AWS PyTorch contêineres de aprendizado profundo 1.5.0 ou posterior

MXNet

AWS Contêineres de aprendizado profundo MXNet 1.6.0 ou posterior

XGBoost

1.0-1, 1.2-1, 1.3-1

SageMaker IA genérica ModelTrainer

Contêineres de treinamento personalizados (disponíveis para TensorFlow PyTorch, MXNet e XGBoost com registro manual de ganchos)

  • Depuração de tensores de saída: Monitore e depure os parâmetros do modelo, como pesos, gradientes, tendenciosos e valores escalares do seu trabalho de treinamento. As estruturas de aprendizado profundo disponíveis são Apache MXNet, TensorFlow PyTorch, e XGBoost.

    Importante

    Para a TensorFlow estrutura com Keras, o SageMaker Debugger descontinua o suporte de alteração de código zero para modelos de depuração criados usando os módulos da versão 2.6 e posteriores. tf.keras TensorFlow Isso se deve às mudanças significativas anunciadas na nota de lançamento TensorFlow 2.6.0. Para obter instruções sobre como atualizar seu script de treinamento, consulte Adapte seu roteiro TensorFlow de treinamento.

    Importante

    A partir da PyTorch v1.12.0 e versões posteriores, o SageMaker Debugger descontinua o suporte de alteração de código zero para modelos de depuração.

    Isso ocorre devido a alterações significativas que fazem com que o SageMaker Debugger interfira na funcionalidade. torch.jit Para obter instruções sobre como atualizar seu script de treinamento, consulte Adapte seu roteiro PyTorch de treinamento.

Se a estrutura ou algoritmo que você deseja treinar e depurar não estiver listado na tabela, acesse o Fórum de AWS discussão e deixe um comentário sobre o SageMaker Debugger.

Regiões da AWS

O Amazon SageMaker Debugger está disponível em todas as regiões em que o Amazon SageMaker AI está em serviço, exceto na seguinte região.

  • Ásia-Pacífico (Jacarta): ap-southeast-3

Para descobrir se a Amazon SageMaker AI está em serviço no seu Região da AWS, consulte Serviços AWS regionais.

Use o Depurador com contêineres de treinamento personalizados

Traga seus contêineres de treinamento para a SageMaker IA e obtenha informações sobre seus trabalhos de treinamento usando o Debugger. Maximize sua eficiência de processamento otimizando seu modelo nas instâncias do Amazon EC2, usando os atributos de monitoramento e depuração.

Para obter mais informações sobre como compilar seu contêiner de treinamento com a biblioteca de cliente sagemaker-debugger, enviá-lo para o Amazon Elastic Container Registry (Amazon ECR), monitorar e depurar, consulte Use o Depurador com contêineres de treinamento personalizados.

Repositórios do Debugger Open-Source GitHub

As APIs do depurador são fornecidas por meio do SDK do SageMaker Python e projetadas para criar configurações de ganchos e regras do Debugger para as operações de IA e API. SageMaker CreateTrainingJob DescribeTrainingJob A biblioteca de clientes sagemaker-debugger fornece ferramentas para registrar hooks e acessar os dados de treinamento por meio de seu atributo de avaliação, por meio de suas operações de API flexíveis e avançadas. Ele suporta as estruturas de aprendizado de máquina TensorFlow PyTorch, MXNet e XGBoost no Python 3.6 e versões posteriores.

Para recursos diretos sobre o Depurador e as operações de API sagemaker-debugger, consulte os seguintes links:

Se você usa o SDK para Java para realizar trabalhos de SageMaker treinamento e quiser configurar as APIs do Debugger, consulte as seguintes referências: