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

Fix srg computer bug

Browse Source
This commit is contained in:
James Betker 2020-09-05 07:59:54 -06:00
parent 0dfd8eaf3b
commit 912a4d9fea
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -253,6 +253,8 @@ class ConfigurableSwitchComputer(nn.Module):
else: else:
x = x1 + rand_feature x = x1 + rand_feature
if not isinstance(x, tuple):
x = (x,)
if self.pre_transform: if self.pre_transform:
x = self.pre_transform(*x) x = self.pre_transform(*x)
if not isinstance(x, tuple): if not isinstance(x, tuple):