Skip to content

Commit 6487d54

Browse files
committed
Fix 'keep_dims is deprecated, use keepdims instead' warning.
1 parent fdfaf56 commit 6487d54

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

nets/resnet_v1_101.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,6 @@ def endpoints(image, is_training):
1414
_, endpoints = resnet_v1_101(image, num_classes=None, is_training=is_training, global_pool=True)
1515

1616
endpoints['model_output'] = endpoints['global_pool'] = tf.reduce_mean(
17-
endpoints['resnet_v1_101/block4'], [1, 2], name='pool5', keep_dims=False)
17+
endpoints['resnet_v1_101/block4'], [1, 2], name='pool5', keepdims=False)
1818

1919
return endpoints, 'resnet_v1_101'

nets/resnet_v1_50.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,6 @@ def endpoints(image, is_training):
1414
_, endpoints = resnet_v1_50(image, num_classes=None, is_training=is_training, global_pool=True)
1515

1616
endpoints['model_output'] = endpoints['global_pool'] = tf.reduce_mean(
17-
endpoints['resnet_v1_50/block4'], [1, 2], name='pool5', keep_dims=False)
17+
endpoints['resnet_v1_50/block4'], [1, 2], name='pool5', keepdims=False)
1818

1919
return endpoints, 'resnet_v1_50'

0 commit comments

Comments
 (0)