Undo SRG change

2020-09-04 17:32:16 -06:00 · 2020-09-04 17:32:16 -06:00 · 44c75f7642
commit 44c75f7642
parent 6657a406ac
1 changed files with 1 additions and 1 deletions
--- a/codes/models/archs/SwitchedResidualGenerator_arch.py
+++ b/codes/models/archs/SwitchedResidualGenerator_arch.py
@ -254,7 +254,7 @@ class ConfigurableSwitchComputer(nn.Module):
                x = x1 + rand_feature

        if self.pre_transform:
-            x = self.pre_transform(x)
+            x = self.pre_transform(*x)
        if not isinstance(x, tuple):
            x = (x,)
        xformed = [torch.utils.checkpoint.checkpoint(t, *x) for t in self.transforms]