Have a question? Connect with the community at the TensorFlow Forum Visit Forum

ParameterGroupOptimizerBuilder

public struct ParameterGroupOptimizerBuilder

Builds a ParameterGroupOptimizer. This is used at essentially the level of a single weight in the model. A mapping from parameter groups selected by ([Bool] to ParameterGroupOptimizer) defines the final optimizer.