|View source on GitHub|
Libraries for working with models in federated learning algorithms.
class FunctionalModel: A model that parameterizes forward pass by model weights.
class ModelWeights: A container for the trainable and non-trainable variables of a
load_functional_model(...): Deserializes a TensorFlow SavedModel at
path to a functional model.
model as a TensorFlow SavedModel to
save_functional_model(...): Serializes a
FunctionalModel as a