fix the core dump bug of DS2's training in docker

7 years ago · 37a29bf181
parent 3f608b3b81
commit 37a29bf181
1 changed files with 2 additions and 1 deletions
--- a/data_utils/featurizer/audio_featurizer.py
+++ b/data_utils/featurizer/audio_featurizer.py
@ -141,7 +141,8 @@ class AudioFeaturizer(object):
        # window weighting, squared Fast Fourier Transform (fft), scaling
        weighting = np.hanning(window_size)[:, None]
        fft = np.fft.rfft(windows * weighting, axis=0)
-        fft = np.absolute(fft)**2
+        fft = np.absolute(fft)
+        fft = fft**2
        scale = np.sum(weighting**2) * sample_rate
        fft[1:-1, :] *= (2.0 / scale)
        fft[(0, -1), :] /= scale