It is not a released tool, but only a snapshot of a private repository. I'm not adding documentation yet, however everything is self-explanatory.

To install pre-commit hooks:

pip install pre-commit
pre-commit install

To fine-tune ASR models:

python train.py --config=configs/base_config.py

A lot of useful information is saved:

• Loss/WER plots
• Model predictions
• Weight dynamics
• Training logs

The file configs/base_config.py contains all details such as:

• The model checkpoint to use (Wav2vec2 or Whisper)
• Training and evaluation datasets
• Training hyperparameters
• Path to save results

See example run in example folder.