Skip to content

Commit 22a0033

Browse files
author
Taylor Robie
authored
fix model call (#4132)
1 parent ddc4226 commit 22a0033

File tree

2 files changed

+0
-4
lines changed

2 files changed

+0
-4
lines changed

official/resnet/cifar10_main.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,6 @@ def __init__(self, resnet_size, data_format=None, num_classes=_NUM_CLASSES,
176176
conv_stride=1,
177177
first_pool_size=None,
178178
first_pool_stride=None,
179-
second_pool_size=8,
180-
second_pool_stride=1,
181179
block_sizes=[num_blocks] * 3,
182180
block_strides=[1, 2, 2],
183181
final_size=64,

official/resnet/imagenet_main.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,8 +235,6 @@ def __init__(self, resnet_size, data_format=None, num_classes=_NUM_CLASSES,
235235
conv_stride=2,
236236
first_pool_size=3,
237237
first_pool_stride=2,
238-
second_pool_size=7,
239-
second_pool_stride=1,
240238
block_sizes=_get_block_sizes(resnet_size),
241239
block_strides=[1, 2, 2, 2],
242240
final_size=final_size,

0 commit comments

Comments
 (0)