Spaces:
Running
on
Zero
Running
on
Zero
StableVITON
/
preprocess
/detectron2
/projects
/Rethinking-BatchNorm
/configs
/mask_rcnn_SyncBNhead.py
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" | |