model/__init__.py

import torch

from .relpose import RelPose
from .pl_trainer import PL_RelPose, keypoint_dict