Skip to content

GitLab

"megatron/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "11f76cd3a3d60d42cd605620f070dfa881e6e795"
Unverified Commit 22a00338 authored by Taylor Robie's avatar Taylor Robie Committed by GitHub
Browse files

fix model call (#4132)

parent ddc4226d
Hide 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): ...@@ -176,8 +176,6 @@ class Cifar10Model(resnet_model.Model):
conv_stride=1, conv_stride=1,
first_pool_size=None, first_pool_size=None,
first_pool_stride=None, first_pool_stride=None,
second_pool_size=8,
second_pool_stride=1,
block_sizes=[num_blocks] * 3, block_sizes=[num_blocks] * 3,
block_strides=[1, 2, 2], block_strides=[1, 2, 2],
final_size=64, final_size=64,
......
official/resnet/imagenet_main.py
View file @ 22a00338
...@@ -235,8 +235,6 @@ class ImagenetModel(resnet_model.Model): ...@@ -235,8 +235,6 @@ class ImagenetModel(resnet_model.Model):
conv_stride=2, conv_stride=2,
first_pool_size=3, first_pool_size=3,
first_pool_stride=2, first_pool_stride=2,
second_pool_size=7,
second_pool_stride=1,
block_sizes=_get_block_sizes(resnet_size), block_sizes=_get_block_sizes(resnet_size),
block_strides=[1, 2, 2, 2], block_strides=[1, 2, 2, 2],
final_size=final_size, 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