in scripts/modules/residual.py [0:0]
def forward(self, x):
if hasattr(self, "proj_conv"):
residual = self.proj_conv(x)
residual = self.proj_bn(residual)
else:
residual = x
x = self.convs(x) + residual
if self.convs.bn1.activation == "leaky_relu":
return functional.leaky_relu(
x, negative_slope=self.convs.bn1.activation_param, inplace=True
)
elif self.convs.bn1.activation == "elu":
return functional.elu(
x, alpha=self.convs.bn1.activation_param, inplace=True
)
elif self.convs.bn1.activation == "identity":
return x