Using MLflow with Hugging Face Transformers

Julien Simon
Jul 15, 2022

--

In this video, I show how to use MLflow with the Transformers library, and why it’s a good idea to store the logs on the Hugging Face hub :)

Let us know how you’d like to use MLflow with Transformers and the Hugging Face hub! Please join the discussion at https://discuss.huggingface.co/t/calling-mlflow-users/20420

Doc: https://huggingface.co/docs/transformers/v4.20.1/en/main_classes/callback#transformers.integrations.MLflowCallback

Notebook: https://gitlab.com/juliensimon/huggingface-demos/-/tree/main/mlflow

--

--

Julien Simon
Julien Simon

Responses (1)