From 44c75f76422d129afcc513094b9cb2db0cb9281f Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 4 Sep 2020 17:32:16 -0600 Subject: [PATCH] Undo SRG change --- codes/models/archs/SwitchedResidualGenerator_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/SwitchedResidualGenerator_arch.py b/codes/models/archs/SwitchedResidualGenerator_arch.py index f6878aab..aa0d740a 100644 --- 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]