fix typo

wy1iu · May 3, 2018 · b89730f · b89730f
1 parent 47e452c
commit b89730f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/baseline/baseline_cnn.py b/baseline/baseline_cnn.py
@@ -151,7 +151,7 @@ def build(self, rgb, n_class, is_training):
             feat = self._conv_layer(feat, ksize, n_out, is_training, name="conv3_"+str(i), bn=True, relu=True, pad='SAME', norm='none', reg=True, orth=False)
         feat = self._max_pool(feat, 'pool3')      
 
-        self.fc6 = self._conv_layer(feat, 4, 256, is_training, "fc6", bn=True, relu=False, pad='VALID', norm='none', reg=True, orth=False)        
+        self.fc6 = self._conv_layer(feat, 4, 256, is_training, "fc6", bn=False, relu=False, pad='VALID', norm='none', reg=True, orth=False)        
 
         self.score = self._conv_layer(self.fc6, 1, n_class, is_training, "score", bn=False, relu=False, pad='VALID', norm='none', reg=True, orth=False, w_norm='none')
 

diff --git a/spherenet.py b/spherenet.py
@@ -151,7 +151,7 @@ def build(self, rgb, n_class, is_training):
             feat = self._conv_layer(feat, ksize, n_out, is_training, name="conv3_"+str(i), bn=True, relu=True, pad='SAME', norm='cosine', reg=False, orth=True)
         feat = self._max_pool(feat, 'pool3')      
 
-        self.fc6 = self._conv_layer(feat, 4, 256, is_training, "fc6", bn=True, relu=False, pad='VALID', norm='cosine', reg=False, orth=True)        
+        self.fc6 = self._conv_layer(feat, 4, 256, is_training, "fc6", bn=False, relu=False, pad='VALID', norm='cosine', reg=False, orth=True)        
 
         self.score = self._conv_layer(self.fc6, 1, n_class, is_training, "score", bn=False, relu=False, pad='VALID', norm='none', reg=True, orth=False, w_norm='none')