cerebras.modelzoo.config_manager.config_loader#

Utility to load the config from yaml or .py config file

Functions

convert_to_dict

Utility to convert config object to dict

create_config_obj_from_params

Convert given dictionary to a config object based on the model name.

flatten_data_params

Config classes package data related params in a sub dict.

flatten_optimizer_params

Config classes package optimizer related params in a sub dict.

flatten_sparsity_params

Config classes package sparsity related params in a sub dict.