| 12345678910111213141516 |
- # -------------------------------------------------------------------------
- # Copyright (c) 2021-2022, NVIDIA Corporation & Affiliates. All rights reserved.
- #
- # This work is made available under the Nvidia Source Code License.
- # To view a copy of this license, visit
- # https://github.com/NVlabs/GroupViT/blob/main/LICENSE
- #
- # Written by Jiarui Xu
- # -------------------------------------------------------------------------
- from .builder import build_model
- from .group_vit import GroupViT
- from .multi_label_contrastive import MultiLabelContrastive
- from .transformer import TextTransformer
- __all__ = ['build_model', 'MultiLabelContrastive', 'GroupViT', 'TextTransformer']
|
