def __init__(
            self,
            num_joints,
            backbone='HRNet',
            trans_encoder='',
            loss='Pose3DLoss', ):
        """
        Modified from METRO network, see https://arxiv.org/abs/2012.09760

        Args:
            backbone (nn.Layer): backbone instance
        """
        super(METRO_Body, self).__init__()
        self.num_joints = num_joints
        self.backbone = backbone
        self.loss = loss
        self.deploy = False

        self.trans_encoder = trans_encoder
        self.conv_learn_tokens = paddle.nn.Conv1D(49, num_joints + 10, 1)
        self.cam_param_fc = paddle.nn.Linear(3, 2)

        Modified from METRO network, see https://arxiv.org/abs/2012.09760

        Args:
            backbone (nn.Layer): backbone instance
        
ppdet/modeling/architectures/pose3d_metro.py