diff --git a/generate.py b/generate.py index ecfd2bf6d..d186df7eb 100644 --- a/generate.py +++ b/generate.py @@ -264,7 +264,7 @@ def main(): tf.summary.audio('generated', decode, wavenet_params['sample_rate']) summaries = tf.summary.merge_all() summary_out = sess.run(summaries, - feed_dict={samples: np.reshape(waveform, [-1, 1])}) + feed_dict={samples: np.reshape(waveform, [1, -1])}) writer.add_summary(summary_out) # Save the result as a wav file.