cerebras.modelzoo.config_manager.config_loader.flatten_data_params#

cerebras.modelzoo.config_manager.config_loader.flatten_data_params(kwargs)[source]#

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

Returns

Flattened dict

Return type

flattened_args

previous

cerebras.modelzoo.config_manager.config_loader.create_config_obj_from_params

next

cerebras.modelzoo.config_manager.config_loader.flatten_optimizer_params