def model_builder():
"""Create a keras model based on the original FedAvg CNN."""
return emnist_models.create_original_fedavg_cnn_model(
only_digits=FLAGS.only_digits)
def model_builder():
"""Create compiled keras model."""
model = emnist_models.create_original_fedavg_cnn_model(
only_digits=FLAGS.digit_only_emnist)
return model
