def METHOD_NAME(self, config) -> NDArrays:
    """Returns weight from a given model. If no model is passed, then a
    local model is created. This can be used to initialize a model in the
    server.
    Returns:
        NDArrays: weights from the model.
    """
    net = get_cifar_model(self.num_classes)
    weights = [val.cpu().numpy() for _, val in net.state_dict().items()]
    return weights