Commit e5b9c984 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

fixed bug: unpool accidentally named upconv

parent 13bea40a
......@@ -503,7 +503,7 @@ class DecoderBlock3D(ConvolutionalBlock3D):
# ATTENTION: As of this code version, only "upconv" works! Debugging is ongoing for upconv and upsample!
# It seems that errors are generated by variable filter sizes and the unorthodox input sizes 91x109x91.
if self.up_mode == 'upconv':
if self.up_mode == 'unpool':
upsampling = self.up(X, pool_indices, output_size=Y_encoder.size())
elif self.up_mode == 'upsample':
upsampling = self.up(X)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment