wazenmai / MIDI-BERT

This is the official repository for the paper, MidiBERT-Piano: Large-scale Pre-training for Symbolic Music Understanding.
MIT License
181 stars 23 forks source link

Classical Piano Compositions #14

Open oatsvine opened 8 months ago

oatsvine commented 8 months ago

I want to give this model a quick try first. You have a nice way to perform an evaluation with minimal setup. However, I noticed your note about checkpoints being trained for pop music. Is there a different checkpoint trained on classical compositions?

wazenmai commented 8 months ago

Hi, indeed, our pre-trained checkpoint is mainly trained on pop music. To be more specifically, we used 5 datasets presented in data_creation. You can get more information about our pre-trained checkpoints and models in the paper. We did not use other datasets to train our model.

Besides, feel free to use your own dataset to train your checkpoints by the instruction of this link :)