From 31b42c2ef0bb17bf6751302bb65f4eb18bfd6c13 Mon Sep 17 00:00:00 2001 From: Magenta Team Date: Wed, 12 Aug 2020 10:25:07 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 326259440 --- ddsp/losses_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ddsp/losses_test.py b/ddsp/losses_test.py index 54a8893a..dd4a2302 100644 --- a/ddsp/losses_test.py +++ b/ddsp/losses_test.py @@ -33,8 +33,8 @@ def test_output_shape_is_correct(self): loudness_weight=1.0, ) - input_audio = tf.random.uniform((3, 8000), dtype=tf.float32) - target_audio = tf.random.uniform((3, 8000), dtype=tf.float32) + input_audio = tf.ones((3, 8000), dtype=tf.float32) + target_audio = tf.ones((3, 8000), dtype=tf.float32) loss = loss_obj(input_audio, target_audio) @@ -47,8 +47,8 @@ class PretrainedCREPEEmbeddingLossTest(tf.test.TestCase): def test_output_shape_is_correct(self): loss_obj = losses.PretrainedCREPEEmbeddingLoss() - input_audio = tf.random.uniform((3, 16000), dtype=tf.float32) - target_audio = tf.random.uniform((3, 16000), dtype=tf.float32) + input_audio = tf.ones((3, 16000), dtype=tf.float32) + target_audio = tf.ones((3, 16000), dtype=tf.float32) loss = loss_obj(input_audio, target_audio)