-
Notifications
You must be signed in to change notification settings - Fork 99
/
hubconf.py
104 lines (104 loc) · 3.03 KB
/
hubconf.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
from official.multimodal.big_sleep import BigGAN, Imagine, biggan_128, biggan_256, biggan_512
from official.multimodal.clip.inference_utils import ClipInferenceUtils
from official.multimodal.clip.models import (
rn50,
rn50x4,
rn50x16,
rn50x64,
rn101,
vit_b_16,
vit_b_32,
vit_l_14,
vit_l_14_336px,
)
from official.multimodal.dalle import (
Generator,
OpenAIDiscreteVAE,
OpenAIDiscreteVAEDecoder,
OpenAIDiscreteVAEEncoder,
VQGanVAE,
coco_512_16_16d_16h_80tsl,
openai_discrete_VAE_decoder,
openai_discrete_VAE_encoder,
vqgan_vae_1024,
)
from official.multimodal.taming_transformer import (
ConditionalSampler,
FastSampler,
Reconstruction,
celebahq_transformer,
drin_transformer,
s_flckr_transformer,
vqgan_gumbel_f8,
vqgan_imagenet_f16_1024,
vqgan_imagenet_f16_16384,
)
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 (
atss_res18_coco_3x_800size,
atss_res34_coco_3x_800size,
atss_res50_coco_3x_800size,
atss_res101_coco_3x_800size,
atss_resx101_coco_2x_800size,
faster_rcnn_res18_coco_3x_800size,
faster_rcnn_res34_coco_3x_800size,
faster_rcnn_res50_coco_3x_800size,
faster_rcnn_res101_coco_3x_800size,
faster_rcnn_resx101_coco_2x_800size,
fcos_res18_coco_3x_800size,
fcos_res34_coco_3x_800size,
fcos_res50_coco_3x_800size,
fcos_res101_coco_3x_800size,
fcos_resx101_coco_2x_800size,
freeanchor_res18_coco_3x_800size,
freeanchor_res34_coco_3x_800size,
freeanchor_res50_coco_3x_800size,
freeanchor_res101_coco_3x_800size,
freeanchor_resx101_coco_2x_800size,
retinanet_res18_coco_3x_800size,
retinanet_res34_coco_3x_800size,
retinanet_res50_coco_3x_800size,
retinanet_res101_coco_3x_800size,
retinanet_resx101_coco_2x_800size,
)
from official.vision.detection.models import ATSS, FCOS, FasterRCNN, FreeAnchor, RetinaNet
from official.vision.detection.tools.utils import DetEvaluator
from official.vision.keypoints.inference import KeypointEvaluator
from official.vision.keypoints.models import (
simplebaseline_res50,
simplebaseline_res101,
simplebaseline_res152,
)
from official.vision.segmentation.configs import (
deeplabv3plus_res101_cityscapes_768size,
deeplabv3plus_res101_voc_512size,
)
from official.vision.segmentation.models import DeepLabV3Plus