[FIX] prevent random_uniform generate from x to x

This commit is contained in:
Yi-Hua Chiu 2019-12-03 18:40:52 +08:00
parent 9b3e4aa9d3
commit ec0ee65eb0

View File

@ -90,7 +90,7 @@ def augment_sparse_warp(spectrogram, time_warping_para=80, interpolation_order=2
# to protect short audio
time_warping_para = tf.math.minimum(
time_warping_para, tf.math.floordiv(tau, 2))
time_warping_para, tf.math.subtract(tf.math.floordiv(tau, 2), 1))
mid_tau = tf.math.floordiv(tau, 2)
mid_freq = tf.math.floordiv(freq_size, 2)