Tutorial: Modelos de aprendizado profundo de ponta a ponta no Azure Databricks

Este notebook de tutorial apresenta um exemplo de ponta a ponta de treinamento de um modelo de aprendizado profundo no Azure Databricks, incluindo o carregamento de dados, visualização dos dados, configuração de uma otimização de hiperparâmetro paralelo e uso do MLflow para examinar os resultados, registrar o modelo e executar inferência em novos dados usando o modelo registrado em uma UDF do Spark.

O notebook usa o PyTorch, um pacote python que fornece computação tensor acelerada por GPU e funcionalidade de alto nível para criar redes de aprendizado profundo.

Quando estiver pronto, você poderá implantar seu modelo usando o serviço de implantação de modelos do Mosaic AI.

Notebook de treinamento de modelo do PyTorch para MLflow

Obter notebook