Add comment

Xilinx · Nov 16, 2023 · 2a57c21 · 2a57c21
1 parent 712f99a
commit 2a57c21
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/brevitas/nn/quant_activation.py b/src/brevitas/nn/quant_activation.py
@@ -13,6 +13,9 @@
 from .quant_layer import QuantNonLinearActLayer as QuantNLAL
 
 
+# Starting from Torch 2.0, nn.Module init function accepts custom *args and **kwargs
+# torch.nn.Sigmoid does not provide its own init method, and the presence of *args + **kwargs
+# conflicts with the dependency injection package
 class Sigmoid(nn.Sigmoid):
 
     def __init__():