ensinando o modelo PyTorch

PyTorch é um pacote Python que fornece computação tensor acelerada por GPU e funcionalidade de alto nível para a construção de redes profundas de aprendizagem.

O MLflow PyTorch Notebook se encaixa em uma rede neural em dados de reconhecimento de dígitos manuscritos MNIST e logs os resultados da execução em um servidor MLflow. Ele logs métricas e pesos de treinamento no formato de evento do TensorFlow localmente e, em seguida upload os no diretório de artefatos da execução do MLflow. Por fim, comece o TensorBoard e leia os logs de eventos localmente.

Quando estiver pronto, o senhor pode implantar seu modelo usando o servindo modelo com o Databricks.

Notebook de treinamento do modelo MLflow PyTorch

Abra o bloco de anotações em outra guia Copiar link para importação