cerebras.modelzoo.common.optim_utils

Helper functions related to configuration of the Optimizer.

Functions

configure_param_groups	Groups the optimizer parameters into non-overlapping groups.
flatten_optimizer_params	Config classes package optimizer related params in a sub dict.
partition_params_group_with_tags	Splits a single param group into multiple param groups based on glob patterns found in optimizer_params.

previous

cerebras.modelzoo.common.model_utils.count_lines.count_lines

next

cerebras.modelzoo.common.optim_utils.configure_param_groups