google / prettytensor

Pretty Tensor: Fluent Networks in TensorFlow
1.24k stars 151 forks source link

conv2d 'kernel' #52

Open InhoChang opened 7 years ago

InhoChang commented 7 years ago

If i want to use conv2d kernel size 1x5 and 5x1 then how can i set the 'kernel' value?

Now i try it like below:

conv2d(kernel=[1, 5], depth=32, activation_fn = tf.nn.relu, edges='SAME', name='layer_conv1_a', batch_normalize = True). \ conv2d(kernel=[5, 1], depth=32, activation_fn = tf.nn.relu, edges='SAME', name='layer_conv1_b'). \

is this right setting for 1x5 and 5x1 conv kernel?