"git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "d2fbf2de0c1eed57643c3ef649b23e2694579b55"
Commit ddc4226d authored by irmo's avatar irmo Committed by Taylor Robie
Browse files

Remove useless fields in resnet_model (#4110)

Refer to commit "use reduce_mean instead of Average Pooling"(d5663b3c), "second_pool_size" and "second_pool_stride" no longer needed in resnet_model
parent 04ef5c2f
......@@ -353,7 +353,7 @@ class Model(object):
def __init__(self, resnet_size, bottleneck, num_classes, num_filters,
kernel_size,
conv_stride, first_pool_size, first_pool_stride,
second_pool_size, second_pool_stride, block_sizes, block_strides,
block_sizes, block_strides,
final_size, version=DEFAULT_VERSION, data_format=None,
dtype=DEFAULT_DTYPE):
"""Creates a model for classifying an image.
......@@ -371,8 +371,6 @@ class Model(object):
If none, the first pooling layer is skipped.
first_pool_stride: stride size for the first pooling layer. Not used
if first_pool_size is None.
second_pool_size: Pool size to be used for the second pooling layer.
second_pool_stride: stride size for the final pooling layer
block_sizes: A list containing n values, where n is the number of sets of
block layers desired. Each value should be the number of blocks in the
i-th set.
......@@ -422,8 +420,6 @@ class Model(object):
self.conv_stride = conv_stride
self.first_pool_size = first_pool_size
self.first_pool_stride = first_pool_stride
self.second_pool_size = second_pool_size
self.second_pool_stride = second_pool_stride
self.block_sizes = block_sizes
self.block_strides = block_strides
self.final_size = final_size
......
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