Update models.py

Author: bmycheez

Burst/models.py

@@ -197,18 +197,18 @@ def forward(self, x):
             RB_outs.append(outR)
         outR = torch.cat(RB_outs, 1)
         outR = self.GFF_R(outR)
-        outR += out
+        
         AB_outs = []
         for i in range(self.A):
             outA = self.AB[i](out)
             AB_outs.append(outA)
         outA = torch.cat(AB_outs, 1)
         outA = self.GFF_A(outA)
-        outA += out
+        
         # outR *= self.softmax(outA)
         out = torch.cat([outR, outA], 1)
         out = self.relu2(self.conv_f(self.res2(out)))
-        out *= 0.2
+        
         out += x
 
         return out