Mod: Changes one line to fix the Deconvolution3D's error (#73)

This commit is contained in:
Tron
2017-04-21 15:48:51 -07:00
committed by Michael Oliver
parent fe227a0803
commit a7520c8f52
+1 -1
View File
@@ -213,7 +213,7 @@ class Deconvolution3D(Convolution3D):
if self.data_format == 'channels_first':
output += K.reshape(self.bias, (1, self.filters, 1, 1, 1))
elif self.data_format == 'channels_last':
output += K.reshape(self.use_bias, (1, 1, 1, 1, self.filters))
output += K.reshape(self.bias, (1, 1, 1, 1, self.filters))
else:
raise ValueError('Invalid data_format: ', self.data_format)
output = self.activation(output)