This is a toolkit for symbolic music generation in pyTorch. You can generate tracks from scratch (unconditional) or continuation for a given track.
- Using pytorch_lightning as backend.
- Using Deepmusic for encoding/decoding MIDI files.
- Supporting different transformer models + vanilla rnn. (vanilla, linear, huggingface transformers)
- Supporting Compound Word and REMI representations. (see references)
Please see main.ipynb to learn how to train a model and generate music!
check out assets directory!
[1] https://github.com/YatingMusic/compound-word-transformer
[2] https://github.com/YatingMusic/remi