mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Get proper contiguous view for backwards compatibility

Browse Source
This commit is contained in:
James Betker 2020-06-16 14:27:16 -06:00
parent 2def96203e
commit f8b67f134b
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/discriminator_vgg_arch.py
View File

@ -57,7 +57,7 @@ class Discriminator_VGG_128(nn.Module):
fea = self.lrelu(self.bn4_0(self.conv4_0(fea)))
fea = self.lrelu(self.bn4_1(self.conv4_1(fea)))
fea = fea.view(fea.size(0), -1)
fea = fea.contiguous().view(fea.size(0), -1)
fea = self.lrelu(self.linear1(fea))
out = self.linear2(fea)
return out