Preparar um modelo do PyTorch
O PyTorch é um pacote Python que fornece computação tensor acelerada por GPU e funcionalidade de alto nível para a construção de redes de aprendizagem profunda.
O notebook MLflow PyTorch se ajusta a uma rede neural em dados de reconhecimento de dígitos manuscritos MNIST e registra os resultados da execução em um servidor MLflow. Ele registra métricas e pesos de treinamento no formato de evento TensorFlow localmente e, em seguida, carrega-os para o diretório de artefatos da execução MLflow. Finalmente, ele inicia o TensorBoard e lê os eventos registrados localmente.
Quando estiveres pronto, poderás implementar o teu modelo usando Implementar modelos usando o Mosaic AI Model Serving.