File size: 958 Bytes
56ef371
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from . import models
from .builder import (
    ARCHITECTURES,
    BACKBONES,
    DECODERS,
    ENCODERS,
    POS_EMBEDDINGS,
    build_architecture,
    build_backbone,
    build_decoder,
    build_encoder,
    build_position_embedding,
)
from .inference_wrapper import UPNWrapper
from .models.architecture import *
from .models.backbone import *
from .models.decoder import *
from .models.encoder import *
from .models.module import *
from .models.utils import *

__all__ = [
    "BACKBONES",
    "POS_EMBEDDINGS",
    "ENCODERS",
    "DECODERS",
    "ARCHITECTURES",
    "build_backbone",
    "build_position_embedding",
    "build_encoder",
    "build_decoder",
    "build_architecture",
    "UPNWrapper",
]

__all__ += (
    models.module.__all__
    + models.utils.__all__
    + models.architecture.__all__
    + models.backbone.__all__
    + models.encoder.__all__
    + models.decoder.__all__
    + models.module.__all__
    + models.utils.__all__
)