From bda4d2556f5f49a6999721dc9dcd5fca59e40c22 Mon Sep 17 00:00:00 2001 From: Difers <707065510@qq.com> Date: Tue, 25 Jul 2023 10:52:56 +0000 Subject: [PATCH] fix multistep error --- test/legacy_test/test_lamb_op.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/legacy_test/test_lamb_op.py b/test/legacy_test/test_lamb_op.py index f6ba331573275..33b2ab65a8372 100644 --- a/test/legacy_test/test_lamb_op.py +++ b/test/legacy_test/test_lamb_op.py @@ -340,8 +340,8 @@ def test_check_output(self): self.inputs['Beta2Pow'] = convert_float_to_uint16(beta2_pow_out) # Randomize gradient for next step - self.inputs['Grad'] = np.random.uniform(-1, 1, (102, 105)).astype( - np.float32 + self.inputs['Grad'] = convert_float_to_uint16( + np.random.uniform(-1, 1, (102, 105)).astype(np.float32) )