From 6217b48e3f92006e65f8c8ad490a32539c58a6b6 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 6 Oct 2020 20:38:47 -0600 Subject: [PATCH] Fix spsr_arch bug --- codes/models/archs/SPSR_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/archs/SPSR_arch.py b/codes/models/archs/SPSR_arch.py index b719ac0f..89ddb060 100644 --- a/codes/models/archs/SPSR_arch.py +++ b/codes/models/archs/SPSR_arch.py @@ -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)