![]() |
Creates a tff.Type
from a tff.learning.Model
or callable that constructs a model.
tff.learning.framework.weights_type_from_model(
model: Union[model_lib.Model, Callable[[], model_lib.Model]]
) -> tff.types.StructType
Used in the notebooks
Used in the tutorials |
---|
Args | |
---|---|
model
|
A tff.learning.Model instance, or a no-arg callable that returns a
model.
|
Returns | |
---|---|
A tff.StructType representing the TFF type of the ModelWeights
structure for model .
|