From e0fc5eb50c3c767dded9e90a0694a29f1057ab91 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 9 Sep 2020 17:12:52 -0600 Subject: [PATCH] Temporary commit - noise --- codes/models/archs/SPSR_arch.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/codes/models/archs/SPSR_arch.py b/codes/models/archs/SPSR_arch.py index 865c374c..29cae50d 100644 --- a/codes/models/archs/SPSR_arch.py +++ b/codes/models/archs/SPSR_arch.py @@ -451,13 +451,13 @@ class SwitchedSpsrWithRef2(nn.Module): x1, a1 = self.sw1(x1, True, identity=x) x2 = x1 - x2 = self.noise_ref_join(x2, torch.randn_like(x2)) + #x2 = self.noise_ref_join(x2, torch.randn_like(x2)) #x2 = self.ref_join2(x2, ref) x2, a2 = self.sw2(x2, True, identity=x1) x_grad = self.grad_conv(x_grad) x_grad_identity = x_grad - x_grad = self.noise_ref_join_grad(x_grad, torch.randn_like(x_grad)) + #x_grad = self.noise_ref_join_grad(x_grad, torch.randn_like(x_grad)) #x_grad = self.ref_join3(x_grad, ref) x_grad = self.grad_ref_join(x_grad, x1) x_grad, a3 = self.sw_grad(x_grad, True, identity=x_grad_identity) @@ -467,7 +467,7 @@ class SwitchedSpsrWithRef2(nn.Module): x_grad_out = self.grad_branch_output_conv(x_grad_out) x_out = x2 - x_out = self.noise_ref_join_conjoin(x_out, torch.randn_like(x_out)) + #x_out = self.noise_ref_join_conjoin(x_out, torch.randn_like(x_out)) #x_out = self.ref_join4(x_out, ref) x_out = self.conjoin_ref_join(x_out, x_grad) x_out, a4 = self.conjoin_sw(x_out, True, identity=x2)