Skip to content

Releases: gwinndr/MusicTransformer-Pytorch

Version 1.0

24 Mar 00:56
a3b3c90
Compare
Choose a tag to compare

Initial release. Utilizes Pytorch train a MusicTransformer model as specified in Huang et al., 2018. The model achieves similar performance to the results from the paper.

Included features:

  • Maestro v2 midi pre-processing
  • Training with Relational Position Representations using the built-in Pytorch Transformer
    • Can build any transformer architecture desired
  • Model evaluation
  • Midi generation
  • Tensorboard reporting (optional)

This version relies on the midi processor provided here:
https://github.com/jason9693/midi-neural-processor

Future releases aim to remove this dependency.