cerebras.modelzoo.common.optim_utils#
Helper functions related to configuration of the Optimizer.
Functions
Groups the optimizer parameters into non-overlapping groups. |
|
Config classes package optimizer related params in a sub dict. |
|
Splits a single param group into multiple param groups based on glob patterns found in optimizer_params. |