Merge pull request Rayhane-mamah#23 from metaln37/master

TacoTrainingHelper bugfix

tacotron/models/helpers.py

@@ -110,15 +110,16 @@ def next_inputs(self, time, outputs, state, sample_ids, stop_token_prediction, n
 				#GTA synthesis stop
 				finished = (time + 1 >= self._lengths)
 
-			if np.random.random() <= self._ratio:
-				next_inputs = self._targets[:, time, :] #Teacher-forcing: return true frame
-			else:
-				next_inputs = outputs[:, -self._output_dim:]
+			next_inputs = tf.cond(
+				tf.equal(tf.mod(time, int(1./self._ratio)), 0), 
+				lambda: self._targets[:, time, :], 
+				lambda: outputs[:,-self._output_dim:])
+
 			#Update the finished state
 			next_state = state.replace(finished=tf.cast(tf.reshape(finished, [-1, 1]), tf.float32))
 			return (finished, next_inputs, next_state)
 
 
 def _go_frames(batch_size, output_dim):
 	'''Returns all-zero <GO> frames for a given batch size and output dimension'''
-	return tf.tile([[0.0]], [batch_size, output_dim])
+	return tf.tile([[0.0]], [batch_size, output_dim])