Skip to content

Commit 6c5de35

Browse files
committed
proposed: Refactor to simplify code
1 parent e9ecb55 commit 6c5de35

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

models/proposed.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,7 @@ def forward(self, x):
5656
branch4 = self.branch4(x)
5757
branch5 = F.interpolate(self.branch5(x), size=(x.size()[2], x.size()[3]), mode="bilinear", align_corners=False)
5858

59-
out = self.outconv(torch.cat([branch1, branch2, branch3, branch4, branch5], dim=1))
60-
return out
59+
return self.outconv(torch.cat([branch1, branch2, branch3, branch4, branch5], dim=1))
6160

6261

6362
class ResidualBlock(nn.Module):

0 commit comments

Comments
 (0)