Pular para o conteúdo principal

Treinar um modelo PyTorch

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

O MLflow PyTorch Notebook encaixa uma rede neural nos dados de reconhecimento de dígitos manuscritos do MNIST e logs executa os resultados em um servidor MLflow. Ele logs treinamento métricas e pesos no formato de evento TensorFlow localmente e, em seguida, carrega-os no diretório de artefatos da MLflow execução. Por fim, ele começa TensorBoard e lê os registros de eventos localmente.

MLflow PyTorch modelo treinamento Notebook

Open notebook in new tab