models/__init__.py

from .appearance_encoder import AppearanceEncoder
from .decoder import Decoder
from .metrics import build_metric
from .pose_encoder import PoseEncoder
from .swin_transformer import build_backbone
from .unet import UNet
from .vae import VariationalAutoencoder