fix syntax

Xilinx · Jul 17, 2023 · a1dca58 · a1dca58
1 parent 680c8fd
commit a1dca58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py
@@ -58,8 +58,8 @@
     nn.AdaptiveAvgPool2d,
     nn.AdaptiveAvgPool3d,
     nn.Identity,
-    torch.nn.ReLU, 
-    torch.nn.LeakyReLU)
+    nn.ReLU, 
+    nn.LeakyReLU)
 
 _scale_invariant_op = (torch.mul, operator.mul, operator.imul, operator.__mul__, operator.__imul__)