Treinar um modelo do PyTorch
O PyTorch é um pacote do Python que fornece computação de tensor acelerada por GPU e funcionalidade de alto nível para a criação de redes de aprendizado profundo.
O notebook do PyTorch para MLflow ajusta uma rede neural de dados de reconhecimento de dígitos manuscritos do MNIST e os resultados da execução de logs a um servidor do MLflow. Ele registra as métricas de treinamento e os pesos no formato de evento do TensorFlow localmente e carrega-os no diretório de artefatos da execução do MLflow. Por fim, inicia o TensorBoard e lê os eventos registrados localmente.
Quando estiver pronto, implante seu modelo usando Implantar modelos usando o Serviço de Modelo de IA do Mosaic.