Mmdetection: frozen BatchNorm

Created on 15 Nov 2018  路  3Comments  路  Source: open-mmlab/mmdetection

mmdetection use common BatchNorm without freeze?
In SSD series batchNorm is unfrozen
I detectron and other faster RCNN batchNorm is frozen.

I want to train with BN and big batch size 4*16. Do I need change something?

PS: this trick is used in SNIPER and in competion WIDER winner.

question

Most helpful comment

You can set the arguments bn_eval and bn_frozen of backbone field in the config file to False, then parameters of BN layers will be updated during training.

All 3 comments

You can set the arguments bn_eval and bn_frozen of backbone field in the config file to False, then parameters of BN layers will be updated during training.

@hellock Thank you!
Batch Norm train with "synchronize BN"?

Synchronized BN is currently not supported and we have the plan to add it later. If you set bn_eval and bn_frozen as False, the normal BN is used.

Was this page helpful?
0 / 5 - 0 ratings