DEV Community

Julien Simon
Julien Simon

Posted on • Originally published at julsimon.Medium on

Using MLflow with Hugging Face Transformers

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

Top comments (0)