OVSegmentor/models/__init__.py

# -------------------------------------------------------------------------
# Copyright (c) 2021-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
#
# NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual
# property and proprietary rights in and to this software, related
# documentation and any modifications thereto.  Any use, reproduction,
# disclosure or distribution of this software and related documentation
# without an express license agreement from NVIDIA CORPORATION is strictly
# prohibited.
#
# Written by Jiarui Xu
# -------------------------------------------------------------------------
# Written by Jilan Xu
# -------------------------------------------------------------------------

from .builder import build_model
from .group_vit import GroupViT
from .multi_label_contrastive import MultiLabelContrastive
from .transformer import TextTransformer
from .transformer import DistilBert, Bert, BertMedium, Roberta

__all__ = ['build_model', 'MultiLabelContrastive', 'GroupViT', 'TextTransformer', \
           'DistilBert', 'Bert', 'BertMedium','Roberta',]