deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

SiLU doesnt support inplace

Browse Source
This commit is contained in:
James Betker 2020-09-23 21:09:13 -06:00
parent eda0eadba2
commit 9a50a7966d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/archs/SwitchedResidualGenerator_arch.py
View File

@ -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) self.downsample = ConvGnSilu(nf, nf_out, kernel_size=1, stride=stride, bias=False, activation=False, norm=True)
else: else:
self.downsample = None self.downsample = None
self.act = SiLU(inplace=True) self.act = SiLU()
def forward(self, x): def forward(self, x):
identity = x identity = x