Watch talks from the 2019 TensorFlow Dev Summit Watch now

tff.learning.framework.EnhancedModel

Class EnhancedModel

Inherits From: Model

Defined in learning/model_utils.py.

A wrapper around a Model that adds sanity checking and metadata helpers.

__init__

__init__(model)

Properties

federated_output_computation

input_spec

local_variables

non_trainable_variables

trainable_variables

weights

Returns a tff.learning.ModelWeights.

Methods

forward_pass

forward_pass(
    batch_input,
    training=True
)

report_local_outputs

report_local_outputs()