README.md

FINETUNE WAV2VEC 2.0 FOR SPEECH RECOGNITION

Table of contents

1. Documentation
2. Installation
3. Usage
4. Logs and Visualization

Documentation

Suppose you need a simple way to fine-tune the Wav2vec 2.0 model for the task of Speech Recognition on your datasets, then you came to the right place.
All documents related to this repo can be found here:

• Wav2vec2ForCTC
• Tutorial
• Code reference

Installation

pip install -r requirements.txt

Usage

1. Prepare your dataset
   • Your dataset can be in .txt or .csv format.
   • path and transcript columns are compulsory. The path column contains the paths to your stored audio files, depending on your dataset location, it can be either absolute paths or relative paths. The transcript column contains the corresponding transcripts to the audio paths.
   • Check out our data_example.csv file for more information.
2. Configure the config.toml file
3. Run
   • Start training:

     python train.py -c config.toml
     

   • Continue to train from resume:

     python train.py -c config.toml -r
     

   • Load specific model and start training:

     python train.py -c config.toml -p path/to/your/model.tar
     

Logs and Visualization

The logs during the training will be stored, and you can visualize it using TensorBoard by running this command:

# specify the <name> in config.json
tensorboard --logdir ~/saved/<name>

# specify a port 8080
tensorboard --logdir ~/saved/<name> --port 8080