diff --git a/TrainingExtensions/torch/src/python/aimet_torch/experimental/v2/quantization/quantizers/float.py b/TrainingExtensions/torch/src/python/aimet_torch/experimental/v2/quantization/quantizers/float.py
index 3c84cf89044..6970f7dda95 100644
--- a/TrainingExtensions/torch/src/python/aimet_torch/experimental/v2/quantization/quantizers/float.py
+++ b/TrainingExtensions/torch/src/python/aimet_torch/experimental/v2/quantization/quantizers/float.py
@@ -212,3 +212,6 @@ def forward(self, input: torch.Tensor):
             maxval = _ieee_float_max_representable_value(exponent_bits, mantissa_bits)
 
         return fake_cast_to_ieee_float(input, maxval, exponent_bits, mantissa_bits)
+
+    def extra_repr(self):
+        return f'exponent_bits={self.exponent_bits}, mantissa_bits={self.mantissa_bits}'