From f39a8ed3f1e529c5d9137cdd344645be108eee92 Mon Sep 17 00:00:00 2001 From: Leo Crestel Mac Date: Thu, 5 Apr 2018 13:40:33 +0200 Subject: [PATCH] Fixed splitting in sample_size blocks to use the last block --- samplernn/audio_reader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/samplernn/audio_reader.py b/samplernn/audio_reader.py index 2f99aa7..5539fae 100644 --- a/samplernn/audio_reader.py +++ b/samplernn/audio_reader.py @@ -97,7 +97,7 @@ def thread_main(self, sess): np.full((pad_elements, 1), 0.0, dtype='float32')], axis=0) if self.sample_size: - while len(audio) > self.sample_size: + while len(audio) >= self.sample_size: piece = audio[:self.sample_size, :] sess.run(self.enqueue, feed_dict={self.sample_placeholder: piece})