fix docs of ste quantizer

sony · Oct 14, 2024 · 4e8cd3a · 4e8cd3a
1 parent a0366a7
commit 4e8cd3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ression_toolkit/trainable_infrastructure/keras/activation_quantizers/ste/symmetric_ste.py b/...ression_toolkit/trainable_infrastructure/keras/activation_quantizers/ste/symmetric_ste.py
@@ -42,7 +42,7 @@ class STESymmetricActivationTrainableQuantizer(BaseKerasActivationTrainableQuant
 
     def __init__(self, quantization_config: TrainableQuantizerActivationConfig, freeze_quant_params: bool = False):
         """
-        Initialize a STEActivationQATQuantizer object with parameters to use
+        Initialize a STESymmetricActivationTrainableQuantizer object with parameters to use
         for the quantization.
 
         Args: