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.

previous

cerebras.modelzoo.config_manager.config_classes.base.sparsity_config.SparsityBaseConfig

next

cerebras.modelzoo.config_manager.config_loader.convert_to_dict