Skip to content

Latest commit

 

History

History
47 lines (35 loc) · 1.21 KB

README.md

File metadata and controls

47 lines (35 loc) · 1.21 KB

SusPedal-Gen

This is the official repository of Learning to Generate Piano Music With Sustain Pedals

Dataset

The dataset used in this project, AIlabs.tw Pop1K7, can be found here.

Environment

  1. Install PyTorch and fast transformer:

    • torch==1.7.0 (Please install it according to your CUDA version.)

    • fast transformer :

      pip install --user pytorch-fast-transformers 
      

      or refer to the original repository

  2. Other requirements:

    pip install -r requirements.txt
    

Usage

Pre-trained models are not included in this repository

Data Preparation

  • To be completed...

Train/Inference

  • Change the MODE to train or inference on lines 36 & 37

Cite This Paper

@article{
	{SusPedal},
	author = {Ching, Joann and Yi-Hsuan, Yang},
	title = {Learning to Generate Piano Music With Sustain Pedals},
	booktitle = {arXiv preprint arXiv:2111.01216},
	year = {2021}
}