Added bias in upsample and downsampling to match paper
@@ -165,13 +165,13 @@ class VNetBlock(TrainableLayer):
n_output_chns=self.n_output_chns,
w_initializer=self.initializers['w'],
w_regularizer=self.regularizers['w'],
- kernel_size=2, stride=2)(res_flow)
+ kernel_size=2, stride=2, with_bias=True)(res_flow)
elif self.func == 'UPSAMPLE':
main_flow = DeconvLayer(name='upsample',
n_output_chns=self.n_output_chns,
w_initializer=self.initializers['w'],
w_regularizer=self.regularizers['w'],
- kernel_size=2, stride=2)(res_flow)
+ kernel_size=2, stride=2, with_bias=True)(res_flow)
elif self.func == 'SAME':
main_flow = ConvLayer(name='conv_1x1x1',
n_output_chns=self.n_output_chns,
