Skip to content

GitLab

Commit 81558853 authored by ThangVu's avatar ThangVu
Browse files

minor

parent bee93859
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet/models/backbones/resnet.py
View file @ 81558853
...@@ -344,7 +344,7 @@ class ResNet(nn.Module): ...@@ -344,7 +344,7 @@ class ResNet(nn.Module):
for m in self.modules(): for m in self.modules():
if isinstance(m, nn.Conv2d): if isinstance(m, nn.Conv2d):
kaiming_init(m) kaiming_init(m)
elif isinstance(m, (nn.BatchNorm, nn.GroupNorm)): elif isinstance(m, (nn.BatchNorm2d, nn.GroupNorm)):
constant_init(m, 1) constant_init(m, 1)
# zero init for last norm layer https://arxiv.org/abs/1706.02677 # zero init for last norm layer https://arxiv.org/abs/1706.02677
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment