Skip to content

GitLab

Commit 1f50270e authored by amirbar's avatar amirbar
Browse files

Fix issue #459

parent 2390974a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
slim/nets/resnet_v1.py
View file @ 1f50270e
......@@ -249,7 +249,7 @@ def resnet_v1_101(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_101.default_image_size = resnet_v1.default_image_size
def resnet_v1_152(inputs,
num_classes=None,
......@@ -271,7 +271,7 @@ def resnet_v1_152(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_152.default_image_size = resnet_v1.default_image_size
def resnet_v1_200(inputs,
num_classes=None,
......@@ -293,3 +293,4 @@ def resnet_v1_200(inputs,
return resnet_v1(inputs, blocks, num_classes, is_training,
global_pool=global_pool, output_stride=output_stride,
include_root_block=True, reuse=reuse, scope=scope)
resnet_v1_200.default_image_size = resnet_v1.default_image_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