diff --git a/direct/nn/conv/modulated_conv.py b/direct/nn/conv/modulated_conv.py index 31ab712b..1e987b6c 100644 --- 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: