From 0716f4e012d1e3f7543bffa8aecc97ce9c903e17 Mon Sep 17 00:00:00 2001 From: Roman Grundkiewicz Date: Mon, 24 Jan 2022 18:31:55 +0000 Subject: Use lower precision --- tests/training/features/quantized-model/test_quantmodel_with_bias.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/training/features/quantized-model/test_quantmodel_with_bias.sh b/tests/training/features/quantized-model/test_quantmodel_with_bias.sh index 8dee56b..d57a1c1 100644 --- a/tests/training/features/quantized-model/test_quantmodel_with_bias.sh +++ b/tests/training/features/quantized-model/test_quantmodel_with_bias.sh @@ -34,7 +34,7 @@ test -e $PREFIX.log # Compare the current output with the expected output cat $PREFIX.log | $MRT_TOOLS/extract-costs.sh > $PREFIX.out -$MRT_TOOLS/diff-nums.py $PREFIX.out $PREFIX.expected -o $PREFIX.diff +$MRT_TOOLS/diff-nums.py -p 0.005 $PREFIX.out $PREFIX.expected -o $PREFIX.diff # make sure that the resulting model has no more than 256 different values (i.e. quantized) $MRT_TOOLS/check-model-unique-vals.py train/model.npz -b 8 --with_bias -- cgit v1.2.3