Spaces:
Running
on
Zero
Running
on
Zero
File size: 166 Bytes
61c2d32 |
1 2 3 4 |
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"
|