search

chrisdonahue / sheetsage

Transcribe music into lead sheets!
https://chrisdonahue.com/sheetsage
Other
297 stars 65 forks source link

How can I train the model with custom dataset? #8

Open CreadDiscans opened 1 year ago

CreadDiscans commented 1 year ago

It looks there is no training code. Can I get the training code?

shansongliu commented 1 year ago

It looks there is no training code. Can I get the training code?

Have you found the training code?

chrisdonahue commented 1 year ago

Hello! Thanks for the interest in the training code.

I did extensive refactoring and cleaning to polish my ugly research codebase for this publicly-released version. Unfortunately, I never got around to integrating the training code into this refactor, so the scope of this codebase is limited to inference of the pre-trained models.

If you're really eager to dive in and try to get my training code working, it can be found here: https://drive.google.com/file/d/1iI86B5UwoACpdrTET2os4G22DC0hQA24/view?usp=share_link

At this point, it may be easier to start from scratch to retrain the model, using the information in the paper as guidance along with this code I've shared. Unfortunately I won't be able to provide support for this code, but if you can get it working, please consider opening a PR.

manhdoan291 commented 10 months ago

@chrisdonahue how to run training and install requirements.txt ? pls, help me!!

manhdoan291 commented 10 months ago

@shansongliu @CreadDiscans @vaclavmuller Have you run the training code yet?