From 962907c365ac7527af39afddd513ce89b1bfbb6a Mon Sep 17 00:00:00 2001 From: georgeyiasemis Date: Wed, 24 Jan 2024 15:53:41 +0100 Subject: [PATCH] Sigmnoid in conv fc --- direct/nn/conv/modulated_conv.py | 2 ++ 1 file changed, 2 insertions(+) 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: