🔧 Fix 🐛 #109, don't call _build() if you want save into pb.

TensorSpeech · Jul 11, 2020 · e8a0d85 · e8a0d85
1 parent 84c0041
commit e8a0d85
Show file tree

Hide file tree

Showing 2 changed files with 271 additions and 1 deletion.
diff --git a/notebooks/fastspeech2_inference.ipynb b/notebooks/fastspeech2_inference.ipynb
diff --git a/tensorflow_tts/models/fastspeech2.py b/tensorflow_tts/models/fastspeech2.py
@@ -227,7 +227,7 @@ def _inference(
         duration_outputs = self.duration_predictor(
             [last_encoder_hidden_states, speaker_ids, attention_mask]
         )  # [batch_size, length]
-        duration_outputs = tf.math.exp(duration_outputs) - 1.0
+        duration_outputs = tf.nn.relu(tf.math.exp(duration_outputs) - 1.0)
         duration_outputs = tf.cast(
             tf.math.round(duration_outputs * speed_ratios), tf.int32
         )