We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4436683 commit 330e1eaCopy full SHA for 330e1ea
models/proposed.py
@@ -125,8 +125,7 @@ def forward(self, x):
125
encode1 = self.encode1(self.encode0(x))
126
encode2 = self.encode2(encode1)
127
encode3 = self.encode3(encode2)
128
- encode4 = self.encode4(encode3)
129
- encode_end = self.encode_end(encode4)
+ encode_end = self.encode_end(self.encode4(encode3))
130
131
# Decoder
132
out = self.decode3(torch.cat([self.upconv3(encode_end), encode3], dim=1))
0 commit comments