ENH: filter size instead of kernel size, in line with num filters

awni · Mar 22, 2017 · 5380f9c · 5380f9c
1 parent 1f90346
commit 5380f9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/network.py b/network.py
@@ -15,10 +15,10 @@ def init_inference(self, config):
 
  for layer in config['conv_layers']:
  num_filters = layer['num_filters']
- kernel_size = layer['kernel_size']
+ filter_size = layer['filter_size']
  stride = layer['stride']
  acts = tf.contrib.layers.convolution2d(acts, num_outputs=num_filters,
- kernel_size=[kernel_size, 1],
+ kernel_size=[filter_size, 1],
  stride=stride)
 
  # Activations should emerge from the convolution with shape