symbolic-music-diffusion
symbolic-music-diffusion copied to clipboard
encode the Lakh dataset Fail
I have converted Lakh MIDI Dataset into NoteSequences with convert_dir_to_note_sequences successfully. Then I tried to encode the Lakh dataset with MusicVAE, use scripts/generate_song_data_beam.py, but I always get prompt "TypeError: PTransform Create: Refusing to treat string as an iterable. (string='../notesequences_tfrecord')" I have checked python package version carefully to ensure that it is the same as requirement.txt, but the problem remained.
I got the same problem. I solved it by making this change to scripts/generate_song_data_beam.py:106
p |= 'tfrecord_list' >> beam.Create(FLAGS.input)
to
p |= 'tfrecord_list' >> beam.Create([FLAGS.input])