cerebras.modelzoo.config_manager.config_loader.flatten_optimizer_params

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

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

previous

cerebras.modelzoo.config_manager.config_loader.flatten_data_params

next

cerebras.modelzoo.config_manager.config_loader.flatten_sparsity_params