cerebras.modelzoo.config_manager.config_loader.flatten_sparsity_params

cerebras.modelzoo.config_manager.config_loader.flatten_sparsity_params(kwargs)

Config classes package sparsity related params in a sub dict. ALthough, if we use native yaml config, they come unrolled. This utility unwraps the sparsity related params(if present) into an unroller sparsity param dict for consistency. :param kwargs: Input args

Returns

Flattened dict

previous

cerebras.modelzoo.config_manager.config_loader.flatten_optimizer_params

next

cerebras.modelzoo.config_manager.config_loader.get_config_from_class