Vishakaraj's picture
Upload folder using huggingface_hub
c709b60
raw
history blame contribute delete
No virus
166 Bytes
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"