Remove keepdims param

Author: maxisme

nets/resnet_v1_101.py

@@ -14,6 +14,6 @@ def endpoints(image, is_training):
         _, endpoints = resnet_v1_101(image, num_classes=None, is_training=is_training, global_pool=True)
 
     endpoints['model_output'] = endpoints['global_pool'] = tf.reduce_mean(
-        endpoints['resnet_v1_101/block4'], [1, 2], name='pool5', keepdims=False)
+        endpoints['resnet_v1_101/block4'], [1, 2], name='pool5')
 
     return endpoints, 'resnet_v1_101'

nets/resnet_v1_50.py

@@ -14,6 +14,6 @@ def endpoints(image, is_training):
         _, endpoints = resnet_v1_50(image, num_classes=None, is_training=is_training, global_pool=True)
 
     endpoints['model_output'] = endpoints['global_pool'] = tf.reduce_mean(
-        endpoints['resnet_v1_50/block4'], [1, 2], name='pool5', keepdims=False)
+        endpoints['resnet_v1_50/block4'], [1, 2], name='pool5')
 
     return endpoints, 'resnet_v1_50'