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.