🔧 Fix #70

dathudeptrai · dathudeptrai · commit 7258ba0b313c · 2020-06-30T12:15:08.000+07:00
diff --git a/examples/tacotron2/decode_tacotron2.py b/examples/tacotron2/decode_tacotron2.py
@@ -113,7 +113,7 @@ def main():
         mel_outputs, post_mel_outputs, stop_outputs, alignment_historys = tacotron2.inference(
             charactor,
             char_length,
-            speaker_ids=tf.zeros(shape=[tf.shape(charactor)[0]]),
+            speaker_ids=tf.zeros(shape=[tf.shape(charactor)[0]], dtype=tf.int32),
         )
 
         # convert to numpy

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ def main():`
`113`	`113`	`mel_outputs, post_mel_outputs, stop_outputs, alignment_historys = tacotron2.inference(`
`114`	`114`	`charactor,`
`115`	`115`	`char_length,`
`116`		`- speaker_ids=tf.zeros(shape=[tf.shape(charactor)[0]]),`
	`116`	`+ speaker_ids=tf.zeros(shape=[tf.shape(charactor)[0]], dtype=tf.int32),`
`117`	`117`	`)`
`118`	`118`
`119`	`119`	`# convert to numpy`