JIT fix

Xilinx · Oct 8, 2024 · 1f71497 · 1f71497
1 parent 0297e87
commit 1f71497
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/core/scaling/standalone.py b/src/brevitas/core/scaling/standalone.py
@@ -383,7 +383,7 @@ def training_forward(self, stats_input: Tensor, threshold: torch.Tensor) -> Tens
             return abs_binary_sign_grad(self.clamp_scaling(self.restrict_scaling(value)))
         else:
             threshold = self.restrict_preprocess(threshold)
-            value = self.restrict_scaling_impl.combine_stats_threshold(value, threshold)
+            value = self.restrict_scaling_impl.combine_stats_threshold(self.value, threshold)
             return abs_binary_sign_grad(self.clamp_scaling(self.restrict_scaling(value)))
 
     @brevitas.jit.script_method