Настройка MLflow в Google Colab

Question

Я часто использую Google Colab для обучения моделей TF / PyTorch, поскольку Colab предоставляет мне среду выполнения GPU / TPU. Кроме того, мне нравится работать с MLflow для хранения и сравнения обученных моделей, отслеживания прогресса, обмена и т. Д. c. Какие существуют решения для использования MLflow с Google Colab?

Answer 1

Попробуйте это. Но большой вопрос в том, как попасть в пользовательский интерфейс

https://github.com/dmatrix/mlflow-tests/blob/master/mllfow_test.ipynb

Один из вариантов - загрузить каталог / drive / mlruns на ваш локальный хост и запустить mlflow ui. на локальном хосте.

Не уверен, будет ли DISPLAY localhost: 0: 0 работать для удаленного сервера для локального отображения на вашем локальном хосте.