def METHOD_NAME(self) -> None:
    """Initialize layers of the transformer head."""
    # cls branch
    self.fc_cls = Linear(self.embed_dims, self.cls_out_channels)
    # reg branch
    self.fc_reg = MLP(self.embed_dims, self.embed_dims, 4, 3)