cerebras.modelzoo.common.utils.model.mup_utils.LRAdjustmentGroup#
- class cerebras.modelzoo.common.utils.model.mup_utils.LRAdjustmentGroup(param_filter, scale=1.0)[source]#
Bases:
object
Stores data for a group of params that share a learning rate scalar. Stores a callable that returns True if a given model param corresponds to the group. Additionally, it stores the scale that should be applied to the LR of the model params that correspond to the group.
param_filter: A string or a list of strings that contains glob expressions used to match whether a given model param name belongs to the group.
scale: The scale that should be applied to the LR of this group
Methods
set_scale