blakelemoine/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

More recurrence fixes for chainedgen

Browse Source
This commit is contained in:
James Betker 2020-10-17 08:35:46 -06:00
parent cf8118a85b
commit 6141aa1110
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
codes/models/archs/ChainedEmbeddingGen.py
View File

@ -81,9 +81,10 @@ class ChainedEmbeddingGenWithStructure(nn.Module):
self.grad_extract = ImageGradientNoPadding() self.grad_extract = ImageGradientNoPadding()
self.upsample = FinalUpsampleBlock2x(64) self.upsample = FinalUpsampleBlock2x(64)
def forward(self, x): def forward(self, x, recurrent=None):
emb = checkpoint(self.spine, x) emb = checkpoint(self.spine, x)
if self.recurrent: if self.recurrent:
fea = torch.cat([x,recurrent], dim=1)
fea = self.initial_conv_rec(x) fea = self.initial_conv_rec(x)
else: else:
fea = self.initial_conv(x) fea = self.initial_conv(x)