The Wayback Machine - http://web.archive.org/web/20221228090921/https://github.com/Unity-Technologies/ml-agents/pull/5732
Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Integrating Hugging Face Hub 🤗 #5732

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

simoninithomas
Copy link

@simoninithomas simoninithomas commented Apr 20, 2022

Hey there 👋 ,

Proposed change(s)

  • Integrating Hugging Face Hub 🤗 in ML-Agents to allow users to download and upload models to the Hub.

I'm Thomas Simonini from Hugging Face 🤗 and we work on the integration of Hugging Face Hub with ML-Agents.
The Hugging Face Hub works as a central place where anyone can share and explore saved models.
The idea is to allow people to share and save their models on the hub.

The PR is a draft for now because none of the checklist are done.

An example of a model card and the tensorboard dashboard integrated in the Hub: https://huggingface.co/ThomasSimonini/Huggy_test

Here's an example:

  1. You train your agent mlagents-learn --config="huggy.yaml" --run-id="Huggy_1"
  2. After the training is done you push your model to the 🤗 Hub with mlagents-push-to-hf --run-id="Huggy_1" --local-dir="results/Huggy_1" --repo-id="ThomasSimonini/Huggy_test" --commit-message="Huggy upload"
  3. Someone else wants to download it, it just need to type mlagents-load-from-hf --repo-id="ThomasSimonini/Huggy_test" --local-dir="./downloads"

Types of change(s)

  • Bug fix
  • New feature
  • Code refactor
  • Breaking change
  • Documentation update
  • Other (please describe)

Checklist

  • Added tests that prove my fix is effective or that my feature works
  • Updated the changelog (if applicable)
  • Updated the documentation (if applicable)
  • Updated the migration guide (if applicable)

Other comments

* push_to_hf
* load_from_hf
@simoninithomas simoninithomas changed the title Integrating Hugging Face Hub Integrating Hugging Face Hub 🤗 Apr 20, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants