models/encoder_decoder/__init__.py

from .vgg import vgg11 as vgg11_ae, vgg11_bn as vgg11_bn_ae
from .vgg import vgg13 as vgg13_ae, vgg13_bn as vgg13_bn_ae
from .vgg import vgg16 as vgg16_ae, vgg16_bn as vgg16_bn_ae
from .vgg import vgg19 as vgg19_ae, vgg19_bn as vgg19_bn_ae
from .resnet import resnet18 as resnet18_ae, resnet34 as resnet34_ae
from .resnet import resnet50 as resnet50_ae, resnet101 as resnet101_ae, resnet152 as resnet152_ae

from .cannet import cannet, cannet_bn
from .csrnet import csrnet, csrnet_bn


__all__ = [
    "vgg11_ae", "vgg11_bn_ae", "vgg13_ae", "vgg13_bn_ae", "vgg16_ae", "vgg16_bn_ae", "vgg19_ae", "vgg19_bn_ae",
    "resnet18_ae", "resnet34_ae", "resnet50_ae", "resnet101_ae", "resnet152_ae",
    "cannet", "cannet_bn",
    "csrnet", "csrnet_bn",
]