mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Temporary commit - noise

Browse Source
This commit is contained in:
James Betker 2020-09-09 17:12:52 -06:00
parent 00da69d450
commit e0fc5eb50c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/models/archs/SPSR_arch.py
View File

@ -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)