Jelajahi Sumber

[slim] Fix a scope name typo in Inception_v1.py

The average pool is used but named 'MaxPool_0a_7x7'.
Change it to 'AvgPool_0a_7x7'.
Jun Kim 8 tahun lalu
induk
melakukan
924d0446a3
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      slim/nets/inception_v1.py

+ 1 - 1
slim/nets/inception_v1.py

@@ -289,7 +289,7 @@ def inception_v1(inputs,
                         is_training=is_training):
       net, end_points = inception_v1_base(inputs, scope=scope)
       with tf.variable_scope('Logits'):
-        net = slim.avg_pool2d(net, [7, 7], stride=1, scope='MaxPool_0a_7x7')
+        net = slim.avg_pool2d(net, [7, 7], stride=1, scope='AvgPool_0a_7x7')
         net = slim.dropout(net,
                            dropout_keep_prob, scope='Dropout_0b')
         logits = slim.conv2d(net, num_classes, [1, 1], activation_fn=None,