cerebras.modelzoo.common.optim_utils.flatten_optimizer_params#
- cerebras.modelzoo.common.optim_utils.flatten_optimizer_params(kwargs)[source]#
Config classes package optimizer related params in a sub dict. ALthough, if we use native yaml config, they come unrolled. This utility unwraps the optimizer related params(if present) into an unroller optimizer param dict for consistency. :param kwargs: Input args dict
- Returns
Flattened dict
- Return type
flattened_args