ParameterGroupOptimizerBuilder

Stay organized with collections Save and categorize content based on your preferences.
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.