Skip to content

Commit 330e1ea

Browse files
committed
proposed: Refactor to simplify code
1 parent 4436683 commit 330e1ea

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
@@ -125,8 +125,7 @@ def forward(self, x):
125125
encode1 = self.encode1(self.encode0(x))
126126
encode2 = self.encode2(encode1)
127127
encode3 = self.encode3(encode2)
128-
encode4 = self.encode4(encode3)
129-
encode_end = self.encode_end(encode4)
128+
encode_end = self.encode_end(self.encode4(encode3))
130129

131130
# Decoder
132131
out = self.decode3(torch.cat([self.upconv3(encode_end), encode3], dim=1))

0 commit comments

Comments
 (0)