fix

6 months ago · ba19c2eee0
parent 1a9a5fab15
commit ba19c2eee0
1 changed files with 1 additions and 1 deletions
--- a/src/models/sdynet.py
+++ b/src/models/sdynet.py
@ -92,7 +92,7 @@ class SDYNetx2(nn.Module):
            output_1 += y
                       
        output_1 /= 4*3
-        output_1 = output_1.view(b, c, h, w)
+        output_1 = output_1.view(b*c, 1, h, w)
        x = output_1

        output_2 = torch.zeros([b*c, 1, h*self.scale, w*self.scale], dtype=x.dtype, device=x.device)