Skip to content

GitLab

Unverified Commit 22a00338 authored by Taylor Robie's avatar Taylor Robie Committed by GitHub
Browse files

fix model call (#4132)

parent ddc4226d
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
official/resnet/cifar10_main.py
View file @ 22a00338
......@@ -176,8 +176,6 @@ class Cifar10Model(resnet_model.Model):
conv_stride=1,
first_pool_size=None,
first_pool_stride=None,
second_pool_size=8,
second_pool_stride=1,
block_sizes=[num_blocks] * 3,
block_strides=[1, 2, 2],
final_size=64,
......
official/resnet/imagenet_main.py
View file @ 22a00338
......@@ -235,8 +235,6 @@ class ImagenetModel(resnet_model.Model):
conv_stride=2,
first_pool_size=3,
first_pool_stride=2,
second_pool_size=7,
second_pool_stride=1,
block_sizes=_get_block_sizes(resnet_size),
block_strides=[1, 2, 2, 2],
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