Add support for weight standardization (#521)
* add support for weight standardization * add ws support for htc heads * add a config file for 20-23-24e lr schedule * update readme of weight standardization
Showing
- README.md 1 addition, 0 deletionsREADME.md
- configs/gn+ws/README.md 54 additions, 0 deletionsconfigs/gn+ws/README.md
- configs/gn+ws/faster_rcnn_r50_fpn_gn_ws_1x.py 165 additions, 0 deletionsconfigs/gn+ws/faster_rcnn_r50_fpn_gn_ws_1x.py
- configs/gn+ws/mask_rcnn_r50_fpn_gn_ws_20_23_24e.py 180 additions, 0 deletionsconfigs/gn+ws/mask_rcnn_r50_fpn_gn_ws_20_23_24e.py
- configs/gn+ws/mask_rcnn_r50_fpn_gn_ws_2x.py 180 additions, 0 deletionsconfigs/gn+ws/mask_rcnn_r50_fpn_gn_ws_2x.py
- configs/gn+ws/mask_rcnn_x101_32x4d_fpn_gn_ws_2x.py 182 additions, 0 deletionsconfigs/gn+ws/mask_rcnn_x101_32x4d_fpn_gn_ws_2x.py
- mmdet/models/backbones/resnet.py 49 additions, 22 deletionsmmdet/models/backbones/resnet.py
- mmdet/models/backbones/resnext.py 18 additions, 6 deletionsmmdet/models/backbones/resnext.py
- mmdet/models/bbox_heads/convfc_bbox_head.py 3 additions, 0 deletionsmmdet/models/bbox_heads/convfc_bbox_head.py
- mmdet/models/mask_heads/fcn_mask_head.py 3 additions, 0 deletionsmmdet/models/mask_heads/fcn_mask_head.py
- mmdet/models/mask_heads/fused_semantic_head.py 5 additions, 0 deletionsmmdet/models/mask_heads/fused_semantic_head.py
- mmdet/models/mask_heads/htc_mask_head.py 1 addition, 0 deletionsmmdet/models/mask_heads/htc_mask_head.py
- mmdet/models/necks/fpn.py 3 additions, 0 deletionsmmdet/models/necks/fpn.py
- mmdet/models/utils/__init__.py 5 additions, 3 deletionsmmdet/models/utils/__init__.py
- mmdet/models/utils/conv_module.py 40 additions, 1 deletionmmdet/models/utils/conv_module.py
- mmdet/models/utils/conv_ws.py 46 additions, 0 deletionsmmdet/models/utils/conv_ws.py
Loading
Please register or sign in to comment