File size: 278 Bytes
ea61c61 |
1 2 3 4 5 6 7 |
from transformers import AutoConfig, AutoModel
from .configuration_maskrcnn import MaskRCNNConfig
from .modeling_maskrcnn import MaskRCNNForInstanceSegmentation
AutoConfig.register("maskrcnn", MaskRCNNConfig)
AutoModel.register(MaskRCNNConfig, MaskRCNNForInstanceSegmentation)
|