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, retinanet_res50_coco_1x_800size, retinanet_res50_coco_1x_800size_syncbn, ) 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, )