AI-ON / Few-Shot-Music-Generation

160 stars 32 forks source link

Finish EpisodeSampler #7

Closed sagelywizard closed 6 years ago

sagelywizard commented 6 years ago

This PR completes the main functionality of the EpisodeSampler. Specifically, this PR adds functionality to do the following:

It's easiest to use this sampler with episode.load_sampler_from_config(path). This function expects a path to a yaml config file. e.g.

query_size: 4
support_size: 4
batch_size: 5
max_len: 500
dataset: lyrics
dataset_path: /Users/username/lyrics_data
split: train

This PR is a squashed version of https://github.com/AI-ON/Few-Shot-Music-Generation/pull/5