-
Notifications
You must be signed in to change notification settings - Fork 24
Open
Description
in FCOS.Pytorch/model/fcos.py
def freeze_bn(module):
if isinstance(module,nn.BatchNorm2d):
module.eval()
classname = module.__class__.__name__
if classname.find('BatchNorm') != -1:
for p in module.parameters(): p.requires_grad=False
Since module.eval() has frozen bn, why you additionally set p.requires_grad=False? Is there another module called BatchNorm*?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels