def METHOD_NAME(self, search_space, dataset, scope=None):
    """
    Same as in darts with a different mixop.
    If you want to checkpoint the dirichlet 'concentration' parameter (beta) add it to the buffer here.
    """
    super().METHOD_NAME(search_space, dataset, scope)
    self.anchor = Dirichlet(
        torch.ones_like(
            torch.nn.utils.parameters_to_vector(self.architectural_weights)
        ).to(self.device)
    )