Поделиться через


Обучение модели PyTorch

PyTorch — это пакет Python, обеспечивающий тензорное вычисление с ускорением GPU и высокоуровневые функции для создания сетей глубокого обучения.

Записная книжка MLflow PyTorch создана для нейронной сети на основе результатов распознавания рукописных цифр MNIST и ведет журнал результатов запусков на сервере MLflow. Она регистрирует метрики обучения и весовые коэффициенты локально в формате событий TensorFlow, а затем передает их в каталог артефактов запуска MLflow. Наконец, она запускает TensorBoard и считывает зарегистрированные локально события.

Когда вы будете готовы к развертыванию модели с помощью модели, обслуживаемой с помощью Azure Databricks.

Записная книжка обучения модели MLflow PyTorch

Получить записную книжку