SUPERB Submission Template
This repository can be used to generate a template so you can submit your pretrained model for evaluation on the leaderboard in the SUPERB Challenge.
Quickstart
1. Create an account and organisation on the Hugging Face Hub
First create an account on the Hugging Face Hub and you can sign up here if you haven't already! Next, create a new organization and invite the SUPERB Hidden Set Committee to join. You will upload your model to a repository under this organization so that members inside it can access the model.
2. Create a template repository on your machine
The next step is to create a template repository on your local machine that contains various files and a CLI to help you validate and submit your pretrained models. The Hugging Face Hub uses Git Large File Storage (LFS) to manage large files, so first install it if you don't have it already. For example, on macOS you can run:
brew install git-lfs
git lfs install
Next, run the following commands to create the repository. We recommend creating a Python virtual environment for the project, e.g. with Anaconda:
# Create and activate a virtual environment
conda create -n superb-submit python=3.8 && conda activate superb-submit
# Install the following libraries
pip install cookiecutter huggingface-hub==0.0.16
# Create the template repository
cookiecutter git+https://huggingface.co/superb/superb-submission
This will ask you to specify your Hugging Face Hub username, password, organisation, and the name of the repository:
hf_hub_username [<huggingface>]:
hf_hub_password [<password>]:
hf_hub_organisation [superb-submissions]:
repo_name [<my-superb-submissions>]:
This will trigger the following steps:
- Create a private dataset repository on the Hugging Face Hub under
{hf_hub_organisation}/{repo_name}
- Clone the repository to your local machine
- Add various template files, commit them locally to the repository, and push them to the Hub
The resulting repository should have the following structure:
my-superb-submission
βββ LICENSE
βββ README.md <- The README with submission instructions
βββ cli.py <- The CLI for validating predictions etc
βββ requirements.txt <- The requirements packages for the submissions
βββ expert.py <- Your model definition
βββ model.pt <- Your model weights
3. Install the dependencies
The final step is to install the project's dependencies:
# Navigate to the template repository
cd my-superb-submission
# Install dependencies
python -m pip install -r requirements.txt
That's it! You're now all set to start pretraining your speech models - see the instructions below on how to submit them to the Hub.
Submitting to the leaderboard
To make a submission to the leaderboard, there are 4 main steps:
Modify
expert.py
andmodel.py
so we can initialize an upstream model following the policy by:upstream = UpstreamExpert(ckpt="./model.pt")
Validate the upstream model meets the requirements in the policy. If everything is correct, you should see the following message: "All submission files validated! Now you can make a submission."
python cli.py validate
Push the predictions to the Hub! If there are no errors, you should see the following message: "Upload successful!"
python cli.py upload "commit message: my best model"
Make a submission at SUPERB website by uniquely indentifying this submission/model with the following information, which will be shown by:
python cli.py info
- Organization Name
- Repository Name
- Commit Hash (full 40 characters)
After you finish the above 4 steps. Please stay tuned and wait for us to get the finetuned results on the hidden set!