fix unittests

Signed-off-by: Shiyang Chen <[email protected]>

Author: sychen52

tests/unittest/trt/quantization/test_mode.py

@@ -44,7 +44,7 @@ def test_any(self):
 
     def test_count(self):
         # Make sure the COUNT value is as expected - change that test if you add a new flag.
-        self.assertEqual(QuantMode.COUNT.value, 1 << 17)
+        self.assertEqual(QuantMode.COUNT.value, 1 << 18)
 
     def test_from_description(self):
         # Test weight only.