Skip to content

Commit 76dc298

Browse files
author
Jianwei Yang
committed
subtle change to resnet.py
1 parent ef7e458 commit 76dc298

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lib/model/fpn/resnet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,7 @@ def train(self, mode=True):
298298
self.RCNN_layer0.eval()
299299
self.RCNN_layer1.eval()
300300
self.RCNN_layer2.train()
301-
self.RCNN_layer3.train()
301+
self.RCNN_layer3.train()
302302
self.RCNN_layer4.train()
303303

304304
self.RCNN_smooth1.train()
@@ -307,10 +307,10 @@ def train(self, mode=True):
307307

308308
self.RCNN_latlayer1.train()
309309
self.RCNN_latlayer2.train()
310-
self.RCNN_latlayer3.train()
310+
self.RCNN_latlayer3.train()
311311

312312
self.RCNN_toplayer.train()
313-
313+
314314
def set_bn_eval(m):
315315
classname = m.__class__.__name__
316316
if classname.find('BatchNorm') != -1:

0 commit comments

Comments
 (0)