From cd9dca99dcd58c9654947fe031fbeec7a1b1f0dd Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9D=B4=EC=83=81=EA=B7=9C/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 5 Dec 2019 09:57:48 +0900 Subject: [PATCH] [gen_golden] Fix passing wrong parameter for random_sample. (#9387) random_sample, used in generating fp32 radom numbers, accepts only 1 parameter. It fixes the python runtime error for this case. Signed-off-by: Sanggyu Lee --- tools/nnpackage_tool/gen_golden/gen_golden.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/nnpackage_tool/gen_golden/gen_golden.py b/tools/nnpackage_tool/gen_golden/gen_golden.py index b42fc7c..7abeebe 100755 --- a/tools/nnpackage_tool/gen_golden/gen_golden.py +++ b/tools/nnpackage_tool/gen_golden/gen_golden.py @@ -124,7 +124,7 @@ if __name__ == '__main__': np.array(np.random.randint(0, 255, this_shape, this_dtype))) elif input_dtypes[idx] == np.float32: input_values.append( - np.array(np.random.random_sample(this_shape, this_dtype))) + np.array(np.random.random_sample(this_shape), this_dtype)) interpreter.set_tensor(input_details[idx]['index'], input_values[idx]) # get output values by running -- 2.7.4