get_parameter_group_ids

lib.training.optimizer.get_parameter_group_ids(trainable_variables: list[Variable]) → dict[int, T.Literal['decay', 'no_decay']]

Obtain the index of each item in the keras model’s trainable weights that belong to each of the optimizer’s parameter groups (ie split by weights that take decay and don’t take decay)

Parameters:

trainable_variables (list[Variable]) – list of trainable variables from keras model

Return type:

dictionary of keras model’s trainable weight index to the name of the parameter group