¡Google I / O regresa del 18 al 20 de mayo! Reserva espacio y crea tu horario Regístrate ahora

tff.learning.ModelWeights

A container for the trainable and non-trainable variables of a Model.

Note this does not include the model's local variables.

It may also be used to hold other values that are parallel to these variables, e.g., tensors corresponding to variable values, or updates to model variables.

Methods

assign_weights_to

View source

Assign these TFF model weights to the weights of a model.

Args
model a tf.keras.Model or tff.learning.Model instance to assign the weights to.

from_model

View source

from_tff_result

View source

non_trainable Instance of member_descriptor
trainable Instance of member_descriptor