coorect output size for reshape and roi pooling
@@ -2133,7 +2133,7 @@ class Reshape(Layer):
@property
def output_size(self):
if self._output_size is None:
- self._output_size = (self.config['height'], self.config['width'], self.config['depth'])
+ self._output_size = (int(self.config['height']), int(self.config['width']), int(self.config['depth']))
return self._output_size
@property
@@ -2423,8 +2423,8 @@ class ROIPooling(Layer):
@property
def output_size(self):
if self._output_size is None:
- self._output_size = (self.config['output_width'], self.config['output_height'],
- self.src_layers[0].output_size[1])
+ self._output_size = (int(self.config['output_width']), int(self.config['output_height']),
+ int(self.src_layers[0].output_size[1]))
return self._output_size
@property
