hubconf.py 1.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
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
M
MegEngine Team 已提交
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
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,
)
30 31 32
from official.vision.detection.configs import (
    faster_rcnn_res50_coco_1x_800size,
    faster_rcnn_res50_coco_1x_800size_syncbn,
33
    retinanet_res50_coco_1x_800size,
34 35
    retinanet_res50_coco_1x_800size_syncbn,
)
36
from official.vision.detection.models import FasterRCNN, RetinaNet
37 38
from official.vision.detection.tools.utils import DetEvaluator
from official.vision.keypoints.inference import KeypointEvaluator
39
from official.vision.keypoints.models import (
40
    mspn_4stage,
41 42 43
    simplebaseline_res50,
    simplebaseline_res101,
    simplebaseline_res152,
44
)
45 46 47 48
from official.vision.segmentation.deeplabv3plus import (
    DeepLabV3Plus,
    deeplabv3plus_res101,
)