mask_rcnn_SyncBNhead.py 166 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
from .mask_rcnn_BNhead import model, dataloader, lr_multiplier, optimizer, train

model.roi_heads.box_head.conv_norm = model.roi_heads.mask_head.conv_norm = "SyncBN"