a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Fix spsr_arch bug

Browse Source
This commit is contained in:
James Betker 2020-10-06 20:38:47 -06:00
parent 4290918359
commit 6217b48e3f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -542,7 +542,7 @@ class Spsr7(nn.Module):
x_grad, grad_fea_std = checkpoint(self.grad_ref_join, x_grad, x1)
x_grad, a3 = self.sw_grad(x_grad, True, identity=x_grad_identity, att_in=(x_grad, ref_embedding))
x_grad = checkpoint(self.grad_lr_conv, x_grad)
x_grad = checkpoint(grad_lr_conv2, x_grad)
x_grad = checkpoint(self.grad_lr_conv2, x_grad)
x_grad_out = checkpoint(self.upsample_grad, x_grad)
x_grad_out = checkpoint(self.grad_branch_output_conv, x_grad_out)