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

More bugs

Browse Source
This commit is contained in:
James Betker 2020-07-06 22:28:07 -06:00
parent d4d4f85fc0
commit 086b2f0570
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

@ -175,6 +175,7 @@ class ConfigurableSwitchedResidualGenerator2(nn.Module):
self.heightened_final_step = heightened_final_step self.heightened_final_step = heightened_final_step
self.attentions = None self.attentions = None
self.upsample_factor = upsample_factor self.upsample_factor = upsample_factor
assert self.upsample_factor == 2 or self.upsample_factor == 4
def forward(self, x): def forward(self, x):
x = self.initial_conv(x) x = self.initial_conv(x)
@ -186,7 +187,6 @@ class ConfigurableSwitchedResidualGenerator2(nn.Module):
self.attentions.append(att) self.attentions.append(att)
x = swx + self.sw_conv(x) x = swx + self.sw_conv(x)
assert x == 2 or x == 4
x = self.upconv1(F.interpolate(x, scale_factor=2, mode="nearest")) x = self.upconv1(F.interpolate(x, scale_factor=2, mode="nearest"))
if self.upsample_factor > 2: if self.upsample_factor > 2:
x = F.interpolate(x, scale_factor=2, mode="nearest") x = F.interpolate(x, scale_factor=2, mode="nearest")