hubconf.py

from official.nlp.bert.model import (
    cased_L_12_H_768_A_12,
    cased_L_24_H_1024_A_16,
    chinese_L_12_H_768_A_12,
    multi_cased_L_12_H_768_A_12,
    uncased_L_12_H_768_A_12,
    uncased_L_24_H_1024_A_16,
    wwm_cased_L_24_H_1024_A_16,
    wwm_uncased_L_24_H_1024_A_16,
)
from official.quantization.models import quantized_resnet18
from official.vision.classification.resnet.model import (
    BasicBlock,
    Bottleneck,
    ResNet,
    resnet18,
    resnet34,
    resnet50,
    resnet101,
    resnet152,
    resnext50_32x4d,
    resnext101_32x8d,
)
from official.vision.classification.shufflenet.model import (
    shufflenet_v2_x0_5,
    shufflenet_v2_x1_0,
    shufflenet_v2_x1_5,
    shufflenet_v2_x2_0,
)
from official.vision.detection.configs import (
    faster_rcnn_res50_coco_1x_800size,
    faster_rcnn_res50_coco_1x_800size_syncbn,
    faster_rcnn_res101_coco_2x_800size,
    faster_rcnn_resx101_coco_2x_800size,
    retinanet_res50_coco_1x_800size,
    retinanet_res50_coco_1x_800size_syncbn,
    retinanet_res101_coco_2x_800size,
    retinanet_resx101_coco_2x_800size,
)
from official.vision.detection.models import FasterRCNN, RetinaNet
from official.vision.detection.tools.utils import DetEvaluator
from official.vision.keypoints.inference import KeypointEvaluator
from official.vision.keypoints.models import (
    mspn_4stage,
    simplebaseline_res50,
    simplebaseline_res101,
    simplebaseline_res152,
)
from official.vision.segmentation.deeplabv3plus import (
    DeepLabV3Plus,
    deeplabv3plus_res101,
)