AI-ON / Few-Shot-Music-Generation

160 stars 32 forks source link

Few-Shot Music Generation

Few-Shot Distribution Learning for Music Generation

Project Status

Community Links

Problem description:

See Introduction section of the proposal.

Why this problem matters:

See Introduction section of the proposal.

How to measure success:

See Experiments section of the proposal.

Datasets:

See Datasets subsection of the proposal.

Relevant Work:

See Related Work section of the proposal.

Contribute

Data

Both the lyrics and freemidi data can be downloaded here. Place the raw-data directory in the home folder of the repository and make sure to unzip both .zip files in the data sub-directories.

For example, for the lyrics data, make sure in the given path the following files and directories exist:

ls Few-Shot-Music-Generation/raw-data/lyrics/
>> lyrics_data test.csv  train.csv  val.csv

Training Models

Sample run (check the different yaml files for different ways to run):

$ CONFIG=lyrics.yaml
$ MODEL=lstm_baseline.yaml
$ TASK=5shot.yaml
python -um train.train --data=config/${CONFIG} --model=config/${MODEL} --task=config/${TASK} --checkpt_dir=/tmp/fewshot/lstm_baseline

To view the tensorboard (only works for lstm_baseline.yaml model):

$ tensorboard --logdir=/tmp/fewshot

If you have any trouble running the code, please create an issue describing your problem.

Logging results

Please log all your results in this spreadsheet.