Watch talks from the 2019 TensorFlow Dev Summit Watch now

tff.learning.framework.EnhancedTrainableModel

Class EnhancedTrainableModel

Inherits From: EnhancedModel, TrainableModel

Defined in learning/model_utils.py.

__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()

train_on_batch

train_on_batch(batch_input)