Fix srg computer bug

2020-09-05 07:59:54 -06:00 · 2020-09-05 07:59:54 -06:00 · 912a4d9fea
commit 912a4d9fea
parent 0dfd8eaf3b
1 changed files with 2 additions and 0 deletions
--- a/codes/models/archs/SwitchedResidualGenerator_arch.py
+++ b/codes/models/archs/SwitchedResidualGenerator_arch.py
@ -253,6 +253,8 @@ class ConfigurableSwitchComputer(nn.Module):
            else:
                x = x1 + rand_feature

+        if not isinstance(x, tuple):
+            x = (x,)
        if self.pre_transform:
            x = self.pre_transform(*x)
        if not isinstance(x, tuple):