Sigmnoid in conv fc

NKI-AI · Jan 24, 2024 · 962907c · 962907c
1 parent 866bc5b
commit 962907c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/direct/nn/conv/modulated_conv.py b/direct/nn/conv/modulated_conv.py
@@ -113,6 +113,7 @@ def __init__(
                 nn.Linear(aux_in_features, fc_hidden_features, bias=fc_bias),
                 nn.PReLU(),
                 nn.Linear(fc_hidden_features, in_channels * out_channels, bias=fc_bias),
+                nn.Sigmoid(),
             )
 
         if bias == ModConv2dBias.PARAM:
@@ -255,6 +256,7 @@ def __init__(
                 nn.Linear(aux_in_features, fc_hidden_features, bias=fc_bias),
                 nn.PReLU(),
                 nn.Linear(fc_hidden_features, in_channels * out_channels, bias=fc_bias),
+                nn.Sigmoid(),
             )
 
         if bias == ModConv2dBias.PARAM: