SiLU doesnt support inplace

2020-09-23 21:09:13 -06:00 · 2020-09-23 21:09:13 -06:00 · 9a50a7966d
commit 9a50a7966d
parent eda0eadba2
1 changed files with 1 additions and 1 deletions
--- 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