subtle change to resnet.py

jwyang · Jan 14, 2018 · 76dc298 · 76dc298
1 parent ef7e458
commit 76dc298
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/model/fpn/resnet.py b/lib/model/fpn/resnet.py
@@ -298,7 +298,7 @@ def train(self, mode=True):
       self.RCNN_layer0.eval()
       self.RCNN_layer1.eval()
       self.RCNN_layer2.train()
-      self.RCNN_layer3.train()      
+      self.RCNN_layer3.train()
       self.RCNN_layer4.train()
 
       self.RCNN_smooth1.train()
@@ -307,10 +307,10 @@ def train(self, mode=True):
 
       self.RCNN_latlayer1.train()
       self.RCNN_latlayer2.train()
-      self.RCNN_latlayer3.train()      
+      self.RCNN_latlayer3.train()
 
       self.RCNN_toplayer.train()
-      
+
       def set_bn_eval(m):
         classname = m.__class__.__name__
         if classname.find('BatchNorm') != -1: