castorini / howl

Wake word detection modeling toolkit for Firefox Voice, supporting open datasets like Speech Commands and Common Voice.
Mozilla Public License 2.0
199 stars 30 forks source link

Add flag to prevent duplicate inference and refactor PyTorch hub config #45

Closed edwinzhng closed 3 years ago

edwinzhng commented 3 years ago

Now the steps to add a new pretrained model are:

  1. Upload the model weights + settings to the howl-models repo and create a new release
  2. Add a short function to load the model in hubconf.py similar to the one for hey_fire_fox
    def hey_fire_fox(pretrained=True, **kwargs):
    """Pretrained model for Firefox Voice"""
    engine, ctx = _load_model(pretrained, "res8", "howl/hey-fire-fox", **kwargs)
    return engine, ctx
edwinzhng commented 3 years ago

@ljj7975 should I add the instructions for uploading a new pretrained model into the README?