Commit 780ecbb3 authored by amirbar's avatar amirbar
Browse files

Fix issue #459

parent 1f50270e
......@@ -226,7 +226,7 @@ def resnet_v1_50(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_50.default_image_size = resnet_v1.default_image_size
def resnet_v1_101(inputs,
num_classes=None,
......@@ -293,4 +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
resnet_v1_200.default_image_size = resnet_v1.default_image_size
\ No newline at end of file
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