fix

Author: Giuseppe5

src/brevitas/config.py

@@ -25,4 +25,4 @@ def env_to_bool(name, default):
 _FULL_STATE_DICT = False
 _IS_INSIDE_QUANT_LAYER = None
 _ONGOING_EXPORT = None
-_RETROCOMPATIBLE_SCALING = False
+_RETROCOMPATIBLE_SCALING = False

src/brevitas/core/restrict_val.py

@@ -142,7 +142,7 @@ def restrict_init_module(self):
 
     def restrict_init_inplace_module(self):
         return Identity()
-    
+
     def retrocompatibility_op(self, x):
         return x
 
@@ -170,7 +170,7 @@ def restrict_init_module(self):
 
     def restrict_init_inplace_module(self):
         return InplaceLogTwo()
-    
+
     def retrocompatibility_op(self, x):
         return self.power_of_two(x)
 

src/brevitas/core/scaling/standalone.py

@@ -150,7 +150,7 @@ def __init__(
 
     @brevitas.jit.script_method
     def forward(self, placeholder: Tensor, threshold: torch.Tensor) -> Tensor:
-        value = abs_binary_sign_grad(self.restrict_clamp_scaling(self.value / threshold))
+        value = abs_binary_sign_grad(self.restrict_clamp_scaling(self.value) / threshold)
         return value
 
     def _load_from_state_dict(
@@ -214,7 +214,6 @@ def forward(self, ignored: torch.Tensor, threshold: torch.Tensor) -> torch.Tenso
             if self.local_loss_mode:
                 return self.stats_scaling_impl(stats, threshold)
             inplace_tensor_mul(self.value.detach(), stats)
-            print(self.restrict_inplace_preprocess)
             value = self.restrict_inplace_preprocess(self.value / threshold)
             value = abs_binary_sign_grad(self.stats_scaling_impl.restrict_clamp_scaling(self.value))
             self.init_done = True