Update resnet_v1.py

snxmv · Jul 5, 2017 · 171b844 · 171b844
1 parent 7b01de7
commit 171b844
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/nets/resnet_v1.py b/lib/nets/resnet_v1.py
@@ -103,21 +103,21 @@ def build_network(self, sess, is_training=True):
     if self._num_layers == 50:
       blocks = [resnet_v1_block('block1', base_depth=64, num_units=3, stride=2),
                 resnet_v1_block('block2', base_depth=128, num_units=4, stride=2),
-                     # use stride 1 for the last conv4 layer
+                # use stride 1 for the last conv4 layer
                 resnet_v1_block('block3', base_depth=256, num_units=6, stride=1),
                 resnet_v1_block('block4', base_depth=512, num_units=3, stride=1)]
 
     elif self._num_layers == 101:
       blocks = [resnet_v1_block('block1', base_depth=64, num_units=3, stride=2),
                 resnet_v1_block('block2', base_depth=128, num_units=4, stride=2),
-                     # use stride 1 for the last conv4 layer
+                # use stride 1 for the last conv4 layer
                 resnet_v1_block('block3', base_depth=256, num_units=23, stride=1),
                 resnet_v1_block('block4', base_depth=512, num_units=3, stride=1)]
 
     elif self._num_layers == 152:
       blocks = [resnet_v1_block('block1', base_depth=64, num_units=3, stride=2),
                 resnet_v1_block('block2', base_depth=128, num_units=8, stride=2),
-                     # use stride 1 for the last conv4 layer
+                # use stride 1 for the last conv4 layer
                 resnet_v1_block('block3', base_depth=256, num_units=36, stride=1),
                 resnet_v1_block('block4', base_depth=512, num_units=3, stride=1)]