我想请问下面这种冻结方式
if cfg.MODEL.BACKBONE.FREEZE:
for p in self.backbone.parameters():
p.requires_grad = False
print('froze backbone parameters')
if cfg.MODEL.BACKBONE.FREEZE_P5:
for connection in [self.backbone.fpn_lateral5, self.backbone.fpn_output5]:
for p in connection.parameters():
p.requires_grad = False
print('frozen P5 in FPN')
是不是已经覆盖了
if freeze_at >= 1:
for p in stem.parameters():
p.requires_grad = False
stem = FrozenBatchNorm2d.convert_frozen_batchnorm(stem)
if freeze_at >= stage_idx:
for block in blocks:
block.freeze()
stages.append(blocks)
这种冻结方式?
这两种冻结方式有什么不同呢?
我想请问下面这种冻结方式 if cfg.MODEL.BACKBONE.FREEZE: for p in self.backbone.parameters(): p.requires_grad = False print('froze backbone parameters')
是不是已经覆盖了 if freeze_at >= 1: for p in stem.parameters(): p.requires_grad = False stem = FrozenBatchNorm2d.convert_frozen_batchnorm(stem) if freeze_at >= stage_idx: for block in blocks: block.freeze() stages.append(blocks) 这种冻结方式? 这两种冻结方式有什么不同呢?