Skip to content

GitLab

Commit 325609e2 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Remove unused _max_pool method (#2186)

parent 70b894bd
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
tutorials/image/cifar10_estimator/model_base.py
View file @ 325609e2
...@@ -211,13 +211,6 @@ class ResNet(object): ...@@ -211,13 +211,6 @@ class ResNet(object):
tf.logging.info('image after unit %s: %s', name_scope, x.get_shape()) tf.logging.info('image after unit %s: %s', name_scope, x.get_shape())
return x return x
def _max_pool(self, x, pool_size, stride):
with tf.name_scope('max_pool') as name_scope:
x = tf.layers.max_pooling2d(
x, pool_size, stride, padding='SAME', data_format=self._data_format)
tf.logging.info('image after unit %s: %s', name_scope, x.get_shape())
return x
def _global_avg_pool(self, x): def _global_avg_pool(self, x):
with tf.name_scope('global_avg_pool') as name_scope: with tf.name_scope('global_avg_pool') as name_scope:
assert x.get_shape().ndims == 4 assert x.get_shape().ndims == 4
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment