Fix datalayer for text2speech (tensorflow 1.13)
@@ -257,6 +257,7 @@ class Text2SpeechDataLayer(DataLayer):
return self._iterator
def build_graph(self):
+ with tf.device('/cpu:0'):
"""Builds data reading graph."""
self._dataset = tf.data.Dataset.from_tensor_slices(self._files)
if self.params['shuffle']:
