From 9a50a7966dee417f7d225a752692d85aba22e67b Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 23 Sep 2020 21:09:13 -0600 Subject: [PATCH] SiLU doesnt support inplace --- codes/models/archs/SwitchedResidualGenerator_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/SwitchedResidualGenerator_arch.py b/codes/models/archs/SwitchedResidualGenerator_arch.py index 5cb94cc4..ad841931 100644 --- a/codes/models/archs/SwitchedResidualGenerator_arch.py +++ b/codes/models/archs/SwitchedResidualGenerator_arch.py @@ -433,7 +433,7 @@ class ResBlock(nn.Module): self.downsample = ConvGnSilu(nf, nf_out, kernel_size=1, stride=stride, bias=False, activation=False, norm=True) else: self.downsample = None - self.act = SiLU(inplace=True) + self.act = SiLU() def forward(self, x): identity = x