Transformer ABC notes genenerator

This reposetory contains solution for Yandex music generation contest. More about this solution you can read in my blog at medium.

Link to article

Data

You may to download data with abc notes from google drive, and unpack in project directory.

Training

Firstly wee need to train tokenizer.

python3 train_tokenizer trainset/abc abc.yttm

Then make some cleaning of the data

python3 clean_data.py trainset/abc cleaned_data

And starts training the tokenizer

python3 train.py cleaned_data

You may to setup some parameters, like gradient accamulation, batch size and e.t.c.

Generation

python3 generate.py testset/abc ABCModel/checkpoint-3/pytorch_model.bin

After that you gets a dirrectory with generated abc notes. You can convert abc to midi with abc2midi tool, or in web service.