From c899cdf1e4971ede74686315b80782ee671ac330 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Sun, 28 Apr 2019 19:28:44 -0700
Subject: [PATCH] bug fix for freezing parameters
---
mmdet/models/backbones/resnet.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/mmdet/models/backbones/resnet.py b/mmdet/models/backbones/resnet.py
index ba7e5fe..7553d57 100644
--- a/mmdet/models/backbones/resnet.py
+++ b/mmdet/models/backbones/resnet.py
@@ -421,12 +421,14 @@ class ResNet(nn.Module):
def _freeze_stages(self):
if self.frozen_stages >= 0:
+ self.norm1.eval()
for m in [self.conv1, self.norm1]:
for param in m.parameters():
param.requires_grad = False
for i in range(1, self.frozen_stages + 1):
m = getattr(self, 'layer{}'.format(i))
+ m.eval()
for param in m.parameters():
param.requires_grad = False
--
GitLab