Function get_model

examples/hfl/src/client.py:194–199

(model, num_channels=None, num_classes=None, data_size=None)

def get_model(model, num_channels=None, num_classes=None, data_size=None):
    if model == "mlp":
        model = mlp.create_model(data_size=data_size, num_classes=num_classes)
    else:
        raise NotImplementedError
    return model


def run(