search

orcasound / aifororcas-livesystem

Real-time AI-assisted killer whale notification system (model and moderator portal) :star:
http://orcahello.ai4orcas.net/
MIT License
37 stars 25 forks source link

Add finetuning.py script for automated retraining on (old/new) blended data #73

Closed aayushmnit closed 2 years ago

aayushmnit commented 3 years ago

image

The finetune.py script outlines the following process for continued update of the model:

  1. Downloading old/new training data and preprocessing it (placeholders)
  2. Blending old/new training data (implemented in data_blender)
  3. Downloading a seed model checkpoint (placeholder)
  4. Finetuning (all layers) the seed model on blended with FastAI framework conveniences
prakruti commented 3 years ago

So one overall comment @aayushmnit we're using the Black formatter for our code now. Instructions to install for VSCode are here https://marcobelo.medium.com/setting-up-python-black-on-visual-studio-code-5318eba4cd00. Please format all added files after installing it. Thanks!