Skip to content

Commit 87e6781

Browse files
authored
Update audio_spec.py
1 parent 5d039b7 commit 87e6781

File tree

1 file changed

+2
-1
lines changed
  • tensorflow_examples/lite/model_maker/core/task/model_spec

1 file changed

+2
-1
lines changed

tensorflow_examples/lite/model_maker/core/task/model_spec/audio_spec.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,8 @@ def _split(self, wav, label):
283283
])
284284
def _preprocess(self, x, label):
285285
"""Preprocess the dataset to extract the spectrum."""
286-
# Add small Gaussian noise to the input x to solve the potential "nan" problem of the preprocess_model
286+
# Add small Gaussian noise to the input x
287+
# to solve the potential "nan" problem of the preprocess_model
287288
x = x + 1e-05 * tf.random.normal(x.shape)
288289
# x has shape (1, EXPECTED_WAVEFORM_LENGTH)
289290
spectrum = self._preprocess_model(x)

0 commit comments

Comments
 (0)