def __init__(self, backbone, head, loss, post_process, flip_test):
        """
        VitPose network, see https://arxiv.org/pdf/2204.12484v2.pdf

        Args:
            backbone (nn.Layer): backbone instance
            post_process (object): `HRNetPostProcess` instance
            
        """
        super(VitPose_TopDown, self).__init__()
        self.backbone = backbone
        self.head = head
        self.loss = loss
        self.post_process = post_process
        self.flip_test = flip_test

        VitPose network, see https://arxiv.org/pdf/2204.12484v2.pdf

        Args:
            backbone (nn.Layer): backbone instance
            post_process (object): `HRNetPostProcess` instance
            
        
ppdet/modeling/architectures/keypoint_vitpose.py