Add missing simd param in qat tutorial

sony · Nov 1, 2023 · 64fb684 · 64fb684
1 parent 143f090
commit 64fb684
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/tutorials/notebooks/example_keras_qat.ipynb b/tutorials/notebooks/example_keras_qat.ipynb
@@ -328,4 +328,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}
diff --git a/tutorials/notebooks/example_keras_qat.py b/tutorials/notebooks/example_keras_qat.py
@@ -55,7 +55,8 @@ def get_tpc():
         quantization_preserving=False,
         fixed_scale=1.0,
         fixed_zero_point=0,
-        weights_multiplier_nbits=0)
+        weights_multiplier_nbits=0,
+        simd_size=None)
 
     default_configuration_options = tp.QuantizationConfigOptions([default_config])
     tp_model = tp.TargetPlatformModel(default_configuration_options)

diff --git a/tutorials/notebooks/example_pytorch_qat.py b/tutorials/notebooks/example_pytorch_qat.py
@@ -173,7 +173,8 @@ def get_tpc():
             quantization_preserving=False,
             fixed_scale=1.0,
             fixed_zero_point=0,
-            weights_multiplier_nbits=0)
+            weights_multiplier_nbits=0,
+            simd_size=None)
 
         default_configuration_options = tp.QuantizationConfigOptions([default_config])
         tp_model = tp.TargetPlatformModel(default_configuration_options)
-Original file line number
+Diff line change
@@ Expand Up / @@ -328,4 +328,4 @@ @@
      },
      "nbformat": 4,
      "nbformat_minor": 5
-    }
+    }