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*?
in FCOS.Pytorch/model/fcos.py
Since module.eval() has frozen bn, why you additionally set
p.requires_grad=False? Is there another module called BatchNorm*?