Missed TensorFlow World? Check out the recap. Learn more

tff.learning.framework.enhance

View source on GitHub

Wraps a tff.learning.Model as an EnhancedModel.

tff.learning.framework.enhance(model)

Args:

Returns:

An EnhancedModel or TrainableEnhancedModel, depending on the type of the input model. If model has already been wrapped as such, this is a no-op.