Entrenamiento de modelos de PyTorch a gran escala con Azure Machine Learning
PyTorch es un paquete de Python que proporciona cálculo de tensores acelerado por GPU y funcionalidad de alto nivel para crear redes de aprendizaje profundo.
El cuaderno de PyTorch de MLflow ajusta una red neuronal en los datos de reconocimiento de dígitos escritos a mano de MNIST y registra los resultados de la ejecución en un servidor de MLflow. Registra las métricas y ponderaciones de entrenamiento en formato de evento TensorFlow localmente y, a continuación, las carga en el directorio de artefactos de la ejecución de MLflow. Por último, inicia TensorBoard y lee los eventos registrados localmente.
Cuando esté listo, puede implementar el modelo mediante Servicio de modelos con Azure Databricks.