2017-02-13 72 views
2

我试图使用keras构建3D卷积图层。它工作正常,但是当我添加子采样参数时,它会崩溃。代码:Keras Convolution3D子采样错误

l_1 = Convolution3D(2, 10,10,10, 
    border_mode='same', 
    name = 'l_1', 
    activation='relu', 
    subsample = (5,5,5) 
    )(inputs) 

的错误是:

Traceback (most recent call last): 
    File "image_proc_09.py", line 244, in <module> 
    )(inputs) 
    File "/usr/local/lib/python2.7/dist-packages/keras/engine/topology.py", line 572, in __call__ 
    self.add_inbound_node(inbound_layers, node_indices, tensor_indices) 
    File "/usr/local/lib/python2.7/dist-packages/keras/engine/topology.py", line 635, in add_inbound_node 
    Node.create_node(self, inbound_layers, node_indices, tensor_indices) 
    File "/usr/local/lib/python2.7/dist-packages/keras/engine/topology.py", line 166, in create_node 
    output_tensors = to_list(outbound_layer.call(input_tensors[0], mask=input_masks[0])) 
    File "/usr/local/lib/python2.7/dist-packages/keras/layers/convolutional.py", line 1234, in call 
    filter_shape=self.W_shape) 
    File "/usr/local/lib/python2.7/dist-packages/keras/backend/theano_backend.py", line 1627, in conv3d 
    dim_ordering, volume_shape, filter_shape) 
    File "/usr/local/lib/python2.7/dist-packages/keras/backend/theano_backend.py", line 1686, in _old_theano_conv3d 
    assert(strides == (1, 1, 1)) 
AssertionError 

我使用theano 0.8.2。

感谢

回答